Hardware
/
litex
mirror of https://github.com/enjoy-digital/litex.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

tools/litex_term/upload: bufferize only chunks of the file instead of the entire file to speedup upload when used on embedded devices (RPI for example)

This commit is contained in:
Florent Kermarrec 2019-09-12 10:21:37 +02:00
parent 16b6b357ca
commit 6a0a1c9d87
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
litex/tools/litex_term.py
View File

@ -193,20 +193,22 @@ class LiteXTerm:
return 1 return 1
def upload(self, filename, address): def upload(self, filename, address):
with open(filename, "rb") as f: f = open(filename, "rb")
data = f.read() f.seek(0, 2)
print("[LXTERM] Uploading {} to 0x{:08x} ({} bytes)...".format(filename, address, len(data))) length = f.tell()
f.seek(0, 0)
print("[LXTERM] Uploading {} to 0x{:08x} ({} bytes)...".format(filename, address, length))
current_address = address current_address = address
position = 0 position = 0
length = len(data)
start = time.time() start = time.time()
while len(data): remaining = length
while remaining:
sys.stdout.write("|{}>{}| {}%\r".format('=' * (20*position//length), sys.stdout.write("|{}>{}| {}%\r".format('=' * (20*position//length),
' ' * (20-20*position//length), ' ' * (20-20*position//length),
100*position//length)) 100*position//length))
sys.stdout.flush() sys.stdout.flush()
frame = SFLFrame() frame = SFLFrame()
frame_data = data[:sfl_payload_length] frame_data = f.read(min(remaining, sfl_payload_length))
frame.cmd = sfl_cmd_load if not self.no_crc else sfl_cmd_load_no_crc frame.cmd = sfl_cmd_load if not self.no_crc else sfl_cmd_load_no_crc
frame.payload = current_address.to_bytes(4, "big") frame.payload = current_address.to_bytes(4, "big")
frame.payload += frame_data frame.payload += frame_data
@ -214,12 +216,10 @@ class LiteXTerm:
return return
current_address += len(frame_data) current_address += len(frame_data)
position += len(frame_data) position += len(frame_data)
try: remaining -= len(frame_data)
data = data[sfl_payload_length:]
except:
data = []
end = time.time() end = time.time()
elapsed = end - start elapsed = end - start
f.close()
print("[LXTERM] Upload complete ({0:.1f}KB/s).".format(length/(elapsed*1024))) print("[LXTERM] Upload complete ({0:.1f}KB/s).".format(length/(elapsed*1024)))
return length return length