61 lines
1.2 KiB
C
61 lines
1.2 KiB
C
|
//
|
||
|
|
// DeflateStream.h
|
||
|
|
// OpenSpades
|
||
|
|
//
|
||
|
|
// Created by yvt on 7/16/13.
|
||
|
|
// Copyright (c) 2013 yvt.jp. All rights reserved.
|
||
|
|
//
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "IStream.h"
|
||
|
|
#include <zlib.h>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
namespace spades {
|
||
|
|
class DeflateStream: public IStream {
|
||
|
|
IStream *baseStream;
|
||
|
|
CompressMode mode;
|
||
|
|
z_stream zstream;
|
||
|
|
bool autoClose;
|
||
|
|
bool valid;
|
||
|
|
uint64_t position;
|
||
|
|
|
||
|
|
// when compressing:
|
||
|
|
// input buffer
|
||
|
|
// when decompressing:
|
||
|
|
// output buffer
|
||
|
|
std::vector<char> buffer;
|
||
|
|
|
||
|
|
|
||
|
|
// input buffer that wasn't used in FillBuffer
|
||
|
|
// is stored here
|
||
|
|
std::vector<char> nextbuffer;
|
||
|
|
|
||
|
|
// decompression only
|
||
|
|
bool reachedEOF;
|
||
|
|
size_t bufferPos;
|
||
|
|
|
||
|
|
void CompressBuffer();
|
||
|
|
void FillBuffer();
|
||
|
|
public:
|
||
|
|
DeflateStream(IStream *stream, CompressMode mode, bool autoClose = false);
|
||
|
|
virtual ~DeflateStream();
|
||
|
|
|
||
|
|
virtual int ReadByte();
|
||
|
|
virtual size_t Read(void *, size_t bytes);
|
||
|
|
|
||
|
|
virtual void WriteByte(int);
|
||
|
|
virtual void Write(const void *, size_t bytes);
|
||
|
|
|
||
|
|
virtual uint64_t GetPosition();
|
||
|
|
virtual void SetPosition(uint64_t);
|
||
|
|
|
||
|
|
virtual uint64_t GetLength();
|
||
|
|
virtual void SetLength(uint64_t);
|
||
|
|
|
||
|
|
/** Must be called when all data was written, in case of compressing, or output will be corrupted */
|
||
|
|
void DeflateEnd();
|
||
|
|
};
|
||
|
|
}
|