CaloCalibClusterDecoratorTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef CALOCALIBHITREC_CALOCALIBCLUSTERDECORATORTOOL_H
#define CALOCALIBHITREC_CALOCALIBCLUSTERDECORATORTOOL_H
 
#include "GaudiKernel/ToolHandle.h"
#include "CaloUtils/CaloClusterCollectionProcessor.h"
#include "StoreGate/ReadHandleKeyArray.h"
#include "CaloCalibHitRec/CaloCalibDefineTypes.h"
#include "CaloCalibHitRec/ICaloCalibClusterTruthAttributerTool.h"
 
//EDM Classes
#include "CaloSimEvent/CaloCalibrationHit.h"
#include "xAODTruth/TruthParticle.h"
 
//EDM Container Classes
#include "xAODCaloEvent/CaloClusterContainer.h"
 
//C++ classes
#include <string>
#include <vector>
#include <set>
#include <map>
#include <atomic>
#include <array>

class CaloCalibClusterDecoratorTool : public AthAlgTool, virtual public CaloClusterCollectionProcessor {
 
public:

  CaloCalibClusterDecoratorTool(const std::string& type, const std::string& name,
                                const IInterface* parent);
 
  using CaloClusterCollectionProcessor::execute;
 
 
  virtual StatusCode execute(const EventContext& ctx,
                             xAOD::CaloClusterContainer* theClusColl) const override final;
 
  
  virtual StatusCode initialize() override;
 
 
  virtual StatusCode finalize() override;

  virtual ~CaloCalibClusterDecoratorTool() {};
 
private:

  StatusCode calculateTruthEnergies(const xAOD::CaloCluster& theCaloCluster,
                                    unsigned int numTruthParticles,
                                    const std::map<Identifier,std::vector<const CaloCalibrationHit*> >& identifierToCaloHitMap,
                                    std::vector<std::pair<unsigned int,double>>& truthIDTrueCalHitEnergy) const;

  SG::ReadHandleKey<std::map<Identifier,std::vector<const CaloCalibrationHit*> > >
    m_mapIdentifierToCalibHitsReadHandleKey{this,"IdentifierToCalibHitsMapName","IdentifierToCalibHitsMap",
                                            "ReadHandleKey for the map between Identifiers and sets of calibration hits"};

  SG::WriteDecorHandleKey<xAOD::CaloClusterContainer>
    m_caloClusterWriteDecorHandleKeyNLeadingTruthParticles{
      this,
      "CaloClusterWriteDecorHandleKey_NLeadingTruthParticles",
      "CaloTopoClustersNew.calclus_NLeadingTruthParticleBarcodeEnergyPairs"};

  Gaudi::Property<unsigned int> m_numTruthParticles{
      this,"NumTruthParticles",100,
      "Set number of truth particles per CaloCluster/PFO for which we store calibration hit energy"};

  Gaudi::Property<bool> m_useCellWeights{
      this,"useCellWeights",true,
      "Toggle whether to use cell weights or not to calculate calibration hit contribution"};

  Gaudi::Property<bool> m_storeFullTruthEnergy{
      this,"StoreFullTruthEnergy",false,
      "If true, include escaped + invisible energy in truth energy calculation"};

  ToolHandle<ICaloCalibClusterTruthAttributerTool>
    m_truthAttributerTool{
      this,"TruthAttributerTool","",
      "ToolHandle to a tool to create the calibration hit truth information that we need for the decoration"};
 
};
 
#endif // CALOCALIBCLUSTERDECORATORTOOL_H