SHS/libs
Archived
3
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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

Browse Source
This commit is contained in:
Пелипенко Иван
2019-10-28 09:10:38 +00:00
parent 58d504dfb2
commit 7d360917f5
2 changed files with 115 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
qad/utils/chunkstream.h
View File

@@ -4,14 +4,20 @@
#include <QDataStream>
#include <QBuffer>
#include <QByteArray>
#include <QMap>
class ChunkStream
{
public:
ChunkStream(const QByteArray & data) {setSource(data);}
ChunkStream(QDataStream & str) {setSource(str);}
ChunkStream(QByteArray * data = 0) {setSource(data);}
enum Version {
Version_1 /*! First, old version */,
Version_2 /*! Second, more optimized version */ = 2,
};
ChunkStream(const QByteArray & data): version_(Version_2) {setSource(data);}
ChunkStream(QDataStream & str): version_(Version_2) {setSource(str);}
ChunkStream(QByteArray * data = 0, Version v = Version_2): version_(v) {setSource(data);}
ChunkStream(Version v): version_(v) {setSource(0);}
~ChunkStream();
template <typename T>
@@ -30,21 +36,38 @@ public:
QDataStream & dataStream() {return stream_;}
QByteArray data() const {return tmp_data;}
bool atEnd() const {return stream_.atEnd();}
Version version() const {return (Version)version_;}
int read() {stream_ >> last_id >> last_data; return last_id;}
int read();
void readAll();
int getID() {return last_id;}
template <typename T>
T getData() const {T ret; QDataStream s(last_data); s.setVersion(QDataStream::Qt_4_8); s >> ret; return ret;}
template <typename T>
void get(T & v) const {v = getData<T>();}
template <typename T>
const ChunkStream & get(int id, T & v) const {
QByteArray ba = data_map.value(id);
if (!ba.isEmpty()) {
QDataStream s(ba); s.setVersion(QDataStream::Qt_4_8);
s >> v;
}
return *this;
}
private:
void _init();
static uint readVInt(QDataStream & s);
static void writeVInt(QDataStream & s, uint val);
int last_id;
uchar version_;
QByteArray * data_, last_data, tmp_data;
QBuffer buffer;
QDataStream stream_;
QMap<int, QByteArray> data_map;
template <typename T> friend ChunkStream & operator <<(ChunkStream & s, const ChunkStream::Chunk<T> & c);
};
template <typename T>
@@ -53,7 +76,19 @@ ChunkStream & operator <<(ChunkStream & s, const ChunkStream::Chunk<T> & c) {
QDataStream bas(&ba, QIODevice::WriteOnly);
bas.setVersion(QDataStream::Qt_4_8);
bas << c.data;
s.dataStream() << c.id << ba;
switch (s.version_) {
case ChunkStream::Version_1:
s.stream_ << c.id << ba;
break;
case ChunkStream::Version_2:
if (s.data_->isEmpty())
s.stream_ << uchar(uchar(s.version_) | 0x80);
ChunkStream::writeVInt(s.stream_, c.id);
ChunkStream::writeVInt(s.stream_, ba.size());
s.stream_.writeRawData(ba.data(), ba.size());
break;
default: break;
}
return s;
}