from typing import Union from tmtccmd.tc import DefaultPusQueueHelper from tmtccmd.tc.pus_200_fsfw_modes import Modes from tmtccmd.util import ObjectIdU32, ObjectIdBase from tmtc.common import pack_mode_cmd_with_info import config.object_ids as obj_ids from pus_tc.prompt_parameters import prompt_parameters_cli, prompt_parameters_gui class OpCodes: THERMAL_CONTROLLER = [obj_ids.THERMAL_CONTROLLER_ID.hex(), "ctrl-th"] CORE_CONTROLLER = [obj_ids.CORE_CONTROLLER_ID.hex(), "ctrl-core"] class Info: THERMAL_CONTROLLER = "Thermal controller" CORE_CONTROLLER = "ACS controller" def pack_cmd_ctrl_to_prompted_mode( q: DefaultPusQueueHelper, object_id: ObjectIdU32, gui: bool ): param_list = [ {"name": "Mode", "defaultValue": "2"}, {"name": "Submode", "defaultValue": "0"}, ] if gui: parameters = prompt_parameters_gui(param_list) else: parameters = prompt_parameters_cli(param_list) mode = int(parameters["Mode"]) if mode < 0 or mode > 2: print("Invalid Mode, defaulting to OFF") mode = 0 submode = int(parameters["Submode"]) pack_mode_cmd_with_info( object_id=object_id.as_bytes, mode=mode, submode=submode, q=q, info=f"Commanding {object_id} to {mode}, {submode}", ) def pack_cmd_ctrl_to_off( q: DefaultPusQueueHelper, object_id: Union[ObjectIdBase, ObjectIdU32] ): pack_mode_cmd_with_info( object_id=object_id.as_bytes, mode=Modes.OFF, submode=0, q=q, info=f"Commanding {object_id} OFF", ) def pack_cmd_ctrl_to_on(q: DefaultPusQueueHelper, object_id: ObjectIdU32): pack_mode_cmd_with_info( object_id=object_id.as_bytes, mode=Modes.ON, submode=0, q=q, info=f"Commanding {object_id} ON", ) def pack_cmd_ctrl_to_nml( q: DefaultPusQueueHelper, object_id: Union[ObjectIdBase, ObjectIdU32] ): pack_mode_cmd_with_info( object_id=object_id.as_bytes, mode=Modes.NORMAL, submode=0, q=q, info=f"Commanding {object_id} NORMAL", ) def get_object_from_op_code(op_code: str): try: return bytes.fromhex(op_code) except: pass if op_code in OpCodes.THERMAL_CONTROLLER: return obj_ids.THERMAL_CONTROLLER_ID if op_code in OpCodes.CORE_CONTROLLER: return obj_ids.CORE_CONTROLLER_ID