TMSelector.cxx

Go to the documentation of this file.

//-----------------------------------------------------------------------------
// $Id: TMSelector.cxx,v 1.1 2006/10/12 07:52:00 liuqg Exp $
//-----------------------------------------------------------------------------
// Filename : TMSelector.cc
// Section  : Tracking
// Owner    : Yoshi Iwasaki
// Email    : yoshihito.iwasaki@kek.jp
//-----------------------------------------------------------------------------
// Description : A class to select a TTrackBase object.
//               See http://bsunsrv1.kek.jp/~yiwasaki/tracking/
//-----------------------------------------------------------------------------
 
#include "TrkReco/TMSelector.h"
#include "TrkReco/TCircle.h"
#include "TrkReco/TMLink.h"
#include "TrkReco/TTrack.h"
#include "TrkReco/TTrackBase.h"
 
TMSelector::TMSelector()
    : _nLinksDefined( false )
    , _nLinks( 0 )
    , _nSuperLayersDefined( false )
    , _nSuperLayers( 0 )
    , _minPtDefined( false )
    , _minPt( 0. )
    , _maxImpactDefined( false )
    , _maxImpact( 0. )
    , _nLinksStereoDefined( false )
    , _nLinksStereo( 0 )
    , _maxDistanceDefined( false )
    , _maxDistance( 0. ) {}
 
TMSelector::TMSelector( const TMSelector& a )
    : _nLinksDefined( a._nLinksDefined )
    , _nLinks( a._nLinks )
    , _nSuperLayersDefined( a._nSuperLayersDefined )
    , _nSuperLayers( a._nSuperLayers )
    , _minPtDefined( a._minPtDefined )
    , _minPt( a._minPt )
    , _maxImpactDefined( a._maxImpactDefined )
    , _maxImpact( a._maxImpact )
    , _nLinksStereoDefined( a._nLinksStereoDefined )
    , _nLinksStereo( a._nLinksStereo )
    , _maxDistanceDefined( a._maxDistanceDefined )
    , _maxDistance( a._maxDistance ) {}
 
TMSelector::~TMSelector() {}
 
bool TMSelector::select( TTrackBase& b ) const {
 
#ifdef TRKRECO_DEBUG_DETAIL
  if ( !b.fitted() ) { std::cout << "    TSelect::select !!! not fitted yet" << std::endl; }
#endif
 
  if ( b.objectType() == Track )
  {
    TTrack& a = (TTrack&)b;
 
    if ( _minPtDefined )
    {
      if ( a.pt() < _minPt )
      {
#ifdef TRKRECO_DEBUG_DETAIL
        std::cout << "    TSelect ... rejected by min. pt(";
        std::cout << a.pt() << ") < ";
        std::cout << _minPt << std::endl;
#endif
        return false;
      }
    }
 
    if ( _maxImpactDefined )
    {
      if ( fabs( a.impact() ) > _maxImpact )
      {
#ifdef TRKRECO_DEBUG_DETAIL
        std::cout << "    TSelect ... rejected by max. impact(";
        std::cout << a.impact() << ") < ";
        std::cout << _maxImpact << std::endl;
#endif
        return false;
      }
    }
  }
  else if ( b.objectType() == Circle )
  {
    TCircle& a = (TCircle&)b;
 
    if ( _minPtDefined )
    {
      if ( a.pt() < _minPt )
      {
#ifdef TRKRECO_DEBUG_DETAIL
        std::cout << "    TSelect ... rejected by min. pt(";
        std::cout << a.pt() << ") < ";
        std::cout << _minPt << std::endl;
#endif
        return false;
      }
    }
 
    if ( _maxImpactDefined )
    {
      if ( fabs( a.impact() ) > _maxImpact )
      {
#ifdef TRKRECO_DEBUG_DETAIL
        std::cout << "    TSelect ... rejected by max. impact(";
        std::cout << a.impact() << ") < ";
        std::cout << _maxImpact << std::endl;
#endif
        return false;
      }
    }
  }
  else
  {
    std::cout << "TMSelector !!! Unknown object type" << std::endl;
    return false;
  }
 
#ifdef TRKRECO_DEBUG_DETAIL
  std::cout << "    TSelect::select ... accepted" << std::endl;
#endif
  return true;
}
 
bool TMSelector::preSelect( const TTrackBase& a ) const {
  if ( _nLinksDefined )
  {
    if ( a.nLinks() < _nLinks )
    {
#ifdef TRKRECO_DEBUG_DETAIL
      std::cout << "    TSelect ... rejected by nLinks(";
      std::cout << a.nLinks() << ") < ";
      std::cout << _nLinks << std::endl;
#endif
      return false;
    }
  }
 
  if ( _nSuperLayersDefined )
  {
    if ( NSuperLayers( a.links() ) < _nSuperLayers )
    {
#ifdef TRKRECO_DEBUG_DETAIL
      std::cout << "    TSelect ... rejected by nSuperLayers(";
      std::cout << NSuperLayers( a.links() ) << ") < ";
      std::cout << _nSuperLayers << std::endl;
#endif
      return false;
    }
  }
 
#ifdef TRKRECO_DEBUG_DETAIL
  std::cout << "    TSelect::preSelect ... accepted" << std::endl;
#endif
  return true;
}