From b6c4cdaffec5749d1523639558e36d3f2b312ffb Mon Sep 17 00:00:00 2001
From: Mosleh Uddin <Mosleh.Uddin@analog.com>
Date: Fri, 26 Jul 2024 09:55:51 -0400
Subject: [PATCH] Add high-level 4p-VNA cst2 module

---
 adi/fmc_4p_vna_cst2.py | 102 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 102 insertions(+)
 create mode 100644 adi/fmc_4p_vna_cst2.py

diff --git a/adi/fmc_4p_vna_cst2.py b/adi/fmc_4p_vna_cst2.py
new file mode 100644
index 000000000..bf266a685
--- /dev/null
+++ b/adi/fmc_4p_vna_cst2.py
@@ -0,0 +1,102 @@
+# Copyright (C) 2024 Analog Devices, Inc.
+#
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without modification,
+# are permitted provided that the following conditions are met:
+#     - Redistributions of source code must retain the above copyright
+#       notice, this list of conditions and the following disclaimer.
+#     - Redistributions in binary form must reproduce the above copyright
+#       notice, this list of conditions and the following disclaimer in
+#       the documentation and/or other materials provided with the
+#       distribution.
+#     - Neither the name of Analog Devices, Inc. nor the names of its
+#       contributors may be used to endorse or promote products derived
+#       from this software without specific prior written permission.
+#     - The use of this software may or may not infringe the patent rights
+#       of one or more patent holders.  This license does not release you
+#       from the requirement that you obtain separate licenses from these
+#       patent holders to use this software.
+#     - Use of the software either in source or binary form, must be run
+#       on or directly connected to an Analog Devices Inc. component.
+#
+# THIS SOFTWARE IS PROVIDED BY ANALOG DEVICES "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
+# INCLUDING, BUT NOT LIMITED TO, NON-INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A
+# PARTICULAR PURPOSE ARE DISCLAIMED.
+#
+# IN NO EVENT SHALL ANALOG DEVICES BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, INTELLECTUAL PROPERTY
+# RIGHTS, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
+# BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
+# THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+from adi.ad9083 import ad9083
+from adi.ad9173 import ad9173
+from adi.adf4371 import adf4371
+from adi.adl5960 import adl5960
+from adi.admv8818 import admv8818
+from adi.adrf5720 import adrf5720
+from adi.ad4696 import ad4696
+from adi.gen_mux import genmux
+from adi.one_bit_adc_dac import one_bit_adc_dac
+
+NUM_PORTS = 4
+
+class fmc_4p_vna_cst2(adrf5720, ad9083, admv8818, genmux, adf4371,
+                      adl5960, one_bit_adc_dac):
+    """ FMCVNA 4-Port Vector Network Analyzer Board (CST2) """
+
+    def __init__(self, uri):
+        self.hsdac  = ad9173(uri)
+        self.lo     = adf4371(uri)
+        self.bpf    = admv8818(uri, device_name="admv8818")
+        self.adcmon = ad4696(uri, device_name="ad4696")
+        self.rfin_attenuator    = adrf5720(uri, device_name="adrf5720-rfin")
+        self.rfin2_attenuator   = adrf5720(uri, device_name="adrf5720-rfin2")
+        self.smix               = one_bit_adc_dac(uri, name="smix")
+        self.lmix               = one_bit_adc_dac(uri, name="lmix")
+
+        self.frontend = []
+        for i in range(0, NUM_PORTS):
+            fe = adl5960(uri, device_name=f"adl5960-{i+1}")
+            self.frontend.append(fe)
+
+        ######## MUX ########
+        self.rfin_mux       = genmux(uri, device_name="mux-rfin0")
+        self.port_rfin_en   = one_bit_adc_dac(uri, name="port-rfin-en")
+
+        self.siglo_mux      = genmux(uri, device_name="mux-adf4371-siglo")
+
+        self.freq_src_sel_mux = genmux(uri, device_name="mux-freq-src-sel")
+        self.lo_rfin          = genmux(uri, device_name="mux-lo-rfin")
+
+        self.hsdac0_mux = genmux(uri, device_name="mux-dac0-rfout")
+        self.hsdac1_mux = genmux(uri, device_name="mux-dac1-rfout")
+
+        ad9083.__init__(self, uri)
+        one_bit_adc_dac.__init__(self, uri, name="misc-gpio-ctrl")
+
+        self._rxadc.set_kernel_buffers_count(1)
+
+
+    @property
+    def active_port(self) -> str:
+        # assume the enabled LED designates the active port
+        return self.rfin_mux.select
+
+
+    @active_port.setter
+    def active_port(self, port: str) -> None:
+        """ Configure all dependencies to enable port """
+        # disable all other ports and enable only active port
+        for i in range(0, NUM_PORTS):
+            enabled_n = 1
+            cur_port = f"d{i+1}"
+            if port == cur_port:
+                enabled_n = 0
+
+            setattr(self.port_rfin_en, f"gpio_{cur_port}", enabled_n)
+
+        self.rfin_mux.select = port
+