mirror of
https://gitea.pjck.top/Cookies/CookiesChartConverter.git
synced 2025-12-14 12:56:54 +08:00
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
import os
|
|
import subprocess
|
|
from pathlib import Path
|
|
from pv_convert import process_video
|
|
|
|
# === 配置 ===
|
|
WANNACRI_PATH = "wannacri" # 需在 PATH 或填写完整路径
|
|
FFMPEG_PATH = "ffmpeg" # 需在 PATH 或填写完整路径
|
|
|
|
def extract_usm(usm_path: Path, output_dir: Path):
|
|
""" 提取 usm 文件内容 """
|
|
subprocess.run([WANNACRI_PATH, "extractusm", str(usm_path),"--key","0x7F4551499DF55E68"], cwd=output_dir, check=True)
|
|
|
|
def convert_ivf_to_mp4(ivf_path: Path, output_mp4_path: Path):
|
|
""" 将 ivf 文件转换为 mp4 """
|
|
subprocess.run([
|
|
FFMPEG_PATH,
|
|
"-y", "-i", str(ivf_path),
|
|
"-c:v", "copy",
|
|
str(output_mp4_path)
|
|
], check=True)
|
|
|
|
import subprocess
|
|
|
|
def get_video_duration(path: Path) -> float:
|
|
"""使用 ffprobe 获取视频时长（单位：秒）"""
|
|
try:
|
|
result = subprocess.run(
|
|
["ffprobe", "-v", "error", "-show_entries", "format=duration",
|
|
"-of", "default=noprint_wrappers=1:nokey=1", str(path)],
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
text=True
|
|
)
|
|
return float(result.stdout.strip())
|
|
except Exception as e:
|
|
print(f"获取视频时长失败: {e}")
|
|
return 0.0
|
|
|
|
def dat_to_mp4(dat_file: str, id: str):
|
|
""" 将 .dat 文件当作 .usm 文件处理，提取并转换为 .mp4 """
|
|
dat_path = Path(dat_file).resolve()
|
|
base_name = dat_path.stem
|
|
work_dir = Path("/Users/bennett/PJCK/CookiesChartConverter") / "work" / id
|
|
|
|
usm_path = Path(dat_file)
|
|
ivf_path = work_dir / "output" / f"{base_name}.dat" / "videos" / f"{base_name}.ivf"
|
|
mp4_path = work_dir / f"{base_name}.mp4"
|
|
|
|
# 直接将 .dat 当作 .usm 文件处理
|
|
print(f"[1/3] 提取 USM 内容 ...")
|
|
extract_usm(usm_path, work_dir)
|
|
|
|
if not ivf_path.exists():
|
|
print(f"❌ 提取失败，未找到 {ivf_path.name}")
|
|
return
|
|
|
|
print(f"[2/3] 转换为 MP4 ...")
|
|
convert_ivf_to_mp4(ivf_path, mp4_path)
|
|
|
|
duration = get_video_duration(mp4_path)
|
|
if duration < 1.0:
|
|
print(f"⚠️ 视频时长 {duration:.2f}s 太短，跳过生成 pv.mp4")
|
|
return None
|
|
|
|
print(f"[3/3] 成功生成：{mp4_path}")
|
|
process_video(mp4_path, work_dir / "pv.mp4")
|
|
return mp4_path
|
|
|
|
|
|
# === 示例用法 ===
|
|
if __name__ == "__main__":
|
|
import sys
|
|
if len(sys.argv) != 2:
|
|
print("用法: python dat_to_mp4.py <xxx.dat>")
|
|
exit(1)
|
|
|
|
dat_file = sys.argv[1]
|
|
mp4_output = dat_to_mp4(dat_file)
|
|
|
|
if mp4_output:
|
|
converted_mp4_path = f"pv.mp4"
|
|
print(f"[4/4] 开始转换为新的 MP4 文件：{converted_mp4_path}")
|
|
|
|
|