Improved firmware for vivado "system" example

1 files changed, 6 insertions, 3 deletions

diff --git a/firmware/makehex.py b/firmware/makehex.py
index 1735d86..70282c2 100644
--- a/firmware/makehex.py
+++ b/firmware/makehex.py
@@ -9,13 +9,16 @@
 
 from sys import argv
 
-with open(argv[1], "rb") as f:
+binfile = argv[1]
+nwords = int(argv[2])
+
+with open(binfile, "rb") as f:
     bindata = f.read()
 
-assert len(bindata) < 60*1024
+assert len(bindata) < 4*nwords
 assert len(bindata) % 4 == 0
 
-for i in range(64*1024//4):
+for i in range(nwords):
     if i < len(bindata) // 4:
         w = bindata[4*i : 4*i+4]
         print("%02x%02x%02x%02x" % (w[3], w[2], w[1], w[0]))