PFlowCalibPFODecoratorAlgorithm.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef PFLOWCALIBPFODECORATORALGORITHM_H
#define PFLOWCALIBPFODECORATORALGORITHM_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "CaloCalibHitRec/CaloCalibDefineTypes.h"
#include "CaloCalibHitRec/ICaloCalibClusterTruthAttributerTool.h"
 
//EDM Classes
#include "CaloSimEvent/CaloCalibrationHit.h"
#include "xAODTruth/TruthParticle.h"
 
//EDM Container Classes
#include "xAODPFlow/FlowElementContainer.h"
 
//C++ classes
#include <map>
#include <vector>
 
//Core classes for some private function classdefs
#include "StoreGate/WriteDecorHandle.h"
 
class PFlowCalibPFODecoratorAlgorithm : public AthReentrantAlgorithm {
 
public:
  using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
  virtual ~PFlowCalibPFODecoratorAlgorithm() {};
  
  /* Gaudi algorithm hooks */
  virtual StatusCode initialize() override;
  virtual StatusCode execute(const EventContext& ctx) const override;
  virtual StatusCode finalize() override;
 
private:
  
  SG::ReadHandleKey<std::map<Identifier,std::vector<const CaloCalibrationHit*> > > m_mapIdentifierToCalibHitsReadHandleKey{this,"IdentifierToCalibHitsMapName","IdentifierToCalibHitsMap","ReadHandleKey for the map between Identifieirs and sets of calibration hits"};
 
  SG::WriteDecorHandleKey<xAOD::FlowElementContainer> m_pfoWriteDecorHandleKeyNLeadingTruthParticles{this,"PFOWriteDecorHandleKey_NLeadingTruthParticles","JetETMissNeutralParticleFlowObjects.calpfo_NLeadingTruthParticleBarcodeEnergyPairs"};
  
  ToolHandle<ICaloCalibClusterTruthAttributerTool> m_truthAttributerTool{this,"TruthAttributerTool",""," ToolHandle to a tool to create the calibration hit truth information that we need for the decoration"};
 
  Gaudi::Property<unsigned int> m_numTruthParticles{this,"NumTruthParticles",20,"Set number of truth particles per CaloCluster/PFO for which we store calibration hit energy"};
 
  // functions to do the links between either PFO or FlowElements
  StatusCode LinkCalibHitPFO(
                 SG::WriteDecorHandle<xAOD::FlowElementContainer, std::vector< std::pair<unsigned int, double> > >& pfoWriteDecorHandle,
                 SG::ReadHandle<std::map<Identifier,std::vector<const CaloCalibrationHit*> > >& CalibHitHandle) const;
  
};
#endif