#include <BkgElectronClassification.h>

Inheritance diagram for DerivationFramework::BkgElectronClassification:

Collaboration diagram for DerivationFramework::BkgElectronClassification:

Public Member Functions
	BkgElectronClassification (const std::string &t, const std::string &n, const IInterface *p)

virtual StatusCode	initialize () override final

virtual StatusCode	addBranches () 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", "Electrons.truthPdgId", "" }

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

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

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_firstEgMotherTruthParticleLink

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

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

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

SG::WriteDecorHandleKey< xAOD::ElectronContainer >	m_lastEgMotherTruthParticleLink

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

Detailed Description

Definition at line 23 of file BkgElectronClassification.h.

Constructor & Destructor Documentation

◆ BkgElectronClassification()

DerivationFramework::BkgElectronClassification::BkgElectronClassification	(	const std::string &	t,
		const std::string &	n,
		const IInterface *	p
	)

Definition at line 15 of file BkgElectronClassification.cxx.

   : base_class(t, n, p)
 {
  
 }

Member Function Documentation

◆ addBranches()

StatusCode DerivationFramework::BkgElectronClassification::addBranches ( ) const

finaloverridevirtual

Definition at line 66 of file BkgElectronClassification.cxx.

 {
  
   const EventContext& ctx = Gaudi::Hive::currentContext();
   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 24 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());
  
   const std::string baseName = m_electronContainer.key();
   m_truthPdgId = baseName + ".truthPdgId";
   //
   m_firstEgMotherTruthType = baseName + ".firstEgMotherTruthType";
   m_firstEgMotherTruthOrigin = baseName + ".firstEgMotherTruthOrigin";
   m_firstEgMotherTruthParticleLink =
     baseName + ".firstEgMotherTruthParticleLink";
   m_firstEgMotherPdgId = baseName + ".firstEgMotherPdgId";
   //
   m_lastEgMotherTruthType = baseName + ".lastEgMotherTruthType";
   m_lastEgMotherTruthOrigin = baseName + ".lastEgMotherTruthOrigin";
   m_lastEgMotherTruthParticleLink = baseName + ".lastEgMotherTruthParticleLink";
   m_lastEgMotherPdgId = baseName + ".lastEgMotherPdgId";
  
   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 43 of file BkgElectronClassification.h.

◆ m_firstEgMotherPdgId

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

private

Definition at line 71 of file BkgElectronClassification.h.

◆ m_firstEgMotherTruthOrigin

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

private

Definition at line 64 of file BkgElectronClassification.h.

◆ m_firstEgMotherTruthParticleLink

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

private

Initial value:

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

Definition at line 66 of file BkgElectronClassification.h.

◆ m_firstEgMotherTruthType

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

private

Definition at line 62 of file BkgElectronClassification.h.

◆ m_lastEgMotherPdgId

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

private

Definition at line 82 of file BkgElectronClassification.h.

◆ m_lastEgMotherTruthOrigin

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

private

Definition at line 75 of file BkgElectronClassification.h.

◆ m_lastEgMotherTruthParticleLink

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

private

Initial value:

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

Definition at line 77 of file BkgElectronClassification.h.

◆ m_lastEgMotherTruthType

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

private

Definition at line 73 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 35 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 50 of file BkgElectronClassification.h.

◆ m_truthPdgId

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

private

Definition at line 60 of file BkgElectronClassification.h.

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

Public Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ BkgElectronClassification()

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