In This Package:

RootIOUserDataProxy.cc

Go to the documentation of this file.

#include "RootIOSvc/RootIOUserDataProxy.h"

#include "TTree.h"
#include <vector>

using namespace std;

RootIOUserDataProxy::RootIOUserDataProxy(const std::string& varName, 
                                         const std::string& cppType,
                                         char* address)
  : m_varName(varName),
    m_cppType(cppType),
    m_address(address)
{
    // If ccpType is not one of these then it is assumed to be an object
    const char* knownPOD[] = { "int", "float", 0 };
    const char* rootTypes[] = { "Int_t", "Float_t", 0 };

    m_isObject = true;
    for (int ind=0; knownPOD[ind]; ++ind) {
        if (m_cppType == knownPOD[ind]) {
            m_isObject = false;
            break;
        }
        if (m_cppType == rootTypes[ind]) {
            m_cppType = knownPOD[ind];
            m_isObject = false;
            break;
        }
    }
}

RootIOUserDataProxy::~RootIOUserDataProxy()
{
}
    

bool RootIOUserDataProxy::branch(TTree* tree)
{
    // handle known object types needing template to trigger write Branch() call
    if (m_cppType == "int") {
        tree->Branch(m_varName.c_str(),address<void>(),"Int_t/I");
        return true;
    }

    if (m_cppType == "float") {
        tree->Branch(m_varName.c_str(),address<void>(),"Float_t/F");
        return true;
    }

    if (m_cppType == "vector<int>") {
        tree->Branch(m_varName.c_str(), address< std::vector<int> >());
        return true;
    }
    
    if (m_cppType == "vector<float>") {
        tree->Branch(m_varName.c_str(), address< std::vector<float> >());
        return true;
    }
    
    return false;
}

---