eive-tmtc/eive_tmtc/tmtc/acs/gps.py

import datetime
import enum
import logging
import struct

from eive_tmtc.config.definitions import CustomServiceList
from eive_tmtc.pus_tm.defs import PrintWrapper
from tmtccmd.config import TmtcDefinitionWrapper, OpCodeEntry
from tmtccmd.config.tmtc import tmtc_definitions_provider
from tmtccmd.tc.pus_200_fsfw_mode import create_mode_command, Mode
from tmtccmd.tc import DefaultPusQueueHelper
from tmtccmd.tc.pus_3_fsfw_hk import (
    make_sid,
    create_request_one_hk_command,
    create_enable_periodic_hk_command_with_interval,
    create_disable_periodic_hk_command,
)
from tmtccmd.util.tmtc_printer import FsfwTmTcPrinter

_LOGGER = logging.getLogger(__name__)


class OpCode:
    OFF = "off"
    ON = "on"
    REQ_OS_HK = ["hk"]
    ENABLE_HK = ["enable_hk"]
    DISABLE_HK = ["disable_hk"]
    RESET_GNSS = ["reset"]


class Info:
    OFF = "Off"
    ON = "On"
    REQ_OS_HK = "Request One-Shot HK"
    ENABLE_HK = "Enable HK"
    DISABLE_HK = "Disable HK"
    RESET_GNSS = "Reset GNSS using reset pin"


class SetId(enum.IntEnum):
    HK = 0


@tmtc_definitions_provider
def add_gps_cmds(defs: TmtcDefinitionWrapper):
    oce = OpCodeEntry()
    oce.add(keys=OpCode.RESET_GNSS, info=Info.RESET_GNSS)
    oce.add(keys=OpCode.REQ_OS_HK, info=Info.REQ_OS_HK)
    oce.add(keys=OpCode.ENABLE_HK, info=Info.ENABLE_HK)
    oce.add(keys=OpCode.DISABLE_HK, info=Info.DISABLE_HK)
    oce.add(keys=OpCode.OFF, info=Info.OFF)
    oce.add(keys=OpCode.ON, info=Info.ON)
    defs.add_service(
        name=CustomServiceList.GPS_CTRL.value,
        info="GPS/GNSS Controller",
        op_code_entry=oce,
    )


def pack_gps_command(object_id: bytes, q: DefaultPusQueueHelper, op_code: str):
    sid = make_sid(object_id=object_id, set_id=SetId.HK)
    if op_code in OpCode.RESET_GNSS:
        # TODO: This needs to be re-implemented
        _LOGGER.warning("Reset pin handling needs to be re-implemented")
    if op_code in OpCode.ENABLE_HK:
        interval = float(input("Please specify interval in floating point seconds: "))
        if interval <= 0:
            raise ValueError("invalid interval")
        q.add_log_cmd(f"GPS: {Info.ENABLE_HK}")
        cmds = create_enable_periodic_hk_command_with_interval(
            diag=False, sid=sid, interval_seconds=interval
        )
        for cmd in cmds:
            q.add_pus_tc(cmd)
    if op_code in OpCode.DISABLE_HK:
        q.add_log_cmd(f"gps: {Info.DISABLE_HK}")
        q.add_pus_tc(create_disable_periodic_hk_command(diag=False, sid=sid))
    if op_code in OpCode.REQ_OS_HK:
        q.add_log_cmd(f"GPS: {Info.REQ_OS_HK}")
        q.add_pus_tc(create_request_one_hk_command(sid=sid))
    if op_code in OpCode.ON:
        q.add_log_cmd(f"GPS: {Info.ON}")
        q.add_pus_tc(create_mode_command(object_id, Mode.ON, 0))
    if op_code in OpCode.OFF:
        q.add_log_cmd(f"GPS: {Info.OFF}")
        q.add_pus_tc(create_mode_command(object_id, Mode.OFF, 0))


def handle_gps_data(pw: PrintWrapper, hk_data: bytes):
    pw.dlog(f"Received GPS data, HK data length {len(hk_data)}")
    current_idx = 0
    fmt_str = "!ddddBBBHBBBBBI"
    inc_len = struct.calcsize(fmt_str)
    (
        lat,
        long,
        alt,
        speed,
        fix,
        sats_in_use,
        sats_in_view,
        year,
        month,
        day,
        hours,
        minutes,
        seconds,
        unix_seconds,
    ) = struct.unpack(fmt_str, hk_data[current_idx : current_idx + inc_len])
    current_idx += inc_len
    if year == 0:
        date_string = "No date string, year is 0"
    else:
        date_string = datetime.datetime(
            year=year, month=month, day=day, hour=hours, minute=minutes, second=seconds
        )
    pw.dlog(f"Lat: {lat} deg")
    pw.dlog(f"Long: {long} deg")
    pw.dlog(f"Altitude: {alt} m | Speed: {speed} m/s")
    pw.dlog(
        f"Fix Type: {fix} | Sats in View {sats_in_view} | Sats in Use {sats_in_use}"
    )
    pw.dlog(f"GNSS Date: {date_string}")
    pw.dlog(f"Unix seconds {unix_seconds}")
    FsfwTmTcPrinter.get_validity_buffer(
        validity_buffer=hk_data[current_idx:], num_vars=14
    )