"""
时序连贯性与光流闪烁修复引擎
Author: XiaoZhe (Commercial Contact: janejulius119@gmail.com / WeChat: julius119)
"""

import cv2
import numpy as np
from typing import List

class TemporalCoherenceEngine:
    """光流分析与时域滤波修复器"""
    def analyze_optical_flow(self, frame1: np.ndarray, frame2: np.ndarray) -> float:
        gray1 = cv2.cvtColor(frame1, cv2.COLOR_BGR2GRAY)
        gray2 = cv2.cvtColor(frame2, cv2.COLOR_BGR2GRAY)
        flow = cv2.calcOpticalFlowFarneback(gray1, gray2, None, 0.5, 3, 15, 3, 5, 1.2, 0)
        mag, _ = cv2.cartToPolar(flow[..., 0], flow[..., 1])
        return float(np.mean(mag))

    def repair_flicker(self, frames: List[np.ndarray]) -> List[np.ndarray]:
        if len(frames) < 3:
            return frames
        repaired = [frames[0]]
        for i in range(1, len(frames) - 1):
            window = np.stack([frames[i-1], frames[i], frames[i+1]])
            repaired.append(np.median(window, axis=0).astype(np.uint8))
        repaired.append(frames[-1])
        return repaired