DerivationFramework::TruthMetaDataWriter Class Reference

#include <TruthMetaDataWriter.h>

Inheritance diagram for DerivationFramework::TruthMetaDataWriter:

Collaboration diagram for DerivationFramework::TruthMetaDataWriter:

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

Private Attributes
ServiceHandle< StoreGateSvc >	m_metaStore {this, "MetaDataStore", "MetaDataStore"}
	Connection to the metadata store.
ServiceHandle< IHepMCWeightSvc >	m_weightSvc {this, "HepMCWeightSvc", "HepMCWeightSvc/HepMCWeightSvc"}
	Service for retrieving the weight names.
xAOD::TruthMetaDataContainer *	m_tmd {}
	The meta data container to be written out.
Gaudi::Property< std::string >	m_metaName {this, "MetaObjectName", "TruthMetaData"}
	SG key and name for meta data.
std::unordered_set< uint32_t >	m_existingMetaDataChan
	Set for tracking the mc channels for which we already added meta data.
ServiceHandle< ITagInfoMgr >	m_tagInfoMgr
	TagInfoMgr to get information out of /TagInfo.
SG::ReadHandleKey< xAOD::EventInfo >	m_eventInfoKey {this, "EventInfoKey", "EventInfo", "EventInfo key"}

Detailed Description

Definition at line 35 of file TruthMetaDataWriter.h.

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 45 of file TruthMetaDataWriter.cxx.

{
 
    //The mcChannelNumber is used as a unique identifier for which truth meta data belongs to
    uint32_t mcChannelNumber = 0;
    // If this fails, we are running on a datatype with no EventInfo.  Such data types should
    //  definitely not be mixing MC samples, so this should be safe (will fall back to 0 above)
    SG::ReadHandle<xAOD::EventInfo> eventInfo(m_eventInfoKey, ctx);
    if (eventInfo.isPresent()) {
      mcChannelNumber = eventInfo->mcChannelNumber();
    }
 
    //Inserting in a (unordered_)set returns an <iterator, boolean> pair, where the boolean
    //is used to check if the key already exists (returns false in the case it exists)
    if( m_existingMetaDataChan.insert(mcChannelNumber).second ) {
        xAOD::TruthMetaData* md = new xAOD::TruthMetaData();
        m_tmd->push_back( md );
 
        // Get the list of weights from the metadata
        std::map<std::string,std::size_t> weight_name_map = m_weightSvc->weightNames(ctx);
 
        std::vector<std::string> orderedWeightNameVec;
        orderedWeightNameVec.reserve( weight_name_map.size() );
        for (auto& entry: weight_name_map) {
            orderedWeightNameVec.push_back(entry.first);
        }
 
        //The map from the HepMC record pairs the weight names with a corresponding index,
        //it is not guaranteed that the indices are ascending when iterating over the map
        std::sort(orderedWeightNameVec.begin(), orderedWeightNameVec.end(),
                  [&](const std::string& i, const std::string& j){return weight_name_map.at(i) < weight_name_map.at(j);});
 
        md->setMcChannelNumber(mcChannelNumber);
        md->setWeightNames( orderedWeightNameVec );

        md->setLhefGenerator( m_tagInfoMgr->findTag("lhefGenerator") );
        md->setGenerators( m_tagInfoMgr->findTag("generators") );
        md->setEvgenProcess( m_tagInfoMgr->findTag("evgenProcess") );
        md->setEvgenTune( m_tagInfoMgr->findTag("evgenTune") );
        md->setHardPDF( m_tagInfoMgr->findTag("hardPDF") );
        md->setSoftPDF( m_tagInfoMgr->findTag("softPDF") );
        // Done getting things from the TagInfo
 
    } // Done making the new truth metadata object
    return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::TruthMetaDataWriter::initialize ( )

finaloverridevirtual

Definition at line 22 of file TruthMetaDataWriter.cxx.

{
    ATH_MSG_VERBOSE("initialize() ...");
 
    ATH_CHECK( m_eventInfoKey.initialize() );
 
    // Initialize the service handles
    CHECK( m_metaStore.retrieve() );
    CHECK( m_weightSvc.retrieve() );
    CHECK( m_tagInfoMgr.retrieve() );
 
    // Create an empty truth meta data container:
    xAOD::TruthMetaDataAuxContainer* aux = new xAOD::TruthMetaDataAuxContainer();
    m_tmd = new xAOD::TruthMetaDataContainer();
    m_tmd->setStore( aux );
    // Record it in the metadata store
    CHECK( m_metaStore->record( aux, m_metaName + "Aux." ) );
    CHECK( m_metaStore->record( m_tmd, m_metaName ) );
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_eventInfoKey

SG::ReadHandleKey<xAOD::EventInfo> DerivationFramework::TruthMetaDataWriter::m_eventInfoKey {this, "EventInfoKey", "EventInfo", "EventInfo key"}

private

Definition at line 59 of file TruthMetaDataWriter.h.

{this, "EventInfoKey", "EventInfo", "EventInfo key"};

m_existingMetaDataChan

std::unordered_set<uint32_t> DerivationFramework::TruthMetaDataWriter::m_existingMetaDataChan

mutableprivate

Set for tracking the mc channels for which we already added meta data.

Definition at line 54 of file TruthMetaDataWriter.h.

m_metaName

Gaudi::Property<std::string> DerivationFramework::TruthMetaDataWriter::m_metaName {this, "MetaObjectName", "TruthMetaData"}

private

SG key and name for meta data.

Definition at line 52 of file TruthMetaDataWriter.h.

{this, "MetaObjectName", "TruthMetaData"}; // FIXME WriteHandle???

m_metaStore

ServiceHandle< StoreGateSvc > DerivationFramework::TruthMetaDataWriter::m_metaStore {this, "MetaDataStore", "MetaDataStore"}

private

Connection to the metadata store.

Definition at line 46 of file TruthMetaDataWriter.h.

{this, "MetaDataStore", "MetaDataStore"};

m_tagInfoMgr

ServiceHandle< ITagInfoMgr > DerivationFramework::TruthMetaDataWriter::m_tagInfoMgr

private

Initial value:

{
      "TagInfoMgr", name()}

TagInfoMgr to get information out of /TagInfo.

Definition at line 56 of file TruthMetaDataWriter.h.

                                             {
      "TagInfoMgr", name()};

m_tmd

xAOD::TruthMetaDataContainer* DerivationFramework::TruthMetaDataWriter::m_tmd {}

private

The meta data container to be written out.

Definition at line 50 of file TruthMetaDataWriter.h.

{};

m_weightSvc

ServiceHandle< IHepMCWeightSvc > DerivationFramework::TruthMetaDataWriter::m_weightSvc {this, "HepMCWeightSvc", "HepMCWeightSvc/HepMCWeightSvc"}

private

Service for retrieving the weight names.

Definition at line 48 of file TruthMetaDataWriter.h.

{this, "HepMCWeightSvc", "HepMCWeightSvc/HepMCWeightSvc"};

---

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

• TruthMetaDataWriter.h
• TruthMetaDataWriter.cxx