Added all functionality to cut_to_length
This commit is contained in:
Hannes committed
1 parent
6c1687d6bc
commit
cb56886a4b
5 files changed
+235
-149
No files matched your search
+92
-72
@@ -1,16 +1,66 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import subprocess
|
||||
import json
|
||||
import sys
|
||||
import shutil
|
||||
from pathlib import Path
|
||||
import cv2
|
||||
import numpy as np
|
||||
from scipy import signal
|
||||
|
||||
def run(cmd):
|
||||
return subprocess.run(cmd, capture_output=True, text=True, check=True)
|
||||
"""Run a subprocess command and raise on error."""
|
||||
subprocess.run(cmd, check=True)
|
||||
|
||||
def get_video_signal(path):
|
||||
cap = cv2.VideoCapture(path)
|
||||
motion_signal = []
|
||||
prev_gray = None
|
||||
|
||||
while True:
|
||||
ret, frame = cap.read()
|
||||
if not ret:
|
||||
break
|
||||
|
||||
# 1. Shrink the frame (makes it faster and removes noise)
|
||||
small_frame = cv2.resize(frame, (160, 90))
|
||||
gray = cv2.cvtColor(small_frame, cv2.COLOR_BGR2GRAY)
|
||||
|
||||
if prev_gray is not None:
|
||||
# 2. Calculate the difference between this frame and the last
|
||||
# This detects the "movement" or "cuts"
|
||||
diff = cv2.absdiff(gray, prev_gray)
|
||||
motion_signal.append(np.sum(diff))
|
||||
else:
|
||||
motion_signal.append(0)
|
||||
|
||||
prev_gray = gray
|
||||
|
||||
cap.release()
|
||||
return np.array(motion_signal)
|
||||
|
||||
def find_offset(video_a, video_b):
|
||||
# print("Extracting signals (this may take a moment)...")
|
||||
sig_a = get_video_signal(video_a)
|
||||
# print("sig_a finished")
|
||||
sig_b = get_video_signal(video_b)
|
||||
# print("sig_b finished")
|
||||
|
||||
# Normalize signals (subtract mean, divide by std dev) to improve correlation
|
||||
sig_a = (sig_a - np.mean(sig_a)) / (np.std(sig_a) + 1e-5)
|
||||
sig_b = (sig_b - np.mean(sig_b)) / (np.std(sig_b) + 1e-5)
|
||||
|
||||
# Perform Cross-Correlation
|
||||
correlation = signal.correlate(sig_a, sig_b, mode='full')
|
||||
lags = signal.correlation_lags(len(sig_a), len(sig_b), mode='full')
|
||||
# plt.plot(lags)
|
||||
|
||||
# The peak of the correlation is our offset in frames
|
||||
best_lag = lags[np.argmax(correlation)]
|
||||
|
||||
return best_lag
|
||||
|
||||
def probe_video(path):
|
||||
"""Get video FPS, duration, codec, and stream start time."""
|
||||
import json
|
||||
cmd = [
|
||||
"ffprobe",
|
||||
"-v", "error",
|
||||
@@ -19,11 +69,9 @@ def probe_video(path):
|
||||
"-show_format",
|
||||
path
|
||||
]
|
||||
|
||||
data = json.loads(run(cmd).stdout)
|
||||
data = json.loads(subprocess.run(cmd, capture_output=True, text=True, check=True).stdout)
|
||||
|
||||
duration = float(data["format"]["duration"])
|
||||
|
||||
for s in data["streams"]:
|
||||
if s["codec_type"] == "video":
|
||||
num, den = s["r_frame_rate"].split("/")
|
||||
@@ -34,95 +82,67 @@ def probe_video(path):
|
||||
|
||||
raise RuntimeError("No video stream found")
|
||||
|
||||
def vaapi_cut(inp, outp, start, end, codec="h264_vaapi"):
|
||||
encoder = codec
|
||||
|
||||
cmd = [
|
||||
def trim_audio(input_mkv, start_time, end_time, out_audio):
|
||||
"""Trim only the first (and only) audio track from input_mkv."""
|
||||
run([
|
||||
"ffmpeg",
|
||||
"-y",
|
||||
"-vaapi_device", "/dev/dri/renderD128",
|
||||
|
||||
"-ss", str(start),
|
||||
"-to", str(end),
|
||||
|
||||
"-i", inp,
|
||||
"-map", "0",
|
||||
|
||||
"-vf", "format=nv12,hwupload",
|
||||
|
||||
"-c:v", encoder,
|
||||
"-qp", "20",
|
||||
|
||||
"-c:a", "copy",
|
||||
"-c:s", "copy",
|
||||
"-c:d", "copy",
|
||||
"-c:t", "copy",
|
||||
|
||||
outp
|
||||
]
|
||||
|
||||
subprocess.run(cmd, check=True)
|
||||
|
||||
def extract_audio(output_mkv, out_audio):
|
||||
# Extract the first (and only) audio track
|
||||
run([
|
||||
"mkvextract",
|
||||
"tracks",
|
||||
output_mkv,
|
||||
f"1:{out_audio}" # track IDs are usually 0 for video, 1 for audio
|
||||
"-i", input_mkv,
|
||||
"-ss", str(start_time),
|
||||
"-to", str(end_time),
|
||||
"-map", "0:a:0", # first audio track
|
||||
"-c:a", "copy", # no re-encode
|
||||
out_audio
|
||||
])
|
||||
|
||||
def mux_audio_into_copy(inp2, audio_file, output_dir):
|
||||
def mux_audio_into_output(inp2, audio_file, output_dir):
|
||||
"""Mux trimmed audio into inp2 and place final file in output_dir."""
|
||||
output_dir = Path(output_dir)
|
||||
output_dir.mkdir(exist_ok=True)
|
||||
|
||||
copied_video = output_dir / Path(inp2).name
|
||||
shutil.copy2(inp2, copied_video)
|
||||
|
||||
final_output = output_dir / "final_with_audio.mkv"
|
||||
final_output = output_dir / f"{Path(inp2).stem}_ger.mkv"
|
||||
|
||||
run([
|
||||
"mkvmerge",
|
||||
"-o", str(final_output),
|
||||
str(copied_video),
|
||||
str(inp2),
|
||||
"--language", "0:deu", # assign German language
|
||||
"--track-name", "0:German (2.0 aac)",
|
||||
str(audio_file)
|
||||
])
|
||||
|
||||
return final_output
|
||||
|
||||
def main():
|
||||
inp = f"/home/honney/projects/mkv/AnoHana The Flower We Saw That Day (2011-2011) [imdbid-tt1913273]/Season 01/AnoHana The Flower We Saw That Day S01E001.mkv"
|
||||
inp2 = f"/home/honney/projects/mkv/Anohana - The Flower We Saw That Day - S01 (BD 1080p HEVC Opus) [derpie]/[derpie] Anohana - The Flower We Saw That Day - S01E01 (BD 1080p HEVC Opus) [Dual Audio].mkv"
|
||||
|
||||
|
||||
outp = "./output.mkv"
|
||||
n1 = 24
|
||||
for i in range(1, 12):
|
||||
# --- Inputs ---
|
||||
inp = f"/home/honney/projects/mkv/AnoHana The Flower We Saw That Day (2011-2011) [imdbid-tt1913273]/Season 01/AnoHana The Flower We Saw That Day S01E0{i:02}.mkv"
|
||||
inp2 = f"/home/honney/projects/mkv/Anohana - The Flower We Saw That Day/S01/AnoHana The Flower We Saw That Day S01E0{i:02}.mkv"
|
||||
n1 = find_offset(inp, inp2) # first frame to keep
|
||||
output_dir = "./output_dir"
|
||||
|
||||
fps, _, _, vstart = probe_video(inp)
|
||||
_, duration, _, _ = probe_video(inp2)
|
||||
# --- Probe video ---
|
||||
fps, _, _, vstart = probe_video(inp)
|
||||
_, duration2, _, _ = probe_video(inp2)
|
||||
|
||||
start = vstart + (n1 / fps)
|
||||
end = duration + start
|
||||
# --- Compute timestamps ---
|
||||
start = vstart + (n1 / fps)
|
||||
end = start + duration2
|
||||
|
||||
if end <= start:
|
||||
raise ValueError("n2 must be greater than n1")
|
||||
if end <= start:
|
||||
raise ValueError("n2 must be greater than n1")
|
||||
|
||||
print(f"fps: {fps}")
|
||||
print(f"cutting from frame {n1} to {end*fps}")
|
||||
print(f"time range: {start:.6f}s -> {end:.6f}s")
|
||||
print(f"fps: {fps}")
|
||||
print(f"Cutting audio from frame {n1} (timestamp {start:.6f}s) to timestamp {end:.6f}s")
|
||||
|
||||
print("Trying stream copy...")
|
||||
vaapi_cut(inp, outp, start, end)
|
||||
# --- Trim only audio ---
|
||||
audio_file = Path(output_dir) / f"{Path(inp2).stem}_german_audio.aac"
|
||||
trim_audio(inp, start, end, audio_file)
|
||||
|
||||
output_dir = "./output_dir"
|
||||
audio_file = Path(output_dir) / "german_audio.aac"
|
||||
# --- Mux trimmed audio ---
|
||||
final_file = mux_audio_into_output(inp2, audio_file, output_dir)
|
||||
|
||||
extract_audio(outp, audio_file)
|
||||
|
||||
final_file = mux_audio_into_copy(inp2, audio_file, output_dir)
|
||||
|
||||
print(f"Final file created at: {final_file}")
|
||||
print(f"Final file with trimmed audio created at: {final_file}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
main()
|
||||
Reference in new issue
Block a user