insert_run_calib.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 <stdlib.h>
 
int const num_item = 9;
TString str_name_list[num_item]={"runcalib",
                 "gaincalib",
                 "resolcalib",
                 "wiregcalib",
                 "ggscalib",
                 "ddgcalib",
                 "entracalib",
                 "zdepcalib",
                 "layergcalib"};
TTree *arr_tree[num_item];
using namespace std;
 
int main(int argc, char *argv[]){
  if(argc<2){
    cout << "plase append the old calibration const root file, Ex. insert_run_calib.exe old.root [runlist.txt]" << endl;
    return 0;
  }
  TString str_old = argv[1];
  TString str_runlist = "runlist.txt";
  if(argc>=3) str_runlist = argv[2];
  cout << "old file: " << str_old << "   runlist file: " << str_runlist <<  endl;
  TFile f_old(str_old); //input
  TFile f_new("new.root","recreate");
  for(int i=0; i<num_item; i++) arr_tree[i] = ((TTree*)f_old.Get(str_name_list[i]))->CloneTree();
  ifstream in(str_runlist);
  if(!in){
    cout << "cannot open runlist.txt" << endl;
    return 0;
  }
  char str[255];
  int in_runno(0);
  int runno;
  Double_t runmean, rungain, runresol;
  while(in){
    in.getline(str, 255);
    sscanf(str ,"%d", &in_runno);
    if(!in_runno) continue; // protect of read at the end
    arr_tree[0]->SetBranchAddress("runno", &runno    );
    arr_tree[0]->SetBranchAddress("runmean", &runmean);
    arr_tree[0]->SetBranchAddress("rungain", &rungain);
    arr_tree[0]->SetBranchAddress("runresol", &runresol);
    int min_diff(10000);
    double temp_runmean(0), temp_rungain(0), temp_runresol(0);
    for(Long64_t i=0; i<arr_tree[0]->GetEntries(); i++){
      arr_tree[0]->GetEntry(i);
      if(abs(runno-in_runno)<min_diff){
    min_diff = abs(runno-in_runno);
    temp_runmean = runmean;
    temp_rungain = rungain;
    temp_runresol = runresol;
      }
    }
    runno = in_runno;
    runmean = temp_runmean;
    rungain = temp_rungain;
    runresol = temp_runresol;
    cout << "runno " << runno << endl;
    arr_tree[0]->Fill();
    in_runno = 0;
  }
 
  f_new.cd();
  for(int j=0; j<num_item; j++) arr_tree[j]->Write(str_name_list[j]);
  f_new.Close();
  f_old.Close();
}