JVoltageOptimizer_IO.hh

Go to the documentation of this file.

#ifndef __READOPTIONS__
#define __READOPTIONS__
 
//c++ standard library
#include <iostream>
 
// Jpp
#include "km3net-dataformat/online/JDAQTimeslice.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JDetector/JDetector.hh"
#include "JDetector/JModuleLocation.hh"
 
//ROOT
 
// namespaces
using namespace std;
using namespace JSUPPORT;   // for JFileScanner
using namespace JLANG;   // for JMultipleFileScanner
using namespace JDETECTOR; // for JModuleLocation
 
/**
 * \author rgruiz
 */
 
 
 
/**
 * Structure to store the different command line arguments for JVoltageOptimizer. 
 */
struct IO{
 
  JMultipleFileScanner<> ifnames ;
 
  string ofname_opt ;
 
  string ofname_cal ;
 
  string ofname_txt ;
 
  string detector_file ;
 
  int string_number ;
 
  int up_pmts ; 
  
  int down_pmts ;
 
  int max_neighbors ;
 
  double saturation_tot ;
 
};
 
 
 
/**
 * Parses the command line options and fills an IO structure with them.
 *
 * \param options an option structure
 * \param argc the number of command line arguments
 * \param argv the command line arguments
 *
 * \return 1 if works 2 if it doesn't work 
 */
inline int read_user_options(IO &options, int argc, char **argv){
 
  int a = 0;
 
  try {
 
    JParser<string> zap;
 
    zap["o"] = make_field(options.ofname_opt) = "out_opt.root" ;
 
    zap["c"] = make_field(options.ofname_cal) = "out_cal.root" ;
 
    zap["t"] = make_field(options.ofname_txt) = "out.txt" ;
 
    zap["f"] = make_field(options.ifnames) ; 
 
    zap["a"] = make_field(options.detector_file) ;
 
    zap["s"] = make_field(options.string_number) ;
 
    zap["u"] = make_field(options.up_pmts) = 1 ; 
 
    zap["d"] = make_field(options.down_pmts) = 4 ;
 
    zap["n"] = make_field(options.max_neighbors) = 2 ;
 
    zap["S"] = make_field(options.saturation_tot) = 26.4 ;
    
    if (zap.read(argc, argv) != 0)
 
      a = 1;
 
  }
 
  catch(const exception &error) {
 
    ERROR(error.what() << endl);
 
    a = 2;
 
  }
 
  return a;
 
}
 
 
/**
 * Writes a .root file with the optional output produced by JVoltageOptimizer.
 *
 * \param filename to output .root filex
 * \param t a root TTree.
 * \param summary_refs a TH2 with information of which reference pmts were used for optimizing the nano-beacon voltages.
 * \param Runs The list of nanobeacon Runs analyzed for the optimization.
 */
inline void write_output_opt(string filename , TTree* t , vector < TH2D* > summary_refs , vector < NBRun* > &Runs){
 
  TFile* f = new TFile(filename.c_str() , "recreate");
 
  f->cd();
 
  t->Write();
 
  int i = 0 ;
 
  char name [100] ;
 
  for (auto & run : Runs){
    
    sprintf(name , "RUN_%d_%2.1fV" , run->getRunNumber() , run->getVoltage()) ;
 
    f->mkdir(name) ;
 
    f->cd(name) ;
 
    summary_refs[i]->Write() ;
 
    for(auto & sm : run->getSuperModules()){
      
      sprintf(name , "RUN_%d_%2.1fV/NB_%d" , run->getRunNumber() , run->getVoltage() , sm->getFloor()) ;
 
      f->mkdir(name) ;
 
      f->cd(name) ;
      
      for(auto & spm : sm->get_ref_pmts()){
 
        spm->getNBPulse()->gettime_vs_tot()->Write() ;
 
        spm->getNBPulse()->gettime_peak()->Write() ;
 
        spm->getNBPulse()->gettot_peak()->Write() ;
 
      }
 
    }
    
    i++ ;
    
  }
  
  f->Close() ;
 
  delete(f) ;
 
}
 
 
/**
 * Writes a .txt file with the result produced by JVoltageOptimizer. 
 * This file contains a list of nanobeacons and optimal voltages
 *
 * \param filename path to the name of the .txt file
 * \param optimal_voltages List of pairs containing optimal voltage and the corresponding run number for each nanobeacon
 */
inline void write_output_txt( string filename , vector < pair < double , int > > optimal_voltages){
 
  ofstream txtfile;
 
  txtfile.open (filename);
 
  txtfile << "# Table of optimal voltages: column 1 = floor , column 2 = voltage" << endl;
  
  int i=0 ; 
 
  for (auto & op : optimal_voltages){
 
    txtfile << i+1 << "\t" << op.first << endl;
 
    i++ ;
 
  }
 
  txtfile.close();
 
}
 
/**
 * Writes a .root file with the optional output produced by JVoltageOptimizer. 
 * This file has the format needed by JInterDomCal to perform a calibration of the DU and contains the pulses corresponding to the optimized voltages.
 *
 * \param filename path to the name of the .root
 * \param optimal_voltages List of pairs containing optimal voltage and the corresponding run number for each nanobeacon
 * \param Runs vector of NBruns
 * \param string_number Number of the DU to calibrate.
 * \param nFloors Number of floors in the detector. 
 */
inline void write_output_cal(string filename , vector< pair < double , int > > &optimal_voltages , vector < NBRun* > &Runs , int string_number , int nFloors){
 
  TFile* out_file = new TFile(filename.c_str() , "recreate") ;
 
  TVectorD Run (2) ;
 
  Run[0] = 999 ;
 
  Run[1] = 999 ;
 
  Run.Write("Run_Info") ;
 
  int Ndoms = nFloors ;
 
  char dirname[200] ;
 
  for( int i = 0 ; i < Ndoms ; ++i ) {
 
    sprintf(dirname, "S%iF%i", string_number, i + 1 ) ;
 
    out_file->mkdir(dirname)->cd() ;
 
  }
  
  out_file->cd();
  
  int nb_index = 0;
 
  for (auto & volt :optimal_voltages){
 
    NBRun* r = Runs[volt.second] ;
 
    vector< pair < SuperModule* , vector < SuperPMT* > > > targets = r->getSuperModules()[nb_index]->get_targets() ;
 
    for (auto & tgt : targets){
 
      int floor = tgt.first->getFloor() ;
 
      sprintf( dirname , "S%iF%i", string_number , floor ) ;
      
      out_file->cd(dirname) ;
 
      for (auto & spm : tgt.second){
 
        spm->getNBPulse()->gettime_vs_tot()->Write() ;
 
      }
 
    }
 
    sprintf(dirname , "S%iF%i", string_number ,  r->getSuperModules()[nb_index]->getFloor()) ;
 
    out_file->cd(dirname) ;
    
    for (auto & spm : r->getSuperModules()[nb_index]->get_ref_pmts()){
 
      spm->getNBPulse()->gettime_vs_tot()->Write() ;
 
    }
 
    out_file->cd();
 
    nb_index ++ ;
 
  }
 
  out_file->Close();
 
  delete(out_file) ;
 
}
 
 
#endif