92 lines
2.4 KiB
Python
92 lines
2.4 KiB
Python
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
|