DerivationFramework::CollectionMakerHelpers Class Reference

#include <CollectionMakerHelpers.h>

Collaboration diagram for DerivationFramework::CollectionMakerHelpers:

Static Public Member Functions
static int	addTruthParticle (const xAOD::TruthParticle &oldPart, xAOD::TruthParticleContainer *partCont, xAOD::TruthVertexContainer *vertCont, std::vector< int > &seenParticles, const int generations, bool includeVertex=true)
	< Helper function to add truth particles to a collection More...
static int	addTruthVertex (const xAOD::TruthVertex &oldVert, xAOD::TruthParticleContainer *partCont, xAOD::TruthVertexContainer *vertCont, std::vector< int > &seenParticles, const int generations)
	Helper function to set up a truth particle based on an old particle. More...
static xAOD::TruthParticle *	setupTruthParticle (const xAOD::TruthParticle &oldPart, xAOD::TruthParticleContainer *partCont)

Detailed Description

Definition at line 20 of file CollectionMakerHelpers.h.

Member Function Documentation

addTruthParticle()

int DerivationFramework::CollectionMakerHelpers::addTruthParticle ( const xAOD::TruthParticle &  oldPart, xAOD::TruthParticleContainer *  partCont, xAOD::TruthVertexContainer *  vertCont, std::vector< int > &  seenParticles, const int  generations, bool  includeVertex = true  )

static

< Helper function to add truth particles to a collection

Helper function to add truth vertices to a collection

Definition at line 55 of file CollectionMakerHelpers.cxx.

                                                                                                              {
    // See if we've seen it - note, could also do this with a unary function on the container itself
    if (std::find(seenParticles.begin(), seenParticles.end(), HepMC::uniqueID(&oldPart)) != seenParticles.end()){
      for (size_t p=0;p<partCont->size();++p){
        // Was it a hit?
        const xAOD::TruthParticle *theParticle = (*partCont)[p];
        if (HepMC::is_same_particle(theParticle,&oldPart)) return p;
      } // Look through the old container
    } // Found it in the old container
    // Now we have seen it
    seenParticles.push_back(HepMC::uniqueID(&oldPart));
    // Make a nw truth particle
    xAOD::TruthParticle* xTruthParticle = setupTruthParticle(oldPart,partCont);
    // Make a link to this particle
    int myIndex = partCont->size()-1;
    ElementLink<xAOD::TruthParticleContainer> eltp(*partCont, myIndex);
    // Decay vertex information
    if (oldPart.hasDecayVtx() && includeVertex) {
        int vertIndex = CollectionMakerHelpers::addTruthVertex( *oldPart.decayVtx(), partCont, vertCont, seenParticles, generations);
        ElementLink<xAOD::TruthVertexContainer> eltv( *vertCont, vertIndex );
        xTruthParticle->setDecayVtxLink( eltv );
        (*vertCont)[vertIndex]->addIncomingParticleLink( eltp );
    }
    // Return a link to this particle
    return myIndex;
}

addTruthVertex()

int DerivationFramework::CollectionMakerHelpers::addTruthVertex ( const xAOD::TruthVertex &  oldVert, xAOD::TruthParticleContainer *  partCont, xAOD::TruthVertexContainer *  vertCont, std::vector< int > &  seenParticles, const int  generations  )

static

Helper function to set up a truth particle based on an old particle.

Definition at line 25 of file CollectionMakerHelpers.cxx.

                                                                                        {
    // Make a new vertex and add it to the container
    xAOD::TruthVertex* xTruthVertex = new xAOD::TruthVertex();
    vertCont->push_back( xTruthVertex );
    // Get a link to this vertex -- will be used to set production vertices on all the next particles
    int myIndex = vertCont->size()-1;
    ElementLink<xAOD::TruthVertexContainer> eltv(*vertCont, myIndex);
    // Set properties
    xTruthVertex->setId(HepMC::status(oldVert));
    xTruthVertex->setBarcode(HepMC::barcode(&oldVert)); // FIXME barcode-based
    xTruthVertex->setX(oldVert.x());
    xTruthVertex->setY(oldVert.y());
    xTruthVertex->setZ(oldVert.z());
    xTruthVertex->setT(oldVert.t());
    // If we are done, then stop here
    if (generations==0) return myIndex;
    // Add all the outgoing particles
    for (size_t n=0;n<oldVert.nOutgoingParticles();++n){
        if (!oldVert.outgoingParticle(n)) continue; // Just in case we removed some truth particles, e.g. G4 decays
        int partIndex = CollectionMakerHelpers::addTruthParticle( *oldVert.outgoingParticle(n), partCont, vertCont, seenParticles, generations-1);
        ElementLink<xAOD::TruthParticleContainer> eltp( *partCont, partIndex);
        xTruthVertex->addOutgoingParticleLink( eltp );
        (*partCont)[partIndex]->setProdVtxLink( eltv );
    }
    // Return a link to this vertex
    return myIndex;
}

setupTruthParticle()

xAOD::TruthParticle * DerivationFramework::CollectionMakerHelpers::setupTruthParticle ( const xAOD::TruthParticle &  oldPart, xAOD::TruthParticleContainer *  partCont  )

static

Definition at line 84 of file CollectionMakerHelpers.cxx.

                                                                                                                                                      {
    // Set up decorators
    const static SG::Decorator< unsigned int > originDecorator("classifierParticleOrigin");
    const static SG::Decorator< unsigned int > typeDecorator("classifierParticleType");
    const static SG::Decorator< unsigned int > outcomeDecorator("classifierParticleOutCome");
    const static SG::Decorator< int > motherIDDecorator("motherID");
    const static SG::Decorator< int > daughterIDDecorator("daughterID");
 
    static const SG::ConstAccessor<unsigned int> classifierParticleTypeAcc("classifierParticleType");
    static const SG::ConstAccessor<unsigned int> classifierParticleOriginAcc("classifierParticleOrigin");
    static const SG::ConstAccessor<unsigned int> classifierParticleOutComeAcc("classifierParticleOutCome");
 
    // Make a truth particle and add it to the container
    xAOD::TruthParticle* xTruthParticle = new xAOD::TruthParticle();
    partCont->push_back( xTruthParticle );
    // Fill with numerical content
    xTruthParticle->setPdgId(oldPart.pdgId());
    xTruthParticle->setBarcode(HepMC::barcode(&oldPart)); // FIXME barcode-based
    xTruthParticle->setStatus(oldPart.status());
    xTruthParticle->setM(oldPart.m());
    xTruthParticle->setPx(oldPart.px());
    xTruthParticle->setPy(oldPart.py());
    xTruthParticle->setPz(oldPart.pz());
    xTruthParticle->setE(oldPart.e());
    // Copy over the polarization information if it's there
    if (oldPart.polarization().valid()){
        xTruthParticle->setPolarizationParameter( oldPart.polarizationParameter( xAOD::TruthParticle::polarizationPhi ) , xAOD::TruthParticle::polarizationPhi );
        xTruthParticle->setPolarizationParameter( oldPart.polarizationParameter( xAOD::TruthParticle::polarizationTheta ) , xAOD::TruthParticle::polarizationTheta );
    }
    // Copy over the decorations if they are available
    typeDecorator(*xTruthParticle) = classifierParticleTypeAcc.withDefault(oldPart, 0);
    originDecorator(*xTruthParticle) = classifierParticleOriginAcc.withDefault(oldPart, 0);
    outcomeDecorator(*xTruthParticle) = classifierParticleOutComeAcc.withDefault(oldPart, 0);
    return xTruthParticle;
}

---

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

• CollectionMakerHelpers.h
• CollectionMakerHelpers.cxx