MsgHandler Class Reference

A class to encode/decode an EvtMessage. More...

#include <MsgHandler.h>

Collaboration diagram for MsgHandler:

Public Member Functions
	MsgHandler (int complevel=0)
	Constructor.
virtual	~MsgHandler ()
	Destructor.
virtual void	clear ()
	Clear object list.
virtual void	add (const TObject *, const std::string &name)
	Add an object to be streamed.
virtual EvtMessage *	encode_msg (ERecordType rectype)
	Stream object list into an EvtMessage.
virtual void	decode_msg (EvtMessage *msg, std::vector< TObject * > &objlist, std::vector< std::string > &namelist)
	Decode an EvtMessage into a vector list of objects with names.

Private Attributes
CharBuffer	m_buf
	EvtMessage character buffer for encode_msg().
CharBuffer	m_compBuf
	EvtMessage character buffer for compressing/decompressing.
std::unique_ptr< TMessage >	m_msg
	Used for serialising objects into m_buf.
InMessage	m_inMsg
	Used for deserializing in decode_msg()
int	m_complevel
	compression algorithm * 100 + compression level.

Detailed Description

A class to encode/decode an EvtMessage.

Definition at line 103 of file MsgHandler.h.

Constructor & Destructor Documentation

MsgHandler()

MsgHandler ( int  complevel = 0 )

explicit

Constructor.

Parameters

complevel

Compression level and algorithm: algorithm*100 + level where algorithm can be one of 0: default root system algorithm (usually zlib) 1: zlib 2: lzma 3: old root compression code and level can be between 1 and 9 (0 disables compression). So for example 101 corresponds to zlib with minimal compression while 209 means lzma with maximal compression.

Definition at line 25 of file MsgHandler.cc.

                                   :
  m_buf(100000),
  m_compBuf(0),
  m_msg(new TMessage(kMESS_OBJECT))
{
  m_complevel = complevel;
 
  //Schema evolution is needed to stream genfit tracks
  //If disabled, streamers will crash when reading data.
  TMessage::EnableSchemaEvolutionForAll();
  m_msg->SetWriteMode();
}

Member Function Documentation

add()

void add ( const TObject *  obj, const std::string &  name  )

virtual

Add an object to be streamed.

Definition at line 46 of file MsgHandler.cc.

{
  m_msg->WriteObject(obj);
 
  int len = m_msg->Length();
  char* buf = m_msg->Buffer();
 
  if (len > c_maxObjectSizeBytes) {
    B2WARNING("MsgHandler: Object " << name << " is very large (" << len  << " bytes), parallel processing may be slow.");
  }
 
  // Put name of object in output buffer including a final 0-byte
  UInt_t nameLength = name.size() + 1;
  m_buf.add(&nameLength, sizeof(nameLength));
  m_buf.add(name.c_str(), nameLength);
  // Copy object into buffer
  m_buf.add(&len, sizeof(len));
  m_buf.add(buf, len);
  m_msg->Reset();
}

clear()

void clear ( )

virtual

Clear object list.

Definition at line 40 of file MsgHandler.cc.

{
  m_buf.clear();
  m_compBuf.clear();
}

decode_msg()

void decode_msg ( EvtMessage *  msg, std::vector< TObject * > &  objlist, std::vector< std::string > &  namelist  )

virtual

Decode an EvtMessage into a vector list of objects with names.

Definition at line 106 of file MsgHandler.cc.

{
  const char* msgptr = msg->msg();
  const char* end = msgptr + msg->msg_size();
 
  if (msg->hasMsgFlags(EvtMessage::c_MsgCompressed)) {
    // apparently message is compressed, let's decompress
    m_compBuf.clear();
    int nzip{0}, nout{0};
    // ROOT wants unsigned char so make a new pointer to the data
    auto* zipptr = (unsigned char*) msgptr;
    // and uncompress everything
    while (zipptr < (unsigned char*)end) {
      // first get a header of the next block so we know how big the output will be
      if (R__unzip_header(&nzip, zipptr, &nout) != 0) {
        B2FATAL("Cannot uncompress message header");
      }
      // no more output? fine
      if (!nout) break;
      if (std::distance(zipptr, (unsigned char*) end) > nzip) {
        B2FATAL("Not enough bytes left to uncompress");
      }
      // otherwise make sure output buffer is large enough
      int old_size = m_compBuf.size();
      m_compBuf.resize(old_size + nout);
      // and uncompress, the amount of bytes will be returned as irep
      int irep{0};
      R__unzip(&nzip, zipptr, &nout, (unsigned char*)(m_compBuf.data() + old_size), &irep);
      // if that is not positive an error happend, bail
      if (irep <= 0) {
        B2FATAL("Cannot uncompress message");
      }
      // otherwise advance pointer by the amount of bytes compressed bytes in the block
      zipptr += nzip;
    }
    // ok, decompressed succesfully, set msg pointer to the correct area
    msgptr = m_compBuf.data();
    end = msgptr + m_compBuf.size();
  }
 
  while (msgptr < end) {
    // Restore object name
    UInt_t nameLength;
    memcpy(&nameLength, msgptr, sizeof(nameLength));
    msgptr += sizeof(nameLength);
    if (nameLength == 0 || std::distance(msgptr, end) < nameLength)
      B2FATAL("Buffer overrun while decoding object name, check length fields!");
 
    // read full string but omit final 0-byte. This safeguards against strings containing 0-bytes
    namelist.emplace_back(msgptr, nameLength - 1);
    msgptr += nameLength;
 
    // Restore object
    UInt_t objlen;
    memcpy(&objlen, msgptr, sizeof(objlen));
    msgptr += sizeof(objlen);
    if (objlen == 0 || std::distance(msgptr, end) < objlen)
      B2FATAL("Buffer overrun while decoding object, check length fields!");
 
    m_inMsg.SetBuffer(msgptr, objlen);
    objlist.push_back(m_inMsg.readTObject());
    msgptr += objlen;
    //no need to call InMessage::Reset() here (done in SetBuffer())
  }
}

encode_msg()

EvtMessage * encode_msg ( ERecordType  rectype )

virtual

Stream object list into an EvtMessage.

Caller is responsible for deletion.

Definition at line 67 of file MsgHandler.cc.

{
  if (rectype == MSG_TERMINATE) {
    auto* eod = new EvtMessage(nullptr, 0, rectype);
    return eod;
  }
 
  // which buffer to send? defaults to uncompressed
  auto buf = &m_buf;
  unsigned int flags = 0;
  // but if we have compression enabled then please compress.
  if (m_complevel > 0) {
    // make sure buffer for the compression is big enough.
    m_compBuf.resize(m_buf.size());
    // And call the root compression function
    const int algorithm = m_complevel / 100;
    const int level = m_complevel % 100;
    int irep{0}, nin{(int)m_buf.size()}, nout{nin};
    R__zipMultipleAlgorithm(level, &nin, m_buf.data(), &nout, m_compBuf.data(), &irep,
                            (ROOT::RCompressionSetting::EAlgorithm::EValues) algorithm);
    // it returns the number of bytes of the output in irep. If that is zero or
    // to big compression failed and we transmit uncompressed.
    if (irep > 0 && irep <= nin) {
      //set correct size of compressed message
      m_compBuf.resize(irep);
      // and set pointer to correct buffer for creating message
      buf = &m_compBuf;
      // also add a flag indicating it's compressed
      flags = EvtMessage::c_MsgCompressed;
    }
  }
 
  auto* evtmsg = new EvtMessage(buf->data(), buf->size(), rectype);
  evtmsg->setMsgFlags(flags);
  clear();
 
  return evtmsg;
}

Member Data Documentation

m_buf

CharBuffer m_buf

private

EvtMessage character buffer for encode_msg().

Definition at line 130 of file MsgHandler.h.

m_compBuf

CharBuffer m_compBuf

private

EvtMessage character buffer for compressing/decompressing.

Definition at line 131 of file MsgHandler.h.

m_complevel

int m_complevel

private

compression algorithm * 100 + compression level.

level can be 0 for no compression to 9 for highes compression, algorithm can be one of default (0), zlib (1) or lzma (2)

Definition at line 134 of file MsgHandler.h.

m_inMsg

InMessage m_inMsg

private

Used for deserializing in decode_msg()

Definition at line 133 of file MsgHandler.h.

m_msg

std::unique_ptr<TMessage> m_msg

private

Used for serialising objects into m_buf.

Definition at line 132 of file MsgHandler.h.

---

The documentation for this class was generated from the following files:

• framework/pcore/include/MsgHandler.h
• framework/pcore/src/MsgHandler.cc