DerivationFramework::HIJetTrackParticleThinningTool Class Reference

#include <HIJetTrackParticleThinningTool.h>

Inheritance diagram for DerivationFramework::HIJetTrackParticleThinningTool:

Collaboration diagram for DerivationFramework::HIJetTrackParticleThinningTool:

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

Private Attributes
StringProperty	m_streamName { this, "StreamName", "", "Name of the stream being thinned" }
SG::ReadHandleKey< xAOD::VertexContainer >	m_vertexKey { this, "PrimaryVertexKey", "PrimaryVertices", "Primary vertex container"}
SG::ReadHandleKey< xAOD::JetContainer >	m_jetKey { this, "JetKey", "", "Jet collection container"}
SG::ReadDecorHandleKey< xAOD::VertexContainer >	m_sumPt2Key { this, "SumPt2Key", "PrimaryVertices.sumPt2", "SumPt2 decoration"}
SG::ThinningHandleKey< xAOD::TrackParticleContainer >	m_inDetSGKey { this, "InDetTrackParticlesKey", "InDetTrackParticles", "" }
ToolHandle< InDet::IInDetTrackSelectionTool >	m_trkSelTool { this, "TrackSelectionTool", "", "Track selection tool"}
Gaudi::Property< std::string >	m_vertexScheme { this, "PrimaryVertexSelection", "sumPt2", "Vertex selection scheme"}
std::atomic< unsigned int >	m_ntot = 0
std::atomic< unsigned int >	m_npass = 0

Detailed Description

Definition at line 30 of file HIJetTrackParticleThinningTool.h.

Constructor & Destructor Documentation

HIJetTrackParticleThinningTool()

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

Definition at line 24 of file HIJetTrackParticleThinningTool.cxx.

                             :
 
        base_class(t,n,p) {}

~HIJetTrackParticleThinningTool()

DerivationFramework::HIJetTrackParticleThinningTool::~HIJetTrackParticleThinningTool ( )

virtual

Definition at line 32 of file HIJetTrackParticleThinningTool.cxx.

                                                                                       {
    }

Member Function Documentation

doThinning()

StatusCode DerivationFramework::HIJetTrackParticleThinningTool::doThinning ( ) const

overridevirtual

Definition at line 59 of file HIJetTrackParticleThinningTool.cxx.

                                                                                   {
        // Get current event context
        const EventContext& ctx = Gaudi::Hive::currentContext();
 
        // Get TrackParticle collection
        SG::ThinningHandle<xAOD::TrackParticleContainer> tracks (m_inDetSGKey, ctx);
 
        // Check event contains tracks, if none then thinning complete!
        unsigned int nTracks = tracks->size();
        if (nTracks==0) return StatusCode::SUCCESS;
 
        // Set up a mask with same entires as full TrackParticle collection
        std::vector<bool> mask(nTracks,false); // reject all tracks by default
        m_ntot += nTracks;
 
        // Define a primary vertex
        const xAOD::Vertex *primary_vertex(nullptr);
        
        // Retrieve vertex container
        SG::ReadHandle<xAOD::VertexContainer> vtxC(m_vertexKey,ctx);
        ATH_CHECK(vtxC.isValid());
 
        // Retrieve jet containers
        SG::ReadHandle<xAOD::JetContainer> importedJets(m_jetKey,ctx);
        ATH_CHECK(importedJets.isValid());
 
        // Variables for tracking best vertex based on scheme
        float ptmax = 0.;
        std::size_t ntrkmax = 0;
 
        // Retrieve sumPt2 decoration
        SG::ReadDecorHandle<xAOD::VertexContainer, float> sumPt2Handle(m_sumPt2Key, ctx);
        ATH_CHECK(sumPt2Handle.isValid());
 
        // Iterate through vertices to find the primary vertex
        for (const xAOD::Vertex* vertex : *vtxC) {
            if (!vertex || vertex->vertexType() != xAOD::VxType::PriVtx) {
                continue;
            }
 
            if (m_vertexScheme == "sumPt2") {
                float sumPt = sumPt2Handle(*vertex);
                if (sumPt > ptmax) {
                    ptmax = sumPt;
                    primary_vertex = vertex;
                }
            }
            else {
                std::size_t ntp = vertex->nTrackParticles();
                if (ntp > ntrkmax) {
                    ntrkmax = ntp;
                    primary_vertex = vertex;
                }
            }
        }
 
        if (!primary_vertex) {
            ATH_MSG_DEBUG("No primary vertex found.");
        }
 
        // Iterate through jets, get associated tracks, set mask if passing thinning
        for (const auto  *jet : *importedJets) {
            std::vector<const xAOD::TrackParticle*> jetTracks;
            bool haveJetTracks = jet->getAssociatedObjects(xAOD::JetAttribute::GhostTrack, jetTracks);
            if ( !haveJetTracks ) { ATH_MSG_WARNING("Associated jet tracks not found!"); }
            else {
                for (auto & jetTrack : jetTracks) {
                    const xAOD::Vertex* vert_trk = primary_vertex;
 
                    asg::AcceptData acceptData = m_trkSelTool->accept(*jetTrack, vert_trk);
                    int index = jetTrack->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::HIJetTrackParticleThinningTool::finalize ( )

overridevirtual

Definition at line 52 of file HIJetTrackParticleThinningTool.cxx.

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

initialize()

StatusCode DerivationFramework::HIJetTrackParticleThinningTool::initialize ( )

overridevirtual

Definition at line 36 of file HIJetTrackParticleThinningTool.cxx.

                                                                             {
 
        // Validate m_vertexScheme
        if (m_vertexScheme != "sumPt2" && m_vertexScheme != "nTracks" ) {
            ATH_MSG_ERROR("Invalid PrimaryVertexSelection: " << m_vertexScheme);
            return StatusCode::FAILURE;
        }
 
        ATH_CHECK(m_vertexKey.initialize());
        ATH_CHECK(m_jetKey.initialize());
        ATH_CHECK(m_sumPt2Key.initialize(m_vertexScheme == "sumPt2"));
        ATH_CHECK(m_trkSelTool.retrieve());
        ATH_CHECK(m_inDetSGKey.initialize(m_streamName));
        return StatusCode::SUCCESS;
    }

Member Data Documentation

m_inDetSGKey

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

private

Definition at line 56 of file HIJetTrackParticleThinningTool.h.

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

m_jetKey

SG::ReadHandleKey<xAOD::JetContainer> DerivationFramework::HIJetTrackParticleThinningTool::m_jetKey { this, "JetKey", "", "Jet collection container"}

private

Definition at line 52 of file HIJetTrackParticleThinningTool.h.

{ this, "JetKey", "", "Jet collection container"};

m_npass

std::atomic<unsigned int> DerivationFramework::HIJetTrackParticleThinningTool::m_npass = 0

mutableprivate

Definition at line 65 of file HIJetTrackParticleThinningTool.h.

m_ntot

std::atomic<unsigned int> DerivationFramework::HIJetTrackParticleThinningTool::m_ntot = 0

mutableprivate

Definition at line 64 of file HIJetTrackParticleThinningTool.h.

m_streamName

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

private

Definition at line 47 of file HIJetTrackParticleThinningTool.h.

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

m_sumPt2Key

SG::ReadDecorHandleKey<xAOD::VertexContainer> DerivationFramework::HIJetTrackParticleThinningTool::m_sumPt2Key { this, "SumPt2Key", "PrimaryVertices.sumPt2", "SumPt2 decoration"}

private

Definition at line 54 of file HIJetTrackParticleThinningTool.h.

{ this, "SumPt2Key", "PrimaryVertices.sumPt2", "SumPt2 decoration"};

m_trkSelTool

ToolHandle<InDet::IInDetTrackSelectionTool> DerivationFramework::HIJetTrackParticleThinningTool::m_trkSelTool { this, "TrackSelectionTool", "", "Track selection tool"}

private

Definition at line 59 of file HIJetTrackParticleThinningTool.h.

{ this, "TrackSelectionTool", "", "Track selection tool"};

m_vertexKey

SG::ReadHandleKey<xAOD::VertexContainer> DerivationFramework::HIJetTrackParticleThinningTool::m_vertexKey { this, "PrimaryVertexKey", "PrimaryVertices", "Primary vertex container"}

private

Definition at line 50 of file HIJetTrackParticleThinningTool.h.

{ this, "PrimaryVertexKey", "PrimaryVertices", "Primary vertex container"};

m_vertexScheme

Gaudi::Property<std::string> DerivationFramework::HIJetTrackParticleThinningTool::m_vertexScheme { this, "PrimaryVertexSelection", "sumPt2", "Vertex selection scheme"}

private

Definition at line 61 of file HIJetTrackParticleThinningTool.h.

{ this, "PrimaryVertexSelection", "sumPt2", "Vertex selection scheme"};

---

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

• HIJetTrackParticleThinningTool.h
• HIJetTrackParticleThinningTool.cxx