transmit_metadata.py

from typing import Annotated, Union

import numpy as np
from pydantic import (
    AfterValidator,
    BaseModel,
)

FloatInfAsNone = Annotated[
    Union[float, None], AfterValidator(lambda x: None if x == np.inf else x)
]


class TransmitSteeringList(BaseModel):
    """Transmit angle and focus parameters."""

    tx_az_angle_deg: list[float]
    tx_el_angle_deg: list[float]
    tx_az_focus: list[FloatInfAsNone]
    tx_el_focus: list[FloatInfAsNone]