DerivationFramework::SV1TrackThinning Class Reference

#include <SV1TrackThinning.h>

Inheritance diagram for DerivationFramework::SV1TrackThinning:

Collaboration diagram for DerivationFramework::SV1TrackThinning:

Public Member Functions
	SV1TrackThinning (const std::string &t, const std::string &n, const IInterface *p)
	~SV1TrackThinning ()
virtual StatusCode	initialize () override
virtual StatusCode	finalize () override
virtual StatusCode	doThinning () const override

Private Attributes
std::atomic< unsigned int >	m_ntot
std::atomic< unsigned int >	m_npass
StringProperty	m_streamName { this, "StreamName", "", "Name of the stream being thinned" }
SG::ThinningHandleKey< xAOD::TrackParticleContainer >	m_inDetSGKey { this, "InDetTrackParticlesKey", "InDetTrackParticles", "" }
std::string	m_jetSGKey
std::string	m_selectionString

Detailed Description

Definition at line 25 of file SV1TrackThinning.h.

Constructor & Destructor Documentation

SV1TrackThinning()

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

Definition at line 20 of file SV1TrackThinning.cxx.

                              :
    base_class(t,n,p),
    m_ntot(0),
    m_npass(0),
    m_jetSGKey(""),
    m_selectionString("")
{
    declareProperty("JetKey", m_jetSGKey);
    declareProperty("SelectionString", m_selectionString);
}

~SV1TrackThinning()

DerivationFramework::SV1TrackThinning::~SV1TrackThinning

(

)

Definition at line 34 of file SV1TrackThinning.cxx.

                                                       {
}

Member Function Documentation

doThinning()

StatusCode DerivationFramework::SV1TrackThinning::doThinning ( ) const

overridevirtual

Definition at line 65 of file SV1TrackThinning.cxx.

{
    const EventContext& ctx = Gaudi::Hive::currentContext();
 
    // Retrieve main TrackParticle collection
    SG::ThinningHandle<xAOD::TrackParticleContainer> importedTrackParticles
      (m_inDetSGKey, ctx);
 
    // Check the event contains tracks
    unsigned int nTracks = importedTrackParticles->size();
    if (nTracks==0) return StatusCode::SUCCESS;
 
    // Set up a mask with the same entries as the full TrackParticle collection
    std::vector<bool> mask;
    mask.assign(nTracks, false); // default: don't keep any tracks
    m_ntot += nTracks;
 
    const xAOD::JetContainer* importedJets(0);
    if (evtStore()->retrieve(importedJets,m_jetSGKey).isFailure()) {
        ATH_MSG_ERROR("No jet collection with name " << m_jetSGKey << " found in StoreGate!");
        return StatusCode::FAILURE;
    }
 
    unsigned int nJets(importedJets->size());
    std::vector<const xAOD::Jet*> jetToCheck; jetToCheck.clear();
 
    // Execute the text parser if requested
    if (!m_selectionString.empty()) {
        std::vector<int> entries =  m_parser->evaluateAsVector();
        unsigned int nEntries = entries.size();
        // check the sizes are compatible
        if (nJets != nEntries ) {
            ATH_MSG_ERROR("Sizes incompatible! Are you sure your selection string used jets??");
            return StatusCode::FAILURE;
        } else {
            // identify which jets to keep for the thinning check
            for (unsigned int i=0; i<nJets; ++i) if (entries[i]==1) jetToCheck.push_back((*importedJets)[i]);
        }
    }
 
    // Retrieve containers
    // ... jets
    if (m_selectionString=="") { // check all jets as user didn't provide a selection string
        for (xAOD::JetContainer::const_iterator jetIt=importedJets->begin(); jetIt!=importedJets->end(); ++jetIt) {
            const xAOD::BTagging* btag = xAOD::BTaggingUtilities::getBTagging( **jetIt );
            const std::vector< ElementLink<xAOD::TrackParticleContainer> >& SV1_trackParticleLinks =
                btag->SV1_TrackParticleLinks();
 
 
            for (const ElementLink<xAOD::TrackParticleContainer> &tp : SV1_trackParticleLinks) {
                int index = (*tp)->index();
                mask[index] = true;
            }
        }
    } else { // check only jets passing user selection string
        for (std::vector<const xAOD::Jet*>::iterator jetIt = jetToCheck.begin(); jetIt!=jetToCheck.end(); ++jetIt) {
            const xAOD::BTagging* btag = xAOD::BTaggingUtilities::getBTagging( **jetIt );
            const std::vector< ElementLink<xAOD::TrackParticleContainer> >& SV1_trackParticleLinks =
                btag->SV1_TrackParticleLinks();
 
 
            for (const ElementLink<xAOD::TrackParticleContainer> &tp : SV1_trackParticleLinks) {
                int index = (*tp)->index();
                mask[index] = true;
            }
        }
    }
 
    // Count up the mask contents
    for (unsigned int i=0; i<nTracks; ++i) {
        if (mask[i]) ++m_npass;
    }
 
    // Execute the thinning service based on the mask. Finish.
    importedTrackParticles.keep (mask);
 
    return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::SV1TrackThinning::finalize ( )

overridevirtual

Definition at line 56 of file SV1TrackThinning.cxx.

{
    ATH_MSG_VERBOSE("finalize() ...");
    ATH_MSG_INFO("Processed "<< m_ntot <<" tracks, "<< m_npass<< " were retained ");
    ATH_CHECK(finalizeParser() );
    return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::SV1TrackThinning::initialize ( )

overridevirtual

Definition at line 38 of file SV1TrackThinning.cxx.

{
    // Decide which collections need to be checked for ID TrackParticles
    ATH_MSG_VERBOSE("initialize() ...");
    ATH_CHECK( m_inDetSGKey.initialize (m_streamName) );
    ATH_MSG_INFO("Using " << m_inDetSGKey << "as the source collection for inner detector track particles");
    if (m_jetSGKey=="") {
        ATH_MSG_FATAL("No jet collection provided for thinning.");
        return StatusCode::FAILURE;
    } else { ATH_MSG_INFO("Inner detector track particles associated with objects in " << m_jetSGKey << " will be retained in this format with the rest being thinned away");}
 
    // Set up the text-parsing machinery for selectiong the jet directly according to user cuts
    if (!m_selectionString.empty()) {
       ATH_CHECK(initializeParser(m_selectionString) );
    }
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_inDetSGKey

SG::ThinningHandleKey<xAOD::TrackParticleContainer> DerivationFramework::SV1TrackThinning::m_inDetSGKey { this, "InDetTrackParticlesKey", "InDetTrackParticles", "" }

private

Definition at line 37 of file SV1TrackThinning.h.

m_jetSGKey

std::string DerivationFramework::SV1TrackThinning::m_jetSGKey

private

Definition at line 39 of file SV1TrackThinning.h.

m_npass

std::atomic<unsigned int> DerivationFramework::SV1TrackThinning::m_npass

private

Definition at line 34 of file SV1TrackThinning.h.

m_ntot

std::atomic<unsigned int> DerivationFramework::SV1TrackThinning::m_ntot

mutableprivate

Definition at line 34 of file SV1TrackThinning.h.

m_selectionString

std::string DerivationFramework::SV1TrackThinning::m_selectionString

private

Definition at line 39 of file SV1TrackThinning.h.

m_streamName

StringProperty DerivationFramework::SV1TrackThinning::m_streamName { this, "StreamName", "", "Name of the stream being thinned" }

private

Definition at line 35 of file SV1TrackThinning.h.

---

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

• SV1TrackThinning.h
• SV1TrackThinning.cxx