Source code for m4opt.constraints._core

from abc import ABC, abstractmethod

import numpy as np
import numpy.typing as npt
from astropy.coordinates import EarthLocation, SkyCoord
from astropy.time import Time




class Constraint(ABC):
    """Base class for field of regard constraints."""



    @abstractmethod
    def __call__(
        self, observer_location: EarthLocation, target_coord: SkyCoord, obstime: Time
    ) -> npt.NDArray[np.bool_]:
        """Evaluate the constraint at a given observer location, target position, and time."""


    def __and__(self, rhs):
        from ._logical import LogicalAndConstraint

        return LogicalAndConstraint(self, rhs)

    def __or__(self, rhs):
        from ._logical import LogicalOrConstraint

        return LogicalOrConstraint(self, rhs)

    def __invert__(self):
        from ._logical import LogicalNotConstraint

        return LogicalNotConstraint(self)