From 5bd8d414cf4573a316f140d39e85b15b7b36a264 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Thu, 2 Apr 2015 16:48:59 +0800
Subject: [PATCH] gateware/amp: add kernel CPU and mailbox modules

---
 artiq/gateware/amp/__init__.py   |  2 ++
 artiq/gateware/amp/kernel_cpu.py | 53 ++++++++++++++++++++++++++++++++
 artiq/gateware/amp/mailbox.py    | 23 ++++++++++++++
 3 files changed, 78 insertions(+)
 create mode 100644 artiq/gateware/amp/__init__.py
 create mode 100644 artiq/gateware/amp/kernel_cpu.py
 create mode 100644 artiq/gateware/amp/mailbox.py

diff --git a/artiq/gateware/amp/__init__.py b/artiq/gateware/amp/__init__.py
new file mode 100644
index 000000000..daa70deb5
--- /dev/null
+++ b/artiq/gateware/amp/__init__.py
@@ -0,0 +1,2 @@
+from artiq.gateware.amp.kernel_cpu import KernelCPU
+from artiq.gateware.amp.mailbox import Mailbox
diff --git a/artiq/gateware/amp/kernel_cpu.py b/artiq/gateware/amp/kernel_cpu.py
new file mode 100644
index 000000000..e85565306
--- /dev/null
+++ b/artiq/gateware/amp/kernel_cpu.py
@@ -0,0 +1,53 @@
+from migen.fhdl.std import *
+from migen.bank.description import *
+from migen.bus import wishbone
+
+from misoclib.cpu import mor1kx
+from misoclib.mem.sdram.frontend.wishbone2lasmi import WB2LASMI
+from misoclib.soc import mem_decoder
+
+
+class KernelCPU(Module):
+    def __init__(self, platform, lasmim,
+                 exec_address=0x41000000,
+                 main_mem_origin=0x40000000,
+                 l2_size=8192):
+        self._reset = CSRStorage(reset=1)
+
+        # # #
+
+        self._wb_slaves = []
+
+        # CPU core
+        self.clock_domains.cd_sys_kernel = ClockDomain()
+        self.comb += [
+            self.cd_sys_kernel.clk.eq(ClockSignal()),
+            self.cd_sys_kernel.rst.eq(self._reset.storage)
+        ]
+        self.submodules.cpu = RenameClockDomains(
+            mor1kx.MOR1KX(platform, exec_address),
+            "sys_kernel")
+
+        # DRAM access
+        # XXX Vivado 2014.X workaround
+        from mibuild.xilinx.vivado import XilinxVivadoToolchain
+        if isinstance(platform.toolchain, XilinxVivadoToolchain):
+            from migen.fhdl.simplify import FullMemoryWE
+            self.submodules.wishbone2lasmi = FullMemoryWE(
+                WB2LASMI(l2_size//4, lasmim))
+        else:
+            self.submodules.wishbone2lasmi = WB2LASMI(l2_size//4, lasmim)
+        self.add_wb_slave(mem_decoder(main_mem_origin),
+                          self.wishbone2lasmi.wishbone)
+
+    def get_csrs(self):
+        return [self._reset]
+
+    def do_finalize(self):
+        self.submodules.wishbonecon = wishbone.InterconnectShared(
+            [self.cpu.ibus, self.cpu.dbus], self._wb_slaves, register=True)
+
+    def add_wb_slave(self, address_decoder, interface):
+        if self.finalized:
+            raise FinalizeError
+        self._wb_slaves.append((address_decoder, interface))
diff --git a/artiq/gateware/amp/mailbox.py b/artiq/gateware/amp/mailbox.py
new file mode 100644
index 000000000..909d53908
--- /dev/null
+++ b/artiq/gateware/amp/mailbox.py
@@ -0,0 +1,23 @@
+from migen.fhdl.std import *
+from migen.bus import wishbone
+
+
+class Mailbox(Module):
+    def __init__(self):
+        self.i1 = wishbone.Interface()
+        self.i2 = wishbone.Interface()
+
+        # # #
+
+        value = Signal(32)
+        for i in self.i1, self.i2:
+            self.sync += [
+                i.ack.eq(0),
+                If(i.cyc & i.stb & ~i.ack, i.ack.eq(1)),
+
+                i.dat_r.eq(value),
+                If(i.cyc & i.stb & i.we,
+                   [If(i.sel[j], value[j*8:j*8+8].eq(i.dat_w[j*8:j*8+8]))
+                    for j in range(4)]
+                )
+            ]