TrackRoiSelector.h

Go to the documentation of this file.

 
#ifndef IDTPM2_TRACKROISELECTOR_H
#define IDTPM2_TRACKROISELECTOR_H
 
#include "TrigSteeringEvent/TrigRoiDescriptor.h"
 
#include <string>
#include <vector>
#include <optional>
 
class TrigRoiDescriptorCollection;
 
namespace IDTPM2 {

  
class TrackRoiSelector { 
 
public:

    TrackRoiSelector() { } 

    virtual ~TrackRoiSelector() = default;

    template<class T>
    bool accept( const T& t, const TrigRoiDescriptor* r ) const;

    template<class T>
    std::vector<const T*> getTracks( const std::vector<const T*>& tracks,
                     const TrigRoiDescriptor* roi ) const;
 
private:
 
  struct exit_point {
    exit_point(double z_=0, double r_=0 ) : z(z_), r(r_) { }  
    double z;
    double r;
  };
 
  
private:

 
#if 1
    exit_point  getExitPoint( float tz0, float teta ) const;
#endif
  
  std::optional<double>   getOuterPhi( float pt, float phi, float r=1000. ) const;
 
}; // class TrackRoiSelector
 
} // namespace IDTPM2
 
 
template<class T>
std::vector<const T*> IDTPM2::TrackRoiSelector::getTracks( const std::vector<const T*>& tracks,
                                                           const TrigRoiDescriptor* roi ) const {
 
  //  std::cout << "\n\n\tSUTT roi filter" << tvec.size() << "\t" << *roi << "\n" << std::endl;
 
  std::vector<const T*> selectedTracks;
 
  for( size_t it=0 ; it<tracks.size() ; it++ ) {
 
    const T* track = tracks.at(it);

    if ( track->pt()==0 ) continue;
    
    bool keep = accept<T>( *track, roi );
 
    //    if ( keep ) std::cout <<  it << "\t" << *tvec[it] << "\taccept: " << keep << std::endl; 
    
    if( keep ) selectedTracks.push_back( track );
  }
 
  //  std::cout << "\tSUTT: " << selectedTracks.size() << std::endl; 
  
  return selectedTracks;
}
 
 
 
 
#endif // > !IDTPM2_TRACKROISELECTOR_H