StreamerInputFile.h

Go to the documentation of this file.

#ifndef IOPool_Streamer_StreamerInputFile_h
#define IOPool_Streamer_StreamerInputFile_h
 
#include "IOPool/Streamer/interface/InitMessage.h"
#include "IOPool/Streamer/interface/EventMessage.h"
#include "IOPool/Streamer/interface/MsgTools.h"
#include "Utilities/StorageFactory/interface/IOTypes.h"
#include "Utilities/StorageFactory/interface/Storage.h"
#include "FWCore/Utilities/interface/propagate_const.h"
 
#include <memory>
 
#include <string>
#include <vector>
 
namespace edm {
  class EventSkipperByID;
  class FileCatalogItem;
  class StreamerInputFile {
  public:
    explicit StreamerInputFile(std::string const& name,
                               std::string const& LFN,
                               std::shared_ptr<EventSkipperByID> eventSkipperByID = std::shared_ptr<EventSkipperByID>());
    explicit StreamerInputFile(std::string const& name,
                               std::shared_ptr<EventSkipperByID> eventSkipperByID = std::shared_ptr<EventSkipperByID>());
 
    explicit StreamerInputFile(std::vector<FileCatalogItem> const& names,
                               std::shared_ptr<EventSkipperByID> eventSkipperByID = std::shared_ptr<EventSkipperByID>());
 
    ~StreamerInputFile();
 
    enum class Next { kEvent, kFile, kStop };
    Next next(); 
    InitMsgView const* startMessage() const { return startMsg_.get(); }
    EventMsgView const* currentRecord() const { return currentEvMsg_.get(); }
    bool newHeader() {
      bool tmp = newHeader_;
      newHeader_ = false;
      return tmp;
    } 
    void closeStreamerFile();
    bool openNextFile();
 
  private:
    void openStreamerFile(std::string const& name, std::string const& LFN);
    IOSize readBytes(char* buf, IOSize nBytes);
    IOOffset skipBytes(IOSize nBytes);
 
    void readStartMessage();
    int readEventMessage();
 
    bool compareHeader();
 
    void logFileAction(char const* msg);
 
    edm::propagate_const<std::shared_ptr<InitMsgView>> startMsg_;
    edm::propagate_const<std::shared_ptr<EventMsgView>> currentEvMsg_;
 
    std::vector<char> headerBuf_; 
    std::vector<char> eventBuf_;  
    unsigned int currentFile_;                   
    std::vector<FileCatalogItem> streamerNames_; 
    bool multiStreams_;                          
    std::string currentFileName_;
    bool currentFileOpen_;
 
    edm::propagate_const<std::shared_ptr<EventSkipperByID>> eventSkipperByID_;
 
    uint32 currRun_;
    uint32 currProto_;
 
    bool newHeader_;
 
    edm::propagate_const<std::unique_ptr<Storage>> storage_;
 
    bool endOfFile_;
  };
}  // namespace edm
 
#endif