M-Labs/artiq
2
0
Fork 2
mirror of https://github.com/m-labs/artiq.git synced 2024-12-26 03:38:25 +08:00
Code Issues Projects Releases Wiki Activity

bit2bin: cleanup

Browse Source
This commit is contained in:
Robert Jördens 2017-11-09 12:52:36 +01:00
parent 76ddb063cf
commit 4880e4225d
1 changed files with 34 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
artiq/frontend/bit2bin.py
View File

@ -17,38 +17,39 @@ def flip32(data):
def bit2bin(bit, bin, flip=False): def bit2bin(bit, bin, flip=False):
with open(bit, "rb") as bitfile: l, = struct.unpack(">H", bit.read(2))
l, = struct.unpack(">H", bitfile.read(2))
if l != 9: if l != 9:
raise ValueError("Missing <0009> header, not a bit file") raise ValueError("Missing <0009> header, not a bit file")
_ = bit.read(l) # unknown data
bitfile.read(l) l, = struct.unpack(">H", bit.read(2))
d = bitfile.read(*struct.unpack(">H", bitfile.read(2))) if l != 1:
if d != b"a": raise ValueError("Missing <0001> header, not a bit file")
raise ValueError("Missing <a> header, not a bit file")
d = bitfile.read(*struct.unpack(">H", bitfile.read(2)))
print("Design name:", d)
while True: while True:
key = bitfile.read(1) key = bit.read(1).decode()
if not key: if not key:
break break
if key in b"bcd": if key in "abcd":
d = bitfile.read(*struct.unpack(">H", bitfile.read(2))) d = bit.read(*struct.unpack(">H", bit.read(2)))
name = {b"b": "Partname", b"c": "Date", b"d": "Time"}[key] assert d.endswith(b"\x00")
print(name, d) d = d.decode()
elif key == b"e": name = {
l, = struct.unpack(">I", bitfile.read(4)) "a": "Design",
print("found binary data length:", l) "b": "Part name",
d = bitfile.read(l) "c": "Date",
"d": "Time"
}[key]
print("{}: {}".format(name, d))
elif key == "e":
l, = struct.unpack(">I", bit.read(4))
print("Bitstream payload length: {:#x}".format(l))
d = bit.read(l)
if flip: if flip:
d = flip32(d) d = flip32(d)
with open(bin, "wb") as f: bin.write(d)
f.write(d)
else: else:
d = bitfile.read(*struct.unpack(">H", bitfile.read(2))) d = bit.read(*struct.unpack(">H", bit.read(2)))
print("Unexpected key: ", key, d) print("Unexpected key: {}: {}".format(key, d))
if __name__ == "__main__": if __name__ == "__main__":
@ -64,4 +65,5 @@ if __name__ == "__main__":
help="Output bin file name") help="Output bin file name")
args = parser.parse_args() args = parser.parse_args()
bit2bin(args.bitfile, args.binfile, args.flip) with open(args.bitfile, "rb") as f, open(args.binfile, "wb") as g:
bit2bin(f, g, args.flip)