mkfit::Event Class Reference

#include <Event.h>

Public Member Functions
int	clean_cms_seedtracks (TrackVec *seed_ptr=nullptr)
int	clean_cms_seedtracks_badlabel ()
int	clean_cms_simtracks ()
	Event (int evtID, int nLayers)
	Event (Validation &v, int evtID, int nLayers)
int	evtID () const
void	fill_hitmask_bool_vectors (int track_algo, std::vector< std::vector< bool >> &layer_masks)
void	fill_hitmask_bool_vectors (std::vector< int > &track_algo_vec, std::vector< std::vector< bool >> &layer_masks)
void	kludge_cms_hit_errors ()
void	print_tracks (const TrackVec &tracks, bool print_hits) const
void	printStats (const TrackVec &, TrackExtraVec &)
void	read_in (DataFile &data_file, FILE *in_fp=0)
int	read_tracks (FILE *fp, TrackVec &tracks, bool skip_reading=false)
void	relabel_bad_seedtracks ()
void	relabel_cmsswtracks_from_seeds ()
void	reset (int evtID)
void	resetLayerHitMap (bool resetSimHits)
int	select_tracks_iter (unsigned int n=0)
void	setInputFromCMSSW (std::vector< HitVec > hits, TrackVec seeds)
int	use_seeds_from_cmsswtracks ()
void	validate ()
void	write_out (DataFile &data_file)
int	write_tracks (FILE *fp, const TrackVec &tracks)

Public Attributes
BeamSpot	beamSpot_
TrackVec	candidateTracks_
TrackExtraVec	candidateTracksExtra_
TrackVec	cmsswTracks_
TrackExtraVec	cmsswTracksExtra_
TrackVec	fitTracks_
TrackExtraVec	fitTracksExtra_
std::vector< std::vector< uint64_t > >	layerHitMasks_
std::vector< HitVec >	layerHits_
TrackVec	seedTracks_
TrackExtraVec	seedTracksExtra_
MCHitInfoVec	simHitsInfo_
TrackVec	simTracks_
TrackExtraVec	simTracksExtra_
TSVec	simTrackStates_
Validation &	validation_

Static Public Attributes
static std::mutex	printmutex

Private Attributes
int	evtID_

Detailed Description

Definition at line 14 of file Event.h.

Constructor & Destructor Documentation

Event() [1/2]

Event::Event ( int  evtID, int  nLayers  )

explicit

Definition at line 21 of file Event.cc.

References layerHitMasks_, layerHits_, and MuonTCMETValueMapProducer_cff::nLayers.

                                     : validation_(*dummyValidation), evtID_(evtID) {
    layerHits_.resize(nLayers);
    layerHitMasks_.resize(nLayers);
  }

Event() [2/2]

Event::Event	(	Validation &	v,
		int	evtID,
		int	nLayers
	)

Definition at line 26 of file Event.cc.

References layerHitMasks_, layerHits_, MuonTCMETValueMapProducer_cff::nLayers, mkfit::Validation::resetValidationMaps(), and validation_.

                                                    : validation_(v), evtID_(evtID) {
    layerHits_.resize(nLayers);
    layerHitMasks_.resize(nLayers);
    validation_.resetValidationMaps();  // need to reset maps for every event.
  }

Member Function Documentation

clean_cms_seedtracks()

int Event::clean_cms_seedtracks

(

TrackVec *

seed_ptr = nullptr

)

Definition at line 578 of file Event.cc.

References funct::abs(), mkfit::Config::c_dpt_common, mkfit::Config::c_drmax_brl, mkfit::Config::c_drmax_els, mkfit::Config::c_drmax_hpt, mkfit::Config::c_dzmax_brl, mkfit::Config::c_dzmax_els, mkfit::Config::c_dzmax_hpt, mkfit::Config::c_etamax_brl, mkfit::Config::c_ptmin_hpt, mkfit::cdist(), ALCARECOTkAlJpsiMuMu_cff::charge, mkfit::TrackBase::charge(), PVValHelper::eta, f, mkfit::TrackBase::invpT(), mkfit::TrackBase::momEta(), mkfit::TrackBase::momPhi(), mkfit::Track::nFoundHits(), nHits, funct::pow(), DiDispStaMuonMonitor_cfi::pt, mkfit::TrackBase::pT(), DetachedQuadStep_cff::seeds, seedTracks_, mathSSE::sqrt(), contentValuesCheck::ss, funct::tan(), mkfit::TrackBase::theta(), mkfit::Config::track1GeVradius, x, mkfit::TrackBase::x(), mkfit::TrackBase::y(), and mkfit::TrackBase::z().

                                                    {
    const float etamax_brl = Config::c_etamax_brl;
    const float dpt_common = Config::c_dpt_common;
    const float dzmax_brl = Config::c_dzmax_brl;
    const float drmax_brl = Config::c_drmax_brl;
    const float ptmin_hpt = Config::c_ptmin_hpt;
    const float dzmax_hpt = Config::c_dzmax_hpt;
    const float drmax_hpt = Config::c_drmax_hpt;
    const float dzmax_els = Config::c_dzmax_els;
    const float drmax_els = Config::c_drmax_els;

    const float dzmax2_inv_brl = 1.f / (dzmax_brl * dzmax_brl);
    const float drmax2_inv_brl = 1.f / (drmax_brl * drmax_brl);
    const float dzmax2_inv_hpt = 1.f / (dzmax_hpt * dzmax_hpt);
    const float drmax2_inv_hpt = 1.f / (drmax_hpt * drmax_hpt);
    const float dzmax2_inv_els = 1.f / (dzmax_els * dzmax_els);
    const float drmax2_inv_els = 1.f / (drmax_els * drmax_els);

    TrackVec &seeds = (seed_ptr != nullptr) ? *seed_ptr : seedTracks_;
    const int ns = seeds.size();

    TrackVec cleanSeedTracks;
    cleanSeedTracks.reserve(ns);
    std::vector<bool> writetrack(ns, true);

    const float invR1GeV = 1.f / Config::track1GeVradius;

    std::vector<int> nHits(ns);
    std::vector<int> charge(ns);
    std::vector<float> oldPhi(ns);
    std::vector<float> pos2(ns);
    std::vector<float> eta(ns);
    std::vector<float> ctheta(ns);
    std::vector<float> invptq(ns);
    std::vector<float> pt(ns);
    std::vector<float> x(ns);
    std::vector<float> y(ns);
    std::vector<float> z(ns);

    for (int ts = 0; ts < ns; ts++) {
      const Track &tk = seeds[ts];
      nHits[ts] = tk.nFoundHits();
      charge[ts] = tk.charge();
      oldPhi[ts] = tk.momPhi();
      pos2[ts] = std::pow(tk.x(), 2) + std::pow(tk.y(), 2);
      eta[ts] = tk.momEta();
      ctheta[ts] = 1.f / std::tan(tk.theta());
      invptq[ts] = tk.charge() * tk.invpT();
      pt[ts] = tk.pT();
      x[ts] = tk.x();
      y[ts] = tk.y();
      z[ts] = tk.z();
    }

    for (int ts = 0; ts < ns; ts++) {
      if (not writetrack[ts])
        continue;  //FIXME: this speed up prevents transitive masking; check build cost!

      const float oldPhi1 = oldPhi[ts];
      const float pos2_first = pos2[ts];
      const float Eta1 = eta[ts];
      const float Pt1 = pt[ts];
      const float invptq_first = invptq[ts];

      //#pragma simd /* Vectorization via simd had issues with icc */
      for (int tss = ts + 1; tss < ns; tss++) {
        const float Pt2 = pt[tss];

        if (charge[tss] != charge[ts])
          continue;

        const float thisDPt = std::abs(Pt2 - Pt1);
        if (thisDPt > dpt_common * (Pt1))
          continue;

        const float Eta2 = eta[tss];
        const float deta2 = std::pow(Eta1 - Eta2, 2);

        const float oldPhi2 = oldPhi[tss];

        const float pos2_second = pos2[tss];
        const float thisDXYSign05 = pos2_second > pos2_first ? -0.5f : 0.5f;

        const float thisDXY = thisDXYSign05 * sqrt(std::pow(x[ts] - x[tss], 2) + std::pow(y[ts] - y[tss], 2));

        const float invptq_second = invptq[tss];

        const float newPhi1 = oldPhi1 - thisDXY * invR1GeV * invptq_first;
        const float newPhi2 = oldPhi2 + thisDXY * invR1GeV * invptq_second;

        const float dphi = cdist(std::abs(newPhi1 - newPhi2));

        const float dr2 = deta2 + dphi * dphi;

        const float thisDZ = z[ts] - z[tss] - thisDXY * (ctheta[ts] + ctheta[tss]);
        const float dz2 = thisDZ * thisDZ;

        if (std::abs(Eta1) < etamax_brl) {
          if (dz2 * dzmax2_inv_brl + dr2 * drmax2_inv_brl < 1.0f)
            writetrack[tss] = false;
        } else if (Pt1 > ptmin_hpt) {
          if (dz2 * dzmax2_inv_hpt + dr2 * drmax2_inv_hpt < 1.0f)
            writetrack[tss] = false;
        } else {
          if (dz2 * dzmax2_inv_els + dr2 * drmax2_inv_els < 1.0f)
            writetrack[tss] = false;
        }
      }

      if (writetrack[ts])
        cleanSeedTracks.emplace_back(seeds[ts]);
    }

    seeds.swap(cleanSeedTracks);

#ifdef DEBUG
    {
      const int ns2 = seeds.size();
      printf("Number of CMS seeds before %d --> after %d cleaning\n", ns, ns2);

      for (int it = 0; it < ns2; it++) {
        const Track &ss = seeds[it];
        printf("  %3i q=%+i pT=%7.3f eta=% 7.3f nHits=%i label=% i\n",
               it,
               ss.charge(),
               ss.pT(),
               ss.momEta(),
               ss.nFoundHits(),
               ss.label());
      }
    }
#endif

    return seeds.size();
  }

clean_cms_seedtracks_badlabel()

int Event::clean_cms_seedtracks_badlabel

(

)

Definition at line 754 of file Event.cc.

References visDQMUpload::buf, seedTracks_, and submitPVValidationJobs::t.

                                           {
    printf("***\n*** REMOVING SEEDS WITH BAD LABEL. This is a development hack. ***\n***\n");
    TrackVec buf;
    seedTracks_.swap(buf);
    std::copy_if(
        buf.begin(), buf.end(), std::back_inserter(seedTracks_), [](const Track &t) { return t.label() >= 0; });
    return seedTracks_.size();
  }

clean_cms_simtracks()

int Event::clean_cms_simtracks

(

)

Definition at line 497 of file Event.cc.

References mkfit::Config::cmsSelMinLayers, dprintf, mps_fire::i, simTracks_, and submitPVValidationJobs::t.

                                 {
    // Sim tracks from cmssw have the following issues:
    // - hits are not sorted by layer;
    // - there are tracks with too low number of hits, even 0;
    // - even with enough hits, there can be too few layers (esp. in endcap);
    // - tracks from secondaries can have extremely low pT.
    // Possible further checks:
    // - make sure enough hits exist in seeding layers.
    //
    // What is done:
    // 1. Hits are sorted by layer;
    // 2. Non-findable tracks are marked with Track::Status::not_findable flag.
    //
    // Returns number of passed simtracks.

    dprintf("Event::clean_cms_simtracks processing %lu simtracks.\n", simTracks_.size());

    int n_acc = 0;
    int i = -1;  //wrap in ifdef DEBUG?
    for (Track &t : simTracks_) {
      i++;

      t.sortHitsByLayer();

      const int lyr_cnt = t.nUniqueLayers();

      //const int lasthit = t.getLastFoundHitPos();
      //const float eta = layerHits_[t.getHitLyr(lasthit)][t.getHitIdx(lasthit)].eta();

      if (lyr_cnt < Config::cmsSelMinLayers)  // || Config::TrkInfo.is_transition(eta))
      {
        dprintf("Rejecting simtrack %d, n_hits=%d, n_layers=%d, pT=%f\n", i, t.nFoundHits(), lyr_cnt, t.pT());
        t.setNotFindable();
      } else {
        dprintf("Accepting simtrack %d, n_hits=%d, n_layers=%d, pT=%f\n", i, t.nFoundHits(), lyr_cnt, t.pT());
        ++n_acc;
      }
    }

    return n_acc;
  }

evtID()

int mkfit::Event::evtID ( ) const

inline

Definition at line 23 of file Event.h.

References evtID_.

Referenced by mkfit::MkBuilder::fit_cands_BH(), and reset().

{ return evtID_; }

fill_hitmask_bool_vectors() [1/2]

void Event::fill_hitmask_bool_vectors	(	int	track_algo,
		std::vector< std::vector< bool >> &	layer_masks
	)

Definition at line 805 of file Event.cc.

References mps_fire::i, createfilelist::int, cmsLHEtoEOSManager::l, layerHitMasks_, layerHits_, and findQualityFiles::size.

                                                                                               {
    // Convert from per-hit uint64_t to per layer bool-vectors for given
    // iteration.

    uint64_t iter_mask = 1 << track_algo;

    const int n_lay = (int)layerHits_.size();
    layer_masks.resize(n_lay);

    for (int l = 0; l < n_lay; ++l) {
      const int n_hit = (int)layerHits_[l].size();
      layer_masks[l].resize(n_hit);

      for (int i = 0; i < n_hit; ++i) {
        layer_masks[l][i] = layerHitMasks_[l][i] & iter_mask;
      }
    }
  }

fill_hitmask_bool_vectors() [2/2]

void Event::fill_hitmask_bool_vectors	(	std::vector< int > &	track_algo_vec,
		std::vector< std::vector< bool >> &	layer_masks
	)

Definition at line 824 of file Event.cc.

References mps_fire::i, createfilelist::int, cmsLHEtoEOSManager::l, layerHitMasks_, layerHits_, and findQualityFiles::size.

                                                                                                               {
    // Convert from per-hit uint64_t to per layer bool-vectors for a list of
    // iterations.
    // A hit mask is set if it is set for _all_ listed iterations.

    uint64_t iter_mask = 0;
    for (auto ta : track_algo_vec)
      iter_mask |= 1 << ta;

    const int n_lay = (int)layerHits_.size();
    layer_masks.resize(n_lay);

    for (int l = 0; l < n_lay; ++l) {
      const int n_hit = (int)layerHits_[l].size();
      layer_masks[l].resize(n_hit);

      for (int i = 0; i < n_hit; ++i) {
        uint64_t hitmasks = layerHitMasks_[l][i];
        layer_masks[l][i] = ((iter_mask ^ hitmasks) & iter_mask) == 0;
      }
    }
  }

kludge_cms_hit_errors()

void Event::kludge_cms_hit_errors

(

)

Definition at line 462 of file Event.cc.

References c, MillePedeFileConverter_cfg::e, relativeConstraints::empty, h, and layerHits_.

Referenced by read_in().

                                    {
    // Enforce Vxy on all layers, Vz on pixb only.

    const float Exy = 15 * 1e-4, Vxy = Exy * Exy;
    const float Ez = 30 * 1e-4, Vz = Ez * Ez;

    int nl = layerHits_.size();

    int cnt = 0;

    for (int il = 0; il < nl; il++) {
      if (layerHits_[il].empty())
        continue;

      for (Hit &h : layerHits_[il]) {
        SVector6 &c = h.error_nc();

        float vxy = c[0] + c[2];
        if (vxy < Vxy) {
          c[0] *= Vxy / vxy;
          c[2] *= Vxy / vxy;
          ++cnt;
        }
        if (il < 4 && c[5] < Vz) {
          c[5] = Vz;
          ++cnt;
        }
      }
    }

    printf("Event::kludge_cms_hit_errors processed %d layers, kludged %d entries.\n", nl, cnt);
  }

print_tracks()

void Event::print_tracks	(	const TrackVec &	tracks,
		bool	print_hits
	)		const

Definition at line 539 of file Event.cc.

References mkfit::getScoreCand(), heavyIonCSV_trainingSettings::idx, layerHits_, nt, alignCSCRings::r, simHitsInfo_, submitPVValidationJobs::t, and tracks.

                                                                        {
    const int nt = tracks.size();

    //WARNING: Printouts for hits will not make any sense if mkFit is not run with a validation flag such as --quality-val
    printf("Event::print_tracks printing %d tracks %s hits:\n", nt, (print_hits ? "with" : "without"));
    for (int it = 0; it < nt; it++) {
      const Track &t = tracks[it];
      printf("  %i with q=%+i pT=%7.3f eta=% 7.3f nHits=%2d  label=%4d findable=%d score=%7.3f chi2=%7.3f\n",
             it,
             t.charge(),
             t.pT(),
             t.momEta(),
             t.nFoundHits(),
             t.label(),
             t.isFindable(),
             getScoreCand(t),
             t.chi2());

      if (print_hits) {
        for (int ih = 0; ih < t.nTotalHits(); ++ih) {
          int lyr = t.getHitLyr(ih);
          int idx = t.getHitIdx(ih);
          if (idx >= 0) {
            const Hit &hit = layerHits_[lyr][idx];
            printf("    hit %2d lyr=%2d idx=%3d pos r=%7.3f z=% 8.3f   mc_hit=%3d mc_trk=%3d\n",
                   ih,
                   lyr,
                   idx,
                   layerHits_[lyr][idx].r(),
                   layerHits_[lyr][idx].z(),
                   hit.mcHitID(),
                   hit.mcTrackID(simHitsInfo_));
          } else
            printf("    hit %2d lyr=%2d idx=%3d\n", ih, t.getHitLyr(ih), t.getHitIdx(ih));
        }
      }
    }
  }

printStats()

void Event::printStats	(	const TrackVec &	trks,
		TrackExtraVec &	trkextras
	)

Definition at line 90 of file Event.cc.

References gather_cfg::cout, newFWLiteAna::found, layerHits_, simHitsInfo_, simTracks_, and hltEgammaHLTExtra_cfi::trks.

                                                                       {
    int miss(0), found(0), fp_10(0), fp_20(0), hit8(0), h8_10(0), h8_20(0);

    for (auto &&trk : trks) {
      auto &&extra = trkextras[trk.label()];
      extra.setMCTrackIDInfo(trk, layerHits_, simHitsInfo_, simTracks_, false, true);
      if (extra.mcTrackID() < 0) {
        ++miss;
      } else {
        auto &&mctrk = simTracks_[extra.mcTrackID()];
        auto pr = trk.pT() / mctrk.pT();
        found++;
        bool h8 = trk.nFoundHits() >= 8;
        bool pt10 = pr > 0.9 && pr < 1.1;
        bool pt20 = pr > 0.8 && pr < 1.2;
        fp_10 += pt10;
        fp_20 += pt20;
        hit8 += h8;
        h8_10 += h8 && pt10;
        h8_20 += h8 && pt20;
      }
    }
    std::cout << "found tracks=" << found << "  in pT 10%=" << fp_10 << "  in pT 20%=" << fp_20
              << "     no_mc_assoc=" << miss << std::endl
              << "  nH >= 8   =" << hit8 << "  in pT 10%=" << h8_10 << "  in pT 20%=" << h8_20 << std::endl;
  }

read_in()

void Event::read_in	(	DataFile &	data_file,
		FILE *	in_fp = 0
	)

Definition at line 206 of file Event.cc.

References mkfit::DataFile::advancePosToNextEvent(), beamSpot_, mkfit::cmsswSeeds, cmsswTracks_, relativeConstraints::empty, mkfit::DataFile::f_fp, personalPlayback::fp, mkfit::DataFile::hasBeamSpot(), mkfit::DataFile::hasCmsswTracks(), mkfit::DataFile::hasHitIterMasks(), mkfit::DataFile::hasSeeds(), mkfit::DataFile::hasSimTrackStates(), heavyIonCSV_trainingSettings::idx, createfilelist::int, kludge_cms_hit_errors(), mkfit::Config::kludgeCmsHitErrors, layerHitMasks_, layerHits_, cms::cuda::nh, nt, mkfit::Config::nTracks, alignCSCRings::r, read_tracks(), mkfit::Config::readCmsswTracks, mkfit::Config::seedInput, seedTracks_, mkfit::Config::silent, simHitsInfo_, simTracks_, simTrackStates_, findQualityFiles::size, mathSSE::sqrt(), contentValuesCheck::ss, submitPVValidationJobs::t, funct::void, x, hit::x, hit::y, and hit::z.

                                                      {
    FILE *fp = in_fp ? in_fp : data_file.f_fp;

    data_file.advancePosToNextEvent(fp);

    int nt = read_tracks(fp, simTracks_);
    Config::nTracks = nt;

    if (data_file.hasSimTrackStates()) {
      int nts;
      fread(&nts, sizeof(int), 1, fp);
      simTrackStates_.resize(nts);
      fread(&simTrackStates_[0], sizeof(TrackState), nts, fp);
    }

    int nl;
    fread(&nl, sizeof(int), 1, fp);
    layerHits_.resize(nl);
    layerHitMasks_.resize(nl);
    for (int il = 0; il < nl; ++il) {
      int nh;
      fread(&nh, sizeof(int), 1, fp);
      layerHits_[il].resize(nh);
      layerHitMasks_[il].resize(nh, 0);  //init to 0 by default
      fread(&layerHits_[il][0], sizeof(Hit), nh, fp);
    }

    if (data_file.hasHitIterMasks()) {
      for (int il = 0; il < nl; ++il) {
        int nh = layerHits_[il].size();
        fread(&layerHitMasks_[il][0], sizeof(uint64_t), nh, fp);
      }
    }

    int nm;
    fread(&nm, sizeof(int), 1, fp);
    simHitsInfo_.resize(nm);
    fread(&simHitsInfo_[0], sizeof(MCHitInfo), nm, fp);

    if (data_file.hasSeeds()) {
      int ns = read_tracks(fp, seedTracks_, Config::seedInput != cmsswSeeds);
      (void)ns;

#ifdef DUMP_SEEDS
      printf("Read %i seedtracks (neg value means actual reading was skipped)\n", ns);
      for (int it = 0; it < ns; it++) {
        const Track &ss = seedTracks_[it];
        printf("  %3i q=%+i pT=%7.3f eta=% 7.3f nHits=%i label=%4i algo=%2i\n",
               it,
               ss.charge(),
               ss.pT(),
               ss.momEta(),
               ss.nFoundHits(),
               ss.label(),
               (int)ss.algorithm());
#ifdef DUMP_SEED_HITS
        for (int ih = 0; ih < seedTracks_[it].nTotalHits(); ++ih) {
          int lyr = seedTracks_[it].getHitLyr(ih);
          int idx = seedTracks_[it].getHitIdx(ih);
          if (idx >= 0) {
            const Hit &hit = layerHits_[lyr][idx];
            printf("    hit %2d lyr=%3d idx=%4d pos r=%7.3f z=% 8.3f   mc_hit=%3d mc_trk=%3d\n",
                   ih,
                   lyr,
                   idx,
                   layerHits_[lyr][idx].r(),
                   layerHits_[lyr][idx].z(),
                   hit.mcHitID(),
                   hit.mcTrackID(simHitsInfo_));
          } else
            printf("    hit %2d idx=%i\n", ih, seedTracks_[it].getHitIdx(ih));
        }
#endif
      }
#endif
    }

    int nert = -99999;
    if (data_file.hasCmsswTracks()) {
      nert = read_tracks(fp, cmsswTracks_, !Config::readCmsswTracks);
      (void)nert;
    }

    /*
    // HACK TO ONLY SELECT ONE PROBLEMATIC TRACK.
    // Note that MC matching gets screwed.
    // Works for MC seeding.
    //
    printf("************** SIM SELECTION HACK IN FORCE ********************\n");
    TrackVec x;
    x.push_back(simTracks_[3]);
    simTracks_.swap(x);
    nt = 1;
  */

#ifdef DUMP_TRACKS
    printf("Read %i simtracks\n", nt);
    for (int it = 0; it < nt; it++) {
      const Track &t = simTracks_[it];
      printf("  %3i q=%+i pT=%7.3f eta=% 7.3f nHits=%2d  label=%4d\n",
             it,
             t.charge(),
             t.pT(),
             t.momEta(),
             t.nFoundHits(),
             t.label());
#ifdef DUMP_TRACK_HITS
      for (int ih = 0; ih < t.nTotalHits(); ++ih) {
        int lyr = t.getHitLyr(ih);
        int idx = t.getHitIdx(ih);
        if (idx >= 0) {
          const Hit &hit = layerHits_[lyr][idx];
          printf("    hit %2d lyr=%2d idx=%3d pos r=%7.3f x=% 8.3f y=% 8.3f z=% 8.3f   mc_hit=%3d mc_trk=%3d\n",
                 ih,
                 lyr,
                 idx,
                 layerHits_[lyr][idx].r(),
                 layerHits_[lyr][idx].x(),
                 layerHits_[lyr][idx].y(),
                 layerHits_[lyr][idx].z(),
                 hit.mcHitID(),
                 hit.mcTrackID(simHitsInfo_));
        } else
          printf("    hit %2d idx=%i\n", ih, t.getHitIdx(ih));
      }
#endif
    }
#endif
#ifdef DUMP_LAYER_HITS
    printf("Read %i layers\n", nl);
    int total_hits = 0;
    for (int il = 0; il < nl; il++) {
      if (layerHits_[il].empty())
        continue;

      printf("Read %i hits in layer %i\n", (int)layerHits_[il].size(), il);
      total_hits += layerHits_[il].size();
      for (int ih = 0; ih < (int)layerHits_[il].size(); ih++) {
        const Hit &hit = layerHits_[il][ih];
        printf("  mcHitID=%5d r=%10g x=%10g y=%10g z=%10g  sx=%10.4g sy=%10.4e sz=%10.4e\n",
               hit.mcHitID(),
               hit.r(),
               hit.x(),
               hit.y(),
               hit.z(),
               std::sqrt(hit.exx()),
               std::sqrt(hit.eyy()),
               std::sqrt(hit.ezz()));
      }
    }
    printf("Total hits in all layers = %d\n", total_hits);
#endif
#ifdef DUMP_REC_TRACKS
    printf("Read %i rectracks\n", nert);
    for (int it = 0; it < nert; it++) {
      const Track &t = cmsswTracks_[it];
      printf("  %i with q=%+i pT=%7.3f eta=% 7.3f nHits=%2d  label=%4d algo=%2d\n",
             it,
             t.charge(),
             t.pT(),
             t.momEta(),
             t.nFoundHits(),
             t.label(),
             (int)t.algorithm());
#ifdef DUMP_REC_TRACK_HITS
      for (int ih = 0; ih < t.nTotalHits(); ++ih) {
        int lyr = t.getHitLyr(ih);
        int idx = t.getHitIdx(ih);
        if (idx >= 0) {
          const Hit &hit = layerHits_[lyr][idx];
          printf("    hit %2d lyr=%2d idx=%3d pos r=%7.3f z=% 8.3f   mc_hit=%3d mc_trk=%3d\n",
                 ih,
                 lyr,
                 idx,
                 hit.r(),
                 hit.z(),
                 hit.mcHitID(),
                 hit.mcTrackID(simHitsInfo_));
        } else
          printf("    hit %2d        idx=%i\n", ih, t.getHitIdx(ih));
      }
#endif
    }
#endif

    if (data_file.hasBeamSpot()) {
      fread(&beamSpot_, sizeof(BeamSpot), 1, fp);
    }

    if (Config::kludgeCmsHitErrors) {
      kludge_cms_hit_errors();
    }

    if (!Config::silent)
      printf("Read complete, %d simtracks on file.\n", nt);
  }

read_tracks()

int Event::read_tracks	(	FILE *	fp,
		TrackVec &	tracks,
		bool	skip_reading = false
	)

Definition at line 429 of file Event.cc.

References personalPlayback::fp, mps_fire::i, and tracks.

Referenced by read_in().

                                                                      {
    // Returns number of read tracks (negative if actual reading was skipped).

    int n_tracks, data_size;
    fread(&n_tracks, sizeof(int), 1, fp);
    fread(&data_size, sizeof(int), 1, fp);

    if (skip_reading) {
      fseek(fp, data_size - 2 * sizeof(int), SEEK_CUR);  // -2 because data_size counts itself and n_tracks too
      n_tracks = -n_tracks;
    } else {
      tracks.resize(n_tracks);

      fread(tracks.data(), sizeof(Track), n_tracks, fp);

      for (int i = 0; i < n_tracks; ++i) {
        tracks[i].resizeHitsForInput();
        fread(tracks[i].beginHitsOnTrack_nc(), sizeof(HitOnTrack), tracks[i].nTotalHits(), fp);
      }
    }

    return n_tracks;
  }

relabel_bad_seedtracks()

void Event::relabel_bad_seedtracks

(

)

Definition at line 778 of file Event.cc.

References seedTracks_, and HLT_2022v12_cff::track.

                                     {
    int newlabel = 0;
    for (auto &&track : seedTracks_) {
      if (track.label() < 0)
        track.setLabel(--newlabel);
    }
  }

relabel_cmsswtracks_from_seeds()

void Event::relabel_cmsswtracks_from_seeds

(

)

Definition at line 786 of file Event.cc.

References cmsswTracks_, iseed, label, and seedTracks_.

                                             {
    std::map<int, int> cmsswLabelMap;
    for (size_t iseed = 0; iseed < seedTracks_.size(); iseed++) {
      for (size_t icmssw = 0; icmssw < cmsswTracks_.size(); icmssw++) {
        if (cmsswTracks_[icmssw].label() == static_cast<int>(iseed)) {
          cmsswLabelMap[icmssw] = seedTracks_[iseed].label();
          break;
        }
      }
    }
    for (size_t icmssw = 0; icmssw < cmsswTracks_.size(); icmssw++) {
      cmsswTracks_[icmssw].setLabel(cmsswLabelMap[icmssw]);
    }
  }

reset()

void Event::reset

(

int

evtID

)

Definition at line 32 of file Event.cc.

References beamSpot_, candidateTracks_, candidateTracksExtra_, cmsswTracks_, cmsswTracksExtra_, evtID(), evtID_, fitTracks_, fitTracksExtra_, cmsLHEtoEOSManager::l, layerHitMasks_, layerHits_, mkfit::Validation::resetValidationMaps(), seedTracks_, seedTracksExtra_, simHitsInfo_, simTracks_, simTracksExtra_, simTrackStates_, and validation_.

                             {
    evtID_ = evtID;

    for (auto &&l : layerHits_) {
      l.clear();
    }
    for (auto &&l : layerHitMasks_) {
      l.clear();
    }

    simHitsInfo_.clear();
    simTrackStates_.clear();
    simTracks_.clear();
    simTracksExtra_.clear();
    seedTracks_.clear();
    seedTracksExtra_.clear();
    candidateTracks_.clear();
    candidateTracksExtra_.clear();
    fitTracks_.clear();
    fitTracksExtra_.clear();
    cmsswTracks_.clear();
    cmsswTracksExtra_.clear();
    beamSpot_ = {};

    validation_.resetValidationMaps();  // need to reset maps for every event.
  }

resetLayerHitMap()

void mkfit::Event::resetLayerHitMap

(

bool

resetSimHits

)

select_tracks_iter()

int Event::select_tracks_iter

(

unsigned int

n = 0

)

Definition at line 718 of file Event.cc.

References mkfit::TrackBase::algorithm(), analysisFilters_cff::algorithms, cmsswTracks_, spr::find(), createfilelist::int, dqmiodumpmetadata::n, nt, and seedTracks_.

                                              {
    if (n == 0)
      return 1;

    unsigned int algorithms[] = {4, 22, 23, 5, 24, 7, 8, 9, 10, 6};  //to be stored somewhere common

    //saving seeds by algorithm
    const int ns = seedTracks_.size();

    TrackVec cleanSeedTracks;
    cleanSeedTracks.reserve(ns);

    for (int ts = 0; ts < ns; ts++) {
      const Track &tk = seedTracks_[ts];
      unsigned int algo = (unsigned int)tk.algorithm();
      if (std::find(algorithms, algorithms + n, algo) != algorithms + n)
        cleanSeedTracks.emplace_back(seedTracks_[ts]);
    }
    seedTracks_.swap(cleanSeedTracks);

    //saving tracks by algorithm
    const int nt = cmsswTracks_.size();

    TrackVec cleanTracks;
    cleanTracks.reserve(nt);

    for (int ts = 0; ts < nt; ts++) {
      const Track &tk = cmsswTracks_[ts];
      unsigned int algo = (unsigned int)tk.algorithm();
      if (std::find(algorithms, algorithms + n, algo) != algorithms + n)
        cleanTracks.emplace_back(cmsswTracks_[ts]);
    }
    cmsswTracks_.swap(cleanTracks);
    return cmsswTracks_.size() + seedTracks_.size();
  }

setInputFromCMSSW()

void Event::setInputFromCMSSW	(	std::vector< HitVec >	hits,
		TrackVec	seeds
	)

Definition at line 455 of file Event.cc.

References hfClusterShapes_cfi::hits, layerHits_, eostools::move(), DetachedQuadStep_cff::seeds, and seedTracks_.

                                                                      {
    layerHits_ = std::move(hits);
    seedTracks_ = std::move(seeds);
  }

use_seeds_from_cmsswtracks()

int Event::use_seeds_from_cmsswtracks

(

)

Definition at line 763 of file Event.cc.

References cmsswTracks_, and seedTracks_.

                                        {
    int ns = seedTracks_.size();

    TrackVec cleanSeedTracks;
    cleanSeedTracks.reserve(ns);

    for (auto &&cmsswtrack : cmsswTracks_) {
      cleanSeedTracks.emplace_back(seedTracks_[cmsswtrack.label()]);
    }

    seedTracks_.swap(cleanSeedTracks);

    return seedTracks_.size();
  }

validate()

void Event::validate

(

)

Definition at line 59 of file Event.cc.

References mkfit::Config::cmssw_val, mkfit::Validation::fillCMSSWEfficiencyTree(), mkfit::Validation::fillCMSSWFakeRateTree(), mkfit::Validation::fillEfficiencyTree(), mkfit::Validation::fillFakeRateTree(), mkfit::Validation::fillFitTree(), mkfit::Config::fit_val, mkfit::Validation::makeCMSSWTkToRecoTksMaps(), mkfit::Validation::makeCMSSWTkToSeedTkMap(), mkfit::Validation::makeRecoTkToRecoTkMaps(), mkfit::Validation::makeRecoTkToSeedTkMapsDumbCMSSW(), mkfit::Validation::makeSeedTkToRecoTkMaps(), mkfit::Validation::makeSimTkToRecoTksMaps(), mkfit::Validation::setTrackExtras(), mkfit::Validation::setTrackScoresDumbCMSSW(), mkfit::Config::sim_val, mkfit::Config::sim_val_for_cmssw, and validation_.

                       {
    // special map needed for sim_val_for_cmssw + set the track scores
    if (Config::sim_val_for_cmssw) {
      validation_.makeRecoTkToSeedTkMapsDumbCMSSW(*this);
      validation_.setTrackScoresDumbCMSSW(*this);
    }

    // standard eff/fr/dr validation
    if (Config::sim_val || Config::sim_val_for_cmssw) {
      validation_.setTrackExtras(*this);
      validation_.makeSimTkToRecoTksMaps(*this);
      validation_.makeSeedTkToRecoTkMaps(*this);
      validation_.fillEfficiencyTree(*this);
      validation_.fillFakeRateTree(*this);
    }

    // special cmssw to mkfit validation
    if (Config::cmssw_val) {
      validation_.makeCMSSWTkToSeedTkMap(*this);
      validation_.makeRecoTkToRecoTkMaps(*this);
      validation_.setTrackExtras(*this);
      validation_.makeCMSSWTkToRecoTksMaps(*this);
      validation_.fillCMSSWEfficiencyTree(*this);
      validation_.fillCMSSWFakeRateTree(*this);
    }

    if (Config::fit_val) {  // fit val for z-phi tuning
      validation_.fillFitTree(*this);
    }
  }

write_out()

void Event::write_out

(

DataFile &

data_file

)

Definition at line 117 of file Event.cc.

References cms::cuda::assert(), align::BeamSpot, beamSpot_, cmsswTracks_, mkfit::DataFile::f_fp, personalPlayback::fp, mkfit::DataFile::hasBeamSpot(), mkfit::DataFile::hasCmsswTracks(), mkfit::DataFile::hasHitIterMasks(), mkfit::DataFile::hasSeeds(), mkfit::DataFile::hasSimTrackStates(), createfilelist::int, layerHitMasks_, layerHits_, mutex, cms::cuda::nh, seedTracks_, simHitsInfo_, simTracks_, simTrackStates_, findQualityFiles::size, and write_tracks().

                                           {
    FILE *fp = data_file.f_fp;

    static std::mutex writemutex;
    std::lock_guard<std::mutex> writelock(writemutex);

    auto start = ftell(fp);
    int evsize = sizeof(int);
    fwrite(&evsize, sizeof(int), 1, fp);  // this will be overwritten at the end

    evsize += write_tracks(fp, simTracks_);

    if (data_file.hasSimTrackStates()) {
      int nts = simTrackStates_.size();
      fwrite(&nts, sizeof(int), 1, fp);
      fwrite(&simTrackStates_[0], sizeof(TrackState), nts, fp);
      evsize += sizeof(int) + nts * sizeof(TrackState);
    }

    int nl = layerHits_.size();
    fwrite(&nl, sizeof(int), 1, fp);
    evsize += sizeof(int);
    for (int il = 0; il < nl; ++il) {
      int nh = layerHits_[il].size();
      fwrite(&nh, sizeof(int), 1, fp);
      fwrite(&layerHits_[il][0], sizeof(Hit), nh, fp);
      evsize += sizeof(int) + nh * sizeof(Hit);
    }

    if (data_file.hasHitIterMasks()) {
      //sizes are the same as in layerHits_
      for (int il = 0; il < nl; ++il) {
        int nh = layerHitMasks_[il].size();
        assert(nh == (int)layerHits_[il].size());
        fwrite(&layerHitMasks_[il][0], sizeof(uint64_t), nh, fp);
        evsize += nh * sizeof(uint64_t);
      }
    }

    int nm = simHitsInfo_.size();
    fwrite(&nm, sizeof(int), 1, fp);
    fwrite(&simHitsInfo_[0], sizeof(MCHitInfo), nm, fp);
    evsize += sizeof(int) + nm * sizeof(MCHitInfo);

    if (data_file.hasSeeds()) {
      evsize += write_tracks(fp, seedTracks_);
    }

    if (data_file.hasCmsswTracks()) {
      evsize += write_tracks(fp, cmsswTracks_);
    }

    if (data_file.hasBeamSpot()) {
      fwrite(&beamSpot_, sizeof(BeamSpot), 1, fp);
      evsize += sizeof(BeamSpot);
    }

    fseek(fp, start, SEEK_SET);
    fwrite(&evsize, sizeof(int), 1, fp);
    fseek(fp, 0, SEEK_END);

    //layerHitMap_ is recreated afterwards

    /*
  printf("write %i tracks\n",nt);
  for (int it = 0; it<nt; it++) {
    printf("track with pT=%5.3f\n",simTracks_[it].pT());
    for (int ih=0; ih<simTracks_[it].nTotalHits(); ++ih) {
      printf("hit lyr:%2d idx=%i\n", simTracks_[it].getHitLyr(ih), simTracks_[it].getHitIdx(ih));
    }
  }
  printf("write %i layers\n",nl);
  for (int il = 0; il<nl; il++) {
    printf("write %i hits in layer %i\n",layerHits_[il].size(),il);
    for (int ih = 0; ih<layerHits_[il].size(); ih++) {
      printf("hit with r=%5.3f x=%5.3f y=%5.3f z=%5.3f\n",layerHits_[il][ih].r(),layerHits_[il][ih].x(),layerHits_[il][ih].y(),layerHits_[il][ih].z());
    }
  }
  */
  }

write_tracks()

int Event::write_tracks	(	FILE *	fp,
		const TrackVec &	tracks
	)

Definition at line 405 of file Event.cc.

References personalPlayback::fp, mps_fire::i, createfilelist::int, and tracks.

Referenced by write_out().

                                                          {
    // Returns total number of bytes written.

    int n_tracks = tracks.size();
    fwrite(&n_tracks, sizeof(int), 1, fp);

    auto start = ftell(fp);
    int data_size = 2 * sizeof(int) + n_tracks * sizeof(Track);
    fwrite(&data_size, sizeof(int), 1, fp);

    fwrite(tracks.data(), sizeof(Track), n_tracks, fp);

    for (int i = 0; i < n_tracks; ++i) {
      fwrite(tracks[i].beginHitsOnTrack(), sizeof(HitOnTrack), tracks[i].nTotalHits(), fp);
      data_size += tracks[i].nTotalHits() * sizeof(HitOnTrack);
    }

    fseek(fp, start, SEEK_SET);
    fwrite(&data_size, sizeof(int), 1, fp);
    fseek(fp, 0, SEEK_END);

    return data_size;
  }

Member Data Documentation

beamSpot_

BeamSpot mkfit::Event::beamSpot_

Definition at line 56 of file Event.h.

Referenced by read_in(), reset(), and write_out().

candidateTracks_

TrackVec mkfit::Event::candidateTracks_

Definition at line 61 of file Event.h.

Referenced by reset().

candidateTracksExtra_

TrackExtraVec mkfit::Event::candidateTracksExtra_

mutable

Definition at line 64 of file Event.h.

Referenced by reset().

cmsswTracks_

TrackVec mkfit::Event::cmsswTracks_

Definition at line 62 of file Event.h.

Referenced by read_in(), relabel_cmsswtracks_from_seeds(), reset(), select_tracks_iter(), use_seeds_from_cmsswtracks(), and write_out().

cmsswTracksExtra_

TrackExtraVec mkfit::Event::cmsswTracksExtra_

mutable

Definition at line 65 of file Event.h.

Referenced by reset().

evtID_

int mkfit::Event::evtID_

private

Definition at line 53 of file Event.h.

Referenced by evtID(), and reset().

fitTracks_

TrackVec mkfit::Event::fitTracks_

Definition at line 61 of file Event.h.

Referenced by reset().

fitTracksExtra_

TrackExtraVec mkfit::Event::fitTracksExtra_

mutable

Definition at line 64 of file Event.h.

Referenced by reset().

layerHitMasks_

std::vector<std::vector<uint64_t> > mkfit::Event::layerHitMasks_

Definition at line 58 of file Event.h.

Referenced by Event(), fill_hitmask_bool_vectors(), read_in(), reset(), and write_out().

layerHits_

std::vector<HitVec> mkfit::Event::layerHits_

Definition at line 57 of file Event.h.

Referenced by Event(), fill_hitmask_bool_vectors(), kludge_cms_hit_errors(), print_tracks(), printStats(), read_in(), reset(), setInputFromCMSSW(), and write_out().

printmutex

std::mutex Event::printmutex

static

Definition at line 68 of file Event.h.

Referenced by mkfit::StdSeq::Quality::quality_print(), mkfit::StdSeq::Quality::quality_process(), and test_standard().

seedTracks_

TrackVec mkfit::Event::seedTracks_

Definition at line 61 of file Event.h.

Referenced by clean_cms_seedtracks(), clean_cms_seedtracks_badlabel(), read_in(), relabel_bad_seedtracks(), relabel_cmsswtracks_from_seeds(), reset(), select_tracks_iter(), setInputFromCMSSW(), use_seeds_from_cmsswtracks(), and write_out().

seedTracksExtra_

TrackExtraVec mkfit::Event::seedTracksExtra_

mutable

Definition at line 64 of file Event.h.

Referenced by reset().

simHitsInfo_

MCHitInfoVec mkfit::Event::simHitsInfo_

Definition at line 59 of file Event.h.

Referenced by print_tracks(), printStats(), read_in(), reset(), and write_out().

simTracks_

TrackVec mkfit::Event::simTracks_

Definition at line 61 of file Event.h.

Referenced by clean_cms_simtracks(), printStats(), read_in(), reset(), and write_out().

simTracksExtra_

TrackExtraVec mkfit::Event::simTracksExtra_

mutable

Definition at line 64 of file Event.h.

Referenced by reset().

simTrackStates_

TSVec mkfit::Event::simTrackStates_

Definition at line 67 of file Event.h.

Referenced by read_in(), reset(), and write_out().

validation_

Validation& mkfit::Event::validation_

Definition at line 50 of file Event.h.

Referenced by Event(), reset(), and validate().