LocalProtonSelector.cxx

Go to the documentation of this file.

#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/IJobOptionsSvc.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/PropertyMgr.h"
 
#include "EventModel/EventModel.h"
#include "EventModel/EventHeader.h"
#include "EvtRecEvent/EvtRecEvent.h"
#include "EvtRecEvent/EvtRecTrack.h"
#include "ParticleID/ParticleID.h"
#include "DTagAlg/LocalProtonSelector.h"
 
#include "VertexFit/IVertexDbSvc.h"
#include "VertexFit/VertexFit.h"
#include "VertexFit/Helix.h"
 
LocalProtonSelector::LocalProtonSelector()
{
        m_pidtype=0;
 
        IJobOptionsSvc* jobSvc;
        Gaudi::svcLocator()->service("JobOptionsSvc", jobSvc);
 
        PropertyMgr m_propMgr;
 
        //Declare the properties
        m_propMgr.declareProperty("RxyCut",               m_VrCut           = 1.0);
        m_propMgr.declareProperty("Vz0Cut",               m_VzCut           = 10.0);
        m_propMgr.declareProperty("CosThetaCut",          m_CosThetaCut     = 0.93);
 
        m_propMgr.declareProperty("PID",                  m_PID     = true);
        m_propMgr.declareProperty("ChiCut",               m_ChiCut  = 8);
 
        m_propMgr.declareProperty("PidUseDedx",           m_useDedx         = true);
        m_propMgr.declareProperty("PidUseTof1",           m_useTof1         = true);
        m_propMgr.declareProperty("PidUseTof2",           m_useTof2         = true);
        m_propMgr.declareProperty("PidUseTofE",           m_useTofE         = false);
        m_propMgr.declareProperty("PidUseTofQ",           m_useTofQ         = false);
        m_propMgr.declareProperty("PidUseEmc",            m_useEmc          = false);
        m_propMgr.declareProperty("PidUseMuc",            m_useMuc          = false);
 
        m_propMgr.declareProperty("PidwithProbability",   m_probability     = true);
        m_propMgr.declareProperty("PidProbCut",           m_PidProbCut      = 0.001);
        m_propMgr.declareProperty("RejectKaon",           m_rejectKaon      = true);
        m_propMgr.declareProperty("RejectPion",           m_rejectPion      = true);
 
        m_propMgr.declareProperty("PidwithLikelihood",    m_likelihood      = false);
        m_propMgr.declareProperty("PidwithNeuronNetwork", m_neuronNetwork   = false);
        m_neuronValCut.clear();
        m_neuronValCut.push_back(1.5);
        m_neuronValCut.push_back(2.5);
        m_propMgr.declareProperty("NeuronValue",      m_neuronValCut);
 
        jobSvc->setMyProperties("LocalProtonSelector", &m_propMgr);
}
 
bool LocalProtonSelector::operator() (CDProton& aProton) {
 
        aProton.setUserTag(1);
        EvtRecTrack* recTrk = const_cast<EvtRecTrack*>( aProton.track() );
 
        // MDC track selection
 
        RecMdcKalTrack::setPidType(RecMdcKalTrack::proton);
        if ( !recTrk->isMdcKalTrackValid() ) return false;
        RecMdcKalTrack* mdcKalTrk = recTrk->mdcKalTrack();
        if ( mdcKalTrk->charge()==0 ) return false;
 
        Hep3Vector xorigin(0,0,0);
        IVertexDbSvc*  vtxsvc;
        Gaudi::svcLocator()->service("VertexDbSvc", vtxsvc);
        if(vtxsvc->isVertexValid()){
                double* dbv = vtxsvc->PrimaryVertex();
                double*  vv = vtxsvc->SigmaPrimaryVertex();
                xorigin.setX(dbv[0]);
                xorigin.setY(dbv[1]);
                xorigin.setZ(dbv[2]);
        }
        HepVector a = mdcKalTrk->getZHelixP();
        HepSymMatrix Ea = mdcKalTrk->getZErrorP();
        HepPoint3D point0(0.,0.,0.);
        HepPoint3D IP(xorigin[0],xorigin[1],xorigin[2]);
        VFHelix helixip3(point0,a,Ea);
        helixip3.pivot(IP);
        HepVector  vecipa = helixip3.a();
 
        double dr=fabs(vecipa[0]);
        double dz=fabs(vecipa[3]);
        double costheta=cos(mdcKalTrk->theta());
        if (  dr>= m_VrCut ) return false;
        if (  dz>= m_VzCut ) return false;
        if ( fabs(costheta) >= m_CosThetaCut ) return false;
 
        if(!m_PID) return true;
 
 
        //ParticleID for Proton
        ParticleID *pid = ParticleID::instance();
        pid->init();
        pid->setMethod(pid->methodProbability());
        pid->setChiMinCut(m_ChiCut);
        pid->setRecTrack(recTrk);
        //pid->usePidSys(pid->useDedx() | pid->useTof1() | pid->useTof2() | pid->useTof());
    pid->usePidSys(pid->useDedx() | pid->useTofCorr() );
        pid->identify(pid->onlyPion() | pid->onlyKaon() | pid->onlyProton());
        pid->calculate();
 
 
        if(!(pid->probProton() > 0. && pid->probProton() > pid->probPion() && pid->probProton() > pid->probKaon())) aProton.setUserTag(2);
 
        if(m_pidtype==0) return true;
 
        if(aProton.userTag()==1)
                return true;
        else
                return false;
 
}
 
LocalProtonSelector protonSelector;