eive-tmtc/eive_tmtc/tmtc/payload/scex.py

209 lines
6.8 KiB
Python
Raw Normal View History

2022-06-21 16:49:11 +02:00
import enum
2022-06-27 20:18:26 +02:00
import json
2022-06-21 16:49:11 +02:00
2022-09-27 17:35:58 +02:00
from spacepackets.ecss import PusTelecommand
2022-11-29 16:53:29 +01:00
from eive_tmtc.config.definitions import CustomServiceList
from tmtccmd.config.tmtc import tmtc_definitions_provider
2023-01-31 12:56:13 +01:00
from tmtccmd.tc.pus_200_fsfw_mode import Mode, pack_mode_data, Subservice
2022-09-27 17:35:58 +02:00
from tmtccmd.tc import service_provider
2022-08-18 14:08:05 +02:00
from tmtccmd.tc.decorator import ServiceProviderParams
from tmtccmd.pus.s8_fsfw_funccmd import create_action_cmd
from tmtccmd.config import OpCodeEntry, TmtcDefinitionWrapper
2022-11-29 16:53:29 +01:00
from eive_tmtc.config.object_ids import SCEX_HANDLER_ID
2022-06-21 16:49:11 +02:00
2022-06-27 20:18:26 +02:00
USE_SCEX_CONF_FILE = True
2023-01-16 14:13:06 +01:00
class OpCode:
2022-06-21 16:49:11 +02:00
PING = ["0", "ping"]
2022-06-27 20:18:26 +02:00
ION_CMD = ["1", "ion"]
TEMP_CMD = ["2", "temp"]
2022-06-30 21:49:25 +02:00
EXP_STATUS_CMD = ["3", "expstatus"]
2022-06-27 20:18:26 +02:00
2022-06-30 21:49:25 +02:00
ONE_CELLS_CMD = ["4", "onecell"]
ALL_CELLS_CMD = ["5", "allcells"]
2022-06-27 20:18:26 +02:00
FRAM = ["6", "fram"]
2022-06-21 16:49:11 +02:00
2022-09-27 17:35:58 +02:00
SWITCH_ON = ["7", "on"]
2022-09-28 17:01:26 +02:00
SWITCH_OFF = ["8", "off"]
2022-09-27 17:35:58 +02:00
2022-06-21 16:49:11 +02:00
2023-01-16 14:13:06 +01:00
class ActionId(enum.IntEnum):
2022-06-21 16:49:11 +02:00
PING = 7
2022-06-21 18:15:17 +02:00
ION_CMD = 4
2022-06-27 20:18:26 +02:00
TEMP_CMD = 3
EXP_STATUS_CMD = 2
ONE_CELLS_CMD = 6
ALL_CELLS_CMD = 5
FRAM = 1
2022-06-21 16:49:11 +02:00
class Info:
PING = "Send Ping command"
2022-06-27 20:18:26 +02:00
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"
2022-06-21 16:49:11 +02:00
2022-09-27 17:35:58 +02:00
SWITCH_ON = "Switch Scex on"
2022-09-28 17:01:26 +02:00
SWITCH_OFF = "Switch Scex off"
2022-09-27 17:35:58 +02:00
2022-06-21 16:49:11 +02:00
@tmtc_definitions_provider
def add_scex_cmds(defs: TmtcDefinitionWrapper):
oce = OpCodeEntry()
2023-01-16 14:13:06 +01:00
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)
2022-06-27 20:18:26 +02:00
2023-01-16 14:13:06 +01:00
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)
2022-06-21 18:15:17 +02:00
defs.add_service(
2022-08-08 16:32:18 +02:00
name=CustomServiceList.SCEX.value, info="SCEX Device", op_code_entry=oce
2022-06-21 16:49:11 +02:00
)
2022-08-12 22:33:16 +02:00
@service_provider(CustomServiceList.SCEX.value)
2023-06-10 14:42:28 +02:00
def pack_scex_cmds(p: ServiceProviderParams): # noqa C901
2022-08-18 14:08:05 +02:00
op_code = p.op_code
q = p.queue_helper
2023-01-16 14:13:06 +01:00
if op_code in OpCode.SWITCH_ON:
2022-09-27 17:35:58 +02:00
q.add_log_cmd(Info.SWITCH_ON)
q.add_pus_tc(
PusTelecommand(
service=200,
2023-01-17 18:37:16 +01:00
subservice=Subservice.TC_MODE_COMMAND,
2023-01-16 15:05:33 +01:00
app_data=pack_mode_data(SCEX_HANDLER_ID, Mode.ON, 0),
2022-09-27 17:35:58 +02:00
)
)
2023-01-16 14:13:06 +01:00
if op_code in OpCode.SWITCH_OFF:
2022-09-28 17:01:26 +02:00
q.add_log_cmd(Info.SWITCH_OFF)
q.add_pus_tc(
PusTelecommand(
service=200,
2023-01-17 18:37:16 +01:00
subservice=Subservice.TC_MODE_COMMAND,
2023-01-16 15:05:33 +01:00
app_data=pack_mode_data(SCEX_HANDLER_ID, Mode.OFF, 0),
2022-09-28 17:01:26 +02:00
)
)
2023-01-16 14:13:06 +01:00
if op_code in OpCode.PING:
q.add_log_cmd(Info.PING)
2022-06-21 16:49:11 +02:00
app_data = bytes([0])
q.add_pus_tc(create_action_cmd(SCEX_HANDLER_ID, ActionId.PING, app_data))
2023-01-16 14:13:06 +01:00
if op_code in OpCode.ION_CMD:
q.add_log_cmd(Info.ION_CMD)
2022-06-27 20:18:26 +02:00
app_data = bytes([0])
q.add_pus_tc(create_action_cmd(SCEX_HANDLER_ID, ActionId.ION_CMD, app_data))
2023-01-16 14:13:06 +01:00
if op_code in OpCode.TEMP_CMD:
q.add_log_cmd(Info.TEMP_CMD)
2022-06-27 20:18:26 +02:00
app_data = bytes([0])
q.add_pus_tc(create_action_cmd(SCEX_HANDLER_ID, ActionId.TEMP_CMD, app_data))
2022-06-27 20:18:26 +02:00
2023-01-16 14:13:06 +01:00
if op_code in OpCode.EXP_STATUS_CMD:
q.add_log_cmd(Info.EXP_STATUS_CMD)
2022-06-21 18:15:17 +02:00
app_data = bytes([0])
2022-08-08 16:32:18 +02:00
q.add_pus_tc(
create_action_cmd(SCEX_HANDLER_ID, ActionId.EXP_STATUS_CMD, app_data)
2022-08-08 16:32:18 +02:00
)
2022-06-21 16:49:11 +02:00
2022-06-21 18:15:17 +02:00
# one cell
2023-01-16 14:13:06 +01:00
if op_code in OpCode.ONE_CELLS_CMD:
q.add_log_cmd(Info.ONE_CELLS_CMD)
2022-08-30 16:32:50 +02:00
app_data = bytearray([0])
2022-06-27 20:18:26 +02:00
# cell number
cn = 0
2022-06-27 20:18:26 +02:00
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
2022-06-30 21:49:25 +02:00
# app_data.extend(struct.pack("!H", first_dac))
2022-08-30 16:32:50 +02:00
app_data.append(cell_select)
2022-06-27 20:18:26 +02:00
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])
2022-08-08 16:32:18 +02:00
q.add_pus_tc(
create_action_cmd(SCEX_HANDLER_ID, ActionId.ONE_CELLS_CMD, app_data)
2022-08-08 16:32:18 +02:00
)
2022-06-27 20:18:26 +02:00
2023-01-16 14:13:06 +01:00
if op_code in OpCode.ALL_CELLS_CMD:
q.add_log_cmd(Info.ALL_CELLS_CMD)
2022-08-30 16:32:50 +02:00
app_data = bytearray([0])
2022-06-30 21:49:25 +02:00
# 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])
2022-08-08 16:32:18 +02:00
q.add_pus_tc(
create_action_cmd(SCEX_HANDLER_ID, ActionId.ALL_CELLS_CMD, app_data)
2022-08-08 16:32:18 +02:00
)
2022-06-30 21:49:25 +02:00
2023-01-16 14:13:06 +01:00
if op_code in OpCode.FRAM:
q.add_log_cmd(Info.FRAM)
2022-06-30 21:49:25 +02:00
app_data = bytes([0])
q.add_pus_tc(create_action_cmd(SCEX_HANDLER_ID, ActionId.FRAM, app_data))
2022-06-27 20:18:26 +02:00
def append_16_bit_val(packet: bytearray, val: int):
packet.append((val >> 8) & 0xFF)
packet.append(val & 0xFF)