アニメエンコードの研究(笑)

import subprocess
from pathlib import Path
import tkinter as tk
from tkinter import filedialog, messagebox

EXTS = {
    ".mkv",
    ".mp4",
    ".m2ts"
}

def pick_input_dir() -> Path | None:
    root = tk.Tk()
    root.withdraw()
    root.attributes("-topmost", True)
    folder = filedialog.askdirectory(title="入力フォルダ")
    root.destroy()
    return Path(folder) if folder else None

def process_videos(input_dir: Path) -> int:
    output_dir = input_dir / "out"
    output_dir.mkdir(exist_ok=True)

    count = 0
    for path in input_dir.iterdir():
        if not path.is_file():
            continue
        if path.suffix.lower() not in EXTS:
            continue
        output_path = output_dir / path.name
        cmd = [
            "nvencc",
            "-i", str(path),
            "--avhw",
            "-c", "hevc",
            "--qvbr", "24",
            "-u", "P6",
            "--output-depth", "10",
            "--lookahead", "32",
            "-b", "4", "--bref-mode", "each",
            "--aq", "--aq-temporal",
            "--audio-codec", "libopus",
            "--audio-bitrate", "128",
            "-o", str(output_path)
        ]

        print("Running"," ".join(cmd))
        result = subprocess.run(cmd)
        if result.returncode != 0:
            print(f"Failed: {path.name}")
            continue
        count += 1
        print(f"Done: {path.name} -> {output_path}")

    return count

def main() -> None:
    input_dir = pick_input_dir()
    if input_dir is None:
        return

    if not input_dir.exists() or not input_dir.is_dir():
        messagebox.showerror("エラー", "有効なフォルダが選択されませんでした。")
        return

    processed = process_videos(input_dir=input_dir)
    messagebox.showinfo("完了",f"{processed}個の動画を処理しました")

if __name__ == "__main__":
    main()