eive-tmtc/pus_tc/devs/gps.py

import enum

from config.definitions import CustomServiceList
from tmtccmd.config import add_op_code_entry, add_service_op_code_entry
from tmtccmd.tc.pus_3_fsfw_hk import make_sid, generate_one_hk_command
from tmtccmd.config.definitions import QueueCommands, ServiceOpCodeDictT
from tmtccmd.logging import get_console_logger
from tmtccmd.tc.definitions import TcQueueT
from tmtccmd.tc.pus_8_funccmd import generate_action_command


from config.object_ids import GPS_HANDLER_1_ID, GPS_HANDLER_0_ID

LOGGER = get_console_logger()


class OpCodes:
    REQ_OS_HK = ["0", "hk-os"]
    RESET_GNSS = ["5", "reset"]


class Info:
    REQ_OS_HK = "Request One-Shot HK"
    RESET_GNSS = "Reset GNSS using reset pin"


class SetIds:
    HK = 0


def add_gps_cmds(cmd_dict: ServiceOpCodeDictT):
    op_code_dict = dict()
    add_op_code_entry(
        op_code_dict=op_code_dict, keys=OpCodes.RESET_GNSS, info=Info.RESET_GNSS
    )
    add_op_code_entry(
        op_code_dict=op_code_dict, keys=OpCodes.REQ_OS_HK, info=Info.REQ_OS_HK
    )
    add_service_op_code_entry(
        srv_op_code_dict=cmd_dict,
        op_code_entry=op_code_dict,
        name=CustomServiceList.GPS_CTRL.value,
        info="GPS/GNSS Controller"
    )


def pack_gps_command(object_id: bytes, tc_queue: TcQueueT, op_code: str):
    if op_code in OpCodes.RESET_GNSS:
        if object_id == GPS_HANDLER_0_ID:
            # TODO: This needs to be re-implemented
            LOGGER.warning("Reset pin handling needs to be re-implemented")
            return
            # tc_queue.appendleft((QueueCommands.PRINT, "Resetting GPS device 0"))
        elif object_id == GPS_HANDLER_1_ID:
            LOGGER.warning("Reset pin handling needs to be re-implemented")
            return
            # tc_queue.appendleft((QueueCommands.PRINT, "Resetting GPS device 1"))
        cmd = generate_action_command(object_id=object_id, action_id=int(op_code))
        tc_queue.appendleft(cmd.pack_command_tuple())
    if op_code in OpCodes.REQ_OS_HK:
        tc_queue.appendleft((QueueCommands.PRINT, f"GMSS: {Info.REQ_OS_HK}"))
        cmd = generate_one_hk_command(sid=make_sid(object_id=object_id, set_id=SetIds.HK), ssc=0)
        tc_queue.appendleft(cmd.pack_command_tuple())