import enum import json from spacepackets.ecss import PusTelecommand from eive_tmtc.config.definitions import CustomServiceList from tmtccmd.config.tmtc import tmtc_definitions_provider from tmtccmd.tc.pus_200_fsfw_mode import Mode, pack_mode_data, Subservice from tmtccmd.tc import service_provider from tmtccmd.tc.decorator import ServiceProviderParams from tmtccmd.pus.s8_fsfw_funccmd import create_action_cmd from tmtccmd.config import OpCodeEntry, TmtcDefinitionWrapper from eive_tmtc.config.object_ids import SCEX_HANDLER_ID USE_SCEX_CONF_FILE = True class OpCode: PING = ["0", "ping"] ION_CMD = ["1", "ion"] TEMP_CMD = ["2", "temp"] EXP_STATUS_CMD = ["3", "expstatus"] ONE_CELLS_CMD = ["4", "onecell"] ALL_CELLS_CMD = ["5", "allcells"] FRAM = ["6", "fram"] SWITCH_ON = ["7", "on"] SWITCH_OFF = ["8", "off"] class ActionId(enum.IntEnum): PING = 7 ION_CMD = 4 TEMP_CMD = 3 EXP_STATUS_CMD = 2 ONE_CELLS_CMD = 6 ALL_CELLS_CMD = 5 FRAM = 1 class Info: PING = "Send Ping command" ION_CMD = "Read Ion" TEMP_CMD = "Read Temperature" EXP_STATUS_CMD = "Read Experiment Status" ONE_CELLS_CMD = "One Cell" ALL_CELLS_CMD = "All Cells" FRAM = "Read FRAM" SWITCH_ON = "Switch Scex on" SWITCH_OFF = "Switch Scex off" @tmtc_definitions_provider def add_scex_cmds(defs: TmtcDefinitionWrapper): oce = OpCodeEntry() oce.add(keys=OpCode.PING, info=Info.PING) oce.add(keys=OpCode.ION_CMD, info=Info.ION_CMD) oce.add(keys=OpCode.TEMP_CMD, info=Info.TEMP_CMD) oce.add(keys=OpCode.EXP_STATUS_CMD, info=Info.EXP_STATUS_CMD) oce.add(keys=OpCode.ONE_CELLS_CMD, info=Info.ONE_CELLS_CMD) oce.add(keys=OpCode.ALL_CELLS_CMD, info=Info.ALL_CELLS_CMD) oce.add(keys=OpCode.FRAM, info=Info.FRAM) oce.add(keys=OpCode.SWITCH_ON, info=Info.SWITCH_ON) oce.add(keys=OpCode.SWITCH_OFF, info=Info.SWITCH_OFF) defs.add_service( name=CustomServiceList.SCEX.value, info="SCEX Device", op_code_entry=oce ) @service_provider(CustomServiceList.SCEX.value) def pack_scex_cmds(p: ServiceProviderParams): # noqa C901 op_code = p.op_code q = p.queue_helper if op_code in OpCode.SWITCH_ON: q.add_log_cmd(Info.SWITCH_ON) q.add_pus_tc( PusTelecommand( service=200, subservice=Subservice.TC_MODE_COMMAND, app_data=pack_mode_data(SCEX_HANDLER_ID, Mode.ON, 0), ) ) if op_code in OpCode.SWITCH_OFF: q.add_log_cmd(Info.SWITCH_OFF) q.add_pus_tc( PusTelecommand( service=200, subservice=Subservice.TC_MODE_COMMAND, app_data=pack_mode_data(SCEX_HANDLER_ID, Mode.OFF, 0), ) ) if op_code in OpCode.PING: q.add_log_cmd(Info.PING) app_data = bytes([0]) q.add_pus_tc(create_action_cmd(SCEX_HANDLER_ID, ActionId.PING, app_data)) if op_code in OpCode.ION_CMD: q.add_log_cmd(Info.ION_CMD) app_data = bytes([0]) q.add_pus_tc(create_action_cmd(SCEX_HANDLER_ID, ActionId.ION_CMD, app_data)) if op_code in OpCode.TEMP_CMD: q.add_log_cmd(Info.TEMP_CMD) app_data = bytes([0]) q.add_pus_tc(create_action_cmd(SCEX_HANDLER_ID, ActionId.TEMP_CMD, app_data)) if op_code in OpCode.EXP_STATUS_CMD: q.add_log_cmd(Info.EXP_STATUS_CMD) app_data = bytes([0]) q.add_pus_tc( create_action_cmd(SCEX_HANDLER_ID, ActionId.EXP_STATUS_CMD, app_data) ) # one cell if op_code in OpCode.ONE_CELLS_CMD: q.add_log_cmd(Info.ONE_CELLS_CMD) app_data = bytearray([0]) # cell number cn = 0 while True: cell_select = input("Which solar cell should be measured? (1-10): ") if not cell_select.isdigit(): print("Invalid cell number. Try again.") continue cell_select = int(cell_select) if cell_select < 1 or cell_select > 10: print( f"Invalid cell number {cell_select}, " f"Please enter a valid number: " ) continue cn = cell_select - 1 break if USE_SCEX_CONF_FILE: with open("template/scex_conf.json") as json_file: json_data = json.load(json_file) first_dac = json_data["first_dac"] last_dac = json_data["last_dac"] res_switch1 = json_data["res_switch1"] res_switch2 = json_data["res_switch2"] dac_weight1 = json_data["dac_weight1"] dac_weight2 = json_data["dac_weight2"] dac_weight3 = json_data["dac_weight3"] # in app_data # app_data.extend(struct.pack("!H", first_dac)) app_data.append(cell_select) append_16_bit_val(packet=app_data, val=first_dac[cn]) append_16_bit_val(packet=app_data, val=last_dac[cn]) append_16_bit_val(packet=app_data, val=res_switch1[cn]) append_16_bit_val(packet=app_data, val=res_switch2[cn]) app_data.append(dac_weight1[cn]) app_data.append(dac_weight2[cn]) app_data.append(dac_weight3[cn]) q.add_pus_tc( create_action_cmd(SCEX_HANDLER_ID, ActionId.ONE_CELLS_CMD, app_data) ) if op_code in OpCode.ALL_CELLS_CMD: q.add_log_cmd(Info.ALL_CELLS_CMD) app_data = bytearray([0]) # cell number cn = 0 if USE_SCEX_CONF_FILE: with open("template/scex_conf.json") as json_file: json_data = json.load(json_file) first_dac = json_data["first_dac"] last_dac = json_data["last_dac"] res_switch1 = json_data["res_switch1"] res_switch2 = json_data["res_switch2"] dac_weight1 = json_data["dac_weight1"] dac_weight2 = json_data["dac_weight2"] dac_weight3 = json_data["dac_weight3"] # in app_data # app_data.extend(struct.pack("!H", first_dac)) append_16_bit_val(packet=app_data, val=first_dac[cn]) append_16_bit_val(packet=app_data, val=last_dac[cn]) append_16_bit_val(packet=app_data, val=res_switch1[cn]) append_16_bit_val(packet=app_data, val=res_switch2[cn]) app_data.append(dac_weight1[cn]) app_data.append(dac_weight2[cn]) app_data.append(dac_weight3[cn]) q.add_pus_tc( create_action_cmd(SCEX_HANDLER_ID, ActionId.ALL_CELLS_CMD, app_data) ) if op_code in OpCode.FRAM: q.add_log_cmd(Info.FRAM) app_data = bytes([0]) q.add_pus_tc(create_action_cmd(SCEX_HANDLER_ID, ActionId.FRAM, app_data)) def append_16_bit_val(packet: bytearray, val: int): packet.append((val >> 8) & 0xFF) packet.append(val & 0xFF)