ITrackSelector.h

Go to the documentation of this file.

 
#ifndef IDTPM2_ITRACKSELECTOR_H
#define IDTPM2_ITRACKSELECTOR_H
 

#include "AsgTools/IAsgTool.h"
 
#include "TrackView.h"
#include "TagNProbe.h"
 
 
namespace IDTPM2 {
 
class ITrackSelector : virtual public asg::IAsgTool {
 
public:
 
  ASG_TOOL_INTERFACE( IDTPM2::ITrackSelector )
 
  virtual std::vector<const TrackView*> selectTracks( const std::vector<const TrackView*>& tv, TagNProbe* tnp=0 ) const { 
      std::vector<const TrackView*> out;
      if ( !highestPTonly() ) {
    for ( size_t i=0 ; i<tv.size() ; i++ ) {
      if ( tnp && !tnp->select(tv[i]) ) continue;
      if ( selectTrack(*tv[i]) ) out.push_back(tv[i]);
    }
      }
      else {
    const TrackView* best = 0;
    for ( size_t i=0 ; i<tv.size() ; i++ ) {
      if ( tnp && !tnp->select(tv[i]) ) continue;
      if ( !selectTrack(*tv[i]) ) continue;
      if ( !best || std::fabs(tv[i]->pt())>std::fabs(best->pt()) ) best = tv[i];
    }
    if ( best ) out.push_back(best);
      }
      return out;
  }
 
  virtual bool selectTrack( const TrackView& t ) const = 0;

  virtual bool highestPTonly() const { return false; }
 
};
 
} // namespace IDTPM2
 
#endif // > ! IDTPM2_ITRACKSELECTOR_H