DerivationFramework::VSITrackParticleThinning Class Reference

#include <VSITrackParticleThinning.h>

Inheritance diagram for DerivationFramework::VSITrackParticleThinning:

Collaboration diagram for DerivationFramework::VSITrackParticleThinning:

Public Member Functions
	VSITrackParticleThinning (const std::string &t, const std::string &n, const IInterface *p)
virtual	~VSITrackParticleThinning ()
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", "" }
StringProperty	m_selectionString { this, "SelectionString", "", "" }
StringProperty	m_trackSelectionString { this, "TrackSelectionString", "", "" }
StringArrayProperty	m_augVerStrings { this, "AugVerStrings", {""}, "List of augVerStrings to consider" }

Detailed Description

Definition at line 29 of file VSITrackParticleThinning.h.

Constructor & Destructor Documentation

VSITrackParticleThinning()

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

Definition at line 22 of file VSITrackParticleThinning.cxx.

                                                                                              :
base_class(t,n,p)
{
}

~VSITrackParticleThinning()

DerivationFramework::VSITrackParticleThinning::~VSITrackParticleThinning ( )

virtual

Definition at line 30 of file VSITrackParticleThinning.cxx.

                                                                       {
}

Member Function Documentation

doThinning()

StatusCode DerivationFramework::VSITrackParticleThinning::doThinning ( ) const

overridevirtual

Definition at line 58 of file VSITrackParticleThinning.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;
    
    // Set elements in the mask to true if associated with a DV
    for ( const auto trkIt: *importedTrackParticles ) {
      for ( const auto &augVerString: m_augVerStrings) {
        SG::ConstAccessor<char> acc("is_svtrk_final" + augVerString);
        if (acc.withDefault (*trkIt, false)) {
          int index = trkIt->index();
          mask[index] = true;
        }
      }
    }
 
    // Apply a track selection string.
    if (!m_trackSelectionString.empty()) {
      std::vector<int> entries =  m_parser[kdvTrackThinning]->evaluateAsVector();
      unsigned int nEntries = entries.size();
      // check the sizes are compatible
      if (nTracks != nEntries ) {
        ATH_MSG_ERROR("Sizes incompatible! Are you sure your track selection string used tracks??");
        return StatusCode::FAILURE;
      } 
      else {
        // identify which dvs to keep for the thinning check
        for (unsigned int i=0; i<nEntries; ++i) {
          if (!mask[i]) continue;
          if (!entries[i]) mask[i] = false;
        }
      }
    }
    
 
    // 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::VSITrackParticleThinning::finalize ( )

overridevirtual

Definition at line 50 of file VSITrackParticleThinning.cxx.

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

initialize()

StatusCode DerivationFramework::VSITrackParticleThinning::initialize ( )

overridevirtual

Definition at line 34 of file VSITrackParticleThinning.cxx.

{
 
    // Decide which collections need to be checked for ID TrackParticles
    ATH_MSG_VERBOSE("initialize() ...");
    ATH_CHECK( m_inDetSGKey.initialize (m_streamName) );
 
    // Set up the text-parsing machinery for selectiong the dv directly according to user cuts
    if (!m_trackSelectionString.empty()) {
       // order must match enum order EJetTrPThinningParser
      ATH_CHECK( initializeParser( { m_selectionString, m_trackSelectionString } ));
    }
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_augVerStrings

StringArrayProperty DerivationFramework::VSITrackParticleThinning::m_augVerStrings { this, "AugVerStrings", {""}, "List of augVerStrings to consider" }

private

Definition at line 50 of file VSITrackParticleThinning.h.

m_inDetSGKey

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

private

Definition at line 43 of file VSITrackParticleThinning.h.

m_npass

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

mutableprivate

Definition at line 39 of file VSITrackParticleThinning.h.

m_ntot

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

mutableprivate

Definition at line 38 of file VSITrackParticleThinning.h.

m_selectionString

StringProperty DerivationFramework::VSITrackParticleThinning::m_selectionString { this, "SelectionString", "", "" }

private

Definition at line 45 of file VSITrackParticleThinning.h.

m_streamName

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

private

Definition at line 41 of file VSITrackParticleThinning.h.

m_trackSelectionString

StringProperty DerivationFramework::VSITrackParticleThinning::m_trackSelectionString { this, "TrackSelectionString", "", "" }

private

Definition at line 47 of file VSITrackParticleThinning.h.

---

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

• VSITrackParticleThinning.h
• VSITrackParticleThinning.cxx