Inheritance diagram for TrackMultiSelector:

Classes
struct	Block

Public Member Functions
	TrackMultiSelector (const edm::ParameterSet &cfg)
	constructor More...

virtual	~TrackMultiSelector ()
	destructor More...

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Types
typedef math::XYZPoint	Point

Private Member Functions
void	produce (edm::Event &evt, const edm::EventSetup &es) override
	process one event More...

short	select (const reco::Track &tk, const reco::BeamSpot &beamSpot, const std::vector< Point > &points)
	return class, or -1 if rejected More...

void	selectVertices (const reco::VertexCollection &vtxs, std::vector< Point > &points)

bool	testVtx (const reco::Track &tk, const reco::BeamSpot &beamSpot, const std::vector< Point > &points, const Block &cut)

Private Attributes
edm::EDGetTokenT< reco::BeamSpot >	beamspot_

double	beamspotD0_

double	beamspotDZsigmas_

std::vector< Block >	blocks_
	filter psets More...

bool	copyExtras_
	copy only the tracks, not extras and rechits (for AOD) More...

bool	copyTrajectories_
	copy also trajectories and trajectory->track associations More...

std::vector< std::string >	labels_
	output labels More...

std::vector< TrackingRecHitRefProd >	rHits_

std::vector< reco::TrackExtraRefProd >	rTrackExtras_

std::vector< reco::TrackRefProd >	rTracks_

std::vector< edm::RefProd< std::vector< Trajectory > > >	rTrajectories_

std::unique_ptr< TrackingRecHitCollection > *	selHits_

std::unique_ptr< reco::TrackExtraCollection > *	selTrackExtras_

std::unique_ptr< reco::TrackCollection > *	selTracks_
	some storage More...

std::unique_ptr< std::vector< Trajectory > > *	selTrajs_

std::unique_ptr< TrajTrackAssociationCollection > *	selTTAss_

bool	splitOutputs_
	split selections in more sets More...

edm::InputTag	src_
	source collection label More...

edm::EDGetTokenT< reco::TrackCollection >	tokenTracks

edm::EDGetTokenT< std::vector< Trajectory > >	tokenTraj

edm::EDGetTokenT< TrajTrackAssociationCollection >	tokenTrajTrack

edm::EDGetTokenT< reco::VertexCollection >	vertices_

double	vtxChi2Prob_

int32_t	vtxNumber_
	vertex cuts More...

size_t	vtxTracks_

std::vector< std::pair< short, reco::TrackRef > >	whereItWent_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

selects a subset of a track collection, copying extra information on demand

Author: Giovanni Petrucciani

Definition at line 32 of file TrackMultiSelector.cc.

Member Typedef Documentation

typedef math::XYZPoint TrackMultiSelector::Point

private

Definition at line 51 of file TrackMultiSelector.cc.

Constructor & Destructor Documentation

TrackMultiSelector::TrackMultiSelector ( const edm::ParameterSet & cfg )

explicit

constructor

Definition at line 124 of file TrackMultiSelector.cc.

References egammaCTFFinalFitWithMaterial_cff::alias, beamspot_, beamspotD0_, beamspotDZsigmas_, blocks_, copyExtras_, copyTrajectories_, edm::ParameterSet::getParameter(), mps_fire::i, checklumidiff::l, labels_, rHits_, rTrackExtras_, rTracks_, rTrajectories_, selHits_, selTrackExtras_, selTracks_, selTrajs_, selTTAss_, splitOutputs_, src_, AlCaHLTBitMon_QueryRunRegistry::string, tokenTracks, tokenTraj, tokenTrajTrack, vertices_, and HistogramManager_cfi::VPSet().

                                                                     :
     src_( cfg.getParameter<edm::InputTag>( "src" ) ),
     copyExtras_(cfg.getUntrackedParameter<bool>("copyExtras", false)),
     copyTrajectories_(cfg.getUntrackedParameter<bool>("copyTrajectories", false)),
     splitOutputs_( cfg.getUntrackedParameter<bool>("splitOutputs", false) ),
     vtxNumber_( cfg.getParameter<int32_t>("vtxNumber") ),
     vtxTracks_( cfg.getParameter<uint32_t>("vtxTracks") ),
     vtxChi2Prob_( cfg.getParameter<double>("vtxChi2Prob") )
 {
     edm::ParameterSet beamSpotPSet = cfg.getParameter<edm::ParameterSet>("beamspot");
     beamspot_   = consumes<reco::BeamSpot>(beamSpotPSet.getParameter<edm::InputTag>("src"));
     beamspotDZsigmas_ = beamSpotPSet.getParameter<double>("dzSigmas");
     beamspotD0_ = beamSpotPSet.getParameter<double>("d0");
     vertices_= consumes<reco::VertexCollection>( cfg.getParameter<edm::InputTag>( "vertices" ) );
     tokenTracks= consumes<reco::TrackCollection>(src_);
     if (copyTrajectories_) {
       tokenTraj= consumes<std::vector<Trajectory> >(src_);
       tokenTrajTrack= consumes<TrajTrackAssociationCollection>(src_);
     }
 
     typedef std::vector<edm::ParameterSet> VPSet;
     VPSet psets = cfg.getParameter<VPSet>("cutSets");
     blocks_.reserve(psets.size());
     for (VPSet::const_iterator it = psets.begin(), ed = psets.end(); it != ed; ++it) {
         blocks_.push_back(TrackMultiSelector::Block(*it));
     }
 
     if (splitOutputs_) {
         char buff[15];
         for (size_t i = 0; i < blocks_.size(); ++i) {
             sprintf(buff,"set%d", static_cast<int>(i+1));
             labels_.push_back(std::string(buff));
         }
     } else {
         labels_.push_back(std::string(""));
     }
 
     std::string alias( cfg.getParameter<std::string>( "@module_label" ) );
     for (std::vector<std::string>::const_iterator li = labels_.begin(), le = labels_.end(); li != le; ++li) {
         const char *l= li->c_str();
         produces<reco::TrackCollection>(l).setBranchAlias( alias + "Tracks" + l);
         if (copyExtras_) {
             produces<reco::TrackExtraCollection>(l).setBranchAlias( alias + "TrackExtras" + l);
             produces<TrackingRecHitCollection>(l).setBranchAlias( alias + "RecHits" + l);
             if (copyTrajectories_) {
                 produces< std::vector<Trajectory> >(l).setBranchAlias( alias + "Trajectories" + l);
                 produces< TrajTrackAssociationCollection >(l).setBranchAlias( alias + "TrajectoryTrackAssociations" + l);
             }
         }
     }
 
     size_t nblocks = splitOutputs_ ? blocks_.size() : 1;
     selTracks_ = new std::unique_ptr<reco::TrackCollection>[nblocks];
     selTrackExtras_ = new std::unique_ptr<reco::TrackExtraCollection>[nblocks];
     selHits_ = new std::unique_ptr<TrackingRecHitCollection>[nblocks];
     selTrajs_ = new std::unique_ptr<std::vector<Trajectory> >[nblocks];
     selTTAss_ = new std::unique_ptr<TrajTrackAssociationCollection >[nblocks];
     rTracks_ = std::vector<reco::TrackRefProd>(nblocks);
     rHits_ = std::vector<TrackingRecHitRefProd>(nblocks);
     rTrackExtras_ = std::vector<reco::TrackExtraRefProd>(nblocks);
     rTrajectories_ = std::vector< edm::RefProd< std::vector<Trajectory> > >(nblocks);
     for (size_t i = 0; i < nblocks; i++) {
         selTracks_[i] = std::make_unique<reco::TrackCollection>();
         selTrackExtras_[i] = std::make_unique<reco::TrackExtraCollection>();
         selHits_[i] = std::make_unique<TrackingRecHitCollection>();
         selTrajs_[i] = std::make_unique<std::vector<Trajectory>>();
         selTTAss_[i] = std::make_unique<TrajTrackAssociationCollection>();
     }
 
 }

TrackMultiSelector::~TrackMultiSelector ( )

virtual

destructor

Definition at line 195 of file TrackMultiSelector.cc.

References selHits_, selTrackExtras_, selTracks_, selTrajs_, and selTTAss_.

                                         {
     delete [] selTracks_;
     delete [] selTrackExtras_;
     delete [] selHits_;
     delete [] selTrajs_;
     delete [] selTTAss_;
 }

Member Function Documentation

void TrackMultiSelector::produce	(	edm::Event &	evt,
		const edm::EventSetup &	es
	)

overrideprivate

process one event

Definition at line 203 of file TrackMultiSelector.cc.

References beamspot_, blocks_, edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, copyExtras_, copyTrajectories_, edm::friendlyname::emptyString(), edm::AssociationMap< Tag >::end(), edm::AssociationMap< Tag >::find(), plotBeamSpotDB::first, edm::Event::getByToken(), edm::Event::getRefBeforePut(), mps_fire::i, reco::Track::innerDetId(), reco::Track::innerMomentum(), reco::Track::innerOk(), reco::Track::innerPosition(), reco::Track::innerStateCovariance(), edm::Ref< C, T, F >::key(), labels_, match(), eostools::move(), gen::n, reco::Track::outerDetId(), reco::Track::outerMomentum(), reco::Track::outerOk(), reco::Track::outerPosition(), reco::Track::outerStateCovariance(), hiPixelPairStep_cff::points, edm::Event::put(), reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), reco::Track::residuals(), rHits_, rTrackExtras_, rTracks_, rTrajectories_, edm::second(), reco::Track::seedDirection(), reco::Track::seedRef(), select(), selectVertices(), selHits_, selTrackExtras_, selTracks_, selTrajs_, selTTAss_, reco::TrackExtraBase::setHits(), reco::TrackExtra::setResiduals(), findQualityFiles::size, splitOutputs_, AlCaHLTBitMon_QueryRunRegistry::string, tokenTracks, tokenTraj, tokenTrajTrack, vertices_, vtxNumber_, and whereItWent_.

 {
     using namespace std;
     using namespace edm;
     using namespace reco;
 
     size_t nblocks = splitOutputs_ ? blocks_.size() : 1;
 
     Handle<TrackCollection> hSrcTrack;
     Handle< vector<Trajectory> > hTraj;
     Handle< TrajTrackAssociationCollection > hTTAss;
 
     edm::Handle<reco::VertexCollection> hVtx;
     evt.getByToken(vertices_, hVtx);
     std::vector<Point> points;
     if (vtxNumber_ != 0) selectVertices(*hVtx, points);
 
     edm::Handle<reco::BeamSpot> hBsp;
     evt.getByToken(beamspot_, hBsp);
 
     evt.getByToken( tokenTracks, hSrcTrack );
 
     for (size_t i = 0; i < nblocks; i++) {
         selTracks_[i] = std::make_unique<TrackCollection>();
         rTracks_[i] = evt.getRefBeforePut<TrackCollection>(labels_[i]);
         if (copyExtras_) {
             selTrackExtras_[i] = std::make_unique<TrackExtraCollection>();
             selHits_[i] = std::make_unique<TrackingRecHitCollection>();
             rHits_[i] = evt.getRefBeforePut<TrackingRecHitCollection>(labels_[i]);
             rTrackExtras_[i] = evt.getRefBeforePut<TrackExtraCollection>(labels_[i]);
         }
     }
 
     if (copyTrajectories_) whereItWent_.resize(hSrcTrack->size());
     size_t current = 0;
     for (TrackCollection::const_iterator it = hSrcTrack->begin(), ed = hSrcTrack->end(); it != ed; ++it, ++current) {
         const Track & trk = * it;
         short where = select(trk, *hBsp, points);
         if (where == -1) {
             if (copyTrajectories_) whereItWent_[current] = std::pair<short, reco::TrackRef>(-1, reco::TrackRef());
             continue;
         }
         if (!splitOutputs_) where = 0;
         selTracks_[where]->push_back( Track( trk ) ); // clone and store
         if (!copyExtras_) continue;
 
         // TrackExtras
         selTrackExtras_[where]->push_back( TrackExtra( trk.outerPosition(), trk.outerMomentum(), trk.outerOk(),
                     trk.innerPosition(), trk.innerMomentum(), trk.innerOk(),
                     trk.outerStateCovariance(), trk.outerDetId(),
                     trk.innerStateCovariance(), trk.innerDetId(),
                     trk.seedDirection(), trk.seedRef() ) );
         selTracks_[where]->back().setExtra( TrackExtraRef( rTrackExtras_[where], selTrackExtras_[where]->size() - 1) );
         TrackExtra & tx = selTrackExtras_[where]->back();
     tx.setResiduals(trk.residuals());
         // TrackingRecHits
         auto& selHitsWhere = selHits_[where];
         auto const firstHitIndex = selHitsWhere->size();
         for( trackingRecHit_iterator hit = trk.recHitsBegin(); hit != trk.recHitsEnd(); ++ hit ) {
             selHitsWhere->push_back( (*hit)->clone() );
         }
         tx.setHits( rHits_[where], firstHitIndex, selHitsWhere->size() - firstHitIndex);
 
         if (copyTrajectories_) {
             whereItWent_[current] = std::pair<short, reco::TrackRef>(where, TrackRef(rTracks_[where], selTracks_[where]->size() - 1));
         }
     }
     if ( copyTrajectories_ ) {
         Handle< vector<Trajectory> > hTraj;
         Handle< TrajTrackAssociationCollection > hTTAss;
         evt.getByToken(tokenTrajTrack, hTTAss);
         evt.getByToken(tokenTraj, hTraj);
         for (size_t i = 0; i < nblocks; i++) {
             rTrajectories_[i] = evt.getRefBeforePut< vector<Trajectory> >(labels_[i]);
             selTrajs_[i] = std::make_unique<std::vector<Trajectory>>();
             selTTAss_[i] = std::make_unique<TrajTrackAssociationCollection>();
         }
         for (size_t i = 0, n = hTraj->size(); i < n; ++i) {
             Ref< vector<Trajectory> > trajRef(hTraj, i);
             TrajTrackAssociationCollection::const_iterator match = hTTAss->find(trajRef);
             if (match != hTTAss->end()) {
                 const Ref<TrackCollection> &trkRef = match->val;
                 short oldKey = static_cast<short>(trkRef.key());
                 if (whereItWent_[oldKey].first != -1) {
                     int where = whereItWent_[oldKey].first;
                     selTrajs_[where]->push_back( Trajectory(*trajRef) );
                     selTTAss_[where]->insert ( Ref< vector<Trajectory> >(rTrajectories_[where], selTrajs_[where]->size() - 1), whereItWent_[oldKey].second );
                 }
             }
         }
     }
 
 
     static const std::string emptyString;
     for (size_t i = 0; i < nblocks; i++) {
         const std::string & lbl = ( splitOutputs_ ? labels_[i] : emptyString);
         evt.put(std::move(selTracks_[i]), lbl);
         if (copyExtras_ ) {
             evt.put(std::move(selTrackExtras_[i]), lbl);
             evt.put(std::move(selHits_[i]), lbl);
             if ( copyTrajectories_ ) {
                 evt.put(std::move(selTrajs_[i]), lbl);
                 evt.put(std::move(selTTAss_[i]), lbl);
             }
         }
     }
 }

short TrackMultiSelector::select	(	const reco::Track &	tk,
		const reco::BeamSpot &	beamSpot,
		const std::vector< Point > &	points
	)

private

return class, or -1 if rejected

Definition at line 329 of file TrackMultiSelector.cc.

References blocks_, chi2n, reco::TrackBase::hitPattern(), reco::TrackBase::normalizedChi2(), reco::TrackBase::numberOfLostHits(), EnergyCorrector::pt, reco::TrackBase::pt(), testVtx(), reco::HitPattern::trackerLayersWithMeasurement(), and eostools::which().

Referenced by produce().

                                                                                                                   {
    uint32_t vlayers = tk.hitPattern().trackerLayersWithMeasurement(), lhits = tk.numberOfLostHits();
    double pt = tk.pt(), chi2n =  tk.normalizedChi2();
    int which = 0;
    for (std::vector<TrackMultiSelector::Block>::const_iterator itb = blocks_.begin(), edb = blocks_.end(); itb != edb; ++itb, ++which) {
         if ( ( itb->vlayers.first <= vlayers ) && ( vlayers <= itb->vlayers.second ) &&
              ( itb->chi2n.first <= chi2n ) && ( chi2n <= itb->chi2n.second ) &&
              ( itb->pt.first    <= pt    ) && ( pt    <= itb->pt.second    ) &&
              ( itb->lhits.first <= lhits ) && ( lhits <= itb->lhits.second ) &&
              testVtx(tk, beamSpot, points, *itb) )
         {
             return which;
         }
     }
     return -1;
 }

void TrackMultiSelector::selectVertices	(	const reco::VertexCollection &	vtxs,
		std::vector< Point > &	points
	)

private

Definition at line 345 of file TrackMultiSelector.cc.

References DEFINE_FWK_MODULE, vtxChi2Prob_, vtxNumber_, and vtxTracks_.

Referenced by produce().

                                                                                                   {
     using namespace reco;
 
     int32_t toTake = vtxNumber_;
     for (VertexCollection::const_iterator it = vtxs.begin(), ed = vtxs.end(); it != ed; ++it) {
         if ((it->tracksSize() >= vtxTracks_)  &&
                 ( (it->chi2() == 0.0) || (TMath::Prob(it->chi2(), static_cast<int32_t>(it->ndof()) ) >= vtxChi2Prob_) ) ) {
             points.push_back(it->position());
             toTake--; if (toTake == 0) break;
         }
     }
 }

bool TrackMultiSelector::testVtx	(	const reco::Track &	tk,
		const reco::BeamSpot &	beamSpot,
		const std::vector< Point > &	points,
		const Block &	cut
	)

inlineprivate

Definition at line 311 of file TrackMultiSelector.cc.

References funct::abs(), beamspotD0_, beamspotDZsigmas_, allConversions_cfi::d0, TrackMultiSelector::Block::d0, reco::TrackBase::d0Error(), TrackMultiSelector::Block::d0Rel, reco::TrackBase::dxy(), PVValHelper::dz, TrackMultiSelector::Block::dz, reco::TrackBase::dz(), reco::TrackBase::dzError(), TrackMultiSelector::Block::dzRel, MillePedeFileConverter_cfg::e, end, hpstanc_transforms::max, point, reco::BeamSpot::position(), and reco::BeamSpot::sigmaZ().

Referenced by select().

                                                            {
     using std::abs;
     double d0Err =abs(tk.d0Error()), dzErr = abs(tk.dzError());  // not fully sure they're > 0!
     if (points.empty()) {
         Point spot = beamSpot.position();
         double dz = abs(tk.dz(spot)), d0 = abs(tk.dxy(spot));
         return ( dz < beamspotDZsigmas_*beamSpot.sigmaZ() ) && ( d0 < beamspotD0_ );
     }
     for (std::vector<Point>::const_iterator point = points.begin(), end = points.end(); point != end; ++point) {
         double dz = abs(tk.dz(*point)), d0 = abs(tk.dxy(*point));
         if ((dz < cut.dz) && (d0 < cut.d0)
         && fabs(dz/std::max(dzErr,1e-9)) < cut.dzRel && (d0/std::max(d0Err,1e-8) < cut.d0Rel )) return true;
     }
     return false;
 }