Create modules/utils.py

modules/utils.py

@@ -0,0 +1,66 @@
+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