PFTrackSelector.h

Go to the documentation of this file.

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

 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
 
#include "AthenaMonitoringKernel/Monitored.h"
 
#include "StoreGate/ReadCondHandleKey.h"
 
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "InDetTrackSelectionTool/IInDetTrackSelectionTool.h"
#include "Particle/TrackParticleContainer.h"
#include "TRT_ReadoutGeometry/TRT_DetElementContainer.h"
#include "eflowRecTrack.h"
#include "xAODEgamma/ElectronContainer.h"
#include "xAODMuon/MuonContainer.h"
#include "xAODTracking/VertexContainer.h"
 
class eflowTrackExtrapolatorBaseAlgTool;
 
class PFTrackSelector : public AthReentrantAlgorithm
{
 
public:
  PFTrackSelector(const std::string& name, ISvcLocator* pSvcLocator);
  ~PFTrackSelector(){};

  StatusCode initialize();
  StatusCode execute(const EventContext& ctx) const;
  StatusCode finalize();
 
private:
  bool selectTrack(const xAOD::TrackParticle& track) const;

  bool isElectron(const xAOD::TrackParticle* track) const;

  bool isMuon(const xAOD::TrackParticle* track) const;

  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_tracksReadHandleKey{
    this,
    "tracksName",
    "InDetTrackParticles",
    "ReadHandleKey for the TrackParticleContainer to be used as input"
  };

  SG::ReadHandleKey<xAOD::ElectronContainer> m_electronsReadHandleKey{
    this,
    "electronsName",
    "eflowRec_selectedElectrons",
    "ReadHandleKey for the ElectronContainer to be used as input"
  };

  SG::ReadHandleKey<xAOD::MuonContainer> m_muonsReadHandleKey{
    this,
    "muonsName",
    "eflowRec_selectedMuons",
    "ReadHandleKey for the MuonContainer to be used as input"
  };

  SG::WriteHandleKey<eflowRecTrackContainer> m_eflowRecTracksWriteHandleKey{
    this,
    "eflowRecTracksOutputName",
    "eflowRecTracks",
    "WriteHandleKey for the eflowRecTrackContainer to write out"
  };
 
  SG::ReadHandleKey<xAOD::VertexContainer> m_vertexKey{ this,
                                                        "VertexContainer",
                                                        "PrimaryVertices",
                                                        "primary vertex container" };

  ToolHandle<eflowTrackExtrapolatorBaseAlgTool> m_theTrackExtrapolatorTool{
    this,
    "trackExtrapolatorTool",
    "Trk::ParticleCaloExtensionTool",
    "ToolHandle for track extrapolation to calorimeter tool"
  };

  ToolHandle<InDet::IInDetTrackSelectionTool>
    m_trackSelectorTool{ this, "trackSelectionTool", "", "ToolHandle to track selection tool provided by tracking CP" };

  Gaudi::Property<float> m_upperTrackPtCut{ this,
                                            "upperTrackPtCut",
                                            100.0,
                                            "Upper limit on track Pt for input tracks" };

  ToolHandle<GenericMonitoringTool> m_monTool{ this, "MonTool", "", "Monitoring tool" };
};
#endif