ParticleIDBase.h

Go to the documentation of this file.

#ifndef ParticleID_ParticleIDBase_H
#define ParticleID_ParticleIDBase_H
//
// define the standard interface of ParticleID package
// PID-sys includes: dE/dx, TOF endcap, TOF barrel inner, TOF barrel outer,
//                   TOF Q, EMC energy deposition and Maybe Muon counter
// the development of particle ID package should follow the rule defined
// in this class.
//
// Author: K.L. He    11/11/2005, created
//
//
#include <string>
 
#ifndef BEAN
#include "EvtRecEvent/EvtRecTrack.h"
#else
#include "DstEvtRecTracks.h"
#endif
 
class ParticleIDBase {
 
 public:
 
  ParticleIDBase();
  virtual ~ParticleIDBase(){;}
 
  virtual void init() = 0;
  virtual void calculate() = 0;
  virtual bool IsPidInfoValid() const = 0;
  virtual double chi(int n) const = 0;
  virtual double prob(int n) const = 0;
  virtual int ndof() const = 0;
//  virtual int neuronPID() const = 0;       //neuron pid
 
  EvtRecTrack* PidTrk() const {return m_trk;}
  void setRecTrack(EvtRecTrack* trk) {m_trk = trk;}
  double chiMinCut() const {return m_chimin_cut;}
  void setChiMinCut(const double chi = 4) {m_chimin_cut = chi;}
  double pdfMinSigmaCut() const {return m_pdfsigmamin_cut;}
  void setPdfMinSigmaCut(const double pdf =4) {m_pdfsigmamin_cut= pdf;}
  double getRunNo() const {return m_runno;}
  void setRunNo(const double runh = 8093) {m_runno = runh;}
 
 
  double p();
  double pt();
  double charge();
 
 public:  // utilities
 
  double xmass(int n);
  double velc();
  double probCalculate(double chi2, int n);
  double pdfCalculate(double offset,double sigma);
  double interpolation(double* x, double* y,double x1);
  double pol2(double x,double* par);
  double pol3(double x,double* par);
  double pol4(double x,double* par);
 
 public:  // specific sub pid sys
 
  int useDedx() const {return USE_DEDX;}
  int useTof() const {return USE_TOF;}
  int useTof1() const {return USE_TOF1;}
  int useTof2() const {return USE_TOF2;}
  int useTofE() const {return USE_TOFE;}
  int useTofQ() const {return USE_TOFQ;}
  int useTofC() const {return USE_TOFC;}
  int useTofCorr() const {return USE_TOFCorr;}
  int useEmc()  const {return USE_EMC;}
  int useMuc()  const {return USE_MUC;}
  int useMrpc() const {return USE_MRPC;}
 
 public: // identify which kind of particles
  int onlyPionKaon() const {return (IDENTIFY_PION | IDENTIFY_KAON);}
  int onlyPionKaonProton() const {return (IDENTIFY_PION | IDENTIFY_KAON | IDENTIFY_PROTON);}
  int onlyPionKaonElectron() const {return (IDENTIFY_PION | IDENTIFY_KAON | IDENTIFY_ELECTRON);}
  int all()  const {return (IDENTIFY_ELECTRON | IDENTIFY_PION | IDENTIFY_KAON | IDENTIFY_PROTON);}
  int onlyElectron() const {return IDENTIFY_ELECTRON;}
  int onlyMuon() const {return IDENTIFY_MUON;}
  int onlyPion() const {return IDENTIFY_PION;}
  int onlyKaon() const {return IDENTIFY_KAON;}
  int onlyProton() const {return IDENTIFY_PROTON;}
 
 public: // Particle ID method
  int methodLikelihood() const {return LIKELIHOOD_PID;}
  int methodProbability() const {return PROBABILITY_PID;}
  int methodNeuronNetwork() const {return NEURONNETWORK_PID;}
 public: //
  int dedxValid() const {return DEDX_VALID;}
  int tofValid() const {return TOF_VALID;}
  int tofeValid() const {return TOFE_VALID;}
  int tofqValid() const {return TOFQ_VALID;}
  int tofcValid() const {return TOFC_VALID;}
  int tofcorrValid() const {return TOFCorr_VALID;}
  int emcValid() const {return EMC_VALID;}
  int mucValid() const {return MUC_VALID;}
  int MrpcValid() const {return MRPC_VALID;}
 protected:
  // path to dir with calibration/configuration/etc data
  static std::string path;
 public:
  void set_path(const char* s_path = 0);
  void set_path(std::string s_path) {path = s_path;}
 
 protected:
  // PID calculation routines(include PID correction)
  virtual int particleIDCalculation() = 0;
//  virtual int neuronPIDCalculation() = 0;
  //huangb add
  virtual int LikelihoodCalculation() =0;
 
 private:
  EvtRecTrack* m_trk;
  double m_runno;
  double m_chimin_cut;
  double m_pdfsigmamin_cut; //the cut is multiple of the sigma. ie. if(m_pdfsigmamin_cut ==3), the cut is 3sigma
  static const int USE_DEDX;
  static const int USE_TOF1;
  static const int USE_TOF2;
  static const int USE_TOF;
  static const int USE_TOFE;
  static const int USE_TOFQ;
  static const int USE_TOFC;
  static const int USE_TOFCorr;
  static const int USE_EMC;
  static const int USE_MUC;
  static const int USE_MRPC;
 
  static const int IDENTIFY_ELECTRON;
  static const int IDENTIFY_MUON;
  static const int IDENTIFY_PION;
  static const int IDENTIFY_KAON;
  static const int IDENTIFY_PROTON;
 
  static const int PROBABILITY_PID;
  static const int LIKELIHOOD_PID;
  static const int NEURONNETWORK_PID;
 
  static const int DEDX_VALID;
  static const int TOF_VALID;
  static const int TOF1_VALID;
  static const int TOF2_VALID;
  static const int TOFE_VALID;
  static const int TOFQ_VALID;
  static const int TOFC_VALID;
  static const int TOFCorr_VALID;
  static const int EMC_VALID;
  static const int MUC_VALID;
  static const int MRPC_VALID;
};
 
#endif