DerivationFramework::Thin_vtxTrk Class Reference

#include <Thin_vtxTrk.h>

Inheritance diagram for DerivationFramework::Thin_vtxTrk:

Collaboration diagram for DerivationFramework::Thin_vtxTrk:

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

Private Attributes
StringProperty	m_streamName { this, "StreamName", "", "Name of the stream being thinned" }
std::atomic< unsigned int >	m_ntot
std::atomic< unsigned int >	m_npass
double	m_acceptanceR
std::atomic< unsigned int >	m_nVtxTot
std::atomic< unsigned int >	m_nVtxPass
SG::ThinningHandleKey< xAOD::TrackParticleContainer >	m_trackParticleContainerName
SG::ThinningHandleKeyArray< xAOD::VertexContainer >	m_vertexContainerName
std::vector< std::string >	m_passFlags
SG::ReadDecorHandleKeyArray< xAOD::VertexContainer >	m_passArray {this, "INTERNALARRAY", {}}
bool	m_thinTracks
bool	m_noFlags

Detailed Description

Definition at line 29 of file Thin_vtxTrk.h.

Constructor & Destructor Documentation

Thin_vtxTrk()

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

Definition at line 19 of file Thin_vtxTrk.cxx.

                                                                                                        :
  base_class(t,n,p),
  m_ntot(0),
  m_npass(0),
  m_acceptanceR(-1.),  // Do not add tracks within a cone from the vertex by default
  m_nVtxTot(0),
  m_nVtxPass(0),
  m_noFlags(false)
{
  declareProperty("TrackParticleContainerName", m_trackParticleContainerName = "InDetTrackParticles");
  declareProperty("VertexContainerNames"      , m_vertexContainerName);
  declareProperty("PassFlags"                 , m_passFlags);
  declareProperty("AcceptanceRadius"          , m_acceptanceR);
  declareProperty("IgnoreFlags"                   , m_noFlags);
  declareProperty("ThinTracks"                , m_thinTracks = true);
}

~Thin_vtxTrk()

DerivationFramework::Thin_vtxTrk::~Thin_vtxTrk ( )

default

Member Function Documentation

doThinning()

StatusCode DerivationFramework::Thin_vtxTrk::doThinning ( ) const

virtual

Definition at line 88 of file Thin_vtxTrk.cxx.

{
  const EventContext& ctx = Gaudi::Hive::currentContext();
  // Retrieve main TrackParticle collection
  SG::ThinningHandle<xAOD::TrackParticleContainer> importedTrackParticles(m_trackParticleContainerName, ctx);
  
  // Check the event contains tracks
  unsigned int nTracks = importedTrackParticles->size();
  if (nTracks==0) return StatusCode::SUCCESS;
  
  // Set up a trackMask with the same entries as the full TrackParticle collection
  std::vector<bool> trackMask(nTracks,false); // default: don't keep any tracks
  m_ntot += nTracks;
  int nVtxTot =0;
  int nVtxPass=0;
 
  std::unordered_map<std::string, SG::ReadDecorHandle<xAOD::VertexContainer, Char_t>> handles;
  handles.reserve(m_passArray.size());
  for(const auto &key : m_passArray){
        auto it = handles.emplace(std::piecewise_construct, std::forward_as_tuple(key.key()), std::forward_as_tuple(key, ctx));
        if(!(*it.first).second.isPresent()) return StatusCode::FAILURE;
  }
 
  // retieve vertex 
  for(const auto& name : m_vertexContainerName){
    SG::ThinningHandle<xAOD::VertexContainer> vertexContainer(name, ctx);
    std::vector<bool> vtxMask(vertexContainer->size(), false); // default: don't keep any vertices
    
    // loop over vertices
    int k = 0;
    for(auto vtxItr = vertexContainer->begin(); vtxItr!=vertexContainer->end(); ++vtxItr, ++k) {
      const xAOD::Vertex* vtx = *vtxItr;
      nVtxTot++;
      
      // check if the vertex passed the required selections criteria
      bool passed = false;
      for(std::vector<std::string>::const_iterator flagItr = m_passFlags.begin(); flagItr!=m_passFlags.end(); ++flagItr) {
        std::string lookupstr = name.key() + '.' + (*flagItr);
        const auto& handle = handles.at(lookupstr);
        if(handle(*vtx) != 0) {
          passed = true;
          break;
        }
      } // end of loop over flags
      
      if(passed || m_noFlags) {
        // vertex passed the selection
        vtxMask[k] = true;
        nVtxPass++;
 
        // Add tracks according to DR selection
        if(m_acceptanceR > 0.){
 
          // determine the sum of the tracks at vertex as centre for the cone
          TLorentzVector centreCandidate;
          for(uint j=0; j<vtx->nTrackParticles(); ++j) {
        centreCandidate += vtx->trackParticle(j)->p4();
          }
 
          for(uint i=0; i<nTracks; ++i) {
        if(!trackMask[i]) {  // do this only for tracks that haven't been selected, yet
          const xAOD::TrackParticle* track = (*importedTrackParticles)[i];
          if(centreCandidate.DeltaR(track->p4()) < m_acceptanceR) trackMask[i]= true;
        }
          }
        }// end adding tracks according to DR selection
 
        if(m_thinTracks) {
          // loop over all tracks
          for(uint i=0; i<nTracks; ++i) {
            if(!trackMask[i]) {  // do this only for tracks that haven't been selected, yet
              const xAOD::TrackParticle* track = (*importedTrackParticles)[i];
              // loop over tracks at vertex
              for(uint j=0; j<vtx->nTrackParticles(); ++j) {
                if(vtx->trackParticle(j) == track) {
                  trackMask[i] = true;  // accept track
                }
              } // end of loop over tracks at vertex
            }
          } // end of loop over all tracks
        }
      }
    } // end of loop over vertices
    
    // Execute the thinning service based on the vtxMask.
    vertexContainer.keep(vtxMask);
  }
  
  // Count up the trackMask contents
  m_npass += std::accumulate(trackMask.begin(), trackMask.end(), 0);
  m_nVtxTot += nVtxTot;
  m_nVtxPass+= nVtxPass;
  if(m_thinTracks || m_acceptanceR > 0.) {
    // Execute the thinning service based on the trackMask. Finish.
    importedTrackParticles.keep(trackMask);
  }
 
  return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::Thin_vtxTrk::finalize

(

)

Definition at line 78 of file Thin_vtxTrk.cxx.

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

initialize()

StatusCode DerivationFramework::Thin_vtxTrk::initialize

(

)

Definition at line 40 of file Thin_vtxTrk.cxx.

{
  // Decide which collections need to be checked for ID TrackParticles
  ATH_MSG_VERBOSE("initialize() ...");
  ATH_CHECK(m_trackParticleContainerName.initialize(m_streamName));
 
 
  if( m_noFlags){
    ATH_MSG_INFO("IgnoreFlags is set, all vertices in the container will be kept");
  }
 
  if( ! m_noFlags){
    if (m_passFlags.empty()) {
      ATH_MSG_FATAL("No pass flags provided for thinning.");
      return StatusCode::FAILURE;
    } else {
      for(auto itr = m_passFlags.begin(); itr!=m_passFlags.end(); ++itr) {
        ATH_MSG_INFO("Vertices must pass the \"" << *itr << "\" selection");
      }
    }
  }
  
  if (m_acceptanceR > 0.) {
      ATH_MSG_INFO("Extra tracks must be within cone of "<<m_acceptanceR<<" from vertex candidate.");
  }
 
  for(auto &handle : m_vertexContainerName){
    ATH_CHECK(handle.initialize(m_streamName));
  }
  for(const auto &tracknames : m_vertexContainerName){
     for(const auto &str : m_passFlags){
        m_passArray.emplace_back(tracknames.key() + '.' + str);
     }
  }
  ATH_CHECK(m_passArray.initialize());
  return StatusCode::SUCCESS;
}

Member Data Documentation

m_acceptanceR

double DerivationFramework::Thin_vtxTrk::m_acceptanceR

private

Definition at line 40 of file Thin_vtxTrk.h.

m_noFlags

bool DerivationFramework::Thin_vtxTrk::m_noFlags

private

Definition at line 48 of file Thin_vtxTrk.h.

m_npass

std::atomic<unsigned int> DerivationFramework::Thin_vtxTrk::m_npass

private

Definition at line 39 of file Thin_vtxTrk.h.

m_ntot

std::atomic<unsigned int> DerivationFramework::Thin_vtxTrk::m_ntot

mutableprivate

Definition at line 39 of file Thin_vtxTrk.h.

m_nVtxPass

std::atomic<unsigned int> DerivationFramework::Thin_vtxTrk::m_nVtxPass

private

Definition at line 41 of file Thin_vtxTrk.h.

m_nVtxTot

std::atomic<unsigned int> DerivationFramework::Thin_vtxTrk::m_nVtxTot

mutableprivate

Definition at line 41 of file Thin_vtxTrk.h.

m_passArray

SG::ReadDecorHandleKeyArray<xAOD::VertexContainer> DerivationFramework::Thin_vtxTrk::m_passArray {this, "INTERNALARRAY", {}}

private

Definition at line 46 of file Thin_vtxTrk.h.

m_passFlags

std::vector<std::string> DerivationFramework::Thin_vtxTrk::m_passFlags

private

Definition at line 45 of file Thin_vtxTrk.h.

m_streamName

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

private

Definition at line 38 of file Thin_vtxTrk.h.

m_thinTracks

bool DerivationFramework::Thin_vtxTrk::m_thinTracks

private

Definition at line 47 of file Thin_vtxTrk.h.

m_trackParticleContainerName

SG::ThinningHandleKey<xAOD::TrackParticleContainer> DerivationFramework::Thin_vtxTrk::m_trackParticleContainerName

private

Definition at line 43 of file Thin_vtxTrk.h.

m_vertexContainerName

SG::ThinningHandleKeyArray<xAOD::VertexContainer> DerivationFramework::Thin_vtxTrk::m_vertexContainerName

private

Definition at line 44 of file Thin_vtxTrk.h.

---

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

• Thin_vtxTrk.h
• Thin_vtxTrk.cxx