CreateExclusiveLiveTimeFile.C

Go to the documentation of this file.

/*
# Copyright (C) 2019 Gabriele Vedovato
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/
 
//
// this macro converts the standard livetime tree to the exclusive livetime tree
// only the zero lag leaf 'live' is updated
//
 
#define APPLY_EXCLUSIVE_LIVETIME_ONLY_TO_ZL  // update only zero lag
 
void CreateExclusiveLiveTimeFile(TString merge_label,TString searchDir_name, TString esegList_name, TString run, TString olabel) {
//
// searchDir_name:   input search dir name (BURST, BURST_BBH, BURST_IMBHB, BURST_HF, BURST_LD)
// esegList_name:    input exclusive segment list (must be the list for zero lag)
// run:        input run (O1,O2,O3a,O3b)
// olabel:     output label used for the output livetime tree file name .root -> .olabel.root      
//
 
  // input exclusive segment list
  TString esegList_path = TString::Format("%s/DATA/%s/DATA/%s/OBSTIME/%s/%s",cwb_config_env,run.Data(),calibVer.Data(),searchDir_name.Data(),esegList_name.Data());
  CWB::Toolbox::checkFile(esegList_path);
  cout << esegList_path << endl;
 
  // input livetime tree file name
  TString ifName = TString::Format("merge/live_%s.%s.root",data_label,merge_label.Data());
  CWB::Toolbox::checkFile(ifName);
  cout << ifName << endl;
 
  // output livetime tree file name
  TString ofName = TString::Format("merge/live_%s.%s.%s.root",data_label,merge_label.Data(),olabel.Data());
  cout << ofName << endl;
 
  vector<waveSegment> esegList = CWB::Toolbox::readSegments(esegList_path);        // read exclusive segments
 
  // open input livetime tree
  TFile *iroot = TFile::Open(ifName);
  if(iroot==NULL) {cout << "CreateExclusiveLiveTimeFile : input root file not found !!!" << endl;exit(1);}
 
  TTree* itree = (TTree *) gROOT->FindObject("liveTime");
  if(itree==NULL) {cout << "CreateExclusiveLiveTimeFile : liveTime tree not found !!!" << endl;exit(1);}
 
  int ientries = (int)itree->GetEntries();
  cout << "livetime entries = " << ientries << endl;
 
  double ilive;
  itree->SetBranchAddress("live", &ilive);
 
  double* istart = new double[NIFO_MAX];
  itree->SetBranchAddress("start", istart);
 
  double* istop = new double[NIFO_MAX];
  itree->SetBranchAddress("stop", istop);
 
  float* ilag = new float[NIFO_MAX+1];
  itree->SetBranchAddress("lag", ilag);
 
  float* islag = new float[NIFO_MAX+1];
  itree->SetBranchAddress("slag", islag);
 
  //open new file to store the output livetime Tree
  TFile oroot(ofName,"recreate");
  //Create an empty clone of the original tree
  TTree *otree = (TTree*)itree->CloneTree(0);
 
  double olive;
  otree->SetBranchAddress("live", &olive);
 
  waveSegment SEG;
  vector<waveSegment> osegList;
 
  double ilive_time=0.;
  double olive_time=0.;
  for (Int_t i=0;i<ientries;i++) {
     if(i%100000==0) cout << i << " / " << ientries << endl;
     itree->GetEntry(i);
 
#ifdef APPLY_EXCLUSIVE_LIVETIME_ONLY_TO_ZL 
     if(ilag[nIFO]==0 && islag[nIFO]==0) {      // zero lag entries
#endif
       // get range segment for this job intersecting with cat1 list (cat1List)
       SEG.start = istart[0];
       SEG.stop  = istop[0];
       osegList.clear();
       osegList.push_back(SEG);
       osegList = CWB::Toolbox::mergeSegLists(esegList,osegList);    // merge exclusive segments list & input job time range
       olive=CWB::Toolbox::getTimeSegList(osegList);
 
       ilive_time+=ilive;
       olive_time+=olive;
#ifdef APPLY_EXCLUSIVE_LIVETIME_ONLY_TO_ZL 
     } else {
       olive=ilive;
     }
#endif
 
     otree->Fill();
  }
  otree->Write();
  oroot.Close();
  iroot->Close();
 
  cout << "ilivetime zero lag : " << ilive_time << " sec\t" << ilive_time/(24*3600.) << " days" << endl;
  cout << "olivetime zero lag : " << olive_time << " sec\t" << olive_time/(24*3600.) << " days" << endl;
 
  exit(0);
}