lsnes/include/library/zlibstream.hpp

#ifndef _library__zlibstream__hpp__included__
#define _library__zlibstream__hpp__included__

#include <vector>
#include <list>
#include <cstdint>
#include <zlib.h>
#define ZLIB_PAGE_STORAGE 65500

/**
 * Zlib stream.
 */
struct zlibstream
{
/**
 * Create new zlib stream.
 *
 * Parameter compression: The compression level.
 */
	zlibstream(unsigned compression);
/**
 * Destroy zlib stream.
 */
	~zlibstream();
/**
 * Reset the stream, adding some initial data.
 *
 * Parameter data: The data to write (may be NULL if datalen=0).
 * Parameter datalen: The length of data.
 */
	void reset(uint8_t* data, size_t datalen);
/**
 * Write data into stream.
 *
 * Parameter data: The data to write.
 * Parameter datalen: The length of data.
 */
	void write(uint8_t* data, size_t datalen);
/**
 * Read the data as buffer.
 *
 * Parameter out: The buffer to write to.
 */
	void read(std::vector<char>& out);
/**
 * Read the data so far as buffer and empty compressed data.
 *
 * This performs a sync flush.
 *
 * Parameter out: The buffer to write to.
 */
	void readsync(std::vector<char>& out);
/**
 * Clear the output stream and add some uncompressed data.
 *
 * This is useful after doing readsync().
 *
 * Parameter data: The data to write (may be NULL if datalen=0).
 * Parameter datalen: The length of data.
 */
	void adddata(uint8_t* data, size_t datalen);
/**
 * Set flag.
 *
 * Parameter f: New flag value.
 */
	void set_flag(bool f);
/**
 * Get flag.
 *
 * Returns: Flag value.
 */
	bool get_flag();
private:
	struct page
	{
		uint8_t data[ZLIB_PAGE_STORAGE];
		size_t used;
	};

	void _read(std::vector<char>& out, int mode);
	void _reset(uint8_t* data, size_t datalen, bool doactually);
	void flushpage(uint8_t* data, size_t datalen, int mode);
	zlibstream(zlibstream&);
	zlibstream& operator=(zlibstream&);
	z_stream z;
	size_t streamsize;
	std::list<page> storage;
	bool flag;
};

#endif