CMS 3D CMS Logo

List of all members | Public Member Functions | Private Types | Private Member Functions | Private Attributes
AnalyticalTrackSelector Class Referencefinal
Inheritance diagram for AnalyticalTrackSelector:
MultiTrackSelector edm::stream::EDProducer<>

Public Member Functions

 AnalyticalTrackSelector (const edm::ParameterSet &cfg)
 constructor More...
 
 ~AnalyticalTrackSelector () override
 destructor More...
 
- Public Member Functions inherited from MultiTrackSelector
 MultiTrackSelector ()
 constructor More...
 
 MultiTrackSelector (const edm::ParameterSet &cfg)
 
 ~MultiTrackSelector () override
 destructor More...
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
bool hasAbilityToProduceInBeginLumis () const final
 
bool hasAbilityToProduceInBeginRuns () const final
 
bool hasAbilityToProduceInEndLumis () const final
 
bool hasAbilityToProduceInEndRuns () const final
 

Private Types

typedef math::XYZPoint Point
 

Private Member Functions

void run (edm::Event &evt, const edm::EventSetup &es) const override
 process one event More...
 

Private Attributes

bool copyExtras_
 copy only the tracks, not extras and rechits (for AOD) More...
 
bool copyTrajectories_
 copy also trajectories and trajectory->track associations More...
 
double maxEta_
 
double minEta_
 eta restrictions More...
 
edm::EDGetTokenT< TrajTrackAssociationCollectionsrcTass_
 
edm::EDGetTokenT< std::vector< Trajectory > > srcTraj_
 

Additional Inherited Members

- Public Types inherited from MultiTrackSelector
using MVACollection = std::vector< float >
 
using QualityMaskCollection = std::vector< unsigned char >
 
- 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, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 
- Protected Types inherited from MultiTrackSelector
typedef math::XYZPoint Point
 
- Protected Member Functions inherited from MultiTrackSelector
void beginStream (edm::StreamID) final
 
Point getBestVertex (const reco::TrackBaseRef, const reco::VertexCollection) const
 
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 produce (edm::Event &evt, const edm::EventSetup &es) final
 process one event More...
 
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 More...
 
void selectVertices (unsigned int tsNum, const reco::VertexCollection &vtxs, std::vector< Point > &points, std::vector< float > &vterr, std::vector< float > &vzerr) const
 
- Protected Attributes inherited from MultiTrackSelector
std::vector< bool > applyAbsCutsIfNoPV_
 
std::vector< bool > applyAdaptedPVCuts_
 
edm::EDGetTokenT< reco::BeamSpotbeamspot_
 
std::vector< double > chi2n_no1Dmod_par_
 
std::vector< double > chi2n_par_
 
std::vector< std::vector< double > > d0_par1_
 
std::vector< std::vector< double > > d0_par2_
 
std::string dbFileName_
 
std::vector< std::vector< double > > dz_par1_
 
std::vector< std::vector< double > > dz_par2_
 
std::vector< GBRForest * > forest_
 
std::vector< std::string > forestLabel_
 
edm::EDGetTokenT< TrackingRecHitCollectionhSrc_
 
std::vector< bool > keepAllTracks_
 
std::vector< double > max_d0_
 Impact parameter absolute cuts. More...
 
std::vector< double > max_d0NoPV_
 
std::vector< double > max_eta_
 
std::vector< int32_t > max_lostHitFraction_
 
std::vector< uint32_t > max_lostLayers_
 
std::vector< int32_t > max_minMissHitOutOrIn_
 
std::vector< double > max_relpterr_
 
std::vector< double > max_z0_
 
std::vector< double > max_z0NoPV_
 
std::vector< uint32_t > min_3Dlayers_
 
std::vector< double > min_eta_
 
std::vector< uint32_t > min_hits_bypass_
 
std::vector< uint32_t > min_layers_
 Cuts on numbers of layers with hits/3D hits/lost hits. More...
 
std::vector< double > min_MVA_
 
std::vector< uint32_t > min_nhits_
 
std::vector< std::string > mvaType_
 
std::vector< std::string > name_
 
std::vector< double > nSigmaZ_
 
std::vector< unsigned int > preFilter_
 
std::vector< reco::TrackBase::TrackQualityqualityToSet_
 
std::vector< std::vector< double > > res_par_
 
std::vector< bool > setQualityBit_
 do I have to set a quality bit? More...
 
edm::EDGetTokenT< reco::TrackCollectionsrc_
 source collection label More...
 
bool useAnyMVA_
 
bool useForestFromDB_
 
std::vector< bool > useMVA_
 
std::vector< bool > useMVAonly_
 
bool useVertices_
 
bool useVtxError_
 
std::vector< StringCutObjectSelector< reco::Vertex > > vertexCut_
 
edm::EDGetTokenT< reco::VertexCollectionvertices_
 
std::vector< int32_t > vtxNumber_
 vertex cuts More...
 

Detailed Description

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

Author
Paolo Azzurri, Giovanni Petrucciani

Definition at line 36 of file AnalyticalTrackSelector.cc.

Member Typedef Documentation

◆ Point

Definition at line 45 of file AnalyticalTrackSelector.cc.

Constructor & Destructor Documentation

◆ AnalyticalTrackSelector()

AnalyticalTrackSelector::AnalyticalTrackSelector ( const edm::ParameterSet cfg)
explicit

constructor

Definition at line 67 of file AnalyticalTrackSelector.cc.

67  : MultiTrackSelector() {
68  //Spoof the pset for each track selector!
69  //Size is always 1!!!
70  qualityToSet_.reserve(1);
71  vtxNumber_.reserve(1);
72  vertexCut_.reserve(1);
73  res_par_.reserve(1);
74  chi2n_par_.reserve(1);
75  chi2n_no1Dmod_par_.reserve(1);
76  d0_par1_.reserve(1);
77  dz_par1_.reserve(1);
78  d0_par2_.reserve(1);
79  dz_par2_.reserve(1);
80  applyAdaptedPVCuts_.reserve(1);
81  max_d0_.reserve(1);
82  max_z0_.reserve(1);
83  nSigmaZ_.reserve(1);
84  min_layers_.reserve(1);
85  min_3Dlayers_.reserve(1);
86  max_lostLayers_.reserve(1);
87  min_hits_bypass_.reserve(1);
88  applyAbsCutsIfNoPV_.reserve(1);
89  max_d0NoPV_.reserve(1);
90  max_z0NoPV_.reserve(1);
91  preFilter_.reserve(1);
92  max_relpterr_.reserve(1);
93  min_nhits_.reserve(1);
94  max_minMissHitOutOrIn_.reserve(1);
95  max_lostHitFraction_.reserve(1);
96  min_eta_.reserve(1);
97  max_eta_.reserve(1);
98  forest_.reserve(1);
99  mvaType_.reserve(1);
100  useMVA_.reserve(1);
101 
102  produces<edm::ValueMap<float>>("MVAVals");
103  //foward compatibility
104  produces<MVACollection>("MVAValues");
105  useAnyMVA_ = false;
106  forest_[0] = nullptr;
107  if (cfg.exists("useAnyMVA"))
108  useAnyMVA_ = cfg.getParameter<bool>("useAnyMVA");
109 
110  src_ = consumes<reco::TrackCollection>(cfg.getParameter<edm::InputTag>("src"));
111  hSrc_ = consumes<TrackingRecHitCollection>(cfg.getParameter<edm::InputTag>("src"));
112  beamspot_ = consumes<reco::BeamSpot>(cfg.getParameter<edm::InputTag>("beamspot"));
113  useVertices_ = cfg.getParameter<bool>("useVertices");
114  useVtxError_ = cfg.getParameter<bool>("useVtxError");
115  if (useVertices_)
116  vertices_ = consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("vertices"));
117  copyExtras_ = cfg.getUntrackedParameter<bool>("copyExtras", false);
118  copyTrajectories_ = cfg.getUntrackedParameter<bool>("copyTrajectories", false);
119  if (copyTrajectories_) {
120  srcTraj_ = consumes<std::vector<Trajectory>>(cfg.getParameter<edm::InputTag>("src"));
121  srcTass_ = consumes<TrajTrackAssociationCollection>(cfg.getParameter<edm::InputTag>("src"));
122  }
123 
124  qualityToSet_.push_back(TrackBase::undefQuality);
125  // parameters for vertex selection
126  vtxNumber_.push_back(useVertices_ ? cfg.getParameter<int32_t>("vtxNumber") : 0);
127  vertexCut_.push_back(useVertices_ ? cfg.getParameter<std::string>("vertexCut") : "");
128  // parameters for adapted optimal cuts on chi2 and primary vertex compatibility
129  res_par_.push_back(cfg.getParameter<std::vector<double>>("res_par"));
130  chi2n_par_.push_back(cfg.getParameter<double>("chi2n_par"));
131  chi2n_no1Dmod_par_.push_back(cfg.getParameter<double>("chi2n_no1Dmod_par"));
132  d0_par1_.push_back(cfg.getParameter<std::vector<double>>("d0_par1"));
133  dz_par1_.push_back(cfg.getParameter<std::vector<double>>("dz_par1"));
134  d0_par2_.push_back(cfg.getParameter<std::vector<double>>("d0_par2"));
135  dz_par2_.push_back(cfg.getParameter<std::vector<double>>("dz_par2"));
136 
137  // Boolean indicating if adapted primary vertex compatibility cuts are to be applied.
138  applyAdaptedPVCuts_.push_back(cfg.getParameter<bool>("applyAdaptedPVCuts"));
139  // Impact parameter absolute cuts.
140  max_d0_.push_back(cfg.getParameter<double>("max_d0"));
141  max_z0_.push_back(cfg.getParameter<double>("max_z0"));
142  nSigmaZ_.push_back(cfg.getParameter<double>("nSigmaZ"));
143  // Cuts on numbers of layers with hits/3D hits/lost hits.
144  min_layers_.push_back(cfg.getParameter<uint32_t>("minNumberLayers"));
145  min_3Dlayers_.push_back(cfg.getParameter<uint32_t>("minNumber3DLayers"));
146  max_lostLayers_.push_back(cfg.getParameter<uint32_t>("maxNumberLostLayers"));
147  min_hits_bypass_.push_back(cfg.getParameter<uint32_t>("minHitsToBypassChecks"));
148  max_relpterr_.push_back(cfg.getParameter<double>("max_relpterr"));
149  min_nhits_.push_back(cfg.getParameter<uint32_t>("min_nhits"));
150  max_minMissHitOutOrIn_.push_back(
151  cfg.existsAs<int32_t>("max_minMissHitOutOrIn") ? cfg.getParameter<int32_t>("max_minMissHitOutOrIn") : 99);
152  max_lostHitFraction_.push_back(
153  cfg.existsAs<double>("max_lostHitFraction") ? cfg.getParameter<double>("max_lostHitFraction") : 1.0);
154  min_eta_.push_back(cfg.getParameter<double>("min_eta"));
155  max_eta_.push_back(cfg.getParameter<double>("max_eta"));
156 
157  // Flag to apply absolute cuts if no PV passes the selection
158  applyAbsCutsIfNoPV_.push_back(cfg.getParameter<bool>("applyAbsCutsIfNoPV"));
159  keepAllTracks_.push_back(cfg.exists("keepAllTracks") ? cfg.getParameter<bool>("keepAllTracks") : false);
160 
161  setQualityBit_.push_back(false);
162  std::string qualityStr = cfg.getParameter<std::string>("qualityBit");
163 
164  if (d0_par1_[0].size() != 2 || dz_par1_[0].size() != 2 || d0_par2_[0].size() != 2 || dz_par2_[0].size() != 2) {
165  edm::LogError("MisConfiguration") << "vector of size less then 2";
166  throw;
167  }
168 
169  if (cfg.exists("qualityBit")) {
170  std::string qualityStr = cfg.getParameter<std::string>("qualityBit");
171  if (!qualityStr.empty()) {
172  setQualityBit_[0] = true;
173  qualityToSet_[0] = TrackBase::qualityByName(cfg.getParameter<std::string>("qualityBit"));
174  }
175  }
176 
177  if (keepAllTracks_[0] && !setQualityBit_[0])
178  throw cms::Exception("Configuration")
179  << "If you set 'keepAllTracks' to true, you must specify which qualityBit to set.\n";
180  if (setQualityBit_[0] && (qualityToSet_[0] == TrackBase::undefQuality))
181  throw cms::Exception("Configuration")
182  << "You can't set the quality bit " << cfg.getParameter<std::string>("qualityBit")
183  << " as it is 'undefQuality' or unknown.\n";
184  if (applyAbsCutsIfNoPV_[0]) {
185  max_d0NoPV_.push_back(cfg.getParameter<double>("max_d0NoPV"));
186  max_z0NoPV_.push_back(cfg.getParameter<double>("max_z0NoPV"));
187  } else { //dummy values
188  max_d0NoPV_.push_back(0.);
189  max_z0NoPV_.push_back(0.);
190  }
191 
192  if (useAnyMVA_) {
193  bool thisMVA = false;
194  if (cfg.exists("useMVA"))
195  thisMVA = cfg.getParameter<bool>("useMVA");
196  useMVA_.push_back(thisMVA);
197  if (thisMVA) {
198  double minVal = -1;
199  if (cfg.exists("minMVA"))
200  minVal = cfg.getParameter<double>("minMVA");
201  min_MVA_.push_back(minVal);
202  mvaType_.push_back(cfg.exists("mvaType") ? cfg.getParameter<std::string>("mvaType") : "Detached");
203  forestLabel_.push_back(cfg.exists("GBRForestLabel") ? cfg.getParameter<std::string>("GBRForestLabel")
204  : "MVASelectorIter0");
205  useMVAonly_.push_back(cfg.exists("useMVAonly") ? cfg.getParameter<bool>("useMVAonly") : false);
206  } else {
207  min_MVA_.push_back(-9999.0);
208  useMVAonly_.push_back(false);
209  mvaType_.push_back("Detached");
210  forestLabel_.push_back("MVASelectorIter0");
211  }
212  } else {
213  useMVA_.push_back(false);
214  useMVAonly_.push_back(false);
215  min_MVA_.push_back(-9999.0);
216  mvaType_.push_back("Detached");
217  forestLabel_.push_back("MVASelectorIter0");
218  }
219 
220  std::string alias(cfg.getParameter<std::string>("@module_label"));
221  produces<reco::TrackCollection>().setBranchAlias(alias + "Tracks");
222  if (copyExtras_) {
223  produces<reco::TrackExtraCollection>().setBranchAlias(alias + "TrackExtras");
224  produces<TrackingRecHitCollection>().setBranchAlias(alias + "RecHits");
225  }
226  if (copyTrajectories_) {
227  produces<std::vector<Trajectory>>().setBranchAlias(alias + "Trajectories");
228  produces<TrajTrackAssociationCollection>().setBranchAlias(alias + "TrajectoryTrackAssociations");
229  }
230 }

References SiStripOfflineCRack_cfg::alias, MultiTrackSelector::applyAbsCutsIfNoPV_, MultiTrackSelector::applyAdaptedPVCuts_, MultiTrackSelector::beamspot_, looper::cfg, MultiTrackSelector::chi2n_no1Dmod_par_, MultiTrackSelector::chi2n_par_, copyExtras_, copyTrajectories_, MultiTrackSelector::d0_par1_, MultiTrackSelector::d0_par2_, MultiTrackSelector::dz_par1_, MultiTrackSelector::dz_par2_, Exception, MultiTrackSelector::forest_, MultiTrackSelector::forestLabel_, MultiTrackSelector::hSrc_, MultiTrackSelector::keepAllTracks_, MultiTrackSelector::max_d0_, MultiTrackSelector::max_d0NoPV_, MultiTrackSelector::max_eta_, MultiTrackSelector::max_lostHitFraction_, MultiTrackSelector::max_lostLayers_, MultiTrackSelector::max_minMissHitOutOrIn_, MultiTrackSelector::max_relpterr_, MultiTrackSelector::max_z0_, MultiTrackSelector::max_z0NoPV_, MultiTrackSelector::min_3Dlayers_, MultiTrackSelector::min_eta_, MultiTrackSelector::min_hits_bypass_, MultiTrackSelector::min_layers_, MultiTrackSelector::min_MVA_, MultiTrackSelector::min_nhits_, MultiTrackSelector::mvaType_, MultiTrackSelector::nSigmaZ_, MultiTrackSelector::preFilter_, MultiTrackSelector::qualityToSet_, MultiTrackSelector::res_par_, MultiTrackSelector::setQualityBit_, findQualityFiles::size, MultiTrackSelector::src_, srcTass_, srcTraj_, AlCaHLTBitMon_QueryRunRegistry::string, MultiTrackSelector::useAnyMVA_, MultiTrackSelector::useMVA_, MultiTrackSelector::useMVAonly_, MultiTrackSelector::useVertices_, MultiTrackSelector::useVtxError_, MultiTrackSelector::vertexCut_, MultiTrackSelector::vertices_, and MultiTrackSelector::vtxNumber_.

◆ ~AnalyticalTrackSelector()

AnalyticalTrackSelector::~AnalyticalTrackSelector ( )
override

destructor

Definition at line 232 of file AnalyticalTrackSelector.cc.

232 {}

Member Function Documentation

◆ run()

void AnalyticalTrackSelector::run ( edm::Event evt,
const edm::EventSetup es 
) const
overrideprivatevirtual

process one event

Reimplemented from MultiTrackSelector.

Definition at line 234 of file AnalyticalTrackSelector.cc.

234  {
235  // storage....
236  std::unique_ptr<reco::TrackCollection> selTracks_;
237  std::unique_ptr<reco::TrackExtraCollection> selTrackExtras_;
238  std::unique_ptr<TrackingRecHitCollection> selHits_;
239  std::unique_ptr<std::vector<Trajectory>> selTrajs_;
240  std::unique_ptr<std::vector<const Trajectory*>> selTrajPtrs_;
241  std::unique_ptr<TrajTrackAssociationCollection> selTTAss_;
242  reco::TrackRefProd rTracks_;
243  reco::TrackExtraRefProd rTrackExtras_;
244  TrackingRecHitRefProd rHits_;
245  edm::RefProd<std::vector<Trajectory>> rTrajectories_;
246  std::vector<reco::TrackRef> trackRefs_;
247 
248  using namespace std;
249  using namespace edm;
250  using namespace reco;
251 
252  Handle<TrackCollection> hSrcTrack;
256 
257  // looking for the beam spot
259  evt.getByToken(beamspot_, hBsp);
260  reco::BeamSpot vertexBeamSpot;
261  vertexBeamSpot = *hBsp;
262 
263  // Select good primary vertices for use in subsequent track selection
265  std::vector<Point> points;
266  std::vector<float> vterr, vzerr;
267  if (useVertices_) {
268  evt.getByToken(vertices_, hVtx);
269  selectVertices(0, *hVtx, points, vterr, vzerr);
270  // Debug
271  LogDebug("SelectVertex") << points.size() << " good pixel vertices";
272  }
273 
274  // Get tracks
275  evt.getByToken(src_, hSrcTrack);
276  // get hits in track..
278  evt.getByToken(hSrc_, hSrcHits);
279  const TrackingRecHitCollection& srcHits(*hSrcHits);
280 
281  selTracks_ = std::make_unique<TrackCollection>();
282  rTracks_ = evt.getRefBeforePut<TrackCollection>();
283  if (copyExtras_) {
284  selTrackExtras_ = std::make_unique<TrackExtraCollection>();
285  selHits_ = std::make_unique<TrackingRecHitCollection>();
287  rTrackExtras_ = evt.getRefBeforePut<TrackExtraCollection>();
288  }
289 
290  if (copyTrajectories_)
291  trackRefs_.resize(hSrcTrack->size());
292 
293  std::vector<float> mvaVals_(hSrcTrack->size(), -99.f);
294  processMVA(evt, es, vertexBeamSpot, *(hVtx.product()), 0, mvaVals_, true);
295 
296  // Loop over tracks
297  size_t current = 0;
298  for (TrackCollection::const_iterator it = hSrcTrack->begin(), ed = hSrcTrack->end(); it != ed; ++it, ++current) {
299  const Track& trk = *it;
300  // Check if this track passes cuts
301 
302  LogTrace("TrackSelection") << "ready to check track with pt=" << trk.pt();
303 
304  float mvaVal = 0;
305  if (useAnyMVA_)
306  mvaVal = mvaVals_[current];
307  bool ok = select(0, vertexBeamSpot, srcHits, trk, points, vterr, vzerr, mvaVal);
308  if (!ok) {
309  LogTrace("TrackSelection") << "track with pt=" << trk.pt() << " NOT selected";
310 
311  if (copyTrajectories_)
312  trackRefs_[current] = reco::TrackRef();
313  if (!keepAllTracks_[0])
314  continue;
315  }
316  LogTrace("TrackSelection") << "track with pt=" << trk.pt() << " selected";
317  selTracks_->push_back(Track(trk)); // clone and store
318  if (ok && setQualityBit_[0]) {
319  selTracks_->back().setQuality(qualityToSet_[0]);
320  if (qualityToSet_[0] == TrackBase::tight) {
321  selTracks_->back().setQuality(TrackBase::loose);
322  } else if (qualityToSet_[0] == TrackBase::highPurity) {
323  selTracks_->back().setQuality(TrackBase::loose);
324  selTracks_->back().setQuality(TrackBase::tight);
325  }
326  if (!points.empty()) {
327  if (qualityToSet_[0] == TrackBase::loose) {
328  selTracks_->back().setQuality(TrackBase::looseSetWithPV);
329  } else if (qualityToSet_[0] == TrackBase::highPurity) {
330  selTracks_->back().setQuality(TrackBase::looseSetWithPV);
331  selTracks_->back().setQuality(TrackBase::highPuritySetWithPV);
332  }
333  }
334  }
335  if (copyExtras_) {
336  // TrackExtras
337  selTrackExtras_->push_back(TrackExtra(trk.outerPosition(),
338  trk.outerMomentum(),
339  trk.outerOk(),
340  trk.innerPosition(),
341  trk.innerMomentum(),
342  trk.innerOk(),
343  trk.outerStateCovariance(),
344  trk.outerDetId(),
345  trk.innerStateCovariance(),
346  trk.innerDetId(),
347  trk.seedDirection(),
348  trk.seedRef()));
349  selTracks_->back().setExtra(TrackExtraRef(rTrackExtras_, selTrackExtras_->size() - 1));
350  TrackExtra& tx = selTrackExtras_->back();
351  tx.setResiduals(trk.residuals());
352  // TrackingRecHits
353  auto const firstHitIndex = selHits_->size();
354  for (trackingRecHit_iterator hit = trk.recHitsBegin(); hit != trk.recHitsEnd(); ++hit) {
355  selHits_->push_back((*hit)->clone());
356  }
357  tx.setHits(rHits_, firstHitIndex, selHits_->size() - firstHitIndex);
358  tx.setTrajParams(trk.extra()->trajParams(), trk.extra()->chi2sX5());
359  }
360  if (copyTrajectories_) {
361  trackRefs_[current] = TrackRef(rTracks_, selTracks_->size() - 1);
362  }
363  }
364  if (copyTrajectories_) {
367  evt.getByToken(srcTass_, hTTAss);
368  evt.getByToken(srcTraj_, hTraj);
369  selTrajs_ = std::make_unique<std::vector<Trajectory>>();
370  rTrajectories_ = evt.getRefBeforePut<vector<Trajectory>>();
371  selTTAss_ = std::make_unique<TrajTrackAssociationCollection>(rTrajectories_, rTracks_);
372  for (size_t i = 0, n = hTraj->size(); i < n; ++i) {
373  Ref<vector<Trajectory>> trajRef(hTraj, i);
375  if (match != hTTAss->end()) {
376  const Ref<TrackCollection>& trkRef = match->val;
377  short oldKey = static_cast<short>(trkRef.key());
378  if (trackRefs_[oldKey].isNonnull()) {
379  selTrajs_->push_back(Trajectory(*trajRef));
380  selTTAss_->insert(Ref<vector<Trajectory>>(rTrajectories_, selTrajs_->size() - 1), trackRefs_[oldKey]);
381  }
382  }
383  }
384  }
385 
386  evt.put(std::move(selTracks_));
387  if (copyExtras_) {
388  evt.put(std::move(selTrackExtras_));
389  evt.put(std::move(selHits_));
390  }
391  if (copyTrajectories_) {
392  evt.put(std::move(selTrajs_));
393  evt.put(std::move(selTTAss_));
394  }
395 }

References MultiTrackSelector::beamspot_, edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, copyExtras_, copyTrajectories_, edm::AssociationMap< Tag >::end(), reco::Track::extra(), edm::AssociationMap< Tag >::find(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), muons_cff::highPurity, MultiTrackSelector::hSrc_, mps_fire::i, reco::Track::innerDetId(), reco::Track::innerMomentum(), reco::Track::innerOk(), reco::Track::innerPosition(), reco::Track::innerStateCovariance(), MultiTrackSelector::keepAllTracks_, edm::Ref< C, T, F >::key(), LogDebug, LogTrace, match(), eostools::move(), dqmiodumpmetadata::n, convertSQLiteXML::ok, reco::Track::outerDetId(), reco::Track::outerMomentum(), reco::Track::outerOk(), reco::Track::outerPosition(), reco::Track::outerStateCovariance(), HLT_2018_cff::points, MultiTrackSelector::processMVA(), edm::Handle< T >::product(), reco::TrackBase::pt(), edm::Event::put(), MultiTrackSelector::qualityToSet_, reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), reco::Track::residuals(), reco::Track::seedDirection(), reco::Track::seedRef(), MultiTrackSelector::select(), MultiTrackSelector::selectVertices(), reco::TrackExtraBase::setHits(), MultiTrackSelector::setQualityBit_, reco::TrackExtra::setResiduals(), reco::TrackExtraBase::setTrajParams(), edm::OwnVector< T, P >::size(), MultiTrackSelector::src_, srcTass_, srcTraj_, MultiTrackSelector::useAnyMVA_, MultiTrackSelector::useVertices_, and MultiTrackSelector::vertices_.

Member Data Documentation

◆ copyExtras_

bool AnalyticalTrackSelector::copyExtras_
private

copy only the tracks, not extras and rechits (for AOD)

Definition at line 50 of file AnalyticalTrackSelector.cc.

Referenced by AnalyticalTrackSelector(), and run().

◆ copyTrajectories_

bool AnalyticalTrackSelector::copyTrajectories_
private

copy also trajectories and trajectory->track associations

Definition at line 52 of file AnalyticalTrackSelector.cc.

Referenced by AnalyticalTrackSelector(), and run().

◆ maxEta_

double AnalyticalTrackSelector::maxEta_
private

Definition at line 55 of file AnalyticalTrackSelector.cc.

◆ minEta_

double AnalyticalTrackSelector::minEta_
private

eta restrictions

Definition at line 54 of file AnalyticalTrackSelector.cc.

◆ srcTass_

edm::EDGetTokenT<TrajTrackAssociationCollection> AnalyticalTrackSelector::srcTass_
private

Definition at line 58 of file AnalyticalTrackSelector.cc.

Referenced by AnalyticalTrackSelector(), and run().

◆ srcTraj_

edm::EDGetTokenT<std::vector<Trajectory> > AnalyticalTrackSelector::srcTraj_
private

Definition at line 57 of file AnalyticalTrackSelector.cc.

Referenced by AnalyticalTrackSelector(), and run().

reco::Track::outerPosition
const math::XYZPoint & outerPosition() const
position of the outermost hit
Definition: Track.h:62
reco::Track::outerMomentum
const math::XYZVector & outerMomentum() const
momentum vector at the outermost hit position
Definition: Track.h:65
edm::RefProd< TrackCollection >
HLT_2018_cff.points
points
Definition: HLT_2018_cff.py:20125
MultiTrackSelector::d0_par1_
std::vector< std::vector< double > > d0_par1_
Definition: MultiTrackSelector.h:107
MultiTrackSelector::max_eta_
std::vector< double > max_eta_
Definition: MultiTrackSelector.h:133
reco::Track::outerStateCovariance
CovarianceMatrix outerStateCovariance() const
outermost trajectory state curvilinear errors
Definition: Track.h:68
edm::Event::getRefBeforePut
RefProd< PROD > getRefBeforePut()
Definition: Event.h:157
reco::Track::outerDetId
unsigned int outerDetId() const
DetId of the detector on which surface the outermost state is located.
Definition: Track.h:79
MultiTrackSelector::select
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
Definition: MultiTrackSelector.cc:330
edm::AssociationMap::find
const_iterator find(const key_type &k) const
find element with specified reference key
Definition: AssociationMap.h:173
mps_fire.i
i
Definition: mps_fire.py:355
dqmiodumpmetadata.n
n
Definition: dqmiodumpmetadata.py:28
MultiTrackSelector::max_relpterr_
std::vector< double > max_relpterr_
Definition: MultiTrackSelector.h:126
edm::Handle::product
T const * product() const
Definition: Handle.h:70
MultiTrackSelector::vtxNumber_
std::vector< int32_t > vtxNumber_
vertex cuts
Definition: MultiTrackSelector.h:99
MultiTrackSelector::min_nhits_
std::vector< uint32_t > min_nhits_
Definition: MultiTrackSelector.h:127
MultiTrackSelector::processMVA
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
Definition: MultiTrackSelector.cc:527
reco::Track::recHitsBegin
trackingRecHit_iterator recHitsBegin() const
Iterator to first hit on the track.
Definition: Track.h:88
MultiTrackSelector::chi2n_no1Dmod_par_
std::vector< double > chi2n_no1Dmod_par_
Definition: MultiTrackSelector.h:106
AnalyticalTrackSelector::copyTrajectories_
bool copyTrajectories_
copy also trajectories and trajectory->track associations
Definition: AnalyticalTrackSelector.cc:52
edm
HLT enums.
Definition: AlignableModifier.h:19
AnalyticalTrackSelector::srcTraj_
edm::EDGetTokenT< std::vector< Trajectory > > srcTraj_
Definition: AnalyticalTrackSelector.cc:57
AnalyticalTrackSelector::srcTass_
edm::EDGetTokenT< TrajTrackAssociationCollection > srcTass_
Definition: AnalyticalTrackSelector.cc:58
MultiTrackSelector::max_z0NoPV_
std::vector< double > max_z0NoPV_
Definition: MultiTrackSelector.h:137
MultiTrackSelector::forestLabel_
std::vector< std::string > forestLabel_
Definition: MultiTrackSelector.h:153
reco
fixed size matrix
Definition: AlignmentAlgorithmBase.h:45
MultiTrackSelector::min_hits_bypass_
std::vector< uint32_t > min_hits_bypass_
Definition: MultiTrackSelector.h:123
reco::Track::outerOk
bool outerOk() const
return true if the outermost hit is valid
Definition: Track.h:50
MultiTrackSelector::useAnyMVA_
bool useAnyMVA_
Definition: MultiTrackSelector.h:91
convertSQLiteXML.ok
bool ok
Definition: convertSQLiteXML.py:98
edm::Handle
Definition: AssociativeIterator.h:50
MultiTrackSelector::applyAbsCutsIfNoPV_
std::vector< bool > applyAbsCutsIfNoPV_
Definition: MultiTrackSelector.h:138
reco::Track::extra
const TrackExtraRef & extra() const
reference to "extra" object
Definition: Track.h:139
edm::Ref
Definition: AssociativeIterator.h:58
reco::Track::innerStateCovariance
CovarianceMatrix innerStateCovariance() const
innermost trajectory state curvilinear errors
Definition: Track.h:71
reco::Track::innerOk
bool innerOk() const
return true if the innermost hit is valid
Definition: Track.h:53
reco::TrackBase::pt
double pt() const
track transverse momentum
Definition: TrackBase.h:608
MultiTrackSelector::mvaType_
std::vector< std::string > mvaType_
Definition: MultiTrackSelector.h:152
reco::TrackExtra
Definition: TrackExtra.h:26
edm::AssociationMap::end
const_iterator end() const
last iterator over the map (read only)
Definition: AssociationMap.h:171
MultiTrackSelector::dz_par2_
std::vector< std::vector< double > > dz_par2_
Definition: MultiTrackSelector.h:110
reco::Track::innerMomentum
const math::XYZVector & innerMomentum() const
momentum vector at the innermost hit position
Definition: Track.h:59
MultiTrackSelector::max_minMissHitOutOrIn_
std::vector< int32_t > max_minMissHitOutOrIn_
Definition: MultiTrackSelector.h:129
MultiTrackSelector::useVertices_
bool useVertices_
Definition: MultiTrackSelector.h:89
MultiTrackSelector::chi2n_par_
std::vector< double > chi2n_par_
Definition: MultiTrackSelector.h:105
reco::Track::recHitsEnd
trackingRecHit_iterator recHitsEnd() const
Iterator to last hit on the track.
Definition: Track.h:91
reco::BeamSpot
Definition: BeamSpot.h:21
reco::Track
Definition: Track.h:27
MultiTrackSelector::selectVertices
void selectVertices(unsigned int tsNum, const reco::VertexCollection &vtxs, std::vector< Point > &points, std::vector< float > &vterr, std::vector< float > &vzerr) const
Definition: MultiTrackSelector.cc:502
MultiTrackSelector::vertices_
edm::EDGetTokenT< reco::VertexCollection > vertices_
Definition: MultiTrackSelector.h:92
reco::TrackExtraCollection
std::vector< TrackExtra > TrackExtraCollection
collection of TrackExtra objects
Definition: TrackExtraFwd.h:10
edm::Event::getByToken
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:528
MultiTrackSelector::src_
edm::EDGetTokenT< reco::TrackCollection > src_
source collection label
Definition: MultiTrackSelector.h:86
edm::OwnVector::const_iterator
Definition: OwnVector.h:41
reco::TrackRef
edm::Ref< TrackCollection > TrackRef
persistent reference to a Track
Definition: TrackFwd.h:20
AlCaHLTBitMon_QueryRunRegistry.string
string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:256
reco::TrackExtraBase::setTrajParams
void setTrajParams(TrajParams tmps, Chi2sFive chi2s)
Definition: TrackExtraBase.h:36
MultiTrackSelector::min_eta_
std::vector< double > min_eta_
Definition: MultiTrackSelector.h:132
reco::TrackExtra::setResiduals
void setResiduals(const TrackResiduals &r)
set the residuals
Definition: TrackExtra.h:132
MultiTrackSelector::d0_par2_
std::vector< std::vector< double > > d0_par2_
Definition: MultiTrackSelector.h:109
edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator
friend struct const_iterator
Definition: AssociationMap.h:274
MultiTrackSelector::max_d0NoPV_
std::vector< double > max_d0NoPV_
Definition: MultiTrackSelector.h:136
reco::Track::seedDirection
const PropagationDirection & seedDirection() const
direction of how the hits were sorted in the original seed
Definition: Track.h:148
LogDebug
#define LogDebug(id)
Definition: MessageLogger.h:670
edm::LogError
Definition: MessageLogger.h:183
match
std::pair< typename Association::data_type::first_type, double > match(Reference key, Association association, bool bestMatchByMaxValue)
Generic matching function.
Definition: Utils.h:10
reco::TrackExtraRef
edm::Ref< TrackExtraCollection > TrackExtraRef
persistent reference to a TrackExtra
Definition: TrackExtraFwd.h:16
MultiTrackSelector::useMVA_
std::vector< bool > useMVA_
Definition: MultiTrackSelector.h:147
MultiTrackSelector::dz_par1_
std::vector< std::vector< double > > dz_par1_
Definition: MultiTrackSelector.h:108
reco::Track::innerPosition
const math::XYZPoint & innerPosition() const
position of the innermost hit
Definition: Track.h:56
edm::Event::put
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:132
reco::Track::innerDetId
unsigned int innerDetId() const
DetId of the detector on which surface the innermost state is located.
Definition: Track.h:82
MultiTrackSelector::min_layers_
std::vector< uint32_t > min_layers_
Cuts on numbers of layers with hits/3D hits/lost hits.
Definition: MultiTrackSelector.h:120
MultiTrackSelector::applyAdaptedPVCuts_
std::vector< bool > applyAdaptedPVCuts_
Definition: MultiTrackSelector.h:112
MultiTrackSelector::MultiTrackSelector
MultiTrackSelector()
constructor
Definition: MultiTrackSelector.cc:17
reco::TrackExtraBase::setHits
void setHits(TrackingRecHitRefProd const &prod, unsigned firstH, unsigned int nH)
Definition: TrackExtraBase.h:30
looper.cfg
cfg
Definition: looper.py:297
MultiTrackSelector::keepAllTracks_
std::vector< bool > keepAllTracks_
Definition: MultiTrackSelector.h:140
MultiTrackSelector::useVtxError_
bool useVtxError_
Definition: MultiTrackSelector.h:90
MultiTrackSelector::preFilter_
std::vector< unsigned int > preFilter_
Definition: MultiTrackSelector.h:143
eostools.move
def move(src, dest)
Definition: eostools.py:511
std
Definition: JetResolutionObject.h:76
MultiTrackSelector::min_3Dlayers_
std::vector< uint32_t > min_3Dlayers_
Definition: MultiTrackSelector.h:121
reco::Track::seedRef
const edm::RefToBase< TrajectorySeed > & seedRef() const
Definition: Track.h:155
Trajectory
Definition: Trajectory.h:38
SiStripOfflineCRack_cfg.alias
alias
Definition: SiStripOfflineCRack_cfg.py:129
Exception
Definition: hltDiff.cc:246
MultiTrackSelector::qualityToSet_
std::vector< reco::TrackBase::TrackQuality > qualityToSet_
Definition: MultiTrackSelector.h:96
reco::Track::residuals
const TrackResiduals & residuals() const
get the residuals
Definition: Track.h:158
MultiTrackSelector::hSrc_
edm::EDGetTokenT< TrackingRecHitCollection > hSrc_
Definition: MultiTrackSelector.h:87
muons_cff.highPurity
highPurity
Definition: muons_cff.py:140
MultiTrackSelector::nSigmaZ_
std::vector< double > nSigmaZ_
Definition: MultiTrackSelector.h:117
MultiTrackSelector::max_lostLayers_
std::vector< uint32_t > max_lostLayers_
Definition: MultiTrackSelector.h:122
MultiTrackSelector::max_d0_
std::vector< double > max_d0_
Impact parameter absolute cuts.
Definition: MultiTrackSelector.h:115
edm::Ref::key
key_type key() const
Accessor for product key.
Definition: Ref.h:250
MultiTrackSelector::max_lostHitFraction_
std::vector< int32_t > max_lostHitFraction_
Definition: MultiTrackSelector.h:130
MultiTrackSelector::beamspot_
edm::EDGetTokenT< reco::BeamSpot > beamspot_
Definition: MultiTrackSelector.h:88
LogTrace
#define LogTrace(id)
Definition: MessageLogger.h:671
MultiTrackSelector::forest_
std::vector< GBRForest * > forest_
Definition: MultiTrackSelector.h:154
MultiTrackSelector::vertexCut_
std::vector< StringCutObjectSelector< reco::Vertex > > vertexCut_
Definition: MultiTrackSelector.h:101
edm::InputTag
Definition: InputTag.h:15
reco::TrackCollection
std::vector< Track > TrackCollection
collection of Tracks
Definition: TrackFwd.h:14
AnalyticalTrackSelector::copyExtras_
bool copyExtras_
copy only the tracks, not extras and rechits (for AOD)
Definition: AnalyticalTrackSelector.cc:50
hit
Definition: SiStripHitEffFromCalibTree.cc:88
edm::OwnVector< TrackingRecHit >
MultiTrackSelector::max_z0_
std::vector< double > max_z0_
Definition: MultiTrackSelector.h:116
findQualityFiles.size
size
Write out results.
Definition: findQualityFiles.py:443
MultiTrackSelector::useMVAonly_
std::vector< bool > useMVAonly_
Definition: MultiTrackSelector.h:148
MultiTrackSelector::setQualityBit_
std::vector< bool > setQualityBit_
do I have to set a quality bit?
Definition: MultiTrackSelector.h:95
MultiTrackSelector::min_MVA_
std::vector< double > min_MVA_
Definition: MultiTrackSelector.h:150
MultiTrackSelector::res_par_
std::vector< std::vector< double > > res_par_
Definition: MultiTrackSelector.h:104