From: Sebastien Bourdeauducq <sebastien@milkymist.org>
Date: Thu, 25 Jul 2013 15:57:17 +0000 (+0200)
Subject: memtest/LFSR: use module decorators
X-Git-Tag: 24jan2021_ls180~2855
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=8e04de524bf4ec77cd131076c9a8de181f83c8bc;p=litex.git

memtest/LFSR: use module decorators
---

diff --git a/milkymist/memtest/__init__.py b/milkymist/memtest/__init__.py
index 6cc7256d..92522295 100644
--- a/milkymist/memtest/__init__.py
+++ b/milkymist/memtest/__init__.py
@@ -4,10 +4,10 @@ from migen.bank.description import *
 from migen.actorlib import dma_lasmi
 from migen.actorlib.spi import *
 
+@DecorateModule(InsertReset)
+@DecorateModule(InsertCE)
 class LFSR(Module):
 	def __init__(self, n_out, n_state=31, taps=[27, 30]):
-		self.ce = Signal()
-		self.reset = Signal()
 		self.o = Signal(n_out)
 
 		###
@@ -20,18 +20,15 @@ class LFSR(Module):
 			curval.insert(0, nv)
 			curval.pop()
 
-		self.sync += If(self.reset,
-				state.eq(0),
-				self.o.eq(0)
-			).Elif(self.ce,
-				state.eq(Cat(*curval[:n_state])),
-				self.o.eq(Cat(*curval))
-			)
+		self.sync += [
+			state.eq(Cat(*curval[:n_state])),
+			self.o.eq(Cat(*curval))
+		]
 
 def _print_lfsr_code():
 	from migen.fhdl import verilog
 	dut = LFSR(3, 4, [3, 2])
-	print(verilog.convert(dut, ios={dut.ce, dut.o}))
+	print(verilog.convert(dut, ios={dut.ce, dut.reset, dut.o}))
 
 class _LFSRTB(Module):
 	def __init__(self, *args, **kwargs):