eive-tmtc/pus_tc/imtq.py

# -*- coding: utf-8 -*-
"""
@file   imtq.py
@brief  Tests for the ISIS IMTQ (Magnettorquer) device handler
@author J. Meier
@date   25.03.2021
"""
from tmtccmd.core.definitions import QueueCommands

from tmtccmd.pus_tc.packer import TcQueueT
from tmtccmd.ecss.tc import PusTelecommand


class ImtqTestProcedure:
    """
    @brief  Use this class to define the tests to perform for the IMTQ Handler.
    @details    Setting all to True will run all tests.
                Setting all to False will only run the tests set to True.
    """
    all = False
    command_dipole = True
    get_commanded_dipole = True


class ImtqActionIds:
    start_actuation_dipole = bytearray([0x0, 0x0, 0x0, 0x02])
    get_commanded_dipole = bytearray([0x0, 0x0, 0x0, 0x03])


def pack_imtq_test_into(object_id: bytearray, tc_queue: TcQueueT) -> TcQueueT:
    tc_queue.appendleft(
        (QueueCommands.PRINT,
         "Testing ISIS IMTQ handler with object id: 0x" + object_id.hex())
    )

    if ImtqTestProcedure.all or ImtqTestProcedure.command_dipole:
        tc_queue.appendleft((QueueCommands.PRINT, "IMTQ: Commanding dipole"))
        command = pack_dipole_command(object_id, 2000, 2000, 2000, 10000)
        command = PusTelecommand(service=8, subservice=128, ssc=20, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())

    if ImtqTestProcedure.all or ImtqTestProcedure.get_commanded_dipole:
        tc_queue.appendleft((QueueCommands.PRINT, "IMTQ: Get commanded dipole"))
        command = object_id + ImtqActionIds.get_commanded_dipole
        command = PusTelecommand(service=8, subservice=128, ssc=20, app_data=command)
        tc_queue.appendleft(command.pack_command_tuple())


def pack_dipole_command(object_id: bytearray, x_dipole: int, y_dipole: int, z_dipole: int, duration: int) -> bytearray:
    """ This function packs the command causing the ISIS IMTQ to generate a dipole.
    @param object_id    The object id of the gomspace device handler.
    @param x_dipole   The dipole of the x coil in 10^-4*Am^2 (max. 2000)
    @param y_dipole   The dipole of the y coil in 10^-4*Am^2 (max. 2000)
    @param z_dipole   The dipole of the z coil in 10^-4*Am^2 (max. 2000)
    @param duration  The duration in milliseconds the dipole will be generated by the coils.
                     When set to 0, the dipole will be generated until a new dipole actuation
                     command is sent.
    """
    action_id = ImtqActionIds.start_actuation_dipole
    command = bytearray()
    command = object_id + action_id
    command.extend(x_dipole.to_bytes(length=2, byteorder='big'))
    command.extend(y_dipole.to_bytes(length=2, byteorder='big'))
    command.extend(z_dipole.to_bytes(length=2, byteorder='big'))
    command.extend(duration.to_bytes(length=2, byteorder='big'))
    return command