Newer
Older
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
#!/usr/bin/env python3
import sys
import zlib
level = -1
if len(sys.argv) > 2:
level = int(sys.argv[2])
input_data = sys.argv[1].encode("utf-8")
try:
with open(sys.argv[1], "rb") as f:
input_data = f.read()
except FileNotFoundError:
input_data = sys.argv[1].encode("utf-8")
output = zlib.compress(input_data, level=level)
cinfo = (output[0] & 0xF0) >> 4
window_size = 1 << (8 + cinfo)
print(f"// CINFO = {cinfo} (window size = {window_size})")
fdict = output[1] & 0x20
if fdict:
print("// FDICT = 1 (preset dictionary in use)")
else:
print("// FDICT = 0 (no preset dictionary)")
is_final = output[2] & 0x01
block_type = (output[2] & 0x06) >> 1
print(f"// first Huffman block: type={block_type} is_final={is_final}")
print(f"// {sys.argv[1]}")
print("unsigned char const inflate_input[] = {" + ", ".join(map(str, output)) + "};")
print(f"unsigned short const inflate_input_size = {len(output)};")
print(f"unsigned char deflate_output[{len(input_data) + 1}];")
print(f"unsigned short const deflate_output_size = {len(input_data) + 1};")