genRunList.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>
 
using namespace std;
 
int main(int argc, char* argv[]){
  if(argc<3){
    cout << "please append path name and root file name [and tree names] [and tag name]" << endl;
    return 1;
  }
 
  TString str_root_file(argv[1]);
  str_root_file.Append(argv[2]);
  TFile f(str_root_file);
  TTree *t(0);
  if(argc>3) t = (TTree*)f.Get(argv[3]);
  else{
    TList *l = f.GetListOfKeys();
    if(l->GetEntries()<1){ 
      cout << "the root file " << argv[2] << " is empty, check it!" << endl;
      return 1;
    }
    else t = (TTree*)f.Get(l->At(0)->GetName());
  }
  Long_t m_totnum = t->GetEntries();
  //  double m_run(0), m_event(0), m_cur_run(0);
  int m_run(0), m_event(0), m_cur_run(0);
  char str_run_name[20];
  TString str_run_name_II("");
  t->SetBranchAddress("run", &m_run);
  t->SetBranchAddress("event", &m_event);
  ofstream f_out;
  cout << "total events: " << m_totnum << endl;
  for(Long_t i=0; i<m_totnum; i++){
 
    t->GetEntry(i);
    if(i%10000==0) cout << "i = " << i << "   run: " << m_run << "   event: " << m_event << endl; 
    if(m_run != m_cur_run && m_run>1){
      if(f_out.is_open()) f_out.close();
      int m_temp_run = (int)m_run;
      if(argc<5){
    sprintf(str_run_name, "%d.txt", m_temp_run);
    str_run_name_II = argv[2];
    str_run_name_II.ReplaceAll("_","");
    str_run_name_II.ReplaceAll(".root","_runnumber");
    str_run_name_II.ReplaceAll("runnumber", str_run_name);
    cout << "run file: " << str_run_name_II << endl;
    f_out.open(str_run_name_II, ios::app);
      }
      else{
    sprintf(str_run_name, "%s_%d.txt", argv[4], m_temp_run);
    cout << "run file: " << str_run_name << endl;
    f_out.open(str_run_name, ios::app);
      }
      m_cur_run = m_run;
    }
    if(!f_out.is_open()){ cout << "cannot open file " << endl; return 1; }
    f_out << (long)m_run << "    " << (long)m_event << endl;
    // cout << (long)m_run << "    " << (long)m_event << endl;
  }
  f_out.close();
 
  return 0;
}