PurityAnalysis.h

Go to the documentation of this file.

/* emacs: this is -*- c++ -*- */
 
 
#ifndef ANALYSIS_PURITYANALYSIS_H
#define ANALYSIS_PURITYANALYSIS_H
 
#include <iostream>
#include <vector>
#include <map>
 
#include "TrigInDetAnalysis/TrackAnalysis.h"
#include "TrigInDetAnalysis/Track.h"
#include "TrigInDetAnalysis/TIDDirectory.h"
#include "TrigInDetAnalysis/Efficiency1D.h"
 
// #include "TrigInDetAnalysis/Efficiency2D.h"
 
#include "Resplot.h"
 
 
#include "TH1F.h"
#include "TH2F.h"
 
 
 
class PurityAnalysis : public TrackAnalysis { 
 
public:
 
  using TrackAnalysis::execute;
 
public:
  
  PurityAnalysis( const std::string& name ) : 
    TrackAnalysis( name ), m_Nreco(0), m_Nref(0), m_Nmatched(0), m_print(false) {
    std::cout << "PurityAnalysis::PurityAnalysis() " << TrackAnalysis::name() << std::endl;
  }  
 
  ~PurityAnalysis() {
    std::cout << "PurityAnalysis::~PurityAnalysis() " << 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:
  TIDDirectory* m_dir = 0;
 
  std::map<std::string, TH1F*> m_histos;
 
  Efficiency1D* m_eff_pt = 0;
  Efficiency1D* m_eff_ptp = 0;
  Efficiency1D* m_eff_ptm = 0;
 
  Efficiency1D* m_eff_eta = 0;
  Efficiency1D* m_eff_phi = 0;
  Efficiency1D* m_eff_z0 = 0;
  Efficiency1D* m_eff_d0 = 0;
  Efficiency1D* m_eff_a0 = 0;
 
  Efficiency1D* m_purity_pt = 0;
  Efficiency1D* m_purity_eta = 0;
  Efficiency1D* m_purity_phi = 0;
  Efficiency1D* m_purity_z0 = 0;
  Efficiency1D* m_purity_d0 = 0;
  Efficiency1D* m_purity_a0 = 0;
 
  //  Efficiency2D* m_eff_pteta;
  //  Efficiency2D* m_eff_etapt;
 
  TH1F* m_hDeltaR = 0;

  int m_Nreco;
  int m_Nref;
  int m_Nmatched;
 
  TH1F* m_hchi2 = 0;

  bool m_print;
 
};
 
 
 
inline std::ostream& operator<<( std::ostream& s, const PurityAnalysis& pa ) {
  return s << pa.name();
}
 
 
 
 
 
#endif  // ANALYSIS_PURITYANALYSIS_H