EGammaCookieCutClusterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_EGammaCookieCutClusterTool_H
#define DERIVATIONFRAMEWORK_EGammaCookieCutClusterTool_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandleKeyArray.h"
#include "xAODEgamma/EgammaContainer.h"
#include "CaloDetDescr/CaloDetDescrManager.h"
#include "egammaCaloUtils/egammaClusterCookieCut.h"
 
#include "CaloUtils/CaloClusterCollectionProcessor.h"
 
namespace DerivationFramework {
 
  class EGammaCookieCutClusterTool
    : 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::WriteHandleKey<xAOD::CaloClusterContainer> m_outClusterContainerKey{
      this,
        "ClusterContainerName",
        "ForwardElectronCookieCutClusters",
        "Name of the output cookie cut cluster container"
        };

    SG::WriteHandleKey<CaloClusterCellLinkContainer> m_outClusterContainerCellLinkKey{
      this,
        "ClusterContainerLinksName",
        "ForwardElectronCookieCutClusters_links",
        "Name of the output cluster container cell links container"
        };

    SG::ReadCondHandleKey<CaloDetDescrManager> m_caloDetDescrMgrKey {
      this,
      "CaloDetDescrManager",
      "CaloDetDescrManager",
      "SG Key for CaloDetDescrManager in the Condition Store"
    };
 
    SG::ReadHandleKey<xAOD::EgammaContainer> m_SGKey_electrons{
      this,
      "SGKey_electrons",
      "ForwardElectrons",
      "SG key of electron container"
    };
 
    SG::WriteDecorHandleKeyArray<xAOD::EgammaContainer>
    m_SGKey_electrons_decorations{
      this,
      "SGKey_electrons_decorations",
      m_SGKey_electrons, {},
      "SG keys for electrons decorations"
    };

    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"
    };

    Gaudi::Property<bool> m_fixCellWeights{
      this,
      "FixCellWeights",
      false,
      "Fix cell weights to one for the cookie-cut cluster"
    };

    Gaudi::Property<bool> m_storeOrigMom{
      this,
      "StoreInputMoments",
      false,
      "Decorate also with the moments from the original cluster"
    };

    Gaudi::Property<bool> m_storeCookMom{
      this,
      "StoreCookedMoments",
      false,
      "Decorate also with the moments from the cookie-cut cluster"
    };
 
    egammaClusterCookieCut::CookieCutPars m_CookieCutPars{};
 
    mutable std::once_flag m_Seen;
    unsigned short m_nDecor = 0;

    Gaudi::Property<std::vector<int>> m_vecM{
      this,
      "Moments",
      {},
      "The moments to be added"
    };

    Gaudi::Property<std::vector<std::string>> m_vecMName{
      this,
      "MomentNames",
      {},
      "The names of the moments to be added"
    };
 
    ToolHandleArray<CaloClusterCollectionProcessor> m_clusterCorrectionTools {
      this,
      "ClusterMomentMaker",
      {},
      "The moment maker"
    };
 
  };
 
}
 
#endif // DERIVATIONFRAMEWORK_EGammaCookieCutClusterTool_H