VtxR3Analysis.cxx

Go to the documentation of this file.

 
 
#include "TrigInDetAnalysisExample/VtxR3Analysis.h"
 
#include "TrigInDetAnalysisUtils/VertexMatcher.h"
 
 
VtxR3Analysis::VtxR3Analysis( const std::string& n ) :
  VertexAnalysis( n ),
  m_initialised(false),
  m_monTool(nullptr),
  m_hnvtx(nullptr),
  m_hzed(nullptr),
  m_hx(nullptr),
  m_hy(nullptr),
  m_hntrax(nullptr),
  m_hnvtx_rec(nullptr),
  m_hzed_rec(nullptr),
  m_hx_rec(nullptr),
  m_hy_rec(nullptr),
  m_hntrax_rec(nullptr),
  m_hzed_res(nullptr),
  m_hx_res(nullptr),
  m_hy_res(nullptr),
  m_rdz_vs_zed(nullptr),
  m_rdz_vs_ntrax(nullptr),
  m_rdz_vs_nvtx(nullptr),
  m_eff_zed(nullptr),
  m_eff_ntrax(nullptr),
  m_eff_nvtx(nullptr),
  m_eff_mu(nullptr),
  m_eff_lb(nullptr)
{}
 
 
void VtxR3Analysis::initialise() { 
 
  m_initialised = true;
 
  //  std::cout << "VtxAnalysis::initialise() " << name() << std::endl;
 
 
  //  mdir = new TIDDirectory(name());
  //  mdir->push();
 
 
 
  m_hnvtx   = new TIDA::Histogram<float>( monTool(),  "vx_nvtx" );
  m_hzed    = new TIDA::Histogram<float>( monTool(),  "vx_zed" );
  m_hx      = new TIDA::Histogram<float>( monTool(),  "vx_x" );
  m_hy      = new TIDA::Histogram<float>( monTool(),  "vx_y" );
  m_hntrax  = new TIDA::Histogram<float>( monTool(),  "vx_ntrax" );
 
 
  m_hnvtx_rec  = new TIDA::Histogram<float>( monTool(),  "vx_nvtx_rec" );
  m_hzed_rec   = new TIDA::Histogram<float>( monTool(),  "vx_zed_rec" );
  m_hx_rec     = new TIDA::Histogram<float>( monTool(),  "vx_x_rec" );
  m_hy_rec     = new TIDA::Histogram<float>( monTool(),  "vx_y_rec" );
  m_hntrax_rec = new TIDA::Histogram<float>( monTool(),  "vx_ntrax_rec" );
  
  m_hzed_res = new TIDA::Histogram<float>( monTool(),  "vx_zed_res" );
  m_hx_res   = new TIDA::Histogram<float>( monTool(),  "vx_x_res" );
  m_hy_res   = new TIDA::Histogram<float>( monTool(),  "vx_y_res" );
 
  m_rdz_vs_zed    = new TIDA::Histogram<float>( monTool(),  "vx_rdz_vs_zed" );
  m_rdz_vs_ntrax  = new TIDA::Histogram<float>( monTool(),  "vx_rdz_vs_ntrax" );
  m_rdz_vs_nvtx   = new TIDA::Histogram<float>( monTool(),  "vx_rdz_vs_nvtx" );
  
  m_eff_zed   = new TIDA::Histogram<float>( monTool(),  "vx_zed_eff" );
  m_eff_ntrax = new TIDA::Histogram<float>( monTool(),  "vx_ntrax_eff" );
  m_eff_nvtx  = new TIDA::Histogram<float>( monTool(),  "vx_nvtx_eff" );
  m_eff_mu    = new TIDA::Histogram<float>( monTool(),  "vx_mu_eff" );
  m_eff_lb    = new TIDA::Histogram<float>( monTool(),  "vx_lb_eff" );
  
 
}
 
 
void VtxR3Analysis::execute( const std::vector<TIDA::Vertex*>& vtx0,
               const std::vector<TIDA::Vertex*>& vtx1,
               const TIDA::Event* tevt ) { 
 
  //  std::cout << "VtxAnalysis::execute() initialised: " << m_initialised << "\tevent: " << tevt << std::endl;
 
  if ( !m_initialised ) return;
 
  //  if ( vtx1.size()<2 ) return;
 
#if 0
    std::cout << "VtxAnalysis::execute() " << name()
          << "\tvtx0.size() " << vtx0.size()
          << "\tvtx1.size() " << vtx1.size()
          << std::endl;
#endif
 
 
    VertexMatcher m("vtx_matcher", 10 );
 
    m.match( vtx0, vtx1 );
 
    m_hnvtx->Fill( vtx0.size() );
    m_hnvtx_rec->Fill( vtx1.size() );
  
    //    for ( unsigned i=0 ; i<vtx0.size() ; i++ ) { 
    for ( unsigned i=0 ; i<vtx0.size() ; i++ ) { 
 
      if ( i>0 ) break;
 
      //      std::cout << i << "\tref z " << vtx0[i]->z();
 
      m_hzed->Fill( vtx0[i]->z() );
      m_hx->Fill( vtx0[i]->x() );
      m_hy->Fill( vtx0[i]->y() );
 
 
      m_hntrax->Fill( vtx0[i]->Ntracks() );
 
      const TIDA::Vertex* mv = m.matched( vtx0[i] ); 
 
      if ( mv ) { 
 
    //  std::cout << "\ttest z " << mv->z() << "  : delta z " << (mv->z()-vtx0[i]->z()) << std::endl;
      
    
    m_hzed_rec->Fill( mv->z() );
    m_hx_rec->Fill( mv->x() );
    m_hy_rec->Fill( mv->y() );
    m_hntrax_rec->Fill( mv->Ntracks() );
 
    m_hzed_res->Fill( mv->z() - vtx0[i]->z() );
    m_hx_res->Fill( mv->x() - vtx0[i]->x() );
    m_hy_res->Fill( mv->y() - vtx0[i]->y() );
       
    m_rdz_vs_zed->Fill(   vtx0[i]->z(),       mv->z() - vtx0[i]->z() );
    m_rdz_vs_ntrax->Fill( vtx0[i]->Ntracks(), mv->z() - vtx0[i]->z() );
    m_rdz_vs_nvtx->Fill( vtx0.size(),  mv->z() - vtx0[i]->z() ); 
 
    m_eff_zed->Fill( vtx0[i]->z(), 1 );
    m_eff_ntrax->Fill( vtx0[i]->Ntracks(), 1 );
    m_eff_nvtx->Fill( vtx0.size(), 1 );
 
    m_eff_mu->Fill( tevt->mu(), 1 );
    m_eff_lb->Fill( tevt->lumi_block(), 1 );
 
      }
      else {
    //  std::cout << "\t" << "------" << std::endl;
 
    m_eff_zed->Fill( vtx0[i]->z(), 0 );
 
    m_eff_ntrax->Fill( vtx0[i]->Ntracks(), 0 );
    m_eff_nvtx->Fill( vtx0.size(), 0 );
    
    m_eff_mu->Fill( tevt->mu(), 0 );
    m_eff_lb->Fill( tevt->lumi_block(), 0 );
      }
      
    }
 
}
 
 
 
  
void VtxR3Analysis::finalise() { 
  //  std::cout << "VtxAnalysis::finalise() " << name() << std::endl;
}