Control_utils.hh

Go to the documentation of this file.

#ifndef __CONTROL_UTILS__
#define __CONTROL_UTILS__


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



/**
 * Produces a TH2 to be read as a table that summarizes which pmts were used as references for each module.
 *
 * \param Run a pointer to a nanobeacon run
 * \return A TH2
 */
inline TH2D* refs_th2(NBRun* Run){
  
  int run_number = Run->getRunNumber() ;

  double voltage = Run->getVoltage() ;

  vector <SuperModule*> SuperMods = Run->getSuperModules() ;
  
  char name[100] ;

  sprintf(name , "refs_Run_%d_%2.1fV", run_number , voltage) ;

  char title[100] ;

  sprintf(title , "Used reference PMTs") ;
  
  TH2D* refs = new TH2D(name , title , 18 , 0.5 , 18.5 , 12 , -0.5 , 11.5) ;
  
  for (auto & sm : SuperMods) {
    
    if (sm->get_ref_pmts().size()>0) {
      
      for (auto & ref : sm->get_ref_pmts()) {
        
        refs->Fill(sm->getFloor() , ref->getChannel()) ;
        
      }
      
    }
    
  }
  
  refs->GetXaxis()->SetTitle("Nanobeacon Floor") ;
  
  refs->GetYaxis()->SetTitle("PMT Channel (FPGA)") ;
  
  refs->SetOption("colz") ;
  
  refs->SetStats(0) ;
  
  return refs ;
  
}



/**
 * Produces a TH2 to be read as a table that summarizes which pmts were used as good references for each module.
 *
 * \param Run a pointer to a nanobeacon run
 * \return A TH2
 */
inline TH2D* good_refs_th2(NBRun* Run){
  
  int run_number = Run->getRunNumber() ;

  double voltage = Run->getVoltage() ;

  vector <SuperModule*> SuperMods = Run->getSuperModules() ;
  
  char name[100] ;

  sprintf(name , "refs_Run_%d_%2.1fV", run_number , voltage) ;

  char title[100] ;

  sprintf(title , "Used reference PMTs") ;
  
  TH2D* refs = new TH2D(name , title , 18 , 0.5 , 18.5 , 12 , -0.5 , 11.5) ;
  
  for (auto & sm : SuperMods) {
    
    if (sm->get_ref_pmts().size()>0) {
      
      for (auto & ref : sm->get_ref_pmts()) {
        
        refs->Fill(sm->getFloor() , ref->getChannel()) ;
        
        if (ref->getNBPulse()->IsGood()==true){

          refs->Fill(sm->getFloor() , ref->getChannel()) ;

        }
        
      }
      
    }
    
  }
  
  refs->GetXaxis()->SetTitle("Nanobeacon Floor") ;
  
  refs->GetYaxis()->SetTitle("PMT Channel (FPGA)") ;
  
  refs->SetOption("colz") ;
  
  refs->SetStats(0) ;
  
  return refs ;
  
}


/**
 * Produces a TH2 to be read as a table that summarizes which pmts were used as good references for each module.
 *
 * \param Run a pointer to a nanobeacon run
 * \return A TH2
 */
inline TH2D* good_tgt_pmts_th2(NBRun* Run){
  
  int run_number = Run->getRunNumber() ;

  double voltage = Run->getVoltage() ;

  vector <SuperModule*> SuperMods = Run->getSuperModules() ;
  
  char name[100] ;

  sprintf(name , "tgt_pmts_Run_%d_%2.1fV", run_number , voltage) ;

  char title[100] ;

  sprintf(title , "Target PMTs used for the closest good source.") ;
  
  TH2D* tgts = new TH2D(name , title , 18 , 0.5 , 18.5 , 19 , 11.5 , 30.5) ;
  
  for (auto & sm : SuperMods) {
    
    if(sm->has_good_sources()==true){

      for (auto & spm : sm->get_closest_good_source().second){

        tgts->Fill(sm->getFloor() , spm->getChannel());

        if (spm->getNBPulse()->IsGood()==true){

          tgts->Fill(sm->getFloor() , spm->getChannel());         

        }

      }

    }
    
  }
  
  tgts->GetXaxis()->SetTitle("Nanobeacon Floor") ;
  
  tgts->GetYaxis()->SetTitle("PMT Channel (FPGA)") ;
  
  tgts->SetOption("colz") ;
  
  tgts->SetStats(0) ;
  
  return tgts ;
  
}



/**
 * Produces a TH2 to be read as a table that summarizes which modules were used as good sources for each target module.
 *
 * \param Run a pointer to a nanobeacon run
 * \return A TH2
 */
inline TH2D* srcs_th2(NBRun* Run){
  
  int run_number = Run->getRunNumber() ;

  double voltage = Run->getVoltage() ;

  vector <SuperModule*> SuperMods = Run->getSuperModules() ;

  char name[100] ;

  sprintf(name , "srcs_Run_%d_%2.1fV", run_number , voltage) ;

  char title[100] ;

  sprintf(title , "Sources for each target module.") ;

  TH2D* srcs = new TH2D(name , title , 18 , 0.5 , 18.5 , 18 , 0.5 , 18.5);

  for(auto & sm : SuperMods){
    
    int thisfloor = sm->getFloor() ;
    
    if(sm->get_sources().size()>0){

      for (auto & src : sm->get_sources()){
      
        int thatfloor = src.first->getFloor();

        srcs->Fill(thisfloor , thatfloor ) ;

      }
    }
    
    if(sm->get_good_sources().size()>0){

      for (auto & src : sm->get_good_sources()){
      
        int thatfloor = src.first->getFloor();
      
        srcs->Fill(thisfloor , thatfloor) ;
      }

      int thatfloor = sm->get_closest_good_source().first->getFloor();
            
      srcs->Fill(thisfloor , thatfloor) ;
    }

  }  

  srcs->GetXaxis()->SetTitle("Target Floor") ;
  
  srcs->GetYaxis()->SetTitle("Source Floor") ;

  srcs->SetOption("colz") ;
  
  srcs->SetStats(0) ;

  return srcs;
}



/**
 * Produces a TH2 to be interpreted as a table that summarizes which modules were targets for each source module.
 *
 * \param Run a pointer to a nanobeacon run
 * \return A TH2
 */
inline TH2D* tgts_th2(NBRun* Run){
  
  int run_number = Run->getRunNumber() ;

  double voltage = Run->getVoltage() ;

  vector <SuperModule*> SuperMods = Run->getSuperModules() ;

  char name[100] ;

  sprintf(name , "tgts_Run_%d_%2.1fV", run_number , voltage) ;

  char title[100] ;

  sprintf(title , "Targets for each source module.") ;

  TH2D* tgts = new TH2D(name , title , 18 , 0.5 , 18.5 , 18 , 0.5 , 18.5) ;

  for(auto & sm : SuperMods){
    
    int thisfloor = sm->getFloor() ;
    
    if(sm->get_targets().size()>0){

      for (auto & tgt : sm->get_targets()){
      
        int thatfloor = tgt.first->getFloor() ;

        tgts->Fill(thisfloor , thatfloor ) ;

      }
    }
    
    if(sm->get_good_targets().size()>0){

      for (auto & tgt : sm->get_good_targets()){
      
        int thatfloor = tgt.first->getFloor() ;
      
        tgts->Fill(thisfloor , thatfloor) ;
      
      }

      int thatfloor = sm->get_closest_good_target().first->getFloor() ;
            
      tgts->Fill(thisfloor , thatfloor) ;
    }

  }

  tgts->SetOption("colz") ;

  tgts->GetXaxis()->SetTitle("Source Floor") ;
  
  tgts->GetYaxis()->SetTitle("Target Floor") ;

  tgts->SetOption("colz") ;
  
  tgts->SetStats(0) ;

  return tgts ;

}


/**
 * Produces a TCanvas with the nanobeacon peak and the fit performed with Roofit. 
 *
 * \param w A Rooworkspace that contains all the information relative to the fit.
 * \param h The histogram with the nanobeacon peak.
 * \return A TCanvas
 */
inline TCanvas* RooCanvas(RooWorkspace w , TH1D* h){
  
  TCanvas * c = new TCanvas( (std::string("canvas_")+h->GetName()).c_str() , (std::string("canvas_")+h->GetName()).c_str() );
          
  c->cd() ;

  RooPlot* frame = w.var("time")->frame() ;

  RooDataHist Data ("data" , "data" , *w.var("time") , Import(*h)) ;

  Data.plotOn(frame) ;

  w.function("Model")->plotOn(frame) ;

  w.function("L1")->plotOn(frame , LineStyle(kDashed)) ;

  w.function("G1")->plotOn(frame , LineStyle(kDashed) , LineColor(kRed)) ;

  frame->Draw() ;

  TLine* line = new TLine ( w.var("mpv")->getVal() , 1 , w.var("mpv")->getVal() , 1e5) ;

  line->SetLineColor(kBlue) ;

  TLine* line2 = new TLine ( w.var("tpeak")->getVal() , 1 , w.var("tpeak")->getVal() , 1e5) ;

  line2->SetLineColor(kBlack) ;

  line->Draw("same") ;

  line2->Draw("same") ;

  return c ;
}

#endif