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