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)
 

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)

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 1260 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().

1260  {
1262 
1263  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
1264  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
1265 
1266  // adaptive seeds per task based on the total estimated amount of work to divide among all threads
1267  const int adaptiveSPT = std::clamp(
1269  dprint("adaptiveSPT " << adaptiveSPT << " fill " << rosi.count() << "/" << eoccs.size() << " region " << region);
1270 
1271  tbb::parallel_for(rosi.tbb_blk_rng_std(adaptiveSPT), [&](const tbb::blocked_range<int> &cands) {
1272  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
1273 
1274  fit_cands(mkfndr.get(), cands.begin(), cands.end(), region);
1275  });
1276  });
1277  }
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:1279

◆ backwardFitBH()

void mkfit::MkBuilder::backwardFitBH ( )

Definition at line 1165 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().

1165  {
1166  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
1167  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
1168 
1169  tbb::parallel_for(rosi.tbb_blk_rng_vec(), [&](const tbb::blocked_range<int> &blk_rng) {
1170  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
1171 
1172  RangeOfSeedIndices rng = rosi.seed_rng(blk_rng);
1173 
1174  while (rng.valid()) {
1175  // final backward fit
1176  fit_cands_BH(mkfndr.get(), rng.m_beg, rng.m_end, region);
1177 
1178  ++rng;
1179  }
1180  });
1181  });
1182  }
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:1184

◆ 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(), findQualityFiles::size, mkfit::EventOfCombCandidates::size(), and std::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(edm::DataFrameContainer &lhs, edm::DataFrameContainer &rhs)
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 683 of file MkBuilder.cc.

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

Referenced by find_tracks_in_layers(), and findTracksStandard().

690  {
691  // XXXX-1 If I miss a layer, insert the original track into tmp_cands
692  // AND do not do it in FindCandidates as the position can be badly
693  // screwed by then. See comment there, too.
694  // One could also do a pre-check ... so as not to use up a slot.
695 
696  // bool debug = true;
697 
698  for (int ti = itrack; ti < end; ++ti) {
699  TrackCand &cand = m_event_of_comb_cands[seed_cand_idx[ti].first][seed_cand_idx[ti].second];
700  WSR_Result &w = mkfndr->m_XWsrResult[ti - itrack];
701 
702  // XXXX-4 Low pT tracks can miss a barrel layer ... and should be stopped
703  const float cand_r =
704  std::hypot(mkfndr->getPar(ti - itrack, MkBase::iP, 0), mkfndr->getPar(ti - itrack, MkBase::iP, 1));
705 
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 (layer_info.is_barrel() && cand_r < layer_info.rin()) {
715  // Fake outside so it does not get processed in FindTracks Std/CE... and
716  // create a stopped replica in barrel and original copy if there is
717  // still chance to hit endcaps.
718  dprintf("Barrel cand propagated to r=%f ... layer is %f - %f\n", cand_r, layer_info.rin(), layer_info.rout());
719 
720  mkfndr->m_XHitSize[ti - itrack] = 0;
721  w.m_wsr = WSR_Outside;
722 
723  tmp_cands[seed_cand_idx[ti].first - start_seed].push_back(cand);
725  dprintf(" creating extra stopped held back candidate\n");
726  tmp_cands[seed_cand_idx[ti].first - start_seed].back().addHitIdx(-2, layer_info.layer_id(), 0);
727  }
728  } else if (w.m_wsr == WSR_Outside) {
729  dprintf(" creating extra held back candidate\n");
730  tmp_cands[seed_cand_idx[ti].first - start_seed].push_back(cand);
731  } else if (w.m_wsr == WSR_Edge) {
732  // Do nothing special here, this case is handled also in MkFinder:findTracks()
733  }
734  }
735  }
static constexpr int iP
Definition: MkBase.h:19
float rin() const
Definition: TrackerInfo.h:63
T w() const
int layer_id() const
Definition: TrackerInfo.h:61
MPlexQI m_XHitSize
Definition: MkFinder.h:316
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
float rout() const
Definition: TrackerInfo.h:64
WSR_Result m_XWsrResult[NN]
Definition: MkFinder.h:315
#define dprintf(...)
Definition: Debug.h:98
bool is_barrel() const
Definition: TrackerInfo.h:73
float getPar(int itrack, int i, int par) const
Definition: MkBase.h:21

◆ 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 958 of file MkBuilder.cc.

References cms::cuda::assert(), mkfit::CandCloner::begin_eta_bin(), mkfit::CandCloner::begin_iteration(), mkfit::CandCloner::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(), 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().

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

◆ find_tracks_load_seeds()

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

Definition at line 606 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().

606  {
607  // This will sort seeds according to iteration configuration.
608  assert(!in_seeds.empty());
609  m_tracks.clear(); // m_tracks can be used for BkFit.
610 
611  m_event_of_comb_cands.reset((int)in_seeds.size(), m_job->max_max_cands());
612 
613  import_seeds(in_seeds, seeds_sorted, [&](const Track &seed, int region, int pos) {
615  });
616  }
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)
void insertSeed(const Track &seed, const track_score_func &score_func, int region, int pos)
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 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 618 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().

624  {
625  int silly_count = 0;
626 
627  seed_cand_vec.clear();
628 
629  auto &iter_params = (iteration_dir == SteeringParams::IT_BkwSearch) ? m_job->m_iter_config.m_backward_params
631 
632  for (int iseed = start_seed; iseed < end_seed; ++iseed) {
634 
635  if (ccand.state() == CombCandidate::Dormant && ccand.pickupLayer() == prev_layer) {
637  }
638  if (!pickup_only && ccand.state() == CombCandidate::Finding) {
639  bool active = false;
640  for (int ic = 0; ic < (int)ccand.size(); ++ic) {
641  if (ccand[ic].getLastHitIdx() != -2) {
642  // Stop candidates with pT<X GeV
643  if (ccand[ic].pT() < iter_params.minPtCut) {
644  ccand[ic].addHitIdx(-2, layer, 0.0f);
645  continue;
646  }
647  // Check if the candidate is close to it's max_r, pi/2 - 0.2 rad (11.5 deg)
648  if (iteration_dir == SteeringParams::IT_FwdSearch && ccand[ic].pT() < 1.2) {
649  const float dphi = std::abs(ccand[ic].posPhi() - ccand[ic].momPhi());
650  if (ccand[ic].posRsq() > 625.f && dphi > 1.371f && dphi < 4.512f) {
651  // printf("Stopping cand at r=%f, posPhi=%.1f momPhi=%.2f pt=%.2f emomEta=%.2f\n",
652  // ccand[ic].posR(), ccand[ic].posPhi(), ccand[ic].momPhi(), ccand[ic].pT(), ccand[ic].momEta());
653  ccand[ic].addHitIdx(-2, layer, 0.0f);
654  continue;
655  }
656  }
657 
658  active = true;
659  seed_cand_vec.push_back(std::pair<int, int>(iseed, ic));
660  ccand[ic].resetOverlaps();
661 
663  if (ccand[ic].hasSillyValues(Const::nan_n_silly_print_bad_cands_every_layer,
665  "Per layer silly check"))
666  ++silly_count;
667  }
668  }
669  }
670  if (!active) {
672  }
673  }
674  }
675 
676  if (Const::nan_n_silly_check_cands_every_layer && silly_count > 0) {
677  m_nan_n_silly_per_layer_count += silly_count;
678  }
679 
680  return seed_cand_vec.size();
681  }
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(), 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,
519  m_job->m_iter_config.m_layer_configs[curr_layer],
520  st_par,
521  m_job->get_mask_for_layer(curr_layer),
522  m_job->m_in_fwd);
523 
524  const LayerOfHits &layer_of_hits = m_job->m_event_of_hits[curr_layer];
525  const LayerInfo &layer_info = trk_info.layer(curr_layer);
526  const FindingFoos &fnd_foos = FindingFoos::get_finding_foos(layer_info.is_barrel());
527  dprint("at layer " << curr_layer << ", nHits in layer " << layer_of_hits.nHits());
528 
529  // Pick up seeds that become active on current layer -- unless already fully loaded.
530  if (curr_tridx < rng.n_proc()) {
531  int prev_tridx = curr_tridx;
532 
533  for (int i = rng.m_beg, ii = 0; i < rng.m_end; ++i, ++ii) {
534  if (trk_llay[ii] == prev_layer)
535  trk_idcs[curr_tridx++] = i;
536  }
537  if (curr_tridx > prev_tridx) {
538  dprintf("added %d seeds, started with %d\n", curr_tridx - prev_tridx, prev_tridx);
539 
540  mkfndr->inputTracksAndHitIdx(cands, trk_idcs, prev_tridx, curr_tridx, false, prev_tridx);
541  }
542  }
543 
544  if (layer_plan_it.is_pickup_only())
545  continue;
546 
547  dcall(pre_prop_print(curr_layer, mkfndr.get()));
548 
549  mkfndr->clearFailFlag();
550  (mkfndr.get()->*fnd_foos.m_propagate_foo)(
551  layer_info.propagate_to(), curr_tridx, prop_config.finding_inter_layer_pflags);
552 
553  dcall(post_prop_print(curr_layer, mkfndr.get()));
554 
555  mkfndr->selectHitIndices(layer_of_hits, curr_tridx);
556 
557  // Stop low-pT tracks that can not reach the current barrel layer.
558  if (layer_info.is_barrel()) {
559  const float r_min_sqr = layer_info.rin() * layer_info.rin();
560  for (int i = 0; i < curr_tridx; ++i) {
561  if (!mkfndr->m_Stopped[i]) {
562  if (mkfndr->radiusSqr(i, MkBase::iP) < r_min_sqr) {
564  mkfndr->m_Stopped[i] = 1;
565  mkfndr->outputTrackAndHitIdx(cands[rng.m_beg + i], i, false);
566  }
567  mkfndr->m_XWsrResult[i].m_wsr = WSR_Outside;
568  mkfndr->m_XHitSize[i] = 0;
569  }
570  } else { // make sure we don't add extra work for AddBestHit
571  mkfndr->m_XWsrResult[i].m_wsr = WSR_Outside;
572  mkfndr->m_XHitSize[i] = 0;
573  }
574  }
575  }
576 
577  // make candidates with best hit
578  dprint("make new candidates");
579 
580  mkfndr->addBestHit(layer_of_hits, curr_tridx, fnd_foos);
581 
582  // Stop tracks that have reached N_max_holes.
583  for (int i = 0; i < curr_tridx; ++i) {
584  if (!mkfndr->m_Stopped[i] && mkfndr->bestHitLastHoT(i).index == -2) {
585  mkfndr->m_Stopped[i] = 1;
586  mkfndr->outputTrackAndHitIdx(cands[rng.m_beg + i], i, false);
587  }
588  }
589 
590  } // end of layer loop
591 
592  mkfndr->outputNonStoppedTracksAndHitIdx(cands, trk_idcs, 0, curr_tridx, false);
593 
594  ++rng;
595  } // end of loop over candidates in a tbb chunk
596 
597  mkfndr->release();
598  }); // end parallel_for over candidates in a region
599  }); // end of parallel_for_each over regions
600  }
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:63
const auto regions_begin() const
Definition: MkJob.h:22
float propagate_to() const
Definition: TrackerInfo.h:69
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
ii
Definition: cuy.py:589
const LayerInfo & layer(int l) const
Definition: TrackerInfo.h:199
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:73
const PropagationConfig & prop_config() const
Definition: TrackerInfo.h:212

◆ findTracksCloneEngine()

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

Definition at line 923 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().

923  {
924  // debug = true;
925 
927 
928  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
929  if (iteration_dir == SteeringParams::IT_BkwSearch && !m_job->steering_params(region).has_bksearch_plan()) {
930  printf("No backward search plan for region %d\n", region);
931  return;
932  }
933 
934  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
935 
936  // adaptive seeds per task based on the total estimated amount of work to divide among all threads
937  const int adaptiveSPT = std::clamp(
939  dprint("adaptiveSPT " << adaptiveSPT << " fill " << rosi.count() << "/" << eoccs.size() << " region " << region);
940 
941  tbb::parallel_for(rosi.tbb_blk_rng_std(adaptiveSPT), [&](const tbb::blocked_range<int> &seeds) {
942  auto cloner = g_exe_ctx.m_cloners.makeOrGet();
943  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
944 
945  cloner->setup(m_job->params());
946 
947  // loop over layers
948  find_tracks_in_layers(*cloner, mkfndr.get(), iteration_dir, seeds.begin(), seeds.end(), region);
949 
950  mkfndr->release();
951  cloner->release();
952  });
953  });
954 
955  // debug = false;
956  }
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:958

◆ findTracksStandard()

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

Definition at line 741 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_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, findQualityFiles::size, mkfit::EventOfCombCandidates::size(), jetUpdater_cfi::sort, mkfit::MkJob::steering_params(), and gpuVertexFinder::while().

Referenced by mkfit::runBuildingTestPlexStandard().

741  {
742  // debug = true;
743 
745 
746  tbb::parallel_for_each(m_job->regions_begin(), m_job->regions_end(), [&](int region) {
747  if (iteration_dir == SteeringParams::IT_BkwSearch && !m_job->steering_params(region).has_bksearch_plan()) {
748  printf("No backward search plan for region %d\n", region);
749  return;
750  }
751 
752  const TrackerInfo &trk_info = m_job->m_trk_info;
753  const SteeringParams &st_par = m_job->steering_params(region);
754  const IterationParams &params = m_job->params();
755  const PropagationConfig &prop_config = trk_info.prop_config();
756 
757  const RegionOfSeedIndices rosi(m_seedEtaSeparators, region);
758 
759  // adaptive seeds per task based on the total estimated amount of work to divide among all threads
760  const int adaptiveSPT = std::clamp(
762  dprint("adaptiveSPT " << adaptiveSPT << " fill " << rosi.count() << "/" << eoccs.size() << " region " << region);
763 
764  // loop over seeds
765  tbb::parallel_for(rosi.tbb_blk_rng_std(adaptiveSPT), [&](const tbb::blocked_range<int> &seeds) {
766  auto mkfndr = g_exe_ctx.m_finders.makeOrGet();
767 
768  const int start_seed = seeds.begin();
769  const int end_seed = seeds.end();
770  const int n_seeds = end_seed - start_seed;
771 
772  std::vector<std::vector<TrackCand>> tmp_cands(n_seeds);
773  for (size_t iseed = 0; iseed < tmp_cands.size(); ++iseed) {
774  tmp_cands[iseed].reserve(2 * params.maxCandsPerSeed); //factor 2 seems reasonable to start with
775  }
776 
777  std::vector<std::pair<int, int>> seed_cand_idx;
778  seed_cand_idx.reserve(n_seeds * params.maxCandsPerSeed);
779 
780  auto layer_plan_it = st_par.make_iterator(iteration_dir);
781 
782  dprintf("Made iterator for %d, first layer=%d ... end layer=%d\n",
783  iteration_dir,
784  layer_plan_it.layer(),
785  layer_plan_it.last_layer());
786 
787  assert(layer_plan_it.is_pickup_only());
788 
789  int curr_layer = layer_plan_it.layer(), prev_layer;
790 
791  dprintf("\nMkBuilder::FindTracksStandard region=%d, seed_pickup_layer=%d, first_layer=%d\n",
792  region,
793  curr_layer,
794  layer_plan_it.next_layer());
795 
796  auto &iter_params = (iteration_dir == SteeringParams::IT_BkwSearch) ? m_job->m_iter_config.m_backward_params
798 
799  // Loop over layers, starting from after the seed.
800  while (++layer_plan_it) {
801  prev_layer = curr_layer;
802  curr_layer = layer_plan_it.layer();
803  mkfndr->setup(prop_config,
804  iter_params,
805  m_job->m_iter_config.m_layer_configs[curr_layer],
806  st_par,
807  m_job->get_mask_for_layer(curr_layer),
808  m_job->m_in_fwd);
809 
810  dprintf("\n* Processing layer %d\n", curr_layer);
811 
812  const LayerOfHits &layer_of_hits = m_job->m_event_of_hits[curr_layer];
813  const LayerInfo &layer_info = trk_info.layer(curr_layer);
814  const FindingFoos &fnd_foos = FindingFoos::get_finding_foos(layer_info.is_barrel());
815 
816  int theEndCand = find_tracks_unroll_candidates(seed_cand_idx,
817  start_seed,
818  end_seed,
819  curr_layer,
820  prev_layer,
821  layer_plan_it.is_pickup_only(),
822  iteration_dir);
823 
824  dprintf(" Number of candidates to process: %d, nHits in layer: %d\n", theEndCand, layer_of_hits.nHits());
825 
826  if (layer_plan_it.is_pickup_only() || theEndCand == 0)
827  continue;
828 
829  // vectorized loop
830  for (int itrack = 0; itrack < theEndCand; itrack += NN) {
831  int end = std::min(itrack + NN, theEndCand);
832 
833  dprint("processing track=" << itrack << ", label="
834  << eoccs[seed_cand_idx[itrack].first][seed_cand_idx[itrack].second].label());
835 
836  //fixme find a way to deal only with the candidates needed in this thread
837  mkfndr->inputTracksAndHitIdx(eoccs.refCandidates(), seed_cand_idx, itrack, end, false);
838 
839  //propagate to layer
840  dcall(pre_prop_print(curr_layer, mkfndr.get()));
841 
842  mkfndr->clearFailFlag();
843  (mkfndr.get()->*fnd_foos.m_propagate_foo)(
844  layer_info.propagate_to(), end - itrack, prop_config.finding_inter_layer_pflags);
845 
846  dcall(post_prop_print(curr_layer, mkfndr.get()));
847 
848  dprint("now get hit range");
849  mkfndr->selectHitIndices(layer_of_hits, end - itrack);
850 
852  mkfndr.get(), layer_info, tmp_cands, seed_cand_idx, region, start_seed, itrack, end);
853 
854  dprint("make new candidates");
855  mkfndr->findCandidates(layer_of_hits, tmp_cands, start_seed, end - itrack, fnd_foos);
856 
857  } //end of vectorized loop
858 
859  // sort the input candidates
860  for (int is = 0; is < n_seeds; ++is) {
861  dprint("dump seed n " << is << " with N_input_candidates=" << tmp_cands[is].size());
862 
863  std::sort(tmp_cands[is].begin(), tmp_cands[is].end(), sortCandByScore);
864  }
865 
866  // now fill out the output candidates
867  for (int is = 0; is < n_seeds; ++is) {
868  if (!tmp_cands[is].empty()) {
869  eoccs[start_seed + is].clear();
870 
871  // Put good candidates into eoccs, process -2 candidates.
872  int n_placed = 0;
873  bool first_short = true;
874  for (int ii = 0; ii < (int)tmp_cands[is].size() && n_placed < params.maxCandsPerSeed; ++ii) {
875  TrackCand &tc = tmp_cands[is][ii];
876 
877  // See if we have an overlap hit available, but only if we have a true hit in this layer
878  // and pT is above the pTCutOverlap
879  if (tc.pT() > params.pTCutOverlap && tc.getLastHitLyr() == curr_layer && tc.getLastHitIdx() >= 0) {
880  CombCandidate &ccand = eoccs[start_seed + is];
881 
882  HitMatch *hm = ccand[tc.originIndex()].findOverlap(
883  tc.getLastHitIdx(), layer_of_hits.refHit(tc.getLastHitIdx()).detIDinLayer());
884 
885  if (hm) {
886  tc.addHitIdx(hm->m_hit_idx, curr_layer, hm->m_chi2);
887  tc.incOverlapCount();
888  }
889  }
890 
891  if (tc.getLastHitIdx() != -2) {
892  eoccs[start_seed + is].emplace_back(tc);
893  ++n_placed;
894  } else if (first_short) {
895  first_short = false;
896  if (tc.score() > eoccs[start_seed + is].refBestShortCand().score()) {
897  eoccs[start_seed + is].setBestShortCand(tc);
898  }
899  }
900  }
901 
902  tmp_cands[is].clear();
903  }
904  }
905 
906  } // end of layer loop
907  mkfndr->release();
908 
909  // final sorting
910  for (int iseed = start_seed; iseed < end_seed; ++iseed) {
911  eoccs[iseed].mergeCandsAndBestShortOne(m_job->params(), st_par.m_track_scorer, true, true);
912  }
913  }); // end parallel-for over chunk of seeds within region
914  }); // end of parallel-for-each over eta regions
915 
916  // debug = false;
917  }
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:618
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:69
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
constexpr int numThreadsFinder
Definition: Config.h:85
ii
Definition: cuy.py:589
const LayerInfo & layer(int l) const
Definition: TrackerInfo.h:199
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:683
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:73
const PropagationConfig & prop_config() const
Definition: TrackerInfo.h:212
const Hit & refHit(int i) const
Definition: HitStructures.h:97

◆ fit_cands()

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

Definition at line 1279 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().

1279  {
1281  const SteeringParams &st_par = m_job->steering_params(region);
1282  const PropagationConfig &prop_config = m_job->m_trk_info.prop_config();
1283  mkfndr->setup_bkfit(prop_config, st_par);
1284 
1285  int step = NN;
1286  for (int icand = start_cand; icand < end_cand; icand += step) {
1287  int end = std::min(icand + NN, end_cand);
1288 
1289  bool chi_debug = false;
1290 
1291 #ifdef DEBUG_FINAL_FIT
1292  bool debug = true;
1293  dprintf("Pre Final fit for %d - %d\n", icand, end);
1294  for (int i = icand; i < end; ++i) {
1295  dprint_tcand(eoccs[i][0], i);
1296  }
1297  chi_debug = true;
1298  static bool first = true;
1299  if (first) {
1300  // ./mkFit ... | perl -ne 'if (/^BKF_OVERLAP/) { s/^BKF_OVERLAP //og; print; }' > bkf_ovlp.rtt
1301  dprintf(
1302  "BKF_OVERLAP event/I:label/I:prod_type/I:is_findable/I:layer/I:is_stereo/I:is_barrel/I:"
1303  "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:"
1304  "sx_t/F:sy_t/F:sz_t/F:d_xy/F:d_z/F\n");
1305  first = false;
1306  }
1307  mkfndr->m_event = m_event;
1308 #endif
1309 
1310  // input tracks
1311  mkfndr->bkFitInputTracks(eoccs, icand, end);
1312 
1313  // fit tracks back to first layer
1314  mkfndr->bkFitFitTracks(m_job->m_event_of_hits, st_par, end - icand, chi_debug);
1315 
1316  // now move one last time to PCA
1317  if (prop_config.backward_fit_to_pca) {
1318  mkfndr->bkFitPropTracksToPCA(end - icand);
1319  }
1320 
1321  mkfndr->bkFitOutputTracks(eoccs, icand, end, prop_config.backward_fit_to_pca);
1322 
1323 #ifdef DEBUG_FINAL_FIT
1324  dprintf("Post Final fit for %d - %d\n", icand, end);
1325  for (int i = icand; i < end; ++i) {
1326  dprint_tcand(eoccs[i][0], i);
1327  }
1328 #endif
1329  }
1330  mkfndr->release();
1331  }
void bkFitInputTracks(TrackVec &cands, int beg, int end)
Definition: MkFinder.cc:1411
void setup_bkfit(const PropagationConfig &pc, const SteeringParams &sp)
Definition: MkFinder.cc:35
void bkFitPropTracksToPCA(const int N_proc)
Definition: MkFinder.cc:1897
void release()
Definition: MkFinder.cc:40
const EventOfHits & m_event_of_hits
Definition: MkJob.h:13
constexpr Matriplex::idx_t NN
Definition: Matrix.h:43
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:1669
EventOfCombCandidates m_event_of_comb_cands
Definition: MkBuilder.h:130
void bkFitOutputTracks(TrackVec &cands, int beg, int end, bool outputProp)
Definition: MkFinder.cc:1468
step
Definition: StallMonitor.cc:98
#define dprintf(...)
Definition: Debug.h:98
const PropagationConfig & prop_config() const
Definition: TrackerInfo.h:212

◆ fit_cands_BH()

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

Definition at line 1184 of file MkBuilder.cc.

References Matriplex::Matriplex< T, D1, D2, N >::At(), 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, and mkfit::MkJob::steering_params().

Referenced by backwardFitBH().

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

◆ 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(), fileCollector::seed, and findQualityFiles::size.

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, 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, findQualityFiles::size, 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

Definition at line 124 of file MkBuilder.h.

Referenced by begin_event(), end_event(), find_tracks_in_layers(), fit_cands(), and fit_cands_BH().

◆ 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