JVoltageOptimizer.cc

Go to the documentation of this file.

// C++ 
#include <string>
#include <iostream>
#include <sstream>
#include <iomanip>
 
 
// ROOT
#include "TTree.h"
 
 
// Nanobeacons
#include "NBRun.hh"
#include "Detector.hh"
#include "JVoltageOptimizer_IO.hh"
#include "JVoltageOptimizer_utils.hh"
#include "Control_utils.hh"
 
 
// namespaces
using namespace std;
 
 
int main(int argc, char **argv) {
 
  IO options ;
 
  read_user_options(options, argc , argv) ;
 
  JDetector detector = loadDetector(options.detector_file) ;
 
  int Nruns = options.ifnames.size() ;
 
  int Nbeacons = getNumberOfFloors(detector) ;
 
  vector < vector < event > > evt_table (Nbeacons , vector < event > (Nruns)) ;
 
  char tname [50] ;
  
  sprintf(tname , "DU_%d" , options.string_number) ;
 
  TTree* t = new TTree(tname , tname) ;
 
  vector < TH2D* > summary_refs ; 
  
  char bname [50] ;
  
  for (int i=0 ; i<Nbeacons ; i++){
    
    sprintf(bname , "NB%d" , i+1) ;
    
    t->Branch(bname , &evt_table[i][0] , "V/D:meanToT_ref/D:meanToT_tgt/D:sigmaToT_ref/D:sigmaToT_tgt/D") ;
  
  }
 
  int run_index = 0 ;
 
  vector < NBRun* > Sruns (Nruns) ;
 
  for(auto & file : options.ifnames){
 
    NBRun* run  = new NBRun (file , detector , options.string_number , options.up_pmts , options.down_pmts , options.max_neighbors) ;
 
    Sruns[run_index] = run ;
 
    double v = run->getVoltage() ;
 
    summary_refs.push_back(refs_th2(run)) ;
    
    cout << "Analyzing run number: " << run->getRunNumber() << " Voltage = " << v << " V." << endl ;
 
    run->analyze(0) ;
 
    run->computeMeanToTs() ;
 
    int i = 0 ;
 
    for (auto & sm : run->getSuperModules()){
 
      sprintf(bname , "NB%d" , i+1) ;
 
      t->SetBranchAddress(bname , &evt_table[i][run_index]) ;
 
      evt_table[i][run_index].V = v ;
 
      evt_table[i][run_index].meanToT_ref = sm->getMeanToT_refs() ;
 
      evt_table[i][run_index].meanToT_tgt = sm->getMeanToT_tgts() ;
      
      evt_table[i][run_index].sigmaToT_ref = sm->getSigmaToT_refs() ;
      
      evt_table[i][run_index].sigmaToT_tgt = sm->getSigmaToT_tgts() ;
      
      i++ ;
 
    }
 
    t->Fill() ;
 
    run_index++ ;
  } 
 
  vector < pair < double , int > > OptimalVoltages = optimize_voltage (evt_table , options.saturation_tot ) ;
 
  write_output_txt( options.ofname_txt , OptimalVoltages ) ;
 
  write_output_opt( options.ofname_opt , t , summary_refs , Sruns) ;
 
  write_output_cal( options.ofname_cal , OptimalVoltages , Sruns , options.string_number , getNumberOfFloors(detector) ) ;
  
}