PhotonsDirectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_PHOTONSDIRECTIONTOOL_H
#define DERIVATIONFRAMEWORK_PHOTONSDIRECTIONTOOL_H
 
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
//
#include "AthenaBaseComps/AthAlgTool.h"
#include "xAODEgamma/PhotonContainer.h"
//
#include <string>
#include <vector>
namespace DerivationFramework {
 
  class PhotonsDirectionTool : 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:
    SG::ReadHandleKey<xAOD::PhotonContainer> m_collName{ this,
        "PhotonContainer",
        "Photons",
        "Input Photons" };
 
    SG::WriteHandleKey<std::vector<float>> m_sgEta{ this,
      "EtaSGEntry",
      "",
      "output Eta vector" };
 
    SG::WriteHandleKey<std::vector<float>> m_sgPhi{ this,
      "PhiSGEntry",
      "",
      "output Phi vector" };
 
    SG::WriteHandleKey<std::vector<float>> m_sgEt{ this,
      "EtSGEntry",
      "",
      "output E vector" };
 
    SG::WriteHandleKey<std::vector<float>> m_sgE{ this,
      "ESGEntry",
      "",
      "output E vector" };
 
    bool m_doEta = false;
    bool m_doPhi = false;
    bool m_doEt = false;
    bool m_doE = false;
  };
}
 
#endif // DERIVATIONFRAMEWORK_PHOTONSDIRECTIONTOOL_H