#ifndef DATASTREAM_H
#define DATASTREAM_H 1

#include <nds/jtypes.h>
#include <stdio.h>
#include "zlib.h"

class DataStream {
	FILE *fd;
	bool ownfd;

	public:
	DataStream();
	DataStream(FILE*);
	DataStream(const char *filename, const char *mode);
	virtual ~DataStream();

	virtual size_t write(const void *ptr, size_t size, size_t nmemb);
	virtual size_t read(void *ptr, size_t size, size_t nmemb);
	virtual long tell() const;
	virtual bool eof() const;

	DataStream &operator <<(s8);
	DataStream &operator >>(s8&);
	DataStream &operator <<(s16);
	DataStream &operator >>(s16&);
	DataStream &operator <<(s32);
	DataStream &operator >>(s32&);

	DataStream &operator <<(u8);
	DataStream &operator >>(u8&);
	DataStream &operator <<(u16);
	DataStream &operator >>(u16&);
	DataStream &operator <<(u32);
	DataStream &operator >>(u32&);

	DataStream &operator <<(bool);
	DataStream &operator >>(bool&);
};

class ZDataStream : public DataStream {
	gzFile gz;

	public:
	ZDataStream(const char *filename, const char *mode);
	virtual ~ZDataStream();

	virtual size_t write(const void *ptr, size_t size, size_t nmemb);
	virtual size_t read(void *ptr, size_t size, size_t nmemb);
	virtual bool eof() const;
	virtual long tell() const;
};

inline DataStream& DataStream::operator <<(u8 i)
{ return *this << (s8)i; }
inline DataStream& DataStream::operator >>(u8 &i)
{ return *this >> (s8&)i; }

inline DataStream& DataStream::operator <<(u16 i)
{ return *this << (s16)i; }
inline DataStream& DataStream::operator >>(u16 &i)
{ return *this >> (s16&)i; }

inline DataStream& DataStream::operator <<(u32 i)
{ return *this << (s32)i; }
inline DataStream& DataStream::operator >>(u32 &i)
{ return *this >> (s32&)i; }

// justified since sizeof(bool) = 1 >.>
inline DataStream& DataStream::operator <<(bool i)
{ return *this << (s8)i; }
inline DataStream& DataStream::operator >>(bool &i)
{ return *this >> (s8&)i; }

#endif /* DATASTREAM_H */