modules/utils.py

import cv2
import numpy as np
import os
from PIL import ImageFont

def parse_color(color_str, default=(255, 255, 255)):
    color_map = {
        "gold": (255, 215, 0), "red": (255, 60, 60), "blue": (60, 120, 255),
        "white": (255, 255, 255), "black": (0, 0, 0), "cyan": (0, 255, 255),
        "magenta": (255, 0, 255), "yellow": (255, 255, 0), "orange": (255, 165, 0),
        "pink": (255, 105, 180), "lime": (50, 255, 50),
    }
    if color_str.lower() in color_map:
        return color_map[color_str.lower()]
    if color_str.startswith("#") and len(color_str) == 7:
        try:
            return (int(color_str[1:3], 16), int(color_str[3:5], 16), int(color_str[5:7], 16))
        except:
            pass
    return default

def get_chinese_font():
    fonts = [
        "/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc",
        "/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
        "/usr/share/fonts/noto-cjk/NotoSansCJK-Regular.ttc",
        "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc",
        "/usr/share/fonts/truetype/wqy/wqy-microhei.ttc",
        "NotoSansCJK-Regular.ttc",
    ]
    for f in fonts:
        if os.path.exists(f):
            return f
    return None

def get_title_font(font_path, size):
    try:
        return ImageFont.truetype(font_path, size) if font_path else ImageFont.load_default()
    except:
        return ImageFont.load_default()

def fit_image_to_canvas(img, target_w, target_h):
    h, w = img.shape[:2]
    scale = min(target_w / w, target_h / h)
    new_w, new_h = int(w * scale), int(h * scale)
    resized = cv2.resize(img, (new_w, new_h))
    canvas = np.zeros((target_h, target_w, 3), dtype=np.uint8)
    x_offset, y_offset = (target_w - new_w) // 2, (target_h - new_h) // 2
    canvas[y_offset:y_offset + new_h, x_offset:x_offset + new_w] = resized
    return canvas

def ken_burns_crop(img, progress, w_out, h_out):
    h, w = img.shape[:2]
    zoom = 1.0 + 0.15 * progress
    crop_w, crop_h = w_out / zoom, h_out / zoom
    dx = max(0, min((w - crop_w) * progress, w - crop_w)) if w > crop_w else 0
    dy = max(0, min((h - crop_h) * progress, h - crop_h)) if h > crop_h else 0
    crop = img[int(dy):int(dy + crop_h), int(dx):int(dx + crop_w)]
    return cv2.resize(crop, (w_out, h_out))

def apply_cinemascope(frame, bar_height_ratio=0.08):
    h = frame.shape[0]
    bar_h = int(h * bar_height_ratio)
    frame[:bar_h, :] = 0
    frame[h - bar_h:, :] = 0
    return frame