LocalKaonSelector.cxx

Go to the documentation of this file.

#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/ISvcLocator.h"
 
#include "EvtRecEvent/EvtRecTrack.h"
#include "LocalKaonSelector.h"
#include "ParticleID/ParticleID.h"
 
#include "VertexDbSvc/IVertexDbSvc.h"
#include "VertexFit/Helix.h"
 
#include "SimplePIDSvc/ISimplePIDSvc.h"
 
DECLARE_COMPONENT( LocalKaonSelector )
 
LocalKaonSelector::LocalKaonSelector( 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( "useSimplePID", m_useSimplePID = true );
 
  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( "RejectPion", m_rejectPion = true );
  declareProperty( "RejectProton", m_rejectProton = false );
 
  declareProperty( "PidwithLikelihood", m_likelihood = false );
  declareProperty( "PidwithNeuronNetwork", m_neuronNetwork = false );
  declareProperty( "NeuronValue", m_neuronValCut = { 1.5, 2.5 } );
}
 
bool LocalKaonSelector::operator()( CDChargedKaon& aKaon ) {
 
  aKaon.setUserTag( 1 );
  EvtRecTrack* recTrk = const_cast<EvtRecTrack*>( aKaon.track() );
 
  // MDC track selection
 
  RecMdcKalTrack::setPidType( RecMdcKalTrack::kaon );
  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->getZHelixK();
  HepSymMatrix Ea = mdcKalTrk->getZErrorK();
  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;
 
  // Simple PID
  if ( m_useSimplePID )
  {
    ISimplePIDSvc* m_simplePIDSvc;
    StatusCode     sc = serviceLocator()->service( "SimplePIDSvc", m_simplePIDSvc );
 
    m_simplePIDSvc->preparePID( recTrk );
    if ( !m_simplePIDSvc->iskaon() ) { aKaon.setUserTag( 2 ); }
  }
  else
  {
    // particleID pacakge
    ParticleID* pid = ParticleID::instance();
    pid->init();
    pid->setMethod( pid->methodProbability() );
    pid->setChiMinCut( 4 );
    pid->setRecTrack( recTrk );
    pid->usePidSys( pid->useDedx() | pid->useTofCorr() );
    pid->identify( pid->onlyPion() | pid->onlyKaon() );
    pid->calculate();
 
    if ( !( pid->probKaon() > 0 && pid->probKaon() > pid->probPion() ) ) aKaon.setUserTag( 2 );
  }
 
  if ( m_pidtype == 0 ) return true;
 
  if ( aKaon.userTag() == 1 ) return true;
  else return false;
}
 
// LocalKaonSelector kaonSelector;