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    : [email protected]
//-----------------------------------------------------------------------------
// Description : A class to select a TTrackBase object.
//               See http://bsunsrv1.kek.jp/~yiwasaki/tracking/
//-----------------------------------------------------------------------------
 
#include "TrkReco/TMSelector.h"
#include "TrkReco/TTrackBase.h"
#include "TrkReco/TTrack.h"
#include "TrkReco/TCircle.h"
#include "TrkReco/TMLink.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;
}