from typing import Any, List, Optional, Sequence, Tuple, TypeVar, Union

from pygame.rect import Rect
from pygame.surface import Surface

from ._common import _ColorValue, _Coordinate, _RectValue

_Offset = TypeVar("_Offset", Tuple[int, int], Sequence[int])

def from_surface(surface: Surface, threshold: int = 127) -> Mask: ...
def from_threshold(
    surface: Surface,
    color: _ColorValue,
    threshold: _ColorValue = (0, 0, 0, 255),
    other_surface: Optional[Surface] = None,
    palette_colors: int = 1,
) -> Mask: ...

class Mask:
    def __init__(self, size: _Coordinate, fill: bool = False) -> None: ...
    def copy(self) -> Mask: ...
    def get_size(self) -> Tuple[int, int]: ...
    def get_rect(self, **kwargs: Any) -> Rect: ...  # Dict type needs to be completed
    def get_at(self, pos: _Coordinate) -> int: ...
    def set_at(self, pos: _Coordinate, value: int = 1) -> None: ...
    def overlap(self, other: Mask, offset: _Offset) -> Union[Tuple[int, int], None]: ...
    def overlap_area(self, other: Mask, offset: _Coordinate) -> int: ...
    def overlap_mask(self, other: Mask, offset: _Coordinate) -> Mask: ...
    def fill(self) -> None: ...
    def clear(self) -> None: ...
    def invert(self) -> None: ...
    def scale(self, size: _Coordinate) -> Mask: ...
    def draw(self, other: Mask, offset: _Coordinate) -> None: ...
    def erase(self, other: Mask, offset: _Coordinate) -> None: ...
    def count(self) -> int: ...
    def centroid(self) -> Tuple[int, int]: ...
    def angle(self) -> float: ...
    def outline(self, every: int = 1) -> List[Tuple[int, int]]: ...
    def convolve(
        self,
        other: Mask,
        output: Optional[Mask] = None,
        offset: _Coordinate = (0, 0),
    ) -> Mask: ...
    def connected_component(
        self, pos: Union[List[int], Tuple[int, int]] = ...
    ) -> Mask: ...
    def connected_components(self, minimum: int = 0) -> List[Mask]: ...
    def get_bounding_rects(self) -> Rect: ...
    def to_surface(
        self,
        surface: Optional[Surface] = None,
        setsurface: Optional[Surface] = None,
        unsetsurface: Optional[Surface] = None,
        setcolor: _ColorValue = (255, 255, 255, 255),
        unsetcolor: _ColorValue = (0, 0, 0, 255),
        dest: Union[_RectValue, _Coordinate] = (0, 0),
    ) -> Surface: ...