pulse_classification.cc File Reference

#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"

Go to the source code of this file.

Functions
void	write_output (string filename, NBRun &run)
	Writes a .root file with some objects that can be used to check the calibration. More...
int	main (int argc, char **argv)

Function Documentation

write_output()

void write_output ( string  filename, NBRun &  run  )

inline

Writes a .root file with some objects that can be used to check the calibration.

Parameters

filename	path to the name of the .root
run	The nanobeacon calibration run

Definition at line 29 of file pulse_classification.cc.

                                                      {
 
  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() ;
 
  }
 
}

main()

int main	(	int	argc,
		char **	argv
	)

Definition at line 131 of file pulse_classification.cc.

                                {
 
  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) ;    
 
}