PhotonPointingTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PhotonVertexSelection_PhotonPointingTool_h
#define PhotonVertexSelection_PhotonPointingTool_h
 
// Framework includes
#include "AsgTools/AsgMetadataTool.h"
 
// Local includes
#include "PhotonVertexSelection/IPhotonPointingTool.h"
 
// EDM includes
#include "xAODEgamma/EgammaContainer.h"
 
// Data handles
#include "AsgTools/CurrentContext.h"
#include "AsgDataHandles/WriteDecorHandle.h"
#include "AsgDataHandles/ReadHandleKey.h"
 
// Forward declarations
class TH1F;
 
namespace CP {

  class PhotonPointingTool : public virtual IPhotonPointingTool,
                             public asg::AsgMetadataTool {

    ASG_TOOL_CLASS(PhotonPointingTool, CP::IPhotonPointingTool)
 
  private:
    TH1F *m_zCorrection;
 
    SG::ReadHandleKey<xAOD::EventInfo> m_evtInfo{
      this,
      "EventInfo",
      "EventInfo",
      "SG key of xAOD::EventInfo"
    };
 
    //Write decoration handle keys
    SG::WriteDecorHandleKey<xAOD::EgammaContainer> m_zvertex;
    SG::WriteDecorHandleKey<xAOD::EgammaContainer> m_errz;
    SG::WriteDecorHandleKey<xAOD::EgammaContainer> m_HPV_zvertex;
    SG::WriteDecorHandleKey<xAOD::EgammaContainer> m_HPV_errz;
 
 
  private:
    float getCorrectedZ(float zPointing, float etas2) const;
    bool m_isMC{};
    std::string m_zOscFileMC, m_zOscFileData;
    std::string m_zvertexDecorName;
    std::string m_errzDecorName;
    std::string m_HPV_zvertexDecorName;
    std::string m_HPV_errzDecorName;
    std::string m_ContainerName;
 
  public:
    PhotonPointingTool(const std::string &name);
    virtual ~PhotonPointingTool();


    virtual StatusCode initialize();



    StatusCode updatePointingAuxdata(const xAOD::EgammaContainer &egammas) const ;

    std::pair<float, float> getCaloPointing(const xAOD::Egamma *egamma) const ;

    std::pair<float, float> getConvPointing(const xAOD::Photon *photon) const ;
 
  }; // class PhotonPointingTool
 
} // namespace CP
 
 
#endif // PhotonVertexSelection_PhotonPointingTool_H