ffmpeg-skill 0.10.0 → 0.12.0

package/scripts/stabilize.py

@@ -0,0 +1,83 @@
+#!/usr/bin/env python3
+"""Stabilize shaky video (FFmpeg's vidstab, two-pass).
+
+Pass 1 (vidstabdetect) analyses camera motion and writes the transforms to a
+temporary file; pass 2 (vidstabtransform) smooths that motion and re-renders
+the frames. The transforms file lives in a temp directory for the duration of
+this run only -- it is not a caller-facing artifact.
+
+--shakiness (1 = barely shaky, fast; 10 = very shaky, slow analysis) and
+--smoothing (how many neighbouring frames to average the camera path over)
+are the two knobs that matter most; --zoom crops in slightly to hide the
+black edges stabilizing can introduce (0 = keep the original framing and let
+edges show; ffmpeg's own --crop-mode is not exposed as a raw flag here).
+
+Examples:
+  python3 stabilize.py shaky.mp4
+  python3 stabilize.py shaky.mp4 --shakiness 8 --smoothing 20 --zoom 5
+"""
+import argparse
+import sys
+import tempfile
+from pathlib import Path
+
+from _common import STATE, add_common, apply_common, aac_args, cfr_args, default_output, die, emit, escape_filter_path, ffmpeg_base, info, probe, require_tool, run, video_args
+
+
+def main() -> int:
+    ap = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
+    ap.add_argument("input")
+    ap.add_argument("-o", "--output", help="output file (default: <name>_stab.<ext>)")
+    ap.add_argument("--shakiness", type=int, default=5, help="1 (barely shaky) .. 10 (very shaky), default 5")
+    ap.add_argument("--smoothing", type=int, default=15, help="frames of camera-path smoothing on each side, default 15")
+    ap.add_argument("--zoom", type=float, default=0.0, help="percent to zoom in to hide stabilization edges, 0..100 (default 0)")
+    ap.add_argument("--crf", type=int, default=18, help="x264 CRF (default 18)")
+    ap.add_argument("--preset", default="medium", help="x264 preset")
+    add_common(ap)
+    args = ap.parse_args()
+    apply_common(args)
+
+    if not 1 <= args.shakiness <= 10:
+        die(f"--shakiness must be 1..10, got {args.shakiness}")
+    if not 0 <= args.smoothing <= 1000:
+        die(f"--smoothing must be 0..1000, got {args.smoothing}")
+    if not 0 <= args.zoom <= 100:
+        die(f"--zoom must be 0..100, got {args.zoom}")
+
+    meta = probe(args.input)
+    if not meta.get("video"):
+        die("input has no video stream")
+    has_audio = bool(meta.get("audio"))
+    output = args.output or default_output(args.input, "stab")
+
+    with tempfile.TemporaryDirectory(prefix="ffmpeg-skill-vidstab-") as tmp:
+        trf = str(Path(tmp) / "transforms.trf")
+        trf_arg = escape_filter_path(trf)
+
+        if not STATE["dry_run"]:
+            ffmpeg = require_tool("ffmpeg")
+            detect_cmd = [ffmpeg, "-hide_banner", "-loglevel", "error", "-nostdin", "-y", "-i", args.input,
+                          "-vf", f"vidstabdetect=shakiness={args.shakiness}:result={trf_arg}", "-f", "null", "-"]
+            proc = run(detect_cmd, check=False)
+            if proc.returncode != 0:
+                die(f"stabilization analysis (pass 1) failed:\n{proc.stderr.strip()[-1500:]}")
+
+        transform_vf = f"vidstabtransform=input={trf_arg}:smoothing={args.smoothing}:zoom={args.zoom:g}:optzoom=1"
+        cmd = ffmpeg_base() + ["-i", args.input, "-vf", transform_vf]
+        cmd += video_args(meta, args.crf, args.preset)
+        cmd += cfr_args(meta)
+        if has_audio:
+            cmd += aac_args()
+        else:
+            cmd += ["-an"]
+        cmd.append(output)
+        run(cmd)
+
+    result = probe(output, role="output")
+    info(f"wrote {output} ({result['duration']:.3f}s, {result['video']['width']}x{result['video']['height']})")
+    emit(output)
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

package/scripts/sync.py

@@ -61,7 +61,7 @@ def decode_mono(path: str, seconds: float, start: float = 0.0) -> List[float]:
            "-vn", "-ac", "1", "-ar", str(SR), "-f", "s16le", "-"]
     proc = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     if proc.returncode != 0 or not proc.stdout:
-        die(f"could not decode audio from {path}:\n{proc.stderr.decode(errors='replace').strip()}")
+        die(f"could not decode audio from {path}:\n{proc.stderr.decode(errors='replace').strip()}", kind="ffmpeg")
     n = len(proc.stdout) // 2
     return [v / 32768.0 for v in struct.unpack(f"<{n}h", proc.stdout[: n * 2])]