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
58
59
60
61
62
63
64
|
#!/usr/bin/env python3
# vim:tabstop=4 softtabstop=4 shiftwidth=4 textwidth=160 smarttab expandtab colorcolumn=160
"""deflate - compression utility for zlib-deflate-nostdlib
deflate compresses the input file (or, if the argument is not a filename, the input string) using zlib
and outputs the compressed data in a format suitable for C/C++ files.
"""
import argparse
import sys
import zlib
def main():
parser = argparse.ArgumentParser(
formatter_class=argparse.RawDescriptionHelpFormatter, description=__doc__
)
parser.add_argument(
"--variable-name",
type=str,
default="inflate_input",
help="variable name for compressed data",
)
parser.add_argument(
"--data-type",
type=str,
default="unsigned char const",
help="variable type for compressed data",
)
parser.add_argument(
"--size-type",
type=str,
default="unsigned short const",
help="variable type for compressed data size",
)
parser.add_argument("--level", type=int, default=-1, help="zlib compression level")
parser.add_argument(
"file_or_string", type=str, help="filename or string to compress"
)
args = parser.parse_args()
try:
with open(args.file_or_string, "rb") as f:
input_data = f.read()
except FileNotFoundError:
input_data = args.file_or_string.encode("utf-8")
output = zlib.compress(input_data, args.level)
fdict = output[1] & 0x20
assert fdict == 0
print(f"// {args.file_or_string}")
print(f"{args.size_type} {args.variable_name}_size = {len(output)};")
print(
f"{args.data_type} {args.variable_name}[] = {{"
+ ", ".join(map(str, output))
+ "};"
)
if __name__ == "__main__":
main()
|
