JRootTypewriter.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
 
#include "TObject.h"
#include "TFile.h"
#include "TError.h"
#include "TKey.h"
#include "TClass.h"
#include "TTree.h"
#include "TString.h"
#include "TRegexp.h"
 
#include "JROOT/JRootClass.hh"
#include "JROOT/JRootTypewriter.hh"
 
#include "JSupport/JLimit.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * Example application to test printing of objects using ROOT dictionary.
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  string    inputFile;
  string    target;
  JLimit    numberOfEvents;
  int       debug;
 
  try {
 
    JParser<> zap;
    
    zap['f'] = make_field(inputFile);
    zap['T'] = make_field(target,         "name of target to be read");
    zap['n'] = make_field(numberOfEvents, "number of events, if applicable")   =  1;
    zap['d'] = make_field(debug)      = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
  gErrorIgnoreLevel = kError;
 
 
  TFile* in = TFile::Open(inputFile.c_str());
 
  if (in == NULL || !in->IsOpen()) {
    FATAL("Error opening file " << inputFile << endl);
  }
 
  const TRegexp regexp(target.c_str());
 
  JRootTypewriter printer;
 
  TIter iter(in->GetListOfKeys());
 
  for (TKey* key; (key = (TKey*) iter.Next()) != NULL; ) {
 
    const TString tag(key->GetName());
 
    DEBUG("Key: " << tag << " match = " << tag.Contains(regexp) << endl);
 
    // option match
 
    if (tag.Contains(regexp) && TClass::GetClass(key->GetClassName())->IsTObject()) {
 
      TClass* cs = TClass::GetClass(key->GetClassName());
 
      if (cs != NULL) {
 
        DEBUG("TClass " << cs->GetName() << endl);
        DEBUG("TKey::ReadObjectAny(..)"  << endl);
 
        void* ps = key->ReadObjectAny(cs);
 
        if (cs == TTree::Class()) {
 
          TTree* ts = (TTree*) ps;
 
          DEBUG("TTree " << ts->GetName() << endl);
 
          vector<JBranchClass> ls;
 
          for (TIter next(ts->GetListOfBranches()); TBranch* p1 = dynamic_cast<TBranch*>(next()); ) {
            ls.push_back(JBranchClass(p1));
          }
 
          // set branch addresses afterwards
 
          for (auto& tp : ls) {
            ts->GetBranch(tp.getName())->SetAddress(tp.getAddress());
          }
 
          for (Long64_t i = numberOfEvents.getLowerLimit(); i < ts->GetEntries() && i != numberOfEvents.getUpperLimit(); ++i) {
 
            cout << ts->GetName() << "[" << i << "]:" << endl;
 
            DEBUG("TTree::GetEvent(" << i << ")" << endl);
 
            ts->GetEvent(i);
 
            for (auto& tp : ls) {
              printer(cout,tp);
            }
          }
 
        } else { 
 
          printer(cout, cs->GetStreamerInfo(), (const char*) ps);
        }
      }
    }
  }
}