forstio/source/buffer.h

#pragma once

#include "error.h"

#include <array>
#include <cstdint>
#include <deque>
#include <list>
#include <string>
#include <vector>

namespace gin {
/*
 * Access class to reduce templated BufferSegments bloat
 */
class Buffer {
private:
	friend class RingBuffer;
	~Buffer() = default;

public:
	virtual size_t readPosition() const = 0;
	virtual size_t readCompositeLength() const = 0;
	virtual size_t readSegmentLength() const = 0;
	virtual void readAdvance(size_t bytes) = 0;

	virtual uint8_t &read(size_t i = 0) = 0;
	virtual const uint8_t &read(size_t i = 0) const = 0;

	virtual size_t writePosition() const = 0;
	virtual size_t writeCompositeLength() const = 0;
	virtual size_t writeSegmentLength() const = 0;
	virtual void writeAdvance(size_t bytes) = 0;

	virtual uint8_t &write(size_t i = 0) = 0;
	virtual const uint8_t &write(size_t i = 0) const = 0;

	/*
	* Sometime buffers need to grow with a little more control
	* than with push and pop for more efficient calls.
	* There is nothing you can do if read hasn't been filled, but at
	* least write can be increased if it is demanded.
	*/
	/// @todo uncomment and implement in child classes
	// virtual Error writeRequireLength(size_t bytes) = 0;

	virtual Error push(const uint8_t &value) = 0;
	virtual Error push(const uint8_t &buffer, size_t size) = 0;
	virtual Error pop(uint8_t &value) = 0;
	virtual Error pop(uint8_t &buffer, size_t size) = 0;

	virtual std::string toString() const = 0;
	virtual std::string toHex() const = 0;
};
/*
* Buffer size meant for default allocation size of the ringbuffer since
* this class currently doesn't support proper resizing
*/
constexpr size_t RING_BUFFER_MAX_SIZE = 4096;
/*
* Buffer wrapping around if read caught up
*/
class RingBuffer final : public Buffer {
private:
	std::vector<uint8_t> buffer;
	size_t read_position;
	size_t write_position;
	bool write_reached_read = false;

public:
	RingBuffer();
	RingBuffer(size_t size);

	inline size_t size() const { return buffer.size(); }

	inline uint8_t &operator[](size_t i) { return buffer[i]; }
	inline const uint8_t &operator[](size_t i) const { return buffer[i]; }

	size_t readPosition() const override;
	size_t readCompositeLength() const override;
	size_t readSegmentLength() const override;
	void readAdvance(size_t bytes) override;

	uint8_t &read(size_t i = 0) override;
	const uint8_t &read(size_t i = 0) const override;

	size_t writePosition() const override;
	size_t writeCompositeLength() const override;
	size_t writeSegmentLength() const override;
	void writeAdvance(size_t bytes) override;

	uint8_t &write(size_t i = 0) override;
	const uint8_t &write(size_t i = 0) const override;

	Error push(const uint8_t &value) override;
	Error push(const uint8_t &buffer, size_t size) override;
	Error pop(uint8_t &value) override;
	Error pop(uint8_t &buffer, size_t size) override;

	std::string toString() const override;
	std::string toHex() const override;
};

/*
 * One time buffer
 */
class ArrayBuffer : public Buffer {
private:
	std::vector<uint8_t> buffer;

	size_t read_position;
	size_t write_position;

public:
	ArrayBuffer(size_t size);

	size_t readPosition() const override;
	size_t readCompositeLength() const override;
	size_t readSegmentLength() const override;
	void readAdvance(size_t bytes) override;

	uint8_t &read(size_t i = 0) override;
	const uint8_t &read(size_t i = 0) const override;

	size_t writePosition() const override;
	size_t writeCompositeLength() const override;
	size_t writeSegmentLength() const override;
	void writeAdvance(size_t bytes) override;

	uint8_t &write(size_t i = 0) override;
	const uint8_t &write(size_t i = 0) const override;

	Error push(const uint8_t &value) override;
	Error push(const uint8_t &buffer, size_t size) override;
	Error pop(uint8_t &value) override;
	Error pop(uint8_t &buffer, size_t size) override;
};

class ChainArrayBuffer : public Buffer {
private:
	std::deque<ArrayBuffer> buffer;

	size_t read_position;
	size_t write_position;

public:
	ChainArrayBuffer();

	size_t readPosition() const override;
	size_t readCompositeLength() const override;
	size_t readSegmentLength() const override;
	void readAdvance(size_t bytes) override;

	uint8_t &read(size_t i = 0) override;
	const uint8_t &read(size_t i = 0) const override;

	size_t writePosition() const override;
	size_t writeCompositeLength() const override;
	size_t writeSegmentLength() const override;
	void writeAdvance(size_t bytes) override;

	uint8_t &write(size_t i = 0) override;
	const uint8_t &write(size_t i = 0) const override;

	Error push(const uint8_t &value) override;
	Error push(const uint8_t &buffer, size_t size) override;
	Error pop(uint8_t &value) override;
	Error pop(uint8_t &buffer, size_t size) override;
};
} // namespace gin