foxx · November 2, 2016 17:09 · Nov 2, 2016
diff --git a/cembed.py b/cembed.py
@@ -0,0 +1,78 @@
+#!/usr/bin/env python
+"""
+Embed files into C arrays
+
+Hacky solution because xxd didn't have enough features
+"""
+
+from __future__ import with_statement
+from __future__ import print_function
+
+import os
+import sys
+import re
+import glob
+
+def chunks(l, n):
+    """Yield successive n-sized chunks from l."""
+    for i in xrange(0, len(l), n):
+        yield l[i:i + n]
+
+
+def main(testdata_path, c_path, h_path):
+    print("Embedding testdata files into C")
+    os.path.isdir(testdata_path)
+
+    cout = ""
+    cout += "#include \"testdata.h\"\n"
+    cout += "#include <glib.h>\n\n"
+
+    hout = ""
+    hout += "#pragma once\n"
+    hout += "#include <glib.h>\n\n"
+
+    for path in glob.glob(testdata_path+'/*'):
+        if not os.path.isfile(path):
+            continue
+        print("Embedding %s" % path)
+
+        name = re.sub('[^a-zA-Z0-9]', '_', path)
+
+        # read file and build array
+        arr = []
+        with open(path, 'rb') as fh:
+            for c in fh.read():
+                nc = format(ord(c), "#04x");
+                arr.append(nc)
+
+        # output as chunks
+        filearr = "{\n"
+        for chunk in chunks(arr, 12):
+            filearr += "        "
+            filearr += ", ".join(chunk)
+            filearr += ",\n"
+        filearr += "    };"
+
+        # build c file
+        cout += "gchar *\n%s()\n{\n    char out[%d] = %s\n" % (name, len(arr), filearr )
+        cout += "    return g_strndup(out, sizeof(out));\n}\n\n"
+
+        # build h file
+        hout += "gchar *\n%s();\n\n" % (name)
+
+    # write C file
+    with open(c_path, "wb") as fh:
+        fh.write(cout)
+
+    # write H file
+    with open(h_path, "wb") as fh:
+        fh.write(hout)
+
+    print("Files generated OK")
+
+if __name__ == '__main__':
+    import sys
+    if (len(sys.argv) != 4):
+        print("syntax: %s <testdata_path> <c_path> <h_path>" % sys.argv[0])
+        sys.exit(1)
+    main(*sys.argv[1:])