from qolab.hardware.basic import BasicInstrument
from qolab.hardware.rf_generator import RFGenerator
import serial


class QOL_LMX2487(RFGenerator):
    """
    QOL lab made RF generator based on TI LMX2487 chip.
    The communication with RF generator is done via nodeMCU controlling esp2866 chip.
    """
    def __init__(self, *args,  port='/dev/ttyUSB0', speed=115200, timeout=1, setToDefaults=True, **kwds):
        super().__init__(*args, **kwds)
        self.config['Device model'] = 'QOL made RF generator based on TI LMX2487 chip'
        self.port = port
        self.speed = speed
        self.timeout = timeout
        self.connection = serial.Serial( self.port, self.speed, timeout=self.timeout)
        self.log = []
        self.logCapacity = 10
        self._FreqFixed = None
        if setToDefaults:
            self.sendSerialCmd("set_lmx2487_board_to_default_state()")

    def add2log(self, text):
        self.log.append(text)
        while len(self.log) > self.logCapacity:
            self.log.pop(0)

    def log2str(self, interval=None):
        strOut = ""
        for e in self.log:
            strOut += e
        return strOut

    def sendSerialCmd(self, cmd):
        self.connection.write(bytes(cmd+'\r','ascii'))
        if "3.4" == serial.__version__:
            # older version style
            resp = self.connection.read_until(terminator=b'> ')
        else:
            # new style after 20180616
            resp = self.connection.read_until(expected=b'> ')
        resp = resp.decode('utf-8')
        self.add2log(resp)
        return resp

    @BasicInstrument.tsdb_append
    def setFreqFixed(self,freq):
        self._FreqFixed = freq
        cmd_str=f'setFreq({freq:.2f})'
        self.sendSerialCmd(cmd_str)

    @BasicInstrument.tsdb_append
    def getFreqFixed(self):
        """
        getFreqFixed unable to get frequency from hardware.
        So we use value from previews setFreqFixed.
        """
        return self._FreqFixed


if __name__ == '__main__': 
    import platform
    if platform.system() == 'Linux':
        rfgen=QOL_LMX2487(port='/dev/ttyUSB0', speed=115200, timeout=1)
    else:
        rfgen=QOL_LMX2487(port='COM4', speed=115200, timeout=1)
    print("testing")
    print('------ Header start -------------')
    print(str.join('\n', rfgen.getHeader()))
    print('------ Header ends  -------------')