1

git-svn-id: svn://db.shs.com.ru/libs@1 a8b55f48-bf90-11e4-a774-851b48703e85

qad_utils/chunkstream.h

@@ -0,0 +1,56 @@
+#ifndef CHUNKSTREAM_H
+#define CHUNKSTREAM_H
+
+#include <QDataStream>
+#include <QBuffer>
+#include <QByteArray>
+
+
+class ChunkStream
+{
+public:
+	ChunkStream(const QByteArray & data) {setSource(data);}
+	ChunkStream(QDataStream & str) {setSource(str);}
+	ChunkStream(QByteArray * data = 0) {setSource(data);}
+	~ChunkStream();
+	
+	template <typename T>
+	struct Chunk {
+		Chunk(int i, const T & d): id(i), data(d) {}
+		int id;
+		T data;
+	};
+	template <typename T> static Chunk<T> chunk(int id, const T & data) {return Chunk<T>(id, data);}
+	
+	void setSource(const QByteArray & data);
+	void setSource(QDataStream & str);
+	void setSource(QByteArray * data);
+	QDataStream & dataStream() {return stream_;}
+	QByteArray data() const {return tmp_data;}
+	bool atEnd() const {return stream_.atEnd();}
+	
+	int read() {stream_ >> last_id >> last_data; return last_id;}
+	int getID() {return last_id;}
+	template <typename T>
+	T getData() const {T ret; QDataStream s(last_data); s >> ret; return ret;}
+	
+private:
+	void _init();
+	
+	int last_id;
+	QByteArray * data_, last_data, tmp_data;
+	QBuffer buffer;
+	QDataStream stream_;
+	
+};
+
+template <typename T>
+ChunkStream & operator <<(ChunkStream & s, const ChunkStream::Chunk<T> & c) {
+	QByteArray ba;
+	QDataStream bas(&ba, QIODevice::WriteOnly);
+	bas << c.data;
+	s.dataStream() << c.id << ba;
+	return s;
+}
+
+#endif // CHUNKSTREAM_H