forked from M-Labs/kirdy
744 lines
28 KiB
Python
744 lines
28 KiB
Python
import socket
|
|
import asyncio
|
|
import json
|
|
|
|
# Data Type Enums
|
|
IP_SETTINGS = "ip_settings"
|
|
TEMP_ADC_FILTER = "temp_adc_filter"
|
|
DATA_F32 = "data_f32"
|
|
DATA_BOOL = "data_bool"
|
|
|
|
TARGET_DEVICE = "device_cmd"
|
|
TARGET_LD = "laser_diode_cmd"
|
|
TARGET_THERMOSTAT = "thermostat_cmd"
|
|
|
|
class CmdDoesNotExist(Exception):
|
|
pass
|
|
|
|
class InvalidDataType(Exception):
|
|
pass
|
|
|
|
class NoAckRecv(Exception):
|
|
pass
|
|
|
|
class StoppedConnecting(Exception):
|
|
pass
|
|
|
|
Filter_Config = {
|
|
"Sinc5Sinc1With50hz60HzRejection": [
|
|
"sinc5sinc1postfilter",
|
|
[
|
|
"F27SPS",
|
|
"F21SPS",
|
|
"F20SPS",
|
|
"F16SPS",
|
|
]
|
|
],
|
|
"Sinc5Sinc1": [
|
|
"sinc5sinc1odr",
|
|
[
|
|
"F31250_0SPS",
|
|
"F15625_0SPS",
|
|
"F10417_0SPS",
|
|
"F5208_0SPS" ,
|
|
"F2597_0SPS" ,
|
|
"F1007_0SPS" ,
|
|
"F503_8SPS" ,
|
|
"F381_0SPS" ,
|
|
"F200_3SPS" ,
|
|
"F100_2SPS" ,
|
|
"F59_52SPS" ,
|
|
"F49_68SPS" ,
|
|
"F20_01SPS" ,
|
|
"F16_63SPS" ,
|
|
"F10_0SPS" ,
|
|
"F5_0SPS" ,
|
|
"F2_5SPS" ,
|
|
"F1_25SPS" ,
|
|
]
|
|
],
|
|
"Sinc3": [
|
|
"sinc3odr",
|
|
[
|
|
"F31250_0SPS",
|
|
"F15625_0SPS",
|
|
"F10417_0SPS",
|
|
"F5208_0SPS" ,
|
|
"F2597_0SPS" ,
|
|
"F1007_0SPS" ,
|
|
"F503_8SPS" ,
|
|
"F381_0SPS" ,
|
|
"F200_3SPS" ,
|
|
"F100_2SPS" ,
|
|
"F59_52SPS" ,
|
|
"F49_68SPS" ,
|
|
"F20_01SPS" ,
|
|
"F16_63SPS" ,
|
|
"F10_0SPS" ,
|
|
"F5_0SPS" ,
|
|
"F2_5SPS" ,
|
|
"F1_25SPS" ,
|
|
]
|
|
],
|
|
"Sinc3WithFineODR": [
|
|
"sinc3fineodr",
|
|
DATA_F32
|
|
],
|
|
}
|
|
|
|
class Device:
|
|
def __init__(self, send_cmd_handler, send_raw_cmd_handler, read_response, cmd_lock):
|
|
self._send_cmd = send_cmd_handler
|
|
self._send_raw_cmd = send_raw_cmd_handler
|
|
self._read_response = read_response
|
|
self._cmd_lock = cmd_lock
|
|
|
|
async def set_ip_settings(self, addr=[192, 168, 1, 128], port=1337, prefix_len=24, gateway=[192, 168, 1, 1]):
|
|
"""
|
|
After calling this fn, the ip settings are immediately saved into flash and will be effective on next reboot.
|
|
"""
|
|
if not(isinstance(addr, list) and isinstance(gateway, list)):
|
|
raise InvalidDataType
|
|
|
|
if not(len(addr) == 4 and len(gateway) == 4):
|
|
raise InvalidDataType
|
|
|
|
if not(isinstance(port, int) and isinstance(prefix_len, int)):
|
|
raise InvalidDataType
|
|
|
|
return await self._send_raw_cmd(
|
|
{
|
|
"device_cmd": "SetIPSettings",
|
|
"ip_settings": {
|
|
"addr": addr,
|
|
"port": port,
|
|
"prefix_len": prefix_len,
|
|
"gateway": gateway,
|
|
}
|
|
}
|
|
)
|
|
|
|
async def set_active_report_mode(self, on):
|
|
"""
|
|
Set active report to be on. If it is on, Kirdy will send status report
|
|
to ALL client socket connections according to the temperature polling rate set.
|
|
"""
|
|
return await self._send_cmd(TARGET_DEVICE, "SetActiveReportMode", on)
|
|
|
|
async def get_status_report(self):
|
|
"""
|
|
Get status of all peripherals in a json object
|
|
|
|
Example of yielded data::
|
|
{
|
|
'ts': 227657, # Relative Timestamp (ms)
|
|
'laser': {
|
|
'pwr_on': False, # Laser Power is On (True/False)
|
|
'pwr_excursion': False, # Was Laser experienced an Overpowered Condition? (True/False)
|
|
'ld_i_set': 0.0, # Laser Diode Output Current (A)
|
|
'pd_i': 2.0000002e-06, # Internal Photodiode Monitor current (A)
|
|
'pd_pwr': None, # Power Readings from Internal Photodiode (W). Return None if pd_mon parameter(s) are not defined.
|
|
'term_status': 'Is50Ohm' # Is the Low Frequency Modulation Input's Impedance 50 Ohm? (Is50Ohm/Not50Ohm)
|
|
},
|
|
'thermostat': {
|
|
'pwr_on': False, # Tec Power is On (True/False)
|
|
'pid_engaged': False, # Is Pid_Engaged. If False, it is in Constant Current Mode (True/False)
|
|
'temp_mon_status': { # Temperature Monitor:
|
|
'status': 'Off', # (To be revised)
|
|
'over_temp_alarm': False # Was Laser Diode experienced an Overtemperature condition (True/False)
|
|
},
|
|
'temperature': 25.03344, # Temperature Readings (Degree Celsius)
|
|
'i_set': 0.0, # Tec Current Set by User/PID Controller(A)
|
|
'tec_i': 0.0024998188, # Tec Current Readings (A)
|
|
'tec_v': -0.00399971 # Tec Voltage Readings (V)
|
|
}
|
|
}
|
|
"""
|
|
retry = 0
|
|
while(retry < 3):
|
|
response = await self._send_cmd(TARGET_DEVICE, "GetStatusReport")
|
|
if response["msg_type"] != "Acknowledge":
|
|
return response
|
|
status_report = await self._read_response()
|
|
if "ts" in status_report:
|
|
return status_report
|
|
else:
|
|
retry += 1
|
|
return None
|
|
|
|
async def get_settings_summary(self):
|
|
"""
|
|
Get the current settings of laser and thermostat in a json object
|
|
|
|
{
|
|
'laser': {
|
|
default_pwr_on': False, # Power On Laser Diode at Startup
|
|
ld_drive_current': { # Laser Diode Output Current(A)
|
|
'value': 0.0, # Value Set
|
|
'max': 0.3 # Max Value Settable
|
|
,
|
|
ld_drive_current_limit': { # Laser Diode Software Current Limit(A)
|
|
'value': 0.3, # Value Set
|
|
'max': 0.3 # Max Value Settable
|
|
,
|
|
pd_responsitivity': { # Laser Diode Software Current Limit(A)
|
|
'responsitivity': None, # Value Set
|
|
'i_dark': 0.0 # Max Value Settable
|
|
,
|
|
ld_pwr_limit': 0.0 # Laser Diode Power Limit(W)
|
|
},
|
|
'thermostat': {
|
|
'default_pwr_on': True, # Power on Thermostat at Startup
|
|
'pid_engaged': True, # True: PID Control Mode | False Constant Current Mode
|
|
'temperature_setpoint': 25.0, # Temperature Setpoint (Degree Celsius)
|
|
'tec_settings': {
|
|
'i_set': { # Current TEC Current Set by PID Controller/User
|
|
'value': 0.04330516, # Value Set
|
|
'max': 1.0 # Max Value Settable
|
|
},
|
|
'max_v': { # Max Voltage Across Tec Terminals
|
|
'value': 4.990857, # Value Set
|
|
'max': 5.0 # Max Value Settable
|
|
},
|
|
'max_i_pos': { # Max Cooling Current Across Tec Terminals
|
|
'value': 0.99628574, # Value Set
|
|
'max': 1.0 # Max Value Settable
|
|
},
|
|
'max_i_neg': { # Max Heating Current Across Tec Terminals
|
|
'value': 0.99628574, # Value Set
|
|
'max': 1.0 # Max Value Settable
|
|
}
|
|
},
|
|
'pid_params': { # PID Controller Parameters
|
|
'kp': 0.15668282, # Proportional Gain
|
|
'ki': 0.0021359625, # Integral Gain
|
|
'kd': 0.8292545, # Derivative Gain
|
|
'output_min': -1.0, # Minimum Current Output (A)
|
|
'output_max': 1.0 # Maximum Current Output (A)
|
|
},
|
|
'temp_adc_settings': { # Temperature ADC Settings (Please read AD7172-2 Documentation)
|
|
'filter_type': 'Sinc5Sinc1With50hz60HzRejection', # Filter Types
|
|
'sinc5sinc1odr': None, # (Unused)
|
|
'sinc3odr': None, # (Unused)
|
|
'sinc5sinc1postfilter': None, # (Unused)
|
|
'sinc3fineodr': None, # (Unused)
|
|
'rate': 16.67 # ADC Sampling Rate (Hz)
|
|
},
|
|
'temp_mon_settings': { # Temperature Monitor Settings
|
|
'upper_limit': 40.0, # Temperature Upper Limit (Degree Celsius)
|
|
'lower_limit': 10.0 # Temperature Lower Limit (Degree Celsius)
|
|
},
|
|
'thermistor_params': { # Thermistor Steinhart-Hart equation parameters
|
|
't0': 25.0, # t0: Degree Celsius
|
|
'r0': 10000.0, # r0: Ohm
|
|
'b': 3900.0 # b: (unitless)
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
|
|
response = await self._send_cmd(TARGET_DEVICE, "GetSettingsSummary")
|
|
if response["msg_type"] != "Acknowledge":
|
|
return response
|
|
return await self._read_response()
|
|
|
|
async def dfu(self):
|
|
"""
|
|
Issuing this cmd will HARD RESET the device and
|
|
put Kirdy into Dfu mode for flashing firmware.
|
|
"""
|
|
return await self._send_cmd(TARGET_DEVICE, "Dfu")
|
|
|
|
async def save_current_settings_to_flash(self):
|
|
"""
|
|
Save the current laser diode and thermostat configurations into flash.
|
|
"""
|
|
return await self._send_cmd(TARGET_DEVICE, "SaveFlashSettings")
|
|
|
|
async def load_current_settings_to_flash(self):
|
|
"""
|
|
Restore the laser diode and thermostat settings from flash
|
|
"""
|
|
return await self._send_cmd(TARGET_DEVICE, "LoadFlashSettings")
|
|
|
|
async def hard_reset(self):
|
|
"""
|
|
Hard Reset Kirdy. The socket connection will be closed by Kirdy.
|
|
Laser diode power and Tec power will be turned off.
|
|
"""
|
|
return await self._send_cmd(TARGET_DEVICE, "HardReset")
|
|
|
|
class Laser:
|
|
def __init__(self, send_cmd_handler, cmd_lock):
|
|
self._send_cmd = send_cmd_handler
|
|
self._cmd_lock = cmd_lock
|
|
|
|
async def set_power_on(self, on):
|
|
"""
|
|
Power Up or Power Down laser diode. Powering up the Laser Diode resets the pwr_excursion status
|
|
- on (True/False)
|
|
"""
|
|
if on:
|
|
return await self._send_cmd(TARGET_LD, "PowerUp", None)
|
|
else:
|
|
return await self._send_cmd(TARGET_LD, "PowerDown", None)
|
|
|
|
async def set_default_pwr_on(self, on):
|
|
"""
|
|
Set whether laser diode is powered up at Startup
|
|
- on (True/False)
|
|
"""
|
|
return await self._send_cmd(TARGET_LD, "SetDefaultPowerOn", on)
|
|
|
|
async def set_ld_terms_short(self, short):
|
|
"""
|
|
Open/Short laser diode terminals.
|
|
- on (True/False)
|
|
"""
|
|
if short:
|
|
return await self._send_cmd(TARGET_LD, "LdTermsShort", None)
|
|
else:
|
|
return await self._send_cmd(TARGET_LD, "LdTermsOpen", None)
|
|
|
|
async def set_i(self, i):
|
|
"""
|
|
Set laser diode output current: Max(0, Min(i_set, i_soft_limit))
|
|
- i: A
|
|
"""
|
|
return await self._send_cmd(TARGET_LD, "SetI", i)
|
|
|
|
async def set_i_soft_limit(self, i_limit):
|
|
"""
|
|
Set laser diode software output current limit
|
|
- i_limit: A
|
|
"""
|
|
return await self._send_cmd(TARGET_LD, "SetISoftLimit", i_limit)
|
|
|
|
async def set_pd_mon_responsitivity(self, responsitivity):
|
|
"""
|
|
Configure the photodiode monitor responsitivity parameter
|
|
- responsitivity: A/W
|
|
"""
|
|
return await self._send_cmd(TARGET_LD, "SetPdResponsitivity", responsitivity)
|
|
|
|
async def set_pd_mon_dark_current(self, dark_current):
|
|
"""
|
|
Configure the photodiode monitor responsitivity parameter
|
|
- dark_current: A/W
|
|
"""
|
|
return await self._send_cmd(TARGET_LD, "SetPdDarkCurrent", dark_current)
|
|
|
|
async def set_ld_pwr_limit(self, pwr_limit):
|
|
"""
|
|
Set power limit for the power excursion monitor
|
|
If the calculated power with the params of pd_mon > pwr_limit,
|
|
overpower protection is triggered.
|
|
- pwr_limit: W
|
|
"""
|
|
return await self._send_cmd(TARGET_LD, "SetLdPwrLimit", pwr_limit)
|
|
|
|
async def clear_alarm(self):
|
|
"""
|
|
Clear the power excursion monitor alarm
|
|
"""
|
|
return await self._send_cmd(TARGET_LD, "ClearAlarm")
|
|
|
|
class Thermostat:
|
|
def __init__(self, send_cmd_handler, send_raw_cmd_handler, cmd_lock):
|
|
self._send_cmd = send_cmd_handler
|
|
self._send_raw_cmd = send_raw_cmd_handler
|
|
self._cmd_lock = cmd_lock
|
|
async def set_power_on(self, on):
|
|
"""
|
|
Power up or power down thermostat
|
|
- Powering up the thermostat resets the pwr_excursion status
|
|
"""
|
|
if on:
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "PowerUp", None)
|
|
else:
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "PowerDown", None)
|
|
|
|
async def set_default_pwr_on(self, on):
|
|
"""
|
|
Set whether thermostat is powered up at Startup
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetDefaultPowerOn", on)
|
|
|
|
async def set_tec_max_v(self, max_v):
|
|
"""
|
|
Set Tec Maximum Voltage Across the TEC Terminals
|
|
- max_v: V
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetTecMaxV", max_v)
|
|
|
|
async def set_tec_max_cooling_i(self, max_i_pos):
|
|
"""
|
|
Set Tec maximum cooling current (Settable Range: 0.0 - 1.0)
|
|
- max_i_pos: A
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetTecMaxIPos", max_i_pos)
|
|
|
|
async def set_tec_max_heating_i(self, max_i_neg):
|
|
"""
|
|
Set Tec maximum heating current (Settable Range: 0.0 - 1.0)
|
|
- max_i_neg: A
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetTecMaxINeg", max_i_neg)
|
|
|
|
async def set_tec_i_out(self, i_out):
|
|
"""
|
|
Set Tec Output Current
|
|
This cmd is only effective in constant current control mode
|
|
or your newly set value will be overwritten by PID Controller Output
|
|
- i_out: A
|
|
"""
|
|
if isinstance(i_out, float):
|
|
return await self._send_raw_cmd({"tec_set_i": i_out})
|
|
elif isinstance(i_out, int):
|
|
return await self._send_raw_cmd({"tec_set_i": float(i_out)})
|
|
else:
|
|
raise InvalidDataType
|
|
|
|
async def set_constant_current_control_mode(self):
|
|
"""
|
|
Disable PID Controller and output current can be controlled with set_tec_i_out() cmd.
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetPidDisEngage", None)
|
|
|
|
async def set_temperature_setpoint(self, temperature):
|
|
"""
|
|
Set Temperature Setpoint for PID Controller. This parameter is not active in constant current control mode
|
|
- temperature: Degree Celsius
|
|
"""
|
|
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetTemperatureSetpoint", temperature)
|
|
|
|
async def set_pid_control_mode(self):
|
|
"""
|
|
Enable PID Controller. Its PID Update Interval is controlled by the Temperature ADC polling rate.
|
|
Please refer to config_temp_adc_filter for the possible polling rate options
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetPidEngage", None)
|
|
|
|
async def set_pid_kp(self, kp):
|
|
"""
|
|
Set Kp parameter for PID Controller
|
|
kp: (unitless)
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetPidKp", kp)
|
|
|
|
async def set_pid_ki(self, ki):
|
|
"""
|
|
Set Ki parameter for PID Controller
|
|
ki: (unitless)
|
|
"""
|
|
await self._send_cmd(TARGET_THERMOSTAT, "SetPidKi", ki)
|
|
|
|
async def set_pid_kd(self, kd):
|
|
"""
|
|
Set Kd parameter for PID Controller
|
|
kd: (unitless)
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetPidKd", kd)
|
|
|
|
async def set_pid_output_max(self, out_max):
|
|
"""
|
|
Set max output limit at the PID Output
|
|
- out_max: A
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetPidOutMax", out_max)
|
|
|
|
async def set_pid_output_min(self, out_min):
|
|
"""
|
|
Set min output limit at the PID Output
|
|
- out_min: A
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetPidOutMin", out_min)
|
|
|
|
async def set_temp_mon_upper_limit(self, upper_limit):
|
|
"""
|
|
Set Temperature Monitor Upper Limit Threshold. Exceeding the limit for too long
|
|
will force the TEC Controller, PID Controller and Laser Diode Power to Shutdown
|
|
- upper_limit: Degree Celsius
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetTempMonUpperLimit", upper_limit)
|
|
|
|
async def set_temp_mon_lower_limit(self, lower_limit):
|
|
"""
|
|
Set Temperature Monitor Lower Limit Threshold. Exceeding the limit for too long
|
|
will force the TEC Controller, PID Controller and Laser Diode Power to Shutdown
|
|
- lower_limit: Degree Celsius
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetTempMonLowerLimit", lower_limit)
|
|
|
|
async def clear_alarm(self):
|
|
"""
|
|
Clear the temperature monitor alarm
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "ClearAlarm")
|
|
|
|
async def set_sh_t0(self, t0):
|
|
"""
|
|
Set t0 Steinhart-Hart parameter for the laser diode NTC
|
|
- t0: Degree Celsius
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetShT0", t0)
|
|
|
|
async def set_sh_r0(self, r0):
|
|
"""
|
|
Set r0 Steinhart-Hart parameter for the laser diode NTC
|
|
- r0: Ohm
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetShR0", r0)
|
|
|
|
async def set_sh_beta(self, beta):
|
|
"""
|
|
Set beta Steinhart-Hart parameter for the laser diode NTC
|
|
- beta: (unitless)
|
|
"""
|
|
return await self._send_cmd(TARGET_THERMOSTAT, "SetShBeta", beta)
|
|
|
|
async def config_temp_adc_filter(self, filter_type, sampling_rate):
|
|
"""
|
|
Configure the temperature adc filter type and sampling rate.
|
|
Please refer to AD7172 datasheet for the usage of various types of filter.
|
|
The actual temperature polling rate is bottlenecked by the processing speed of the MCU and
|
|
performs differently under different kinds of workload. Please verify the polling rate with
|
|
the timestamp.
|
|
"""
|
|
if not(filter_type in Filter_Config.keys()):
|
|
raise InvalidDataType
|
|
if Filter_Config[filter_type][1] != DATA_F32:
|
|
if not(sampling_rate in Filter_Config[filter_type][1]):
|
|
raise InvalidDataType
|
|
else:
|
|
if not(isinstance(sampling_rate, float)):
|
|
raise InvalidDataType
|
|
|
|
cmd = {}
|
|
cmd["thermostat_cmd"] = "ConfigTempAdcFilter"
|
|
cmd["temp_adc_filter"] = {
|
|
"filter_type": filter_type,
|
|
Filter_Config[filter_type][0]: sampling_rate,
|
|
}
|
|
|
|
return await self._send_raw_cmd(cmd)
|
|
|
|
class Kirdy:
|
|
def __init__(self):
|
|
self._reader = None
|
|
self._writer = None
|
|
self._connecting_task = None
|
|
self._cmd_lock = asyncio.Lock()
|
|
self._report_mode_on = False
|
|
self.timeout = None
|
|
self.device = Device(self._send_cmd_handler, self._send_raw_cmd_handler, self._read_response, self._cmd_lock)
|
|
self.laser = Laser(self._send_cmd_handler, self._cmd_lock)
|
|
self.thermostat = Thermostat(self._send_cmd_handler, self._send_raw_cmd_handler, self._cmd_lock)
|
|
|
|
self._cmd_list = {
|
|
TARGET_DEVICE: {
|
|
"SetIPSettings": IP_SETTINGS,
|
|
"SetActiveReportMode": DATA_BOOL,
|
|
"GetStatusReport": None,
|
|
"GetSettingsSummary": None,
|
|
"Dfu": None,
|
|
"SaveFlashSettings": None,
|
|
"LoadFlashSettings": None,
|
|
"HardReset": None,
|
|
},
|
|
TARGET_LD: {
|
|
# LD Drive Related
|
|
"SetDefaultPowerOn": DATA_BOOL,
|
|
"PowerUp": None,
|
|
"PowerDown": None,
|
|
"LdTermsShort": None,
|
|
"LdTermsOpen": None,
|
|
"SetI": DATA_F32,
|
|
"SetISoftLimit": DATA_F32,
|
|
# PD Mon Related
|
|
"SetPdResponsitivity": DATA_F32,
|
|
"SetPdDarkCurrent": DATA_F32,
|
|
"SetLdPwrLimit": DATA_F32,
|
|
"ClearAlarm": None,
|
|
},
|
|
TARGET_THERMOSTAT: {
|
|
"SetDefaultPowerOn": DATA_BOOL,
|
|
"PowerUp": DATA_F32,
|
|
"PowerDown": DATA_F32,
|
|
# TEC Controller Settings
|
|
"SetTecMaxV": DATA_F32,
|
|
"SetTecMaxIPos": DATA_F32,
|
|
"SetTecMaxINeg": DATA_F32,
|
|
"SetTecIOut": DATA_F32,
|
|
"SetTemperatureSetpoint": DATA_F32,
|
|
# PID Controller Settings
|
|
"SetPidEngage": None,
|
|
"SetPidDisEngage": None,
|
|
"SetPidKp": DATA_F32,
|
|
"SetPidKi": DATA_F32,
|
|
"SetPidKd": DATA_F32,
|
|
"SetPidOutMin": DATA_F32,
|
|
"SetPidOutMax": DATA_F32,
|
|
# Temperature Adc Internal Filters Settings
|
|
"ConfigTempAdcFilter": TEMP_ADC_FILTER,
|
|
# Temperature Monitor Settings
|
|
"SetTempMonUpperLimit": DATA_F32,
|
|
"SetTempMonLowerLimit": DATA_F32,
|
|
"ClearAlarm": None,
|
|
# Thermistor Parameter Settings
|
|
"SetShT0": DATA_F32,
|
|
"SetShR0": DATA_F32,
|
|
"SetShBeta": DATA_F32,
|
|
}
|
|
}
|
|
|
|
async def start_session(self, host='192.168.1.128', port=1337, timeout=None):
|
|
self._connecting_task = asyncio.create_task(
|
|
asyncio.wait_for(asyncio.open_connection(host, port), timeout)
|
|
)
|
|
self.timeout = timeout
|
|
try:
|
|
self._reader, self._writer = await self._connecting_task
|
|
writer_sock = self._writer.get_extra_info("socket")
|
|
writer_sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
|
|
|
|
except asyncio.CancelledError:
|
|
raise StoppedConnecting
|
|
finally:
|
|
self._connecting_task = None
|
|
|
|
def connecting(self):
|
|
"""Returns True if client is connecting"""
|
|
return self._connecting_task is not None
|
|
|
|
def connected(self):
|
|
"""Returns True if client is connected"""
|
|
return self._writer is not None
|
|
|
|
async def _cmd(self, *cmd):
|
|
async with self._cmd_lock:
|
|
# protect the read-write process from being cancelled midway
|
|
line = await asyncio.shield(self._read_write(cmd))
|
|
|
|
async def end_session(self):
|
|
"""End session to Kirdy if connected, cancel connection if connecting"""
|
|
if self._connecting_task is not None:
|
|
self._connecting_task.cancel()
|
|
|
|
if self._writer is None:
|
|
return
|
|
|
|
# Reader needn't be closed
|
|
self._writer.close()
|
|
await self._writer.wait_closed()
|
|
self._reader = None
|
|
self._writer = None
|
|
|
|
async def _read_response(self, buffer_size=16384):
|
|
"""
|
|
Decode newline delimited Json objects and return the latest json received inside the buffer.
|
|
- buffer_size: Integer
|
|
"""
|
|
|
|
try:
|
|
raw_response = await asyncio.wait_for(self._reader.read(buffer_size), self.timeout)
|
|
except asyncio.exceptions.TimeoutError:
|
|
return {
|
|
"msg_type": "Internal Timeout"
|
|
}
|
|
|
|
response = raw_response.decode('utf-8', errors='ignore').split("\n")
|
|
for item in reversed(response):
|
|
try:
|
|
return json.loads(item)
|
|
except json.decoder.JSONDecodeError:
|
|
pass
|
|
return { "msg_type": "Internal No json object found in response" }
|
|
|
|
async def _send_raw_cmd_handler(self, cmd, lock=True):
|
|
if lock:
|
|
async with self._cmd_lock:
|
|
return await asyncio.shield(self._send_raw_cmd(cmd))
|
|
else:
|
|
return await asyncio.shield(self._send_raw_cmd(cmd))
|
|
|
|
# If the cmd involves a cmd specific data type,
|
|
# checking is done separately within the functions being called
|
|
async def _send_raw_cmd(self, cmd):
|
|
retry = 0
|
|
while retry < 10:
|
|
self._writer.write(bytes(json.dumps(cmd), "UTF-8"))
|
|
await self._writer.drain()
|
|
response = await self._read_response()
|
|
try:
|
|
if response["msg_type"] == "Acknowledge":
|
|
return response
|
|
except:
|
|
retry += 1
|
|
await asyncio.sleep(0.1)
|
|
raise NoAckRecv
|
|
|
|
async def _send_cmd_handler(self, target, cmd, data=None, lock=True):
|
|
if lock:
|
|
async with self._cmd_lock:
|
|
return await asyncio.shield(self._send_cmd(target, cmd, data))
|
|
else:
|
|
return await asyncio.shield(self._send_cmd(target, cmd, data))
|
|
|
|
async def _send_cmd(self, target, cmd, data):
|
|
cmd_dict = {}
|
|
|
|
if not(target in self._cmd_list.keys()) or not(cmd in self._cmd_list[target].keys()):
|
|
raise CmdDoesNotExist
|
|
cmd_dict[target] = cmd
|
|
|
|
if self._cmd_list[target][cmd] == DATA_F32:
|
|
if isinstance(data, float):
|
|
cmd_dict[DATA_F32] = data
|
|
elif isinstance(data, int):
|
|
cmd_dict[DATA_F32] = float(data)
|
|
elif self._cmd_list[target][cmd] == DATA_BOOL:
|
|
if isinstance(data, bool):
|
|
cmd_dict[DATA_BOOL] = data
|
|
else:
|
|
raise InvalidDataType
|
|
elif self._cmd_list[target][cmd] == None:
|
|
pass
|
|
else:
|
|
# Undefined Data Type
|
|
raise CmdDoesNotExist
|
|
|
|
retry = 0
|
|
while retry < 10:
|
|
self._writer.write(bytes(json.dumps(cmd_dict), "UTF-8"))
|
|
await self._writer.drain()
|
|
response = await self._read_response()
|
|
try:
|
|
if response["msg_type"] == "Acknowledge":
|
|
return response
|
|
except:
|
|
retry += 1
|
|
await asyncio.sleep(0.1)
|
|
raise NoAckRecv
|
|
|
|
async def report_mode(self, report_interval = 0.0, buffer_size = 16384):
|
|
"""
|
|
Start reporting device status in json object. Optional report_interval can be added to discard unwanted samples.
|
|
Only the latest status report received within the buffer is returned.
|
|
- polling interval: float/int (unit: seconds)
|
|
- buffer_size: int
|
|
"""
|
|
await self.device.set_active_report_mode(True)
|
|
self._report_mode_on = True
|
|
|
|
while self._report_mode_on:
|
|
await asyncio.sleep(report_interval)
|
|
async with self._cmd_lock:
|
|
yield await self._read_response(buffer_size)
|
|
|
|
await self.device.set_active_report_mode(False)
|
|
|
|
def stop_report_mode(self):
|
|
self._report_mode_on = False
|