From 3680b48216cf2c0cf5d31919a428c29ea28ce574 Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Wed, 18 Feb 2015 21:45:36 +0100
Subject: [PATCH] dump/sigrok: fix against real dumps, now able to import and
 export

---
 litescope/host/dump/sigrok.py | 33 +++++++++++++++++++--------------
 1 file changed, 19 insertions(+), 14 deletions(-)

diff --git a/litescope/host/dump/sigrok.py b/litescope/host/dump/sigrok.py
index 0d664cec..e37ec77c 100644
--- a/litescope/host/dump/sigrok.py
+++ b/litescope/host/dump/sigrok.py
@@ -8,7 +8,7 @@ from collections import OrderedDict
 from litescope.host.dump import *
 
 class SigrokDump(Dump):
-	def __init__(self, init_dump=None):
+	def __init__(self, init_dump=None, samplerate=50000000):
 		Dump.__init__(self)
 		if init_dump:
 			self.vars = init_dump.vars
@@ -25,21 +25,20 @@ class SigrokDump(Dump):
 sigrok version = 0.2.0
 [device 1]
 driver = litescope
-capturefile = {}
+capturefile = logic-1
 unitsize = 1
 total probes = {}
-samplerate = {} MHz
+samplerate = {} KHz
 """.format(
-		name,
 		len(self.vars),
-		50, # XXX add parameter
+		self.samplerate//1000,
 	)
 		for i, var in enumerate(self.vars):
-			r += "probe{} = {}\n".format(i, var.name)
+			r += "probe{} = {}\n".format(i+1, var.name)
 		f.write(r)
 		f.close()
 
-	def write_data(self, name):
+	def write_data(self):
 		# XXX are probes limited to 1 bit?
 		data_bits = math.ceil(len(self.vars)/8)*8
 		data_len = 0
@@ -55,7 +54,7 @@ samplerate = {} MHz
 				except:
 					pass
 			datas.append(data)
-		f = open(name, "wb")
+		f = open("logic-1", "wb")
 		for data in datas:
 			f.write(data.to_bytes(data_bits//8, "big"))
 		f.close()
@@ -65,7 +64,7 @@ samplerate = {} MHz
 		os.chdir(name)
 		f.write("version")
 		f.write("metadata")
-		f.write(name)
+		f.write("logic-1")
 		os.chdir("..")
 		f.close()
 
@@ -77,7 +76,7 @@ samplerate = {} MHz
 		os.chdir(name)
 		self.write_version()
 		self.write_metadata(name)
-		self.write_data(name)
+		self.write_data()
 		os.chdir("..")
 		self.zip(name)
 		shutil.rmtree(name)
@@ -101,14 +100,20 @@ samplerate = {} MHz
 				index = int(m.group(1))
 				name = m.group(2)
 				probes[name] = index
+			m = re.search("samplerate = ([0-9]+) kHz", l, re.I)
+			if m is not None:
+				self.samplerate = int(m.group(1))*1000
+			m = re.search("samplerate = ([0-9]+) mHz", l, re.I)
+			if m is not None:
+				self.samplerate = int(m.group(1))*1000000
 		f.close()
 		return probes
 
-	def read_data(self, name, total_probes):
+	def read_data(self, name, nprobes):
 		datas = []
-		f = open(name, "rb")
+		f = open("logic-1", "rb")
 		while True:
-			data = f.read(math.ceil(total_probes/8))
+			data = f.read(math.ceil(nprobes/8))
 			if data == bytes('', "utf-8"):
 				break
 			data = int.from_bytes(data, "big")
@@ -129,7 +134,7 @@ samplerate = {} MHz
 		for k, v in probes.items():
 			probe_data = []
 			for data in datas:
-				probe_data.append((data >> v) & 0x1)
+				probe_data.append((data >> (v-1)) & 0x1)
 			self.add(Var(k, 1, probe_data))
 
 if __name__ == '__main__':
-- 
2.30.2