pulse_classification.cc

Go to the documentation of this file.

// c++ standard library
#include <string>
#include <iostream>
#include <sstream>
#include <iomanip>
 
#include "TCanvas.h"
#include "RooPlot.h"
#include "RooDerivative.h"
#include "TLine.h"
#include "TF1.h"
 
#include "NBRun.hh"
#include "Detector.hh"
#include "pulse_classification_IO.hh"
#include "Control_utils.hh"
 
// namespaces
 
using namespace std;
 
 
/**
 * Writes a .root file with some objects that can be used to check the calibration.
 *
 * \param filename path to the name of the .root
 * \param run The nanobeacon calibration run
 */
inline void write_output(string filename , NBRun& run){
 
  if (filename.length()==0){
 
    cout << "Omitting calibration checks..." << endl ;
 
    return ;
 
  }else{
 
    cout << "Writing pulses to file: " << endl ;
 
    cout << filename << endl ;
 
    cout << "------------------------------------" << endl << endl ;
 
    TFile outfile(filename.c_str() , "recreate") ;
    
    outfile.cd() ;
 
    outfile.mkdir("REF/Good") ;
    outfile.mkdir("REF/Weak") ;
    outfile.mkdir("REF/Saturated") ;
    
    outfile.mkdir("TGT/Good") ;
    outfile.mkdir("TGT/Weak") ;
    outfile.mkdir("TGT/Saturated") ;
 
    vector <SuperModule*> SuperMods = run.getSuperModules() ;
 
      
    for (auto & sm : SuperMods){
 
      for (auto & spm : sm->get_ref_pmts()){
   
        if (spm->getNBPulse()->IsGood()==true) outfile.cd("REF/Good") ;
      
        if (spm->getNBPulse()->IsSaturatedHit()==true) outfile.cd("REF/Saturated") ;
      
        if (spm->getNBPulse()->IsWeak()==true) outfile.cd("REF/Weak") ;
      
        spm->getNBPulse()->getHtime_full()->Write() ;
      
        if (spm->getNBPulse()->IsFitted()==true){
        
          RooWorkspace w = spm->getNBPulse()->getWorkspace() ;
        
          TH1D* h = spm->getNBPulse()->getHtime_full() ;
        
          TCanvas* c1 = RooCanvas(w , h) ;
        
          c1->Write() ;
        
          w.Write() ;
        
        }
      
        outfile.cd() ;
      
      }
             
      for (auto & sms : sm->get_sources()){
 
        for (auto & spm : sms.second){
   
          if (spm->getNBPulse()->IsGood()==true)  outfile.cd("TGT/Good") ;
            
          if (spm->getNBPulse()->IsSaturatedHit()==true) outfile.cd("TGT/Saturated") ;
            
          if (spm->getNBPulse()->IsWeak()==true) outfile.cd("TGT/Weak") ;
        
          spm->getNBPulse()->getHtime_full()->Write() ;
        
          if (spm->getNBPulse()->IsFitted()==true){
          
            RooWorkspace w = spm->getNBPulse()->getWorkspace() ;
 
            TH1D* h = spm->getNBPulse()->getHtime_full() ;
 
            TCanvas* c1 = RooCanvas(w , h) ;
 
            c1->Write() ;
 
            w.Write() ;
 
          }
        
          outfile.cd() ;
 
        }
 
      }
 
    }
 
    outfile.Close() ;
 
  }
 
}
 
 
int main(int argc, char **argv) {
 
  IO options ;
 
  read_user_options(options, argc , argv);
 
  JDetector detector = loadDetector(options.detector_file) ;
 
  NBRun run (options.ifname , detector , options.string_number , options.up_pmts , options.down_pmts , options.number_neighbors) ;
 
  run.analyze(options.analysis_level) ;
 
  write_output(options.ofname , run) ;    
 
}