1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
#pragma once
#include <cstdint>
#include <string>
namespace TehImage
{
struct HuffmanTree
{
uint16_t val = 0xFFFF;
HuffmanTree* left = nullptr; // 0
HuffmanTree* right = nullptr; // 1
HuffmanTree() = default;
~HuffmanTree();
void free();
};
struct StreamData
{
uint8_t* data;
unsigned long length;
unsigned long pos;
};
class ZLibInflator
{
private:
HuffmanTree tree;
HuffmanTree distTree;
bool staticTree = false;
bool haveTree = false;
public:
ZLibInflator() = default;
~ZLibInflator() = default;
static bool nextBit(StreamData* stream);
static uint16_t nextBits(StreamData* stream, int bits); // Max 16 bits
void calculateCodes(uint8_t* lengths, uint16_t* codesOut, int codeCount);
void buildHuffmanTree(uint8_t* lengths, uint16_t* codes, int codeCount);
void buildHuffmanTree(uint8_t* lengths, uint16_t* codes, int codeCount, HuffmanTree* treeOut);
void buildStaticHuffmanTree();
void buildStaticHuffmanTree(HuffmanTree* treeOut, HuffmanTree* distTreeOut);
void buildDynamicHuffmanTree(StreamData* stream);
void buildDynamicHuffmanTree(StreamData* stream, HuffmanTree* treeOut, HuffmanTree* distTreeOut);
uint16_t getNextCode(StreamData* stream);
uint16_t getNextCode(StreamData* stream, HuffmanTree* tree);
int decodeData(uint8_t* data, unsigned long length, uint8_t* out, unsigned long outLength);
};
}
|
