TrackSelector.cxx

Go to the documentation of this file.

 
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODEgamma/ElectronContainer.h"
 
#include "TrackSelector.h"

#include "find_bin.h"
 
IDTPM2::TrackSelector::TrackSelector( const std::string& name )
  : GenericTrackSelector( name ) { }
 
 
 
StatusCode IDTPM2::TrackSelector::initialize() {
  ATH_CHECK( GenericTrackSelector::initialize() );  
  if ( m_useOfflineTool ) ATH_CHECK( m_offlineTool.retrieve() );
  return StatusCode::SUCCESS;
}
 

 
template<typename T, typename U=T> 
bool select_min( const T& x, const std::vector<T>& xbins,
         const U& y, const std::vector<U>& ycuts ) { return y>=ycuts.at( IDTPM2::bins::find_bin( xbins, x ) ); }
 
template<typename T, typename U=T> 
bool select_max( const T& x, const std::vector<T>& xbins,
         const U& y, const std::vector<U>& ycuts ) { return y<=ycuts.at( IDTPM2::bins::find_bin( xbins, x ) ); }
 
 
  
bool IDTPM2::TrackSelector::selectTrack( const TrackView& t ) const {
 
  if ( m_useOfflineTool ) {
    const xAOD::TrackParticle* track = t.as<xAOD::TrackParticle>();
    
    if ( !track ) {
      const xAOD::Electron* electron = t.as<xAOD::Electron>();
      if ( electron ) track = electron->trackParticle();
      else return false;
    }
    
    if ( ! m_offlineTool->accept( track ) ) return false; // TODO vertex needs to be provided here
  }
  
  if ( !GenericTrackSelector::selectTrack(t) ) return false;

  if ( !m_etaBins.empty() ) {
    // if (!m_minHitsVec.empty() && !nHitsSelVec(*track, m_minHitsVec, m_etaBins))  return false;
    // if (!m_minPtVec.empty() && !minPtSelVec(*track, m_minPtVec, m_etaBins))  return false;
    // if (!m_maxD0Vec.empty() && !maxD0SelVec(*track, m_maxD0Vec, m_etaBins))  return false;
    // if (!m_maxZ0Vec.empty() && !maxZ0SelVec(*track, m_maxZ0Vec, m_etaBins))  return false;
 
    //  if (!m_minHitsVec.empty() && !vector_select( track, m_minHitsVec, m_etaBins))  return false;
    //  if (!m_minPtVec.empty() && !minPtSelVec(*track, m_minPtVec, m_etaBins))  return false;
    //  if (!m_maxD0Vec.empty() && !maxD0SelVec(*track, m_maxD0Vec, m_etaBins))  return false;
    //  if (!m_maxZ0Vec.empty() && !maxZ0SelVec(*track, m_maxZ0Vec, m_etaBins))  return false;

    if ( !m_minHitsVec.empty() && !select_min<float,unsigned>( t.eta(), m_etaBins, t.nSi(), m_minHitsVec.value() ))  return false;
    if ( !m_minPtVec.empty()   && !select_min<float>( t.eta(), m_etaBins, t.pt(),  m_minPtVec.value() ))    return false;
    if ( !m_maxZ0Vec.empty()   && !select_max<float>( t.eta(), m_etaBins, std::fabs(t.z0()), m_maxZ0Vec.value() ))  return false;
    if ( !m_maxD0Vec.empty()   && !select_max<float>( t.eta(), m_etaBins, std::fabs(t.d0()), m_maxD0Vec.value() ))  return false;
 
  }
  
  return true;
}