DerivationFramework::TVAAugmentationTool Class Reference

#include <TVAAugmentationTool.h>

Inheritance diagram for DerivationFramework::TVAAugmentationTool:

Collaboration diagram for DerivationFramework::TVAAugmentationTool:

Public Member Functions
	TVAAugmentationTool (const std::string &t, const std::string &n, const IInterface *p)
virtual StatusCode	initialize ()
virtual StatusCode	addBranches () const

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

Private Attributes
std::string	m_linkName
std::string	m_trackName
std::string	m_vertexName
ToolHandle< CP::ITrackVertexAssociationTool >	m_tool
SG::WriteDecorHandleKey< xAOD::TrackParticleContainer >	m_vtxDec_key {this, "vtxDecKey", "", "Decoration for associated vertex"}

Detailed Description

Definition at line 17 of file TVAAugmentationTool.h.

Member Typedef Documentation

vtxLink_t

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

private

Definition at line 31 of file TVAAugmentationTool.h.

Constructor & Destructor Documentation

TVAAugmentationTool()

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

Definition at line 11 of file TVAAugmentationTool.cxx.

                          :
    base_class(t, n, p)
  {
    declareProperty("LinkName", m_linkName, "The name of the output links");
    declareProperty("TrackName", m_trackName="InDetTrackParticles");
    declareProperty("VertexName", m_vertexName="PrimaryVertices");
    declareProperty("TVATool", m_tool);
  }

Member Function Documentation

addBranches()

StatusCode DerivationFramework::TVAAugmentationTool::addBranches ( ) const

virtual

Definition at line 34 of file TVAAugmentationTool.cxx.

  {
 
    SG::WriteDecorHandle<xAOD::TrackParticleContainer, vtxLink_t> vtxDec_handle(m_vtxDec_key);
 
    const xAOD::VertexContainer* vertices = nullptr;
    ATH_CHECK(evtStore()->retrieve(vertices, m_vertexName) );
    const xAOD::TrackParticleContainer* tracks = nullptr;
    ATH_CHECK(evtStore()->retrieve(tracks, m_trackName) );
 
    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 ( )

virtual

Definition at line 23 of file TVAAugmentationTool.cxx.

  {
    ATH_MSG_INFO("Initialising TVAAugmentationTool " << name() );
    ATH_CHECK( m_tool.retrieve() );
 
    m_vtxDec_key = m_trackName + "." + m_linkName;
    ATH_CHECK(m_vtxDec_key.initialize());
 
    return StatusCode::SUCCESS;
  }

Member Data Documentation

m_linkName

std::string DerivationFramework::TVAAugmentationTool::m_linkName

private

Definition at line 26 of file TVAAugmentationTool.h.

m_tool

ToolHandle<CP::ITrackVertexAssociationTool> DerivationFramework::TVAAugmentationTool::m_tool

private

Definition at line 29 of file TVAAugmentationTool.h.

m_trackName

std::string DerivationFramework::TVAAugmentationTool::m_trackName

private

Definition at line 27 of file TVAAugmentationTool.h.

m_vertexName

std::string DerivationFramework::TVAAugmentationTool::m_vertexName

private

Definition at line 28 of file TVAAugmentationTool.h.

m_vtxDec_key

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

private

Definition at line 32 of file TVAAugmentationTool.h.

---

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

• TVAAugmentationTool.h
• TVAAugmentationTool.cxx