DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_TRUTHISOLATIONTOOL_H
#define DERIVATIONFRAMEWORK_TRUTHISOLATIONTOOL_H
 
#include <string>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandleKeyArray.h"
#include "Gaudi/Property.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "DerivationFrameworkMCTruth/DecayGraphHelper.h"
#include "xAODTruth/TruthParticleContainer.h"
#include "GaudiKernel/ToolHandle.h"
 
namespace DerivationFramework {
 
  class TruthIsolationTool : public AthAlgTool, public IAugmentationTool {
    public: 
      TruthIsolationTool(const std::string& t, const std::string& n, const IInterface* p);
      ~TruthIsolationTool();
      virtual StatusCode initialize() override;
      virtual StatusCode addBranches() const override;
 
    private:
      SG::ReadHandleKey<xAOD::TruthParticleContainer> m_isoParticlesKey
        {this, "isoParticlesKey", "TruthParticles",  "Name of TruthParticle key for input"};      
      SG::ReadHandleKey<xAOD::TruthParticleContainer> m_allParticlesKey
        {this, "allParticlesKey", "TruthParticles", "Name of Truthparticle key to find in iso cone"}; 
      SG::WriteDecorHandleKeyArray<xAOD::TruthParticleContainer, float> m_isoDecorKeys
        {this, "DoNotSet_isoDecorKeys", {}, "WriteDecorHandleKeyArray - set internally but must be property"}; 
      Gaudi::Property<std::vector<float> > m_coneSizes
        {this, "IsolationConeSizes", {0.2}, "Vector of sizes of dR cone in which to include particles"};
      Gaudi::Property<bool> m_chargedOnly
        {this, "ChargedParticlesOnly", false, "Only keep charged particles in isolation cone"};
      Gaudi::Property<std::vector<int> > m_listOfPIDs
        {this, "particleIDsToCalculate", {11,13,22}, "List of the pdgIDs of particles for which to calculate isolation"};
      Gaudi::Property<std::vector<int> > m_excludeFromCone
        {this, "excludeIDsFromCone", {}, "List of the pdgIDs of particles to exclude from the cone when calculating isolation"};
      Gaudi::Property<std::string> m_isoVarNamePrefix
        {this, "IsolationVarNamePrefix", "", "Prefix of name of the variable to add to output xAOD"};
      Gaudi::Property<bool> m_includeNonInteracting
        {this, "IncludeNonInteracting", false, "Include non-interacting particles in the isolation definition"};
      Gaudi::Property<bool>  m_variableR
        {this, "VariableR", false, "Use radius that shrinks with pT in isolation"};
 
      std::vector<float> m_coneSizesSort;
 
      //private helper functions
      void calcIsos(const xAOD::TruthParticle* particle,
          const std::vector<const xAOD::TruthParticle*> &,
          std::vector<float>&) const;
      static float calculateDeltaR2(const xAOD::IParticle *p1, float eta2, float phi2) ;
  }; 
}
 
#endif // DERIVATIONFRAMEWORK_TRUTHISOLATIONTOOL_H