feat/ ver0.1
This commit is contained in:
@@ -71,10 +71,10 @@ def _load_weight_json(svo_path: Path, settings: Settings) -> Dict[str, Any]:
|
||||
|
||||
|
||||
def _find_preview_videos(output_dir: Path) -> Tuple[Optional[Path], Optional[Path]]:
|
||||
previews = sorted(output_dir.glob("*preview*.mp4"))
|
||||
previews = sorted(output_dir.rglob("*preview*.mp4"))
|
||||
if len(previews) >= 2:
|
||||
return previews[0], previews[1]
|
||||
all_mp4 = sorted(output_dir.glob("*.mp4"))
|
||||
all_mp4 = sorted(output_dir.rglob("*.mp4"))
|
||||
if len(all_mp4) >= 2:
|
||||
return all_mp4[0], all_mp4[1]
|
||||
if len(all_mp4) == 1:
|
||||
@@ -84,6 +84,46 @@ def _find_preview_videos(output_dir: Path) -> Tuple[Optional[Path], Optional[Pat
|
||||
return None, None
|
||||
|
||||
|
||||
def _split_sbs_video(src: Path, left_dst: Path, right_dst: Path) -> bool:
|
||||
"""Split a side-by-side stereo video (W x H where W == 2*H_single) into left/right halves.
|
||||
|
||||
Returns True if split succeeded, False otherwise (caller should fall back to copy).
|
||||
"""
|
||||
probe = subprocess.run(
|
||||
[
|
||||
"ffprobe", "-v", "quiet", "-print_format", "json",
|
||||
"-show_streams", str(src),
|
||||
],
|
||||
capture_output=True, text=True,
|
||||
)
|
||||
if probe.returncode != 0:
|
||||
return False
|
||||
import json as _json
|
||||
try:
|
||||
streams = _json.loads(probe.stdout).get("streams", [])
|
||||
vstream = next((s for s in streams if s.get("codec_type") == "video"), None)
|
||||
if vstream is None:
|
||||
return False
|
||||
w, h = int(vstream["width"]), int(vstream["height"])
|
||||
except Exception:
|
||||
return False
|
||||
half_w = w // 2
|
||||
if half_w < 1 or w < h:
|
||||
return False
|
||||
|
||||
for crop, dst in [
|
||||
(f"crop={half_w}:{h}:{half_w}:0", left_dst),
|
||||
(f"crop={half_w}:{h}:0:0", right_dst),
|
||||
]:
|
||||
r = subprocess.run(
|
||||
["ffmpeg", "-y", "-i", str(src), "-vf", crop, "-an", "-q:v", "5", str(dst)],
|
||||
capture_output=True, text=True,
|
||||
)
|
||||
if r.returncode != 0:
|
||||
return False
|
||||
return True
|
||||
|
||||
|
||||
def _publish_media(
|
||||
left: Optional[Path],
|
||||
right: Optional[Path],
|
||||
@@ -94,6 +134,13 @@ def _publish_media(
|
||||
right_dst = settings.media_root / "latest_right.mp4"
|
||||
base = settings.public_base_url.rstrip("/")
|
||||
|
||||
if left is not None and left == right and left.is_file():
|
||||
if _split_sbs_video(left, left_dst, right_dst):
|
||||
return (
|
||||
f"{base}/media/{left_dst.name}",
|
||||
f"{base}/media/{right_dst.name}",
|
||||
)
|
||||
|
||||
def publish(src: Optional[Path], dst: Path) -> str:
|
||||
if src is None or not src.is_file():
|
||||
return ""
|
||||
|
||||
Reference in New Issue
Block a user