From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Wed, 18 Feb 2015 10:59:35 +0000 (+0100)
Subject: continue sigrok export (should almost work)
X-Git-Tag: 24jan2021_ls180~2575^2~21
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=130212039e85ca88d972ddc08c264ac4f61239cc;p=litex.git

continue sigrok export (should almost work)
---

diff --git a/litescope/host/dump.py b/litescope/host/dump.py
index af354c0e..c6788959 100644
--- a/litescope/host/dump.py
+++ b/litescope/host/dump.py
@@ -1,5 +1,6 @@
 import sys
 import os
+import math
 import shutil
 import datetime
 import zipfile
@@ -320,31 +321,44 @@ total probes = {}
 samplerate = {} MHz
 """.format(
 		name,
-		8,  # XXX add parameter
+		len(self.dump.vars),
 		50, # XXX add parameter
 	)
-		# XXX add probe names
+		for i, var in enumerate(self.dump.vars):
+			r += "probe{} = {}\n".format(i, var.name)
 		f.write(r)
 		f.close()
 
 	def create_data(self, name):
+		# XXX are probes limited to 1 bit?
+		data_bits = math.ceil(len(self.dump.vars)/8)*8
+		data_len = 0
+		for var in self.dump.vars:
+			data_len = max(data_len, len(var))
+		datas = []
+		for i in range(data_len):
+			data = 0
+			for j, var in enumerate(reversed(self.dump.vars)):
+				data = data << 1
+				try:
+					data |= var.values[i] %2
+				except:
+					pass
+			datas.append(data)
+
 		f = open(name, "wb")
-		# XXX
-		for i in range(16):
-			f.write(bytes(i))
+		for data in datas:
+			f.write(data.to_bytes(data_bits//8, "big"))
 		f.close()
 
 	def zip(self, name):
-		def zipdir(path, zip):
-			for root, dirs, files in os.walk(path):
-				for file in files:
-					# XXX
-					os.chdir(root)
-					zip.write(file)
-					os.chdir("..")
-		zipf = zipfile.ZipFile(name + ".sr", 'w')
-		zipdir(name, zipf)
-		zipf.close()
+		f = zipfile.ZipFile(name + ".sr", "w")
+		os.chdir(name)
+		f.write("version")
+		f.write("metadata")
+		f.write(name)
+		os.chdir("..")
+		f.close()
 
 	def write(self, filename):
 		name, ext = os.path.splitext(filename)
@@ -357,6 +371,7 @@ samplerate = {} MHz
 		self.create_data(name)
 		os.chdir("..")
 		self.zip(name)
+		shutil.rmtree(name)
 
 def main():
 	dump = Dump()