eive-tmtc/pus_tc/pdu1.py

# -*- coding: utf-8 -*-
"""
@file   tmtcc_tc_pdu1.py
@brief  PDU2 tests
@details PDU2 is mounted on the X4 slot of the P60 dock
@author J. Meier
@date   17.12.2020
"""
import enum
from tmtccmd.config.definitions import QueueCommands

from tmtccmd.tc.packer import TcQueueT
from tmtccmd.ecss.tc import PusTelecommand
from gomspace.gomspace_common import *
from pus_tc.p60dock import P60DockConfigTable
from gomspace.gomspace_pdu_definitions import *


class Pdu1OpCodes(enum.Enum):
    STAR_TRACKER_ON = "1"
    STAR_TRACKER_OFF = "4"
    ACS_A_SIDE_ON = "6"
    ACS_A_SIDE_OFF = "7"
    PRINT_SWITCH_STATE = "17"


class PDU1TestProcedure:
    """
    @brief  Use this class to define the tests to perform for the PDU2.
    @details    Setting all to True will run all tests.
                Setting all to False will only run the tests set to True.
    """
    all = False
    reboot = False
    ping = False
    read_temperature = False
    turn_channel_2_on = False    # Star Tracker connected to this channel (5V)
    turn_channel_2_off = False
    turn_channel_3_on = False  # MTQ connected to this channel (5V)
    turn_channel_3_off = False


def pack_pdu1_test_into(object_id: bytearray, tc_queue: TcQueueT, op_code: str):
    tc_queue.appendleft((QueueCommands.PRINT, "Commanding PDU1"))

    if op_code == Pdu1OpCodes.STAR_TRACKER_ON.value:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn star tracker on"))
        command = pack_set_param_command(object_id, PDUConfigTable.out_en_2.parameter_address,
                                         PDUConfigTable.out_en_2.parameter_size, Channel.on)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "3":
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn SUS nominal on"))
        command = pack_set_param_command(object_id, PDUConfigTable.out_en_4.parameter_address,
                                         PDUConfigTable.out_en_4.parameter_size, Channel.on)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == Pdu1OpCodes.STAR_TRACKER_OFF.value:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn star tracker off"))
        command = pack_set_param_command(object_id, PDUConfigTable.out_en_2.parameter_address,
                                         PDUConfigTable.out_en_2.parameter_size, Channel.off)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "5":
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn SUS nominal off"))
        command = pack_set_param_command(object_id, PDUConfigTable.out_en_4.parameter_address,
                                         PDUConfigTable.out_en_4.parameter_size, Channel.off)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == Pdu1OpCodes.ACS_A_SIDE_ON.value:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn ACS Side A on"))
        command = pack_set_param_command(
            object_id, PDUConfigTable.out_en_7.parameter_address,
            PDUConfigTable.out_en_7.parameter_size, Channel.on
        )
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == Pdu1OpCodes.ACS_A_SIDE_OFF.value:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn ACS Side A off"))
        command = pack_set_param_command(
            object_id, PDUConfigTable.out_en_7.parameter_address,
            PDUConfigTable.out_en_7.parameter_size, Channel.off
        )
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == Pdu1OpCodes.PRINT_SWITCH_STATE.value:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Print Switch Status"))
        command = generate_action_command(
            object_id=object_id, action_id=GomspaceDeviceActionIds.PRINT_SWITCH_STATE
        )
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "8":
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn SUS nominal on"))
        command = pack_set_param_command(object_id, PDUConfigTable.out_en_4.parameter_address,
                                         PDUConfigTable.out_en_4.parameter_size, Channel.on)
        command = PusTelecommand(service=8, subservice=128, ssc=36, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "9":
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn SUS nominal off"))
        command = pack_set_param_command(object_id, PDUConfigTable.out_en_4.parameter_address,
                                         PDUConfigTable.out_en_4.parameter_size, Channel.off)
        command = PusTelecommand(service=8, subservice=128, ssc=36, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())

    if PDU1TestProcedure.all or PDU1TestProcedure.ping:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Ping Test"))
        ping_data = bytearray([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
        command = pack_ping_command(object_id, ping_data)
        tc_queue.appendleft(command.pack_command_tuple())
    if PDU1TestProcedure.all or PDU1TestProcedure.read_temperature:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Testing temperature reading"))
        command = pack_get_param_command(
            object_id, TableIds.hk, PDUHkTable.temperature.parameter_address,
            PDUHkTable.temperature.parameter_size
        )
        tc_queue.appendleft(command.pack_command_tuple())
    if PDU1TestProcedure.all or PDU1TestProcedure.turn_channel_2_on:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn channel 2 on (Star Tracker)"))
        command = pack_set_param_command(
            object_id, PDUConfigTable.out_en_2.parameter_address,
            PDUConfigTable.out_en_2.parameter_size, Channel.on
        )
        tc_queue.appendleft(command.pack_command_tuple())
    if PDU1TestProcedure.all or PDU1TestProcedure.turn_channel_2_off:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn channel 2 off (Star Tracker)"))
        command = pack_set_param_command(
            object_id, PDUConfigTable.out_en_2.parameter_address,
            PDUConfigTable.out_en_2.parameter_size, Channel.off
        )
        tc_queue.appendleft(command.pack_command_tuple())
    if PDU1TestProcedure.all or PDU1TestProcedure.turn_channel_3_on:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn channel 3 on (MTQ)"))
        command = pack_set_param_command(
            object_id, PDUConfigTable.out_en_3.parameter_address,
            PDUConfigTable.out_en_3.parameter_size, Channel.on
        )
        tc_queue.appendleft(command.pack_command_tuple())
    if PDU1TestProcedure.all or PDU1TestProcedure.turn_channel_3_off:
        tc_queue.appendleft((QueueCommands.PRINT, "PDU1: Turn channel 3 off (MTQ)"))
        command = pack_set_param_command(
            object_id, PDUConfigTable.out_en_3.parameter_address,
            PDUConfigTable.out_en_3.parameter_size, Channel.off
        )
        tc_queue.appendleft(command.pack_command_tuple())