DerivationFramework::DiTauTrackParticleThinning Class Reference

#include <DiTauTrackParticleThinning.h>

Inheritance diagram for DerivationFramework::DiTauTrackParticleThinning:

Collaboration diagram for DerivationFramework::DiTauTrackParticleThinning:

Public Member Functions
	DiTauTrackParticleThinning (const std::string &t, const std::string &n, const IInterface *p)
virtual	~DiTauTrackParticleThinning ()
virtual StatusCode	initialize () override
virtual StatusCode	finalize () override
virtual StatusCode	doThinning (const EventContext &ctx) 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::ReadHandleKey< xAOD::DiTauJetContainer >	m_ditauKey { this, "DiTauKey","",""}
SG::ThinningHandleKey< xAOD::TrackParticleContainer >	m_inDetSGKey { this, "InDetTrackParticlesKey", "InDetTrackParticles", "" }
Gaudi::Property< std::string >	m_selectionString { this, "SelectionString", "", ""}

Detailed Description

Definition at line 29 of file DiTauTrackParticleThinning.h.

Constructor & Destructor Documentation

DiTauTrackParticleThinning()

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

Definition at line 19 of file DiTauTrackParticleThinning.cxx.

                                                                                                  :
base_class(t,n,p)
{}

~DiTauTrackParticleThinning()

DerivationFramework::DiTauTrackParticleThinning::~DiTauTrackParticleThinning ( )

virtualdefault

Member Function Documentation

doThinning()

StatusCode DerivationFramework::DiTauTrackParticleThinning::doThinning ( const EventContext & ctx ) const

overridevirtual

Definition at line 54 of file DiTauTrackParticleThinning.cxx.

{
 
 
    // 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;
    
    // Retrieve containers
    // ... ditaus
    SG::ReadHandle<xAOD::DiTauJetContainer> importedDiTaus(m_ditauKey,ctx);
    if (!importedDiTaus.isValid()) {
        ATH_MSG_ERROR("No ditau collection with name " << m_ditauKey.key() << " found in StoreGate!");
        return StatusCode::FAILURE;
    }
    unsigned int nDiTaus(importedDiTaus->size());
    std::vector<const xAOD::DiTauJet*> ditauToCheck; ditauToCheck.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 (nDiTaus != nEntries ) {
          ATH_MSG_ERROR("Sizes incompatible! Are you sure your selection string used ditaus??");
            return StatusCode::FAILURE;
        } else {
          // identify which ditaus to keep for the thinning check
          for (unsigned int i=0; i<nDiTaus; ++i) if (entries[i]==1) ditauToCheck.push_back((*importedDiTaus)[i]);
        }
    }
    
    // Set elements in the mask to true if there is a corresponding ElementLink from a reconstructed object
    // ... ditaus
    if (m_selectionString=="") { // check all ditaus as user didn't provide a selection string
      for (const auto *ditauIt : *importedDiTaus) {
        for (unsigned int i=0; i<ditauIt->nTracks(); ++i) {
          int index = ditauIt->trackLinks().at(i).index();
          mask[index] = true;
        }
        for (unsigned int i=0; i<ditauIt->nIsoTracks(); ++i) {
          int index = ditauIt->isoTrackLinks().at(i).index();
          mask[index] = true;
        }
      }
    } else { // check only ditaus passing user selection string
        for (auto & ditauIt : ditauToCheck) {
          for (unsigned int i=0; i<ditauIt->nTracks(); ++i) {
              int index = ditauIt->trackLinks().at(i).index();
              mask[index] = true;
          }
          for (unsigned int i=0; i<ditauIt->nIsoTracks(); ++i) {
              int index = ditauIt->isoTrackLinks().at(i).index();
              mask[index] = true;
          }
        }
    }
 
    // Count up the mask contents
    unsigned int n_pass=0;
    for (unsigned int i=0; i<nTracks; ++i) {
        if (mask[i]) ++n_pass;
    }
    m_npass += n_pass;
 
    // Execute the thinning service based on the mask. Finish.
    importedTrackParticles.keep (mask);
 
    return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::DiTauTrackParticleThinning::finalize ( )

overridevirtual

Definition at line 45 of file DiTauTrackParticleThinning.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::DiTauTrackParticleThinning::initialize ( )

overridevirtual

Definition at line 29 of file DiTauTrackParticleThinning.cxx.

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

Member Data Documentation

m_ditauKey

SG::ReadHandleKey<xAOD::DiTauJetContainer> DerivationFramework::DiTauTrackParticleThinning::m_ditauKey { this, "DiTauKey","",""}

private

Definition at line 42 of file DiTauTrackParticleThinning.h.

{ this, "DiTauKey","",""};

m_inDetSGKey

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

private

Definition at line 44 of file DiTauTrackParticleThinning.h.

{ this, "InDetTrackParticlesKey", "InDetTrackParticles", "" };

m_npass

std::atomic<unsigned int> DerivationFramework::DiTauTrackParticleThinning::m_npass {}

mutableprivate

Definition at line 39 of file DiTauTrackParticleThinning.h.

{};

m_ntot

std::atomic<unsigned int> DerivationFramework::DiTauTrackParticleThinning::m_ntot {}

mutableprivate

Definition at line 38 of file DiTauTrackParticleThinning.h.

{};

m_selectionString

Gaudi::Property<std::string> DerivationFramework::DiTauTrackParticleThinning::m_selectionString { this, "SelectionString", "", ""}

private

Definition at line 46 of file DiTauTrackParticleThinning.h.

{ this, "SelectionString", "", ""};

m_streamName

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

private

Definition at line 40 of file DiTauTrackParticleThinning.h.

{ this, "StreamName", "", "Name of the stream being thinned" };

---

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

• DiTauTrackParticleThinning.h
• DiTauTrackParticleThinning.cxx