StarRailCopilot/tasks/map/control/waypoint.py

from dataclasses import dataclass, field

from module.base.timer import Timer


@dataclass
class Waypoint:
    # Position to goto, (x, y)
    position: tuple
    # Position diff < threshold is considered as arrived
    # `threshold` is used first if it is set
    threshold: int = None
    # If `threshold` is not set, `waypoint_threshold` and `endpoint_threshold` are used
    waypoint_threshold: int = 10
    endpoint_threshold: int = 3
    # Max move speed, 'run_2x', 'straight_run', 'run', 'walk'
    # See MapControl._goto() for details of each speed level
    speed: str = 'straight_run'

    """
    The following attributes are only be used if this waypoint is the end point of goto()
    """
    # True to enable endpoint optimizations, character will smoothly approach target position
    # False to stop all controls at arrive
    end_opt: bool = True
    # Set rotation after arrive, 0~360
    end_rotation: int = None
    end_rotation_threshold: int = 15

    """
    Walk
    """
    # A list of expected events, e.g. ['enemy', 'item']
    # - "enemy", finished any combat
    # - "item", destroyed any destructive objects
    # - "interact", have map interact option (interact is not handled)
    # - callable, A function that returns bool, True represents stop
    # Or empty list [] for just walking
    expected_end: list = field(default_factory=lambda: [])
    # If triggered any expected event, consider arrive and stop walking
    early_stop: bool = True
    # Confirm timer if arrived but didn't trigger any expected event
    unexpected_confirm: Timer = field(default_factory=lambda: Timer(2, count=6))

    def __str__(self):
        return f'Waypoint({self.position})'

    __repr__ = __str__

    def run_2x(self) -> "Waypoint":
        """
        Product a Waypoint object with overridden "speed",
        see Waypoint class for args.
        """
        self.speed = 'run_2x'
        return self

    def straight_run(self) -> "Waypoint":
        self.speed = 'straight_run'
        return self

    def run(self) -> "Waypoint":
        self.speed = 'run'
        return self

    def walk(self) -> "Waypoint":
        self.speed = 'walk'
        return self

    def set_threshold(self, threshold) -> "Waypoint":
        self.threshold = threshold
        return self

    def get_threshold(self, end):
        """
        Args:
            end: True if this is an end point

        Returns:
            int
        """
        if self.threshold is not None:
            return self.threshold
        if end:
            return self.endpoint_threshold
        else:
            return self.waypoint_threshold

    @staticmethod
    def expected_to_str(results: list) -> list[str]:
        return [result.__name__ if callable(result) else str(result) for result in results]

    def match_results(self, results) -> list[str]:
        """
        Args:
            results:

        Returns:
            list[str]: A list if matched results
        """
        if not results and not self.expected_end:
            return []

        results = set(self.expected_to_str(results))
        expected_end = set(self.expected_to_str(self.expected_end))
        same = results.intersection(expected_end)

        return list(same)


def ensure_waypoint(point) -> Waypoint:
    """
    Args:
        point: Position (x, y) or Waypoint object

    Returns:
        Waypoint:
    """

    if isinstance(point, Waypoint):
        return point
    return Waypoint(point)


def ensure_waypoints(points) -> list[Waypoint]:
    if not isinstance(points, (list, tuple)):
        points = [points]
    return [ensure_waypoint(point) for point in points]