TrackSubsetProcessor.h

#ifndef TrackSubsetProcessor_h
#define TrackSubsetProcessor_h 1


#include "marlin/Processor.h"
#include "lcio.h"
#include "EVENT/Track.h"
#include "MarlinTrk/IMarlinTrkSystem.h"

#include "Math/ProbFunc.h"




using namespace lcio ;
using namespace marlin ;


class TrackSubsetProcessor : public Processor {
  
 public:
  
  virtual Processor*  newProcessor() { return new TrackSubsetProcessor ; }
  
  
  TrackSubsetProcessor() ;
  
  virtual void init() ;
  
  virtual void processRunHeader( LCRunHeader* run ) ;
  
  virtual void processEvent( LCEvent * evt ) ; 
  
  
  virtual void check( LCEvent * evt ) ; 
  
  
  virtual void end() ;
  
  
 protected:

  void removeShortTracks( std::vector< EVENT::Track*>& tracks ) ; 
  
  
  std::vector< std::string > _trackInputColNames;
  
  std::string _trackOutputColName;
  
  MarlinTrk::IMarlinTrkSystem* _trkSystem;
  std::string _trkSystemName ;
  
  bool _MSOn ;
  bool _ElossOn ;
  bool _SmoothOn ;
  bool _removeShortTracks ;


  float _initialTrackError_d0;
  float _initialTrackError_phi0;
  float _initialTrackError_omega;
  float _initialTrackError_z0;
  float _initialTrackError_tanL;
  
  double _maxChi2PerHit;
  
  float _bField;
  
  int _nRun ;
  int _nEvt ;
  
  double _omega;
  
} ;


class TrackCompatibility{
  
  
public:
  
  
  inline bool operator()( Track* trackA, Track* trackB ){
    
    
    std::vector< TrackerHit* > hitsA = trackA->getTrackerHits();
    std::vector< TrackerHit* > hitsB = trackB->getTrackerHits();
    
    
    for( unsigned i=0; i < hitsA.size(); i++){
      
      for( unsigned j=0; j < hitsB.size(); j++){
        
        if ( hitsA[i] == hitsB[j] ) return false;      // a hit is shared -> incompatible
      
      }
      
    }
    
    return true;      
    
    
  }
  
  
};


class TrackQI{
  
public:
  
  TrackQI( MarlinTrk::IMarlinTrkSystem* trkSystem ): _trkSystem(trkSystem){}
  
  inline double operator()( Track* track ){
    
     return ROOT::Math::chisquared_cdf_c( track->getChi2() , track->getNdf() );   
       
    
  }
  
protected:
  
  MarlinTrk::IMarlinTrkSystem* _trkSystem;
    
};




#endif