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

import enum
from typing import Dict, Tuple

from eive_tmtc.config.object_ids import PL_SUBSYSTEM_ID
from spacepackets.ecss import PusTelecommand
from eive_tmtc.tmtc.common import pack_mode_cmd_with_info
from tmtccmd.config import CmdTreeNode
from tmtccmd.tmtc import DefaultPusQueueHelper
from tmtccmd.pus.s200_fsfw_mode import Subservice as ModeSubservice


class ModeId(enum.IntEnum):
    OFF = 0
    SUPV_ONLY = 10
    MPSOC_STREAM = 11
    CAM_STREAM = 12
    EARTH_OBSV = 13
    SCEX = 14


class OpCode(str, enum.Enum):
    OFF = "off"
    REPORT_ALL_MODES = "report_modes"
    MPSOC_STREAM = "mode_mpsoc_stream"
    CAM_STREAM = "mode_cam_stream"
    EARTH_OBSV = "mode_eart_obsv"


class Info(str, enum.Enum):
    OFF = "Off Command"
    REPORT_ALL_MODES = "Report all modes"
    MPSOC_STREAM = "MPSoC Stream Mode"
    CAM_STREAM = "Camera Stream Mode"
    EARTH_OBSV = "Earth Observation Mode"


HANDLER_LIST: Dict[str, Tuple[int, str]] = {
    OpCode.OFF: (ModeId.OFF, Info.OFF),
    OpCode.MPSOC_STREAM: (ModeId.MPSOC_STREAM, Info.MPSOC_STREAM),
}


def create_payload_subsystem_cmd(q: DefaultPusQueueHelper, cmd_str: str):
    info_prefix = "Payload Subsystem"
    if cmd_str == OpCode.REPORT_ALL_MODES:
        q.add_log_cmd(f"{info_prefix}: {Info.REPORT_ALL_MODES}")
        q.add_pus_tc(
            PusTelecommand(
                service=200,
                subservice=ModeSubservice.TC_MODE_ANNOUNCE_RECURSIVE,
                app_data=PL_SUBSYSTEM_ID,
            )
        )
    mode_info_tup = HANDLER_LIST.get(cmd_str)
    assert mode_info_tup is not None
    if mode_info_tup is None:
        return
    pack_mode_cmd_with_info(
        object_id=PL_SUBSYSTEM_ID,
        info=f"{info_prefix}: {mode_info_tup[1]}",
        mode=mode_info_tup[0],
        submode=0,
        q=q,
    )


def create_payload_subsystem_node() -> CmdTreeNode:
    payload_node = CmdTreeNode("payload", "Payload Subsystem")
    subsystem_node = CmdTreeNode("subsystem", "Subsystem Commands")
    subsystem_node.add_child(CmdTreeNode(OpCode.OFF, Info.OFF))
    subsystem_node.add_child(CmdTreeNode(OpCode.MPSOC_STREAM, Info.MPSOC_STREAM))
    subsystem_node.add_child(CmdTreeNode(OpCode.EARTH_OBSV, Info.EARTH_OBSV))
    subsystem_node.add_child(CmdTreeNode(OpCode.CAM_STREAM, Info.CAM_STREAM))

    subsystem_node.add_child(
        CmdTreeNode(OpCode.REPORT_ALL_MODES, Info.REPORT_ALL_MODES)
    )
    payload_node.add_child(subsystem_node)
    return payload_node