# Copyright (C) 2019 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.context_manager import context_manager
from adi.rx_tx import rx_tx
from adi.ad9361 import ad9361


class FMComms5(ad9361):
    """ FMComms5 Dual Transceiver Evaluation Board """

    _complex_data = True
    _split_cores = True
    _rx_channel_names = [
        "voltage0",
        "voltage1",
        "voltage2",
        "voltage3",
        "voltage4",
        "voltage5",
        "voltage6",
        "voltage7",
    ]
    _tx_channel_names = [
        "voltage0",
        "voltage1",
        "voltage2",
        "voltage3",
        "voltage4",
        "voltage5",
        "voltage6",
        "voltage7",
    ]
    _device_name = ""

    def __init__(self, uri=""):
        context_manager.__init__(self, uri, self._device_name)
        self._ctrl = self._ctx.find_device("ad9361-phy")
        self._ctrl_b = self._ctx.find_device("ad9361-phy-B")
        self._rxadc = self._ctx.find_device("cf-ad9361-A")
        self._txdac = self._ctx.find_device("cf-ad9361-dds-core-lpc")
        self._rxadc_chip_b = self._ctx.find_device("cf-ad9361-B")
        self._txdac_chip_b = self._ctx.find_device("cf-ad9361-dds-core-B")
        rx_tx.__init__(self)

    @property
    def filter(self):
        """Load FIR filter file. Provide path to filter file to attribute"""
        return self._get_iio_dev_attr("filter_fir_config")

    @filter.setter
    def filter(self, value):
        with open(value, "r") as file:
            data = file.read()
        self.sample_rate = 3000000
        self._set_iio_attr("out", "voltage_filter_fir_en", False, 0)
        self._set_iio_attr("out", "voltage_filter_fir_en", False, 0, self._ctrl_b)
        self._set_iio_dev_attr_str("filter_fir_config", data)
        self._set_iio_dev_attr_str("filter_fir_config", data, self._ctrl_b)
        self._set_iio_attr("out", "voltage_filter_fir_en", False, 1)
        self._set_iio_attr("out", "voltage_filter_fir_en", False, 1, self._ctrl_b)

    @property
    def loopback_chip_b(self):
        """loopback_chip_b: Set loopback mode of second transceiver. Options are:
        0 (Disable), 1 (Digital), 2 (RF)"""
        return self._get_iio_debug_attr("loopback", self._ctrl_b)

    @loopback_chip_b.setter
    def loopback_chip_b(self, value):
        self._set_iio_debug_attr_str("loopback", value, self._ctrl_b)

    @property
    def gain_control_mode_chip_b(self):
        """gain_control_mode_chip_b: Mode of receive path AGC of second transceiver.
        Options are: slow_attack, fast_attack, manual"""
        return self._get_iio_attr("voltage0", "gain_control_mode", False, self._ctrl_b)

    @gain_control_mode_chip_b.setter
    def gain_control_mode_chip_b(self, value):
        self._set_iio_attr("voltage0", "gain_control_mode", False, value, self._ctrl_b)

    @property
    def rx_hardwaregain_chip_b(self):
        """rx_hardwaregain_chip_b: Gain applied to RX path of second transceiver.
        Only applicable when gain_control_mode is set to 'manual'"""
        return self._get_iio_attr("voltage0", "hardwaregain", False, self._ctrl_b)

    @rx_hardwaregain_chip_b.setter
    def rx_hardwaregain_chip_b(self, value):
        if self.gain_control_mode == "manual":
            self._set_iio_attr("voltage0", "hardwaregain", False, value, self._ctrl_b)

    @property
    def tx_hardwaregain_chip_b(self):
        """tx_hardwaregain: Attenuation applied to TX path of second transceiver"""
        return self._get_iio_attr("voltage0", "hardwaregain", True, self._ctrl_b)

    @tx_hardwaregain_chip_b.setter
    def tx_hardwaregain_chip_b(self, value):
        self._set_iio_attr("voltage0", "hardwaregain", True, value, self._ctrl_b)

    @property
    def rx_rf_bandwidth_chip_b(self):
        """rx_rf_bandwidth_chip_b: Bandwidth of front-end analog filter of RX path
         of second transceiver"""
        return self._get_iio_attr("voltage0", "rf_bandwidth", False, self._ctrl_b)

    @rx_rf_bandwidth_chip_b.setter
    def rx_rf_bandwidth_chip_b(self, value):
        self._set_iio_attr("voltage0", "rf_bandwidth", False, value, self._ctrl_b)

    @property
    def tx_rf_bandwidth_chip_b(self):
        """tx_rf_bandwidth_chip_b: Bandwidth of front-end analog filter of TX path
         of second transceiver"""
        return self._get_iio_attr("voltage0", "rf_bandwidth", True, self._ctrl_b)

    @tx_rf_bandwidth_chip_b.setter
    def tx_rf_bandwidth_chip_b(self, value):
        self._set_iio_attr("voltage0", "rf_bandwidth", True, value, self._ctrl_b)

    @property
    def sample_rate(self):
        """sample_rate: Sample rate RX and TX paths in samples per second of
        second transceiver"""
        return self._get_iio_attr("voltage0", "sampling_frequency", False)

    @sample_rate.setter
    def sample_rate(self, rate):
        if rate < 521e3:
            raise ValueError(
                "Error: Does not currently support sample rates below 521e3"
            )

        # fmt: off
        # The following was converted from ad9361_set_bb_rate() in libad9361-iio
        if (rate <= 20000000):
            dec = 4
            fir = [-15,-27,-23,-6,17,33,31,9,-23,-47,-45,-13,34,69,67,21,-49,-102,-99,-32,69,146,143,48,-96,-204,-200,-69,129,278,275,97,-170,
                -372,-371,-135,222,494,497,187,-288,-654,-665,-258,376,875,902,363,-500,-1201,-1265,-530,699,1748,1906,845,-1089,-2922,-3424,
                -1697,2326,7714,12821,15921,15921,12821,7714,2326,-1697,-3424,-2922,-1089,845,1906,1748,699,-530,-1265,-1201,-500,363,902,875,
                376,-258,-665,-654,-288,187,497,494,222,-135,-371,-372,-170,97,275,278,129,-69,-200,-204,-96,48,143,146,69,-32,-99,-102,-49,21,
                67,69,34,-13,-45,-47,-23,9,31,33,17,-6,-23,-27,-15]
            taps = 128
        elif (rate <= 40000000):
            dec = 2;
            fir = [-0,0,1,-0,-2,0,3,-0,-5,0,8,-0,-11,0,17,-0,-24,0,33,-0,-45,0,61,-0,-80,0,104,-0,-134,0,169,-0,
                -213,0,264,-0,-327,0,401,-0,-489,0,595,-0,-724,0,880,-0,-1075,0,1323,-0,-1652,0,2114,-0,-2819,0,4056,-0,-6883,0,20837,32767,
                20837,0,-6883,-0,4056,0,-2819,-0,2114,0,-1652,-0,1323,0,-1075,-0,880,0,-724,-0,595,0,-489,-0,401,0,-327,-0,264,0,-213,-0,
                169,0,-134,-0,104,0,-80,-0,61,0,-45,-0,33,0,-24,-0,17,0,-11,-0,8,0,-5,-0,3,0,-2,-0,1,0,-0, 0]
            taps = 128
        elif (rate <= 53333333):
            dec = 2
            fir = [-4,0,8,-0,-14,0,23,-0,-36,0,52,-0,-75,0,104,-0,-140,0,186,-0,-243,0,314,-0,-400,0,505,-0,-634,0,793,-0,-993,0,1247,-0,-1585,
                0,2056,-0,-2773,0,4022,-0,-6862,0,20830,32767,20830,0,-6862,-0,4022,0,-2773,-0,2056,0,-1585,-0,1247,0,-993,-0,793,0,-634,-0,505,0,
                -400,-0,314,0,-243,-0,186,0,-140,-0,104,0,-75,-0,52,0,-36,-0,23,0,-14,-0,8,0,-4,0]
            taps = 96
        else:
            dec = 2
            fir = [-58,0,83,-0,-127,0,185,-0,-262,0,361,-0,-488,0,648,-0,-853,0,1117,-0,-1466,0,1954,-0,-2689,0,3960,-0,-6825,0,20818,32767,
                20818,0,-6825,-0,3960,0,-2689,-0,1954,0,-1466,-0,1117,0,-853,-0,648,0,-488,-0,361,0,-262,-0,185,0,-127,-0,83,0,-58,0]
            taps = 64
        # fmt: on

        current_rate = self._get_iio_attr("voltage0", "sampling_frequency", False)

        if self._get_iio_attr("out", "voltage_filter_fir_en", False):
            if current_rate <= (25000000 // 12):
                self._set_iio_attr("voltage0", "sampling_frequency", False, 3000000)
                self._set_iio_attr(
                    "voltage0", "sampling_frequency", False, 3000000, self._ctrl_b
                )
            self._set_iio_attr("out", "voltage_filter_fir_en", False, 0)
            self._set_iio_attr("out", "voltage_filter_fir_en", False, 0, self._ctrl_b)

        # Assemble FIR filter config string
        fir_config_string = ""
        fir_config_string += "RX 3 GAIN -6 DEC " + str(dec) + "\n"
        fir_config_string += "TX 3 GAIN 0 INT " + str(dec) + "\n"
        for i in range(taps):
            fir_config_string += str(fir[i]) + "," + str(fir[i]) + "\n"
        fir_config_string += "\n"
        self._set_iio_dev_attr_str("filter_fir_config", fir_config_string)
        self._set_iio_dev_attr_str("filter_fir_config", fir_config_string, self._ctrl_b)

        if rate <= (25000000 // 12):
            readbuf = self._get_iio_dev_attr_str("tx_path_rates", self._ctrl_b)
            dacrate = int(readbuf.split(" ")[1].split(":")[1])
            txrate = int(readbuf.split(" ")[5].split(":")[1])
            max_rate = (dacrate // txrate) * 16
            if max_rate < taps:
                self._set_iio_attr("voltage0", "sampling_frequency", False, 3000000)
                self._set_iio_attr(
                    "voltage0", "sampling_frequency", False, 3000000, self._ctrl_b
                )
            self._set_iio_attr("out", "voltage_filter_fir_en", False, 1)
            self._set_iio_attr("out", "voltage_filter_fir_en", False, 1, self._ctrl_b)
            self._set_iio_attr("voltage0", "sampling_frequency", False, rate)
            self._set_iio_attr(
                "voltage0", "sampling_frequency", False, rate, self._ctrl_b
            )
        else:
            self._set_iio_attr("voltage0", "sampling_frequency", False, rate)
            self._set_iio_attr(
                "voltage0", "sampling_frequency", False, rate, self._ctrl_b
            )
            self._set_iio_attr("out", "voltage_filter_fir_en", False, 1)
            self._set_iio_attr("out", "voltage_filter_fir_en", False, 1, self._ctrl_b)

    @property
    def rx_lo_chip_b(self):
        """rx_lo_chip_b: Carrier frequency of RX path of second transceiver"""
        return self._get_iio_attr("altvoltage0", "frequency", True, self._ctrl_b)

    @rx_lo_chip_b.setter
    def rx_lo_chip_b(self, value):
        self._set_iio_attr("altvoltage0", "frequency", True, value, self._ctrl_b)

    @property
    def tx_lo_chip_b(self):
        """tx_lo_chip_b: Carrier frequency of TX path of second transceiver"""
        return self._get_iio_attr("altvoltage1", "frequency", True, self._ctrl_b)

    @tx_lo_chip_b.setter
    def tx_lo_chip_b(self, value):
        self._set_iio_attr("altvoltage1", "frequency", True, value, self._ctrl_b)
