QRootFileReaderBase.cc

Go to the documentation of this file.

#include "QRootFileReaderBase.hh"
#include "QEvent.hh"
#include "QEventAssembler.hh"
#include "QBaseTree.hh"
#include "QObject.hh"
#include "QString.hh"
#include "QChain.hh"
#include "QTree.hh"
#include <TLeaf.h>
#include <TFile.h>
#include <TEntryList.h>
#include <TFriendElement.h>
#include "QVectorC.hh"
#include "QRawEvent.hh"
#include <sstream>
 
using namespace Diana;
 
using std::string;
 
//_____________________________________________________________________________
QRootFileReaderBase::~QRootFileReaderBase() {
  // std::map<std::string,QObject **>::iterator it;
  // for(it=fObjectList.begin();it!=fObjectList.end();it++){
  //   if(*(it->second)) delete *(it->second);
  // }
  // fObjectList.clear();
}
//_____________________________________________________________________________
void QRootFileReaderBase::LoadQChain() {
  fObjectList.clear();
 
  // init chain
  fChain = new QChain();
  std::vector<std::string>::iterator iter;
  int err;
  //const char* file0 = "";
  for(iter = fFileList.begin(); iter!= fFileList.end(); iter++) {
    Debug("Adding to chain %s", iter->c_str());
    err = fChain->Add(iter->c_str(),0);
    //if(iter == fFileList.begin()) file0 = iter->c_str();
    if(err == 0) Panic("Cannot open file %s", iter->c_str());
  }
  
  fChain->SetCacheSize(100000000);
  fChain->AddBranchToCache("*",kTRUE);  
  fEvents = fChain->GetEntries();
  fChain->SetEstimate(fEvents);
  Info("Number of events: %d", fEvents);
  
}
//_____________________________________________________________________________
const QError& QRootFileReaderBase::JumpToEvent(Long64_t event) {
  if(event > fEvents) {
    fErr = QERR_OUT_OF_RANGE;
    char buf[128];
    snprintf(buf,128,"Total events: %llu; event requested: %llu", fEvents, event);
    fErr.SetDescription(buf);
  } else {
    fErr = QERR_SUCCESS;
    fEventNumber = event;
  }
  return fErr;
}
//_____________________________________________________________________________
void QRootFileReaderBase::AddToEvent(const char* brname, QObject** evData,
                                     QEventAssembler& eva) {
  QEventLabel label(brname);
  std::string owner = label.GetOwner();
  std::string name = label.GetName();
  
  Debug("Adding \"%s\" of owned by \"%s\"", name.c_str(), owner.c_str());
  eva.Add(owner.c_str(),name.c_str(), evData);
}
//_____________________________________________________________________________
std::vector<TBranch*> QRootFileReaderBase::GetBranches(QChain* chain,
                                                       QEventAssembler& eva) {
  chain->ResetLists();
  TObjArray* branches = chain->GetListOfAllBranches();
  
  Int_t nbranches = branches->GetEntriesFast();
  std::vector<TBranch*> myBranches;
  std::stringstream msg;
  bool firstCall = false;
  if(eva.GetLabels().empty()) {
    firstCall = true;
  }
  for (Int_t br = 0; br < nbranches; br++) {
    TBranch* branch = (TBranch*) branches->UncheckedAt(br);
    const char* brname = branch->GetName();
    std::string brnamecut = brname;
    brnamecut = brnamecut.substr(0,brnamecut.size()-1);
    
    QEventLabel label = brnamecut;
    QObject** oldobj = eva.Get(label.owner.c_str(),label.name.c_str());
    bool addBranch = firstCall || (oldobj && *oldobj &&
                                   (*oldobj)->IsRequired());
    // if required add anyway
    if(addBranch) {
      myBranches.push_back(branch);
      msg<<" "<<brname;
      if(!firstCall) (*oldobj)->SetIsRequired();
    }
    
  }
  Debug("Active branches: %s",msg.str().c_str());
  return myBranches;
}
//_____________________________________________________________________________
void QRootFileReaderBase::SetBranches(QEventAssembler& eva) {
  std::vector<TBranch*> branches = GetBranches(fChain,eva);
  Int_t nbranches = branches.size();
  fChain->SetBranchStatus("*",0);
  
  for (Int_t br = 0; br < nbranches; br++) {
    TBranch* branch = branches[br];
    // assume one leaf per branch
    const char* brname = branch->GetName();
    TLeaf* leaf = branch->GetLeaf(brname);
    const char* leafType = leaf->GetTypeName();
    Debug("Found branch \"%s\" of type \"%s\"", brname, leafType);
    TClass* thisClass = TClass::GetClass(leafType,false,false);
    if(thisClass && thisClass->InheritsFrom("Diana::QObject")) {
      std::string brnamecut = brname;
      brnamecut = brnamecut.substr(0,brnamecut.size()-1);
      std::string brnamed = brnamecut + "*";
      fChain->SetBranchStatus(brnamed.c_str(),1);
      if(fObjectList.find(brname) == fObjectList.end()) {
        QObject** evData = new QObject*(0);
        //      QObject *evMem = (QObject *)thisClass.New();
        //      *evData=evMem;
        branch->SetAddress(evData);
        AddToEvent(brnamecut.c_str(),evData,eva);
        fObjectList[brname] = evData;
        (*evData)->SetIsRequired(GetBool(brnamecut,fInclusive, false ));
      } else {
        branch->SetAddress(fObjectList[brname]);
      }
    } else {
      Error("Type \"%s\" (in branch \"%s\") does not inherit from "
            "Diana::QObject: skipped",leafType,brname);
    }
  }
}
//_____________________________________________________________________________