From 83428961ad3dd74c5aa58859da035630c2fc06cf Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Fri, 22 Jun 2018 00:04:22 +0800
Subject: [PATCH] sayma: add SAWG and JESD to DRTIO master

---
 .../sayma_drtio/repository/sines_drtio.py     |  2 +-
 artiq/gateware/targets/sayma_amc.py           | 21 +++++++++++++------
 2 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/artiq/examples/sayma_drtio/repository/sines_drtio.py b/artiq/examples/sayma_drtio/repository/sines_drtio.py
index b7c26bdcb..7d4e2cf35 100644
--- a/artiq/examples/sayma_drtio/repository/sines_drtio.py
+++ b/artiq/examples/sayma_drtio/repository/sines_drtio.py
@@ -5,7 +5,7 @@ class SAWGTestDRTIO(EnvExperiment):
     def build(self):
         self.setattr_device("core")
         self.setattr_device("ttl_sma_out")
-        self.sawgs = [self.get_device("sawg"+str(8+i)) for i in range(8)]
+        self.sawgs = [self.get_device("sawg"+str(i)) for i in range(16)]
 
     @kernel
     def run(self):
diff --git a/artiq/gateware/targets/sayma_amc.py b/artiq/gateware/targets/sayma_amc.py
index c8c33a5c3..ce469bacf 100755
--- a/artiq/gateware/targets/sayma_amc.py
+++ b/artiq/gateware/targets/sayma_amc.py
@@ -230,9 +230,6 @@ class Master(MiniSoC, AMPSoC, RTMCommon):
         RTMCommon.__init__(self)
         self.config["HMC830_REF"] = "150"
 
-        if with_sawg:
-            warnings.warn("SAWG is not implemented yet, ignoring.")
-
         platform = self.platform
         rtio_clk_freq = 150e6
 
@@ -299,10 +296,22 @@ class Master(MiniSoC, AMPSoC, RTMCommon):
         rtio_channels.append(rtio.Channel.from_phy(phy))
 
         self.submodules.ad9154_crg = jesd204_tools.UltrascaleCRG(platform)
+        if with_sawg:
+            cls = AD9154
+        else:
+            cls = AD9154NoSAWG
+        self.submodules.ad9154_0 = cls(platform, self.crg, self.ad9154_crg, 0)
+        self.submodules.ad9154_1 = cls(platform, self.crg, self.ad9154_crg, 1)
         self.csr_devices.append("ad9154_crg")
-        platform.add_false_path_constraints(
-            self.crg.cd_sys.clk,
-            self.ad9154_crg.cd_jesd.clk)
+        self.csr_devices.append("ad9154_0")
+        self.csr_devices.append("ad9154_1")
+        self.config["HAS_AD9154"] = None
+        self.add_csr_group("ad9154", ["ad9154_0", "ad9154_1"])
+        self.config["RTIO_FIRST_SAWG_CHANNEL"] = len(rtio_channels)
+        rtio_channels.extend(rtio.Channel.from_phy(phy)
+                                for sawg in self.ad9154_0.sawgs +
+                                            self.ad9154_1.sawgs
+                                for phy in sawg.phys)
 
         self.submodules.rtio_moninj = rtio.MonInj(rtio_channels)
         self.csr_devices.append("rtio_moninj")