#include <TruthIsolationTool.h>

Inheritance diagram for DerivationFramework::TruthIsolationTool:

Collaboration diagram for DerivationFramework::TruthIsolationTool:

Public Member Functions
virtual StatusCode	initialize () override final
virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Member Functions
void	calcIsos (const xAOD::TruthParticle particle, const std::vector< const xAOD::TruthParticle > &, std::vector< float > &) const

Static Private Member Functions
static float	calculateDeltaR2 (const xAOD::IParticle *p1, float eta2, float phi2)

Private Attributes
SG::ReadHandleKey< xAOD::TruthParticleContainer >	m_isoParticlesKey {this, "isoParticlesKey", "TruthParticles", "Name of TruthParticle key for input"}
	Parameter: input collection key.
SG::ReadHandleKey< xAOD::TruthParticleContainer >	m_allParticlesKey {this, "allParticlesKey", "TruthParticles", "Name of Truthparticle key to find in iso cone"}
	Parameter: input collection key for particles used in calculation.
SG::WriteDecorHandleKeyArray< xAOD::TruthParticleContainer, float >	m_isoDecorKeys {this, "DoNotSet_isoDecorKeys", {}, "WriteDecorHandleKeyArray - set internally but must be property"}
	Decor handle key array.
Gaudi::Property< std::vector< float > >	m_coneSizes {this, "IsolationConeSizes", {0.2}, "Vector of sizes of dR cone in which to include particles"}
	Parameter: Cone size for Isolation.
Gaudi::Property< bool >	m_chargedOnly {this, "ChargedParticlesOnly", false, "Only keep charged particles in isolation cone"}
	Parameter: only use charged particles for iso?
Gaudi::Property< std::vector< int > >	m_listOfPIDs {this, "particleIDsToCalculate", {11,13,22}, "List of the pdgIDs of particles for which to calculate isolation"}
	Parameter: List of pdgIDs of particles to dress.
Gaudi::Property< std::vector< int > >	m_excludeFromCone {this, "excludeIDsFromCone", {}, "List of the pdgIDs of particles to exclude from the cone when calculating isolation"}
	Parameter: List of pdgIDs to exclude from cone calculation.
Gaudi::Property< std::string >	m_isoVarNamePrefix {this, "IsolationVarNamePrefix", "", "Prefix of name of the variable to add to output xAOD"}
	Parameter: name of output variable.
Gaudi::Property< bool >	m_includeNonInteracting {this, "IncludeNonInteracting", false, "Include non-interacting particles in the isolation definition"}
	Parameter: Include non-interacting particles?
Gaudi::Property< bool >	m_variableR {this, "VariableR", false, "Use radius that shrinks with pT in isolation"}
	Parameter: Use variable radius?
std::vector< float >	m_coneSizesSort

Detailed Description

Definition at line 21 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

Member Function Documentation

◆ addBranches()

StatusCode TruthIsolationTool::addBranches ( const EventContext & ctx ) const

finaloverridevirtual

Even if this is with some dummy value

Definition at line 43 of file DerivationFramework/DerivationFrameworkMCTruth/src/TruthIsolationTool.cxx.

{
    // Event context
 
    // Retrieve the truth collections
    SG::ReadHandle<xAOD::TruthParticleContainer> isoTruthParticles(m_isoParticlesKey, ctx);
    if (!isoTruthParticles.isValid()) {
      ATH_MSG_ERROR("Couldn't retrieve collection with name " << m_isoParticlesKey);
      return StatusCode::FAILURE;
    }
 
    SG::ReadHandle<xAOD::TruthParticleContainer> allTruthParticles(m_allParticlesKey, ctx);
    if (!allTruthParticles.isValid()) {
      ATH_MSG_ERROR("Couldn't retrieve collection with name " << m_allParticlesKey);
      return StatusCode::FAILURE;
    }
 
    //get struct of helper functions
    DerivationFramework::DecayGraphHelper decayHelper;
 
    // Isolation is applied to selected particles only
    std::vector<const xAOD::TruthParticle*> listOfParticlesForIso;
    decayHelper.constructListOfFinalParticles(isoTruthParticles.ptr(), listOfParticlesForIso, m_listOfPIDs);
 
    // Vectors to store particles which will be dressed
    std::vector<const xAOD::TruthParticle*>  candidateParticlesList;
 
    std::vector<int> emptyList;
    //make a list of all candidate particles that could fall inside the cone of the particle of interest from listOfParticlesForIso
    decayHelper.constructListOfFinalParticles(allTruthParticles.ptr(), candidateParticlesList, emptyList, true, m_chargedOnly);
 
    std::vector<SG::WriteDecorHandle< xAOD::TruthParticleContainer, float > >
      decorator_iso = m_isoDecorKeys.makeHandles (ctx);
 
    //All isolation must filled for all Particles.
    for ( unsigned int icone = 0; icone < m_coneSizesSort.size(); ++icone ) {
      for (const auto* part : *isoTruthParticles) {
        decorator_iso.at(icone)(*part) = -1;
      }
    }
 
    // Standard particle loop over final state particles of interest
    for (const auto* part : listOfParticlesForIso) {
      std::vector<float> isolationsCalcs(m_coneSizesSort.size(), 0.0);
      calcIsos(part, candidateParticlesList, isolationsCalcs);
      for ( unsigned int icone = 0; icone < m_coneSizesSort.size(); ++icone ) {
        decorator_iso.at(icone)(*part) = isolationsCalcs.at(icone);
      }
    }
 
    return StatusCode::SUCCESS;
}

◆ calcIsos()

void TruthIsolationTool::calcIsos	(	const xAOD::TruthParticle *	particle,
		const std::vector< const xAOD::TruthParticle * > &	candidateParticlesList,
		std::vector< float > &	isoCalcs ) const

private

Definition at line 97 of file DerivationFramework/DerivationFrameworkMCTruth/src/TruthIsolationTool.cxx.

{
    //given a bare TruthParticle particle, calculate the isolation(s) using the
    //particles in candidateParticlesList, filling isoCalcs in order corresponding
    //to the sorted cone sizes
 
    float part_eta = particle->eta();
    float part_phi = particle->phi();
    for (const auto* cand_part : candidateParticlesList) {
      if (find(m_excludeFromCone.begin(), m_excludeFromCone.end(), cand_part->pdgId()) != m_excludeFromCone.end()) {
        //skip if we find a particle in the exclude list
        continue;
      }
      if (!m_includeNonInteracting && !MC::isInteracting(cand_part->pdgId())){
        // Do not include non-interacting particles, and this particle is non-interacting
        continue;
      }
      if (!HepMC::is_same_particle(cand_part,particle)) {
        //iteration over sorted cone sizes
        for ( unsigned int icone = 0; icone < m_coneSizesSort.size(); ++icone ) {
          const float dr2 = calculateDeltaR2(cand_part, part_eta, part_phi);
          const float coneSize = m_coneSizesSort.at(icone);
          if (dr2 < coneSize*coneSize &&
              (!m_variableR || dr2*particle->pt()*particle->pt() < 100000000.)) {
            //sum the transverse momenta
            isoCalcs.at(icone) = isoCalcs.at(icone) + cand_part->pt();
          } else {
            //break out of the loop safely since the cone sizes are sorted descending
            break;
          }
        }
      }
    }
    return;
}

◆ calculateDeltaR2()

float TruthIsolationTool::calculateDeltaR2	(	const xAOD::IParticle *	p1,
		float	eta2,
		float	phi2 )

staticprivate

Definition at line 135 of file DerivationFramework/DerivationFrameworkMCTruth/src/TruthIsolationTool.cxx.

{
  //calculate dR^2 this way to hopefully do fewer sqrt and TVector3::Pseudorapidity calls
  float phi1 = p1->phi();
  float eta1 = p1->eta();
  float deltaPhi = std::abs(phi1-phi2);
  if (deltaPhi>M_PI) deltaPhi = 2.0*M_PI - deltaPhi;
  float deltaPhiSq = deltaPhi * deltaPhi;
  float deltaEtaSq = (eta1-eta2)*(eta1-eta2);
  return deltaPhiSq+deltaEtaSq;
}

◆ initialize()

StatusCode TruthIsolationTool::initialize ( )

finaloverridevirtual

Definition at line 18 of file DerivationFramework/DerivationFrameworkMCTruth/src/TruthIsolationTool.cxx.

{
 
    // Initialise input keys
    ATH_CHECK(m_isoParticlesKey.initialize());
    ATH_CHECK(m_allParticlesKey.initialize());
 
    //sort (descsending) the cone sizes vector to optimize calculation
    m_coneSizesSort = m_coneSizes;
    std::sort(m_coneSizesSort.begin(), m_coneSizesSort.end(), [](float a, float b){return a>b;});
 
    // Decorations depend on the list of cone sizes
    // FIXME set decorations properly in the configuration
    for ( auto csize_itr : m_coneSizesSort ) {
      std::ostringstream sizess;
      if (m_variableR) sizess << "var";
      sizess << m_isoVarNamePrefix.value() << (int)((csize_itr)*100.);
      m_isoDecorKeys.emplace_back(m_isoParticlesKey.key()+"."+sizess.str());
    }
    ATH_CHECK(m_isoDecorKeys.initialize());
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

◆ m_allParticlesKey

SG::ReadHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthIsolationTool::m_allParticlesKey {this, "allParticlesKey", "TruthParticles", "Name of Truthparticle key to find in iso cone"}

private

Parameter: input collection key for particles used in calculation.

Definition at line 34 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

35{this, "allParticlesKey", "TruthParticles", "Name of Truthparticle key to find in iso cone"};

◆ m_chargedOnly

Gaudi::Property<bool> DerivationFramework::TruthIsolationTool::m_chargedOnly {this, "ChargedParticlesOnly", false, "Only keep charged particles in isolation cone"}

private

Parameter: only use charged particles for iso?

Definition at line 43 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

44{this, "ChargedParticlesOnly", false, "Only keep charged particles in isolation cone"};

◆ m_coneSizes

Gaudi::Property<std::vector<float> > DerivationFramework::TruthIsolationTool::m_coneSizes {this, "IsolationConeSizes", {0.2}, "Vector of sizes of dR cone in which to include particles"}

private

Parameter: Cone size for Isolation.

Definition at line 40 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

41{this, "IsolationConeSizes", {0.2}, "Vector of sizes of dR cone in which to include particles"};

◆ m_coneSizesSort

std::vector<float> DerivationFramework::TruthIsolationTool::m_coneSizesSort

private

Definition at line 61 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

◆ m_excludeFromCone

Gaudi::Property<std::vector<int> > DerivationFramework::TruthIsolationTool::m_excludeFromCone {this, "excludeIDsFromCone", {}, "List of the pdgIDs of particles to exclude from the cone when calculating isolation"}

private

Parameter: List of pdgIDs to exclude from cone calculation.

Definition at line 49 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

50{this, "excludeIDsFromCone", {}, "List of the pdgIDs of particles to exclude from the cone when calculating isolation"};

◆ m_includeNonInteracting

Gaudi::Property<bool> DerivationFramework::TruthIsolationTool::m_includeNonInteracting {this, "IncludeNonInteracting", false, "Include non-interacting particles in the isolation definition"}

private

Parameter: Include non-interacting particles?

Definition at line 55 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

56{this, "IncludeNonInteracting", false, "Include non-interacting particles in the isolation definition"};

◆ m_isoDecorKeys

SG::WriteDecorHandleKeyArray<xAOD::TruthParticleContainer, float> DerivationFramework::TruthIsolationTool::m_isoDecorKeys {this, "DoNotSet_isoDecorKeys", {}, "WriteDecorHandleKeyArray - set internally but must be property"}

private

Decor handle key array.

Definition at line 37 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

38{this, "DoNotSet_isoDecorKeys", {}, "WriteDecorHandleKeyArray - set internally but must be property"};

◆ m_isoParticlesKey

SG::ReadHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthIsolationTool::m_isoParticlesKey {this, "isoParticlesKey", "TruthParticles", "Name of TruthParticle key for input"}

private

Parameter: input collection key.

Definition at line 31 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

32{this, "isoParticlesKey", "TruthParticles", "Name of TruthParticle key for input"};

◆ m_isoVarNamePrefix

Gaudi::Property<std::string> DerivationFramework::TruthIsolationTool::m_isoVarNamePrefix {this, "IsolationVarNamePrefix", "", "Prefix of name of the variable to add to output xAOD"}

private

Parameter: name of output variable.

Definition at line 52 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

53{this, "IsolationVarNamePrefix", "", "Prefix of name of the variable to add to output xAOD"};

◆ m_listOfPIDs

Gaudi::Property<std::vector<int> > DerivationFramework::TruthIsolationTool::m_listOfPIDs {this, "particleIDsToCalculate", {11,13,22}, "List of the pdgIDs of particles for which to calculate isolation"}

private

Parameter: List of pdgIDs of particles to dress.

Definition at line 46 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

47{this, "particleIDsToCalculate", {11,13,22}, "List of the pdgIDs of particles for which to calculate isolation"};

◆ m_variableR

Gaudi::Property<bool> DerivationFramework::TruthIsolationTool::m_variableR {this, "VariableR", false, "Use radius that shrinks with pT in isolation"}

private

Parameter: Use variable radius?

Definition at line 58 of file DerivationFramework/DerivationFrameworkMCTruth/DerivationFrameworkMCTruth/TruthIsolationTool.h.

59{this, "VariableR", false, "Use radius that shrinks with pT in isolation"};

The documentation for this class was generated from the following files:

Public Member Functions

Private Member Functions

Static Private Member Functions

Private Attributes

Detailed Description

Member Function Documentation

◆ addBranches()

◆ calcIsos()

◆ calculateDeltaR2()

◆ initialize()

Member Data Documentation

◆ m_allParticlesKey

◆ m_chargedOnly

◆ m_coneSizes

◆ m_coneSizesSort

◆ m_excludeFromCone

◆ m_includeNonInteracting

◆ m_isoDecorKeys

◆ m_isoParticlesKey

◆ m_isoVarNamePrefix

◆ m_listOfPIDs

◆ m_variableR