DataStream.h

#ifndef _dcl_dis_data_stream_h_
#define _dcl_dis_data_stream_h_

// the class member, DataStream::BufferType is causing warnign 4251.
// disable it until a proper fix is found, as instructed from the enlightening article:
// http://www.unknownroad.com/rtfm/VisualStudio/warningC4251.html
#if _MSC_VER
#pragma warning( push )
#pragma warning( disable : 4251 )
#endif

#include <string>                 // for typedef, member
#include <vector>                 // for typedef, member
#include <dis6/utils/Endian.h>           // for enum
#include <dis6/opendis6_export.h>       // for library symbols
#include <cstdlib>                // for size_t and NULL definition
#include <cstring>                // for memcpy

namespace DIS
{
   class OPENDIS6_EXPORT DataStream
   {
   public:
      DataStream(Endian stream);

      DataStream(const char* buffer, size_t length, Endian stream);
      ~DataStream();

      const char& operator [](unsigned int offset) const;

      void SetStream(const char* buffer, size_t length, Endian order);

      // write operations
      DataStream& operator <<(bool c);
      DataStream& operator <<(char c);
      DataStream& operator <<(unsigned char c);
      DataStream& operator <<(float c);
      DataStream& operator <<(double c);
      DataStream& operator <<(int c);
      DataStream& operator <<(unsigned int c);
      DataStream& operator <<(long long c);
      DataStream& operator <<(unsigned long long c);
      DataStream& operator <<(unsigned short c);
      DataStream& operator <<(short c);

      // read operations
      DataStream& operator >>(bool& c);
      DataStream& operator >>(char& c);
      DataStream& operator >>(unsigned char& c);
      DataStream& operator >>(float& c);
      DataStream& operator >>(double& c);
      DataStream& operator >>(int& c);
      DataStream& operator >>(unsigned int& c);
      DataStream& operator >>(long long& c);
      DataStream& operator >>(unsigned long long& c);
      DataStream& operator >>(unsigned short& c);
      DataStream& operator >>(short& c);

      Endian GetStreamEndian() const;
      Endian GetMachineEndian() const;

      size_t GetWritePos() const;
      size_t GetReadPos() const;

      size_t size() const;

      void clear();

      bool empty() const;

   private:
      template<typename T, typename IterT>
      void IncrementPointer(IterT& iter)
      {
         iter += sizeof(T);
      }

      template<typename T, typename IterT>
      void DecrementPointer(IterT& iter)
      {
         iter -= sizeof(T);
      }

      template<typename T>
      void WriteAlgorithm(T t)
      {
         char* ch = reinterpret_cast<char*>( &t );
         DoFlip( ch , sizeof(T) );
         DoWrite( ch , sizeof(T) );
         IncrementPointer<T>( _write_pos );
      }

      template<typename T>
      void ReadAlgorithm(T& t)
      {
         char ch[sizeof(T)];
         DoRead( ch , sizeof(T) );
         DoFlip( ch , sizeof(T) );
         memcpy(&t, ch, sizeof(t));
         IncrementPointer<T>( _read_pos );
      }

      void DoFlip(char* buf, size_t bufsize);

      void DoWrite(const char* buf, size_t bufsize);

      void DoRead(char* ch, size_t bufsize);


      typedef std::vector<char> BufferType;
      //const BufferType& GetBuffer() const;

      BufferType _buffer;

      size_t _read_pos;
      size_t _write_pos;

      Endian _stream_endian;

      Endian _machine_endian;
   };
}

#if _MSC_VER
#pragma warning( pop )
#endif

#endif  // _dcl_dis_data_stream_h_