#include <MucPID.h>

Inheritance diagram for MucPID:

Public Member Functions
void	init ()

void	calculate ()

bool	IsPidInfoValid () const

double	chi (int n) const

double	prob (int n) const

int	ndof () const

double	hits () const

double	depth () const

double	chi2 () const

double	distance () const

double	delta_phi () const

double	val_muc1 () const

int	neuronPID () const

Public Member Functions inherited from ParticleIDBase
	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

EvtRecTrack *	PidTrk () const

void	setRecTrack (EvtRecTrack *trk)

double	chiMinCut () const

void	setChiMinCut (const double chi=4)

double	chiMaxCut () const

void	setChiMaxCut (const double chi=6)

double	pdfMinSigmaCut () const

void	setPdfMinSigmaCut (const double pdf=4)

double	getRunNo () const

void	setRunNo (const double runh=8093)

double	p ()

double	pt ()

double	charge ()

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)

int	useDedx () const

int	useTof () const

int	useTof1 () const

int	useTof2 () const

int	useTofE () const

int	useTofQ () const

int	useTofC () const

int	useTofCorr () const

int	useEmc () const

int	useMuc () const

int	onlyPionKaon () const

int	onlyPionKaonProton () const

int	onlyPionKaonElectron () const

int	all () const

int	onlyElectron () const

int	onlyMuon () const

int	onlyPion () const

int	onlyKaon () const

int	onlyProton () const

int	methodLikelihood () const

int	methodProbability () const

int	methodNeuronNetwork () const

int	dedxValid () const

int	tofValid () const

int	tofeValid () const

int	tofqValid () const

int	tofcValid () const

int	tofcorrValid () const

int	emcValid () const

int	mucValid () const

void	set_path (const char *s_path=0)

void	set_path (std::string s_path)

Static Public Member Functions
static MucPID *	instance ()

Protected Member Functions
int	particleIDCalculation ()

int	neuronPIDCalculation ()

int	LikelihoodCalculation ()

virtual int	particleIDCalculation ()=0

virtual int	LikelihoodCalculation ()=0

Additional Inherited Members
Static Protected Attributes inherited from ParticleIDBase
static std::string	path = ""

Detailed Description

Definition at line 20 of file MucPID.h.

Member Function Documentation

◆ calculate()

void MucPID::calculate ( )

virtual

Implements ParticleIDBase.

Definition at line 101 of file MucPID.cxx.

                       {
   if(particleIDCalculation() == 0) m_ndof = 1;
}

◆ chi()

double MucPID::chi ( int n ) const

inlinevirtual

Implements ParticleIDBase.

Definition at line 29 of file MucPID.h.

29{return m_chi[n];}

◆ chi2()

double MucPID::chi2 ( ) const

inline

Definition at line 34 of file MucPID.h.

34{return m_chi2;}

Referenced by init().

◆ delta_phi()

double MucPID::delta_phi ( ) const

inline

Definition at line 36 of file MucPID.h.

36{return m_muc_delta_phi;}

◆ depth()

double MucPID::depth ( ) const

inline

Definition at line 33 of file MucPID.h.

33{return m_depth;}

Referenced by init().

◆ distance()

double MucPID::distance ( ) const

inline

Definition at line 35 of file MucPID.h.

35{return m_distance;}

Referenced by init().

◆ hits()

double MucPID::hits ( ) const

inline

Definition at line 32 of file MucPID.h.

32{return m_hits;}

Referenced by init().

◆ init()

void MucPID::init ( )

virtual

Implements ParticleIDBase.

Definition at line 67 of file MucPID.cxx.

                  {
   for(int i = 0; i < 5; i++) {
      m_chi[i] = 99.0;
      m_prob[i] = -1.0;
   }
   m_chimin = 99.;
   m_ndof = 0;
   m_hits = -99;
   m_depth = -999;
   m_val_muc=-99;
 
   std::string neural = path + "/share/neural.root";
   std::string muc = path + "/share/muc.txt";
   double ptrk,phi,theta,depth,hits,chi2,distance,muc_delta_phi;
   int type;
   if(!m_trainTree_muc) {
      m_trainTree_muc = new TTree("m_trainTree_muc","m_trainTree_muc");
      m_trainTree_muc->Branch("ptrk",&ptrk,"ptrk/D");
      m_trainTree_muc->Branch("phi",&phi,"phi/D");
      m_trainTree_muc->Branch("theta",&theta,"theta/D");
      m_trainTree_muc->Branch("depth",&depth,"depth/D");
      m_trainTree_muc->Branch("hits",&hits,"hits/D");
      m_trainTree_muc->Branch("chi2",&chi2,"chi2/D");
      m_trainTree_muc->Branch("distance",&distance,"distance/D");
      m_trainTree_muc->Branch("muc_delta_phi",&muc_delta_phi,"muc_delta_phi/D");
      m_trainTree_muc->Branch("type",&type,"type/I");
   }
   if(!m_mlp_muc) {
      m_mlp_muc = new TMultiLayerPerceptron("@ptrk,@phi,@theta,@depth,@hits,@chi2,@distance,@muc_delta_phi:16:type",m_trainTree_muc);
      m_mlp_muc->LoadWeights(muc.c_str());
   }
}

Referenced by ParticleID::init().

◆ instance()

MucPID * MucPID::instance ( )

static

Definition at line 22 of file MucPID.cxx.

                          {
 
   if(!m_pointer) m_pointer = new MucPID();
   return m_pointer;
}

Referenced by ParticleID::init().

◆ IsPidInfoValid()

bool MucPID::IsPidInfoValid ( ) const

inlinevirtual

Implements ParticleIDBase.

Definition at line 28 of file MucPID.h.

28{return (m_ndof> 0);}

◆ LikelihoodCalculation()

int MucPID::LikelihoodCalculation ( )

inlineprotectedvirtual

Implements ParticleIDBase.

Definition at line 46 of file MucPID.h.

46{return -1;}

◆ ndof()

int MucPID::ndof ( ) const

inlinevirtual

Implements ParticleIDBase.

Definition at line 31 of file MucPID.h.

31{return m_ndof;}

◆ neuronPID()

int MucPID::neuronPID ( ) const

inline

Definition at line 39 of file MucPID.h.

39{return -1;}

◆ neuronPIDCalculation()

int MucPID::neuronPIDCalculation ( )

inlineprotected

Definition at line 45 of file MucPID.h.

45{return -1;}

◆ particleIDCalculation()

int MucPID::particleIDCalculation ( )

protectedvirtual

Implements ParticleIDBase.

Definition at line 105 of file MucPID.cxx.

                                  {
   int irc = -1;
   EvtRecTrack* recTrk = PidTrk();
   if(!(recTrk->isMdcTrackValid())) return irc;
   RecMdcTrack* mdcTrk = recTrk->mdcTrack();
 
   m_depth = -99;
   m_hits =  -99;
   m_chi2 =-99;
   m_distance =-99;
   m_muc_delta_phi =-99;
 
   double ptrk = mdcTrk->p();
   double m_ptrk = ptrk;
   double m_pt = mdcTrk->pxy();
   double phi = mdcTrk->phi();
   double theta = mdcTrk->theta();
   double cost = cos(mdcTrk->theta());
   if(ptrk<0.5) return irc;
   if(fabs(cost)>0.83) return irc;
   if(!(recTrk->isExtTrackValid())) return irc;
   RecExtTrack* extTrk = recTrk->extTrack();
   if(extTrk->mucVolumeNumber() == -1) return irc;
   if (!(recTrk->isMucTrackValid())) return irc;
   RecMucTrack* mucTrk = recTrk->mucTrack();
 
   //  if(mucTrk->maxHitsInLayer()< 0) return irc;
   if(mucTrk->depth()>100000) return irc;
 
   m_hits = mucTrk->maxHitsInLayer();
   m_depth = mucTrk->depth();
   m_distance = mucTrk->distance();
   m_chi2 = mucTrk->chi2();
   /* Hep3Vector phi_muc;
    phi_muc.set(mucTrk->xPos(),mucTrk->yPos(),0);
    Hep3Vector phi_mdc;
    phi_mdc.set(mdcTrk->px(),mdcTrk->py(),0);
    m_muc_delta_phi = phi_muc.angle(phi_mdc);
    if(m_muc_delta_phi<0) m_muc_delta_phi = -m_muc_delta_phi;        */
   m_muc_delta_phi= mucTrk->deltaPhi();
   m_muc_delta_phi=3.14159-m_muc_delta_phi;
   theta = cos(theta);
   params_muc1[0] = m_ptrk;
   params_muc1[1] = phi;
   params_muc1[2] = theta;
   params_muc1[3] = m_depth;
   params_muc1[4] = m_hits;
   params_muc1[5] = m_chi2;
   params_muc1[6] = m_distance;
   params_muc1[7] = m_muc_delta_phi;
 
   m_val_muc = m_mlp_muc->Evaluate(0,params_muc1);
   if(m_pt<0) m_pt = -m_pt;
   int pindex = int((m_ptrk-0.5)/0.1);
   int bindex = int((m_val_muc-0.5)/0.01);
   if(bindex>300||bindex<0) return irc;
   if(pindex>11) pindex=11;
   if(pindex<0) pindex=0;
   m_prob[0] = m_p_h[pindex][bindex];;
   m_prob[1] = m_m_h[pindex][bindex];
   m_prob[2] = m_p_h[pindex][bindex];
   m_prob[3] = m_p_h[pindex][bindex];
   m_prob[4] = m_p_h[pindex][bindex];
   for(int i =0; i<5; i++) {
      if(m_prob[i]==0) m_prob[i] = 0.001;
   }
   float ppp[5];
   for(int i =0; i<5; i++) {
      ppp[i] = 0.0;
   }
 
   for(int j=0; j<bindex; j++) {
      ppp[1]+= m_m_h[pindex][j]*0.01;
      ppp[2]+= m_p_h[pindex][j]*0.01;
   }
   if(ppp[1]>0&&ppp[1]<1)
      CALG(ppp[1],m_chi[1]);
   if(ppp[2]>0&&ppp[2]<1)
      CALG(ppp[2],m_chi[2]);
   if(ppp[1]<=0||ppp[1]>=1)
      m_chi[1]=-99;
   if(ppp[2]<=0||ppp[2]>=1)
      m_chi[2]=-99;
 
   m_chi[3]=m_chi[2];
   m_chi[4]=m_chi[2];
   m_chi[0] =m_chi[2];
   m_ndof = 1;
   irc = 0;
   return irc;
}

Referenced by calculate().

◆ prob()

double MucPID::prob ( int n ) const

inlinevirtual

Implements ParticleIDBase.

Definition at line 30 of file MucPID.h.

30{return m_prob[n];}

◆ val_muc1()

double MucPID::val_muc1 ( ) const

inline

Definition at line 37 of file MucPID.h.

37{return m_val_muc;}

The documentation for this class was generated from the following files:

Public Member Functions

Static Public Member Functions

Protected Member Functions

Additional Inherited Members

Detailed Description

Member Function Documentation

◆ calculate()

◆ chi()

◆ chi2()

◆ delta_phi()

◆ depth()

◆ distance()

◆ hits()

◆ init()

◆ instance()

◆ IsPidInfoValid()

◆ LikelihoodCalculation()

◆ ndof()

◆ neuronPID()

◆ neuronPIDCalculation()

◆ particleIDCalculation()

◆ prob()

◆ val_muc1()