DerivationFramework::TagAndProbeTrackParticleThinning Class Reference

#include <TagAndProbeTrackParticleThinning.h>

Inheritance diagram for DerivationFramework::TagAndProbeTrackParticleThinning:

Collaboration diagram for DerivationFramework::TagAndProbeTrackParticleThinning:

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

Private Attributes
StringProperty	m_selectionString
std::atomic< unsigned int >	m_ntot {0}
std::atomic< unsigned int >	m_npass {0}
StringProperty	m_streamName
SG::ThinningHandleKey< xAOD::TrackParticleContainer >	m_inDetParticlesKey
SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexContainerKey
SG::ReadHandleKey< xAOD::MuonContainer >	m_muonContainerKey
DoubleProperty	m_d0SignifCut
DoubleProperty	m_z0Cut
DoubleProperty	m_massCut
std::mutex	m_mutex

Detailed Description

Definition at line 32 of file TagAndProbeTrackParticleThinning.h.

Constructor & Destructor Documentation

TagAndProbeTrackParticleThinning()

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

Definition at line 14 of file TagAndProbeTrackParticleThinning.cxx.

                                                                  :
  base_class(t, n, p)
{}

~TagAndProbeTrackParticleThinning()

virtual DerivationFramework::TagAndProbeTrackParticleThinning::~TagAndProbeTrackParticleThinning ( )

virtualdefault

Member Function Documentation

doThinning()

StatusCode DerivationFramework::TagAndProbeTrackParticleThinning::doThinning ( ) const

overridevirtual

Definition at line 44 of file TagAndProbeTrackParticleThinning.cxx.

{
 
  const EventContext& ctx = Gaudi::Hive::currentContext();
 
  // Retrieve main TrackParticle collection
  SG::ThinningHandle<xAOD::TrackParticleContainer> importedTrackParticles
    (m_inDetParticlesKey, ctx);
 
  // Check the event contains tracks
  size_t 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
 
  std::lock_guard<std::mutex> lock(m_mutex);    
 
  m_ntot += nTracks;
 
  // Execute the text parser and update the mask
  std::vector<int> entries;
  if (m_parser) {
    entries =  m_parser->evaluateAsVector();
    unsigned int nEntries = entries.size();
    // check the sizes are compatible
    if (nTracks != nEntries ) {
      ATH_MSG_ERROR("Sizes incompatible! Are you sure your selection string used ID TrackParticles?");
      return StatusCode::FAILURE;
    } else {
      // set mask
      for (unsigned int i=0; i<nTracks; ++i) if (entries[i]==1) mask[i]=true;
    }
  }
 
  // Retrieve vertex collection
  SG::ReadHandle<xAOD::VertexContainer> primaryVertices(m_vertexContainerKey,ctx);
  ATH_CHECK( primaryVertices.isValid() );
  
  const xAOD::Vertex* priVtx = nullptr;
  
  for( const auto* vtx : *primaryVertices ) {
    if( vtx->vertexType() == xAOD::VxType::PriVtx ) {
      priVtx = vtx;
      break;
    }
  }
 
  if( !priVtx ) return StatusCode::SUCCESS;
  
  // Retrieve muon collection
  SG::ReadHandle<xAOD::MuonContainer> muons(m_muonContainerKey,ctx);
  ATH_CHECK( muons.isValid() );
 
  // Count the mask
  for (size_t i=0; i<nTracks; ++i) {
    
    const auto* trk = importedTrackParticles->at(i);
 
    // Keep only d0 significancd is small enough
    if( std::abs( xAOD::TrackingHelpers::d0significance( trk ) ) > m_d0SignifCut ) {
      continue;
    }
    
    // Keep only delta(z0) is less than 10 mm
    if( std::abs( trk->z0() - priVtx->z() + trk->vz() ) > m_z0Cut ) {
      continue;
    }
 
    // probe track keeping
    for (const auto muon : *muons) {
      const auto* muon_trk = muon->trackParticle(xAOD::Muon::InnerDetectorTrackParticle);
 
      if( muon_trk ) {
    if( muon_trk == trk) continue;
      }
 
      if( (muon->p4() + trk->p4()).M() > m_massCut ) {
    mask.at(i) = (mask.at(i) && true);
    ++m_npass;
      }
    }
        
  }
 
  assert( importedTrackParticles->size() == mask.size() );
 
  // Execute the thinning service based on the mask.
  importedTrackParticles.keep (mask);
  
  return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::TagAndProbeTrackParticleThinning::finalize ( )

overridevirtual

Definition at line 36 of file TagAndProbeTrackParticleThinning.cxx.

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

initialize()

StatusCode DerivationFramework::TagAndProbeTrackParticleThinning::initialize ( )

overridevirtual

Definition at line 21 of file TagAndProbeTrackParticleThinning.cxx.

{
  //check xAOD::InDetTrackParticle collection
  ATH_CHECK( m_inDetParticlesKey.initialize (m_streamName) );
  ATH_MSG_DEBUG("Using " << m_inDetParticlesKey << "as the source collection for inner detector track particles");
  ATH_CHECK( m_vertexContainerKey.initialize() );
  ATH_CHECK( m_muonContainerKey.initialize() );
 
  if (!m_selectionString.empty()) {
    ATH_CHECK(initializeParser(m_selectionString));
  }
 
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_d0SignifCut

DoubleProperty DerivationFramework::TagAndProbeTrackParticleThinning::m_d0SignifCut

private

Initial value:

{
      this, "d0SignifCut", 5.}

Definition at line 59 of file TagAndProbeTrackParticleThinning.h.

m_inDetParticlesKey

SG::ThinningHandleKey<xAOD::TrackParticleContainer> DerivationFramework::TagAndProbeTrackParticleThinning::m_inDetParticlesKey

private

Initial value:

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

Definition at line 51 of file TagAndProbeTrackParticleThinning.h.

m_massCut

DoubleProperty DerivationFramework::TagAndProbeTrackParticleThinning::m_massCut

private

Initial value:

{
      this, "MassCut", 30.e3}

Definition at line 63 of file TagAndProbeTrackParticleThinning.h.

m_muonContainerKey

SG::ReadHandleKey< xAOD::MuonContainer > DerivationFramework::TagAndProbeTrackParticleThinning::m_muonContainerKey

private

Initial value:

{
      this, "MuonContainerKey", "Muons"}

Definition at line 55 of file TagAndProbeTrackParticleThinning.h.

m_mutex

std::mutex DerivationFramework::TagAndProbeTrackParticleThinning::m_mutex

mutableprivate

Definition at line 67 of file TagAndProbeTrackParticleThinning.h.

m_npass

std::atomic<unsigned int> DerivationFramework::TagAndProbeTrackParticleThinning::m_npass {0}

private

Definition at line 47 of file TagAndProbeTrackParticleThinning.h.

m_ntot

std::atomic<unsigned int> DerivationFramework::TagAndProbeTrackParticleThinning::m_ntot {0}

mutableprivate

Definition at line 47 of file TagAndProbeTrackParticleThinning.h.

m_selectionString

StringProperty DerivationFramework::TagAndProbeTrackParticleThinning::m_selectionString

private

Initial value:

{
      this, "SelectionString", "", "track selections"}

Definition at line 42 of file TagAndProbeTrackParticleThinning.h.

m_streamName

StringProperty DerivationFramework::TagAndProbeTrackParticleThinning::m_streamName

private

Initial value:

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

Definition at line 49 of file TagAndProbeTrackParticleThinning.h.

m_vertexContainerKey

SG::ReadHandleKey< xAOD::VertexContainer > DerivationFramework::TagAndProbeTrackParticleThinning::m_vertexContainerKey

private

Initial value:

{
      this, "VertexContainerKey", "PrimaryVertices"}

Definition at line 53 of file TagAndProbeTrackParticleThinning.h.

m_z0Cut

DoubleProperty DerivationFramework::TagAndProbeTrackParticleThinning::m_z0Cut

private

Initial value:

{
      this, "z0Cut", 10.}

Definition at line 61 of file TagAndProbeTrackParticleThinning.h.

---

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

• TagAndProbeTrackParticleThinning.h
• TagAndProbeTrackParticleThinning.cxx