DerivationFramework::HITrackParticleThinningTool Class Reference

#include <HITrackParticleThinningTool.h>

Inheritance diagram for DerivationFramework::HITrackParticleThinningTool:

Collaboration diagram for DerivationFramework::HITrackParticleThinningTool:

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

Private Attributes
StringProperty	m_streamName { this, "StreamName", "", "Name of the stream being thinned" }
SG::ThinningHandleKey< xAOD::TrackParticleContainer >	m_inDetSGKey { this, "InDetTrackParticlesKey", "InDetTrackParticles", "" }
std::string	m_vertex_key
std::string	m_vertex_scheme
ToolHandle< InDet::IInDetTrackSelectionTool >	m_trkSelTool
	track selection tool which can be optionally used for N_trk and sum pt cuts
std::atomic< unsigned int >	m_ntot
std::atomic< unsigned int >	m_npass

Detailed Description

Definition at line 29 of file HITrackParticleThinningTool.h.

Constructor & Destructor Documentation

HITrackParticleThinningTool()

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

Definition at line 23 of file HITrackParticleThinningTool.cxx.

                                                                              : 
    
    base_class(t,n,p),
    m_ntot(0),
    m_npass(0)
    {
      //declareProperty("InDetTrackParticlesKey", m_TP_key="InDetTrackParticles");
      declareProperty("PrimaryVertexKey", m_vertex_key="PrimaryVertices");
      declareProperty("PrimaryVertexSelection", m_vertex_scheme="sumPt2");
      declareProperty("TrackSelectionTool", m_trkSelTool, "Track selection tool" );
    }

~HITrackParticleThinningTool()

DerivationFramework::HITrackParticleThinningTool::~HITrackParticleThinningTool

(

)

Definition at line 38 of file HITrackParticleThinningTool.cxx.

                                                                               {
  }  

Member Function Documentation

doThinning()

StatusCode DerivationFramework::HITrackParticleThinningTool::doThinning ( ) const

overridevirtual

Definition at line 67 of file HITrackParticleThinningTool.cxx.

  {
    // Get the current event context
    const EventContext& ctx = Gaudi::Hive::currentContext();
 
    // Get the track container
    SG::ThinningHandle<xAOD::TrackParticleContainer> tracks (m_inDetSGKey, ctx);
 
    //Define a primary vertex 
    const xAOD::Vertex *primary_vertex(nullptr);
 
    // Retrieve vertex container
    const xAOD::VertexContainer* vtxC(nullptr);
    if (evtStore()->retrieve(vtxC, m_vertex_key).isFailure()) {
        ATH_MSG_ERROR("Failed to retrieve VertexContainer with key: " << m_vertex_key);
        return StatusCode::FAILURE;
    }
 
    // Check event contains tracks, if none then thinning complete!
    unsigned int nTracks = tracks->size();
    if (nTracks==0) return StatusCode::SUCCESS;
 
    // Variables to track the best vertex based on the scheme
    float ptmax = 0.;
    size_t ntrkmax = 0;
 
    // Iterate through the vertices to find the primary vertex
    for (auto vertex : *vtxC) {
        if (!vertex || vertex->vertexType() != xAOD::VxType::PriVtx) {
            continue;
        }
 
        if (m_vertex_scheme.compare("sumPt2") == 0) {
            // Define the accessor for the sumPt2 variable
            SG::ConstAccessor<float> sumPt2Acc("sumPt2");
 
            // Check if the sumPt2 variable is available for the vertex
            if (sumPt2Acc.isAvailable(*vertex)) {
                float sumPT = sumPt2Acc(*vertex);  // Access the sumPt2 variable
                if (sumPT > ptmax) {
                    ptmax = sumPT;
                    primary_vertex = vertex;
                }
            }
        } else {
            size_t ntp = vertex->nTrackParticles();
            if (ntp > ntrkmax) {
                ntrkmax = ntp;
                primary_vertex = vertex;
            }
        }
    }
 
    // Warn if not primary vertex is found
    if (!primary_vertex) {
        ATH_MSG_DEBUG("No primary vertex found.");
    }
 
    // Loop over tracks, see if they pass, set mask
    std::vector<bool> mask(nTracks,false); // reject all tracks by default
    m_ntot += nTracks;
 
    for (auto tp : *tracks) {
        if (tp) {
            const xAOD::Vertex* vert_trk = primary_vertex;
 
            asg::AcceptData acceptData = m_trkSelTool->accept(*tp, vert_trk);
            int index = tp->index();
            mask[index] = static_cast<bool>(acceptData);
        }
    }
 
    // Count up mask contents
    unsigned int n_pass=0;
    for (unsigned int i=0; i<nTracks; ++i) {
        if (mask[i]) ++n_pass;
    }
    m_npass += n_pass;
    
    tracks.keep (mask);
 
    return StatusCode::SUCCESS;
    }  

finalize()

StatusCode DerivationFramework::HITrackParticleThinningTool::finalize ( )

override

Definition at line 59 of file HITrackParticleThinningTool.cxx.

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

initialize()

StatusCode DerivationFramework::HITrackParticleThinningTool::initialize ( )

override

Definition at line 42 of file HITrackParticleThinningTool.cxx.

                                                                         {
        if(!m_trkSelTool)
        {
            ATH_MSG_ERROR("InDetTrackSelectionTool not set. Cannot initialize.");
            return StatusCode::FAILURE;
        }
 
        // Validate m_vertex_scheme
        if (m_vertex_scheme != "sumPt2" && m_vertex_scheme != "nTracks") {
            ATH_MSG_ERROR("Invalid PrimaryVertexSelection: " << m_vertex_scheme);
            return StatusCode::FAILURE;
        }
 
        ATH_CHECK( m_inDetSGKey.initialize(m_streamName) );
        return StatusCode::SUCCESS;
    }

Member Data Documentation

m_inDetSGKey

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

private

Definition at line 48 of file HITrackParticleThinningTool.h.

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

m_npass

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

mutableprivate

Definition at line 56 of file HITrackParticleThinningTool.h.

m_ntot

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

mutableprivate

Definition at line 55 of file HITrackParticleThinningTool.h.

m_streamName

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

private

Definition at line 46 of file HITrackParticleThinningTool.h.

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

m_trkSelTool

ToolHandle< InDet::IInDetTrackSelectionTool > DerivationFramework::HITrackParticleThinningTool::m_trkSelTool

private

track selection tool which can be optionally used for N_trk and sum pt cuts

Definition at line 53 of file HITrackParticleThinningTool.h.

m_vertex_key

std::string DerivationFramework::HITrackParticleThinningTool::m_vertex_key

private

Definition at line 51 of file HITrackParticleThinningTool.h.

m_vertex_scheme

std::string DerivationFramework::HITrackParticleThinningTool::m_vertex_scheme

private

Definition at line 52 of file HITrackParticleThinningTool.h.

---

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

• HITrackParticleThinningTool.h
• HITrackParticleThinningTool.cxx