#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 More...

std::atomic< unsigned int >	m_ntot

std::atomic< unsigned int >	m_npass

Detailed Description

Definition at line 32 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.

38 {

39 }

Member Function Documentation

◆ doThinning()

StatusCode DerivationFramework::HITrackParticleThinningTool::doThinning ( ) const

overridevirtual

Definition at line 66 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);
    
     m_ntot+=tracks->size(); 
  
     //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;
     }
  
     // 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;
     mask.reserve(tracks->size());
  
     for (auto tp : *tracks) {
         if (tp) {
             const xAOD::Vertex* vert_trk = primary_vertex;
  
             asg::AcceptData acceptData = m_trkSelTool->accept(*tp, vert_trk);
             mask.push_back(static_cast<bool>(acceptData));
         } else { 
             mask.push_back(false);
         }
     }
     
     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());
         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 51 of file HITrackParticleThinningTool.h.