TrackRoiSelectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRACKPERFMON_TRACKROISELECTIONTOOL_H
#define INDETTRACKPERFMON_TRACKROISELECTIONTOOL_H


#include "AsgTools/IAsgTool.h"
#include "AsgTools/AsgTool.h"
#include "TrigDecisionTool/TrigDecisionTool.h"
#include "AthLinks/ElementLink.h"

#include <string>
#include <vector>
 
class TrigRoiDescriptorCollection;
 
namespace IDTPM {
 
  class TrackAnalysisCollections;
 
  typedef struct {
    float z;
    float r;
    float tantheta;
  } exitPoint_t;
 
  class TrackRoiSelectionTool :
      public virtual asg::IAsgTool,
      public asg::AsgTool {
 
  public:
 
    ASG_TOOL_CLASS( TrackRoiSelectionTool, IAsgTool );

    TrackRoiSelectionTool( const std::string& name );

    virtual ~TrackRoiSelectionTool() = default;

    virtual StatusCode initialize() override;

    StatusCode selectTracksInRoI(
        TrackAnalysisCollections& trkAnaColls,
        const ElementLink< TrigRoiDescriptorCollection >& roiLink );
  
    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* >& tvec,
        const TrigRoiDescriptor* r ) const;

    std::vector< const xAOD::TrackParticle* > getTrigTracks( 
        const std::vector< const xAOD::TrackParticle* >& tvec,
        const ElementLink< TrigRoiDescriptorCollection >& roiLink ) const;
 
  private:

    exitPoint_t getExitPoint( float tz0, float teta ) const;
 
    float getOuterPhi( float pt, float phi, float r=1000. ) const;

    SG::ReadHandleKey< xAOD::TrackParticleContainer > m_triggerTrkParticleName {
        this, "TriggerTrkParticleContainerName", "", "Name of container of trigger tracks" };

    PublicToolHandle< Trig::TrigDecisionTool > m_trigDecTool {
        this, "TrigDecisionTool", "Trig::TrigDecisionTool/TrigDecisionTool", "" };
 
  }; // class TrackRoiSelectionTool
 
} // namespace IDTPM
 
 
#endif // > !INDETTRACKPERFMON_TRACKROISELECTIONTOOL_H