StreamerFileWriter.h

Go to the documentation of this file.

#ifndef IOPool_Streamer_StreamerFileWriter_h
#define IOPool_Streamer_StreamerFileWriter_h 

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "IOPool/Streamer/interface/StreamerOutputFile.h"
#include "IOPool/Streamer/interface/InitMsgBuilder.h"
#include "IOPool/Streamer/interface/EventMsgBuilder.h"
#include "IOPool/Streamer/interface/InitMessage.h"
#include "IOPool/Streamer/interface/EventMessage.h"
#include "IOPool/Streamer/interface/MsgTools.h"

#include <iostream>
#include <vector>
#include <memory>
#include <string>

namespace edm
{
  struct StreamerFileWriterHeaderParams
  {
    uint32 runNumber;
    uint32 hltCount;
    const char* headerPtr;
    uint32 headerSize;

    uint32 fragmentIndex;
    uint32 fragmentCount;
    const char* dataPtr;
    uint32 dataSize;
  };

  struct StreamerFileWriterEventParams
  {
    std::vector<unsigned char> hltBits;
    const char* headerPtr;
    uint32 headerSize;

    uint32 fragmentIndex;
    uint32 fragmentCount;
    const char* dataPtr;
    uint32 dataSize;
  };

  class ParameterSetDescription;
  class StreamerFileWriter 
  {
  public:

    explicit StreamerFileWriter(edm::ParameterSet const& ps);
    explicit StreamerFileWriter(std::string const& fileName);
    ~StreamerFileWriter();

    static void fillDescription(ParameterSetDescription& desc);

    void doOutputHeader(InitMsgBuilder const& init_message);    
    void doOutputHeader(InitMsgView const& init_message);    
    void doOutputHeaderFragment(StreamerFileWriterHeaderParams const&);

    void doOutputEvent(EventMsgBuilder const& msg);
    void doOutputEvent(EventMsgView const& msg);
    void doOutputEventFragment(StreamerFileWriterEventParams const&);

    void start(){}
    void stop();
    // Returns the sizes of EOF records, call them after 
    // u called stop, just before destruction
    uint32 getStreamEOFSize() const {return stream_eof_size_;}

    uint32 get_adler32() const { return stream_writer_->adler32();}

  private:
    void updateHLTStats(std::vector<uint8> const& packedHlt);

    std::auto_ptr<StreamerOutputFile> stream_writer_;
    uint32 hltCount_;
    std::vector<uint32> hltStats_;
    uint32 stream_eof_size_;

  };
}
#endif