From 948fefa69a3cc8348a4cb42763aac8edcd9e4400 Mon Sep 17 00:00:00 2001
From: Robert Jordens <rj@m-labs.hk>
Date: Mon, 29 Feb 2016 11:48:29 +0100
Subject: [PATCH] gateware.spi: style

---
 artiq/gateware/spi.py | 34 ++++++++++++++++------------------
 1 file changed, 16 insertions(+), 18 deletions(-)

diff --git a/artiq/gateware/spi.py b/artiq/gateware/spi.py
index dabb89a40..f485d345a 100644
--- a/artiq/gateware/spi.py
+++ b/artiq/gateware/spi.py
@@ -282,10 +282,9 @@ class SPIMaster(Module):
         ])
         assert len(xfer) <= len(bus.dat_w)
 
-        # SPI
-        spi = SPIMachine(data_width, clock_width=len(config.div_read),
-                         bits_width=len(xfer.read_length))
-        self.submodules += spi
+        self.submodules.spi = spi = SPIMachine(
+            data_width, clock_width=len(config.div_read),
+            bits_width=len(xfer.read_length))
 
         wb_we = Signal()
         pending = Signal()
@@ -344,12 +343,12 @@ class SPIMaster(Module):
 
         self.comb += [
             clk_t.oe.eq(~config.offline),
+            clk_t.o.eq((spi.cg.clk & spi.cs) ^ config.clk_polarity),
             mosi_t.oe.eq(~config.offline & spi.cs &
                          (spi.oe | ~config.half_duplex)),
-            clk_t.o.eq((spi.cg.clk & spi.cs) ^ config.clk_polarity),
+            mosi_t.o.eq(spi.reg.o),
             spi.reg.i.eq(Mux(config.half_duplex, mosi_t.i,
                              getattr(pads, "miso", mosi_t.i))),
-            mosi_t.o.eq(spi.reg.o),
         ]
 
 
@@ -438,32 +437,31 @@ def _test_gen(bus):
                   hex(wdata), hex(rdata), hex(a), hex(b))
 
 
-
 class _TestPads:
     def __init__(self):
-        self.cs_n = Signal(3)
+        self.cs_n = Signal(2)
         self.clk = Signal()
         self.mosi = Signal()
         self.miso = Signal()
 
 
+class _TestTristate(Module):
+    def __init__(self, t):
+        oe = Signal()
+        self.comb += [
+            t.target.eq(t.o),
+            oe.eq(t.oe),
+            t.i.eq(t.o),
+        ]
+
 if __name__ == "__main__":
     from migen.fhdl.specials import Tristate
 
-    class T(Module):
-        def __init__(self, t):
-            oe = Signal()
-            self.comb += [
-                t.target.eq(t.o),
-                oe.eq(t.oe),
-                t.i.eq(t.o),
-            ]
-    Tristate.lower = staticmethod(lambda dr: T(dr))
-
     pads = _TestPads()
     dut = SPIMaster(pads)
     dut.comb += pads.miso.eq(pads.mosi)
     # from migen.fhdl.verilog import convert
     # print(convert(dut))
 
+    Tristate.lower = _TestTristate
     run_simulation(dut, _test_gen(dut.bus), vcd_name="spi_master.vcd")