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

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