DerivationFramework::BPhysPVThinningTool Class Reference

#include <BPhysPVThinningTool.h>

Inheritance diagram for DerivationFramework::BPhysPVThinningTool:

Collaboration diagram for DerivationFramework::BPhysPVThinningTool:

Public Member Functions
	BPhysPVThinningTool (const std::string &t, const std::string &n, const IInterface *p)
	Constructor with parameters. More...
	~BPhysPVThinningTool ()
	Destructor. More...
virtual StatusCode	initialize () override
virtual StatusCode	finalize () override
virtual StatusCode	doThinning () const override
	Check that the current event passes this filter. More...

Private Attributes
StringProperty	m_streamName { this, "StreamName", "", "Name of the stream being thinned" }
SG::ReadHandleKeyArray< xAOD::VertexContainer >	m_BPhyCandList
SG::ThinningHandleKey< xAOD::TrackParticleContainer >	m_TrackContainerName
SG::ThinningHandleKey< xAOD::VertexContainer >	m_PVContainerName
std::atomic< unsigned int >	m_ntot
std::atomic< unsigned int >	m_npass
std::atomic< unsigned int >	m_tracks_kept
bool	m_keepTracks

Detailed Description

Definition at line 23 of file BPhysPVThinningTool.h.

Constructor & Destructor Documentation

BPhysPVThinningTool()

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

Constructor with parameters.

Definition at line 26 of file BPhysPVThinningTool.cxx.

                              :
    base_class(t,n,p),
    m_TrackContainerName("InDetTrackParticles"),
    m_PVContainerName("PrimaryVertices"),
    m_ntot(0),
    m_npass(0), m_tracks_kept(0), m_keepTracks(false)
{
    declareProperty("CandidateCollections" , m_BPhyCandList);
    declareProperty("KeepPVTracks", m_keepTracks);
    declareProperty("TrackParticleContainerName", m_TrackContainerName);
    declareProperty("PrimaryVertexContainerName", m_PVContainerName);
}

~BPhysPVThinningTool()

DerivationFramework::BPhysPVThinningTool::~BPhysPVThinningTool

(

)

Destructor.

Definition at line 42 of file BPhysPVThinningTool.cxx.

                                                             {
}

Member Function Documentation

doThinning()

StatusCode DerivationFramework::BPhysPVThinningTool::doThinning ( ) const

overridevirtual

Check that the current event passes this filter.

Definition at line 64 of file BPhysPVThinningTool.cxx.

{
    const EventContext& ctx = Gaudi::Hive::currentContext();
    // Get the track container
    SG::ThinningHandle<xAOD::VertexContainer> PV_col(m_PVContainerName, ctx);
    if(!PV_col.isValid()) {
        ATH_MSG_ERROR ("Couldn't retrieve VertexContainer with key PrimaryVertices");
        return StatusCode::FAILURE;
    }
    m_ntot+=PV_col->size();
    // Loop over tracks, see if they pass, set mask
    std::vector<bool> mask(PV_col->size(), false);
 
    BPhysHelper::pv_type pvtypes[] = {BPhysHelper::PV_MAX_SUM_PT2,
                      BPhysHelper::PV_MIN_A0,
                      BPhysHelper::PV_MIN_Z0,
                      BPhysHelper::PV_MIN_Z0_BA};
 
    
 
    for(auto &str : m_BPhyCandList) {
        SG::ReadHandle<xAOD::VertexContainer> Container(str, ctx);
        ATH_CHECK(Container.isValid());
        size_t s = Container->size();
        for(size_t i = 0; i<s; i++) {
            xAOD::BPhysHelper vtx(Container->at(i));
 
            for(size_t i =0; i < 4; i++) {
                 const xAOD::Vertex* origPv = vtx.origPv(pvtypes[i]);
                 if(origPv==nullptr) continue;
                 auto pvit = std::find (PV_col->begin(), PV_col->end(), origPv);
                 if(pvit == PV_col->end()) {
                    ATH_MSG_WARNING("PV not found in container");
                    continue;
                 }
                size_t x = std::distance(PV_col->begin(), pvit);
                mask.at(x) = true;
            }
 
        }
    }
 
    m_npass += std::accumulate(mask.begin(), mask.end(), 0);
 
    if(m_keepTracks){
         SG::ThinningHandle<xAOD::TrackParticleContainer> importedTrackParticles(m_TrackContainerName, ctx);
         std::vector<bool> trackmask(importedTrackParticles->size(), false);
         size_t pvnum = mask.size();
         for(size_t i =0; i<pvnum;i++){
            if(mask[i] == false) continue;
            auto vtx =  PV_col->at(i);
            size_t s = vtx->nTrackParticles();
            for(size_t j=0;j<s;j++){
                auto trackit = std::find(importedTrackParticles->begin(), importedTrackParticles->end(), vtx->trackParticle(j));
                if(trackit == importedTrackParticles->end()){
                    ATH_MSG_WARNING("track not found in container");
                    continue;
                }
                size_t x = std::distance(importedTrackParticles->begin(), trackit);
                trackmask.at(x) = true;
            }
         }
      importedTrackParticles.keep(trackmask);
      m_tracks_kept += std::accumulate(trackmask.begin(), trackmask.end(), 0);
    }
    PV_col.keep(mask);
 
    return StatusCode::SUCCESS;
}

finalize()

StatusCode DerivationFramework::BPhysPVThinningTool::finalize ( )

overridevirtual

Definition at line 55 of file BPhysPVThinningTool.cxx.

{
    ATH_MSG_VERBOSE("finalize() ...");
    ATH_MSG_INFO("Processed "<< m_ntot <<" PV, "<< m_npass<< " were retained ");
    if(m_keepTracks) ATH_MSG_INFO("Additional tracks kept " << m_tracks_kept);
    return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::BPhysPVThinningTool::initialize ( )

overridevirtual

Definition at line 46 of file BPhysPVThinningTool.cxx.

{
    ATH_MSG_VERBOSE("initialize() ...");
    ATH_CHECK(m_BPhyCandList.initialize());
    if(not m_TrackContainerName.key().empty()) ATH_CHECK(m_TrackContainerName.initialize(m_streamName));
    ATH_CHECK(m_PVContainerName.initialize(m_streamName));
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_BPhyCandList

SG::ReadHandleKeyArray<xAOD::VertexContainer> DerivationFramework::BPhysPVThinningTool::m_BPhyCandList

private

Definition at line 41 of file BPhysPVThinningTool.h.

m_keepTracks

bool DerivationFramework::BPhysPVThinningTool::m_keepTracks

private

Definition at line 47 of file BPhysPVThinningTool.h.

m_npass

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

mutableprivate

Definition at line 45 of file BPhysPVThinningTool.h.

m_ntot

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

mutableprivate

Definition at line 44 of file BPhysPVThinningTool.h.

m_PVContainerName

SG::ThinningHandleKey<xAOD::VertexContainer> DerivationFramework::BPhysPVThinningTool::m_PVContainerName

private

Definition at line 43 of file BPhysPVThinningTool.h.

m_streamName

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

private

Definition at line 40 of file BPhysPVThinningTool.h.

m_TrackContainerName

SG::ThinningHandleKey<xAOD::TrackParticleContainer> DerivationFramework::BPhysPVThinningTool::m_TrackContainerName

private

Definition at line 42 of file BPhysPVThinningTool.h.

m_tracks_kept

std::atomic<unsigned int> DerivationFramework::BPhysPVThinningTool::m_tracks_kept

mutableprivate

Definition at line 46 of file BPhysPVThinningTool.h.

---

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

• BPhysPVThinningTool.h
• BPhysPVThinningTool.cxx