TruthParticleID/McParticleTools/src/TruthIsolationTool.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MCPARTICLETOOLS_TRUTHISOLATIONTOOL_H 
#define MCPARTICLETOOLS_TRUTHISOLATIONTOOL_H 
 
// STL includes
#include <string>
#include <list>
#include <unordered_map>
 
// FrameWork includes
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
// McParticleKernel includes
#include "McParticleKernel/ITruthIsolationTool.h"
 
// McParticleEvent includes
#include "McParticleEvent/TruthParticleParamDefs.h"
#include "HepPDT/ParticleDataTable.hh"
 
// Forward declarations
class McEventCollection;
class TruthEtIsolations;
#include "AtlasHepMC/GenEvent_fwd.h"
#include "AtlasHepMC/GenParticle_fwd.h"
 
class TruthIsolationTool : virtual public ITruthIsolationTool,
                                   public AthAlgTool
{ 

  // Public methods: 
 public: 
  TruthIsolationTool( const std::string& type,
              const std::string& name, 
              const IInterface* parent );
  virtual ~TruthIsolationTool(); 
 
  // Athena algorithm's Hooks
  StatusCode initialize();

  // Const methods: 

  const std::string&
  etIsolationsName( const std::string& mcEvtName ) const;

  // Non-const methods: 

  StatusCode buildEtIsolations( const std::string& mcEvtName, 
                ITruthIsolationTool::ParticleSelect sel );

  StatusCode registerAlias( const std::string& originalMcEvtColl,
                const std::string& aliasMcEvtColl );

  // Private methods: 
 private: 
  
  StatusCode buildEtIsolations( const std::string& mcEvtName,
                const HepMC::GenEvent* genEvt,
                const std::size_t genIdx,
                TruthEtIsolations& etIsols, 
                ITruthIsolationTool::ParticleSelect sel );

  void computeIso( const std::list<HepMC::ConstGenParticlePtr>& parts, 
           const HepMC::ConstGenParticlePtr& p,
           TruthEtIsolations& etIsolations, 
           ITruthIsolationTool::ParticleSelect sel );

  void setupTruthEtIsolationsPrefix( Gaudi::Details::PropertyBase& truthEtIsolationsPrefix );

  void setupMcEventsOutput( Gaudi::Details::PropertyBase& mcEventsOutputName );

  // Private data: 
 private: 

  const HepPDT::ParticleDataTable * m_pdt;

  StringProperty m_prefix;

  StringProperty m_mcEventsOutputName;

  DoubleProperty m_ptGamMin;
 
  typedef std::unordered_map<std::string, std::string> EtIsolMap_t;
  EtIsolMap_t m_etIsolMap;
 
}; 
 
 
#endif