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 "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);
  }
 
  if (debug >= debug_t) {
    for (TIter i(in->GetListOfKeys()); TKey* key = dynamic_cast<TKey*>(i.Next()); ) {
      cout << "key: " << key->GetName() << ' ' << key->GetClassName() << endl;
    }
  }
 
  TKey* key = in->GetKey(target.c_str());
  
  if (key == NULL) {
    FATAL("No key for " << target << endl);
  }
 
  JRootTypewriter printer;
 
  TClass* cs = TClass::GetClass(key->GetClassName());
 
  if (cs != NULL) {
 
    void* ps = key->ReadObjectAny(cs);
 
    if (cs == TTree::Class()) {
 
      TTree* ts = (TTree*) ps;
 
      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;
 
        ts->GetEvent(i);
 
        for (auto& tp : ls) {
          printer(cout,tp);
        }
      }
 
    } else { 
 
      printer(cout, cs->GetStreamerInfo(), (const char*) ps);
    }
  }
}