from pprint import pp
from driver.kirdy import Kirdy, FilterConfig
import signal
import time
import asyncio

async def enter_dfu_mode(kirdy: Kirdy):
    """
        Enter Device Firmware Upgrade(Dfu) mode
        Please refer to README.md for firmware update instructions.
    """
    await kirdy.device.dfu()

async def active_report(kirdy: Kirdy):
    """
        Configure Kirdy to actively report status to connected socket
        Press Ctrl + C to exit active report mode
    """
    async for data in kirdy.report_mode():
        pp(data)

async def device_cfg(kirdy: Kirdy):
    """
        Configure Kirdy's network and board specific transconductance settings.
        These configs are saved to flash immediately after command is processed. 
    """
    # Kirdy rev0_3's gain and transconductance varies between boards to maximize the
    # PD current range resolution.
    await kirdy.device.set_pd_mon_fin_gain(1.0)
    await kirdy.device.set_pd_mon_transconductance(1/1000.0)

    # Network Settings will be updated on next reboot.
    await kirdy.device.set_ip_settings(
        addr="192.168.1.128",
        port=1337,
        prefix_len=24,
        gateway="192.168.1.1"
    )
    # Hard reset Kirdy.
    await kirdy.device.hard_reset()

async def ld_thermostat_cfg(kirdy: Kirdy):
    """
        Control and config laser diode and thermostat parameters.
    """
    # Load the laser diode and thermostat settings from flash memory.
    await kirdy.device.restore_settings_from_flash()

    # Power off the laser diode & thermostat and clear any asserted alarm
    await kirdy.laser.set_power_on(False)
    await kirdy.laser.clear_alarm()
    await kirdy.thermostat.set_power_on(False)
    await kirdy.thermostat.clear_alarm()

    # Set the laser diode terminals not to be shorted
    await kirdy.laser.set_ld_terms_short(False)

    # Do not power up the laser & thermostat during initial startup
    await kirdy.laser.set_default_pwr_on(False)
    await kirdy.thermostat.set_default_pwr_on(False)

    await kirdy.laser.set_i(0)

    # Configure the laser diode output power limit and photodiode parameters
    # Exceeding the power limit triggers overpower protection alarm.
    # The laser diode power will be cut off upon alarm assertion while the thermostat power remains unchanged.
    await kirdy.laser.set_ld_pwr_limit(0.0)
    await kirdy.laser.set_pd_mon_dark_current(0.0)
    await kirdy.laser.set_pd_mon_responsitivity(0.0)

    # Configure the thermostat NTC thermistor parameters.
    await kirdy.thermostat.set_sh_r0(10.0 * 1000)
    await kirdy.thermostat.set_sh_t0(25)
    await kirdy.thermostat.set_sh_beta(3900)

    # Configure the thermostat TEC settings.
    # The actual output current is limited by value set below. 
    await kirdy.thermostat.set_tec_max_cooling_i(1.0)
    await kirdy.thermostat.set_tec_max_heating_i(1.0)
    await kirdy.thermostat.set_tec_max_v(4.0)

    # Configure the thermostat temperature monitor limits.
    # Exceeding the temperature limits trigger over temperature protection alarm.
    # Both laser diode and thermostat power will be cut off upon alarm assertion.
    await kirdy.thermostat.set_temp_mon_upper_limit(70)
    await kirdy.thermostat.set_temp_mon_lower_limit(0)

    # Configure the thermostat ADC Filter Setting / PID Update Rate / Report Rate.
    # The ADC sampling rate determines the report and pid update rate.
    # The chosen filter and sampling rate affects the noise of the readings.
    # For details, please refer to the AD7172 datasheet.
    await kirdy.thermostat.config_temp_adc_filter(FilterConfig.Sinc5Sinc1With50hz60HzRejection.f16sps)

    # Configure the thermostat PID parameters.
    # You can configure the PID parameter by the included autotune script.
    # You should configure the filter sampling rate first before applying pid parameters.
    await kirdy.thermostat.set_temperature_setpoint(25)
    await kirdy.thermostat.set_pid_kp(0.15668282198105507)
    await kirdy.thermostat.set_pid_ki(0.0001281321)
    await kirdy.thermostat.set_pid_kd(13.82367)
    await kirdy.thermostat.set_pid_output_max(1.0)
    await kirdy.thermostat.set_pid_output_min(-1.0)

    # Configure thermostat to run in PID control mode
    await kirdy.thermostat.set_pid_control_mode()
    
    # When control mode is switched from PID to constant current(CC) control mode,
    # thermostat keeps its instantaneous output current unchanged.
    # Thermostat output current should only be set if it is in CC control mode
    # or the value set will be overwritten by PID output. 
    await kirdy.thermostat.set_constant_current_control_mode()
    await kirdy.thermostat.set_tec_i_out(0.0)

    # Save the current settings to flash memory
    await kirdy.device.save_current_settings_to_flash()

    # Power on the laser diode and thermostat
    await kirdy.laser.set_power_on(True)
    await kirdy.thermostat.set_power_on(True)

    pp(await kirdy.device.get_settings_summary())
    pp(await kirdy.device.get_status_report())

async def main():
    kirdy = Kirdy()
    kirdy.start_session(host='192.168.1.128', port=1337)
    await kirdy.wait_until_connected()

    await ld_thermostat_cfg(kirdy)
    # await active_report(kirdy)
    # await device_cfg(kirdy)
    # await enter_dfu_mode(kirdy)

    await kirdy.end_session()

if __name__ == "__main__":
    asyncio.run(main())