You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.8 KiB
67 lines
1.8 KiB
#!/usr/bin/env python |
|
|
|
''' |
|
script to create ap_romfs_embedded.h from a set of static files |
|
|
|
Andrew Tridgell |
|
May 2017 |
|
''' |
|
|
|
import os, sys, tempfile, gzip |
|
|
|
def write_encode(out, s): |
|
out.write(s.encode()) |
|
|
|
def embed_file(out, f, idx): |
|
'''embed one file''' |
|
try: |
|
contents = open(f,'rb').read() |
|
except Exception: |
|
print("Failed to embed %s" % f) |
|
return False |
|
write_encode(out, 'static const uint8_t ap_romfs_%u[] = {' % idx) |
|
|
|
# compress it |
|
compressed = tempfile.NamedTemporaryFile() |
|
f = open(compressed.name, "wb") |
|
with gzip.GzipFile(fileobj=f, mode='wb', filename='', compresslevel=9, mtime=0) as g: |
|
g.write(contents) |
|
f.close() |
|
|
|
compressed.seek(0) |
|
b = bytearray(compressed.read()) |
|
compressed.close() |
|
|
|
for c in b: |
|
write_encode(out, '%u,' % c) |
|
write_encode(out, '};\n\n'); |
|
return True |
|
|
|
def create_embedded_h(filename, files): |
|
'''create a ap_romfs_embedded.h file''' |
|
|
|
out = open(filename, "wb") |
|
write_encode(out, '''// generated embedded files for AP_ROMFS\n\n''') |
|
|
|
for i in range(len(files)): |
|
(name, filename) = files[i] |
|
if not embed_file(out, filename, i): |
|
return False |
|
|
|
write_encode(out, '''const AP_ROMFS::embedded_file AP_ROMFS::files[] = {\n''') |
|
|
|
for i in range(len(files)): |
|
(name, filename) = files[i] |
|
print(("Embedding file %s:%s" % (name, filename)).encode()) |
|
write_encode(out, '{ "%s", sizeof(ap_romfs_%u), ap_romfs_%u },\n' % (name, i, i)) |
|
write_encode(out, '};\n') |
|
out.close() |
|
return True |
|
|
|
if __name__ == '__main__': |
|
import sys |
|
flist = [] |
|
for i in range(1, len(sys.argv)): |
|
f = sys.argv[i] |
|
flist.append((f, f)) |
|
create_embedded_h("/tmp/ap_romfs_embedded.h", flist)
|
|
|