tt::StubAssociation Class Reference

Class to associate reconstrucable TrackingParticles with TTStubs and vice versa. It may associate multiple TPs with a TTStub and can therefore be used to associate TTTracks with TrackingParticles. More...

#include <StubAssociation.h>

Public Member Functions
std::vector< TPPtr >	associate (const std::vector< TTStubRef > &ttStubRefs) const
std::vector< TPPtr >	associateFinal (const std::vector< TTStubRef > &ttStubRefs) const
std::vector< TPPtr >	findTrackingParticlePtrs (const TTStubRef &ttStubRef) const
std::vector< TTStubRef >	findTTStubRefs (const TPPtr &tpPtr) const
const std::map< TPPtr, std::vector< TTStubRef > > &	getTrackingParticleToTTStubsMap () const
const std::map< TTStubRef, std::vector< TPPtr > > &	getTTStubToTrackingParticlesMap () const
void	insert (const TPPtr &tpPtr, const std::vector< TTStubRef > &ttSTubRefs)
int	numStubs () const
int	numTPs () const
	StubAssociation ()
	StubAssociation (const Setup *setup)
	~StubAssociation ()

Private Attributes
const std::vector< TPPtr >	emptyTPPtrs_
const std::vector< TTStubRef >	emptyTTStubRefs_
std::map< TPPtr, std::vector< TTStubRef > >	mapTPPtrsTTStubRefs_
std::map< TTStubRef, std::vector< TPPtr > >	mapTTStubRefsTPPtrs_
const Setup *	setup_

Detailed Description

Class to associate reconstrucable TrackingParticles with TTStubs and vice versa. It may associate multiple TPs with a TTStub and can therefore be used to associate TTTracks with TrackingParticles.

Author

Thomas Schuh

Date

2020, Apr

Definition at line 19 of file StubAssociation.h.

Constructor & Destructor Documentation

StubAssociation() [1/2]

tt::StubAssociation::StubAssociation ( )

inline

Definition at line 21 of file StubAssociation.h.

References setup_.

{ setup_ = nullptr; }

StubAssociation() [2/2]

tt::StubAssociation::StubAssociation ( const Setup *  setup )

inline

Definition at line 22 of file StubAssociation.h.

: setup_(setup) {}

~StubAssociation()

tt::StubAssociation::~StubAssociation ( )

inline

Definition at line 23 of file StubAssociation.h.

{}

Member Function Documentation

associate()

vector< TPPtr > tt::StubAssociation::associate

(

const std::vector< TTStubRef > &

ttStubRefs

)

const

Definition at line 33 of file StubAssociation.cc.

References edm::eventsetup::heterocontainer::insert(), createfilelist::int, visualization-live-secondInstance_cfg::m, and AlCaHLTBitMon_ParallelJobs::p.

                                                                                    {
    // count associated layer for each TP
    map<TPPtr, set<int>> m;
    map<TPPtr, set<int>> mPS;
    for (const TTStubRef& ttStubRef : ttStubRefs) {
      for (const TPPtr& tpPtr : findTrackingParticlePtrs(ttStubRef)) {
        const int layerId = setup_->layerId(ttStubRef);
        m[tpPtr].insert(layerId);
        if (setup_->psModule(ttStubRef))
          mPS[tpPtr].insert(layerId);
      }
    }
    // count matched TPs
    auto acc = [this](int sum, const pair<TPPtr, set<int>>& p) {
      return sum + ((int)p.second.size() < setup_->tpMinLayers() ? 0 : 1);
    };
    const int nTPs = accumulate(m.begin(), m.end(), 0, acc);
    vector<TPPtr> tpPtrs;
    tpPtrs.reserve(nTPs);
    // fill and return matched TPs
    for (const auto& p : m)
      if ((int)p.second.size() >= setup_->tpMinLayers() && (int)mPS[p.first].size() >= setup_->tpMinLayersPS())
        tpPtrs.push_back(p.first);
    return tpPtrs;
  }

associateFinal()

std::vector< TPPtr > tt::StubAssociation::associateFinal

(

const std::vector< TTStubRef > &

ttStubRefs

)

const

Definition at line 60 of file StubAssociation.cc.

References RPCNoise_example::check, and spr::find().

                                                                                               {
    // Get all TPs that are matched to these stubs in at least 'tpMinLayers' layers and 'tpMinLayersPS' ps layers
    vector<TPPtr> tpPtrs = associate(ttStubRefs);
    // remove TPs with more then 'tpMaxBadStubs2S' not associated 2S stubs and more then 'tpMaxBadStubsPS' not associated PS stubs
    auto check = [this, &ttStubRefs](const TPPtr& tpPtr) {
      int bad2S(0);
      int badPS(0);
      for (const TTStubRef& ttStubRef : ttStubRefs) {
        const vector<TPPtr>& tpPtrs = findTrackingParticlePtrs(ttStubRef);
        if (find(tpPtrs.begin(), tpPtrs.end(), tpPtr) == tpPtrs.end())
          setup_->psModule(ttStubRef) ? badPS++ : bad2S++;
      }
      if (badPS > setup_->tpMaxBadStubsPS() || bad2S > setup_->tpMaxBadStubs2S())
        return true;
      return false;
    };
    tpPtrs.erase(remove_if(tpPtrs.begin(), tpPtrs.end(), check), tpPtrs.end());
    return tpPtrs;
  }

findTrackingParticlePtrs()

vector< TPPtr > tt::StubAssociation::findTrackingParticlePtrs

(

const TTStubRef &

ttStubRef

)

const

Definition at line 20 of file StubAssociation.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it.

                                                                                          {
    const auto it = mapTTStubRefsTPPtrs_.find(ttStubRef);
    const vector<TPPtr> res = it != mapTTStubRefsTPPtrs_.end() ? it->second : emptyTPPtrs_;
    return res;
  }

findTTStubRefs()

vector< TTStubRef > tt::StubAssociation::findTTStubRefs

(

const TPPtr &

tpPtr

)

const

Definition at line 27 of file StubAssociation.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it.

                                                                            {
    const auto it = mapTPPtrsTTStubRefs_.find(tpPtr);
    return it != mapTPPtrsTTStubRefs_.end() ? it->second : emptyTTStubRefs_;
  }

getTrackingParticleToTTStubsMap()

const std::map<TPPtr, std::vector<TTStubRef> >& tt::StubAssociation::getTrackingParticleToTTStubsMap ( ) const

inline

Definition at line 31 of file StubAssociation.h.

References mapTPPtrsTTStubRefs_.

                                                                                       {
      return mapTPPtrsTTStubRefs_;
    }

getTTStubToTrackingParticlesMap()

const std::map<TTStubRef, std::vector<TPPtr> >& tt::StubAssociation::getTTStubToTrackingParticlesMap ( ) const

inline

Definition at line 27 of file StubAssociation.h.

References mapTTStubRefsTPPtrs_.

                                                                                       {
      return mapTTStubRefsTPPtrs_;
    }

insert()

void tt::StubAssociation::insert	(	const TPPtr &	tpPtr,
		const std::vector< TTStubRef > &	ttSTubRefs
	)

Definition at line 13 of file StubAssociation.cc.

Referenced by tt::StubAssociator::produce().

                                                                                      {
    mapTPPtrsTTStubRefs_.insert({tpPtr, ttSTubRefs});
    for (const TTStubRef& ttSTubRef : ttSTubRefs)
      mapTTStubRefsTPPtrs_[ttSTubRef].push_back(tpPtr);
  }

numStubs()

int tt::StubAssociation::numStubs ( ) const

inline

Definition at line 39 of file StubAssociation.h.

References mapTTStubRefsTPPtrs_.

{ return mapTTStubRefsTPPtrs_.size(); };

numTPs()

int tt::StubAssociation::numTPs ( ) const

inline

Definition at line 41 of file StubAssociation.h.

References mapTPPtrsTTStubRefs_.

{ return mapTPPtrsTTStubRefs_.size(); };

Member Data Documentation

emptyTPPtrs_

const std::vector<TPPtr> tt::StubAssociation::emptyTPPtrs_

private

Definition at line 55 of file StubAssociation.h.

emptyTTStubRefs_

const std::vector<TTStubRef> tt::StubAssociation::emptyTTStubRefs_

private

Definition at line 57 of file StubAssociation.h.

mapTPPtrsTTStubRefs_

std::map<TPPtr, std::vector<TTStubRef> > tt::StubAssociation::mapTPPtrsTTStubRefs_

private

Definition at line 53 of file StubAssociation.h.

Referenced by getTrackingParticleToTTStubsMap(), and numTPs().

mapTTStubRefsTPPtrs_

std::map<TTStubRef, std::vector<TPPtr> > tt::StubAssociation::mapTTStubRefsTPPtrs_

private

Definition at line 51 of file StubAssociation.h.

Referenced by getTTStubToTrackingParticlesMap(), and numStubs().

setup_

const Setup* tt::StubAssociation::setup_

private

Definition at line 49 of file StubAssociation.h.

Referenced by upgradeWorkflowComponents.UpgradeWorkflow::setup(), and StubAssociation().