"""Threshold-based temperature profile segmentation into Ramp/Soak/Cool phases."""

from dataclasses import dataclass


@dataclass
class DataSegment:
    """A contiguous segment of the temperature profile."""
    label: str
    start_frame: int
    end_frame: int
    avg_temp: float


def segment_profile(avg_temps: list[float], threshold: float = 40.0) -> list[DataSegment]:
    """Scan average temperatures and identify Ramp/Soak/Cool phases.

    Args:
        avg_temps: Per-frame average temperatures.
        threshold: Temperature threshold (°C) defining Soak entry/exit.

    Returns:
        List of DataSegment objects covering the entire profile.
    """
    if not avg_temps:
        return []

    segments: list[DataSegment] = []
    n = len(avg_temps)
    i = 0

    while i < n and avg_temps[i] < threshold:
        i += 1
    if i > 0:
        avg = sum(avg_temps[:i]) / i
        segments.append(DataSegment("Ramp", 0, i - 1, round(avg, 2)))

    while i < n:
        start = i
        above = avg_temps[i] >= threshold
        while i < n and (avg_temps[i] >= threshold) == above:
            i += 1
        label = "Soak" if above else "Cool"
        count = i - start
        avg = sum(avg_temps[start:i]) / count
        segments.append(DataSegment(label, start, i - 1, round(avg, 2)))

    return segments