egammaForwardBuilder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMAALGS_EGAMMAFORWARDBUILDER_H
#define EGAMMAALGS_EGAMMAFORWARDBUILDER_H

 
#include <Gaudi/Accumulators.h>
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/EventContext.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
//
#include "EventKernel/IParticle.h"
#include "StoreGate/ReadCondHandleKey.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
//
#include "CaloDetDescr/CaloDetDescrManager.h"
#include "CaloEvent/CaloClusterCellLinkContainer.h"
#include "xAODCaloEvent/CaloCluster.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
//
#include "xAODEgamma/Egamma.h"
#include "xAODEgamma/ElectronContainer.h"
//
#include "xAODTracking/VertexContainer.h"
#include "xAODEventInfo/EventInfo.h"
//
#include "egammaCaloUtils/egammaClusterCookieCut.h"
//
#include "egammaInterfaces/IEMTrackMatchBuilder.h"
#include "egammaInterfaces/IegammaBaseTool.h"
#include "egammaInterfaces/IegammaOQFlagsBuilder.h"
#include "EgammaAnalysisInterfaces/IegammaMVASvc.h"
 
 
#include "EgammaAnalysisInterfaces/IAsgForwardElectronIsEMSelector.h"
#include "EgammaAnalysisInterfaces/IAsgElectronLikelihoodTool.h"
#include "EgammaAnalysisInterfaces/IForwardElectronCalib.h"
//
#include <memory>
#include <string>
 
 
 
 
class egammaForwardBuilder : public AthReentrantAlgorithm
{
public:
  egammaForwardBuilder(const std::string& name, ISvcLocator* pSvcLocator);

  ~egammaForwardBuilder() = default;

  virtual StatusCode initialize() override final;

  virtual StatusCode finalize() override final;

  virtual StatusCode execute(const EventContext& ctx) const override final;
 
private:
  StatusCode RetrieveEMTrackMatchBuilder();
  StatusCode ExecObjectQualityTool(const EventContext& ctx, xAOD::Egamma* eg) const;
  std::pair<unsigned int, float> getnPVmu(const EventContext& ctx) const;

  void setAllTrackCaloMatchValues(
    xAOD::Electron *el,
    const std::array<xAOD::EgammaParameters::TrackCaloMatchType, 4> &match_parameters,
    const std::array<double, 4> &match_values
  ) const;
 

  ToolHandle<IegammaOQFlagsBuilder> m_objectQualityTool{
    this,
    "ObjectQualityTool",
    "",
    "Name of the object quality tool (empty tool name ignored)"
  };

  ToolHandle<IEMTrackMatchBuilder> m_trackMatchBuilder{
    this,
    "TrackMatchBuilderTool",
    "EMTrackMatchBuilder",
    "Tool that matches tracks to egammaRecs (Fwd)"
  };

  Gaudi::Property<bool> m_doEnergyCal {
    this,
    "doEnergyCal",
    false,
    "Boolean to do energy calibration"
  };
 

  Gaudi::Property<bool> m_dopTCal {
    this,
    "dopTCal",
    false,
    "Boolean to do pT calibration"
  };
 
  Gaudi::Property<bool> m_fwdDNN {
    this,
    "dofwdDNN",
    false,
    "Boolean to do forward DNN identification"
  };
 
  
  SG::ReadHandleKey<xAOD::VertexContainer> m_pVtxKey{
    this,
    "PrimaryVerticesContainerKey",
    "",
    "Name of the primary vertex container" };

  SG::ReadHandleKey<xAOD::EventInfo> m_eiKey{
    this,
    "EventInfoKey",
    "",
    "Name of the EventInfo object" };

  ServiceHandle<IegammaMVASvc> m_MVACalibSvc{
    this,
    "MVACalibSvc",
    "egammaMVASvc",
    "calibration service" };

  SG::ReadHandleKey<xAOD::CaloClusterContainer> m_topoClusterKey{
    this,
    "TopoClusterName",
    "",
    "Name of the input cluster collection"
  };

  SG::ReadCondHandleKey<CaloDetDescrManager> m_caloDetDescrMgrKey {
    this,
    "CaloDetDescrManager",
    "CaloDetDescrManager",
    "SG Key for CaloDetDescrManager in the Condition Store"
  };

  SG::WriteHandleKey<xAOD::ElectronContainer> m_electronOutputKey{
    this,
    "ElectronOutputName",
    "",
    "Name of Electron Container to be created"
  };

  SG::WriteHandleKey<xAOD::CaloClusterContainer> m_outClusterContainerKey{
    this,
    "ClusterContainerName",
    ""
    "Name of the output EM cluster container"
  };

  SG::WriteHandleKey<CaloClusterCellLinkContainer> m_outClusterContainerCellLinkKey{
    this,
    "ClusterContainerLinks", "",
    "Key of the output cluster container cell links; Taken from associated container"};

  Gaudi::Property<bool> m_doTrackMatching {
    this,
    "doTrackMatching",
    false,
    "Boolean to do track matching"
  };

  Gaudi::Property<bool> m_doCookieCutting {
    this,
    "doCookieCutting",
    false,
    "Boolean to do cookie cutting"
  };

  Gaudi::Property<int> m_maxDelEtaCells{
    this,
    "MaxWindowDelEtaCells",
    3,
    "Size of maximum search window in eta"
  };

  Gaudi::Property<int> m_maxDelPhiCells{
    this,
    "MaxWindowDelPhiCells",
    3,
    "Size of maximum search window in phi"
  };

  Gaudi::Property<float> m_maxDelR{
    this,
    "MaxWindowDelR",
    0.3,
    "Cone size to collect cells around hottest-cell FCAL"
  };
 
  mutable Gaudi::Accumulators::Counter<> m_AllClusters {};
  mutable Gaudi::Accumulators::Counter<> m_MatchedClusters {};
 
  egammaClusterCookieCut::CookieCutPars m_CookieCutPars{};
 
protected:
  ToolHandleArray<IAsgForwardElectronIsEMSelector> m_forwardElectronIsEMSelectors {
    this,
    "forwardelectronIsEMselectors",
    {},
    "The selectors that we need to apply to the FwdElectron object"
  };
 
  ToolHandleArray<IAsgElectronLikelihoodTool> m_forwardElectronNNSelectors {
    this,
    "forwardelectronNNselectors",
    {},
    "The selectors that we need to apply to the FwdElectron object"
  };
 
 
  ToolHandle<IForwardElectronCalib> m_forwardElectronpTCalib {
    this,
    "forwardelectronNNCalib",
    {},
    "The tool that we need to calibrate the pT of the FwdElectron object"
  };
 
 
  Gaudi::Property<std::vector<std::string>> m_forwardElectronIsEMSelectorResultNames {
    this,
    "forwardelectronIsEMselectorResultNames",
    {},
    "The selector result names"
  };
 
  Gaudi::Property<std::vector<std::string>> m_forwardElectronNNSelectorResultNames {
    this,
    "forwardelectronNNselectorResultNames",
    {},
    "The selector result names"
  };
 
};
#endif