@camera.ui/server 0.0.10 → 0.0.11-alpha.2

package/dist/server/src/polyglot/node/decoder/wasm/test/imageUtils.py

@@ -1,106 +0,0 @@
-import asyncio
-import time
-from typing import Any, Literal
-
-import numpy as np
-from PIL import Image
-
-width = 640
-height = 360
-
-
-def yuv420_to_rgb(yuv_bytes: bytes, width: int, height: int) -> np.ndarray[Any, Any]:
-    frame_size = width * height
-    uv_size = frame_size // 4
-
-    raw_data = np.frombuffer(yuv_bytes, dtype=np.uint8)
-
-    y = raw_data[:frame_size].reshape((height, width)).astype(np.float32) - 16.0
-    u = (
-        raw_data[frame_size : frame_size + uv_size].reshape((height // 2, width // 2)).astype(np.float32)
-        - 128.0
-    )
-    v = raw_data[frame_size + uv_size :].reshape((height // 2, width // 2)).astype(np.float32) - 128.0
-
-    # Upsample U and V to match Y dimensions in one step
-    u = u.repeat(2, axis=0).repeat(2, axis=1)
-    v = v.repeat(2, axis=0).repeat(2, axis=1)
-
-    # Perform YUV to RGB conversion
-    r = y + 1.402 * v
-    g = y - 0.344136 * u - 0.714136 * v
-    b = y + 1.772 * u
-
-    # return np.clip(np.stack((r, g, b), axis=-1), 0, 255).astype(np.uint8)
-    rgb = np.stack((r, g, b), axis=-1)
-    return np.clip(rgb, 0, 255).astype(np.uint8)
-
-
-def yuv420_to_rgba(yuv_bytes: bytes, width: int, height: int) -> np.ndarray[Any, Any]:
-    rgb = yuv420_to_rgb(yuv_bytes, width, height)
-    alpha = np.full((height, width, 1), 255, dtype=np.uint8)  # Alpha direkt als uint8
-    rgba = np.concatenate((rgb, alpha), axis=-1)
-    return rgba
-
-
-def yuv420_to_gray(yuv_bytes: bytes, width: int, height: int) -> np.ndarray[Any, Any]:
-    frame_size = width * height
-    raw_data = np.frombuffer(yuv_bytes, dtype=np.uint8)
-    return raw_data[:frame_size].reshape((height, width)).astype(np.uint8)
-
-
-def read_yuv_file():
-    with open("frame.yuv", "rb") as f:
-        return f.read()
-
-
-def main(format: Literal["rgb", "rgba", "gray"]):
-    yuv_bytes = read_yuv_file()
-
-    if format == "rgb":
-        rgb = yuv420_to_rgb(yuv_bytes, width, height)
-        rgb_image = Image.fromarray(rgb)
-        rgb_image.save("frame_rgb.jpg")
-    elif format == "rgba":
-        rgba = yuv420_to_rgba(yuv_bytes, width, height)
-        rgba_image = Image.fromarray(rgba)
-        rgba_image.save("frame_rgba.png")
-    elif format == "gray":
-        gray = yuv420_to_gray(yuv_bytes, width, height)
-        gray_image = Image.fromarray(gray)
-        gray_image.save("frame_gray.jpg")
-
-
-async def benchmark(format: Literal["rgb", "rgba", "gray"]):
-    yuv_bytes = read_yuv_file()
-
-    fps = 10
-    total_frames = 3000
-    average_latency_in_ms = 6
-    frame_duration = 1.0 / fps
-    current_frame = 0
-
-    start = time.time()
-
-    while current_frame < total_frames:
-        if format == "rgb":
-            rgb = yuv420_to_rgb(yuv_bytes, width, height)
-            Image.fromarray(rgb)
-        elif format == "rgba":
-            rgba = yuv420_to_rgba(yuv_bytes, width, height)
-            Image.fromarray(rgba)
-        elif format == "gray":
-            gray = yuv420_to_gray(yuv_bytes, width, height)
-            Image.fromarray(gray)
-
-        current_frame += 1
-        await asyncio.sleep(frame_duration + average_latency_in_ms / 1000.0)
-
-    end = time.time()
-    print("Simulation completed! Total duration:", (end - start) * 1000, "ms")
-
-
-format: Literal["rgb", "rgba", "gray"] = "gray"
-
-main(format)
-asyncio.run(benchmark(format))