DerivationFramework::TauThinningTool Class Reference

#include <TauThinningTool.h>

Inheritance diagram for DerivationFramework::TauThinningTool:

Collaboration diagram for DerivationFramework::TauThinningTool:

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

Private Attributes
std::atomic< unsigned int >	m_ntot = 0
std::atomic< unsigned int >	m_npass = 0
StringProperty	m_streamName { this, "StreamName", "", "Name of the stream being thinned" }
Gaudi::Property< std::string >	m_selectionString { this, "SelectionString", "", "" }
SG::ThinningHandleKey< xAOD::TauJetContainer >	m_taus { this, "Taus", "TauJets", "" }
SG::ThinningHandleKey< xAOD::TauTrackContainer >	m_tauTracks { this, "TauTracks", "TauTracks", "" }
SG::ThinningHandleKey< xAOD::TrackParticleContainer >	m_trackParticles { this, "TrackParticles", "InDetTrackParticles", "" }
SG::ThinningHandleKey< xAOD::PFOContainer >	m_neutralPFOs { this, "TauNeutralPFOs", "TauNeutralParticleFlowObjects", "" }
SG::ThinningHandleKey< xAOD::VertexContainer >	m_secondaryVertices { this, "TauSecondaryVertices", "TauSecondaryVertices", "" }

Detailed Description

Definition at line 24 of file TauThinningTool.h.

Constructor & Destructor Documentation

TauThinningTool()

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

Definition at line 12 of file TauThinningTool.cxx.

                                                   :
  base_class(t,n,p)
{
}

~TauThinningTool()

virtual DerivationFramework::TauThinningTool::~TauThinningTool ( )

virtualdefault

Member Function Documentation

doThinning()

StatusCode DerivationFramework::TauThinningTool::doThinning ( ) const

overridevirtual

Definition at line 45 of file TauThinningTool.cxx.

{
  const EventContext& ctx = Gaudi::Hive::currentContext();
 
  // retrieve containers and thin them
  SG::ThinningHandle<xAOD::TauJetContainer> taus(m_taus, ctx);
  taus.thinAll();
  size_t nTaus = taus->size();
 
  SG::ThinningHandle<xAOD::TauTrackContainer> tauTracks(m_tauTracks, ctx);
  tauTracks.thinAll();
 
  SG::ThinningHandle<xAOD::TrackParticleContainer> trackParticles(m_trackParticles, ctx);
  trackParticles.thinAll();
 
  SG::ThinningHandle<xAOD::PFOContainer> neutralPFOs(m_neutralPFOs, ctx);
  neutralPFOs.thinAll();
 
  SG::ThinningHandle<xAOD::VertexContainer> secondaryVertices(m_secondaryVertices, ctx);
  secondaryVertices.thinAll();
 
 
  std::vector<const xAOD::TauJet*> tausToKeep;
 
  // execute the text parser if requested
  if (!m_selectionString.empty()) {
    std::vector<int> entries =  m_parser->evaluateAsVector();
    size_t nEntries = entries.size();
    if (nTaus != nEntries) {
      ATH_MSG_ERROR("Incompatible sizes: " << nTaus << " vs " << nEntries << "! Please check your selection string uses the appropriate tau container.");
      return StatusCode::FAILURE;
    }
    // identify which taus to keep
    for (size_t i=0; i<nTaus; ++i) if (entries[i]==1) tausToKeep.push_back(taus->at(i));    
  }
  // use all taus if no selection string is passed
  else {
    for (size_t i=0; i<nTaus; ++i) tausToKeep.push_back(taus->at(i));
  }
 
  // protection against duplicate taus -- built from different seed jets, but end up having same (eta,phi)
  if( tausToKeep.size() > 0){
    for(size_t i=0; i < tausToKeep.size()-1; i++){
      const auto* aTau=tausToKeep[i];
      auto it = std::remove_if(tausToKeep.begin()+i+1,tausToKeep.end(),[aTau](const xAOD::TauJet* bTau) {return aTau->p4().DeltaR(bTau->p4()) < 0.01;});
      tausToKeep.erase (it, tausToKeep.end());
    }
  }   
 
  // keep the various tau-related objects for taus passing the selection
  for (const auto* tau : tausToKeep) {
    // tau
    taus.keep(tau->index());
 
    // classifiedCharged tau tracks
    for (const xAOD::TauTrack* track : tau->tracks()) {
      tauTracks.keep(track->index());
      
      // associated ID track
      trackParticles.keep(track->track()->index());
    }
 
    // neutral PFOs
    for (size_t i=0; i<tau->nNeutralPFOs(); i++) {
      neutralPFOs.keep(tau->neutralPFO(i)->index());
    }  
 
    // secondary vertex
    if (tau->secondaryVertex() != nullptr) {
      secondaryVertices.keep(tau->secondaryVertex()->index());
    }
  }
 
  // increment counters
  m_npass += tausToKeep.size();
  m_ntot  += nTaus;
 
  return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::TauThinningTool::finalize ( )

overridevirtual

Definition at line 37 of file TauThinningTool.cxx.

{
  ATH_MSG_INFO("Processed " << m_ntot << " taus, " << m_npass << " were kept");
  ATH_CHECK( finalizeParser() );
  return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::TauThinningTool::initialize ( )

overridevirtual

Definition at line 20 of file TauThinningTool.cxx.

{
  ATH_CHECK( m_taus.initialize(m_streamName) );
  ATH_CHECK( m_tauTracks.initialize(m_streamName) );
  ATH_CHECK( m_trackParticles.initialize(m_streamName) );
  ATH_CHECK( m_neutralPFOs.initialize(m_streamName) );
  ATH_CHECK( m_secondaryVertices.initialize(m_streamName) );
 
  // set up the text-parsing machinery for selecting taus according to user cuts
  if (!m_selectionString.empty()) {
    ATH_MSG_INFO("Selection string for " << m_taus.key() << ": " << m_selectionString);
    ATH_CHECK( initializeParser(m_selectionString) );
  }
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_neutralPFOs

SG::ThinningHandleKey<xAOD::PFOContainer> DerivationFramework::TauThinningTool::m_neutralPFOs { this, "TauNeutralPFOs", "TauNeutralParticleFlowObjects", "" }

private

Definition at line 40 of file TauThinningTool.h.

m_npass

std::atomic<unsigned int> DerivationFramework::TauThinningTool::m_npass = 0

mutableprivate

Definition at line 34 of file TauThinningTool.h.

m_ntot

std::atomic<unsigned int> DerivationFramework::TauThinningTool::m_ntot = 0

mutableprivate

Definition at line 33 of file TauThinningTool.h.

m_secondaryVertices

SG::ThinningHandleKey<xAOD::VertexContainer> DerivationFramework::TauThinningTool::m_secondaryVertices { this, "TauSecondaryVertices", "TauSecondaryVertices", "" }

private

Definition at line 41 of file TauThinningTool.h.

m_selectionString

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

private

Definition at line 36 of file TauThinningTool.h.

m_streamName

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

private

Definition at line 35 of file TauThinningTool.h.

m_taus

SG::ThinningHandleKey<xAOD::TauJetContainer> DerivationFramework::TauThinningTool::m_taus { this, "Taus", "TauJets", "" }

private

Definition at line 37 of file TauThinningTool.h.

m_tauTracks

SG::ThinningHandleKey<xAOD::TauTrackContainer> DerivationFramework::TauThinningTool::m_tauTracks { this, "TauTracks", "TauTracks", "" }

private

Definition at line 38 of file TauThinningTool.h.

m_trackParticles

SG::ThinningHandleKey<xAOD::TrackParticleContainer> DerivationFramework::TauThinningTool::m_trackParticles { this, "TrackParticles", "InDetTrackParticles", "" }

private

Definition at line 39 of file TauThinningTool.h.

---

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

• TauThinningTool.h
• TauThinningTool.cxx