238 std::unique_ptr<reco::TrackCollection> selTracks_;
239 std::unique_ptr<reco::TrackExtraCollection> selTrackExtras_;
240 std::unique_ptr<TrackingRecHitCollection> selHits_;
241 std::unique_ptr<std::vector<Trajectory>> selTrajs_;
242 std::unique_ptr<std::vector<const Trajectory*>> selTrajPtrs_;
243 std::unique_ptr<TrajTrackAssociationCollection> selTTAss_;
248 std::vector<reco::TrackRef> trackRefs_;
252 using namespace reco;
263 vertexBeamSpot = *hBsp;
268 std::vector<Point> points;
269 std::vector<float> vterr, vzerr;
274 LogDebug(
"SelectVertex") << points.size() <<
" good pixel vertices";
284 selTracks_ = std::make_unique<TrackCollection>();
287 selTrackExtras_ = std::make_unique<TrackExtraCollection>();
288 selHits_ = std::make_unique<TrackingRecHitCollection>();
294 trackRefs_.resize(hSrcTrack->size());
296 std::vector<float> mvaVals_(hSrcTrack->size(), -99.f);
297 processMVA(evt, es, vertexBeamSpot, *vtxPtr, 0, mvaVals_,
true);
301 for (TrackCollection::const_iterator
it = hSrcTrack->begin(), ed = hSrcTrack->end();
it != ed; ++
it, ++current) {
305 LogTrace(
"TrackSelection") <<
"ready to check track with pt=" <<
trk.pt();
309 mvaVal = mvaVals_[current];
310 bool ok =
select(0, vertexBeamSpot, srcHits,
trk, points, vterr, vzerr, mvaVal);
312 LogTrace(
"TrackSelection") <<
"track with pt=" <<
trk.pt() <<
" NOT selected";
319 LogTrace(
"TrackSelection") <<
"track with pt=" <<
trk.pt() <<
" selected";
327 selTracks_->back().setQuality(TrackBase::tight);
329 if (!points.empty()) {
331 selTracks_->back().setQuality(TrackBase::looseSetWithPV);
333 selTracks_->back().setQuality(TrackBase::looseSetWithPV);
334 selTracks_->back().setQuality(TrackBase::highPuritySetWithPV);
346 trk.outerStateCovariance(),
348 trk.innerStateCovariance(),
352 selTracks_->back().setExtra(
TrackExtraRef(rTrackExtras_, selTrackExtras_->size() - 1));
356 auto const firstHitIndex = selHits_->size();
358 selHits_->push_back((*hit)->clone());
360 tx.
setHits(rHits_, firstHitIndex, selHits_->size() - firstHitIndex);
364 trackRefs_[current] =
TrackRef(rTracks_, selTracks_->size() - 1);
372 selTrajs_ = std::make_unique<std::vector<Trajectory>>();
374 selTTAss_ = std::make_unique<TrajTrackAssociationCollection>(rTrajectories_, rTracks_);
375 for (
size_t i = 0,
n = hTraj->size();
i <
n; ++
i) {
380 short oldKey =
static_cast<short>(trkRef.
key());
381 if (trackRefs_[oldKey].isNonnull()) {
383 selTTAss_->insert(
Ref<vector<Trajectory>>(rTrajectories_, selTrajs_->size() - 1), trackRefs_[oldKey]);
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
bool get(ProductID const &oid, Handle< PROD > &result) const
edm::EDGetTokenT< TrajTrackAssociationCollection > srcTass_
std::vector< bool > keepAllTracks_
bool copyTrajectories_
copy also trajectories and trajectory->track associations
std::vector< Track > TrackCollection
collection of Tracks
std::vector< Vertex > VertexCollection
collection of Vertex objects
edm::Ref< TrackExtraCollection > TrackExtraRef
persistent reference to a TrackExtra
bool getByToken(EDGetToken token, Handle< PROD > &result) const
edm::EDGetTokenT< TrackingRecHitCollection > hSrc_
edm::EDGetTokenT< std::vector< Trajectory > > srcTraj_
edm::EDGetTokenT< reco::VertexCollection > vertices_
key_type key() const
Accessor for product key.
const_iterator find(const key_type &k) const
find element with specified reference key
const_iterator end() const
last iterator over the map (read only)
edm::EDGetTokenT< reco::BeamSpot > beamspot_
std::vector< bool > setQualityBit_
do I have to set a quality bit?
void processMVA(edm::Event &evt, const edm::EventSetup &es, const reco::BeamSpot &beamspot, const reco::VertexCollection &vertices, int selIndex, std::vector< float > &mvaVals_, bool writeIt=false) const
void selectVertices(unsigned int tsNum, const reco::VertexCollection &vtxs, std::vector< Point > &points, std::vector< float > &vterr, std::vector< float > &vzerr) const
bool select(unsigned tsNum, const reco::BeamSpot &vertexBeamSpot, const TrackingRecHitCollection &recHits, const reco::Track &tk, const std::vector< Point > &points, std::vector< float > &vterr, std::vector< float > &vzerr, double mvaVal) const
return class, or -1 if rejected
RefProd< PROD > getRefBeforePut()
std::vector< TrackExtra > TrackExtraCollection
collection of TrackExtra objects
std::vector< reco::TrackBase::TrackQuality > qualityToSet_
edm::Ref< TrackCollection > TrackRef
persistent reference to a Track
bool copyExtras_
copy only the tracks, not extras and rechits (for AOD)
std::pair< typename Association::data_type::first_type, double > match(Reference key, Association association, bool bestMatchByMaxValue)
Generic matching function.
edm::EDGetTokenT< reco::TrackCollection > src_
source collection label