kinemotion 0.39.0__py3-none-any.whl → 0.40.0__py3-none-any.whl

kinemotion/core/debug_overlay_utils.py

@@ -1,5 +1,11 @@
 """Shared debug overlay utilities for video rendering."""
 
+import os
+import shutil
+import subprocess
+import time
+from pathlib import Path
+
 import cv2
 import numpy as np
 
@@ -11,7 +17,7 @@ def create_video_writer(
     display_width: int,
     display_height: int,
     fps: float,
-) -> tuple[cv2.VideoWriter, bool]:
+) -> tuple[cv2.VideoWriter, bool, str]:
     """
     Create a video writer with fallback codec support.
 
@@ -24,28 +30,43 @@ def create_video_writer(
         fps: Frames per second
 
     Returns:
-        Tuple of (video_writer, needs_resize)
+        Tuple of (video_writer, needs_resize, used_codec)
     """
     needs_resize = (display_width != width) or (display_height != height)
 
-    # Try H.264 codec first (better quality/compatibility), fallback to mp4v
-    fourcc = cv2.VideoWriter_fourcc(*"avc1")
-    writer = cv2.VideoWriter(output_path, fourcc, fps, (display_width, display_height))
+    # Try browser-compatible codecs first
+    # avc1/h264: H.264 (Most compatible)
+    # vp09: VP9 (Good compatibility)
+    # mp4v: MPEG-4 (Poor browser support, last resort)
+    codecs_to_try = ["avc1", "h264", "vp09", "mp4v"]
 
-    # Check if writer opened successfully, fallback to mp4v if not
-    if not writer.isOpened():
-        fourcc = cv2.VideoWriter_fourcc(*"mp4v")
-        writer = cv2.VideoWriter(
-            output_path, fourcc, fps, (display_width, display_height)
-        )
+    writer = None
+    used_codec = "mp4v"  # Default fallback
 
-    if not writer.isOpened():
+    for codec in codecs_to_try:
+        try:
+            fourcc = cv2.VideoWriter_fourcc(*codec)
+            writer = cv2.VideoWriter(
+                output_path, fourcc, fps, (display_width, display_height)
+            )
+            if writer.isOpened():
+                used_codec = codec
+                if codec == "mp4v":
+                    print(
+                        f"Warning: Fallback to {codec} codec. "
+                        "Video may not play in browsers."
+                    )
+                break
+        except Exception:
+            continue
+
+    if writer is None or not writer.isOpened():
         raise ValueError(
             f"Failed to create video writer for {output_path} with dimensions "
             f"{display_width}x{display_height}"
         )
 
-    return writer, needs_resize
+    return writer, needs_resize, used_codec
 
 
 def write_overlay_frame(
@@ -95,11 +116,12 @@ class BaseDebugOverlayRenderer:
             display_height: Display height (considering SAR)
             fps: Frames per second
         """
+        self.output_path = output_path
         self.width = width
         self.height = height
         self.display_width = display_width
         self.display_height = display_height
-        self.writer, self.needs_resize = create_video_writer(
+        self.writer, self.needs_resize, self.used_codec = create_video_writer(
             output_path, width, height, display_width, display_height, fps
         )
 
@@ -133,9 +155,67 @@ class BaseDebugOverlayRenderer:
         write_overlay_frame(self.writer, frame, self.display_width, self.display_height)
 
     def close(self) -> None:
-        """Release video writer."""
+        """Release video writer and re-encode if possible."""
         self.writer.release()
 
+        # Post-process with ffmpeg ONLY if we fell back to the incompatible mp4v codec
+        if self.used_codec == "mp4v" and shutil.which("ffmpeg"):
+            temp_path = None
+            try:
+                temp_path = str(
+                    Path(self.output_path).with_suffix(
+                        ".temp" + Path(self.output_path).suffix
+                    )
+                )
+
+                # Convert to H.264 with yuv420p pixel format for browser compatibility
+                # -y: Overwrite output file
+                # -vcodec libx264: Use H.264 codec
+                # -pix_fmt yuv420p: Required for wide browser support (Chrome,
+                #                   Safari, Firefox)
+                # -preset fast: Reasonable speed/compression tradeoff
+                # -crf 23: Standard quality
+                # -an: Remove audio (debug video has no audio)
+                cmd = [
+                    "ffmpeg",
+                    "-y",
+                    "-i",
+                    self.output_path,
+                    "-vcodec",
+                    "libx264",
+                    "-pix_fmt",
+                    "yuv420p",
+                    "-preset",
+                    "fast",
+                    "-crf",
+                    "23",
+                    "-an",
+                    temp_path,
+                ]
+
+                # Suppress output unless error
+                reencode_start = time.time()
+                subprocess.run(
+                    cmd,
+                    check=True,
+                    stdout=subprocess.DEVNULL,
+                    stderr=subprocess.PIPE,
+                )
+                reencode_duration = time.time() - reencode_start
+                print(f"Debug video re-encoded in {reencode_duration:.2f}s")
+
+                # Overwrite original file
+                os.replace(temp_path, self.output_path)
+
+            except subprocess.CalledProcessError as e:
+                print(f"Warning: Failed to re-encode debug video with ffmpeg: {e}")
+                if temp_path and os.path.exists(temp_path):
+                    os.remove(temp_path)
+            except Exception as e:
+                print(f"Warning: Error during video post-processing: {e}")
+                if temp_path and os.path.exists(temp_path):
+                    os.remove(temp_path)
+
     def __enter__(self) -> "BaseDebugOverlayRenderer":
         return self
 

{kinemotion-0.39.0.dist-info → kinemotion-0.40.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kinemotion
-Version: 0.39.0
+Version: 0.40.0
 Summary: Video-based kinematic analysis for athletic performance
 Project-URL: Homepage, https://github.com/feniix/kinemotion
 Project-URL: Repository, https://github.com/feniix/kinemotion

{kinemotion-0.39.0.dist-info → kinemotion-0.40.0.dist-info}/RECORD

@@ -12,7 +12,7 @@ kinemotion/cmj/validation_bounds.py,sha256=9ZTo68fl3ooyWjXXyTMRLpK9tFANa_rQf3oHh
 kinemotion/core/__init__.py,sha256=HsqolRa60cW3vrG8F9Lvr9WvWcs5hCmsTzSgo7imi-4,1278
 kinemotion/core/auto_tuning.py,sha256=wtCUMOhBChVJNXfEeku3GCMW4qED6MF-O_mv2sPTiVQ,11324
 kinemotion/core/cli_utils.py,sha256=zbnifPhD-OYofJioeYfJtshuWcl8OAEWtqCGVF4ctAI,7966
-kinemotion/core/debug_overlay_utils.py,sha256=TyUb5okv5qw8oeaX3jsUO_kpwf1NnaHEAOTm-8LwTno,4587
+kinemotion/core/debug_overlay_utils.py,sha256=-HRHw5Ew7hVH0MWwdutZV1JwPqQwzQ6Jex7xHWc3q88,7511
 kinemotion/core/determinism.py,sha256=NwVrHqJiVxxFHTBPVy8aDBJH2SLIcYIpdGFp7glblB8,2515
 kinemotion/core/experimental.py,sha256=IK05AF4aZS15ke85hF3TWCqRIXU1AlD_XKzFz735Ua8,3640
 kinemotion/core/filtering.py,sha256=GsC9BB71V07LJJHgS2lsaxUAtJsupcUiwtZFDgODh8c,11417
@@ -31,8 +31,8 @@ kinemotion/dropjump/kinematics.py,sha256=kH-XM66wlOCYMpjvyb6_Qh5ZebyOfFZ47rmhgE1
 kinemotion/dropjump/metrics_validator.py,sha256=CrTlGup8q2kyPXtA6HNwm7_yq0AsBaDllG7RVZdXmYA,9342
 kinemotion/dropjump/validation_bounds.py,sha256=5b4I3CKPybuvrbn-nP5yCcGF_sH4Vtyw3a5AWWvWnBk,4645
 kinemotion/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kinemotion-0.39.0.dist-info/METADATA,sha256=ytvO76fvfntV_Vpjkc-4BezahvOO91a2I6X3PUM0eAk,26020
-kinemotion-0.39.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-kinemotion-0.39.0.dist-info/entry_points.txt,sha256=zaqnAnjLvcdrk1Qvj5nvXZCZ2gp0prS7it1zTJygcIY,50
-kinemotion-0.39.0.dist-info/licenses/LICENSE,sha256=KZajvqsHw0NoOHOi2q0FZ4NBe9HdV6oey-IPYAtHXfg,1088
-kinemotion-0.39.0.dist-info/RECORD,,
+kinemotion-0.40.0.dist-info/METADATA,sha256=0X_uEEzEi-DH9d4HwfseQVEzlDmbJRw96Dn1o-m5DbI,26020
+kinemotion-0.40.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+kinemotion-0.40.0.dist-info/entry_points.txt,sha256=zaqnAnjLvcdrk1Qvj5nvXZCZ2gp0prS7it1zTJygcIY,50
+kinemotion-0.40.0.dist-info/licenses/LICENSE,sha256=KZajvqsHw0NoOHOi2q0FZ4NBe9HdV6oey-IPYAtHXfg,1088
+kinemotion-0.40.0.dist-info/RECORD,,