CMS 3D CMS Logo

List of all members | Public Types | Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes
mkfit::MkBuilder Class Reference

#include <MkBuilder.h>

Public Types

typedef std::vector< std::pair< int, int > > CandIdx_t
 
using insert_seed_foo = void(const Track &, int, int, int)
 

Public Member Functions

void backwardFit ()
 
void backwardFitBH ()
 
void begin_event (MkJob *job, Event *ev, const char *build_type)
 
void beginBkwSearch ()
 
void compactifyHitStorageForBestCand (bool remove_seed_hits, int backward_fit_min_hits)
 
void end_event ()
 
void endBkwSearch ()
 
void export_best_comb_cands (TrackVec &out_vec, bool remove_missing_hits=false)
 
void export_tracks (TrackVec &out_vec)
 
int filter_comb_cands (filter_candidates_func filter, bool attempt_all_cands)
 
void find_min_max_hots_size ()
 
void find_tracks_handle_missed_layers (MkFinder *mkfndr, const LayerInfo &layer_info, std::vector< std::vector< TrackCand >> &tmp_cands, const std::vector< std::pair< int, int >> &seed_cand_idx, const int region, const int start_seed, const int itrack, const int end)
 
void find_tracks_in_layers (CandCloner &cloner, MkFinder *mkfndr, SteeringParams::IterationType_e iteration_dir, const int start_seed, const int end_seed, const int region)
 
void find_tracks_load_seeds (const TrackVec &in_seeds, const bool seeds_sorted)
 
void find_tracks_load_seeds_BH (const TrackVec &in_seeds, const bool seeds_sorted)
 
int find_tracks_unroll_candidates (std::vector< std::pair< int, int >> &seed_cand_vec, int start_seed, int end_seed, int layer, int prev_layer, bool pickup_only, SteeringParams::IterationType_e iteration_dir)
 
void findTracksBestHit (SteeringParams::IterationType_e iteration_dir=SteeringParams::IT_FwdSearch)
 
void findTracksCloneEngine (SteeringParams::IterationType_e iteration_dir=SteeringParams::IT_FwdSearch)
 
void findTracksStandard (SteeringParams::IterationType_e iteration_dir=SteeringParams::IT_FwdSearch)
 
void fit_cands (MkFinder *mkfndr, int start_cand, int end_cand, int region)
 
void fit_cands_BH (MkFinder *mkfndr, int start_cand, int end_cand, int region)
 
void import_seeds (const TrackVec &in_seeds, const bool seeds_sorted, std::function< insert_seed_foo > insert_seed)
 
std::pair< int, int > max_hits_layer (const EventOfHits &eoh) const
 
 MkBuilder (bool silent=true)
 
const EventOfCombCandidatesref_eocc () const
 
const TrackVecref_tracks () const
 
TrackVecref_tracks_nc ()
 
void release_memory ()
 
void seed_post_cleaning (TrackVec &tv)
 
void select_best_comb_cands (bool clear_m_tracks=false, bool remove_missing_hits=false)
 
int total_cands () const
 
 ~MkBuilder ()=default
 

Static Public Member Functions

static std::unique_ptr< MkBuildermake_builder (bool silent=true)
 
static void populate ()
 

Private Member Functions

void fit_one_seed_set (TrackVec &simtracks, int itrack, int end, MkFitter *mkfttr, const bool is_brl[])
 

Private Attributes

Eventm_event = nullptr
 
EventOfCombCandidates m_event_of_comb_cands
 
MkJobm_job = nullptr
 
std::atomic< int > m_nan_n_silly_per_layer_count
 
std::vector< int > m_seedEtaSeparators
 
std::vector< int > m_seedMaxLastLayer
 
std::vector< int > m_seedMinLastLayer
 
bool m_silent
 
TrackVec m_tracks
 

Detailed Description

Definition at line 28 of file MkBuilder.h.

Member Typedef Documentation

◆ CandIdx_t

typedef std::vector<std::pair<int, int> > mkfit::MkBuilder::CandIdx_t

Definition at line 32 of file MkBuilder.h.

◆ insert_seed_foo

using mkfit::MkBuilder::insert_seed_foo = void(const Track &, int, int, int)

Definition at line 30 of file MkBuilder.h.

Constructor & Destructor Documentation

◆ MkBuilder()

mkfit::MkBuilder::MkBuilder ( bool  silent = true)
inline

Definition at line 34 of file MkBuilder.h.

◆ ~MkBuilder()

mkfit::MkBuilder::~MkBuilder ( )
default

Member Function Documentation

◆ backwardFit()

void mkfit::MkBuilder::backwardFit ( )

Definition at line 1270 of file MkBuilder.cc.

References HLT_2023v12_cff::cands, dprint, fit_cands(), mkfit::g_exe_ctx, m_event_of_comb_cands, mkfit::ExecutionContext::m_finders, m_job, m_seedEtaSeparators, mkfit::Config::numSeedsPerTask, mkfit::Config::numThreadsEvents, mkfit::Config::numThreadsFinder, nano_mu_digi_cff::region, mkfit::MkJob::regions_begin(), mkfit::MkJob::regions_end(), and mkfit::EventOfCombCandidates::size().

Referenced by mkfit::Shell::ProcessEvent(), mkfit::run_OneIteration(), and mkfit::runBtpCe_MultiIter().

1270  {
1272 
1273  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
1274  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
1275 
1276  // adaptive seeds per task based on the total estimated amount of work to divide among all threads
1277  const int adaptiveSPT = std::clamp(
1279  dprint("adaptiveSPT " << adaptiveSPT << " fill " << rosi.count() << "/" << eoccs.size() << " region " << region);
1280 
1281  tbb::parallel_for(rosi.tbb_blk_rng_std(adaptiveSPT), [&](const tbb::blocked_range<int> &cands) {
1282  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
1283 
1284  fit_cands(mkfndr.get(), cands.begin(), cands.end(), region);
1285  });
1286  });
1287  }
const auto regions_begin() const
Definition: MkJob.h:22
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
ExecutionContext g_exe_ctx
Definition: MkBuilder.cc:49
Pool< MkFinder > m_finders
Definition: MkBuilder.cc:40
constexpr int numThreadsFinder
Definition: Config.h:85
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
#define dprint(x)
Definition: Debug.h:95
constexpr int numSeedsPerTask
Definition: Config.h:87
constexpr int numThreadsEvents
Definition: Config.h:86
const auto regions_end() const
Definition: MkJob.h:23
void fit_cands(MkFinder *mkfndr, int start_cand, int end_cand, int region)
Definition: MkBuilder.cc:1289

◆ backwardFitBH()

void mkfit::MkBuilder::backwardFitBH ( )

Definition at line 1174 of file MkBuilder.cc.

References fit_cands_BH(), mkfit::g_exe_ctx, mkfit::ExecutionContext::m_finders, m_job, m_seedEtaSeparators, nano_mu_digi_cff::region, mkfit::MkJob::regions_begin(), and mkfit::MkJob::regions_end().

Referenced by mkfit::runBuildingTestPlexBestHit(), mkfit::runBuildingTestPlexCloneEngine(), and mkfit::runBuildingTestPlexStandard().

1174  {
1175  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
1176  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
1177 
1178  tbb::parallel_for(rosi.tbb_blk_rng_vec(), [&](const tbb::blocked_range<int> &blk_rng) {
1179  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
1180 
1181  RangeOfSeedIndices rng = rosi.seed_rng(blk_rng);
1182 
1183  while (rng.valid()) {
1184  // final backward fit
1185  fit_cands_BH(mkfndr.get(), rng.m_beg, rng.m_end, region);
1186 
1187  ++rng;
1188  }
1189  });
1190  });
1191  }
const auto regions_begin() const
Definition: MkJob.h:22
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
ExecutionContext g_exe_ctx
Definition: MkBuilder.cc:49
Pool< MkFinder > m_finders
Definition: MkBuilder.cc:40
const auto regions_end() const
Definition: MkJob.h:23
void fit_cands_BH(MkFinder *mkfndr, int start_cand, int end_cand, int region)
Definition: MkBuilder.cc:1193

◆ begin_event()

void mkfit::MkBuilder::begin_event ( MkJob job,
Event ev,
const char *  build_type 
)

Definition at line 194 of file MkBuilder.cc.

References gather_cfg::cout, makeMEIFBenchmarkPlots::ev, mps_fire::i, m_event, mkfit::MkJob::m_iter_config, mkfit::IterationConfig::m_iteration_index, m_job, m_nan_n_silly_per_layer_count, m_seedEtaSeparators, m_seedMaxLastLayer, m_seedMinLastLayer, m_silent, mkfit::IterationConfig::m_track_algorithm, and mkfit::MkJob::num_regions().

Referenced by mkfit::Shell::ProcessEvent(), mkfit::run_OneIteration(), mkfit::runBtpCe_MultiIter(), mkfit::runBuildingTestPlexBestHit(), mkfit::runBuildingTestPlexCloneEngine(), mkfit::runBuildingTestPlexDumbCMSSW(), and mkfit::runBuildingTestPlexStandard().

194  {
196 
197  m_job = job;
198  m_event = ev;
199 
203 
204  for (int i = 0; i < m_job->num_regions(); ++i) {
205  m_seedEtaSeparators[i] = 0;
206  m_seedMinLastLayer[i] = 9999;
207  m_seedMaxLastLayer[i] = 0;
208  }
209 
210  if (!m_silent) {
211  std::cout << "MkBuilder building tracks with '" << build_type << "'"
212  << ", iteration_index=" << job->m_iter_config.m_iteration_index
213  << ", track_algorithm=" << job->m_iter_config.m_track_algorithm << std::endl;
214  }
215  }
const IterationConfig & m_iter_config
Definition: MkJob.h:12
std::atomic< int > m_nan_n_silly_per_layer_count
Definition: MkBuilder.h:137
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
Event * m_event
Definition: MkBuilder.h:124
std::vector< int > m_seedMaxLastLayer
Definition: MkBuilder.h:135
std::vector< int > m_seedMinLastLayer
Definition: MkBuilder.h:134
int num_regions() const
Definition: MkJob.h:21

◆ beginBkwSearch()

void mkfit::MkBuilder::beginBkwSearch ( )
inline

◆ compactifyHitStorageForBestCand()

void mkfit::MkBuilder::compactifyHitStorageForBestCand ( bool  remove_seed_hits,
int  backward_fit_min_hits 
)
inline

Definition at line 62 of file MkBuilder.h.

References mkfit::EventOfCombCandidates::compactifyHitStorageForBestCand(), and m_event_of_comb_cands.

Referenced by mkfit::Shell::ProcessEvent(), mkfit::run_OneIteration(), and mkfit::runBtpCe_MultiIter().

62  {
63  m_event_of_comb_cands.compactifyHitStorageForBestCand(remove_seed_hits, backward_fit_min_hits);
64  }
void compactifyHitStorageForBestCand(bool remove_seed_hits, int backward_fit_min_hits)
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130

◆ end_event()

void mkfit::MkBuilder::end_event ( )

◆ endBkwSearch()

void mkfit::MkBuilder::endBkwSearch ( )
inline

◆ export_best_comb_cands()

void mkfit::MkBuilder::export_best_comb_cands ( TrackVec out_vec,
bool  remove_missing_hits = false 
)

Definition at line 384 of file MkBuilder.cc.

References relativeConstraints::empty, mkfit::TrackCand::exportTrack(), mps_fire::i, m_event_of_comb_cands, and mkfit::EventOfCombCandidates::size().

Referenced by mkfit::Shell::ProcessEvent(), mkfit::run_OneIteration(), mkfit::runBuildingTestPlexCloneEngine(), mkfit::runBuildingTestPlexStandard(), and select_best_comb_cands().

384  {
386  out_vec.reserve(out_vec.size() + eoccs.size());
387  for (int i = 0; i < eoccs.size(); i++) {
388  // Take the first candidate, if it exists.
389  if (!eoccs[i].empty()) {
390  const TrackCand &bcand = eoccs[i].front();
391  out_vec.emplace_back(bcand.exportTrack(remove_missing_hits));
392  }
393  }
394  }
Track exportTrack(bool remove_missing_hits=false) const
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130

◆ export_tracks()

void mkfit::MkBuilder::export_tracks ( TrackVec out_vec)

Definition at line 396 of file MkBuilder.cc.

References m_tracks, and submitPVValidationJobs::t.

Referenced by mkfit::runBtpCe_MultiIter().

396  {
397  out_vec.reserve(out_vec.size() + m_tracks.size());
398  for (auto &t : m_tracks) {
399  out_vec.emplace_back(t);
400  }
401  }
TrackVec m_tracks
Definition: MkBuilder.h:127

◆ filter_comb_cands()

int mkfit::MkBuilder::filter_comb_cands ( filter_candidates_func  filter,
bool  attempt_all_cands 
)

Definition at line 301 of file MkBuilder.cc.

References cms::cuda::assert(), mkfit::EventOfCombCandidates::cands_in_backward_rep(), dprintf, ALCARECOTkAlBeamHalo_cff::filter, mps_fire::i, createfilelist::int, dqmiolumiharvest::j, m_event_of_comb_cands, m_job, m_seedEtaSeparators, mkfit::MkJob::num_regions(), TriggerAnalyzer::passed, mkfit::EventOfCombCandidates::resizeAfterFiltering(), mkfit::EventOfCombCandidates::size(), and edm::swap().

Referenced by mkfit::Shell::ProcessEvent(), mkfit::run_OneIteration(), and mkfit::runBtpCe_MultiIter().

301  {
303  int i = 0, place_pos = 0;
304 
305  dprintf("MkBuilder::filter_comb_cands Entering filter size eoccs.size=%d\n", eoccs.size());
306 
307  std::vector<int> removed_cnts(m_job->num_regions());
308  while (i < eoccs.size()) {
309  if (eoccs.cands_in_backward_rep())
310  eoccs[i].repackCandPostBkwSearch(0);
311  bool passed = filter(eoccs[i].front(), *m_job);
312 
313  if (!passed && attempt_all_cands) {
314  for (int j = 1; j < (int)eoccs[i].size(); ++j) {
315  if (eoccs.cands_in_backward_rep())
316  eoccs[i].repackCandPostBkwSearch(j);
317  if (filter(eoccs[i][j], *m_job)) {
318  eoccs[i][0] = eoccs[i][j]; // overwrite front, no need to std::swap() them
319  passed = true;
320  break;
321  }
322  }
323  }
324  if (passed) {
325  if (place_pos != i)
326  std::swap(eoccs[place_pos], eoccs[i]);
327  ++place_pos;
328  } else {
329  assert(eoccs[i].front().getEtaRegion() < m_job->num_regions());
330  ++removed_cnts[eoccs[i].front().getEtaRegion()];
331  }
332  ++i;
333  }
334 
335  int n_removed = 0;
336  for (int reg = 0; reg < m_job->num_regions(); ++reg) {
337  dprintf("MkBuilder::filter_comb_cands reg=%d: n_rem_was=%d removed_in_r=%d n_rem=%d, es_was=%d es_new=%d\n",
338  reg,
339  n_removed,
340  removed_cnts[reg],
341  n_removed + removed_cnts[reg],
342  m_seedEtaSeparators[reg],
343  m_seedEtaSeparators[reg] - n_removed - removed_cnts[reg]);
344 
345  n_removed += removed_cnts[reg];
346  m_seedEtaSeparators[reg] -= n_removed;
347  }
348 
349  eoccs.resizeAfterFiltering(n_removed);
350 
351  dprintf("MkBuilder::filter_comb_cands n_removed = %d, eoccs.size=%d\n", n_removed, eoccs.size());
352 
353  return n_removed;
354  }
size
Write out results.
void resizeAfterFiltering(int n_removed)
assert(be >=bs)
void swap(Association< C > &lhs, Association< C > &rhs)
Definition: Association.h:112
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
int num_regions() const
Definition: MkJob.h:21
#define dprintf(...)
Definition: Debug.h:98

◆ find_min_max_hots_size()

void mkfit::MkBuilder::find_min_max_hots_size ( )

Definition at line 356 of file MkBuilder.cc.

References mps_fire::i, m_event_of_comb_cands, m_seedEtaSeparators, SiStripPI::max, and SiStripPI::min.

356  {
358  int min[5], max[5], gmin = 0, gmax = 0;
359  int i = 0;
360  for (int reg = 0; reg < 5; ++reg) {
361  min[reg] = 9999;
362  max[reg] = 0;
363  for (; i < m_seedEtaSeparators[reg]; i++) {
364  min[reg] = std::min(min[reg], eoccs[i].hotsSize());
365  max[reg] = std::max(max[reg], eoccs[i].hotsSize());
366  }
367  gmin = std::max(gmin, min[reg]);
368  gmax = std::max(gmax, max[reg]);
369  }
370  // clang-format off
371  printf("MkBuilder::find_min_max_hots_size MIN %3d -- [ %3d | %3d | %3d | %3d | %3d ] "
372  "MAX %3d -- [ %3d | %3d | %3d | %3d | %3d ]\n",
373  gmin, min[0], min[1], min[2], min[3], min[4],
374  gmax, max[0], max[1], max[2], max[3], max[4]);
375  // clang-format on
376  }
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130

◆ find_tracks_handle_missed_layers()

void mkfit::MkBuilder::find_tracks_handle_missed_layers ( MkFinder mkfndr,
const LayerInfo layer_info,
std::vector< std::vector< TrackCand >> &  tmp_cands,
const std::vector< std::pair< int, int >> &  seed_cand_idx,
const int  region,
const int  start_seed,
const int  itrack,
const int  end 
)

Definition at line 686 of file MkBuilder.cc.

References dprintf, mkfit::MkBase::iP, mkfit::LayerInfo::is_barrel(), mkfit::LayerInfo::layer_id(), m_event_of_comb_cands, mkfit::MkFinder::m_XWsrResult, mkfit::MkBase::radius(), mkfit::TrackerInfo::Reg_Barrel, nano_mu_digi_cff::region, mkfit::LayerInfo::rin(), mkfit::LayerInfo::rout(), w(), mkfit::WSR_Edge, mkfit::WSR_Failed, and mkfit::WSR_Outside.

Referenced by find_tracks_in_layers(), and findTracksStandard().

693  {
694  // XXXX-1 If I miss a layer, insert the original track into tmp_cands
695  // AND do not do it in FindCandidates as the position can be badly
696  // screwed by then. See comment there, too.
697  // One could also do a pre-check ... so as not to use up a slot.
698 
699  // bool debug = true;
700 
701  for (int ti = itrack; ti < end; ++ti) {
702  TrackCand &cand = m_event_of_comb_cands[seed_cand_idx[ti].first][seed_cand_idx[ti].second];
703  WSR_Result &w = mkfndr->m_XWsrResult[ti - itrack];
704 
705  // Low pT tracks can miss a barrel layer ... and should be stopped
706  dprintf("WSR Check label %d, seed %d, cand %d score %f -> wsr %d, in_gap %d\n",
707  cand.label(),
708  seed_cand_idx[ti].first,
709  seed_cand_idx[ti].second,
710  cand.score(),
711  w.m_wsr,
712  w.m_in_gap);
713 
714  if (w.m_wsr == WSR_Failed) {
715  // Fake outside so it does not get processed in FindTracks BH/Std/CE.
716  // [ Should add handling of WSR_Failed there, perhaps. ]
717  w.m_wsr = WSR_Outside;
718 
719  if (layer_info.is_barrel()) {
720  dprintf("Barrel cand propagation failed, got to r=%f ... layer is %f - %f\n",
721  mkfndr->radius(ti - itrack, MkBase::iP),
722  layer_info.rin(),
723  layer_info.rout());
724  // In barrel region, create a stopped replica. In transition region keep the original copy
725  // as there is still a chance to hit endcaps.
726  tmp_cands[seed_cand_idx[ti].first - start_seed].push_back(cand);
728  dprintf(" creating extra stopped held back candidate\n");
729  tmp_cands[seed_cand_idx[ti].first - start_seed].back().addHitIdx(-2, layer_info.layer_id(), 0);
730  }
731  }
732  // Never happens for endcap / propToZ
733  } else if (w.m_wsr == WSR_Outside) {
734  dprintf(" creating extra held back candidate\n");
735  tmp_cands[seed_cand_idx[ti].first - start_seed].push_back(cand);
736  } else if (w.m_wsr == WSR_Edge) {
737  // Do nothing special here, this case is handled also in MkFinder:findTracks()
738  }
739  }
740  }
float radius(int itrack, int i) const
Definition: MkBase.h:24
static constexpr int iP
Definition: MkBase.h:19
float rin() const
Definition: TrackerInfo.h:66
T w() const
int layer_id() const
Definition: TrackerInfo.h:64
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
float rout() const
Definition: TrackerInfo.h:67
WSR_Result m_XWsrResult[NN]
Definition: MkFinder.h:315
#define dprintf(...)
Definition: Debug.h:98
bool is_barrel() const
Definition: TrackerInfo.h:76

◆ find_tracks_in_layers()

void mkfit::MkBuilder::find_tracks_in_layers ( CandCloner cloner,
MkFinder mkfndr,
SteeringParams::IterationType_e  iteration_dir,
const int  start_seed,
const int  end_seed,
const int  region 
)

Definition at line 967 of file MkBuilder.cc.

References cms::cuda::assert(), mkfit::CandCloner::begin_eta_bin(), mkfit::CandCloner::begin_iteration(), mkfit::CandCloner::begin_layer(), mkfit::MkFinder::begin_layer(), gpuPixelDoublets::cc, mkfit::MkBase::clearFailFlag(), mkfit::MkFinder::copyOutParErr(), dprint, dprintf, mkfit::CandCloner::end_eta_bin(), mkfit::CandCloner::end_iteration(), mkfit::CandCloner::end_layer(), mkfit::MkFinder::end_layer(), find_tracks_handle_missed_layers(), find_tracks_unroll_candidates(), mkfit::MkFinder::findCandidatesCloneEngine(), mkfit::PropagationConfig::finding_inter_layer_pflags, dqmdumpme::first, mkfit::FindingFoos::get_finding_foos(), mkfit::MkJob::get_mask_for_layer(), mps_fire::i, cuy::ii, mkfit::MkFinder::inputTracksAndHitIdx(), mkfit::MkFinder::inputTracksAndHits(), createfilelist::int, mkfit::LayerInfo::is_barrel(), iseed, mkfit::SteeringParams::IT_BkwSearch, mkfit::TrackerInfo::layer(), mkfit::IterationConfig::m_backward_params, m_event, m_event_of_comb_cands, mkfit::MkJob::m_event_of_hits, mkfit::MkJob::m_in_fwd, mkfit::MkJob::m_iter_config, m_job, mkfit::MkFinder::m_Label, mkfit::IterationConfig::m_layer_configs, mkfit::IterationConfig::m_params, mkfit::FindingFoos::m_propagate_foo, mkfit::SteeringParams::m_track_scorer, mkfit::MkJob::m_trk_info, mkfit::SteeringParams::make_iterator(), SiStripPI::min, mkfit::LayerOfHits::nHits(), mkfit::NN, mkfit::MkJob::params(), submitPVValidationJobs::params, mkfit::TrackerInfo::prop_config(), mkfit::LayerInfo::propagate_to(), mkfit::EventOfCombCandidates::refCandidates(), mkfit::EventOfCombCandidates::refCandidates_nc(), nano_mu_digi_cff::region, offlineSlimmedPrimaryVertices_cfi::score, edm::second(), mkfit::MkFinder::selectHitIndices(), mkfit::MkFinder::setup(), mkfit::MkJob::steering_params(), mkfit::MkFinder::updateWithLoadedHit(), and gpuVertexFinder::while().

Referenced by findTracksCloneEngine().

972  {
974  const TrackerInfo &trk_info = m_job->m_trk_info;
975  const SteeringParams &st_par = m_job->steering_params(region);
976  const IterationParams &params = m_job->params();
977  const PropagationConfig &prop_config = trk_info.prop_config();
978 
979  const int n_seeds = end_seed - start_seed;
980 
981  std::vector<std::pair<int, int>> seed_cand_idx;
982  std::vector<UpdateIndices> seed_cand_update_idx;
983  seed_cand_idx.reserve(n_seeds * params.maxCandsPerSeed);
984  seed_cand_update_idx.reserve(n_seeds * params.maxCandsPerSeed);
985 
986  std::vector<std::vector<TrackCand>> extra_cands(n_seeds);
987  for (int ii = 0; ii < n_seeds; ++ii)
988  extra_cands[ii].reserve(params.maxCandsPerSeed);
989 
990  cloner.begin_eta_bin(&eoccs, &seed_cand_update_idx, &extra_cands, start_seed, n_seeds);
991 
992  // Loop over layers, starting from after the seed.
993 
994  auto layer_plan_it = st_par.make_iterator(iteration_dir);
995 
996  dprintf("Made iterator for %d, first layer=%d ... end layer=%d\n",
997  iteration_dir,
998  layer_plan_it.layer(),
999  layer_plan_it.last_layer());
1000 
1001  assert(layer_plan_it.is_pickup_only());
1002 
1003  int curr_layer = layer_plan_it.layer(), prev_layer;
1004 
1005  dprintf(
1006  "\nMkBuilder::find_tracks_in_layers region=%d, seed_pickup_layer=%d, first_layer=%d; start_seed=%d, "
1007  "end_seed=%d\n",
1008  region,
1009  curr_layer,
1010  layer_plan_it.next_layer(),
1011  start_seed,
1012  end_seed);
1013 
1014  auto &iter_params = (iteration_dir == SteeringParams::IT_BkwSearch) ? m_job->m_iter_config.m_backward_params
1016 
1017  // Loop over layers according to plan.
1018  while (++layer_plan_it) {
1019  prev_layer = curr_layer;
1020  curr_layer = layer_plan_it.layer();
1021  mkfndr->setup(prop_config,
1023  iter_params,
1024  m_job->m_iter_config.m_layer_configs[curr_layer],
1025  st_par,
1026  m_job->get_mask_for_layer(curr_layer),
1027  m_event,
1028  region,
1029  m_job->m_in_fwd);
1030 
1031  const bool pickup_only = layer_plan_it.is_pickup_only();
1032 
1033  const LayerInfo &layer_info = trk_info.layer(curr_layer);
1034  const LayerOfHits &layer_of_hits = m_job->m_event_of_hits[curr_layer];
1035  const FindingFoos &fnd_foos = FindingFoos::get_finding_foos(layer_info.is_barrel());
1036 
1037  dprintf("\n\n* Processing layer %d, %s\n\n", curr_layer, pickup_only ? "pickup only" : "full finding");
1038  mkfndr->begin_layer(layer_of_hits);
1039 
1040  const int theEndCand = find_tracks_unroll_candidates(
1041  seed_cand_idx, start_seed, end_seed, curr_layer, prev_layer, pickup_only, iteration_dir);
1042 
1043  dprintf(" Number of candidates to process: %d, nHits in layer: %d\n", theEndCand, layer_of_hits.nHits());
1044 
1045  // Don't bother messing with the clone engine if there are no candidates
1046  // (actually it crashes, so this protection is needed).
1047  // If there are no cands on this iteration, there won't be any later on either,
1048  // by the construction of the seed_cand_idx vector.
1049  // XXXXMT There might be cases in endcap where all tracks will miss the
1050  // next layer, but only relevant if we do geometric selection before.
1051 
1052  if (pickup_only || theEndCand == 0)
1053  continue;
1054 
1055  cloner.begin_layer(curr_layer);
1056 
1057  //vectorized loop
1058  for (int itrack = 0; itrack < theEndCand; itrack += NN) {
1059  const int end = std::min(itrack + NN, theEndCand);
1060 
1061 #ifdef DEBUG
1062  dprintf("\nProcessing track=%d, start_seed=%d, n_seeds=%d, theEndCand=%d, end=%d, nn=%d, end_eq_tec=%d\n",
1063  itrack,
1064  start_seed,
1065  n_seeds,
1066  theEndCand,
1067  end,
1068  end - itrack,
1069  end == theEndCand);
1070  dprintf(" (seed,cand): ");
1071  for (int i = itrack; i < end; ++i)
1072  dprintf("(%d,%d) ", seed_cand_idx[i].first, seed_cand_idx[i].second);
1073  dprintf("\n");
1074 #endif
1075 
1076  mkfndr->inputTracksAndHitIdx(eoccs.refCandidates(), seed_cand_idx, itrack, end, false);
1077 
1078 #ifdef DEBUG
1079  for (int i = itrack; i < end; ++i)
1080  dprintf(" track %d, idx %d is from seed %d\n", i, i - itrack, mkfndr->m_Label(i - itrack, 0, 0));
1081 #endif
1082 
1083  // propagate to current layer
1084  mkfndr->clearFailFlag();
1085  (mkfndr->*fnd_foos.m_propagate_foo)(
1086  layer_info.propagate_to(), end - itrack, prop_config.finding_inter_layer_pflags);
1087 
1088  dprint("now get hit range");
1089 
1090  mkfndr->selectHitIndices(layer_of_hits, end - itrack);
1091 
1093  mkfndr, layer_info, extra_cands, seed_cand_idx, region, start_seed, itrack, end);
1094 
1095  // copy_out the propagated track params, errors only.
1096  // Do not, keep cands at last valid hit until actual update,
1097  // this requires change to propagation flags used in MkFinder::updateWithLastHit()
1098  // from intra-layer to inter-layer.
1099  // mkfndr->copyOutParErr(eoccs.refCandidates_nc(), end - itrack, true);
1100 
1101  dprint("make new candidates");
1102  cloner.begin_iteration();
1103 
1104  mkfndr->findCandidatesCloneEngine(layer_of_hits, cloner, start_seed, end - itrack, fnd_foos);
1105 
1106  cloner.end_iteration();
1107  } //end of vectorized loop
1108 
1109  cloner.end_layer();
1110 
1111  // Update loop of best candidates. CandCloner prepares the list of those
1112  // that need update (excluding all those with negative last hit index).
1113 
1114  const int theEndUpdater = seed_cand_update_idx.size();
1115 
1116  for (int itrack = 0; itrack < theEndUpdater; itrack += NN) {
1117  const int end = std::min(itrack + NN, theEndUpdater);
1118 
1119  mkfndr->inputTracksAndHits(eoccs.refCandidates(), layer_of_hits, seed_cand_update_idx, itrack, end, true);
1120 
1121  mkfndr->updateWithLoadedHit(end - itrack, fnd_foos);
1122 
1123  // copy_out the updated track params, errors only (hit-idcs and chi2 already set)
1124  mkfndr->copyOutParErr(eoccs.refCandidates_nc(), end - itrack, false);
1125  }
1126 
1127  // Check if cands are sorted, as expected.
1128 #ifdef DEBUG
1129  for (int iseed = start_seed; iseed < end_seed; ++iseed) {
1130  auto &cc = eoccs[iseed];
1131 
1132  for (int i = 0; i < ((int)cc.size()) - 1; ++i) {
1133  if (cc[i].score() < cc[i + 1].score()) {
1134  printf("CloneEngine - NOT SORTED: layer=%d, iseed=%d (size=%lu)-- %d : %f smaller than %d : %f\n",
1135  curr_layer,
1136  iseed,
1137  cc.size(),
1138  i,
1139  cc[i].score(),
1140  i + 1,
1141  cc[i + 1].score());
1142  }
1143  }
1144  }
1145 #endif
1146  mkfndr->end_layer();
1147  } // end of layer loop
1148 
1149  cloner.end_eta_bin();
1150 
1151  // final sorting
1152  for (int iseed = start_seed; iseed < end_seed; ++iseed) {
1153  eoccs[iseed].mergeCandsAndBestShortOne(m_job->params(), st_par.m_track_scorer, true, true);
1154  }
1155  }
void(MkBase::* m_propagate_foo)(float, const int, const PropagationFlags &)
Definition: FindingFoos.h:23
const IterationConfig & m_iter_config
Definition: MkJob.h:12
int find_tracks_unroll_candidates(std::vector< std::pair< int, int >> &seed_cand_vec, int start_seed, int end_seed, int layer, int prev_layer, bool pickup_only, SteeringParams::IterationType_e iteration_dir)
Definition: MkBuilder.cc:621
IterationParams m_params
IterationParams m_backward_params
void inputTracksAndHitIdx(const std::vector< Track > &tracks, int beg, int end, bool inputProp)
Definition: MkFinder.cc:99
uint32_t cc[maxCellsPerHit]
Definition: gpuFishbone.h:49
void selectHitIndices(const LayerOfHits &layer_of_hits, const int N_proc, bool fill_binsearch_only=false)
Definition: MkFinder.cc:282
void setup(const PropagationConfig &pc, const IterationConfig &ic, const IterationParams &ip, const IterationLayerConfig &ilc, const SteeringParams &sp, const std::vector< bool > *ihm, const Event *ev, int region, bool infwd)
Definition: MkFinder.cc:30
void inputTracksAndHits(const std::vector< CombCandidate > &tracks, const LayerOfHits &layer_of_hits, const std::vector< UpdateIndices > &idxs, int beg, int end, bool inputProp)
Definition: MkFinder.cc:149
float propagate_to() const
Definition: TrackerInfo.h:72
assert(be >=bs)
const auto & params() const
Definition: MkJob.h:27
const EventOfHits & m_event_of_hits
Definition: MkJob.h:13
track_score_func m_track_scorer
unsigned int nHits() const
Definition: HitStructures.h:68
U second(std::pair< T, U > const &p)
void end_iteration()
Definition: CandCloner.cc:63
PropagationFlags finding_inter_layer_pflags
iterator make_iterator(IterationType_e type) const
std::vector< IterationLayerConfig > m_layer_configs
constexpr Matriplex::idx_t NN
Definition: Matrix.h:43
Event * m_event
Definition: MkBuilder.h:124
void clearFailFlag()
Definition: MkBase.h:96
void begin_layer(const LayerOfHits &layer_of_hits)
Definition: MkFinder.cc:68
ii
Definition: cuy.py:589
const LayerInfo & layer(int l) const
Definition: TrackerInfo.h:202
const auto & steering_params(int i)
Definition: MkJob.h:25
const TrackerInfo & m_trk_info
Definition: MkJob.h:10
bool m_in_fwd
Definition: MkJob.h:18
void begin_layer(int lay)
Definition: CandCloner.cc:46
const std::vector< CombCandidate > & refCandidates() const
int iseed
Definition: AMPTWrapper.h:134
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
std::vector< CombCandidate > & refCandidates_nc()
void find_tracks_handle_missed_layers(MkFinder *mkfndr, const LayerInfo &layer_info, std::vector< std::vector< TrackCand >> &tmp_cands, const std::vector< std::pair< int, int >> &seed_cand_idx, const int region, const int start_seed, const int itrack, const int end)
Definition: MkBuilder.cc:686
static const FindingFoos & get_finding_foos(bool is_barrel)
Definition: FindingFoos.cc:18
void updateWithLoadedHit(int N_proc, const FindingFoos &fnd_foos)
Definition: MkFinder.cc:1649
void begin_iteration()
Definition: CandCloner.cc:59
#define dprint(x)
Definition: Debug.h:95
void copyOutParErr(std::vector< CombCandidate > &seed_cand_vec, int N_proc, bool outputProp) const
Definition: MkFinder.cc:1680
void findCandidatesCloneEngine(const LayerOfHits &layer_of_hits, CandCloner &cloner, const int offset, const int N_proc, const FindingFoos &fnd_foos)
Definition: MkFinder.cc:1449
const std::vector< bool > * get_mask_for_layer(int layer)
Definition: MkJob.h:33
void begin_eta_bin(EventOfCombCandidates *e_o_ccs, std::vector< UpdateIndices > *update_list, std::vector< std::vector< TrackCand >> *extra_cands, int start_seed, int n_seeds)
Definition: CandCloner.cc:25
while(__syncthreads_or(more))
#define dprintf(...)
Definition: Debug.h:98
MPlexQI m_Label
Definition: MkFinder.h:286
bool is_barrel() const
Definition: TrackerInfo.h:76
void end_layer()
Definition: MkFinder.cc:88
const PropagationConfig & prop_config() const
Definition: TrackerInfo.h:215

◆ find_tracks_load_seeds()

void mkfit::MkBuilder::find_tracks_load_seeds ( const TrackVec in_seeds,
const bool  seeds_sorted 
)

Definition at line 609 of file MkBuilder.cc.

References cms::cuda::assert(), import_seeds(), mkfit::EventOfCombCandidates::insertSeed(), m_event_of_comb_cands, m_job, m_tracks, mkfit::MkJob::max_max_cands(), nano_mu_digi_cff::region, mkfit::EventOfCombCandidates::reset(), fileCollector::seed, and mkfit::MkJob::steering_params().

Referenced by mkfit::Shell::ProcessEvent(), mkfit::run_OneIteration(), mkfit::runBtpCe_MultiIter(), mkfit::runBuildingTestPlexCloneEngine(), and mkfit::runBuildingTestPlexStandard().

609  {
610  // This will sort seeds according to iteration configuration.
611  assert(!in_seeds.empty());
612  m_tracks.clear(); // m_tracks can be used for BkFit.
613 
614  m_event_of_comb_cands.reset((int)in_seeds.size(), m_job->max_max_cands());
615 
616  import_seeds(in_seeds, seeds_sorted, [&](const Track &seed, int seed_idx, int region, int pos) {
618  });
619  }
void import_seeds(const TrackVec &in_seeds, const bool seeds_sorted, std::function< insert_seed_foo > insert_seed)
Definition: MkBuilder.cc:228
void insertSeed(const Track &seed, int seed_idx, const track_score_func &score_func, int region, int pos)
assert(be >=bs)
void reset(int new_capacity, int max_cands_per_seed, int expected_num_hots=128)
TrackVec m_tracks
Definition: MkBuilder.h:127
const auto & steering_params(int i)
Definition: MkJob.h:25
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
int max_max_cands() const
Definition: MkJob.h:31

◆ find_tracks_load_seeds_BH()

void mkfit::MkBuilder::find_tracks_load_seeds_BH ( const TrackVec in_seeds,
const bool  seeds_sorted 
)

Definition at line 439 of file MkBuilder.cc.

References cms::cuda::assert(), dcall, import_seeds(), m_tracks, nano_mu_digi_cff::region, and fileCollector::seed.

Referenced by mkfit::runBuildingTestPlexBestHit().

439  {
440  // bool debug = true;
441  assert(!in_seeds.empty());
442  m_tracks.resize(in_seeds.size());
443 
444  import_seeds(in_seeds, seeds_sorted, [&](const Track &seed, int seed_idx, int region, int pos) {
445  m_tracks[pos] = seed;
446  m_tracks[pos].setNSeedHits(seed.nTotalHits());
447  m_tracks[pos].setEtaRegion(region);
448  });
449 
450  //dump seeds
451  dcall(print_seeds(m_tracks));
452  }
void import_seeds(const TrackVec &in_seeds, const bool seeds_sorted, std::function< insert_seed_foo > insert_seed)
Definition: MkBuilder.cc:228
assert(be >=bs)
#define dcall(x)
Definition: Debug.h:97
TrackVec m_tracks
Definition: MkBuilder.h:127

◆ find_tracks_unroll_candidates()

int mkfit::MkBuilder::find_tracks_unroll_candidates ( std::vector< std::pair< int, int >> &  seed_cand_vec,
int  start_seed,
int  end_seed,
int  layer,
int  prev_layer,
bool  pickup_only,
SteeringParams::IterationType_e  iteration_dir 
)

Definition at line 621 of file MkBuilder.cc.

References funct::abs(), mkfit::CombCandidate::Dormant, f, mkfit::CombCandidate::Finding, mkfit::CombCandidate::Finished, cms::cuda::for(), createfilelist::int, iseed, mkfit::SteeringParams::IT_BkwSearch, mkfit::SteeringParams::IT_FwdSearch, mkfit::IterationConfig::m_backward_params, m_event_of_comb_cands, mkfit::MkJob::m_iter_config, m_job, m_nan_n_silly_per_layer_count, mkfit::IterationConfig::m_params, mkfit::Const::nan_n_silly_check_cands_every_layer, mkfit::Const::nan_n_silly_fixup_bad_cands_every_layer, mkfit::Const::nan_n_silly_print_bad_cands_every_layer, mkfit::CombCandidate::pickupLayer(), pv::pT, mkfit::CombCandidate::setState(), mkfit::CombCandidate::size(), and mkfit::CombCandidate::state().

Referenced by find_tracks_in_layers(), and findTracksStandard().

627  {
628  int silly_count = 0;
629 
630  seed_cand_vec.clear();
631 
632  auto &iter_params = (iteration_dir == SteeringParams::IT_BkwSearch) ? m_job->m_iter_config.m_backward_params
634 
635  for (int iseed = start_seed; iseed < end_seed; ++iseed) {
637 
638  if (ccand.state() == CombCandidate::Dormant && ccand.pickupLayer() == prev_layer) {
640  }
641  if (!pickup_only && ccand.state() == CombCandidate::Finding) {
642  bool active = false;
643  for (int ic = 0; ic < (int)ccand.size(); ++ic) {
644  if (ccand[ic].getLastHitIdx() != -2) {
645  // Stop candidates with pT<X GeV
646  if (ccand[ic].pT() < iter_params.minPtCut) {
647  ccand[ic].addHitIdx(-2, layer, 0.0f);
648  continue;
649  }
650  // Check if the candidate is close to it's max_r, pi/2 - 0.2 rad (11.5 deg)
651  if (iteration_dir == SteeringParams::IT_FwdSearch && ccand[ic].pT() < 1.2) {
652  const float dphi = std::abs(ccand[ic].posPhi() - ccand[ic].momPhi());
653  if (ccand[ic].posRsq() > 625.f && dphi > 1.371f && dphi < 4.512f) {
654  // printf("Stopping cand at r=%f, posPhi=%.1f momPhi=%.2f pt=%.2f emomEta=%.2f\n",
655  // ccand[ic].posR(), ccand[ic].posPhi(), ccand[ic].momPhi(), ccand[ic].pT(), ccand[ic].momEta());
656  ccand[ic].addHitIdx(-2, layer, 0.0f);
657  continue;
658  }
659  }
660 
661  active = true;
662  seed_cand_vec.push_back(std::pair<int, int>(iseed, ic));
663  ccand[ic].resetOverlaps();
664 
666  if (ccand[ic].hasSillyValues(Const::nan_n_silly_print_bad_cands_every_layer,
668  "Per layer silly check"))
669  ++silly_count;
670  }
671  }
672  }
673  if (!active) {
675  }
676  }
677  }
678 
679  if (Const::nan_n_silly_check_cands_every_layer && silly_count > 0) {
680  m_nan_n_silly_per_layer_count += silly_count;
681  }
682 
683  return seed_cand_vec.size();
684  }
const IterationConfig & m_iter_config
Definition: MkJob.h:12
IterationParams m_params
IterationParams m_backward_params
for(int i=first, nt=offsets[nh];i< nt;i+=gridDim.x *blockDim.x)
std::atomic< int > m_nan_n_silly_per_layer_count
Definition: MkBuilder.h:137
void setState(SeedState_e ss)
constexpr bool nan_n_silly_fixup_bad_cands_every_layer
Definition: Config.h:25
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
double f[11][100]
trk_cand_vec_type::size_type size() const
int iseed
Definition: AMPTWrapper.h:134
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
constexpr bool nan_n_silly_check_cands_every_layer
Definition: Config.h:23
constexpr bool nan_n_silly_print_bad_cands_every_layer
Definition: Config.h:24
SeedState_e state() const

◆ findTracksBestHit()

void mkfit::MkBuilder::findTracksBestHit ( SteeringParams::IterationType_e  iteration_dir = SteeringParams::IT_FwdSearch)

Definition at line 454 of file MkBuilder.cc.

References cms::cuda::assert(), HLT_2023v12_cff::cands, dcall, dprint, dprintf, mkfit::PropagationConfig::finding_inter_layer_pflags, mkfit::g_exe_ctx, mkfit::FindingFoos::get_finding_foos(), mkfit::MkJob::get_mask_for_layer(), mps_fire::i, cuy::ii, mkfit::MkBase::iP, mkfit::LayerInfo::is_barrel(), mkfit::SteeringParams::IT_BkwSearch, label, mkfit::TrackerInfo::layer(), m_event, mkfit::MkJob::m_event_of_hits, mkfit::ExecutionContext::m_finders, mkfit::MkJob::m_in_fwd, mkfit::MkJob::m_iter_config, m_job, mkfit::IterationConfig::m_layer_configs, mkfit::IterationConfig::m_params, mkfit::FindingFoos::m_propagate_foo, m_seedEtaSeparators, m_tracks, mkfit::MkJob::m_trk_info, mkfit::SteeringParams::make_iterator(), SiStripPI::min, mkfit::LayerOfHits::nHits(), mkfit::NN, mkfit::TrackerInfo::prop_config(), mkfit::LayerInfo::propagate_to(), mkfit::TrackerInfo::Reg_Barrel, nano_mu_digi_cff::region, mkfit::MkJob::regions_begin(), mkfit::MkJob::regions_end(), mkfit::LayerInfo::rin(), mkfit::MkJob::steering_params(), and mkfit::WSR_Outside.

Referenced by mkfit::runBuildingTestPlexBestHit().

454  {
455  // bool debug = true;
456 
458 
459  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
460  if (iteration_dir == SteeringParams::IT_BkwSearch && !m_job->steering_params(region).has_bksearch_plan()) {
461  printf("No backward search plan for region %d\n", region);
462  return;
463  }
464 
465  // XXXXXX Select endcap / barrel only ...
466  // if (region != TrackerInfo::Reg_Endcap_Neg && region != TrackerInfo::Reg_Endcap_Pos)
467  // if (region != TrackerInfo::Reg_Barrel)
468  // return;
469 
470  const SteeringParams &st_par = m_job->steering_params(region);
471  const TrackerInfo &trk_info = m_job->m_trk_info;
472  const PropagationConfig &prop_config = trk_info.prop_config();
473 
474  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
475 
476  tbb::parallel_for(rosi.tbb_blk_rng_vec(), [&](const tbb::blocked_range<int> &blk_rng) {
477  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
478 
479  RangeOfSeedIndices rng = rosi.seed_rng(blk_rng);
480 
481  std::vector<int> trk_idcs(NN); // track indices in Matriplex
482  std::vector<int> trk_llay(NN); // last layer on input track
483 
484  while (rng.valid()) {
485  dprint(std::endl << "processing track=" << rng.m_beg << ", label=" << cands[rng.m_beg].label());
486 
487  int prev_layer = 9999;
488 
489  for (int i = rng.m_beg, ii = 0; i < rng.m_end; ++i, ++ii) {
490  int llay = cands[i].getLastHitLyr();
491  trk_llay[ii] = llay;
492  prev_layer = std::min(prev_layer, llay);
493 
494  dprintf(" %2d %2d %2d lay=%3d prev_layer=%d\n", ii, i, cands[i].label(), llay, prev_layer);
495  }
496  int curr_tridx = 0;
497 
498  auto layer_plan_it = st_par.make_iterator(iteration_dir);
499 
500  dprintf("Made iterator for %d, first layer=%d ... end layer=%d\n",
501  iteration_dir,
502  layer_plan_it.layer(),
503  layer_plan_it.last_layer());
504 
505  assert(layer_plan_it.is_pickup_only());
506 
507  int curr_layer = layer_plan_it.layer();
508 
509  mkfndr->m_Stopped.setVal(0);
510 
511  // Loop over layers, starting from after the seed.
512  // Consider inverting loop order and make layer outer, need to
513  // trade off hit prefetching with copy-out of candidates.
514  while (++layer_plan_it) {
515  prev_layer = curr_layer;
516  curr_layer = layer_plan_it.layer();
517  mkfndr->setup(prop_config,
520  m_job->m_iter_config.m_layer_configs[curr_layer],
521  st_par,
522  m_job->get_mask_for_layer(curr_layer),
523  m_event,
524  region,
525  m_job->m_in_fwd);
526 
527  const LayerOfHits &layer_of_hits = m_job->m_event_of_hits[curr_layer];
528  const LayerInfo &layer_info = trk_info.layer(curr_layer);
529  const FindingFoos &fnd_foos = FindingFoos::get_finding_foos(layer_info.is_barrel());
530  dprint("at layer " << curr_layer << ", nHits in layer " << layer_of_hits.nHits());
531 
532  // Pick up seeds that become active on current layer -- unless already fully loaded.
533  if (curr_tridx < rng.n_proc()) {
534  int prev_tridx = curr_tridx;
535 
536  for (int i = rng.m_beg, ii = 0; i < rng.m_end; ++i, ++ii) {
537  if (trk_llay[ii] == prev_layer)
538  trk_idcs[curr_tridx++] = i;
539  }
540  if (curr_tridx > prev_tridx) {
541  dprintf("added %d seeds, started with %d\n", curr_tridx - prev_tridx, prev_tridx);
542 
543  mkfndr->inputTracksAndHitIdx(cands, trk_idcs, prev_tridx, curr_tridx, false, prev_tridx);
544  }
545  }
546 
547  if (layer_plan_it.is_pickup_only())
548  continue;
549 
550  dcall(pre_prop_print(curr_layer, mkfndr.get()));
551 
552  mkfndr->clearFailFlag();
553  (mkfndr.get()->*fnd_foos.m_propagate_foo)(
554  layer_info.propagate_to(), curr_tridx, prop_config.finding_inter_layer_pflags);
555 
556  dcall(post_prop_print(curr_layer, mkfndr.get()));
557 
558  mkfndr->selectHitIndices(layer_of_hits, curr_tridx);
559 
560  // Stop low-pT tracks that can not reach the current barrel layer.
561  if (layer_info.is_barrel()) {
562  const float r_min_sqr = layer_info.rin() * layer_info.rin();
563  for (int i = 0; i < curr_tridx; ++i) {
564  if (!mkfndr->m_Stopped[i]) {
565  if (mkfndr->radiusSqr(i, MkBase::iP) < r_min_sqr) {
567  mkfndr->m_Stopped[i] = 1;
568  mkfndr->outputTrackAndHitIdx(cands[rng.m_beg + i], i, false);
569  }
570  mkfndr->m_XWsrResult[i].m_wsr = WSR_Outside;
571  mkfndr->m_XHitSize[i] = 0;
572  }
573  } else { // make sure we don't add extra work for AddBestHit
574  mkfndr->m_XWsrResult[i].m_wsr = WSR_Outside;
575  mkfndr->m_XHitSize[i] = 0;
576  }
577  }
578  }
579 
580  // make candidates with best hit
581  dprint("make new candidates");
582 
583  mkfndr->addBestHit(layer_of_hits, curr_tridx, fnd_foos);
584 
585  // Stop tracks that have reached N_max_holes.
586  for (int i = 0; i < curr_tridx; ++i) {
587  if (!mkfndr->m_Stopped[i] && mkfndr->bestHitLastHoT(i).index == -2) {
588  mkfndr->m_Stopped[i] = 1;
589  mkfndr->outputTrackAndHitIdx(cands[rng.m_beg + i], i, false);
590  }
591  }
592 
593  } // end of layer loop
594 
595  mkfndr->outputNonStoppedTracksAndHitIdx(cands, trk_idcs, 0, curr_tridx, false);
596 
597  ++rng;
598  } // end of loop over candidates in a tbb chunk
599 
600  mkfndr->release();
601  }); // end parallel_for over candidates in a region
602  }); // end of parallel_for_each over regions
603  }
void(MkBase::* m_propagate_foo)(float, const int, const PropagationFlags &)
Definition: FindingFoos.h:23
const IterationConfig & m_iter_config
Definition: MkJob.h:12
IterationParams m_params
static constexpr int iP
Definition: MkBase.h:19
float rin() const
Definition: TrackerInfo.h:66
const auto regions_begin() const
Definition: MkJob.h:22
float propagate_to() const
Definition: TrackerInfo.h:72
assert(be >=bs)
const EventOfHits & m_event_of_hits
Definition: MkJob.h:13
unsigned int nHits() const
Definition: HitStructures.h:68
#define dcall(x)
Definition: Debug.h:97
char const * label
PropagationFlags finding_inter_layer_pflags
iterator make_iterator(IterationType_e type) const
std::vector< IterationLayerConfig > m_layer_configs
TrackVec m_tracks
Definition: MkBuilder.h:127
constexpr Matriplex::idx_t NN
Definition: Matrix.h:43
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
ExecutionContext g_exe_ctx
Definition: MkBuilder.cc:49
Pool< MkFinder > m_finders
Definition: MkBuilder.cc:40
Event * m_event
Definition: MkBuilder.h:124
ii
Definition: cuy.py:589
const LayerInfo & layer(int l) const
Definition: TrackerInfo.h:202
const auto & steering_params(int i)
Definition: MkJob.h:25
const TrackerInfo & m_trk_info
Definition: MkJob.h:10
bool m_in_fwd
Definition: MkJob.h:18
std::vector< Track > TrackVec
static const FindingFoos & get_finding_foos(bool is_barrel)
Definition: FindingFoos.cc:18
#define dprint(x)
Definition: Debug.h:95
const auto regions_end() const
Definition: MkJob.h:23
const std::vector< bool > * get_mask_for_layer(int layer)
Definition: MkJob.h:33
#define dprintf(...)
Definition: Debug.h:98
bool is_barrel() const
Definition: TrackerInfo.h:76
const PropagationConfig & prop_config() const
Definition: TrackerInfo.h:215

◆ findTracksCloneEngine()

void mkfit::MkBuilder::findTracksCloneEngine ( SteeringParams::IterationType_e  iteration_dir = SteeringParams::IT_FwdSearch)

Definition at line 932 of file MkBuilder.cc.

References dprint, find_tracks_in_layers(), mkfit::g_exe_ctx, mkfit::SteeringParams::IT_BkwSearch, mkfit::ExecutionContext::m_cloners, m_event_of_comb_cands, mkfit::ExecutionContext::m_finders, m_job, m_seedEtaSeparators, mkfit::Config::numSeedsPerTask, mkfit::Config::numThreadsEvents, mkfit::Config::numThreadsFinder, mkfit::MkJob::params(), nano_mu_digi_cff::region, mkfit::MkJob::regions_begin(), mkfit::MkJob::regions_end(), DetachedQuadStep_cff::seeds, mkfit::EventOfCombCandidates::size(), and mkfit::MkJob::steering_params().

Referenced by mkfit::Shell::ProcessEvent(), mkfit::run_OneIteration(), mkfit::runBtpCe_MultiIter(), and mkfit::runBuildingTestPlexCloneEngine().

932  {
933  // debug = true;
934 
936 
937  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
938  if (iteration_dir == SteeringParams::IT_BkwSearch && !m_job->steering_params(region).has_bksearch_plan()) {
939  printf("No backward search plan for region %d\n", region);
940  return;
941  }
942 
943  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
944 
945  // adaptive seeds per task based on the total estimated amount of work to divide among all threads
946  const int adaptiveSPT = std::clamp(
948  dprint("adaptiveSPT " << adaptiveSPT << " fill " << rosi.count() << "/" << eoccs.size() << " region " << region);
949 
950  tbb::parallel_for(rosi.tbb_blk_rng_std(adaptiveSPT), [&](const tbb::blocked_range<int> &seeds) {
951  auto cloner = g_exe_ctx.m_cloners.makeOrGet();
952  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
953 
954  cloner->setup(m_job->params());
955 
956  // loop over layers
957  find_tracks_in_layers(*cloner, mkfndr.get(), iteration_dir, seeds.begin(), seeds.end(), region);
958 
959  mkfndr->release();
960  cloner->release();
961  });
962  });
963 
964  // debug = false;
965  }
const auto regions_begin() const
Definition: MkJob.h:22
const auto & params() const
Definition: MkJob.h:27
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
ExecutionContext g_exe_ctx
Definition: MkBuilder.cc:49
Pool< MkFinder > m_finders
Definition: MkBuilder.cc:40
constexpr int numThreadsFinder
Definition: Config.h:85
Pool< CandCloner > m_cloners
Definition: MkBuilder.cc:38
const auto & steering_params(int i)
Definition: MkJob.h:25
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
#define dprint(x)
Definition: Debug.h:95
constexpr int numSeedsPerTask
Definition: Config.h:87
constexpr int numThreadsEvents
Definition: Config.h:86
const auto regions_end() const
Definition: MkJob.h:23
void find_tracks_in_layers(CandCloner &cloner, MkFinder *mkfndr, SteeringParams::IterationType_e iteration_dir, const int start_seed, const int end_seed, const int region)
Definition: MkBuilder.cc:967

◆ findTracksStandard()

void mkfit::MkBuilder::findTracksStandard ( SteeringParams::IterationType_e  iteration_dir = SteeringParams::IT_FwdSearch)

Definition at line 746 of file MkBuilder.cc.

References mkfit::TrackCand::addHitIdx(), cms::cuda::assert(), dcall, dprint, dprintf, relativeConstraints::empty, find_tracks_handle_missed_layers(), find_tracks_unroll_candidates(), mkfit::PropagationConfig::finding_inter_layer_pflags, dqmdumpme::first, mkfit::g_exe_ctx, mkfit::FindingFoos::get_finding_foos(), mkfit::MkJob::get_mask_for_layer(), mkfit::TrackCand::getLastHitIdx(), mkfit::TrackCand::getLastHitLyr(), cuy::ii, mkfit::TrackCand::incOverlapCount(), createfilelist::int, mkfit::LayerInfo::is_barrel(), iseed, mkfit::SteeringParams::IT_BkwSearch, label, mkfit::TrackerInfo::layer(), mkfit::IterationConfig::m_backward_params, mkfit::HitMatch::m_chi2, m_event, m_event_of_comb_cands, mkfit::MkJob::m_event_of_hits, mkfit::ExecutionContext::m_finders, mkfit::HitMatch::m_hit_idx, mkfit::MkJob::m_in_fwd, mkfit::MkJob::m_iter_config, m_job, mkfit::IterationConfig::m_layer_configs, mkfit::IterationConfig::m_params, mkfit::FindingFoos::m_propagate_foo, m_seedEtaSeparators, mkfit::SteeringParams::m_track_scorer, mkfit::MkJob::m_trk_info, mkfit::SteeringParams::make_iterator(), SiStripPI::min, mkfit::LayerOfHits::nHits(), mkfit::NN, mkfit::Config::numSeedsPerTask, mkfit::Config::numThreadsEvents, mkfit::Config::numThreadsFinder, mkfit::TrackCand::originIndex(), mkfit::MkJob::params(), submitPVValidationJobs::params, mkfit::TrackerInfo::prop_config(), mkfit::LayerInfo::propagate_to(), mkfit::TrackBase::pT(), mkfit::EventOfCombCandidates::refCandidates(), mkfit::LayerOfHits::refHit(), nano_mu_digi_cff::region, mkfit::MkJob::regions_begin(), mkfit::MkJob::regions_end(), mkfit::TrackBase::score(), edm::second(), DetachedQuadStep_cff::seeds, mkfit::EventOfCombCandidates::size(), jetUpdater_cfi::sort, mkfit::MkJob::steering_params(), and gpuVertexFinder::while().

Referenced by mkfit::runBuildingTestPlexStandard().

746  {
747  // debug = true;
748 
750 
751  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
752  if (iteration_dir == SteeringParams::IT_BkwSearch && !m_job->steering_params(region).has_bksearch_plan()) {
753  printf("No backward search plan for region %d\n", region);
754  return;
755  }
756 
757  const TrackerInfo &trk_info = m_job->m_trk_info;
758  const SteeringParams &st_par = m_job->steering_params(region);
759  const IterationParams &params = m_job->params();
760  const PropagationConfig &prop_config = trk_info.prop_config();
761 
762  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
763 
764  // adaptive seeds per task based on the total estimated amount of work to divide among all threads
765  const int adaptiveSPT = std::clamp(
767  dprint("adaptiveSPT " << adaptiveSPT << " fill " << rosi.count() << "/" << eoccs.size() << " region " << region);
768 
769  // loop over seeds
770  tbb::parallel_for(rosi.tbb_blk_rng_std(adaptiveSPT), [&](const tbb::blocked_range<int> &seeds) {
771  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
772 
773  const int start_seed = seeds.begin();
774  const int end_seed = seeds.end();
775  const int n_seeds = end_seed - start_seed;
776 
777  std::vector<std::vector<TrackCand>> tmp_cands(n_seeds);
778  for (size_t iseed = 0; iseed < tmp_cands.size(); ++iseed) {
779  tmp_cands[iseed].reserve(2 * params.maxCandsPerSeed); //factor 2 seems reasonable to start with
780  }
781 
782  std::vector<std::pair<int, int>> seed_cand_idx;
783  seed_cand_idx.reserve(n_seeds * params.maxCandsPerSeed);
784 
785  auto layer_plan_it = st_par.make_iterator(iteration_dir);
786 
787  dprintf("Made iterator for %d, first layer=%d ... end layer=%d\n",
788  iteration_dir,
789  layer_plan_it.layer(),
790  layer_plan_it.last_layer());
791 
792  assert(layer_plan_it.is_pickup_only());
793 
794  int curr_layer = layer_plan_it.layer(), prev_layer;
795 
796  dprintf("\nMkBuilder::FindTracksStandard region=%d, seed_pickup_layer=%d, first_layer=%d\n",
797  region,
798  curr_layer,
799  layer_plan_it.next_layer());
800 
801  auto &iter_params = (iteration_dir == SteeringParams::IT_BkwSearch) ? m_job->m_iter_config.m_backward_params
803 
804  // Loop over layers, starting from after the seed.
805  while (++layer_plan_it) {
806  prev_layer = curr_layer;
807  curr_layer = layer_plan_it.layer();
808  mkfndr->setup(prop_config,
810  iter_params,
811  m_job->m_iter_config.m_layer_configs[curr_layer],
812  st_par,
813  m_job->get_mask_for_layer(curr_layer),
814  m_event,
815  region,
816  m_job->m_in_fwd);
817 
818  const LayerOfHits &layer_of_hits = m_job->m_event_of_hits[curr_layer];
819  const LayerInfo &layer_info = trk_info.layer(curr_layer);
820  const FindingFoos &fnd_foos = FindingFoos::get_finding_foos(layer_info.is_barrel());
821 
822  dprintf("\n* Processing layer %d\n", curr_layer);
823  mkfndr->begin_layer(layer_of_hits);
824 
825  int theEndCand = find_tracks_unroll_candidates(seed_cand_idx,
826  start_seed,
827  end_seed,
828  curr_layer,
829  prev_layer,
830  layer_plan_it.is_pickup_only(),
831  iteration_dir);
832 
833  dprintf(" Number of candidates to process: %d, nHits in layer: %d\n", theEndCand, layer_of_hits.nHits());
834 
835  if (layer_plan_it.is_pickup_only() || theEndCand == 0)
836  continue;
837 
838  // vectorized loop
839  for (int itrack = 0; itrack < theEndCand; itrack += NN) {
840  int end = std::min(itrack + NN, theEndCand);
841 
842  dprint("processing track=" << itrack << ", label="
843  << eoccs[seed_cand_idx[itrack].first][seed_cand_idx[itrack].second].label());
844 
845  //fixme find a way to deal only with the candidates needed in this thread
846  mkfndr->inputTracksAndHitIdx(eoccs.refCandidates(), seed_cand_idx, itrack, end, false);
847 
848  //propagate to layer
849  dcall(pre_prop_print(curr_layer, mkfndr.get()));
850 
851  mkfndr->clearFailFlag();
852  (mkfndr.get()->*fnd_foos.m_propagate_foo)(
853  layer_info.propagate_to(), end - itrack, prop_config.finding_inter_layer_pflags);
854 
855  dcall(post_prop_print(curr_layer, mkfndr.get()));
856 
857  dprint("now get hit range");
858  mkfndr->selectHitIndices(layer_of_hits, end - itrack);
859 
861  mkfndr.get(), layer_info, tmp_cands, seed_cand_idx, region, start_seed, itrack, end);
862 
863  dprint("make new candidates");
864  mkfndr->findCandidates(layer_of_hits, tmp_cands, start_seed, end - itrack, fnd_foos);
865 
866  } //end of vectorized loop
867 
868  // sort the input candidates
869  for (int is = 0; is < n_seeds; ++is) {
870  dprint("dump seed n " << is << " with N_input_candidates=" << tmp_cands[is].size());
871 
872  std::sort(tmp_cands[is].begin(), tmp_cands[is].end(), sortCandByScore);
873  }
874 
875  // now fill out the output candidates
876  for (int is = 0; is < n_seeds; ++is) {
877  if (!tmp_cands[is].empty()) {
878  eoccs[start_seed + is].clear();
879 
880  // Put good candidates into eoccs, process -2 candidates.
881  int n_placed = 0;
882  bool first_short = true;
883  for (int ii = 0; ii < (int)tmp_cands[is].size() && n_placed < params.maxCandsPerSeed; ++ii) {
884  TrackCand &tc = tmp_cands[is][ii];
885 
886  // See if we have an overlap hit available, but only if we have a true hit in this layer
887  // and pT is above the pTCutOverlap
888  if (tc.pT() > params.pTCutOverlap && tc.getLastHitLyr() == curr_layer && tc.getLastHitIdx() >= 0) {
889  CombCandidate &ccand = eoccs[start_seed + is];
890 
891  HitMatch *hm = ccand[tc.originIndex()].findOverlap(
892  tc.getLastHitIdx(), layer_of_hits.refHit(tc.getLastHitIdx()).detIDinLayer());
893 
894  if (hm) {
895  tc.addHitIdx(hm->m_hit_idx, curr_layer, hm->m_chi2);
896  tc.incOverlapCount();
897  }
898  }
899 
900  if (tc.getLastHitIdx() != -2) {
901  eoccs[start_seed + is].emplace_back(tc);
902  ++n_placed;
903  } else if (first_short) {
904  first_short = false;
905  if (tc.score() > eoccs[start_seed + is].refBestShortCand().score()) {
906  eoccs[start_seed + is].setBestShortCand(tc);
907  }
908  }
909  }
910 
911  tmp_cands[is].clear();
912  }
913  }
914  mkfndr->end_layer();
915  } // end of layer loop
916  mkfndr->release();
917 
918  // final sorting
919  for (int iseed = start_seed; iseed < end_seed; ++iseed) {
920  eoccs[iseed].mergeCandsAndBestShortOne(m_job->params(), st_par.m_track_scorer, true, true);
921  }
922  }); // end parallel-for over chunk of seeds within region
923  }); // end of parallel-for-each over eta regions
924 
925  // debug = false;
926  }
void(MkBase::* m_propagate_foo)(float, const int, const PropagationFlags &)
Definition: FindingFoos.h:23
const IterationConfig & m_iter_config
Definition: MkJob.h:12
size
Write out results.
void addHitIdx(int hitIdx, int hitLyr, float chi2)
int find_tracks_unroll_candidates(std::vector< std::pair< int, int >> &seed_cand_vec, int start_seed, int end_seed, int layer, int prev_layer, bool pickup_only, SteeringParams::IterationType_e iteration_dir)
Definition: MkBuilder.cc:621
IterationParams m_params
IterationParams m_backward_params
float pT() const
Definition: Track.h:171
const auto regions_begin() const
Definition: MkJob.h:22
float propagate_to() const
Definition: TrackerInfo.h:72
assert(be >=bs)
const auto & params() const
Definition: MkJob.h:27
const EventOfHits & m_event_of_hits
Definition: MkJob.h:13
track_score_func m_track_scorer
unsigned int nHits() const
Definition: HitStructures.h:68
U second(std::pair< T, U > const &p)
#define dcall(x)
Definition: Debug.h:97
char const * label
PropagationFlags finding_inter_layer_pflags
float score() const
Definition: Track.h:187
iterator make_iterator(IterationType_e type) const
std::vector< IterationLayerConfig > m_layer_configs
constexpr Matriplex::idx_t NN
Definition: Matrix.h:43
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
ExecutionContext g_exe_ctx
Definition: MkBuilder.cc:49
Pool< MkFinder > m_finders
Definition: MkBuilder.cc:40
Event * m_event
Definition: MkBuilder.h:124
constexpr int numThreadsFinder
Definition: Config.h:85
ii
Definition: cuy.py:589
const LayerInfo & layer(int l) const
Definition: TrackerInfo.h:202
const auto & steering_params(int i)
Definition: MkJob.h:25
const TrackerInfo & m_trk_info
Definition: MkJob.h:10
bool m_in_fwd
Definition: MkJob.h:18
const std::vector< CombCandidate > & refCandidates() const
int iseed
Definition: AMPTWrapper.h:134
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
void find_tracks_handle_missed_layers(MkFinder *mkfndr, const LayerInfo &layer_info, std::vector< std::vector< TrackCand >> &tmp_cands, const std::vector< std::pair< int, int >> &seed_cand_idx, const int region, const int start_seed, const int itrack, const int end)
Definition: MkBuilder.cc:686
static const FindingFoos & get_finding_foos(bool is_barrel)
Definition: FindingFoos.cc:18
int getLastHitIdx() const
#define dprint(x)
Definition: Debug.h:95
constexpr int numSeedsPerTask
Definition: Config.h:87
constexpr int numThreadsEvents
Definition: Config.h:86
const auto regions_end() const
Definition: MkJob.h:23
int getLastHitLyr() const
const std::vector< bool > * get_mask_for_layer(int layer)
Definition: MkJob.h:33
while(__syncthreads_or(more))
int originIndex() const
#define dprintf(...)
Definition: Debug.h:98
bool is_barrel() const
Definition: TrackerInfo.h:76
const PropagationConfig & prop_config() const
Definition: TrackerInfo.h:215
const Hit & refHit(int i) const

◆ fit_cands()

void mkfit::MkBuilder::fit_cands ( MkFinder mkfndr,
int  start_cand,
int  end_cand,
int  region 
)

Definition at line 1289 of file MkBuilder.cc.

References mkfit::PropagationConfig::backward_fit_to_pca, mkfit::MkFinder::bkFitFitTracks(), mkfit::MkFinder::bkFitInputTracks(), mkfit::MkFinder::bkFitOutputTracks(), mkfit::MkFinder::bkFitPropTracksToPCA(), debug, dprintf, dqmdumpme::first, mps_fire::i, m_event, m_event_of_comb_cands, mkfit::MkJob::m_event_of_hits, m_job, mkfit::MkJob::m_trk_info, SiStripPI::min, mkfit::NN, mkfit::TrackerInfo::prop_config(), nano_mu_digi_cff::region, mkfit::MkFinder::release(), mkfit::MkFinder::setup_bkfit(), and mkfit::MkJob::steering_params().

Referenced by backwardFit().

1289  {
1291  const SteeringParams &st_par = m_job->steering_params(region);
1292  const PropagationConfig &prop_config = m_job->m_trk_info.prop_config();
1293  mkfndr->setup_bkfit(prop_config, st_par, m_event);
1294 
1295  int step = NN;
1296  for (int icand = start_cand; icand < end_cand; icand += step) {
1297  int end = std::min(icand + NN, end_cand);
1298 
1299  bool chi_debug = false;
1300 
1301 #ifdef DEBUG_FINAL_FIT
1302  bool debug = true;
1303  dprintf("Pre Final fit for %d - %d\n", icand, end);
1304  for (int i = icand; i < end; ++i) {
1305  dprint_tcand(eoccs[i][0], i);
1306  }
1307  chi_debug = true;
1308  static bool first = true;
1309  if (first) {
1310  // ./mkFit ... | perl -ne 'if (/^BKF_OVERLAP/) { s/^BKF_OVERLAP //og; print; }' > bkf_ovlp.rtt
1311  dprintf(
1312  "BKF_OVERLAP event/I:label/I:prod_type/I:is_findable/I:layer/I:is_stereo/I:is_barrel/I:"
1313  "pt/F:pt_cur/F:eta/F:phi/F:phi_cur/F:r_cur/F:z_cur/F:chi2/F:isnan/I:isfin/I:gtzero/I:hit_label/I:"
1314  "sx_t/F:sy_t/F:sz_t/F:d_xy/F:d_z/F\n");
1315  first = false;
1316  }
1317 #endif
1318 
1319  // input tracks
1320  mkfndr->bkFitInputTracks(eoccs, icand, end);
1321 
1322  // fit tracks back to first layer
1323  mkfndr->bkFitFitTracks(m_job->m_event_of_hits, st_par, end - icand, chi_debug);
1324 
1325  // now move one last time to PCA
1326  if (prop_config.backward_fit_to_pca) {
1327  mkfndr->bkFitPropTracksToPCA(end - icand);
1328  }
1329 
1330  mkfndr->bkFitOutputTracks(eoccs, icand, end, prop_config.backward_fit_to_pca);
1331 
1332 #ifdef DEBUG_FINAL_FIT
1333  dprintf("Post Final fit for %d - %d\n", icand, end);
1334  for (int i = icand; i < end; ++i) {
1335  dprint_tcand(eoccs[i][0], i);
1336  }
1337 #endif
1338  }
1339  mkfndr->release();
1340  }
void bkFitInputTracks(TrackVec &cands, int beg, int end)
Definition: MkFinder.cc:1701
void bkFitPropTracksToPCA(const int N_proc)
Definition: MkFinder.cc:2187
void release()
Definition: MkFinder.cc:56
const EventOfHits & m_event_of_hits
Definition: MkJob.h:13
constexpr Matriplex::idx_t NN
Definition: Matrix.h:43
void setup_bkfit(const PropagationConfig &pc, const SteeringParams &sp, const Event *ev)
Definition: MkFinder.cc:50
Event * m_event
Definition: MkBuilder.h:124
const auto & steering_params(int i)
Definition: MkJob.h:25
const TrackerInfo & m_trk_info
Definition: MkJob.h:10
#define debug
Definition: HDRShower.cc:19
void bkFitFitTracks(const EventOfHits &eventofhits, const SteeringParams &st_par, const int N_proc, bool chiDebug=false)
Definition: MkFinder.cc:1959
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
void bkFitOutputTracks(TrackVec &cands, int beg, int end, bool outputProp)
Definition: MkFinder.cc:1758
step
Definition: StallMonitor.cc:83
#define dprintf(...)
Definition: Debug.h:98
const PropagationConfig & prop_config() const
Definition: TrackerInfo.h:215

◆ fit_cands_BH()

void mkfit::MkBuilder::fit_cands_BH ( MkFinder mkfndr,
int  start_cand,
int  end_cand,
int  region 
)

Definition at line 1193 of file MkBuilder.cc.

References mkfit::PropagationConfig::backward_fit_to_pca, mkfit::MkFinder::bkFitFitTracksBH(), mkfit::MkFinder::bkFitInputTracks(), mkfit::MkFinder::bkFitOutputTracks(), mkfit::MkFinder::bkFitPropTracksToPCA(), debug, dprintf, mkfit::Event::evtID(), f, mps_fire::i, mkfit::MkBase::iP, mkfit::MkFinder::m_Chi2, m_event, m_event_of_comb_cands, mkfit::MkJob::m_event_of_hits, m_job, mkfit::MkBase::m_Par, m_tracks, mkfit::MkJob::m_trk_info, SiStripPI::min, mkfit::NN, mkfit::TrackerInfo::prop_config(), nano_mu_digi_cff::region, mkfit::MkFinder::setup_bkfit(), and mkfit::MkJob::steering_params().

Referenced by backwardFitBH().

1193  {
1194  const SteeringParams &st_par = m_job->steering_params(region);
1195  const PropagationConfig &prop_config = m_job->m_trk_info.prop_config();
1196  mkfndr->setup_bkfit(prop_config, st_par, m_event);
1197 #ifdef DEBUG_FINAL_FIT
1199  bool debug = true;
1200 #endif
1201 
1202  for (int icand = start_cand; icand < end_cand; icand += NN) {
1203  const int end = std::min(icand + NN, end_cand);
1204 
1205 #ifdef DEBUG_FINAL_FIT
1206  dprintf("Pre Final fit for %d - %d\n", icand, end);
1207  for (int i = icand; i < end; ++i) {
1208  dprint_tcand(eoccs[i][0], i);
1209  }
1210 #endif
1211 
1212  bool chi_debug = false;
1213 #ifdef DEBUG_BACKWARD_FIT_BH
1214  redo_fit:
1215 #endif
1216 
1217  // input candidate tracks
1218  mkfndr->bkFitInputTracks(m_tracks, icand, end);
1219 
1220  // perform fit back to first layer on track
1221  mkfndr->bkFitFitTracksBH(m_job->m_event_of_hits, st_par, end - icand, chi_debug);
1222 
1223  // now move one last time to PCA
1224  if (prop_config.backward_fit_to_pca) {
1225  mkfndr->bkFitPropTracksToPCA(end - icand);
1226  }
1227 
1228 #ifdef DEBUG_BACKWARD_FIT_BH
1229  // Dump tracks with pT > 2 and chi2/dof > 20. Assumes MPT_SIZE=1.
1230  if (!chi_debug && 1.0f / mkfndr->m_Par[MkBase::iP].At(0, 3, 0) > 2.0f &&
1231  mkfndr->m_Chi2(0, 0, 0) / (eoccs[icand][0].nFoundHits() * 3 - 6) > 20.0f) {
1232  chi_debug = true;
1233 #ifdef MKFIT_STANDALONE
1234  printf("CHIHDR Event %d, Cand %3d, pT %f, chipdof %f ### NOTE x,y,z in cm, sigmas, deltas in mum ### !!!\n",
1235  m_event->evtID(),
1236 #else
1237  printf("CHIHDR Cand %3d, pT %f, chipdof %f ### NOTE x,y,z in cm, sigmas, deltas in mum ### !!!\n",
1238 #endif
1239  icand,
1240  1.0f / mkfndr->m_Par[MkBase::iP].At(0, 3, 0),
1241  mkfndr->m_Chi2(0, 0, 0) / (eoccs[icand][0].nFoundHits() * 3 - 6));
1242  // clang-format off
1243  printf("CHIHDR %3s %10s"
1244  " %10s %10s %10s %10s %11s %11s %11s"
1245  " %10s %10s %10s %10s %11s %11s %11s"
1246  " %10s %10s %10s %10s %10s %11s %11s\n",
1247  "lyr", "chi2",
1248  "x_h", "y_h", "z_h", "r_h", "sx_h", "sy_h", "sz_h",
1249  "x_t", "y_t", "z_t", "r_t", "sx_t", "sy_t", "sz_t",
1250  "pt", "phi", "theta", "phi_h", "phi_t", "d_xy", "d_z");
1251  // clang-format on
1252  goto redo_fit;
1253  }
1254 #endif
1255 
1256  // copy out full set of info at last propagated position
1257  mkfndr->bkFitOutputTracks(m_tracks, icand, end, prop_config.backward_fit_to_pca);
1258 
1259 #ifdef DEBUG_FINAL_FIT
1260  dprintf("Post Final fit for %d - %d\n", icand, end);
1261  for (int i = icand; i < end; ++i) {
1262  dprint_tcand(eoccs[i][0], i);
1263  }
1264 #endif
1265  }
1266  }
static constexpr int iP
Definition: MkBase.h:19
void bkFitInputTracks(TrackVec &cands, int beg, int end)
Definition: MkFinder.cc:1701
MPlexLV m_Par[2]
Definition: MkBase.h:102
void bkFitPropTracksToPCA(const int N_proc)
Definition: MkFinder.cc:2187
int evtID() const
Definition: Event.h:23
const EventOfHits & m_event_of_hits
Definition: MkJob.h:13
TrackVec m_tracks
Definition: MkBuilder.h:127
constexpr Matriplex::idx_t NN
Definition: Matrix.h:43
void setup_bkfit(const PropagationConfig &pc, const SteeringParams &sp, const Event *ev)
Definition: MkFinder.cc:50
double f[11][100]
Event * m_event
Definition: MkBuilder.h:124
const auto & steering_params(int i)
Definition: MkJob.h:25
const TrackerInfo & m_trk_info
Definition: MkJob.h:10
#define debug
Definition: HDRShower.cc:19
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
MPlexQF m_Chi2
Definition: MkFinder.h:285
void bkFitOutputTracks(TrackVec &cands, int beg, int end, bool outputProp)
Definition: MkFinder.cc:1758
void bkFitFitTracksBH(const EventOfHits &eventofhits, const SteeringParams &st_par, const int N_proc, bool chiDebug=false)
Definition: MkFinder.cc:1806
#define dprintf(...)
Definition: Debug.h:98
const PropagationConfig & prop_config() const
Definition: TrackerInfo.h:215

◆ fit_one_seed_set()

void mkfit::MkBuilder::fit_one_seed_set ( TrackVec simtracks,
int  itrack,
int  end,
MkFitter mkfttr,
const bool  is_brl[] 
)
private

◆ import_seeds()

void mkfit::MkBuilder::import_seeds ( const TrackVec in_seeds,
const bool  seeds_sorted,
std::function< insert_seed_foo insert_seed 
)

Definition at line 228 of file MkBuilder.cc.

References mkfit::binnor< C, A1, A2, NB_first, NB_count >::begin_registration(), dprintf, PVValHelper::eta, mkfit::binnor< C, A1, A2, NB_first, NB_count >::finalize_registration(), mps_fire::i, dqmiolumiharvest::j, mkfit::HitOnTrack::layer, mkfit::MkJob::m_event_of_hits, mkfit::MkJob::m_iter_config, m_job, mkfit::binnor< C, A1, A2, NB_first, NB_count >::m_ranks, mkfit::IterationConfig::m_seed_partitioner, m_seedEtaSeparators, m_seedMaxLastLayer, m_seedMinLastLayer, mkfit::MkJob::m_trk_info, SiStripPI::max, SiStripPI::min, mkfit::MkJob::num_regions(), mkfit::Const::PI, mkfit::binnor< C, A1, A2, NB_first, NB_count >::register_entry_safe(), and fileCollector::seed.

Referenced by find_tracks_load_seeds(), and find_tracks_load_seeds_BH().

230  {
231  // bool debug = true;
232 
233  const int size = in_seeds.size();
234 
236  std::vector<unsigned> ranks;
237  if (!seeds_sorted) {
238  // We don't care about bins in phi, use low N to reduce overall number of bins.
240  axis<float, unsigned short, 8, 8> ax_eta(-3.0, 3.0, 64u);
242  part.m_phi_eta_foo = [&](float phi, float eta) { phi_eta_binnor.register_entry_safe(phi, eta); };
243 
244  phi_eta_binnor.begin_registration(size);
246  phi_eta_binnor.finalize_registration();
247  ranks.swap(phi_eta_binnor.m_ranks);
248  } else {
250  }
251 
252  for (int i = 0; i < size; ++i) {
253  int j = seeds_sorted ? i : ranks[i];
254  int reg = part.m_region[j];
255  ++m_seedEtaSeparators[reg];
256  }
257 
258  // Sum up region counts to contain actual ending indices and prepare insertion cursors.
259  // Fix min/max layers.
260  std::vector<int> seed_cursors(m_job->num_regions());
261  for (int reg = 1; reg < m_job->num_regions(); ++reg) {
262  seed_cursors[reg] = m_seedEtaSeparators[reg - 1];
263  m_seedEtaSeparators[reg] += m_seedEtaSeparators[reg - 1];
264  }
265 
266  // Actually imports seeds, detect last-hit layer range per region.
267  for (int i = 0; i < size; ++i) {
268  int j = seeds_sorted ? i : ranks[i];
269  int reg = part.m_region[j];
270  const Track &seed = in_seeds[j];
271  insert_seed(seed, j, reg, seed_cursors[reg]++);
272 
273  HitOnTrack hot = seed.getLastHitOnTrack();
276  }
277 
278  // Fix min/max layers
279  for (int reg = 0; reg < m_job->num_regions(); ++reg) {
280  if (m_seedMinLastLayer[reg] == 9999)
281  m_seedMinLastLayer[reg] = -1;
282  if (m_seedMaxLastLayer[reg] == 0)
283  m_seedMaxLastLayer[reg] = -1;
284  }
285 
286  // clang-format off
287  dprintf("MkBuilder::import_seeds finished import of %d seeds (last seeding layer min, max):\n"
288  " ec- = %d(%d,%d), t- = %d(%d,%d), brl = %d(%d,%d), t+ = %d(%d,%d), ec+ = %d(%d,%d).\n",
289  size,
295  // dcall(print_seeds(m_event_of_comb_cands));
296  // clang-format on
297  }
const IterationConfig & m_iter_config
Definition: MkJob.h:12
size
Write out results.
const EventOfHits & m_event_of_hits
Definition: MkJob.h:13
constexpr float PI
Definition: Config.h:7
partition_seeds_func m_seed_partitioner
std::vector< int > m_seedEtaSeparators
Definition: MkBuilder.h:133
const TrackerInfo & m_trk_info
Definition: MkJob.h:10
part
Definition: HCALResponse.h:20
std::vector< int > m_seedMaxLastLayer
Definition: MkBuilder.h:135
std::vector< int > m_seedMinLastLayer
Definition: MkBuilder.h:134
int num_regions() const
Definition: MkJob.h:21
#define dprintf(...)
Definition: Debug.h:98

◆ make_builder()

std::unique_ptr< MkBuilder > mkfit::MkBuilder::make_builder ( bool  silent = true)
static

Definition at line 166 of file MkBuilder.cc.

References CastorRawToDigi_cfi::silent.

Referenced by test_standard().

166 { return std::make_unique<MkBuilder>(silent); }

◆ max_hits_layer()

std::pair< int, int > mkfit::MkBuilder::max_hits_layer ( const EventOfHits eoh) const

Definition at line 170 of file MkBuilder.cc.

References MainPageGenerator::l, numberPerLSFilter_cff::maxN, and mkfit::EventOfHits::nLayers().

170  {
171  int maxN = 0;
172  int maxL = 0;
173  for (int l = 0; l < eoh.nLayers(); ++l) {
174  int lsize = eoh[l].nHits();
175  if (lsize > maxN) {
176  maxN = lsize;
177  maxL = eoh[l].layer_id();
178  }
179  }
180  return {maxN, maxL};
181  }
int nLayers() const

◆ populate()

void mkfit::MkBuilder::populate ( )
static

Definition at line 168 of file MkBuilder.cc.

References mkfit::g_exe_ctx, mkfit::Config::numThreadsFinder, and mkfit::ExecutionContext::populate().

Referenced by mkfit::MkBuilderWrapper::populate(), and test_standard().

ExecutionContext g_exe_ctx
Definition: MkBuilder.cc:49
constexpr int numThreadsFinder
Definition: Config.h:85
void populate(int n_thr)
Definition: MkBuilder.cc:42

◆ ref_eocc()

const EventOfCombCandidates& mkfit::MkBuilder::ref_eocc ( ) const
inline

Definition at line 73 of file MkBuilder.h.

References m_event_of_comb_cands.

Referenced by mkfit::Shell::ProcessEvent().

73 { return m_event_of_comb_cands; }
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130

◆ ref_tracks()

const TrackVec& mkfit::MkBuilder::ref_tracks ( ) const
inline

◆ ref_tracks_nc()

TrackVec& mkfit::MkBuilder::ref_tracks_nc ( )
inline

Definition at line 71 of file MkBuilder.h.

References m_tracks.

Referenced by mkfit::runBtpCe_MultiIter().

71 { return m_tracks; }
TrackVec m_tracks
Definition: MkBuilder.h:127

◆ release_memory()

void mkfit::MkBuilder::release_memory ( )

Definition at line 222 of file MkBuilder.cc.

References m_event_of_comb_cands, m_tracks, mkfit::EventOfCombCandidates::releaseMemory(), and createJobs::tmp.

Referenced by mkfit::run_OneIteration(), and mkfit::runBtpCe_MultiIter().

222  {
223  TrackVec tmp;
224  m_tracks.swap(tmp);
226  }
TrackVec m_tracks
Definition: MkBuilder.h:127
std::vector< Track > TrackVec
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
tmp
align.sh
Definition: createJobs.py:716

◆ seed_post_cleaning()

void mkfit::MkBuilder::seed_post_cleaning ( TrackVec tv)

Definition at line 407 of file MkBuilder.cc.

References submitPVResolutionJobs::count, mps_fire::i, createfilelist::int, m_silent, mkfit::Const::nan_n_silly_check_seeds, mkfit::Const::nan_n_silly_fixup_bad_seeds, mkfit::Const::nan_n_silly_print_bad_seeds, and mkfit::Const::nan_n_silly_remove_bad_seeds.

Referenced by mkfit::Shell::ProcessEvent(), mkfit::run_OneIteration(), and mkfit::runBtpCe_MultiIter().

407  {
409  int count = 0;
410 
411  for (int i = 0; i < (int)tv.size(); ++i) {
412  bool silly = tv[i].hasSillyValues(Const::nan_n_silly_print_bad_seeds,
414  "Post-cleaning seed silly value check and fix");
415  if (silly) {
416  ++count;
418  // XXXX MT
419  // Could do somethin smarter here: set as Stopped ? check in seed cleaning ?
420  tv.erase(tv.begin() + i);
421  --i;
422  }
423  }
424  }
425 
426  if (count > 0 && !m_silent) {
427  printf("Nan'n'Silly detected %d silly seeds (fix=%d, remove=%d).\n",
428  count,
431  }
432  }
433  }
constexpr bool nan_n_silly_print_bad_seeds
Definition: Config.h:19
constexpr bool nan_n_silly_check_seeds
Definition: Config.h:18
constexpr bool nan_n_silly_remove_bad_seeds
Definition: Config.h:21
constexpr bool nan_n_silly_fixup_bad_seeds
Definition: Config.h:20

◆ select_best_comb_cands()

void mkfit::MkBuilder::select_best_comb_cands ( bool  clear_m_tracks = false,
bool  remove_missing_hits = false 
)

Definition at line 378 of file MkBuilder.cc.

References export_best_comb_cands(), and m_tracks.

Referenced by mkfit::runBtpCe_MultiIter(), mkfit::runBuildingTestPlexCloneEngine(), and mkfit::runBuildingTestPlexStandard().

378  {
379  if (clear_m_tracks)
380  m_tracks.clear();
381  export_best_comb_cands(m_tracks, remove_missing_hits);
382  }
TrackVec m_tracks
Definition: MkBuilder.h:127
void export_best_comb_cands(TrackVec &out_vec, bool remove_missing_hits=false)
Definition: MkBuilder.cc:384

◆ total_cands()

int mkfit::MkBuilder::total_cands ( ) const

Definition at line 183 of file MkBuilder.cc.

References mps_fire::i, m_event_of_comb_cands, and mkfit::EventOfCombCandidates::size().

183  {
184  int res = 0;
185  for (int i = 0; i < m_event_of_comb_cands.size(); ++i)
187  return res;
188  }
size
Write out results.
Definition: Electron.h:6
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130

Member Data Documentation

◆ m_event

Event* mkfit::MkBuilder::m_event = nullptr
private

◆ m_event_of_comb_cands

EventOfCombCandidates mkfit::MkBuilder::m_event_of_comb_cands
private

◆ m_job

MkJob* mkfit::MkBuilder::m_job = nullptr
private

◆ m_nan_n_silly_per_layer_count

std::atomic<int> mkfit::MkBuilder::m_nan_n_silly_per_layer_count
private

Definition at line 137 of file MkBuilder.h.

Referenced by begin_event(), and find_tracks_unroll_candidates().

◆ m_seedEtaSeparators

std::vector<int> mkfit::MkBuilder::m_seedEtaSeparators
private

◆ m_seedMaxLastLayer

std::vector<int> mkfit::MkBuilder::m_seedMaxLastLayer
private

Definition at line 135 of file MkBuilder.h.

Referenced by begin_event(), and import_seeds().

◆ m_seedMinLastLayer

std::vector<int> mkfit::MkBuilder::m_seedMinLastLayer
private

Definition at line 134 of file MkBuilder.h.

Referenced by begin_event(), and import_seeds().

◆ m_silent

bool mkfit::MkBuilder::m_silent
private

Definition at line 139 of file MkBuilder.h.

Referenced by begin_event(), and seed_post_cleaning().

◆ m_tracks

TrackVec mkfit::MkBuilder::m_tracks
private