egammaTruthAlg.h

Go to the documentation of this file.

// This file's extension implies that it's C, but it's really -*- C++ -*-.
/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
#ifndef EGAMMAD3PDANALYSIS_EGAMMATRUTHALG_H
#define EGAMMAD3PDANALYSIS_EGAMMATRUTHALG_H
 
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "xAODTruth/TruthParticle.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "RecoToolInterfaces/IParticleCaloExtensionTool.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "GaudiKernel/ToolHandle.h"
#include <string>
 
 
namespace D3PD {
 
 
class egammaTruthAlg
  : public AthReentrantAlgorithm
{
public:
  egammaTruthAlg (const std::string& name,
                  ISvcLocator* svcloc);
 
 
  virtual StatusCode initialize() override;
 
 
  virtual StatusCode execute (const EventContext& ctx) const override;
 
 
private:
  bool isAccepted (const xAOD::TruthParticle& tp,
                   const xAOD::TruthParticleContainer& cont,
                   float& iso) const;
 
 
  float computeIso (const xAOD::TruthParticle& tp,
                    const xAOD::TruthParticleContainer& cont) const;
 
 
  StatusCode findImpact (const xAOD::TruthParticle& tp,
                         float& etaCalo,
                         float& phiCalo,
                         float& depthCalo) const;
 
 
  std::string m_auxPrefix;
 
  SG::ReadHandleKey<xAOD::TruthParticleContainer> m_inputKey
    { this, "InputKey", "", "SG key for the input container." };
 
  SG::WriteHandleKey<xAOD::TruthParticleContainer> m_outputKey
    { this, "OutputKey", "", "SG key for the output container." };
 
  float m_electronPtMin;
 
  float m_photonPtMin;
 
  float m_etaMax;
 
  float m_isoCone;
 
  float m_photonEtIsoMax;
 
  ToolHandle<Trk::IParticleCaloExtensionTool> m_exten
  { this, "ParticleCaloExtensionTool", "", "Extrapolator to calorimeter." };
};
 
 
} // namespace D3PD
 
 
#endif // not EGAMMAD3PDANALYSIS_EGAMMATRUTHALG_H