from time import sleep
from typing import Dict, Final
from .generic import ModbusRTUDevice


class SensorWiredIAQ(ModbusRTUDevice):
    """
    Wired sensor measuring temperature, relative humidity,
    carbon dioxide and VOC, optionally particulate matter
    """

    DEVICE_CLASS: Final[str] = "IAQ_Wired"
    DEVICE_CODE: Final[int] = 0x0010

    input_registers: Dict[str, int] = {
        "T": 30003,  # deg C
        "T_F": 30004,  # deg F
        "RH": 30005,  # %, from SHT4x
        "CO2": 30006,  # ppm
        "VOC_INDEX": 30007,  # VOC index as calculated by Sensirion library (1 to 500, average 100)
        "VOC_TICKS": 30008,  # raw VOC ticks
        "PMC_MASS_1_0": 30009,  # ug / m^3
        "PMC_MASS_2_5": 30010,  # ug / m^3
        "PMC_MASS_4_0": 30011,  # ug / m^3
        "PMC_MASS_10_0": 30012,  # ug / m^3
        "PMC_NUMBER_0_5": 30013,  # 1 / m^3
        "PMC_NUMBER_1_0": 30014,  # 1 / m^3
        "PMC_NUMBER_2_5": 30015,  # 1 / m^3
        "PMC_NUMBER_4_0": 30016,  # 1 / m^3
        "PMC_NUMBER_10_0": 30017,  # 1 / m^3
        "TYPICAL_PARTICLE_SIZE": 30018,  # nm
        "T_SCD4x": 30019,  # deg C
        "T_SCD4x_F": 30020,  # deg F
        "RH_SCD4x": 30021  # %
    } | ModbusRTUDevice.input_registers
    holding_registers: Dict[str, int] = {
        "MODBUS_ADDR": 40001,
        "BAUDRATE": 40002,
        "LED_ON": 40003,
        "LED_BRIGHTNESS": 40004,
        "LED_SMOOTH": 40005,
        "CO2_ALERT_LIMIT1": 40006,
        "CO2_ALERT_LIMIT2": 40007,
        "SCD4x_T_OFFSET": 40008
    } | ModbusRTUDevice.holding_registers

    @property
    def CO2(self) -> int:
        return int(self.read_register(self.input_registers["CO2"]))

    @property
    def T(self) -> float:
        return self.read_register(self.input_registers["T"], signed=True) / 10

    @property
    def RH(self) -> float:
        return self.read_register(self.input_registers["RH"])

    @property
    def VOC(self):
        return self.read_register(self.input_registers["VOC_INDEX"])

    @property
    def LED(self) -> int:
        return int(self.read_register(self.holding_registers["LED_BRIGHTNESS"]))

    @LED.setter
    def LED(self, value: int):
        if value == 0:
            self.write_register(self.holding_registers["LED_ON"], 0)
        else:
            self.write_register(self.holding_registers["LED_BRIGHTNESS"], value)
            sleep(0.1)
            self.write_register(self.holding_registers["LED_ON"], 1)