eive-tmtc/pus_tc/devs/rad_sensor.py

101 lines
3.3 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
"""
@file rad_sensor.py
@brief Tests for the radiation sensor handler
@author J. Meier
@date 01.07.2021
"""
2022-03-31 11:36:50 +02:00
import struct
2022-05-23 18:25:25 +02:00
from config.definitions import CustomServiceList
2021-10-01 10:55:56 +02:00
from spacepackets.ecss.tc import PusTelecommand
2022-05-05 16:23:17 +02:00
from pus_tc.service_200_mode import pack_mode_data, Modes
2022-07-04 15:22:53 +02:00
from tmtccmd.tc import QueueHelper
2022-05-23 18:25:25 +02:00
from tmtccmd.tc.pus_3_fsfw_hk import generate_one_hk_command, make_sid
from tmtccmd.utility import ObjectId
2022-05-23 15:49:21 +02:00
class SetIds:
HK = 3
2022-05-23 18:25:25 +02:00
class OpCodes:
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"
2022-03-31 11:36:50 +02:00
class CommandIds:
START_CONVERSIONS = 2
READ_CONVERSIONS = 3
ENABLE_DEBUG_OUTPUT = 4
DISABLE_DEBUG_OUTPUT = 5
2022-05-23 18:25:25 +02:00
def add_rad_sens_cmds(cmd_dict: ServiceOpCodeDictT):
2022-05-23 18:25:25 +02:00
op_code_dict = dict()
add_op_code_entry(op_code_dict=op_code_dict, info=Info.ON, keys=OpCodes.ON)
add_op_code_entry(op_code_dict=op_code_dict, info=Info.OFF, keys=OpCodes.OFF)
add_op_code_entry(op_code_dict=op_code_dict, info=Info.NORMAL, keys=OpCodes.NORMAL)
add_op_code_entry(
op_code_dict=op_code_dict, info=Info.REQ_OS_HK, keys=OpCodes.REQ_HK_ONCE
)
add_op_code_entry(
op_code_dict=op_code_dict, info=Info.DEBUG_ON, keys=OpCodes.DEBUG_ON
)
add_op_code_entry(
op_code_dict=op_code_dict, info=Info.DEBUG_OFF, keys=OpCodes.DEBUG_OFF
)
add_service_op_code_entry(
srv_op_code_dict=cmd_dict,
name=CustomServiceList.RAD_SENSOR.value,
info="Radiation Sensor",
op_code_entry=op_code_dict,
)
2022-03-31 11:36:50 +02:00
2022-07-04 15:22:53 +02:00
def pack_rad_sensor_test_into(object_id: ObjectId, q: QueueHelper, op_code: str):
q.add_log_cmd(f"Commanding Radiation sensor handler {object_id}")
2022-03-31 11:36:50 +02:00
2022-05-23 18:25:25 +02:00
if op_code in OpCodes.ON:
2022-07-04 15:22:53 +02:00
rad_sensor_mode_cmd(object_id, Modes.ON, Info.ON, q)
2022-05-23 18:25:25 +02:00
if op_code in OpCodes.NORMAL:
2022-07-04 15:22:53 +02:00
rad_sensor_mode_cmd(object_id, Modes.NORMAL, Info.NORMAL, q)
2022-05-23 18:25:25 +02:00
if op_code in OpCodes.OFF:
2022-07-04 15:22:53 +02:00
rad_sensor_mode_cmd(object_id, Modes.OFF, Info.OFF, q)
2022-05-23 18:25:25 +02:00
if op_code in OpCodes.REQ_HK_ONCE:
2022-07-04 15:22:53 +02:00
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=SetIds.HK))
2022-05-23 18:25:25 +02:00
)
if op_code in OpCodes.DEBUG_ON:
2022-07-04 15:22:53 +02:00
q.add_log_cmd(f"Rad sensor: {Info.DEBUG_ON}")
2022-05-23 18:25:25 +02:00
command = object_id.as_bytes + struct.pack("!I", CommandIds.ENABLE_DEBUG_OUTPUT)
2022-07-04 15:22:53 +02:00
q.add_pus_tc(PusTelecommand(service=8, subservice=128, app_data=command))
2022-05-23 18:25:25 +02:00
if op_code in OpCodes.DEBUG_OFF:
2022-07-04 15:22:53 +02:00
q.add_log_cmd(f"Rad sensor: {Info.DEBUG_OFF}")
2022-05-23 18:25:25 +02:00
command = object_id.as_bytes + struct.pack(
"!I", CommandIds.DISABLE_DEBUG_OUTPUT
)
2022-07-04 15:22:53 +02:00
q.add_pus_tc(PusTelecommand(service=8, subservice=128, app_data=command))
2022-05-23 18:25:25 +02:00
2022-07-04 15:22:53 +02:00
def rad_sensor_mode_cmd(object_id: ObjectId, mode: Modes, info: str, q: QueueHelper):
q.add_log_cmd(f"Rad sensor: {info}")
2022-05-23 18:25:25 +02:00
mode_data = pack_mode_data(object_id.as_bytes, mode, 0)
2022-07-04 15:22:53 +02:00
q.add_pus_tc(PusTelecommand(service=200, subservice=1, app_data=mode_data))