add deflate utility

1 files changed, 64 insertions, 0 deletions

diff --git a/util/deflate b/util/deflate
new file mode 100755
index 0000000..925779f
--- /dev/null
+++ b/util/deflate
@@ -0,0 +1,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()