"""
Image properties provider.

Delegates color-profile guessing and dominant-color extraction to
cores.vision.color — no duplicated k-means logic.
"""

from __future__ import annotations

import numpy as np

from config.settings import Settings, settings as _default_settings
from cores.vision import guess_color_profile, dominant_colors
from pipeline.feature_extraction import PipelineOutput
from providers.base import BaseProvider, ProviderCapability


class ImagePropertiesProvider(BaseProvider):
    name = "image_properties"
    capability = ProviderCapability.IMAGE_ANALYSIS

    def __init__(self, settings: Settings | None = None) -> None:
        super().__init__(settings=settings or _default_settings)

    def is_available(self) -> bool:
        return True

    def _run(self, pipeline_output: PipelineOutput) -> tuple[dict, dict]:
        img: np.ndarray = pipeline_output.image
        h, w = img.shape[:2]
        channels = img.shape[2] if img.ndim == 3 else 1
        aspect = round(w / h, 4) if h > 0 else 0
        megapixels = round((w * h) / 1_000_000, 4)
        profile = guess_color_profile(img)
        colors = dominant_colors(img, k=5)

        raw = {
            "width": w, "height": h, "channels": channels,
            "aspect_ratio": aspect, "megapixels": megapixels,
            "color_profile": profile, "dominant_colors": colors,
        }
        normalized = {
            "quality_score": None,
            "width": w, "height": h, "channels": channels,
            "color_profile": profile, "dominant_colors": colors,
            "aspects": {"aspect_ratio": aspect, "megapixels": megapixels},
        }
        return raw, normalized