# ruff: noqa: D101
# Missing docstring in public class


from __future__ import annotations

import sys
from typing import TYPE_CHECKING, Literal, TypeAlias, Union

import numpy as np
from arro3.core.types import ArrowArrayExportable, ArrowStreamExportable
from numpy.typing import NDArray

if sys.version_info >= (3, 12):
    from typing import TypedDict
else:
    from typing_extensions import TypedDict

if TYPE_CHECKING:
    from collections.abc import Sequence

    import pandas as pd
    import pyarrow as pa

    from lonboard._base import BaseExtension


IntFloat: TypeAlias = int | float
Units = Literal["meters", "common", "pixels"]

ColorAccessorInput = Union[
    list[int],
    tuple[int, int, int],
    tuple[int, ...],
    str,
    NDArray[np.uint8],
    "pa.FixedSizeListArray",
    "pa.ChunkedArray",
    ArrowArrayExportable,
    ArrowStreamExportable,
]
FloatAccessorInput = Union[
    int,
    float,
    NDArray[np.number],
    "pd.Series",
    "pa.FloatingPointArray",
    "pa.ChunkedArray",
    ArrowArrayExportable,
    ArrowStreamExportable,
]
H3AccessorInput = Union[
    NDArray[np.object_],
    NDArray[np.str_],
    NDArray[np.uint64],
    "pd.Series",
    "pa.StringArray",
    "pa.LargeStringArray",
    "pa.StringViewArray",
    "pa.UInt64Array",
    "pa.ChunkedArray",
    ArrowArrayExportable,
    ArrowStreamExportable,
]
A5AccessorInput = H3AccessorInput
NormalAccessorInput = Union[
    list[int],
    tuple[int, int, int],
    tuple[int, ...],
    NDArray[np.floating],
    "pa.FixedSizeListArray",
    "pa.ChunkedArray",
    ArrowArrayExportable,
    ArrowStreamExportable,
]
PointAccessorInput = Union[
    NDArray[np.float64],
    "pa.FixedSizeListArray",
    "pa.ChunkedArray",
    ArrowArrayExportable,
    ArrowStreamExportable,
]
TextAccessorInput = Union[
    str,
    NDArray[np.str_],
    "pd.Series",
    "pa.StringArray",
    "pa.LargeStringArray",
    "pa.ChunkedArray",
    ArrowArrayExportable,
    ArrowStreamExportable,
]


class BaseLayerKwargs(TypedDict, total=False):
    extensions: Sequence[BaseExtension]
    pickable: bool
    visible: bool
    opacity: IntFloat
    auto_highlight: bool


class ArcLayerKwargs(BaseLayerKwargs, total=False):
    great_circle: bool
    num_segments: int
    width_units: Units
    width_scale: IntFloat
    width_min_pixels: IntFloat
    width_max_pixels: IntFloat
    get_source_color: ColorAccessorInput
    get_target_color: ColorAccessorInput
    get_width: FloatAccessorInput
    get_height: FloatAccessorInput
    get_tilt: FloatAccessorInput


class BitmapLayerKwargs(BaseLayerKwargs, total=False):
    image: str
    bounds: (
        tuple[IntFloat, IntFloat, IntFloat, IntFloat]
        | tuple[
            tuple[IntFloat, IntFloat],
            tuple[IntFloat, IntFloat],
            tuple[IntFloat, IntFloat],
            tuple[IntFloat, IntFloat],
        ]
    )

    desaturate: IntFloat
    transparent_color: Sequence[IntFloat]
    tint_color: Sequence[IntFloat]


class BitmapTileLayerKwargs(BaseLayerKwargs, total=False):
    data: str | Sequence[str]
    tile_size: int
    zoom_offset: int
    max_zoom: int
    min_zoom: int
    extent: Sequence[IntFloat]
    max_cache_size: int
    # max_cache_byte_size: int  # noqa: ERA001
    refinement_strategy: Literal["best-available", "no-overlap", "never"]
    max_requests: int
    desaturate: IntFloat
    transparent_color: Sequence[IntFloat]
    tint_color: Sequence[IntFloat]


class ColumnLayerKwargs(BaseLayerKwargs, total=False):
    disk_resolution: int
    radius: IntFloat
    angle: IntFloat
    offset: tuple[IntFloat, IntFloat]
    coverage: IntFloat
    elevation_scale: IntFloat
    filled: bool
    stroked: bool
    extruded: bool
    wireframe: bool
    flat_shading: bool
    radius_units: Units
    line_width_units: Units
    line_width_scale: IntFloat
    line_width_min_pixels: IntFloat
    line_width_max_pixels: IntFloat
    get_fill_color: ColorAccessorInput
    get_line_color: ColorAccessorInput
    get_elevation: FloatAccessorInput
    get_line_width: FloatAccessorInput


class PathLayerKwargs(BaseLayerKwargs, total=False):
    width_units: Units
    width_scale: IntFloat
    width_min_pixels: IntFloat
    width_max_pixels: IntFloat
    joint_rounded: bool
    cap_rounded: bool
    miter_limit: int
    billboard: bool
    get_color: ColorAccessorInput
    get_width: FloatAccessorInput


class PointCloudLayerKwargs(BaseLayerKwargs, total=False):
    size_units: Units
    point_size: IntFloat
    get_color: ColorAccessorInput
    get_normal: NormalAccessorInput


class PolygonLayerKwargs(BaseLayerKwargs, total=False):
    stroked: bool
    filled: bool
    extruded: bool
    wireframe: bool
    elevation_scale: IntFloat
    line_width_units: Units
    line_width_scale: IntFloat
    line_width_min_pixels: IntFloat
    line_width_max_pixels: IntFloat
    line_joint_rounded: bool
    line_miter_limit: IntFloat
    get_fill_color: ColorAccessorInput
    get_line_color: ColorAccessorInput
    get_line_width: FloatAccessorInput
    get_elevation: FloatAccessorInput


class H3HexagonLayerKwargs(PolygonLayerKwargs, total=False):
    high_precision: bool


class A5LayerKwargs(PolygonLayerKwargs, total=False):
    pass


class S2LayerKwargs(PolygonLayerKwargs, total=False):
    pass


class GeohashLayerKwargs(PolygonLayerKwargs, total=False):
    pass


class RasterLayerKwargs(BaseLayerKwargs, total=False):
    tile_size: int
    zoom_offset: int
    max_zoom: int
    min_zoom: int
    extent: tuple[IntFloat, IntFloat, IntFloat, IntFloat]
    max_cache_size: int
    debounce_time: int


class ScatterplotLayerKwargs(BaseLayerKwargs, total=False):
    radius_units: Units
    radius_scale: IntFloat
    radius_min_pixels: IntFloat
    radius_max_pixels: IntFloat
    line_width_units: Units
    line_width_scale: IntFloat
    line_width_min_pixels: IntFloat
    line_width_max_pixels: IntFloat
    stroked: bool
    filled: bool
    billboard: bool
    antialiasing: bool
    get_radius: FloatAccessorInput
    get_fill_color: ColorAccessorInput
    get_line_color: ColorAccessorInput
    get_line_width: FloatAccessorInput


class SolidPolygonLayerKwargs(BaseLayerKwargs, total=False):
    filled: bool
    extruded: bool
    wireframe: bool
    elevation_scale: IntFloat
    get_elevation: FloatAccessorInput
    get_fill_color: ColorAccessorInput
    get_line_color: ColorAccessorInput


class TripsLayerKwargs(BaseLayerKwargs, total=False):
    width_units: Units
    width_scale: IntFloat
    width_min_pixels: IntFloat
    width_max_pixels: IntFloat
    joint_rounded: bool
    cap_rounded: bool
    miter_limit: IntFloat
    billboard: bool
    fade_trail: bool
    trail_length: IntFloat
    current_time: IntFloat
    get_color: ColorAccessorInput
    get_width: FloatAccessorInput


class HeatmapLayerKwargs(BaseLayerKwargs, total=False):
    radius_pixels: IntFloat
    intensity: IntFloat
    threshold: IntFloat
    color_domain: tuple[IntFloat, IntFloat]
    aggregation: Literal["SUM", "MEAN"]
    weights_texture_size: int
    debounce_timeout: int
    get_weight: FloatAccessorInput
