Create modules/lut.py

modules/lut.py

@@ -0,0 +1,47 @@
+import numpy as np
+
+LUTS = {
+    "cinematic": {
+        "name": "🎬 电影橙青",
+        "matrix": np.array([[1.1, 0.0, -0.1], [0.0, 1.05, 0.0], [-0.05, 0.0, 1.15]]),
+        "warmth": (5, 0, -5), "contrast": 1.15,
+    },
+    "bw": {
+        "name": "⬛ 经典黑白",
+        "matrix": np.array([[0.3, 0.59, 0.11], [0.3, 0.59, 0.11], [0.3, 0.59, 0.11]]),
+        "warmth": (0, 0, 0), "contrast": 1.3,
+    },
+    "cyberpunk": {
+        "name": "🌆 赛博朋克",
+        "matrix": np.array([[0.8, 0.0, 0.3], [0.0, 0.9, 0.1], [0.1, 0.0, 1.3]]),
+        "warmth": (-10, 0, 15), "contrast": 1.2,
+    },
+    "warm": {
+        "name": "☀️ 日系暖阳",
+        "matrix": np.array([[1.1, 0.05, 0.0], [0.0, 1.0, -0.05], [-0.05, -0.05, 0.9]]),
+        "warmth": (15, 5, -10), "contrast": 0.95,
+    },
+    "noir": {
+        "name": "🌙 夜景蓝调",
+        "matrix": np.array([[0.7, 0.0, 0.2], [0.0, 0.8, 0.1], [0.1, 0.0, 1.2]]),
+        "warmth": (-15, -5, 10), "contrast": 1.25,
+    },
+}
+
+def apply_lut(frame, lut_name):
+    lut = LUTS.get(lut_name)
+    if not lut:
+        return frame
+    frame_float = frame.astype(np.float32)
+    h, w = frame.shape[:2]
+    pixels = frame_float.reshape(-1, 3)
+    transformed = pixels @ lut["matrix"].T
+    transformed = np.clip(transformed, 0, 255).reshape(h, w, 3)
+    b, g, r = cv2.split(transformed.astype(np.uint8))
+    r = cv2.add(r, lut["warmth"][0])
+    g = cv2.add(g, lut["warmth"][1])
+    b = cv2.add(b, lut["warmth"][2])
+    frame = cv2.merge([b, g, r])
+    frame = frame.astype(np.float32)
+    frame = (frame - 128) * lut["contrast"] + 128
+    return np.clip(frame, 0, 255).astype(np.uint8)