MRootFileReader.cc

Go to the documentation of this file.

#include "MRootFileReader.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;
 
REGISTER_MODULE(MRootFileReader)
 
using std::string;
 
//_____________________________________________________________________________
// Init method is called before event loop
void MRootFileReader::Init(QEventAssembler& eva) {
 
  // Reads multiple or single files
  InitFileManager();
  fDeleteInput = GetBool("DeleteInputFiles",false,false);
  fRWCommon->fDeleteInputFiles = fDeleteInput;
  fInclusive = GetBool("InclusiveMode",true,false);
  fRWCommon->fInclusiveMode = fInclusive;
  fRootInput = NULL;
 
  // Load the chain
  LoadQChain();
 
  // read in the cut string
  fEntryList = 0;
  fEntryListEntry = 0;
  std::string cut = GetString("Cut","",false);
  if (cut != "") {
   Warn("The \"Cut\" option is not orthodox. Users must give to Marco Vignati"
         " developer 0.89 Euro for each use");
    fChain->Draw(">>anakaEventList", cut.c_str(), "entrylist");
    fEntryList = (TEntryList*) gDirectory->Get("anakaEventList");
    Long64_t nsel = fEntryList->GetN();
    if(nsel == 0) {
      Error("The Cut selected zero entries");
      return;
    }
    fEvents = fEntryList->GetN();
    fChain->SetEntryList(fEntryList);
    Info("The Cut selected %lu entries",fEvents);
    
  }    
  
  // Init branch 
  fEventNumber = 0;
  fRootInput = 0;
  fFileCounter = -1;//initialized to -1 because it should be always positive
  if(fEvents > 0 ) {
    SetBranches(eva);
    //MV FIXME: Move to QReader (see FIXME in QReader.cc:47
    //QGlobalReaderDispatcher::GetInstance().SetCurrentReaderFilename(file0);
  } else {
    Error("Zero entries!");
    return;
  }
  
  // Get Aliases
  TList* falias = fChain->GetListOfAllAliases();
  if(!falias)  {
    Debug("Found no alias in the first file");
  } else {
    TIter next(falias);
    Debug("Found %d aliases in the first file",falias->GetSize());
    while(TObject *obj = next()) {
      const char* name = obj->GetName();
      const char* path = obj->GetTitle();
      Debug("Alias %s defined in file as %s",name,path);
      eva.GetAliases().Add(path,name);
    }
  }
}
//_____________________________________________________________________________
// Doit method is called for each event, getting the event as argument
bool MRootFileReader::Do( QEventAssembler& eva) 
{
  if(fEntryList) {
    if(fEntryListEntry >= fEvents) return false;
    Int_t treenum=0;
    fEventNumber = fEntryList->GetEntryAndTree(fEntryListEntry,treenum);
    fEventNumber += fChain->GetTreeOffset()[treenum];
    fEntryListEntry++;
  } else {
    if(fEventNumber >= fEvents)
      return false;
  }
  fChain->GetEntry(fEventNumber);
  
  if(fChain->GetFile()->GetFileCounter() != fFileCounter) {
    fRootInput = fChain->GetFile();
    fFileCounter = fChain->GetFile()->GetFileCounter();
 
    // here we can set the required only branches from the event
    SetBranches(eva);
    // fChain->SetCacheSize(50000000);
    //  fChain->AddBranchToCache("*",kTRUE);
    // reget as we have reset branches
    fChain->GetEntry(fEventNumber);
    // notify open new file
    std::string fileName = fChain->GetFile()->GetName();
    fRWCommon->fReaderTreeEvents[fileName] = fChain->GetTree()->GetEntries();
    OpenNewFile(fileName);
    Info("Current file is %s", fileName.c_str());
  } 
  fEventNumber++;
  return true;
}
//_____________________________________________________________________________
// Done method is called after event loop
void MRootFileReader::Done() {   
  delete fChain;
  fRootInput = NULL;
  if(fDeleteInput) {
    std::vector<std::string>::iterator iter;
    for(iter = fFileList.begin(); iter!= fFileList.end(); iter++) {
      Info("Removing %s",iter->c_str());
      remove(iter->c_str());
    }
    if(fReadFileList) {
      remove(fFileListPath.c_str());
      Info("Removing %s", fFileListPath.c_str());
    }
  }
}
//_____________________________________________________________________________
const QError& MRootFileReader::JumpToEvent(Long64_t event) {
  if(fEntryList) {
    fErr = QERR_NOT_IMPLEMENTED;
    char buf[128];
    snprintf(buf,128,"Cut option is not compatible with JumpToEvent");
    fErr.SetDescription(buf);
  }
  else
    QRootFileReaderBase::JumpToEvent(event);
  return fErr;
}
//_____________________________________________________________________________