139 lines
3.3 KiB
Python
139 lines
3.3 KiB
Python
from collections import Iterable
|
|
from decimal import *
|
|
|
|
import config as cfg
|
|
from data import LedState, BatteryStatus
|
|
|
|
# trick the pycharm type-checker into thinking Callable is in scope, not used at runtime
|
|
# noinspection PyUnreachableCode
|
|
if False:
|
|
from typing import Callable
|
|
|
|
|
|
def read_bool(register, bit):
|
|
# type: (int, int) -> Callable[[BatteryStatus], bool]
|
|
|
|
def get_value(status):
|
|
# type: (BatteryStatus) -> bool
|
|
value = status.modbus_data[register - cfg.BASE_ADDRESS]
|
|
return value & (1 << bit) > 0
|
|
|
|
return get_value
|
|
|
|
|
|
def read_float(register, scale_factor=1.0, offset=0.0, places=2):
|
|
# type: (int, float, float) -> Callable[[BatteryStatus], float]
|
|
|
|
def get_value(status):
|
|
# type: (BatteryStatus) -> float
|
|
value = status.modbus_data[register - cfg.BASE_ADDRESS]
|
|
|
|
if value >= 0x8000: # convert to signed int16
|
|
value -= 0x10000 # fiamm stores their integers signed AND with sign-offset @#%^&!
|
|
|
|
result = (value+offset)*scale_factor
|
|
return round(result,places)
|
|
|
|
return get_value
|
|
|
|
|
|
def read_hex_string(register, count):
|
|
# type: (int, int) -> Callable[[BatteryStatus], str]
|
|
"""
|
|
reads count consecutive modbus registers from start_address,
|
|
and returns a hex representation of it:
|
|
e.g. for count=4: DEAD BEEF DEAD BEEF.
|
|
"""
|
|
start = register - cfg.BASE_ADDRESS
|
|
end = start + count
|
|
|
|
def get_value(status):
|
|
# type: (BatteryStatus) -> str
|
|
return ' '.join(['{0:0>4X}'.format(x) for x in status.modbus_data[start:end]])
|
|
|
|
return get_value
|
|
|
|
|
|
def read_led_state(register, led):
|
|
# type: (int, int) -> Callable[[BatteryStatus], int]
|
|
|
|
read_lo = read_bool(register, led * 2)
|
|
read_hi = read_bool(register, led * 2 + 1)
|
|
|
|
def get_value(status):
|
|
# type: (BatteryStatus) -> int
|
|
|
|
lo = read_lo(status)
|
|
hi = read_hi(status)
|
|
|
|
if hi:
|
|
if lo:
|
|
return LedState.blinking_fast
|
|
else:
|
|
return LedState.blinking_slow
|
|
else:
|
|
if lo:
|
|
return LedState.on
|
|
else:
|
|
return LedState.off
|
|
|
|
return get_value
|
|
|
|
|
|
def read_bitmap(register):
|
|
# type: (int) -> Callable[[BatteryStatus], bitmap]
|
|
|
|
def get_value(status):
|
|
# type: (BatteryStatus) -> bitmap
|
|
value = status.modbus_data[register - cfg.BASE_ADDRESS]
|
|
return value
|
|
|
|
return get_value
|
|
|
|
def read_limb_string(register):
|
|
# type: (int) -> Callable[[BatteryStatus], bitmap]
|
|
|
|
def get_value(status):
|
|
# type: (BatteryStatus) -> bitmap
|
|
value = status.modbus_data[register - cfg.BASE_ADDRESS]
|
|
|
|
string1_disabled = int((value & 0b00001) != 0)
|
|
string2_disabled = int((value & 0b00010) != 0)
|
|
string3_disabled = int((value & 0b00100) != 0)
|
|
string4_disabled = int((value & 0b01000) != 0)
|
|
string5_disabled = int((value & 0b10000) != 0)
|
|
n_limb_strings = string1_disabled+string2_disabled+string3_disabled+string4_disabled+string5_disabled
|
|
if n_limb_strings>=2:
|
|
return True
|
|
else:
|
|
|
|
return False
|
|
return get_value
|
|
|
|
def append_unit(unit):
|
|
# type: (unicode) -> Callable[[unicode], unicode]
|
|
|
|
def get_text(v):
|
|
# type: (unicode) -> unicode
|
|
return "{0}{1}".format(str(v), unit)
|
|
|
|
return get_text
|
|
|
|
|
|
def mean(numbers):
|
|
# type: (Iterable[float] | Iterable[int]) -> float
|
|
return float("{:.2f}".format(float(sum(numbers)) / len(numbers)))
|
|
|
|
def ssum(numbers):
|
|
# type: (Iterable[float] | Iterable[int]) -> float
|
|
return float("{:.2f}".format(float(sum(numbers))))
|
|
|
|
|
|
def first(ts):
|
|
return next(t for t in ts)
|
|
|
|
def return_in_list(ts):
|
|
return ts
|
|
|
|
|