PhotonVertexSelectionWrapper.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_PHOTONVERTEXSELECTIONWRAPPER_H
#define DERIVATIONFRAMEWORK_PHOTONVERTEXSELECTIONWRAPPER_H
 
#include <string>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
 
#include "GaudiKernel/ToolHandle.h"
#include "PhotonVertexSelection/IPhotonPointingTool.h"
 
#include "StoreGate/ReadHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandle.h"
#include "StoreGate/WriteHandleKey.h"
#include "xAODEgamma/PhotonContainer.h"
#include "xAODTracking/VertexContainer.h"
 
// namespace CP
//{
//  class IPhotonPointingTool;
//}
 
namespace DerivationFramework {
 
  class PhotonVertexSelectionWrapper : public extends<AthAlgTool, IAugmentationTool>
  {
  public:
 
    using base_class::base_class;
 
    virtual StatusCode initialize() override final;
    virtual StatusCode addBranches(const EventContext& ctx) const override final;
 
  private:
    ToolHandle<CP::IPhotonPointingTool> m_photonPointingTool{
      this,
        "PhotonPointingTool",
        "",
        "Handle to the photon pointing tool"
        };
 
    SG::ReadHandleKey<xAOD::PhotonContainer> m_photonContainer{
      this,
      "PhotonContainer",
      "Photons",
      "SG key of input photon container"
    };
 
    SG::ReadHandleKey<xAOD::VertexContainer> m_vertexContainer{
      this,
      "VertexContainer",
      "PrimaryVertices",
      "SG key of input vertex container"
    };
 
    // Write decoration handle keys
    // these are not really configurable
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_vtxPt{ this, "pt", m_vertexContainer, "", "" };
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_vtxEta{ this,
      "eta",
      m_vertexContainer, "",
      "" };
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_vtxPhi{ this,
      "phi",
      m_vertexContainer, "",
      "" };
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_vtxSumPt{ this,
      "sumPt",
      m_vertexContainer,  "",
      "" };
    SG::WriteDecorHandleKey<xAOD::VertexContainer> m_vtxSumPt2{ this,
      "sumPt2",
      m_vertexContainer, "",
      "" };
  };
}
 
#endif // DERIVATIONFRAMEWORK_PHOTONVERTEXSELECTIONWRAPPER_H