CreateMergedHVETO.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/>.
*/
 
 
#include "ReadChunkList.C"
#include "MergeSegments.C"
 
#define CHUNK_FILE_LIST         "Chunk_List.txt"
#define CHUNK_MAX_SIZE          100
 
#define EXECUTE   
 
void CreateMergedHVETO(TString run, TString ifo, TString calibVer) {
 
  if(run!="O1" && run!="O2") {
    cout << "CreateMergedHVETO : Error - run type not available !!!" << endl;
    exit(1);
  } 
  if(ifo!="L1" && ifo!="H1" && ifo!="V1") {
    cout << "CreateMergedHVETO : Error - ifo type not available !!!" << endl;
    exit(1);
  } 
 
  // get CWB_CONFIG
  char cwb_config_env[1024] = "";
  if(gSystem->Getenv("CWB_CONFIG")!=NULL) {
    strcpy(cwb_config_env,TString(gSystem->Getenv("CWB_CONFIG")).Data());
  }
 
  char chunk_file_list[1024];
  sprintf(chunk_file_list,"%s/%s/CHUNKS/%s",cwb_config_env,run.Data(),CHUNK_FILE_LIST);
  cout << chunk_file_list << endl;
 
  int    chunk[CHUNK_MAX_SIZE];
  double start[CHUNK_MAX_SIZE];
  double stop[CHUNK_MAX_SIZE];
 
  int nChunks = ReadChunkList(chunk_file_list, chunk, start, stop);
 
  char vetoes_dir[1024];
  sprintf(vetoes_dir,"%s/%s/DATA/%s/HVETO",cwb_config_env,run.Data(),calibVer.Data());
  char vetoes_dir_tmp[1024];
  sprintf(vetoes_dir_tmp,"%s/%s/DATA/%s/HVETO/tmp",cwb_config_env,run.Data(),calibVer.Data());
 
  // create temporary vetoes_dir_tmp dir
  cout << vetoes_dir_tmp << endl << endl;
#ifdef EXECUTE
  CWB::Toolbox::mkDir(vetoes_dir_tmp,true,true);
#endif
 
  // merge douplicate entries 
  char ihveto_file[1024];
  char ohveto_file[1024];
  for(int k=0;k<nChunks;k++) {
    sprintf(ihveto_file,"%s/%s/HVETO_%s_ANALYSIS%d_SEGMENTS.txt",vetoes_dir,ifo.Data(),ifo.Data(),chunk[k]);
    sprintf(ohveto_file,"%s/HVETO_%s_ANALYSIS%d_SEGMENTS_MERGED.txt",vetoes_dir_tmp,ifo.Data(),chunk[k]);
    cout << "ihveto_file  : " << ihveto_file << endl;
    cout << "ohveto_file : " << ohveto_file << endl;
    cout << endl;
#ifdef EXECUTE
    MergeSegments(ihveto_file,ohveto_file,false);
#endif
  }
 
  // merge hveto files
  char cmd[1024];
  sprintf(cmd,"rm %s/%s/HVETO_%s_ANALYSIS_SEGMENTS_MERGED.txt",vetoes_dir,ifo.Data(),ifo.Data());
  cout << cmd << endl;
#ifdef EXECUTE
  gSystem->Exec(cmd);
#endif
  for(int k=0;k<nChunks;k++) {
    if(k==0) { 
      sprintf(cmd,"cat %s/HVETO_%s_ANALYSIS%d_SEGMENTS_MERGED.txt >  %s/%s/HVETO_%s_ANALYSIS_SEGMENTS_MERGED.txt",
              vetoes_dir_tmp,ifo.Data(),chunk[k],vetoes_dir,ifo.Data(),ifo.Data());
    } else {
      sprintf(cmd,"cat %s/HVETO_%s_ANALYSIS%d_SEGMENTS_MERGED.txt >> %s/%s/HVETO_%s_ANALYSIS_SEGMENTS_MERGED.txt",
              vetoes_dir_tmp,ifo.Data(),chunk[k],vetoes_dir,ifo.Data(),ifo.Data());
    }
    cout << cmd << endl;
#ifdef EXECUTE
    gSystem->Exec(cmd);
#endif
  }
 
  // remove temporary vetoes_dir_tmp dir
  cout << vetoes_dir_tmp << endl << endl;
#ifdef EXECUTE
  CWB::Toolbox::rmDir(vetoes_dir_tmp,false);
#endif
 
  exit(0);
}