LocalProtonSelector.cxx

Go to the documentation of this file.

#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/ISvcLocator.h"
 
#include "EvtRecEvent/EvtRecTrack.h"
#include "LocalProtonSelector.h"
#include "ParticleID/ParticleID.h"
 
#include "VertexDbSvc/IVertexDbSvc.h"
#include "VertexFit/Helix.h"
 
DECLARE_COMPONENT( LocalProtonSelector )
 
LocalProtonSelector::LocalProtonSelector( const std::string& type, const std::string& name,
                                          const IInterface* parent )
    : base_class( type, name, parent ) {
  m_pidtype = 0;
 
  // Declare the properties
  declareProperty( "RxyCut", m_VrCut = 1.0 );
  declareProperty( "Vz0Cut", m_VzCut = 10.0 );
  declareProperty( "CosThetaCut", m_CosThetaCut = 0.93 );
 
  declareProperty( "PID", m_PID = true );
  declareProperty( "ChiCut", m_ChiCut = 8 );
 
  declareProperty( "PidUseDedx", m_useDedx = true );
  declareProperty( "PidUseTof1", m_useTof1 = true );
  declareProperty( "PidUseTof2", m_useTof2 = true );
  declareProperty( "PidUseTofE", m_useTofE = false );
  declareProperty( "PidUseTofQ", m_useTofQ = false );
  declareProperty( "PidUseEmc", m_useEmc = false );
  declareProperty( "PidUseMuc", m_useMuc = false );
 
  declareProperty( "PidwithProbability", m_probability = true );
  declareProperty( "PidProbCut", m_PidProbCut = 0.001 );
  declareProperty( "RejectKaon", m_rejectKaon = true );
  declareProperty( "RejectPion", m_rejectPion = true );
 
  declareProperty( "PidwithLikelihood", m_likelihood = false );
  declareProperty( "PidwithNeuronNetwork", m_neuronNetwork = false );
  declareProperty( "NeuronValue", m_neuronValCut = { 1.5, 2.5 } );
}
 
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;
  StatusCode    sc = serviceLocator()->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;