eive-tmtc/pus_tc/system/acs.py

import enum
from tmtccmd.tc.definitions import TcQueueT, QueueCommands
from spacepackets.ecss.tc import PusTelecommand
from tmtccmd.tc.service_200_mode import pack_mode_data, Modes, Subservices
from config.object_ids import ACS_BOARD_ASS_ID


class AcsOpCodes:
    ACS_ASS_A_SIDE = ["0", "acs-a"]
    ACS_ASS_B_SIDE = ["1", "acs-b"]
    ACS_ASS_DUAL_MODE = ["2", "acs-d"]
    ACS_ASS_OFF = ["3", "acs-off"]
    ACS_ASS_A_ON = ["4", "acs-ao"]
    ACS_ASS_B_ON = ["5", "acs-bo"]
    ACS_ASS_DUAL_ON = ["6", "acs-do"]
    SUS_ASS_NOM_SIDE = ["10", "sus-nom"]
    SUS_ASS_RED_SIDE = ["11", "sus-red"]
    SUS_ASS_DUAL_MODE = ["12", "sus-d"]
    SUS_ASS_OFF = ["13", "sus-off"]


class DualSideSubmodes(enum.IntEnum):
    A_SIDE = 0
    B_SIDE = 1
    DUAL_SIDE = 2


def pack_acs_command(tc_queue: TcQueueT, op_code: str):
    if op_code in AcsOpCodes.ACS_ASS_A_SIDE:
        command_acs_board(
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.A_SIDE,
            tc_queue=tc_queue,
            info="Switching to ACS board assembly A side",
        )
    if op_code in AcsOpCodes.ACS_ASS_B_SIDE:
        command_acs_board(
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.B_SIDE,
            tc_queue=tc_queue,
            info="Switching to ACS board assembly B side",
        )
    if op_code in AcsOpCodes.ACS_ASS_DUAL_MODE:
        command_acs_board(
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.DUAL_SIDE,
            tc_queue=tc_queue,
            info="Switching to ACS board assembly dual mode",
        )
    if op_code in AcsOpCodes.ACS_ASS_A_ON:
        command_acs_board(
            mode=Modes.ON,
            submode=DualSideSubmodes.A_SIDE,
            tc_queue=tc_queue,
            info="Switching ACS board assembly A side on",
        )
    if op_code in AcsOpCodes.ACS_ASS_B_ON:
        command_acs_board(
            mode=Modes.ON,
            submode=DualSideSubmodes.B_SIDE,
            tc_queue=tc_queue,
            info="Switching ACS board assembly B side on",
        )
    if op_code in AcsOpCodes.ACS_ASS_DUAL_ON:
        command_acs_board(
            mode=Modes.ON,
            submode=DualSideSubmodes.B_SIDE,
            tc_queue=tc_queue,
            info="Switching ACS board assembly dual side on",
        )
    if op_code in AcsOpCodes.ACS_ASS_OFF:
        command_acs_board(
            mode=Modes.OFF,
            submode=0,
            tc_queue=tc_queue,
            info="Switching to ACS board assembly off",
        )
    if op_code in AcsOpCodes.SUS_ASS_NOM_SIDE:
        command_acs_board(
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.A_SIDE,
            tc_queue=tc_queue,
            info="Switching to SUS board to nominal side",
        )
    if op_code in AcsOpCodes.SUS_ASS_RED_SIDE:
        command_acs_board(
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.B_SIDE,
            tc_queue=tc_queue,
            info="Switching to SUS board to redundant side",
        )
    if op_code in AcsOpCodes.SUS_ASS_OFF:
        command_acs_board(
            mode=Modes.OFF,
            submode=0,
            tc_queue=tc_queue,
            info="Switching SUS board off",
        )


def command_acs_board(mode: Modes, submode: int, tc_queue: TcQueueT, info: str):
    tc_queue.appendleft((QueueCommands.PRINT, info))
    mode_data = pack_mode_data(
        object_id=ACS_BOARD_ASS_ID,
        mode=mode,
        submode=submode,
    )
    cmd = PusTelecommand(
        service=200, subservice=Subservices.COMMAND_MODE_COMMAND, app_data=mode_data
    )
    tc_queue.appendleft(cmd.pack_command_tuple())