#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 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 20 of file DiTauTrackParticleThinning.cxx.

                                                                                                   :
 base_class(t,n,p)
 {}

◆ ~DiTauTrackParticleThinning()

DerivationFramework::DiTauTrackParticleThinning::~DiTauTrackParticleThinning ( )

virtualdefault

Member Function Documentation

◆ doThinning()

StatusCode DerivationFramework::DiTauTrackParticleThinning::doThinning ( ) const

overridevirtual

Definition at line 55 of file DiTauTrackParticleThinning.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;
     
     // 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 46 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 30 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.

◆ m_inDetSGKey

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

private

Definition at line 44 of file DiTauTrackParticleThinning.h.