DerivationFramework::PixeldEdxTrackParticleThinning Class Reference

#include <PixeldEdxTrackParticleThinning.h>

Inheritance diagram for DerivationFramework::PixeldEdxTrackParticleThinning:

Collaboration diagram for DerivationFramework::PixeldEdxTrackParticleThinning:

Public Member Functions
	PixeldEdxTrackParticleThinning (const std::string &t, const std::string &n, const IInterface *p)
virtual	~PixeldEdxTrackParticleThinning ()=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
DoubleProperty	m_unprescalePtCut
DoubleProperty	m_globalScale
DoubleProperty	m_d0SignifCut
DoubleProperty	m_z0Cut
DoubleProperty	m_etaCut
std::vector< double >	m_pTbins
std::vector< unsigned long long > m_counter	ATLAS_THREAD_SAFE
std::vector< unsigned long long > m_counter_picked	ATLAS_THREAD_SAFE
std::mutex	m_mutex

Static Private Attributes
static const std::vector< double >	m_preScales

Detailed Description

Definition at line 36 of file PixeldEdxTrackParticleThinning.h.

Constructor & Destructor Documentation

PixeldEdxTrackParticleThinning()

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

Definition at line 26 of file PixeldEdxTrackParticleThinning.cxx.

                                                                                                          :
  base_class(t, n, p)
{}

~PixeldEdxTrackParticleThinning()

virtual DerivationFramework::PixeldEdxTrackParticleThinning::~PixeldEdxTrackParticleThinning ( )

virtualdefault

Member Function Documentation

doThinning()

StatusCode DerivationFramework::PixeldEdxTrackParticleThinning::doThinning ( ) const

overridevirtual

Definition at line 71 of file PixeldEdxTrackParticleThinning.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
  if (m_parser) {
    std::vector<int> 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;
 
  auto getBin = [&]( const double pT ) -> size_t {
    for( size_t ip = 0; ip < m_pTbins.size()-1; ip++ ) {
      if( pT > m_pTbins.at(ip) && pT < m_pTbins.at(ip+1) ) {
        return ip;
      }
    }
    return 0;
  };
    
  // Count the mask
  for (size_t i=0; i<nTracks; ++i) {
    
    const auto* trk = importedTrackParticles->at(i);
    
    // Keep all tracks if IBL is overflow
    if( trk->numberOfIBLOverflowsdEdx() > 0 ) {
      
      mask.at(i) = true; ++m_npass; continue;
      
    }
    
    
    // Keep all track for pT above 10 GeV
    if( std::abs( trk->pt() ) > m_unprescalePtCut ) {
      
      mask.at(i) = true; ++m_npass; continue;
      
    }      
    
    // Hereafter only low-pT and no-overflow tracks
    
    // 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;
    }
    
    // Reject forward low-pT tracks
    if( std::abs( trk->eta() ) > m_etaCut ) {
      continue;
    }
    
    // Prescaled track keeping
    auto bin = getBin( std::abs( trk->pt() ) );
    
    const auto preScale   = static_cast<unsigned long long>( std::floor( m_preScales.at( bin ) * m_globalScale ) );
    const auto preScale10 = std::max( static_cast<unsigned long long>( std::floor( m_preScales.at( bin ) * m_globalScale / 10. ) ), 1ull );
        
    float dEdx { 0 };
    trk->summaryValue( dEdx, xAOD::pixeldEdx );
 
    // Increment the m_counter
    m_counter.at(bin)++;
 
    // Relatively higher dE/dx tracks
    static const float dEdxThr { static_cast<float>(pow( 10, 0.1 )) };
    if( dEdx > dEdxThr ) {
      
      mask.at(i) = ( (m_counter.at(bin) % preScale10) == 0 );
      
      // There are also tracks with dE/dx == -1.
    } else if( dEdx > 0. ) {
      
      mask.at(i) = ( (m_counter.at(bin) % preScale) == 0 );
      
    } else {
      
      mask.at(i) = false;
      
    }
    
    if( mask.back() ) { ++m_npass; m_counter_picked.at(bin)++; }
    
  }
 
  assert( importedTrackParticles->size() == mask.size() );
 
  // Execute the thinning service based on the mask.
  importedTrackParticles.keep (mask);
  
  return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::PixeldEdxTrackParticleThinning::finalize ( )

overridevirtual

Definition at line 62 of file PixeldEdxTrackParticleThinning.cxx.

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

initialize()

StatusCode DerivationFramework::PixeldEdxTrackParticleThinning::initialize ( )

overridevirtual

Definition at line 33 of file PixeldEdxTrackParticleThinning.cxx.

{
  ATH_MSG_DEBUG("initialize() ...");
    
  //check xAOD::InDetTrackParticle collection
  ATH_CHECK( m_inDetParticlesKey.initialize (m_streamName) );
  ATH_MSG_INFO("Using " << m_inDetParticlesKey << "as the source collection for inner detector track particles");
  ATH_CHECK( m_vertexContainerKey.initialize() );
  //check availability of xAOD::TrackStateValidation and xAOD::TrackMeasurementValidation containers
 
  if (!m_selectionString.empty()) {
    ATH_CHECK(initializeParser(m_selectionString));
  }
 
  m_counter.resize(50);
  m_counter_picked.resize(50);
  
  if( 0 == m_pTbins.size() ) {
    for( size_t ip = 0; ip <= 50; ++ip ) {
      m_pTbins.emplace_back( pow( 10, 2.0 + 0.04 * ip ) );
    }
  }
 
 
  return StatusCode::SUCCESS;
}

Member Data Documentation

ATLAS_THREAD_SAFE [1/2]

std::vector<unsigned long long> m_counter DerivationFramework::PixeldEdxTrackParticleThinning::ATLAS_THREAD_SAFE

mutableprivate

Definition at line 76 of file PixeldEdxTrackParticleThinning.h.

ATLAS_THREAD_SAFE [2/2]

std::vector<unsigned long long> m_counter_picked DerivationFramework::PixeldEdxTrackParticleThinning::ATLAS_THREAD_SAFE

mutableprivate

Definition at line 77 of file PixeldEdxTrackParticleThinning.h.

m_d0SignifCut

DoubleProperty DerivationFramework::PixeldEdxTrackParticleThinning::m_d0SignifCut

private

Initial value:

{
      this, "d0SignifCut", 5.}

Definition at line 65 of file PixeldEdxTrackParticleThinning.h.

m_etaCut

DoubleProperty DerivationFramework::PixeldEdxTrackParticleThinning::m_etaCut

private

Initial value:

{
      this, "EtaCut", 2.}

Definition at line 69 of file PixeldEdxTrackParticleThinning.h.

m_globalScale

DoubleProperty DerivationFramework::PixeldEdxTrackParticleThinning::m_globalScale

private

Initial value:

{
      this, "GlobalScale", 1.}

Definition at line 63 of file PixeldEdxTrackParticleThinning.h.

m_inDetParticlesKey

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

private

Initial value:

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

Definition at line 55 of file PixeldEdxTrackParticleThinning.h.

m_mutex

std::mutex DerivationFramework::PixeldEdxTrackParticleThinning::m_mutex

mutableprivate

Definition at line 78 of file PixeldEdxTrackParticleThinning.h.

m_npass

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

private

Definition at line 51 of file PixeldEdxTrackParticleThinning.h.

m_ntot

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

mutableprivate

Definition at line 51 of file PixeldEdxTrackParticleThinning.h.

m_preScales

const std::vector< double > DerivationFramework::PixeldEdxTrackParticleThinning::m_preScales

staticprivate

Initial value:

= {
  150, 150, 150, 150, 150, 150, 150, 150, 150, 150,
  150, 150, 150, 150, 150, 150, 150, 150, 150, 150,
  150, 150, 150, 150, 150, 124.261, 111.448, 97.7357, 85.5616, 72.3294,
  61.0017, 50.8592, 41.6827, 33.7413, 26.9884, 21.3822, 16.8352, 13.1081, 10.2368, 7.96723,
  6.23636, 4.91757, 3.91519, 3.15519, 2.57768, 2.13622, 1.80524, 1.53102, 1.30677, 1.13582
}

Definition at line 73 of file PixeldEdxTrackParticleThinning.h.

m_pTbins

std::vector<double> DerivationFramework::PixeldEdxTrackParticleThinning::m_pTbins

private

Definition at line 72 of file PixeldEdxTrackParticleThinning.h.

m_selectionString

StringProperty DerivationFramework::PixeldEdxTrackParticleThinning::m_selectionString

private

Initial value:

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

Definition at line 46 of file PixeldEdxTrackParticleThinning.h.

m_streamName

StringProperty DerivationFramework::PixeldEdxTrackParticleThinning::m_streamName

private

Initial value:

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

Definition at line 53 of file PixeldEdxTrackParticleThinning.h.

m_unprescalePtCut

DoubleProperty DerivationFramework::PixeldEdxTrackParticleThinning::m_unprescalePtCut

private

Initial value:

{
      this, "UnprescalePtCut", 10.e3}

Definition at line 61 of file PixeldEdxTrackParticleThinning.h.

m_vertexContainerKey

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

private

Initial value:

{
      this, "VertexContainerKey", "PrimaryVertices"}

Definition at line 57 of file PixeldEdxTrackParticleThinning.h.

m_z0Cut

DoubleProperty DerivationFramework::PixeldEdxTrackParticleThinning::m_z0Cut

private

Initial value:

{
      this, "z0Cut", 3.}

Definition at line 67 of file PixeldEdxTrackParticleThinning.h.

---

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

• PixeldEdxTrackParticleThinning.h
• PixeldEdxTrackParticleThinning.cxx