packed_multi_message.h

Go to the documentation of this file.

//                                                         //
// Copyright (c) 2003-2014 by The University of Queensland //
// Centre for Geoscience Computing                         //
// http://earth.uq.edu.au/centre-geoscience-computing      //
//                                                         //
// Primary Business: Brisbane, Queensland, Australia       //
// Licensed under the Open Software License version 3.0    //
// http://www.opensource.org/licenses/osl-3.0.php          //
//                                                         //

#ifndef __PACKED_MULTI_MESSAGE_H
#define __PACKED_MULTI_MESSAGE_H

//--- MPI ---
#include <mpi.h>

//--- project includes ---
#include "tml/message/multi_message_slab.h"

//--- STL includes ---
#include <string>
using std::string;


class TML_PackedMultiMessage
{
 private:
  MPI_Comm m_comm;  
  char* m_vbuffer;
  int m_vbuffersize; 
  int *m_position; 
  int *m_rpos;     

  int *m_recvcount;
  int *m_displ; //<! the diplacements of the slices in the buffer
  int m_size;
  int m_int_increment,m_dbl_increment; 

 protected:
  void grow();
  void growTo(int);

 public:
  TML_PackedMultiMessage(MPI_Comm,int isize=64);
  virtual ~TML_PackedMultiMessage();

  TML_PackedMultiMessageSlab operator[](int);

  char* buffer(){return m_vbuffer;}; // make protected or make TML_Comm friend?
  int* offsets(){return m_displ;};
  int* sizes(){return m_rpos;};

  void clear();
  void begin_pack(int);
  void begin_unpack(int);

  void append(int,int);
  void append(double,int);
  void append(const string&,int);
  void append(bool,int);

  int pop_int(int);
  double pop_double(int);
  //string pop_string(int);
  string pop_string();
  bool pop_bool(int);
};

#endif // __PACKED_MULTI_MESSAGE_H