import subprocess from pathlib import Path import os from utils.helpers import get_media_info, get_media_duration # --- 標準化設定 --- TARGET_WIDTH = 1920 TARGET_HEIGHT = 1080 TARGET_FPS = 30 def escape_ffmpeg_path_for_filter(path: Path) -> str: """為在 FFmpeg 濾鏡圖中安全使用而轉義路徑。""" path_str = str(path.resolve()) if os.name == 'nt': return path_str.replace('\\', '\\\\').replace(':', '\\:') else: return path_str.replace("'", "'\\\\\\''") def run_step6_assemble_video( project_path: Path, logo_video: Path, open_video: Path, end_video: Path, p_loop_count: int, transition_duration: float = 1.0 ): """ 使用 FFmpeg 的 xfade 濾鏡組裝最終影片，並在過程中統一影片屬性與進行色彩校正。 """ try: # --- 1. 路徑與檔案檢查 --- output_dir = project_path / "output" test_dir = output_dir / "test" audio_path = output_dir / "combined_audio.wav" ass_path = output_dir / "subtitles.ass" bg_final_path = output_dir / "bg_final.mp4" final_video_path = output_dir / "final_video_full_sequence.mp4" for file_path in [logo_video, open_video, end_video, audio_path, ass_path]: if not file_path or not file_path.exists(): return False, f"❌ 找不到必要的輸入檔案: {file_path}", None base_videos = sorted([p for p in test_dir.iterdir() if p.is_file() and p.suffix.lower() in ['.mp4', '.mov']]) if not base_videos: return False, "❌ `test` 資料夾中沒有影片可供合成。", None # --- 2. 一步式生成主要內容影片 (bg_final.mp4) --- if not bg_final_path.exists(): print(f"⚙️ 準備一步式生成主要內容影片 (循環 {p_loop_count} 次)...") looped_video_list = base_videos * p_loop_count inputs_cmd = [] for video_path in looped_video_list: inputs_cmd.extend(["-i", str(video_path)]) inputs_cmd.extend(["-i", str(audio_path)]) ass_path_str = escape_ffmpeg_path_for_filter(ass_path) filter_parts = [] stream_count = len(looped_video_list) # 【核心修改】定義色彩校正濾鏡 color_correction_filter = "eq=brightness=-0.05:contrast=0.95:saturation=0.7" # 將標準化 (尺寸、影格率) 與色彩校正合併 for i in range(stream_count): filter_parts.append( f"[{i}:v]scale={TARGET_WIDTH}:{TARGET_HEIGHT}:force_original_aspect_ratio=decrease,pad={TARGET_WIDTH}:{TARGET_HEIGHT}:(ow-iw)/2:(oh-ih)/2,setsar=1,fps={TARGET_FPS},{color_correction_filter}[v_scaled_{i}]" ) last_v_out = "[v_scaled_0]" for i in range(stream_count - 1): offset = sum(get_media_duration(v) for v in looped_video_list[:i+1]) - (i + 1) * transition_duration v_in_next = f"[v_scaled_{i+1}]" v_out_current = f"[v_out{i+1}]" if i < stream_count - 2 else "[bg_video_stream]" filter_parts.append(f"{last_v_out}{v_in_next}xfade=transition=fade:duration={transition_duration}:offset={offset:.4f}{v_out_current}") last_v_out = v_out_current drawbox_filter = "drawbox=w=iw*0.8:h=ih*0.8:x=(iw-iw*0.8)/2:y=(ih-ih*0.8)/2:color=0x808080@0.7:t=fill" filter_parts.append(f"[bg_video_stream]{drawbox_filter}[v_with_mask]") filter_parts.append(f"[v_with_mask]ass=filename='{ass_path_str}'[final_v]") filter_complex_str = ";".join(filter_parts) ffmpeg_main_content_cmd = [ "ffmpeg", "-y", *inputs_cmd, "-filter_complex", filter_complex_str, "-map", "[final_v]", "-map", f"{stream_count}:a:0", "-c:v", "libx264", "-pix_fmt", "yuv420p", "-preset", "fast", "-c:a", "aac", "-b:a", "192k", "-shortest", str(bg_final_path) ] print("🚀 正在執行 FFmpeg 主要內容合成指令 (含色彩校正)...") result = subprocess.run(ffmpeg_main_content_cmd, check=True, capture_output=True, text=True, encoding='utf-8') if result.returncode != 0: raise subprocess.CalledProcessError(result.returncode, result.args, stderr=result.stderr) print(f"✅ 主要內容影片已生成: {bg_final_path}") # --- 3. 合成最終的影片序列 (logo -> open -> bg_final -> end) --- if final_video_path.exists(): print(f"✅ 最終影片已存在，跳過組裝: {final_video_path}") return True, f"✅ 影片已存在於 {final_video_path}", final_video_path print("🎬 開始動態生成 xfade 合成指令...") videos_to_concat = [logo_video, open_video, bg_final_path, end_video] final_inputs_cmd, final_filter_parts, video_infos = [], [], [] for video_path in videos_to_concat: final_inputs_cmd.extend(["-i", str(video_path)]) info = get_media_info(video_path) video_infos.append(info) for i, info in enumerate(video_infos): if not info["has_audio"]: duration = info['duration'] final_filter_parts.append(f"anullsrc=r=44100:cl=stereo:d={duration:.4f}[silent_a_{i}]") for i in range(len(videos_to_concat)): # 注意：這裡我們不對 logo/open/end 影片進行調色，以保留它們的原始風格 final_filter_parts.append( f"[{i}:v]scale={TARGET_WIDTH}:{TARGET_HEIGHT}:force_original_aspect_ratio=decrease,pad={TARGET_WIDTH}:{TARGET_HEIGHT}:(ow-iw)/2:(oh-ih)/2,setsar=1,fps={TARGET_FPS}[v_final_scaled_{i}]" ) last_v_out = "[v_final_scaled_0]" last_a_out = "[0:a]" if video_infos[0]["has_audio"] else "[silent_a_0]" for i in range(len(videos_to_concat) - 1): offset = sum(info['duration'] for info in video_infos[:i+1]) - (i + 1) * transition_duration v_in_next = f"[v_final_scaled_{i+1}]" a_in_next = f"[{i+1}:a]" if video_infos[i+1]["has_audio"] else f"[silent_a_{i+1}]" v_out_current = f"[v_out{i+1}]" if i < len(videos_to_concat) - 2 else "[video]" a_out_current = f"[a_out{i+1}]" if i < len(videos_to_concat) - 2 else "[audio]" final_filter_parts.append(f"{last_v_out}{v_in_next}xfade=transition=fade:duration={transition_duration}:offset={offset:.4f}{v_out_current}") final_filter_parts.append(f"{last_a_out}{a_in_next}acrossfade=d={transition_duration}{a_out_current}") last_v_out, last_a_out = v_out_current, a_out_current final_filter_complex_str = ";".join(final_filter_parts) ffmpeg_concat_cmd = [ "ffmpeg", "-y", *final_inputs_cmd, "-filter_complex", final_filter_complex_str, "-map", "[video]", "-map", "[audio]", "-c:v", "libx264", "-pix_fmt", "yuv420p", "-preset", "fast", "-movflags", "+faststart", "-c:a", "aac", "-b:a", "192k", str(final_video_path) ] print("🚀 執行 FFmpeg 最終序列合成指令...") result = subprocess.run(ffmpeg_concat_cmd, check=True, capture_output=True, text=True, encoding='utf-8') if result.returncode != 0: raise subprocess.CalledProcessError(result.returncode, result.args, stderr=result.stderr) print(f"✅ 影片已成功組裝並儲存至 {final_video_path}") return True, f"✅ 影片已成功組裝並儲存至 {final_video_path}", final_video_path except subprocess.CalledProcessError as e: error_output = e.stderr if e.stderr else str(e) return False, f"❌ FFmpeg 執行失敗:\n{error_output}", None except Exception as e: return False, f"❌ 發生未預期錯誤: {e}", None