import enum from eive_tmtc.config.definitions import CustomServiceList from tmtccmd.config.tmtc import ( tmtc_definitions_provider, TmtcDefinitionWrapper, OpCodeEntry, ) from tmtccmd.tc import service_provider, DefaultPusQueueHelper from tmtccmd.tc.decorator import ServiceProviderParams from tmtccmd.tc.pus_200_fsfw_modes import Mode from eive_tmtc.config.object_ids import ACS_BOARD_ASS_ID from eive_tmtc.tmtc.common import pack_mode_cmd_with_info class AcsOpCodes: ACS_ASS_A_SIDE = ["0", "a"] ACS_ASS_B_SIDE = ["1", "b"] ACS_ASS_DUAL_MODE = ["2", "d"] ACS_ASS_OFF = ["3", "off"] ACS_ASS_A_ON = ["4", "ao"] ACS_ASS_B_ON = ["5", "bo"] ACS_ASS_DUAL_ON = ["6", "do"] class DualSideSubmode(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: pack_mode_cmd_with_info( object_id=ACS_BOARD_ASS_ID, mode=Mode.NORMAL, submode=DualSideSubmode.A_SIDE, q=q, info="Switching to ACS board assembly A side", ) if op_code in AcsOpCodes.ACS_ASS_B_SIDE: pack_mode_cmd_with_info( object_id=ACS_BOARD_ASS_ID, mode=Mode.NORMAL, submode=DualSideSubmode.B_SIDE, q=q, info="Switching to ACS board assembly B side", ) if op_code in AcsOpCodes.ACS_ASS_DUAL_MODE: pack_mode_cmd_with_info( object_id=ACS_BOARD_ASS_ID, mode=Mode.NORMAL, submode=DualSideSubmode.DUAL_SIDE, q=q, info="Switching to ACS board assembly dual mode", ) if op_code in AcsOpCodes.ACS_ASS_A_ON: pack_mode_cmd_with_info( object_id=ACS_BOARD_ASS_ID, mode=Mode.ON, submode=DualSideSubmode.A_SIDE, q=q, info="Switching ACS board assembly A side on", ) if op_code in AcsOpCodes.ACS_ASS_B_ON: pack_mode_cmd_with_info( object_id=ACS_BOARD_ASS_ID, mode=Mode.ON, submode=DualSideSubmode.B_SIDE, q=q, info="Switching ACS board assembly B side on", ) if op_code in AcsOpCodes.ACS_ASS_DUAL_ON: pack_mode_cmd_with_info( object_id=ACS_BOARD_ASS_ID, mode=Mode.ON, submode=DualSideSubmode.B_SIDE, q=q, info="Switching ACS board assembly dual side on", ) if op_code in AcsOpCodes.ACS_ASS_OFF: pack_mode_cmd_with_info( object_id=ACS_BOARD_ASS_ID, mode=Mode.OFF, submode=0, q=q, info="Switching to ACS board assembly off", ) @service_provider(CustomServiceList.ACS_BRD_ASS) def pack_acs_command_provider(p: ServiceProviderParams): op_code = p.op_code q = p.queue_helper pack_acs_command(q, op_code) @tmtc_definitions_provider def add_acs_board_cmds(defs: TmtcDefinitionWrapper): oce = OpCodeEntry() oce.add( keys=AcsOpCodes.ACS_ASS_A_SIDE, info="Switch to ACS board A side", ) oce.add( keys=AcsOpCodes.ACS_ASS_B_SIDE, info="Switch to ACS board B side", ) oce.add( keys=AcsOpCodes.ACS_ASS_DUAL_MODE, info="Switch to ACS board dual mode", ) oce.add( keys=AcsOpCodes.ACS_ASS_A_ON, info="Switch ACS board A side on", ) oce.add( keys=AcsOpCodes.ACS_ASS_B_ON, info="Switch ACS board B side on", ) oce.add( keys=AcsOpCodes.ACS_ASS_DUAL_ON, info="Switch ACS board dual mode on", ) oce.add( keys=AcsOpCodes.ACS_ASS_OFF, info="Switch off ACS board", ) defs.add_service( name=CustomServiceList.ACS_BRD_ASS.value, info="ACS Board Assemblie", op_code_entry=oce, )