JPrintChain.cc File Reference

Auxiliary program to print ROOT TChain information. More...

#include <string>
#include <iostream>
#include <iomanip>
#include <set>
#include <memory>
#include <string.h>
#include "TError.h"
#include "TROOT.h"
#include "TFile.h"
#include "TKey.h"
#include "TTree.h"
#include "TChain.h"
#include "JSystem/JGlob.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Detailed Description

Auxiliary program to print ROOT TChain information.

Author

mdejong

Definition in file JPrintChain.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 66 of file JPrintChain.cc.

{
  using namespace std;
  using namespace JPP;

  vector<string>     inputFile;
  int                debug;

  try { 

    JParser<> zap("Auxiliary program to print ROOT TChain information.");
    
    zap['f'] = make_field(inputFile);
    zap['d'] = make_field(debug)           = 1;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }

  gErrorIgnoreLevel = kFatal;

  inputFile = getFilenames(inputFile);

  set<JChain> buffer;

  for (vector<string>::const_iterator file_name = inputFile.begin(); file_name != inputFile.end(); ++file_name) {

    TFile* file = TFile::Open(file_name->c_str());

    if (file != NULL) {

      TIter iter(file->GetListOfKeys(), kIterBackward);

      for (TKey* key; (key = (TKey*) iter.Next()) != NULL; ) {

        TKey* p = dynamic_cast<TKey*>(file->GetListOfKeys()->Before(key));

        if (p == NULL || strcmp(key->GetName(), p->GetName()) != 0) {                      // select last key

          TTree* tree = dynamic_cast<TTree*>(key->ReadObj());

          if (tree != NULL) {
            buffer.insert(JChain(tree->GetName()));
          }
        }
      }

      file->Close();

      delete file;
    }
  }

  for (auto& chain : buffer) {

    for (vector<string>::const_iterator file_name = inputFile.begin(); file_name != inputFile.end(); ++file_name) {
      chain->Add(file_name->c_str());
    }

    cout << setw(24) << left  <<  chain->GetName()    << ' '
         << setw(10) << right <<  chain->GetEntries() << endl;
  }
}