#include <BkgElectronClassification.h>

Inheritance diagram for DerivationFramework::BkgElectronClassification:

Collaboration diagram for DerivationFramework::BkgElectronClassification:

Public Member Functions
virtual StatusCode	initialize () override final

virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Attributes
ToolHandle< IMCTruthClassifier >	m_mcTruthClassifier
	MCTruthClassifier. More...

SG::ReadHandleKey< xAOD::ElectronContainer >	m_electronContainer
	input electron container More...

SG::ReadHandleKey< xAOD::TruthParticleContainer >	m_truthContainer
	Input truth particle container. More...

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_truthPdgId { this, "DoNotSet_truthPdgId", m_electronContainer, "truthPdgId", "" }

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_firstEgMotherTruthType { this, "firstEgMotherTruthType", m_electronContainer, "firstEgMotherTruthType", "" }

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_firstEgMotherTruthOrigin { this, "firstEgMotherTruthOrigin", m_electronContainer, "firstEgMotherTruthOrigin", "" }

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_firstEgMotherTruthParticleLink

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_firstEgMotherPdgId { this, "firstEgMotherPdgId", m_electronContainer, "firstEgMotherPdgId", "" }

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_lastEgMotherTruthType { this, "lastEgMotherTruthType", m_electronContainer, "lastEgMotherTruthType", "" }

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_lastEgMotherTruthOrigin { this, "lastEgMotherTruthOrigin", m_electronContainer, "lastEgMotherTruthOrigin", "" }

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_lastEgMotherTruthParticleLink

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_lastEgMotherPdgId { this, "lastEgMotherPdgId", m_electronContainer, "lastEgMotherPdgId", "" }

Detailed Description

Definition at line 23 of file BkgElectronClassification.h.

Member Function Documentation

◆ addBranches()

StatusCode DerivationFramework::BkgElectronClassification::addBranches ( const EventContext & ctx ) const

finaloverridevirtual

Definition at line 44 of file BkgElectronClassification.cxx.

 {
  
   SG::ReadHandle<xAOD::ElectronContainer> electrons{ m_electronContainer, ctx };
   SG::ReadHandle<xAOD::TruthParticleContainer> truthContainer{ m_truthContainer,
                                                                ctx };
  
   // Access for the already existing info
   static const SG::AuxElement::Accessor<int> tT("truthType");
   static const SG::AuxElement::Accessor<int> tO("truthOrigin");
   static const SG::AuxElement::Accessor<
     ElementLink<xAOD::TruthParticleContainer>>
     tPL("truthParticleLink");
  
   // pdg iD
   SG::WriteDecorHandle<xAOD::ElectronContainer, int> tPdgID(m_truthPdgId, ctx);
   // first mother decorations
   SG::WriteDecorHandle<xAOD::ElectronContainer, int> firstEgMotherTT(
     m_firstEgMotherTruthType, ctx);
   SG::WriteDecorHandle<xAOD::ElectronContainer, int> firstEgMotherTO(
     m_firstEgMotherTruthOrigin, ctx);
   SG::WriteDecorHandle<xAOD::ElectronContainer, int> firstEgMotherPdgID(
     m_firstEgMotherPdgId, ctx);
   SG::WriteDecorHandle<xAOD::ElectronContainer,
                        ElementLink<xAOD::TruthParticleContainer>>
     firstEgMotherTPL(m_firstEgMotherTruthParticleLink, ctx);
  
   // last mother decorations
   SG::WriteDecorHandle<xAOD::ElectronContainer, int> lastEgMotherTT(
     m_lastEgMotherTruthType, ctx);
   SG::WriteDecorHandle<xAOD::ElectronContainer, int> lastEgMotherTO(
     m_lastEgMotherTruthOrigin, ctx);
   SG::WriteDecorHandle<xAOD::ElectronContainer, int> lastEgMotherPdgID(
     m_lastEgMotherPdgId, ctx);
   SG::WriteDecorHandle<xAOD::ElectronContainer,
                        ElementLink<xAOD::TruthParticleContainer>>
     lastEgMotherTPL(m_lastEgMotherTruthParticleLink, ctx);
   //
  
   for (const xAOD::Electron* el : *electrons) {
     tPdgID(*el) = 0;
     if (tPL.isAvailable(*el) && tPL(*el).isValid()) {
       tPdgID(*el) = (*tPL(*el))->pdgId();
     }
     // Use the Helpers for electron from electron or photon
     // Add Extra Decoration from Egamma helpers in case of BkgElectron (Electron
     // coming for a photon) Go back to the first/last electron/photon Generator
     // mother and classify this one
     // First the one entering the Geant, the first we meet on the way back
     firstEgMotherTT(*el) = 0;
     firstEgMotherTO(*el) = 0;
     firstEgMotherTPL(*el) = ElementLink<xAOD::TruthParticleContainer>();
     firstEgMotherPdgID(*el) = 0;
     const xAOD::TruthParticle* firstElTruth =
       xAOD::EgammaHelpers::getBkgElectronMother(el, false);
  
     MCTruthPartClassifier::Info mcinfo(ctx);
     if (firstElTruth) {
       auto res = m_mcTruthClassifier->particleTruthClassifier(firstElTruth, &mcinfo);
       firstEgMotherTT(*el) = res.first;
       firstEgMotherTO(*el) = res.second;
       firstEgMotherPdgID(*el) = firstElTruth->pdgId();
       ElementLink<xAOD::TruthParticleContainer> link(
         firstElTruth, *truthContainer, ctx);
       firstEgMotherTPL(*el) = link;
     }
  
     // The last electron / photon  we meet on the way back towards the Generator
     // vertex
     lastEgMotherTT(*el) = 0;
     lastEgMotherTO(*el) = 0;
     lastEgMotherTPL(*el) = ElementLink<xAOD::TruthParticleContainer>();
     lastEgMotherPdgID(*el) = 0;
     const xAOD::TruthParticle* lastElTruth =
       xAOD::EgammaHelpers::getBkgElectronMother(el);
  
     if (lastElTruth) {
       auto res = m_mcTruthClassifier->particleTruthClassifier(lastElTruth, &mcinfo);
       lastEgMotherTT(*el) = res.first;
       lastEgMotherTO(*el) = res.second;
       lastEgMotherPdgID(*el) = lastElTruth->pdgId();
       ElementLink<xAOD::TruthParticleContainer> link(
         lastElTruth, *truthContainer, ctx);
       lastEgMotherTPL(*el) = link;
     }
   }
   return StatusCode::SUCCESS;
 }

◆ initialize()

StatusCode DerivationFramework::BkgElectronClassification::initialize ( )

finaloverridevirtual

Definition at line 16 of file BkgElectronClassification.cxx.

 {
  
   ATH_MSG_DEBUG("Initializing " << name() << "...");
   ATH_CHECK(m_mcTruthClassifier.retrieve());
   ATH_MSG_DEBUG("Retrieved tool " << m_mcTruthClassifier);
  
   ATH_CHECK(m_electronContainer.initialize());
   ATH_CHECK(m_truthContainer.initialize());
  
   ATH_CHECK(m_truthPdgId.initialize());
   //
   ATH_CHECK(m_firstEgMotherTruthType.initialize());
   ATH_CHECK(m_firstEgMotherTruthOrigin.initialize());
   ATH_CHECK(m_firstEgMotherTruthParticleLink.initialize());
   ATH_CHECK(m_firstEgMotherPdgId.initialize());
   //
   ATH_CHECK(m_lastEgMotherTruthType.initialize());
   ATH_CHECK(m_lastEgMotherTruthOrigin.initialize());
   ATH_CHECK(m_lastEgMotherTruthParticleLink.initialize());
   ATH_CHECK(m_lastEgMotherPdgId.initialize());
  
   ATH_MSG_DEBUG("Initialization successful");
  
   return StatusCode::SUCCESS;
 }

Member Data Documentation

◆ m_electronContainer

SG::ReadHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_electronContainer

private

Initial value:

{
      this,
      "ElectronContainerName",
      "Electrons",
      "Input Electrons"
    }

input electron container

Definition at line 42 of file BkgElectronClassification.h.

◆ m_firstEgMotherPdgId

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_firstEgMotherPdgId { this, "firstEgMotherPdgId", m_electronContainer, "firstEgMotherPdgId", "" }

private

Definition at line 69 of file BkgElectronClassification.h.

◆ m_firstEgMotherTruthOrigin

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_firstEgMotherTruthOrigin { this, "firstEgMotherTruthOrigin", m_electronContainer, "firstEgMotherTruthOrigin", "" }

private

Definition at line 62 of file BkgElectronClassification.h.

◆ m_firstEgMotherTruthParticleLink

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_firstEgMotherTruthParticleLink

private

Initial value:

{ this,
      "firstEgMotherTruthParticleLink",
      m_electronContainer, "firstEgMotherTruthParticleLink",
      "" }

Definition at line 64 of file BkgElectronClassification.h.

◆ m_firstEgMotherTruthType

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_firstEgMotherTruthType { this, "firstEgMotherTruthType", m_electronContainer, "firstEgMotherTruthType", "" }

private

Definition at line 60 of file BkgElectronClassification.h.

◆ m_lastEgMotherPdgId

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_lastEgMotherPdgId { this, "lastEgMotherPdgId", m_electronContainer, "lastEgMotherPdgId", "" }

private

Definition at line 80 of file BkgElectronClassification.h.

◆ m_lastEgMotherTruthOrigin

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_lastEgMotherTruthOrigin { this, "lastEgMotherTruthOrigin", m_electronContainer, "lastEgMotherTruthOrigin", "" }

private

Definition at line 73 of file BkgElectronClassification.h.

◆ m_lastEgMotherTruthParticleLink

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_lastEgMotherTruthParticleLink

private

Initial value:

{ this,
      "lastEgMotherTruthParticleLink",
      m_electronContainer, "lastEgMotherTruthParticleLink",
      "" }

Definition at line 75 of file BkgElectronClassification.h.

◆ m_lastEgMotherTruthType

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_lastEgMotherTruthType { this, "lastEgMotherTruthType", m_electronContainer, "lastEgMotherTruthType", "" }

private

Definition at line 71 of file BkgElectronClassification.h.

◆ m_mcTruthClassifier

ToolHandle<IMCTruthClassifier> DerivationFramework::BkgElectronClassification::m_mcTruthClassifier

private

Initial value:

{
      this,
        "MCTruthClassifierTool",
        "",
        "Handle to the MCTruthClassifier"
        }

MCTruthClassifier.

Definition at line 34 of file BkgElectronClassification.h.

◆ m_truthContainer

SG::ReadHandleKey<xAOD::TruthParticleContainer> DerivationFramework::BkgElectronClassification::m_truthContainer

private

Initial value:

{
      this,
      "TruthParticleContainerName",
      "TruthParticles",
      "Input Truth Particles"
    }

Input truth particle container.

Definition at line 49 of file BkgElectronClassification.h.

◆ m_truthPdgId

SG::WriteDecorHandleKey<xAOD::ElectronContainer> DerivationFramework::BkgElectronClassification::m_truthPdgId { this, "DoNotSet_truthPdgId", m_electronContainer, "truthPdgId", "" }

private

Definition at line 58 of file BkgElectronClassification.h.

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

Public Member Functions

Private Attributes

Detailed Description

Member Function Documentation

◆ addBranches()

◆ initialize()

Member Data Documentation

◆ m_electronContainer

◆ m_firstEgMotherPdgId

◆ m_firstEgMotherTruthOrigin

◆ m_firstEgMotherTruthParticleLink

◆ m_firstEgMotherTruthType

◆ m_lastEgMotherPdgId

◆ m_lastEgMotherTruthOrigin

◆ m_lastEgMotherTruthParticleLink

◆ m_lastEgMotherTruthType

◆ m_mcTruthClassifier

◆ m_truthContainer

◆ m_truthPdgId