DerivationFramework::TruthPVCollectionMaker Class Reference

#include <TruthPVCollectionMaker.h>

Inheritance diagram for DerivationFramework::TruthPVCollectionMaker:

Collaboration diagram for DerivationFramework::TruthPVCollectionMaker:

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

Private Attributes
SG::ReadHandleKey< xAOD::TruthEventContainer >	m_eventsKey {this, "EventsKey", "TruthEvents"}
	Input event collection (navigates to the vertices)
SG::WriteHandleKey< xAOD::TruthVertexContainer >	m_outVtxKey {this, "NewCollectionName", ""}
	Output collection name.

Detailed Description

Definition at line 18 of file TruthPVCollectionMaker.h.

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 34 of file TruthPVCollectionMaker.cxx.

{
    // Retrieve truth collections
    SG::ReadHandle<xAOD::TruthEventContainer> importedTruthEvents{m_eventsKey, ctx};
    if (!importedTruthEvents.isValid()) {
        ATH_MSG_ERROR("No TruthEvent collection with name " << m_eventsKey.fullKey() << " found in StoreGate!");
        return StatusCode::FAILURE;
    }
 
    // Create the new vertex containers
    SG::WriteHandle<xAOD::TruthVertexContainer> writeHandleVtx{m_outVtxKey, ctx};
    ATH_CHECK(writeHandleVtx.record(std::make_unique<xAOD::TruthVertexContainer>(),
                                    std::make_unique<xAOD::TruthVertexAuxContainer>()));
    xAOD::TruthVertexContainer* newVertexCollection = writeHandleVtx.ptr();
    ATH_MSG_DEBUG( "Recorded new TruthVertexContainer with key: " << m_outVtxKey.fullKey());
 
    // Go through the events, add one vertex for each event
    for (const auto * event : *importedTruthEvents){
        // Just in case there is a place-holder
        if (!event) continue;
        // Try with the signal process vertex
        const xAOD::TruthVertex* old_vert(event->signalProcessVertex());
        // If that's not there, grab the first vertex
        if (!old_vert && event->nTruthVertices()>0){
            old_vert = event->truthVertex(0);
        }
        if (old_vert){
            // Hit -- copy it!
            // Make a new vertex and add it to the container
            xAOD::TruthVertex* xTruthVertex = new xAOD::TruthVertex();
            newVertexCollection->push_back( xTruthVertex );
            // Set properties
            xTruthVertex->setStatus(HepMC::status(old_vert));
            xTruthVertex->setUid(HepMC::uniqueID(old_vert));
            xTruthVertex->setX(old_vert->x());
            xTruthVertex->setY(old_vert->y());
            xTruthVertex->setZ(old_vert->z());
            xTruthVertex->setT(old_vert->t());
        } else {
            ATH_MSG_WARNING("No signal vertex or vertices associated to an input event!");
        }
    }
    return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::TruthPVCollectionMaker::initialize ( )

finaloverridevirtual

Definition at line 19 of file TruthPVCollectionMaker.cxx.

{
    ATH_MSG_VERBOSE("initialize() ...");
 
    // Check configuration, print errors, warning, and information for the user
    ATH_CHECK(m_eventsKey.initialize());
    ATH_MSG_INFO("Using " << m_eventsKey.fullKey() << " as the source collections for new truth collections");
    ATH_CHECK(m_outVtxKey.initialize());
    ATH_MSG_INFO("New truth vertex collection key: " << m_outVtxKey.fullKey() );
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_eventsKey

SG::ReadHandleKey<xAOD::TruthEventContainer> DerivationFramework::TruthPVCollectionMaker::m_eventsKey {this, "EventsKey", "TruthEvents"}

private

Input event collection (navigates to the vertices)

Definition at line 27 of file TruthPVCollectionMaker.h.

{this, "EventsKey", "TruthEvents"}; 

m_outVtxKey

SG::WriteHandleKey<xAOD::TruthVertexContainer> DerivationFramework::TruthPVCollectionMaker::m_outVtxKey {this, "NewCollectionName", ""}

private

Output collection name.

Definition at line 28 of file TruthPVCollectionMaker.h.

{this, "NewCollectionName", ""}; 

---

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

• TruthPVCollectionMaker.h
• TruthPVCollectionMaker.cxx