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();

    bool 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();

  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 openNextFile();
    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