compare_trees.cxx

Go to the documentation of this file.

#include <iostream>
#include <fstream>
#include <iomanip>
#include <cstring>
#include <string>
#include <vector>
#include <TFile.h>
#include <TTree.h>
#include <TString.h>
#include <TBranch.h>
#include <TLeaf.h>
#include <TH1F.h>
 
using namespace std;
int const num_item = 9;
TString str_name_list[num_item]={"runcalib",
                 "gaincalib",
                 "resolcalib",
                 "wiregcalib",
                 "ggscalib",
                 "ddgcalib",
                 "entracalib",
                 "zdepcalib",
                 "layergcalib"};
 
int main(int argc, char *argv[]){
  if(argc<3){
    cout << "please append two root files' names you want to compare " << endl;
    return 1;
  }
 
 
  TFile f1(argv[1]);
  TTree *t1(0);
  TFile f2(argv[2]);
  TTree *t2(0);
 
  int m_flag(0);
  TH1F *h1 = new TH1F("h1","h1",100000,-100000,100000);
  TH1F *h2 = new TH1F("h2","h2",100000,-100000,100000);
 
  for(int i=0; i<num_item; i++){
    m_flag = 0;
    t1 = (TTree*)f1.Get(str_name_list[i]);
    t2 = (TTree*)f2.Get(str_name_list[i]);
    TObjArray *l_b1 = t1->GetListOfBranches();
    TObjArray *l_b2 = t1->GetListOfBranches();
    if(l_b1->GetSize() != l_b2->GetSize()){
      cout << "tree " << str_name_list[i] << " is different" << endl;
      break;
    }
    //    cout << "size " << l_b1->GetEntries() << endl;
    for(int j=0; j<l_b1->GetEntries(); j++){
      // cout << "j= " << j << "  branch name " << l_b1->At(j)->GetName() << endl;
      t1->Project("h1", l_b1->At(j)->GetName());
      t2->Project("h2", l_b2->At(j)->GetName());
      if(h1->GetMean() - h2->GetMean()){
        cout << "tree " << str_name_list[i] << " is different" << endl;
        m_flag = 1;
        break;
      } 
    }// end of loop branches in trees
    if(m_flag) continue;
    else cout << "tree " << str_name_list[i] << " is same" << endl;
  }// end of loop tree
 
  return 0;
}