eive-tmtc/pus_tc/devs/syrlinks_hk_handler.py

# -*- coding: utf-8 -*-
"""
@file   syrlinks_hk_handler.py
@brief  Syrlinks Hk Handler tests
@author J. Meier
@date   13.12.2020
"""

from tmtccmd.config.definitions import QueueCommands
from tmtccmd.tc.definitions import TcQueueT
from tmtccmd.tc.service_3_housekeeping import make_sid, generate_one_hk_command
from spacepackets.ecss.tc import PusTelecommand
from tmtccmd.tc.service_200_mode import pack_mode_data, Modes
import struct


class SetIds:
    RX_REGISTERS_DATASET = 1
    TX_REGISTERS_DATASET = 2


class CommandIds:
    SET_TX_MODE_STANDBY = bytearray([0x0, 0x0, 0x0, 0x3])
    SET_TX_MODE_MODULATION = bytearray([0x0, 0x0, 0x0, 0x4])
    SET_TX_MODE_CW = bytearray([0x0, 0x0, 0x0, 0x5])
    READ_TX_STATUS = bytearray([0x0, 0x0, 0x0, 0x7])
    READ_TX_WAVEFORM = bytearray([0x0, 0x0, 0x0, 0x8])
    READ_TX_AGC_VALUE_HIGH_BYTE = bytearray([0x0, 0x0, 0x0, 0x9])
    READ_TX_AGC_VALUE_LOW_BYTE = bytearray([0x0, 0x0, 0x0, 0x9])
    WRITE_LCL_CONFIG = 11
    READ_LCL_CONFIG_REGISTER = 12
    READ_RX_STATUS_REGISTERS = 2


def pack_syrlinks_command(
    object_id: bytearray, tc_queue: TcQueueT, op_code: str
) -> TcQueueT:
    tc_queue.appendleft(
        (
            QueueCommands.PRINT,
            "Testing Syrlinks with object id: 0x" + object_id.hex(),
        )
    )

    if op_code == "0":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Set mode off"))
        command = pack_mode_data(object_id, Modes.OFF, 0)
        command = PusTelecommand(service=200, subservice=1, ssc=9, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "1":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Set mode on"))
        command = pack_mode_data(object_id, Modes.ON, 0)
        command = PusTelecommand(service=200, subservice=1, ssc=10, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "2":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Mode Normal"))
        command = pack_mode_data(object_id, Modes.NORMAL, 0)
        command = PusTelecommand(service=200, subservice=1, ssc=11, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "3":
        tc_queue.appendleft((QueueCommands.PRINT, "syrlinks: Set TX mode standby"))
        command = object_id + CommandIds.SET_TX_MODE_STANDBY
        command = PusTelecommand(service=8, subservice=128, ssc=10, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "4":
        tc_queue.appendleft((QueueCommands.PRINT, "syrlinks: Set TX mode modulation"))
        command = object_id + CommandIds.SET_TX_MODE_MODULATION
        command = PusTelecommand(service=8, subservice=128, ssc=11, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "5":
        tc_queue.appendleft((QueueCommands.PRINT, "syrlinks: Set TX mode CW"))
        command = object_id + CommandIds.SET_TX_MODE_CW
        command = PusTelecommand(service=8, subservice=128, ssc=12, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "6":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Get RX Registers"))
        sid = make_sid(object_id, SetIds.RX_REGISTERS_DATASET)
        command = generate_one_hk_command(sid, 200)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "7":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Get TX Registers"))
        sid = make_sid(object_id, SetIds.TX_REGISTERS_DATASET)
        command = generate_one_hk_command(sid, 201)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "8":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Read TX status"))
        command = object_id + CommandIds.READ_TX_STATUS
        command = PusTelecommand(service=8, subservice=128, ssc=13, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "9":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Read TX waveform"))
        command = object_id + CommandIds.READ_TX_WAVEFORM
        command = PusTelecommand(service=8, subservice=128, ssc=14, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "10":
        tc_queue.appendleft(
            (QueueCommands.PRINT, "Syrlinks: Read TX AGC value high byte")
        )
        command = object_id + CommandIds.READ_TX_AGC_VALUE_HIGH_BYTE
        command = PusTelecommand(service=8, subservice=128, ssc=15, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "11":
        tc_queue.appendleft(
            (QueueCommands.PRINT, "Syrlinks: Read TX AGC value low byte")
        )
        command = object_id + CommandIds.READ_TX_AGC_VALUE_LOW_BYTE
        command = PusTelecommand(service=8, subservice=128, ssc=16, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "12":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Write LCL config"))
        command = object_id + struct.pack("!I", CommandIds.WRITE_LCL_CONFIG)
        command = PusTelecommand(service=8, subservice=128, ssc=17, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "13":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Read RX status registers"))
        command = object_id + struct.pack("!I", CommandIds.READ_RX_STATUS_REGISTERS)
        command = PusTelecommand(service=8, subservice=128, ssc=18, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())
    if op_code == "14":
        tc_queue.appendleft((QueueCommands.PRINT, "Syrlinks: Read LCL config register"))
        command = object_id + struct.pack("!I", CommandIds.READ_LCL_CONFIG_REGISTER)
        command = PusTelecommand(service=8, subservice=128, ssc=19, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())