Utilities/interface/DAQSourceModelsScoutingRun3.h

0001 #ifndef EventFilter_Utilities_DAQSourceModelsScoutingRun3_h
0002 #define EventFilter_Utilities_DAQSourceModelsScoutingRun3_h
0003
0004 #include "EventFilter/Utilities/interface/DAQSource.h"
0005 #include "EventFilter/Utilities/interface/DAQSourceModels.h"
0006 #include "DataFormats/L1ScoutingRawData/interface/SDSRawDataCollection.h"
0007 #include "DataFormats/L1ScoutingRawData/interface/SDSNumbering.h"
0008
0009 #include "FWCore/Framework/interface/Event.h"
0010 #include "DataFormats/Provenance/interface/EventAuxiliary.h"
0011 #include "DataFormats/Provenance/interface/EventID.h"
0012
0013 #include <sys/types.h>
0014 #include <filesystem>
0015 #include <sstream>
0016 #include <iostream>
0017 #include <memory>
0018 #include <vector>
0019
0020 class DataModeScoutingRun3 : public DataMode {
0021 public:
0022   DataModeScoutingRun3(DAQSource* daqSource) : DataMode(daqSource) {}
0023   ~DataModeScoutingRun3() override {}
0024   std::vector<std::shared_ptr<const edm::DaqProvenanceHelper>>& makeDaqProvenanceHelpers() override;
0025   void readEvent(edm::EventPrincipal& eventPrincipal) override;
0026
0027   void fillSDSRawDataCollection(SDSRawDataCollection& rawData, char* buff, size_t len);
0028
0029   //reuse FRD file and event headers
0030   int dataVersion() const override { return detectedFRDversion_; }
0031   void detectVersion(unsigned char* fileBuf, uint32_t fileHeaderOffset) override {
0032     detectedFRDversion_ = *((uint16_t*)(fileBuf + fileHeaderOffset));
0033   }
0034   uint32_t headerSize() const override { return edm::streamer::FRDHeaderVersionSize[detectedFRDversion_]; }
0035   bool versionCheck() const override { return detectedFRDversion_ <= edm::streamer::FRDHeaderMaxVersion; }
0036
0037   uint64_t dataBlockSize() const override {
0038     // get event size from the first data source (main)
0039     return events_[0]->size();
0040   }
0041
0042   void makeDataBlockView(unsigned char* addr, RawInputFile* rawFile) override {
0043     std::vector<uint64_t> const& fileSizes = rawFile->fileSizes_;
0044     fileHeaderSize_ = rawFile->rawHeaderSize_;
0045     numFiles_ = fileSizes.size();
0046
0047     // initalize vectors keeping tracks of valid orbits and completed blocks
0048     sourceValidOrbitPair_.clear();
0049     completedBlocks_.clear();
0050     for (unsigned int i = 0; i < fileSizes.size(); i++) {
0051       completedBlocks_.push_back(false);
0052     }
0053
0054     //add offset address for each file payload
0055     dataBlockAddrs_.clear();
0056     dataBlockAddrs_.push_back(addr);
0057     dataBlockMaxAddrs_.clear();
0058     dataBlockMaxAddrs_.push_back(addr + fileSizes[0] - fileHeaderSize_);
0059     auto fileAddr = addr;
0060     for (unsigned int i = 1; i < fileSizes.size(); i++) {
0061       fileAddr += fileSizes[i - 1];
0062       dataBlockAddrs_.push_back(fileAddr);
0063       dataBlockMaxAddrs_.push_back(fileAddr + fileSizes[i] - fileHeaderSize_);
0064     }
0065
0066     dataBlockMax_ = rawFile->currentChunkSize();
0067     blockCompleted_ = false;
0068     //set event cached as we set initial address here
0069     bool result = makeEvents();
0070     assert(result);
0071     eventCached_ = true;
0072     setDataBlockInitialized(true);
0073   }
0074
0075   bool nextEventView(RawInputFile*) override;
0076   bool blockChecksumValid() override { return true; }
0077   bool checksumValid() override;
0078   std::string getChecksumError() const override;
0079
0080   uint32_t run() const override {
0081     assert(!events_.empty());
0082     return events_[0]->run();
0083   }
0084
0085   bool dataBlockCompleted() const override { return blockCompleted_; }
0086
0087   bool requireHeader() const override { return true; }
0088
0089   bool fitToBuffer() const override { return true; }
0090   void unpackFile(RawInputFile* file) override {}
0091
0092   bool dataBlockInitialized() const override { return dataBlockInitialized_; }
0093
0094   void setDataBlockInitialized(bool val) override { dataBlockInitialized_ = val; };
0095
0096   void setTCDSSearchRange(uint16_t MINTCDSuTCAFEDID, uint16_t MAXTCDSuTCAFEDID) override { return; }
0097
0098   void makeDirectoryEntries(std::vector<std::string> const& baseDirs,
0099                             std::vector<int> const& numSources,
0100                             std::vector<int> const& sourceIDs,
0101                             std::string const& sourceIdentifier,
0102                             std::string const& runDir) override;
0103
0104   std::pair<bool, std::vector<std::string>> defineAdditionalFiles(std::string const& primaryName,
0105                                                                   bool fileListMode) const override;
0106
0107 private:
0108   bool makeEvents();
0109   std::vector<std::shared_ptr<const edm::DaqProvenanceHelper>> daqProvenanceHelpers_;  //
0110   uint16_t detectedFRDversion_ = 0;
0111   size_t fileHeaderSize_ = 0;
0112   size_t headerSize_ = 0;
0113   std::vector<std::unique_ptr<edm::streamer::FRDEventMsgView>> events_;
0114   unsigned char* dataBlockAddr_ = nullptr;
0115   std::vector<unsigned char*> dataBlockAddrs_;
0116   std::vector<unsigned char*> dataBlockMaxAddrs_;
0117   size_t dataBlockMax_ = 0;
0118   short numFiles_ = 0;
0119   bool dataBlockInitialized_ = false;
0120   bool blockCompleted_ = true;
0121   bool eventCached_ = false;
0122   std::vector<std::filesystem::path> buPaths_;
0123   std::vector<int> buNumSources_;
0124
0125   // keep track of valid (=aligned) orbits from different data sources
0126   std::vector<std::pair<int, int>> sourceValidOrbitPair_;
0127   unsigned int currOrbit_ = 0xFFFFFFFF;
0128
0129   std::vector<bool> completedBlocks_;
0130 };
0131
0132 #endif  // EventFilter_Utilities_DAQSourceModelsScoutingRun3_h