EvtMessage.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <RtypesCore.h>
 
struct timeval;
 
namespace Belle2 {

  enum ERecordType { MSG_EVENT, MSG_BEGIN_RUN, MSG_END_RUN, MSG_TERMINATE, MSG_NORECORD, MSG_STREAMERINFO };

  struct EvtHeader {
    EvtHeader(UInt_t aSize, ERecordType aRectype): size(aSize), rectype(aRectype) {}
    UInt_t size; 
    ERecordType rectype; 
    Long64_t time_sec{0}; 
    Long64_t time_usec{0}; 
    UInt_t src{(UInt_t) - 1}; 
    UInt_t dest{(UInt_t) - 1}; 
    UInt_t flags{0}; 
    UInt_t nObjects{0}; 
    UInt_t nArrays{0}; 
    UInt_t reserved[6] {0}; 
    UInt_t version{0xBEEFED01};
  };

  class EvtMessage {
 
  public:
    const static unsigned int c_MaxEventSize = 200000000;

    enum EMessageFlags {
      c_MsgCompressed = 1
    };

    explicit EvtMessage(char* buf = nullptr);
    EvtMessage(const char* msg, int size, ERecordType type);
    EvtMessage(const EvtMessage& evtmsg);
    ~EvtMessage();

    EvtMessage& operator=(const EvtMessage& obj);
 
 
    // Access functions
    char* buffer();
    void  buffer(const char*);

    int   size() const;
    int   paddedSize() const;
    int   msg_size() const;

    unsigned int getVersion() const { return ((getHeader()->version & 0xFFFFFF00) == 0xBEEFED00) ? (getHeader()->version & 0xFF) : 0;}
    unsigned int getMsgFlags() const { return getVersion() > 0 ? getHeader()->flags : 0; }
    void setMsgFlags(unsigned int flags) { header()->flags = flags;}
    void addMsgFlags(unsigned int flags) { header()->flags |= flags;}
    bool hasMsgFlags(unsigned int flags) const { return (getMsgFlags() & flags) == flags; }

    ERecordType type() const;
    void type(ERecordType);

    struct timeval time() const;
    void setTime(const struct timeval& time);

    int   src() const;
    void  src(int src);

    int   dest() const;
    void  dest(int dest);

    EvtHeader* header();
    const EvtHeader* getHeader() const;
    char* msg();
 
  private:
    void setMsg(const char* msg, int size, ERecordType type);
 
    char* m_data;         
    bool m_ownsBuffer; 
  };

}