LRootGlobalReader.cc

Go to the documentation of this file.

#include "LRootGlobalReader.hh"
#include "QObject.hh"
#include "TH1I.h"
#include "TH1D.h"
#include "TH1F.h"
#include "TObjString.h"
#include "QDianaR.hh"
#include "TROOT.h"
 
REGISTER_GLOBAL_READER(LRootGlobalReader,".root")
 
using namespace Diana;
 
 
LRootGlobalReader::LRootGlobalReader() : QGlobalReader("RootGlobalReader")
{
    fQObjectCache = new std::map<std::string, QObject*>;
    fFile = NULL;
}
 
LRootGlobalReader::~LRootGlobalReader()
{
    delete fQObjectCache;
}
 
QError LRootGlobalReader::Open(const std::string& filename, const std::string& opt)
{
    QError err = QERR_SUCCESS;
    if (!fFile) fFile = new TFile(filename.c_str(),"READ"); 
    if(fFile->IsZombie()) {
        err = QERR_CANNOT_OPEN_FILE;
        err.SetDescription("Cannot open " + filename);
        delete fFile;
        fFile = NULL;
    }
    else if(fFile->TestBit(TFile::kRecovered)) {
        err = QERR_CANNOT_OPEN_FILE;
        err.SetDescription("File " + filename + " not closed");
        delete fFile;
        fFile = NULL;
    }
    return err;
}
 
 
QError LRootGlobalReader::Close()
{
    QError err = QERR_SUCCESS;
    if(fFile){
        // MV FIXME this check if for cint. When we do ".q" TFiles are automatically deleted
        // by root so that this delete results in a double free.
        if(gROOT->GetListOfFiles()->GetSize() > 0) delete fFile;
    }
    fFile = NULL;
    std::map<std::string, QObject*>::iterator iter;
    for(iter = fQObjectCache->begin(); iter != fQObjectCache->end(); iter++) 
        delete iter->second;
 
    return err;
}
 
bool LRootGlobalReader::ReadingIsSafe() const
{
    if(fFile != NULL) return true;
    return false;
}
 
std::string LRootGlobalReader::GetFullPath(const std::string& name) const
{
    std::string fullpath = std::string(ROOT_GLOBAL_DIR) + "/" + name;
    return fullpath; 
}
 
const QObject* LRootGlobalReader::GetQObject(const std::string& name, const std::string& className, QError& err) const 
{
    // check for already existing object
    err = QERR_SUCCESS;
    std::map<std::string, QObject*>::const_iterator object = fQObjectCache->find(name);
    if(object != fQObjectCache->end()) return object->second;
    // get object from root file
    TObject* tobj = NULL;
    if(ReadingIsSafe()) fFile->GetObject(GetFullPath(name).c_str(),tobj);
    QObject* obj = dynamic_cast<QObject*>(tobj);
    if(obj) (*fQObjectCache)[name] = obj;
    if( obj == NULL) {
        err = QERR_UNKNOWN_ERR;
        err.SetDescription(GetName(),__LINE__,std::string("Could not find object \"")+name+"\"");
    }
    return obj;
}