eive-tmtc/pus_tc/system/acs.py

import enum

from tmtccmd.tc import DefaultPusQueueHelper
from tmtccmd.tc.pus_200_fsfw_modes import Modes
from config.object_ids import ACS_BOARD_ASS_ID, SUS_BOARD_ASS_ID

from .common import command_mode


class AcsOpCodes:
    ACS_ASS_A_SIDE = ["0", "a"]
    ACS_ASS_B_SIDE = ["1", "b"]
    ACS_ASS_DUAL_MODE = ["2", "dual"]
    ACS_ASS_OFF = ["3", "off"]
    ACS_ASS_A_ON = ["4", "a_on"]
    ACS_ASS_B_ON = ["5", "b_on"]
    ACS_ASS_DUAL_ON = ["6", "dual_on"]


class SusOpCodes:
    SUS_ASS_NOM_SIDE = ["0", "nom"]
    SUS_ASS_RED_SIDE = ["1", "red"]
    SUS_ASS_DUAL_MODE = ["2", "dual"]
    SUS_ASS_OFF = ["3", "off"]


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


def pack_acs_command(q: DefaultPusQueueHelper, op_code: str):
    if op_code in AcsOpCodes.ACS_ASS_A_SIDE:
        command_mode(
            object_id=ACS_BOARD_ASS_ID,
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.A_SIDE,
            q=q,
            info="Switching to ACS board assembly A side",
        )
    if op_code in AcsOpCodes.ACS_ASS_B_SIDE:
        command_mode(
            object_id=ACS_BOARD_ASS_ID,
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.B_SIDE,
            q=q,
            info="Switching to ACS board assembly B side",
        )
    if op_code in AcsOpCodes.ACS_ASS_DUAL_MODE:
        command_mode(
            object_id=ACS_BOARD_ASS_ID,
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.DUAL_SIDE,
            q=q,
            info="Switching to ACS board assembly dual mode",
        )
    if op_code in AcsOpCodes.ACS_ASS_A_ON:
        command_mode(
            object_id=ACS_BOARD_ASS_ID,
            mode=Modes.ON,
            submode=DualSideSubmodes.A_SIDE,
            q=q,
            info="Switching ACS board assembly A side on",
        )
    if op_code in AcsOpCodes.ACS_ASS_B_ON:
        command_mode(
            object_id=ACS_BOARD_ASS_ID,
            mode=Modes.ON,
            submode=DualSideSubmodes.B_SIDE,
            q=q,
            info="Switching ACS board assembly B side on",
        )
    if op_code in AcsOpCodes.ACS_ASS_DUAL_ON:
        command_mode(
            object_id=ACS_BOARD_ASS_ID,
            mode=Modes.ON,
            submode=DualSideSubmodes.B_SIDE,
            q=q,
            info="Switching ACS board assembly dual side on",
        )
    if op_code in AcsOpCodes.ACS_ASS_OFF:
        command_mode(
            object_id=ACS_BOARD_ASS_ID,
            mode=Modes.OFF,
            submode=0,
            q=q,
            info="Switching to ACS board assembly off",
        )


def pack_sus_cmds(q: DefaultPusQueueHelper, op_code: str):
    if op_code in SusOpCodes.SUS_ASS_NOM_SIDE:
        command_mode(
            object_id=SUS_BOARD_ASS_ID,
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.A_SIDE,
            q=q,
            info="Switching to SUS board to nominal side",
        )
    if op_code in SusOpCodes.SUS_ASS_RED_SIDE:
        command_mode(
            object_id=SUS_BOARD_ASS_ID,
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.B_SIDE,
            q=q,
            info="Switching to SUS board to redundant side",
        )
    if op_code in SusOpCodes.SUS_ASS_OFF:
        command_mode(
            object_id=SUS_BOARD_ASS_ID,
            mode=Modes.OFF,
            submode=0,
            q=q,
            info="Switching SUS board off",
        )
    if op_code in SusOpCodes.SUS_ASS_DUAL_MODE:
        command_mode(
            object_id=SUS_BOARD_ASS_ID,
            mode=Modes.NORMAL,
            submode=DualSideSubmodes.DUAL_SIDE,
            q=q,
            info="Switching to SUS board to dual side",
        )