JVoltageOptimizer_IO.hh

Go to the documentation of this file.

#ifndef __READOPTIONS__
#define __READOPTIONS__

//c++ standard library
#include <iostream>

// Jpp
#include "JDAQ/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