DerivationFramework::AugmentationToolExample Class Reference

#include <AugmentationToolExample.h>

Inheritance diagram for DerivationFramework::AugmentationToolExample:

Collaboration diagram for DerivationFramework::AugmentationToolExample:

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

Private Attributes
SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexContainerKey {this, "VertexContainer", "PrimaryVertices",""}
SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_trackPartContainerKey {this, "TrackParticleContainer", "InDetTrackParticles", "Container to be decorated"}
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_exampleDecorKey {this, "ExampleDecorKey", m_trackPartContainerKey, "DFDecoratorExample", "Decoration"}
SG::WriteHandleKey< std::vector< float > >	m_decisionKey {this, "DecisionKey", "DFAugmentationExample", "Write decision to SG for access by downstream algs"}

Detailed Description

Definition at line 17 of file AugmentationToolExample.h.

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 32 of file AugmentationToolExample.cxx.

  {
    // Set up the vector
    std::unique_ptr<std::vector<float> > track_z0_PV(new std::vector<float>());
 
    // Set up the decorators
    SG::WriteDecorHandle< xAOD::TrackParticleContainer, float > decorator(m_exampleDecorKey, ctx);
 
    // CALCULATION OF THE NEW VARIABLE
    // Get Primary vertex
    SG::ReadHandle<xAOD::VertexContainer> vertices(m_vertexContainerKey, ctx);
    if (!vertices.isValid()) {
      ATH_MSG_ERROR ("Couldn't retrieve VertexContainer with key PrimaryVertices");
      return StatusCode::FAILURE;
    }
 
    const xAOD::Vertex* pv{};
    for (const xAOD::Vertex* vx : *vertices) {
      if (vx->vertexType() == xAOD::VxType::PriVtx) {
        pv = vx;
        break;
      }
    }
 
    // Get the track container
    SG::ReadHandle<xAOD::TrackParticleContainer> tracks(m_trackPartContainerKey, ctx);
    if (!tracks.isValid()) {
      ATH_MSG_ERROR ("Couldn't retrieve TrackParticleContainer with key InDetTrackParticles");
      return StatusCode::FAILURE;
    }
 
    // Get track z0 w.r.t PV: this is what we're adding
    for (const auto *trackParticle : *tracks) {
      if (pv) {
        float z0wrtPV = trackParticle->z0() + trackParticle->vz() - pv->z(); // CALCULATE THE QUANTITY
        track_z0_PV->push_back(z0wrtPV); // ADD TO VECTOR
        decorator(*trackParticle) = z0wrtPV; // DECORATE THE TRACK
      } else {
        track_z0_PV->push_back(999.);
        decorator(*trackParticle) = 999.;
      }
    }
 
    // Write decision to SG for access by downstream algs
    SG::WriteHandle<std::vector<float> > decision(m_decisionKey, ctx);
    if (!decision.isValid()) {
      ATH_MSG_ERROR("Tool is attempting to write StoreGate keys which already exists. Please use a different key");
      return StatusCode::FAILURE;
    } else {
      decision = std::move(track_z0_PV);
    }
 
    return StatusCode::SUCCESS;
  }

initialize()

StatusCode DerivationFramework::AugmentationToolExample::initialize ( )

finaloverridevirtual

Definition at line 23 of file AugmentationToolExample.cxx.

  {
    ATH_CHECK(m_vertexContainerKey.initialize());
    ATH_CHECK(m_trackPartContainerKey.initialize());
    ATH_CHECK(m_exampleDecorKey.initialize());
    ATH_CHECK(m_decisionKey.initialize());
    return StatusCode::SUCCESS;
  }

Member Data Documentation

m_decisionKey

SG::WriteHandleKey<std::vector<float> > DerivationFramework::AugmentationToolExample::m_decisionKey {this, "DecisionKey", "DFAugmentationExample", "Write decision to SG for access by downstream algs"}

private

Definition at line 27 of file AugmentationToolExample.h.

m_exampleDecorKey

SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> DerivationFramework::AugmentationToolExample::m_exampleDecorKey {this, "ExampleDecorKey", m_trackPartContainerKey, "DFDecoratorExample", "Decoration"}

private

Definition at line 26 of file AugmentationToolExample.h.

m_trackPartContainerKey

SG::ReadHandleKey<xAOD::TrackParticleContainer> DerivationFramework::AugmentationToolExample::m_trackPartContainerKey {this, "TrackParticleContainer", "InDetTrackParticles", "Container to be decorated"}

private

Definition at line 25 of file AugmentationToolExample.h.

m_vertexContainerKey

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::AugmentationToolExample::m_vertexContainerKey {this, "VertexContainer", "PrimaryVertices",""}

private

Definition at line 24 of file AugmentationToolExample.h.

---

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

• AugmentationToolExample.h
• AugmentationToolExample.cxx