EMClusterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMATOOLS_EMCLUSTERTOOL_H
#define EGAMMATOOLS_EMCLUSTERTOOL_H
 
#include "egammaInterfaces/IEMClusterTool.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/EventContext.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "EgammaAnalysisInterfaces/IegammaMVASvc.h"
#include "egammaInterfaces/IegammaSwTool.h"
 
#include "xAODCaloEvent/CaloCluster.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "xAODEgamma/EgammaEnums.h"
#include "xAODEgamma/EgammaFwd.h"
 
class CaloClusterCellLink;

 
class EMClusterTool final
  : public AthAlgTool
  , virtual public IEMClusterTool
{
 
public:
  EMClusterTool(const std::string& type,
                const std::string& name,
                const IInterface* parent);

  virtual ~EMClusterTool() = default;

  virtual StatusCode initialize() override final;
  virtual StatusCode contExecute(
    const EventContext& ctx,
    xAOD::ElectronContainer* electronContainer,
    xAOD::PhotonContainer* photonContainer) const override final;
 
private:
  void setNewCluster(const EventContext& ctx,
                     xAOD::Egamma* eg,
                     xAOD::CaloClusterContainer* outputClusterContainer) const;

  SG::WriteHandleKey<xAOD::CaloClusterContainer> m_outputClusterContainerKey{
    this,
    "OutputClusterContainerName",
    "",
    "Name of the output cluster container"
  };

  SG::WriteHandleKey<CaloClusterCellLinkContainer>
    m_outputClusterContainerCellLinkKey{
      this,
      "DoNotSet_OutputClusterContainerLinks",
      "",
      "Key of the output cluster container cell links; Do not set! Name taken "
      "from associated container"
    };

  ServiceHandle<IegammaMVASvc> m_MVACalibSvc{ this,
                                              "MVACalibSvc",
                                              "egammaMVASvc",
                                              "calibration service" };
};
 
#endif // EGAMMATOOLS_EMCLUSTERTOOL_H