From e099f4d52f60511a9016d7e9489124956d3a73da Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sebastien@milkymist.org>
Date: Sun, 4 Dec 2011 22:41:50 +0100
Subject: [PATCH] Reset insertion

---
 migen/fhdl/convtools.py | 76 +++++++++++++++++++++++++++++++++++++++++
 migen/fhdl/verilog.py   | 58 +++----------------------------
 2 files changed, 81 insertions(+), 53 deletions(-)
 create mode 100644 migen/fhdl/convtools.py

diff --git a/migen/fhdl/convtools.py b/migen/fhdl/convtools.py
new file mode 100644
index 00000000..df2855b7
--- /dev/null
+++ b/migen/fhdl/convtools.py
@@ -0,0 +1,76 @@
+from .structure import *
+
+class Namespace:
+	def __init__(self):
+		self.counts = {}
+		self.sigs = {}
+	
+	def GetName(self, sig):
+		try:
+			n = self.sigs[sig]
+			if n:
+				return sig.name + "_" + str(n)
+			else:
+				return sig.name
+		except KeyError:
+			try:
+				n = self.counts[sig.name]
+			except KeyError:
+				n = 0
+			self.sigs[sig] = n
+			self.counts[sig.name] = n + 1
+			if n:
+				return sig.name + "_" + str(n)
+			else:
+				return sig.name
+
+def ListSignals(node):
+	if isinstance(node, Constant):
+		return set()
+	elif isinstance(node, Signal):
+		return {node}
+	elif isinstance(node, Operator):
+		l = list(map(ListSignals, node.operands))
+		return set().union(*l)
+	elif isinstance(node, Slice):
+		return ListSignals(node.value)
+	elif isinstance(node, Cat):
+		l = list(map(ListSignals, node.l))
+		return set().union(*l)
+	elif isinstance(node, Assign):
+		return ListSignals(node.l) | ListSignals(node.r)
+	elif isinstance(node, StatementList):
+		l = list(map(ListSignals, node.l))
+		return set().union(*l)
+	elif isinstance(node, If):
+		return ListSignals(node.cond) | ListSignals(node.t) | ListSignals(node.f)
+	elif isinstance(node, Fragment):
+		return ListSignals(node.comb) | ListSignals(node.sync)
+	else:
+		raise TypeError
+
+def ListTargets(node):
+	if isinstance(node, Signal):
+		return {node}
+	elif isinstance(node, Slice):
+		return ListTargets(node.value)
+	elif isinstance(node, Cat):
+		l = list(map(ListTargets, node.l))
+		return set().union(*l)
+	elif isinstance(node, Assign):
+		return ListTargets(node.l)
+	elif isinstance(node, StatementList):
+		l = list(map(ListTargets, node.l))
+		return set().union(*l)
+	elif isinstance(node, If):
+		return ListTargets(node.t) | ListTargets(node.f)
+	else:
+		raise TypeError
+
+def InsertReset(rst, sl):
+	targets = ListTargets(sl)
+	resetcode = []
+	for t in targets:
+		if not t.variable:
+			resetcode.append(Assign(t, t.reset))
+	return If(rst == 1, resetcode, sl)
\ No newline at end of file
diff --git a/migen/fhdl/verilog.py b/migen/fhdl/verilog.py
index a3f02624..e5acfd1e 100644
--- a/migen/fhdl/verilog.py
+++ b/migen/fhdl/verilog.py
@@ -1,60 +1,12 @@
 from .structure import *
+from .convtools import *
 from functools import partial
 
-class Namespace:
-	def __init__(self):
-		self.counts = {}
-		self.sigs = {}
-	
-	def GetName(self, sig):
-		try:
-			n = self.sigs[sig]
-			if n:
-				return sig.name + "_" + str(n)
-			else:
-				return sig.name
-		except KeyError:
-			try:
-				n = self.counts[sig.name]
-			except KeyError:
-				n = 0
-			self.sigs[sig] = n
-			self.counts[sig.name] = n + 1
-			if n:
-				return sig.name + "_" + str(n)
-			else:
-				return sig.name
-
-def ListSignals(node):
-	if isinstance(node, Constant):
-		return set()
-	elif isinstance(node, Signal):
-		return {node}
-	elif isinstance(node, Operator):
-		l = list(map(ListSignals, node.operands))
-		return set().union(*l)
-	elif isinstance(node, Slice):
-		return ListSignals(node.value)
-	elif isinstance(node, Cat):
-		l = list(map(ListSignals, node.l))
-		return set().union(*l)
-	elif isinstance(node, Assign):
-		return ListSignals(node.l) | ListSignals(node.r)
-	elif isinstance(node, StatementList):
-		l = list(map(ListSignals, node.l))
-		return set().union(*l)
-	elif isinstance(node, If):
-		return ListSignals(node.cond) | ListSignals(node.t) | ListSignals(node.f)
-	elif isinstance(node, Fragment):
-		return ListSignals(node.comb) | ListSignals(node.sync)
-	else:
-		raise TypeError
-
-def Convert(f, ins, outs, name="top"):
+def Convert(f, ins, outs, name="top", clkname="sys_clk", rstname="sys_rst"):
 	ns = Namespace()
 	
-	clks = Signal(name="sys_clk")
-	rsts = Signal(name="sys_rst")
+	clks = Signal(name=clkname)
+	rsts = Signal(name=rstname)
 	clk = ns.GetName(clks)
 	rst = ns.GetName(rsts)
 	
@@ -129,7 +81,7 @@ def Convert(f, ins, outs, name="top"):
 	
 	if f.sync.l:
 		r += "always @(posedge " + clk + ") begin\n"
-		r += printnode(1, f.sync)
+		r += printnode(1, InsertReset(rsts, f.sync))
 		r += "end\n\n"
 	
 	r += "endmodule\n"
-- 
2.30.2