DerivationFramework::TVAAugmentationTool Class Reference

#include <TVAAugmentationTool.h>

Inheritance diagram for DerivationFramework::TVAAugmentationTool:

Collaboration diagram for DerivationFramework::TVAAugmentationTool:

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

Private Types
using	vtxLink_t = ElementLink< xAOD::VertexContainer >

Private Attributes
SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_trackName {this, "TrackName", "InDetTrackParticles"}
SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexName {this, "VertexName", "PrimaryVertices"}
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_vtxDec_key {this, "LinkName", m_trackName, "", "Decoration for associated vertex"}
PublicToolHandle< CP::ITrackVertexAssociationTool >	m_tool {this, "TVATool", ""}

Detailed Description

Definition at line 18 of file TVAAugmentationTool.h.

Member Typedef Documentation

vtxLink_t

using DerivationFramework::TVAAugmentationTool::vtxLink_t = ElementLink<xAOD::VertexContainer>

private

Definition at line 33 of file TVAAugmentationTool.h.

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 21 of file TVAAugmentationTool.cxx.

  {
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, vtxLink_t> vtxDec_handle(m_vtxDec_key, ctx);
 
    SG::ReadHandle<xAOD::VertexContainer> vertices{m_vertexName, ctx};
    SG::ReadHandle<xAOD::TrackParticleContainer> tracks{m_trackName, ctx};
 
    xAOD::TrackVertexAssociationMap matchMap = m_tool->getMatchMap(*tracks, *vertices);
 
    for (const xAOD::Vertex* ivtx : *vertices)
      for (const xAOD::TrackParticle* itrk : matchMap[ivtx])
        vtxDec_handle(*itrk).toContainedElement(*vertices, ivtx);
 
    return StatusCode::SUCCESS;
  }

initialize()

StatusCode DerivationFramework::TVAAugmentationTool::initialize ( )

finaloverridevirtual

Definition at line 10 of file TVAAugmentationTool.cxx.

  {
    ATH_MSG_DEBUG("Initialising TVAAugmentationTool " << name() );
    ATH_CHECK( m_trackName.initialize() );
    ATH_CHECK( m_vertexName.initialize() );
    ATH_CHECK(m_vtxDec_key.initialize());
    ATH_CHECK( m_tool.retrieve() );
 
    return StatusCode::SUCCESS;
  }

Member Data Documentation

m_tool

PublicToolHandle<CP::ITrackVertexAssociationTool> DerivationFramework::TVAAugmentationTool::m_tool {this, "TVATool", ""}

private

Definition at line 31 of file TVAAugmentationTool.h.

m_trackName

SG::ReadHandleKey<xAOD::TrackParticleContainer> DerivationFramework::TVAAugmentationTool::m_trackName {this, "TrackName", "InDetTrackParticles"}

private

Definition at line 28 of file TVAAugmentationTool.h.

m_vertexName

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::TVAAugmentationTool::m_vertexName {this, "VertexName", "PrimaryVertices"}

private

Definition at line 29 of file TVAAugmentationTool.h.

m_vtxDec_key

SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> DerivationFramework::TVAAugmentationTool::m_vtxDec_key {this, "LinkName", m_trackName, "", "Decoration for associated vertex"}

private

Definition at line 30 of file TVAAugmentationTool.h.

---

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

• TVAAugmentationTool.h
• TVAAugmentationTool.cxx