SigAnalysis.h

Go to the documentation of this file.

/* emacs: this is -*- c++ -*- */
#ifndef TIDAEXAMPLE_SIGANALYSIS_H
#define TIDAEXAMPLE_SIGANALYSIS_H
 
#include <iostream>
#include <vector>
#include <map>
 
#include "TrigInDetAnalysis/TrackAnalysis.h"
#include "TrigInDetAnalysis/Track.h"
#include "TrigInDetAnalysis/Efficiency1D.h"
#include "TrigInDetAnalysis/TIDDirectory.h"
 
#include "TH1F.h"
#include "TH2F.h"
 
 
class SigAnalysis : public TrackAnalysis { 
   
public:
 
  using TrackAnalysis::execute;
 
public:
  
  SigAnalysis( const std::string& name ) : TrackAnalysis( name ), m_Nreco(0), m_print(true), m_icount(0), m_dir(0) {
    std::cout << "SigAnalysis::SigAnalysis() " << TrackAnalysis::name() << std::endl;
  }  
 
  ~SigAnalysis() {
    std::cout << "SigAnalysis::~SigAnalysis() " << name() << std::endl;
    std::map<std::string, TH1F*>::iterator hitr=m_histos.begin();
    std::map<std::string, TH1F*>::iterator hend=m_histos.end();
    for ( ; hitr!=hend ; hitr++ ) delete hitr->second;     
 
    Efficiency1D* heff[4]    = {    m_eff_pt,    m_eff_eta,    m_eff_phi,    m_eff_z0 };
    Efficiency1D* hpurity[4] = { m_purity_pt, m_purity_eta, m_purity_phi, m_purity_z0 };
 
    for ( int i=4 ; i-- ; ) { delete heff[i]; delete hpurity[i]; } 
 
    delete m_eff_ptp;
    delete m_eff_ptm;
 
  }  
  
  
 
  virtual void initialise();
 
  virtual void execute(const std::vector<TIDA::Track*>& reftracks,
                       const std::vector<TIDA::Track*>& testtracks,
                       TrackAssociator* matcher );
 
  virtual void finalise();
  
  void setprint(bool p) { m_print=p; }
 
private:
  
  void addHistogram( TH1F* h ) { 
      std::string name = h->GetName();
      m_histos.insert( std::map<std::string, TH1F*>::value_type( name, h) );
  }
  
  TH1F* find( const std::string& n ) { 
      std::map<std::string, TH1F*>::iterator hmitr = m_histos.find(n);
      if ( hmitr!=m_histos.end() ) return hmitr->second;
      else                         return 0; 
  }
 
private:
 
  std::map<std::string, TH1F*> m_histos;
 
  Efficiency1D* m_eff_pt = nullptr;
  Efficiency1D* m_eff_ptp = nullptr;
  Efficiency1D* m_eff_ptm = nullptr;
 
  Efficiency1D* m_eff_eta = nullptr;
  Efficiency1D* m_eff_phi = nullptr;
  Efficiency1D* m_eff_z0 = nullptr;
  Efficiency1D* m_eff_d0 = nullptr;
  Efficiency1D* m_eff_a0 = nullptr;
 
  Efficiency1D* m_purity_pt = nullptr;
  Efficiency1D* m_purity_eta = nullptr;
  Efficiency1D* m_purity_phi = nullptr;
  Efficiency1D* m_purity_z0 = nullptr;
  Efficiency1D* m_purity_d0 = nullptr;
  Efficiency1D* m_purity_a0 = nullptr;
 
 
  TH2F* m_h2 = nullptr;
  TH2F* m_h2m = nullptr;
  TH2F* m_h2r = nullptr;
 
  int m_Nreco;
 
  bool m_print;
 
  int m_icount;
 
  TIDDirectory* m_dir;
 
};
 
 
 
inline std::ostream& operator<<( std::ostream& s, const SigAnalysis& /*_s*/ ) { 
  return s;
}
 
 
 
 
 
#endif  // TIDAEXAMPLE_SIGANALYSIS_H