ctodata: make stripping the last byte optional

This commit is contained in:
Campbell Barton 2019-02-15 08:39:15 +11:00
parent 5c432cd11b
commit de9026de6e
1 changed files with 17 additions and 7 deletions

View File

@ -27,25 +27,35 @@
import sys
if len(sys.argv) < 2:
sys.stdout.write("Usage: ctodata <c_file>\n")
argv = sys.argv[:]
strip_byte = False
if "--strip-byte" in argv:
argv.remove("--strip-byte")
strip_byte = True
if len(argv) < 2:
sys.stdout.write("Usage: ctodata <c_file> [--strip-byte]\n")
sys.exit(1)
filename = sys.argv[1]
filename = argv[1]
try:
fpin = open(filename, "r")
except:
sys.stdout.write("Unable to open input %s\n" % sys.argv[1])
sys.stdout.write("Unable to open input %s\n" % argv[1])
sys.exit(1)
data = fpin.read().rsplit("{")[-1].split("}")[0]
data = data.replace(",", " ")
data = data.split()
data = [int(v) for v in data]
# for some reason all data gets trailing byte
last = data.pop()
assert(last == 0)
if strip_byte:
# String data gets trailing byte.
last = data.pop()
assert(last == 0)
data = bytes(data)
dname = filename + ".ctodata"