TrigTrackSelector.h

Go to the documentation of this file.

/* emacs: this is -*- c++ -*- */
#ifndef TIDAUTILS_TRIGTRACKSELECTOR_H
#define TIDAUTILS_TRIGTRACKSELECTOR_H
 
#include "TrigInDetAnalysisUtils/TIDA_newtracking.h"
 
#include "TrigInDetEvent/TrigInDetTrackCollection.h"
#include "TrigInDetTruthEvent/TrigInDetTrackTruthMap.h"
 
#include "Particle/TrackParticle.h"
#include "Particle/TrackParticleContainer.h"
 
#include "McParticleEvent/TruthParticleContainer.h"
#include "AtlasHepMC/GenEvent.h"
#include "AtlasHepMC/GenVertex.h"
#include "AtlasHepMC/GenParticle.h"
#include "xAODTruth/TruthParticleContainer.h"
 
 
 
#include "TrigInDetAnalysisUtils/particleType.h"
 
#include "TrigInDetAnalysis/TrackSelector.h"
#include "TrigInDetAnalysis/Track.h"
#include "TrigInDetAnalysis/TrackFilter.h"
 
#include "TrkParameters/TrackParameters.h"
// #include "TrkParameters/Perigee.h"
#include "TrkTrack/TrackCollection.h"
#include "TrkTrack/Track.h"
#include "TrkTrackSummary/TrackSummary.h"
#include "TrkToolInterfaces/ITrackSummaryTool.h"
#include "GaudiKernel/ToolHandle.h" 
#include "TMath.h"
 
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
 
class TrigTrackSelector : public TrackSelector { 
 
public:
 
  typedef std::vector< ElementLink< xAOD::TrackParticleContainer > > TrackParticleLinks_t;
 
public:
 
 
  TrigTrackSelector( TrackFilter* selector ) : TrigTrackSelector( selector, default_radius(), 0, 0 ) { } 
 
  TrigTrackSelector( TrackFilter* selector, double radius ) : TrigTrackSelector( selector, radius, 0, 0 ) { } 
 
  TrigTrackSelector( TrackFilter* selector, int selectPdgId, int selectParentPdgId ) : TrigTrackSelector( selector, default_radius(), selectPdgId, selectParentPdgId ) { } 
 
  TrigTrackSelector( TrackFilter* selector, double radius, int selectPdgId, int selectParentPdgId );
 
  template<typename T>
  TrigTrackSelector( T& tm,  TrackFilter* selector, double radius=default_radius(), int selectPdgId=0, int selectParentPdgId=0 ) : 
    TrigTrackSelector( tm.begin(), tm.end(), selector, radius, selectPdgId, selectParentPdgId ) { 
  }
 
  template<typename T>
  TrigTrackSelector( T tbegin, T tend,  TrackFilter* selector, double radius=default_radius(), int selectPdgId=0, int selectParentPdgId=0 ) : 
    TrigTrackSelector( selector, radius, selectPdgId, selectParentPdgId ) { 
    selectTracks( tbegin, tend );
  }
 
 
  ~TrigTrackSelector() { clear(); }
 
  virtual TrackSelector* clone() override { return new TrigTrackSelector(*this); }
 
  void setBeamline( double x, double y, double z=0) { m_xBeam = x; m_yBeam = y; m_zBeam=z; }
 
  double getBeamX() const { return m_xBeam; } 
  double getBeamY() const { return m_yBeam; } 
  double getBeamZ() const { return m_zBeam; } 
 
  void correctTracks(bool b=true) { m_correctTrkTracks = b; } 
 
  virtual void clear() override { for ( size_t i=m_tracks.size() ; i-- ; ) delete m_tracks[i]; m_tracks.clear(); }   
 
 
  bool selectTrack( const TrigInDetTrack* track, const TrigInDetTrackTruthMap* truthMap=0 );
 
 
  // extract all the tracks from a TrigInDetTrack collection and associated TruthMap and convert them
  void selectTracks( const TrigInDetTrackCollection* trigtracks, const TrigInDetTrackTruthMap* truthMap=0 );
 
  // add a TrackParticle 
  bool selectTrack( const Rec::TrackParticle* track );
  
 
  // extract all the tracks from a TrackParticle collection and add them
  void selectTracks( const Rec::TrackParticleContainer* trigtracks );
 
 
  // extract all the tracks from a TrackParticle collection and add them
  void selectTracks( const TruthParticleContainer* truthtracks );
 
  // extract all the tracks from a TrackParticle collection and add them
  void selectTracks( const xAOD::TruthParticleContainer* truthtracks );
 
  // extract all the tracks from a TrackParticle collection and add them
  void truthBeamline( const xAOD::TruthParticleContainer* truthtracks, double& x0, double& y0 );
 
 
  bool selectTrack( HepMC::ConstGenParticlePtr track );
 
 
  // add a TruthParticle 
  bool selectTrack( const TruthParticle& track );
 
  // add a TruthParticle 
  bool selectTrack( const TruthParticle* track );
 
  bool selectTrack( const xAOD::TruthParticle* track, double x0=0, double y0=0 );
 
 
  // make a TIDA::Track from a GenParticle 
  TIDA::Track* makeTrack( HepMC::ConstGenParticlePtr track );
 
  // make a TIDA::Track from a TruthParticle 
  TIDA::Track* makeTrack( const TruthParticle* track, unsigned long tid=0 );
 
 
  // add a Trk::Track
  bool selectTrack( const Trk::Track* track );
 
  
  // extract all the tracks from a TrackCollection and add them
  void selectTracks( const TrackCollection* trigtracks );
 
 
 
  bool selectTrack( const xAOD::TrackParticle* track, void* =0);
 
  void selectTracks( const xAOD::TrackParticleContainer* tracks, void* =0);
 
 
  void selectTracks( xAOD::TrackParticleContainer::const_iterator trackitr, 
             xAOD::TrackParticleContainer::const_iterator trackend, void* =0);
 
 
  void selectTracks( const TrackParticleLinks_t& tracks );
 
  void correctToBeamline( double& z0,    double& dz0, 
              double& d0,    double& dd0, 
              double  theta, double  phi );
 
  
protected:
 
 
  const xAOD::TruthParticle* fromAncestor( const int pdg_id,  const xAOD::TruthParticle *p) const;
 
  const xAOD::TruthParticle* fromAncestor( const std::vector<int>& ids,  const xAOD::TruthParticle *p) const;
 
public:
 
  static double default_radius() { return s_default_radius; }
 
private:
 
  unsigned long m_id;
 
  double m_xBeam;
  double m_yBeam;
  double m_zBeam;
 
  bool   m_correctTrkTracks;
 
  double m_radius;
 
  int m_selectPdgId;
  int m_selectParentPdgId;
 
  static const double s_default_radius;
 
};
 
#endif // TIDAUTILS_TRIGTRACKSELECTOR_H