File indexing completed on 2025-03-02 23:54:18
0001 #ifndef RecoTracker_MkFitCore_standalone_Event_h
0002 #define RecoTracker_MkFitCore_standalone_Event_h
0003
0004 #include "RecoTracker/MkFitCore/interface/Config.h"
0005 #include "RecoTracker/MkFitCore/interface/Track.h"
0006 #include "RecoTracker/MkFitCore/interface/BeamSpot.h"
0007 #include "Validation.h"
0008
0009 #include <mutex>
0010
0011 namespace mkfit {
0012
0013 struct DataFile;
0014
0015 class Event {
0016 public:
0017 explicit Event(int evtID, int nLayers);
0018 Event(Validation &v, int evtID, int nLayers);
0019
0020 void reset(int evtID);
0021 void validate();
0022 void printStats(const TrackVec &, TrackExtraVec &);
0023
0024 int evtID() const { return evtID_; }
0025 void resetLayerHitMap(bool resetSimHits);
0026
0027 void write_out(DataFile &data_file);
0028 void read_in(DataFile &data_file, FILE *in_fp = 0);
0029 int write_tracks(FILE *fp, const TrackVec &tracks);
0030 int read_tracks(FILE *fp, TrackVec &tracks, bool skip_reading = false);
0031
0032 void setInputFromCMSSW(std::vector<HitVec> hits, TrackVec seeds);
0033
0034 int use_seeds_from_cmsswtracks();
0035 int clean_cms_simtracks();
0036 int clean_cms_seedtracks(
0037 TrackVec *seed_ptr = nullptr);
0038 int clean_cms_seedtracks_badlabel();
0039 void relabel_bad_seedtracks();
0040 void relabel_cmsswtracks_from_seeds();
0041
0042 int select_tracks_iter(unsigned int n = 0);
0043
0044 void fill_hitmask_bool_vectors(int track_algo, std::vector<std::vector<bool>> &layer_masks);
0045 void fill_hitmask_bool_vectors(std::vector<int> &track_algo_vec, std::vector<std::vector<bool>> &layer_masks);
0046
0047 void print_tracks(const TrackVec &tracks, bool print_hits) const;
0048
0049 Validation &validation_;
0050
0051
0052 struct SimLabelFromHits {
0053 int n_hits = 0, n_match = 0, label = -1;
0054 float good_frac() const { return (float)n_match / n_hits; }
0055 bool is_set() const { return label >= 0; }
0056 };
0057 void setCurrentSeedTracks(const TrackVec &seeds);
0058 const Track ¤tSeed(int i) const;
0059 SimLabelFromHits simLabelForCurrentSeed(int i) const;
0060 void resetCurrentSeedTracks();
0061
0062 private:
0063 int evtID_;
0064 const TrackVec *currentSeedTracks_ = nullptr;
0065 mutable std::vector<SimLabelFromHits> currentSeedSimFromHits_;
0066
0067 public:
0068 BeamSpot beamSpot_;
0069 std::vector<HitVec> layerHits_;
0070 std::vector<std::vector<uint64_t>> layerHitMasks_;
0071 MCHitInfoVec simHitsInfo_;
0072
0073 TrackVec simTracks_, seedTracks_, candidateTracks_, fitTracks_;
0074 TrackVec cmsswTracks_;
0075
0076 mutable TrackExtraVec simTracksExtra_, seedTracksExtra_, candidateTracksExtra_, fitTracksExtra_;
0077 mutable TrackExtraVec cmsswTracksExtra_;
0078
0079 TSVec simTrackStates_;
0080 static std::mutex printmutex;
0081 };
0082
0083 typedef std::vector<Event> EventVec;
0084
0085 struct DataFileHeader {
0086 int f_magic = 0xBEEF;
0087 int f_format_version = 7;
0088 int f_sizeof_track = sizeof(Track);
0089 int f_sizeof_hit = sizeof(Hit);
0090 int f_sizeof_hot = sizeof(HitOnTrack);
0091 int f_n_layers = -1;
0092 int f_n_events = -1;
0093
0094 int f_extra_sections = 0;
0095
0096 DataFileHeader() = default;
0097 };
0098
0099 struct DataFile {
0100 enum ExtraSection {
0101 ES_SimTrackStates = 0x1,
0102 ES_Seeds = 0x2,
0103 ES_CmsswTracks = 0x4,
0104 ES_HitIterMasks = 0x8,
0105 ES_BeamSpot = 0x10
0106 };
0107
0108 FILE *f_fp = 0;
0109 long f_pos = sizeof(DataFileHeader);
0110
0111 DataFileHeader f_header;
0112
0113 std::mutex f_next_ev_mutex;
0114
0115
0116
0117 bool hasSimTrackStates() const { return f_header.f_extra_sections & ES_SimTrackStates; }
0118 bool hasSeeds() const { return f_header.f_extra_sections & ES_Seeds; }
0119 bool hasCmsswTracks() const { return f_header.f_extra_sections & ES_CmsswTracks; }
0120 bool hasHitIterMasks() const { return f_header.f_extra_sections & ES_HitIterMasks; }
0121 bool hasBeamSpot() const { return f_header.f_extra_sections & ES_BeamSpot; }
0122
0123 int openRead(const std::string &fname, int expected_n_layers);
0124 void openWrite(const std::string &fname, int n_layers, int n_ev, int extra_sections = 0);
0125
0126 void rewind();
0127
0128 int advancePosToNextEvent(FILE *fp);
0129
0130 void skipNEvents(int n_to_skip);
0131
0132 void close();
0133 void CloseWrite(int n_written);
0134 };
0135
0136 void print(std::string pfx, int itrack, const Track &trk, const Event &ev);
0137
0138 }
0139 #endif