3escollatedpacket.h

//
// author: Kazys Stepanas
//
#ifndef _3ESCOLLATEDPACKET_H_
#define _3ESCOLLATEDPACKET_H_

#include "3es-core.h"

#include "3esconnection.h"
#include "3espacketheader.h"

namespace tes
{
  struct CollatedPacketMessage;
  struct CollatedPacketZip;
  class PacketWriter;

  class _3es_coreAPI CollatedPacket : public Connection
  {
  public:
    static const size_t Overhead;
    static const unsigned InitialCursorOffset;
    static const uint16_t MaxPacketSize;

  public:
    CollatedPacket(bool compress, uint16_t bufferSize = 0xff00u);

    CollatedPacket(unsigned bufferSize, unsigned maxPacketSize);

    ~CollatedPacket();

    bool compressionEnabled() const;

    unsigned maxPacketSize() const;

    void reset();

    int add(const PacketWriter &packet);

    int add(const uint8_t *buffer, uint16_t byteCount);

    bool finalise();

    const uint8_t *buffer(unsigned &byteCount) const;

    unsigned collatedBytes() const;

    //-------------------------------------------
    // Connection methods.
    //-------------------------------------------

    void close() override;

    void setActive(bool active) override;

    bool active() const override;

    const char *address() const override;

    uint16_t port() const override;

    bool isConnected() const override;

    int create(const Shape &shape) override;

    int destroy(const Shape &shape) override;

    int update(const Shape &shape) override;

    int updateTransfers(unsigned byteLimit) override;

    int updateFrame(float dt, bool flush = true) override;

    unsigned referenceResource(const Resource *resource) override;

    unsigned releaseResource(const Resource *resource) override;

    bool sendServerInfo(const ServerInfoMessage &info) override;

    int send(const uint8_t *data, int byteCount) override;

  private:
    void init(bool compress, unsigned bufferSize, unsigned maxPacketSize);

    static void expand(unsigned expandBy, uint8_t *&buffer, unsigned &bufferSize, unsigned currentDataCount, unsigned maxPacketSize);

    CollatedPacketZip *_zip;  
    uint8_t *_buffer;         
    uint8_t *_finalBuffer;
    unsigned _bufferSize;     
    unsigned _finalBufferSize;
    unsigned _finalPacketCursor;  
    unsigned _cursor;         
    unsigned _maxPacketSize;  
    bool _finalised;          
    bool _active;             
  };


  inline bool CollatedPacket::compressionEnabled() const
  {
    return _zip != nullptr;
  }


  inline unsigned CollatedPacket::maxPacketSize() const
  {
    return _maxPacketSize;
  }


  inline unsigned CollatedPacket::collatedBytes() const
  {
    return _cursor;
  }
}

#endif // _3ESCOLLATEDPACKET_H_