Back to home page

Project CMSSW displayed by LXR

 
 

    


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();  //special mode --> use only seeds which generated cmssw reco track
0035     int clean_cms_simtracks();
0036     int clean_cms_seedtracks(
0037         TrackVec *seed_ptr = nullptr);    //operates on seedTracks_; returns the number of cleaned seeds
0038     int clean_cms_seedtracks_badlabel();  //operates on seedTracks_, removes those with label == -1;
0039     void relabel_bad_seedtracks();
0040     void relabel_cmsswtracks_from_seeds();
0041 
0042     int select_tracks_iter(unsigned int n = 0);  //for cmssw input
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     // For seed access in deep data dumpers.
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 &currentSeed(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_;  // XXXX Read/Write of BeamSpot + file-version bump or extra-section to be added.
0069     std::vector<HitVec> layerHits_;
0070     std::vector<std::vector<uint64_t>> layerHitMasks_;  //aligned with layerHits_
0071     MCHitInfoVec simHitsInfo_;
0072 
0073     TrackVec simTracks_, seedTracks_, candidateTracks_, fitTracks_;
0074     TrackVec cmsswTracks_;
0075     // validation sets these, so needs to be mutable
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;  //last update with ph2 geom
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);  //override nevents in the header and close
0134   };
0135 
0136   void print(std::string pfx, int itrack, const Track &trk, const Event &ev);
0137 
0138 }  // end namespace mkfit
0139 #endif