AuroraRuntime/Source/Compression/Compressors/DeflateCompressor.hpp

194 lines
5.4 KiB
C++

/***
Copyright (C) 2022 J Reece Wilson (a/k/a "Reece"). All rights reserved.
File: DeflateCompressor.hpp
Date: 2022-2-15
Author: Reece
***/
#pragma once
#include "zlib.h"
namespace Aurora::Compression
{
struct ZIPDeflate : BaseStream
{
CompressInfo meta;
AuInt8 bits_;
ZIPDeflate(const CompressInfo &meta, AuInt8 bits) : meta(meta), BaseStream(meta.uInternalStreamSize), bits_(bits)
{}
~ZIPDeflate()
{
if (this->init_)
{
deflateEnd(&this->ctx_);
}
}
bool Init(const AuSPtr<IO::IStreamReader> &pReader) override
{
this->pReader_ = pReader;
if (!this->IsValid())
{
SysPushErrorMem();
return false;
}
this->ctx_.next_in = this->din_;
auto ret = deflateInit2(&this->ctx_, meta.uCompressionLevel, Z_DEFLATED, this->bits_, 8, Z_DEFAULT_STRATEGY);
if (ret < Z_OK)
{
this->SetLastError(ret, zError(ret));
return false;
}
this->init_ = true;
this->SetArray(this->din_);
this->SetOutArray(this->dout_);
return true;
}
AuStreamReadWrittenPair_t Ingest_s(AuUInt32 input) override
{
AuUInt32 done {}, read {};
if (!this->pReader_)
{
return {};
}
while (read < input)
{
read += IngestForInPointer<Bytef, uInt>(this->pReader_, this->ctx_.next_in, this->ctx_.avail_in, input - read, this);
if (!this->ctx_.avail_in)
{
return {read, done};
}
do
{
auto [pMainDOut, uMainDOutLength] = this->GetDOutPair();
this->ctx_.avail_out = uMainDOutLength;
this->ctx_.next_out = (Bytef *)pMainDOut;
if (!this->ctx_.avail_out)
{
break;
}
auto ret = deflate(&this->ctx_, Z_NO_FLUSH);
if (ret < Z_OK)
{
this->SetLastError(ret, zError(ret));
this->pReader_.reset();
return AuMakePair(read, 0);
}
auto have = uMainDOutLength - this->ctx_.avail_out;
done += have;
if (!Write2(reinterpret_cast<const AuUInt8 *>(pMainDOut),
have))
{
this->pReader_.reset();
this->SetLastError(0x69, "OOM");
return AuMakePair(read, 0);
}
}
while (this->ctx_.avail_out == 0);
}
return {read, done};
}
bool Flush() override
{
return RunFlush(Z_SYNC_FLUSH);
}
bool Finish() override
{
return RunFlush(Z_FINISH);
}
bool RunFlush(int type)
{
if (!this->pReader_)
{
return false;
}
while (this->ctx_.avail_in)
{
do
{
auto [pMainDOut, uMainDOutLength] = this->GetDOutPair();
this->ctx_.avail_out = uMainDOutLength;
this->ctx_.next_out = (Bytef *)pMainDOut;
auto ret = deflate(&this->ctx_, Z_FULL_FLUSH);
if (ret < Z_OK)
{
this->SetLastError(ret, zError(ret));
this->pReader_.reset();
return false;
}
auto have = uMainDOutLength - this->ctx_.avail_out;
if (!Write2(reinterpret_cast<const AuUInt8 *>(this->dout_), have))
{
this->pReader_.reset();
this->SetLastError(0x69, "OOM");
return false;
}
}
while (this->ctx_.avail_out == 0);
}
if (!this->ctx_.avail_in)
{
do
{
auto [pMainDOut, uMainDOutLength] = this->GetDOutPair();
this->ctx_.avail_out = uMainDOutLength;
this->ctx_.next_out = (Bytef *)pMainDOut;
auto ret = deflate(&this->ctx_, type);
if (ret < Z_OK)
{
this->SetLastError(ret, zError(ret));
this->pReader_.reset();
return false;
}
auto have = uMainDOutLength - this->ctx_.avail_out;
if (!Write2(pMainDOut, have))
{
this->pReader_.reset();
this->SetLastError(0x69, "OOM");
return false;
}
} while (this->ctx_.avail_out == 0);
return true;
}
return true;
}
private:
AuSPtr<IO::IStreamReader> pReader_;
z_stream ctx_ {};
bool init_ {};
unsigned char din_[kChunkSize];
unsigned char dout_[kChunkSize];
};
}