JMergeNB.cc

Go to the documentation of this file.

#include <iostream>
#include <string>
#include <vector>
#include <map>
 
#include "Jeep/JParser.hh"
 
#include "TH1.h"
#include "TFile.h"
#include "TKey.h"
 
using namespace std;
using namespace JPP;
 
int main(int argc , char** argv){
 
  vector<string> inputFiles;
  string outFile;
 
  try {
 
    JParser<> zap;
 
    zap['f'] = make_field(inputFiles);
    zap['o'] = make_field(outFile  ) = "out.root";
 
    zap(argc,argv);
  }
  catch(const exception &error) {
    ERROR(error.what() << endl);
  }
 
  typedef map<TString,TH1*> map_type;
  map_type zmap;
 
  for(vector<string>::const_iterator i = inputFiles.begin(); i != inputFiles.end(); ++i){
    cout << " Now processing: " << *i << endl;
    TFile in(i->c_str() , "read");
 
    TIter iter(in.GetListOfKeys());
 
    for(TKey *key; (key = (TKey*) iter.Next()) != NULL; ){
 
      TH1* h = dynamic_cast<TH1*>(key->ReadObj());
 
      if( h != NULL){
 
        map_type::iterator p = zmap.find(h->GetName());
 
        if (p == zmap.end()){
 
          p = zmap.insert( make_pair( h->GetName(), (TH1*) h->Clone() )).first;
          
        } else {
 
          p->second->Add(h);
        }
      }
    }
 
    for (map_type::iterator i = zmap.begin(); i != zmap.end(); ++i) {
      i->second->SetDirectory(0);
    }
 
    in.Close();
  }
 
  TFile output(outFile.c_str() , "recreate") ;
  for (map_type::iterator i = zmap.begin(); i != zmap.end(); ++i) {
    
    TH1D* hist = (TH1D*) i->second;
    hist->Write();
  } 
  output.Close();
}