egammaSelectedTrackCopy.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 */
 
#ifndef EGAMMAALGS_EGAMMASELECTEDTRACKCOPY_H
#define EGAMMAALGS_EGAMMASELECTEDTRACKCOPY_H
 
#include "egammaInterfaces/IEMExtrapolationTools.h"
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/EventContext.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/ReadDecorHandleKey.h"
#include "StoreGate/ReadCondHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "CaloDetDescr/CaloDetDescrManager.h"
 
#include "AthContainers/ConstDataVector.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "xAODCaloEvent/CaloClusterFwd.h"
#include "xAODTracking/TrackParticleContainerFwd.h"
#include "xAODTracking/TrackParticleFwd.h"
 
#include "egammaInterfaces/IegammaCaloClusterSelector.h"
#include <Gaudi/Accumulators.h>
 
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"

class egammaSelectedTrackCopy : public AthReentrantAlgorithm
{
public:
  egammaSelectedTrackCopy(const std::string& name, ISvcLocator* pSvcLocator);
 
  virtual StatusCode initialize() override final;
  virtual StatusCode finalize() override final;
  virtual StatusCode execute(const EventContext& ctx) const override final;
 
private:
  bool selectTrack(const EventContext& ctx,
                   const xAOD::CaloCluster* cluster,
                   const xAOD::TrackParticle* track,
                   bool trkTRT,
                   const CaloDetDescrManager& caloDD) const;
    
  ToolHandle<IEMExtrapolationTools> m_extrapolationTool {
    this,
    "ExtrapolationTool",
    "EMExtrapolationTools",
    "Extrapolation tool"
  };

  ToolHandle<IegammaCaloClusterSelector> m_egammaCaloClusterSelector {
    this,
    "egammaCaloClusterSelector",
    "egammaCaloClusterSelector",
    "Tool that makes the cluster selection"
  };

  SG::ReadHandleKey<xAOD::CaloClusterContainer> m_clusterContainerKey {
    this,
    "ClusterContainerName",
    "egammaTopoCluster",
    "Input calo cluster for seeding"
  };

  SG::ReadHandleKey<xAOD::CaloClusterContainer> m_fwdClusterContainerKey {
    this,
    "FwdClusterContainerName",
    "",
    "Input calo cluster for seeding"
  };
 
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trackParticleContainerKey {
    this,
    "TrackParticleContainerName",
    "InDetTrackParticles",
    "Input TrackParticles to select from"
  };
 
  SG::ReadDecorHandleKey<xAOD::TrackParticleContainer> m_trackParticleTimeDecorKey{
    this,
    "TrackParticleTimeDecoration",
    "",
    "Time assigned to this track"};
 
  SG::ReadCondHandleKey<CaloDetDescrManager> m_caloDetDescrMgrKey {
    this,
    "CaloDetDescrManager",
    "CaloDetDescrManager",
    "SG Key for CaloDetDescrManager in the Condition Store"
  };
 
  SG::WriteHandleKey<ConstDataVector<xAOD::TrackParticleContainer>> m_OutputTrkPartContainerKey {
    this,
    "OutputTrkPartContainerName",
    "egammaSelectedTrackParticles",
    "Output selected TrackParticles"
  };

  Gaudi::Property<double> m_broadDeltaEta {
    this,
    "broadDeltaEta",
    0.2,
    "Value of broad cut for delta eta"
  };
 
  Gaudi::Property<double> m_broadDeltaPhi {
    this,
    "broadDeltaPhi",
    0.3,
    "Value of broad cut for delta phi"
  };

  Gaudi::Property<double> m_narrowDeltaEta{
    this,
    "narrowDeltaEta",
    0.05,
    "Value of narrow cut for delta eta"
  };
 
  Gaudi::Property<double> m_narrowDeltaPhi{
    this,
    "narrowDeltaPhi",
    0.05,
    "Value of narrow cut for delta phi"
  };
 
  Gaudi::Property<double> m_narrowDeltaPhiBrem{
    this,
    "narrowDeltaPhiBrem",
    0.2,
    "Value of the narrow cut for delta phi in the brem direction"
  };
 
  Gaudi::Property<double> m_narrowRescale{
    this,
    "narrowDeltaPhiRescale",
    0.05,
    "Value of the narrow cut for delta phi Rescale"
  };
 
  Gaudi::Property<double> m_narrowRescaleBrem{
    this,
    "narrowDeltaPhiRescaleBrem",
    0.1,
    "Value of the narrow cut for delta phi Rescale Brem"
  };

  Gaudi::Property<bool> m_doForwardTracks { 
    this,
    "doFwdTracks",
    false,
    "Boolean to select tracks in the forward region"
  };
 
  mutable Gaudi::Accumulators::Counter<> m_AllClusters {};
  mutable Gaudi::Accumulators::Counter<> m_AllFwdClusters {};
  mutable Gaudi::Accumulators::Counter<> m_SelectedClusters {};
  mutable Gaudi::Accumulators::Counter<> m_AllTracks {};
  mutable Gaudi::Accumulators::Counter<> m_SelectedTracks {};
  mutable Gaudi::Accumulators::Counter<> m_SelectedFwdTracks {};
  mutable Gaudi::Accumulators::Counter<> m_AllSiTracks {};
  mutable Gaudi::Accumulators::Counter<> m_SelectedSiTracks {};
  mutable Gaudi::Accumulators::Counter<> m_AllTRTTracks {};
  mutable Gaudi::Accumulators::Counter<> m_SelectedTRTTracks {};
};
#endif