ops-sat-rs/pytmtc/opssat_tmtc/common.py

from __future__ import annotations

import dataclasses
import enum
import struct

EXPERIMENT_ID = 278
EXPERIMENT_APID = 1024 + EXPERIMENT_ID


class UniqueId(enum.IntEnum):
    Controller = 0
    PusEventManagement = 1
    PusRouting = 2
    PusTest = 3
    PusAction = 4
    PusMode = 5
    PusHk = 6
    UdpServer = 7
    TcpServer = 8
    TcpSppClient = 9
    PusScheduler = 10
    CameraHandler = 11


class EventSeverity(enum.IntEnum):

    INFO = 0
    LOW = 1
    MEDIUM = 2
    HIGH = 3


@dataclasses.dataclass
class EventU32:
    severity: EventSeverity
    group_id: int
    unique_id: int

    @classmethod
    def unpack(cls, data: bytes) -> EventU32:
        if len(data) < 4:
            raise ValueError("passed data too short")
        event_raw = struct.unpack("!I", data[0:4])[0]
        return cls(
            severity=EventSeverity((event_raw >> 30) & 0b11),
            group_id=(event_raw >> 16) & 0x3FFF,
            unique_id=event_raw & 0xFFFF,
        )


class AcsId(enum.IntEnum):
    MGM_0 = 0


class AcsHkIds(enum.IntEnum):
    MGM_SET = 1


def make_unique_id(unique_id: int) -> bytes:
    return struct.pack("!I", unique_id)


def make_action_cmd_header(unique_id: int, action_id: int) -> bytes:
    byte_string = bytearray(struct.pack("!I", unique_id))
    byte_string.extend(struct.pack("!I", action_id))
    return byte_string