from comm import *


class AppTest:
    def __init__(self, adr_module, adr_sensor):
        self.serial_port = SerialComm('/dev/ttyUSB0')
        self.module = adr_module
        self.sensor = adr_sensor

    def finish(self):
        self.serial_port.close()

    def cmd_version(self):
        resp = self.serial_port.request(self.module, 0, CMD_VERSION,[])
        version = chr(resp[3])+chr(resp[4])+chr(resp[5])
        print("Version: "+version)

    def cmd_echo(self):
        msg=[65,66,67,68,69,70]
        resp = self.serial_port.request(self.module, 0, CMD_ECHO,msg)
        echo = ""
        for r in range(len(msg)):
            echo = echo + chr(resp[3+r])
        print("Echo:"+echo)

    def cmd_set_param(self, param):        
        resp = self.serial_port.request(self.module, self.sensor, (SET+CMD_PARAM),[])     

    def cmd_get_param(self, param):        
        resp = self.serial_port.request(self.module, self.sensor, (CMD_PARAM),[])     

    def cmd_sensor_cnt(self):
        resp = self.serial_port.request(self.module, 0, CMD_SENSOR_CNT,[])
        return resp[3]

    def cmd_sensor_type(self, index):
        resp = self.serial_port.request(self.module, index, CMD_SENSOR_TYPE,[])
        return resp[3]

    def cmd_sensors_type(self, pocet):
        resp = self.serial_port.request(self.module, 0, CMD_SENSOR_TYPE,[])
        typy = [];
        for i in range(pocet):
            typy.append([resp[3+2*i],resp[4+2*i]])
        return typy

    def cmd_sensor_get_params(self, index):
        resp = self.serial_port.request(self.module, index, CMD_PARAM,[])
        return resp

    def cmd_sensor_get_data(self, index):
        resp = self.serial_port.request(self.module, index, CMD_DATA,[])
        return resp

if __name__ == "__main__":
    
    app = AppTest(0x05, 0x0)
    # app.cmd_version()
    # app.cmd_version()
    # app.cmd_echo()
    
    
    pocet = app.cmd_sensor_cnt()
    print("Sensor type")
    for i in range(pocet):
        print(app.cmd_sensor_type(i+1))
    print("Sensors type")
    print(app.cmd_sensors_type(pocet))
    par = app.cmd_sensor_get_params(1)
    print(par)
    app.cmd_sensor_get_data(1)
    app.cmd_sensor_get_data(2)
    app.finish()