Run the processor.
156 {
157
158
159 auto tg{geometry()};
160
161
162 if (!event.
exists(track_collection_, track_collection_event_passname_))
163 return;
164 auto tracks{
165 event.getCollection<
ldmx::Track>(track_collection_, track_passname_)};
166
167
168 if (!event.
exists(meas_collection_, meas_collection_event_passname_))
return;
169 auto measurements{
171
173
174
175 Acts::GainMatrixUpdater updater;
176 Acts::GsfExtensions<Acts::VectorMultiTrajectory> gsf_extensions;
177 gsf_extensions.updater.connect<
178 &Acts::GainMatrixUpdater::operator()<Acts::VectorMultiTrajectory>>(
179 &updater);
180 gsf_extensions.calibrator
182 Acts::VectorMultiTrajectory>>(&calibrator);
183
184
185 struct SurfaceAccessor {
186 const Acts::TrackingGeometry* tracking_geometry_;
187
188 const Acts::Surface* operator()(const Acts::SourceLink& sourceLink) const {
189 const auto& index_source_link =
191 return tracking_geometry_->findSurface(index_source_link.geometryId());
192 }
193 };
194
195 SurfaceAccessor m_sl_surface_accessor{tg.getTG().get()};
196
197 gsf_extensions.surfaceAccessor.connect<&SurfaceAccessor::operator()>(
198 &m_sl_surface_accessor);
199 gsf_extensions.mixtureReducer.connect<&Acts::reduceMixtureLargestWeights>();
200
201
202
203
204 Acts::PropagatorOptions<Acts::StepperPlainOptions,
205 Acts::NavigatorPlainOptions, ActionList, AbortList>
206 propagator_options(geometryContext(), magneticFieldContext());
207
208 propagator_options.pathLimit = std::numeric_limits<double>::max();
209
210
211 propagator_options.loopProtection =
212 false;
213
214
215 auto& m_interactor =
216 propagator_options.actionList.get<Acts::MaterialInteractor>();
217 m_interactor.multipleScattering = true;
218 m_interactor.energyLoss = true;
219 m_interactor.recordInteractions = false;
220
221
222 auto& s_logger =
223 propagator_options.actionList.get<Acts::detail::SteppingLogger>();
224 s_logger.sterile = true;
225
226 propagator_options.stepping.maxStepSize =
227 propagator_step_size_ * Acts::UnitConstants::mm;
228 propagator_options.maxSteps = propagator_max_steps_;
229
230
231
232
233
234
235 std::shared_ptr<const Acts::PerigeeSurface> origin_surface =
236 Acts::Surface::makeShared<Acts::PerigeeSurface>(
237 Acts::Vector3(0., 0., 0.));
238
239 std::shared_ptr<const Acts::PerigeeSurface> tagger_layer_surface =
240 Acts::Surface::makeShared<Acts::PerigeeSurface>(
241 Acts::Vector3(-700., 0., 0.));
242
243 std::shared_ptr<const Acts::PerigeeSurface> reference_surface =
244 origin_surface;
245 if (tagger_tracking_) {
246 reference_surface = tagger_layer_surface;
247 }
248
249
250
251
252
253
254
255
256
257 Acts::GsfOptions<Acts::VectorMultiTrajectory> gsf_options{
258 geometryContext(), magneticFieldContext(), calibrationContext()};
259
260 gsf_options.extensions = gsf_extensions;
261 gsf_options.propagatorPlainOptions = propagator_options;
262 gsf_options.referenceSurface = &(*reference_surface);
263 gsf_options.maxComponents = max_components_;
264 gsf_options.weightCutoff = weight_cutoff_;
265 gsf_options.abortOnError = abort_on_error_;
266 gsf_options.disableAllMaterialHandling = disable_all_material_handling_;
267
268
269 std::vector<ldmx::Track> out_tracks;
270
271
272 Acts::VectorTrackContainer vtc;
273 Acts::VectorMultiTrajectory mtj;
274 Acts::TrackContainer tc{vtc, mtj};
275
276
277 unsigned int itrk = 0;
278
279 for (auto& track : tracks) {
280
281 std::vector<ldmx::Measurement> meas_on_track;
282
283
284 std::vector<Acts::SourceLink> fit_track_source_links;
285
286 for (auto imeas : track.getMeasurementsIdxs()) {
287 auto meas = measurements.at(imeas);
288 meas_on_track.push_back(meas);
289
290
291
292 const Acts::Surface* hit_surface =
293 tg.geo::TrackingGeometry::getSurface(meas.getLayerID());
294
295
297 fit_track_source_links.push_back(Acts::SourceLink(idx_sl));
298 }
299
300
301 std::reverse(meas_on_track.begin(), meas_on_track.end());
302 std::reverse(fit_track_source_links.begin(), fit_track_source_links.end());
303
304 for (auto m : meas_on_track) {
305 ldmx_log(debug) << "Measurement:\n" << m << "\n";
306 }
307
308 ldmx_log(debug) << "GSF Refitting";
309
310
311
312 std::shared_ptr<Acts::PerigeeSurface> perigee =
313 Acts::Surface::makeShared<Acts::PerigeeSurface>(Acts::Vector3(
314 track.getPerigeeX(), track.getPerigeeY(), track.getPerigeeZ()));
315
316 Acts::BoundTrackParameters trk_btp =
317 tracking::sim::utils::boundTrackParameters(track, perigee);
318
319 std::shared_ptr<Acts::Surface> beam_origin_surface =
320 tracking::sim::utils::unboundSurface(-700);
321
322 const std::shared_ptr<Acts::Surface> target_surface =
323 tracking::sim::utils::unboundSurface(0.);
324
325 const std::shared_ptr<Acts::Surface> ecal_surface =
326 tracking::sim::utils::unboundSurface(240.5);
327
328 Acts::BoundTrackParameters trk_btp_b_o =
329 tracking::sim::utils::boundTrackParameters(track, perigee);
330
331 if (tagger_tracking_) {
332 if (!track.getTrackState(ldmx::TrackStateType::AtBeamOrigin)
333 .has_value()) {
334 ldmx_log(warn) << "Failed retreiving AtBeamOrigin TrackState for "
335 "track. Skipping..";
336 continue;
337 }
338
339 auto ts = track.getTrackState(ldmx::TrackStateType::AtBeamOrigin).value();
340 trk_btp_b_o = tracking::sim::utils::btp(
341 ts, beam_origin_surface,
342 11);
343 } else {
344 if (!track.getTrackState(ldmx::TrackStateType::AtTarget).has_value()) {
345 ldmx_log(warn)
346 << "Failed retreiving AtTarget TrackState for track. Skipping..";
347 continue;
348 }
349 auto ts = track.getTrackState(ldmx::TrackStateType::AtTarget).value();
350 trk_btp_b_o = tracking::sim::utils::btp(ts, target_surface, 11);
351 }
352 const Acts::BoundVector& trkpars = trk_btp.parameters();
353 ldmx_log(debug) << "CKF Track parameters" << std::endl
354 << trkpars[0] << " " << trkpars[1] << " " << trkpars[2]
355 << " " << trkpars[3] << " " << trkpars[4] << " "
356 << trkpars[5] << std::endl
357 << "Perigee Surface" << std::endl
358 << track.getPerigeeX() << " " << track.getPerigeeY() << " "
359 << track.getPerigeeZ();
360
361 Acts::Vector3 trk_pos = trk_btp.position(geometryContext());
362
363 ldmx_log(debug) << trk_pos(0) << " " << trk_pos(1) << " " << trk_pos(2)
364 << std::endl;
365
366 const Acts::BoundVector& trkpars_b_o = trk_btp_b_o.parameters();
367 ldmx_log(debug) << "CKF BeamOrigin track parameters" << std::endl
368 << trkpars_b_o[0] << " " << trkpars_b_o[1] << " "
369 << trkpars_b_o[2] << " " << trkpars_b_o[3] << " "
370 << trkpars_b_o[4] << " " << trkpars_b_o[5] << " ";
371
372 Acts::Vector3 trk_pos_b_o = trk_btp_b_o.position(geometryContext());
373 ldmx_log(debug) << trk_pos_b_o(0) << " " << trk_pos_b_o(1) << " "
374 << trk_pos_b_o(2) << std::endl;
375
376 auto gsf_refit_result =
377 gsf_->fit(fit_track_source_links.begin(), fit_track_source_links.end(),
378 trk_btp_b_o, gsf_options, tc);
379
380 if (!gsf_refit_result.ok()) {
381 ldmx_log(warn) << "GSF re-fit failed" << std::endl;
382 continue;
383 }
384
385 if (tc.size() < 1) continue;
386
387 auto gsftrk = tc.getTrack(itrk);
388 calculateTrackQuantities(gsftrk);
389
390 const Acts::BoundVector& perigee_pars = gsftrk.parameters();
391 const Acts::BoundMatrix& trk_cov = gsftrk.covariance();
392 const Acts::Surface& perigee_surface = gsftrk.referenceSurface();
393
394 ldmx_log(debug)
395 << "Found track:" << std::endl
396 << "Track states " << gsftrk.nTrackStates() << std::endl
397 << perigee_pars[Acts::eBoundLoc0] << " "
398 << perigee_pars[Acts::eBoundLoc1] << " "
399 << perigee_pars[Acts::eBoundPhi] << " "
400 << perigee_pars[Acts::eBoundTheta] << " "
401 << perigee_pars[Acts::eBoundQOverP] << std::endl
402 << "Reference Surface" << std::endl
403 << " " << perigee_surface.transform(geometryContext()).translation()(0)
404 << " " << perigee_surface.transform(geometryContext()).translation()(1)
405 << " " << perigee_surface.transform(geometryContext()).translation()(2)
406 << std::endl;
407
409
410 bool success = false;
411 if (tagger_tracking_) {
412 ldmx_log(debug) << "Target extrapolation";
414
415 success = trk_extrap_->trackStateAtSurface(
416 gsftrk, target_surface, ts_at_target, ldmx::TrackStateType::AtTarget);
417
418 if (success) trk.addTrackState(ts_at_target);
419 } else {
420 ldmx_log(debug) << "Ecal Extrapolation";
422 success = trk_extrap_->trackStateAtSurface(
423 gsftrk, ecal_surface, ts_at_ecal, ldmx::TrackStateType::AtECAL);
424
425 if (success) trk.addTrackState(ts_at_ecal);
426 }
427
428 trk.setPerigeeLocation(
429 perigee_surface.transform(geometryContext()).translation()(0),
430 perigee_surface.transform(geometryContext()).translation()(1),
431 perigee_surface.transform(geometryContext()).translation()(2));
432
433 trk.setChi2(gsftrk.chi2());
434 trk.setNhits(gsftrk.nMeasurements());
435 trk.setNdf(gsftrk.nMeasurements() - 5);
437 tracking::sim::utils::convertActsToLdmxPars(perigee_pars));
438 std::vector<double> v_trk_cov;
439 tracking::sim::utils::flatCov(trk_cov, v_trk_cov);
440 trk.setPerigeeCov(v_trk_cov);
441 Acts::Vector3 trk_momentum = gsftrk.momentum();
442 trk.setMomentum(trk_momentum(0), trk_momentum(1), trk_momentum(2));
443
444
445 trk.setTrackID(track.getTrackID());
446 trk.setPdgID(track.getPdgID());
447 trk.setTruthProb(track.getTruthProb());
448
449 itrk++;
450
451 out_tracks.push_back(trk);
452
453 }
454
455 event.add(out_trk_collection_, out_tracks);
456}
A source link that stores just an index_.
bool exists(const std::string &name, const std::string &passName, bool unique=true) const
Check for the existence of an object or collection with the given name and pass name in the event.
Implementation of a track object.
void setPerigeeParameters(const std::vector< double > &par)
d_0 z_0 phi_0 theta q/p t
void calibrate1d(const Acts::GeometryContext &, const Acts::CalibrationContext &, const Acts::SourceLink &genericSourceLink, typename traj_t::TrackStateProxy trackState) const
Find the measurement corresponding to the source link.