120 lines
4.1 KiB
Python
120 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
@file rad_sensor.py
|
|
@brief Tests for the radiation sensor handler
|
|
@author J. Meier
|
|
@date 01.07.2021
|
|
"""
|
|
import enum
|
|
import struct
|
|
|
|
from eive_tmtc.config.definitions import CustomServiceList
|
|
from eive_tmtc.pus_tm.defs import PrintWrapper
|
|
|
|
from spacepackets.ecss.tc import PusTelecommand
|
|
from tmtccmd.config import TmtcDefinitionWrapper, OpCodeEntry
|
|
from tmtccmd.config.tmtc import tmtc_definitions_provider
|
|
from tmtccmd.tc import DefaultPusQueueHelper
|
|
from tmtccmd.tc.pus_200_fsfw_mode import Mode, pack_mode_data
|
|
from tmtccmd.tc.pus_3_fsfw_hk import generate_one_hk_command, make_sid
|
|
from tmtccmd.util import ObjectIdU32
|
|
from tmtccmd.util.tmtc_printer import FsfwTmTcPrinter
|
|
|
|
|
|
class SetId(enum.IntEnum):
|
|
HK = 3
|
|
|
|
|
|
class OpCode:
|
|
ON = ["0", "on"]
|
|
NORMAL = ["1", "normal"]
|
|
OFF = ["2", "off"]
|
|
REQ_HK_ONCE = ["3", "hk-os"]
|
|
DEBUG_ON = ["10", "dbg-on"]
|
|
DEBUG_OFF = ["11", "dbg-off"]
|
|
|
|
|
|
class Info:
|
|
ON = "Switch Rad Sensor on"
|
|
NORMAL = "Switch Rad Sensor normal"
|
|
OFF = "Switch Rad sensor off"
|
|
REQ_OS_HK = "Request one-shot HK"
|
|
DEBUG_ON = "Switch debug output on"
|
|
DEBUG_OFF = "Switch debug output off"
|
|
|
|
|
|
class CommandId:
|
|
START_CONVERSIONS = 2
|
|
READ_CONVERSIONS = 3
|
|
ENABLE_DEBUG_OUTPUT = 4
|
|
DISABLE_DEBUG_OUTPUT = 5
|
|
|
|
|
|
@tmtc_definitions_provider
|
|
def add_rad_sens_cmds(defs: TmtcDefinitionWrapper):
|
|
oce = OpCodeEntry()
|
|
oce.add(info=Info.ON, keys=OpCode.ON)
|
|
oce.add(info=Info.OFF, keys=OpCode.OFF)
|
|
oce.add(info=Info.NORMAL, keys=OpCode.NORMAL)
|
|
oce.add(info=Info.REQ_OS_HK, keys=OpCode.REQ_HK_ONCE)
|
|
oce.add(info=Info.DEBUG_ON, keys=OpCode.DEBUG_ON)
|
|
oce.add(info=Info.DEBUG_OFF, keys=OpCode.DEBUG_OFF)
|
|
defs.add_service(
|
|
name=CustomServiceList.RAD_SENSOR.value,
|
|
info="Radiation Sensor",
|
|
op_code_entry=oce,
|
|
)
|
|
|
|
|
|
def pack_rad_sensor_test_into(
|
|
object_id: ObjectIdU32, q: DefaultPusQueueHelper, op_code: str
|
|
):
|
|
q.add_log_cmd(f"Commanding Radiation sensor handler {object_id}")
|
|
|
|
if op_code in OpCode.ON:
|
|
rad_sensor_mode_cmd(object_id, Mode.ON, Info.ON, q)
|
|
if op_code in OpCode.NORMAL:
|
|
rad_sensor_mode_cmd(object_id, Mode.NORMAL, Info.NORMAL, q)
|
|
if op_code in OpCode.OFF:
|
|
rad_sensor_mode_cmd(object_id, Mode.OFF, Info.OFF, q)
|
|
if op_code in OpCode.REQ_HK_ONCE:
|
|
q.add_log_cmd(f"Rad sensor: {Info.REQ_OS_HK}")
|
|
q.add_pus_tc(
|
|
generate_one_hk_command(sid=make_sid(object_id.as_bytes, set_id=SetId.HK))
|
|
)
|
|
if op_code in OpCode.DEBUG_ON:
|
|
q.add_log_cmd(f"Rad sensor: {Info.DEBUG_ON}")
|
|
command = object_id.as_bytes + struct.pack("!I", CommandId.ENABLE_DEBUG_OUTPUT)
|
|
q.add_pus_tc(PusTelecommand(service=8, subservice=128, app_data=command))
|
|
if op_code in OpCode.DEBUG_OFF:
|
|
q.add_log_cmd(f"Rad sensor: {Info.DEBUG_OFF}")
|
|
command = object_id.as_bytes + struct.pack("!I", CommandId.DISABLE_DEBUG_OUTPUT)
|
|
q.add_pus_tc(PusTelecommand(service=8, subservice=128, app_data=command))
|
|
|
|
|
|
def rad_sensor_mode_cmd(
|
|
object_id: ObjectIdU32, mode: Mode, info: str, q: DefaultPusQueueHelper
|
|
):
|
|
q.add_log_cmd(f"Rad sensor: {info}")
|
|
mode_data = pack_mode_data(object_id.as_bytes, mode, 0)
|
|
q.add_pus_tc(PusTelecommand(service=200, subservice=1, app_data=mode_data))
|
|
|
|
|
|
def handle_rad_sensor_data(printer: FsfwTmTcPrinter, set_id: int, hk_data: bytes):
|
|
if set_id == SetId.HK:
|
|
pw = PrintWrapper(printer)
|
|
current_idx = 0
|
|
pw.dlog("Received Radiation Sensor HK data")
|
|
fmt_str = "!fHHHHHH"
|
|
inc_len = struct.calcsize(fmt_str)
|
|
(temp, ain0, ain1, ain4, ain5, ain6, ain7) = struct.unpack(
|
|
fmt_str, hk_data[current_idx : current_idx + inc_len]
|
|
)
|
|
ain_dict = {0: ain0, 1: ain1, 4: ain4, 5: ain5, 6: ain6, 7: ain7}
|
|
pw.dlog(f"Temperature: {temp} C")
|
|
pw.dlog(f"AIN Channel | Raw Value (hex) | Raw Value (dec)")
|
|
for idx, val in ain_dict.items():
|
|
pw.dlog(f"{idx} | {val:#06x} | {str(val).ljust(5)}")
|
|
current_idx += inc_len
|
|
printer.print_validity_buffer(validity_buffer=hk_data[current_idx:], num_vars=7)
|