SHA256
421 lines
15 KiB
C#
421 lines
15 KiB
C#
using System;
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Threading;
|
|
using ADOFAIRenderer.Util;
|
|
using MelonLoader;
|
|
using MelonLoader.Utils;
|
|
using UnityEngine;
|
|
|
|
namespace ADOFAIRenderer.Component
|
|
{
|
|
public class RenderManager : MonoBehaviour
|
|
{
|
|
public static int Countdown;
|
|
public static int FrameCount;
|
|
public static bool SkipPressToStart;
|
|
public static string LevelPath = "";
|
|
public static RenderManager Instance = null!;
|
|
private static int _originalInputOffset;
|
|
private static GameObject? _debugMonitorObj;
|
|
private static GameObject? _escapeHandlerObj;
|
|
private static string? _currentOutPath;
|
|
|
|
public static RenderSettings Configuration;
|
|
|
|
public struct RenderSettings
|
|
{
|
|
public double TargetFPS;
|
|
public double VideoFPS;
|
|
|
|
public int RenderWidth;
|
|
public int RenderHeight;
|
|
|
|
public int Bitrate;
|
|
public string Codec;
|
|
public double EndWaitTime;
|
|
|
|
public RenderSettings(double targetFPS, double videoFPS, int renderWidth, int renderHeight, int bitrate, string codec, double endWaitTime)
|
|
{
|
|
TargetFPS = targetFPS;
|
|
VideoFPS = videoFPS;
|
|
|
|
RenderWidth = renderWidth;
|
|
RenderHeight = renderHeight;
|
|
|
|
Bitrate = bitrate;
|
|
Codec = codec;
|
|
EndWaitTime = endWaitTime;
|
|
}
|
|
}
|
|
|
|
public static SortedList<int, byte[]> ImageQueue = new SortedList<int, byte[]>();
|
|
public static Thread? RenderThread;
|
|
public static int PassCount;
|
|
public static bool EndRequested;
|
|
public static bool EndSuccessful;
|
|
public static int WonFrame = -1;
|
|
|
|
private static bool _renderSetup;
|
|
|
|
private void Awake()
|
|
{
|
|
Instance = this;
|
|
}
|
|
|
|
private void Update()
|
|
{
|
|
if (_renderSetup && ADOBase.lm)
|
|
{
|
|
ADOBase.lm.hideDifficultyUI = true;
|
|
}
|
|
|
|
if (_renderSetup && scnGame.instance && !scnGame.instance.isLoading)
|
|
{
|
|
_renderSetup = false;
|
|
scrCamera.instance.forceRTCam = true;
|
|
scrCamera.instance.SetupRTCam(true);
|
|
|
|
if (!scrCamera.instance.gameObject.GetComponent<PostRenderHook>())
|
|
scrCamera.instance.gameObject.AddComponent<PostRenderHook>();
|
|
|
|
_originalInputOffset = scrConductor.currentPreset.inputOffset;
|
|
scrConductor.currentPreset.inputOffset = 0;
|
|
|
|
RDC.auto = true;
|
|
if (scrController.instance)
|
|
{
|
|
scrController.instance.noFail = true;
|
|
scrController.instance.noFailInfiniteMargin = true;
|
|
}
|
|
|
|
RenderMod.TimeBase = Time.timeAsDouble;
|
|
RenderMod.UnscaledTimeBase = Time.unscaledTimeAsDouble;
|
|
RenderMod.DspTimeBase = AudioSettings.dspTime;
|
|
RenderMod.FrameCountBase = Time.frameCount;
|
|
|
|
RenderMod.IsRendering = true;
|
|
Countdown = 3; WonFrame = -1;
|
|
|
|
if (!_escapeHandlerObj)
|
|
{
|
|
_escapeHandlerObj = new GameObject("EscapeHandlerCanvas");
|
|
_escapeHandlerObj.AddComponent<EscapeHandlerUI>();
|
|
DontDestroyOnLoad(_escapeHandlerObj);
|
|
}
|
|
|
|
scrController.instance!.StartCoroutine(SetupPlay());
|
|
}
|
|
|
|
if (!RenderMod.IsRendering) return;
|
|
if (!scrController.instance || scrController.instance.state == States.Fail || scrController.instance.state == States.Fail2)
|
|
{
|
|
RenderMod.IsRendering = RenderMod.IsCapturing = false;
|
|
StopFFmpeg(false);
|
|
EndRender();
|
|
}
|
|
else if (scrController.instance.state == States.Won)
|
|
{
|
|
if (WonFrame == -1)
|
|
{
|
|
WonFrame = FrameCount;
|
|
}
|
|
|
|
var extraFrames = (int)Math.Ceiling(Configuration.EndWaitTime * Configuration.TargetFPS);
|
|
|
|
if (FrameCount - WonFrame < extraFrames) return;
|
|
RenderMod.IsRendering = RenderMod.IsCapturing = false;
|
|
StopFFmpeg(true);
|
|
}
|
|
}
|
|
|
|
public void SetupCaptureDirectory()
|
|
{
|
|
var renderDir = Path.Combine(MelonEnvironment.GameRootDirectory, "Render");
|
|
if (!Directory.Exists(renderDir))
|
|
{
|
|
Directory.CreateDirectory(renderDir);
|
|
}
|
|
}
|
|
|
|
public void SetupGameScene()
|
|
{
|
|
GCS.customLevelPaths = new[] { LevelPath };
|
|
GCS.checkpointNum = 0;
|
|
GCS.customLevelId = "";
|
|
GCS.nextSpeedRun = 1;
|
|
GCS.customLevelIndex = 0;
|
|
|
|
ADOBase.loader.LoadSceneWithTransition(WipeDirection.StartsFromRight, "scnGame");
|
|
SkipPressToStart = true;
|
|
_renderSetup = true;
|
|
}
|
|
|
|
private IEnumerator SetupPlay()
|
|
{
|
|
for (; Countdown > 0; Countdown--)
|
|
{
|
|
yield return new WaitForSecondsRealtime(1);
|
|
}
|
|
|
|
FrameCount = 0;
|
|
|
|
StartFFmpegThread();
|
|
|
|
RenderMod.IsCapturing = true;
|
|
scrController.instance.paused = scrController.instance.audioPaused = false;
|
|
scrController.instance.levelWasSkipped = true;
|
|
|
|
scrConductor.instance.song.clip = null;
|
|
scrConductor.instance.song.time = -1000;
|
|
}
|
|
|
|
private static string FindFFmpegPath()
|
|
{
|
|
var isWindows = Application.platform == RuntimePlatform.WindowsPlayer;
|
|
var ffmpegName = isWindows ? "ffmpeg.exe" : "ffmpeg";
|
|
|
|
// 1. Mods/ADOFAIRenderer/ffmpeg
|
|
var path1 = Path.Combine(MelonEnvironment.ModsDirectory, "ADOFAIRenderer", ffmpegName);
|
|
if (File.Exists(path1)) return path1;
|
|
|
|
// 2. Mods/ffmpeg
|
|
var path2 = Path.Combine(MelonEnvironment.ModsDirectory, ffmpegName);
|
|
if (File.Exists(path2)) return path2;
|
|
|
|
// 3. GameRootDirectory/ffmpeg
|
|
var path3 = Path.Combine(MelonEnvironment.GameRootDirectory, ffmpegName);
|
|
return File.Exists(path3) ? path3 : "";
|
|
}
|
|
|
|
private void StartFFmpegThread()
|
|
{
|
|
var ffmpegPath = FindFFmpegPath();
|
|
if (string.IsNullOrEmpty(ffmpegPath))
|
|
{
|
|
Melon<RenderMod>.Logger.Error("ffmpeg 바이너리를 찾을 수 없습니다. Mods/ADOFAIRenderer/ 에 ffmpeg를 넣어주세요.");
|
|
return;
|
|
}
|
|
|
|
if (!_debugMonitorObj)
|
|
{
|
|
_debugMonitorObj = new GameObject("DebugMonitorCanvas");
|
|
_debugMonitorObj.AddComponent<DebugMonitorUI>();
|
|
DontDestroyOnLoad(_debugMonitorObj);
|
|
}
|
|
|
|
FFmpegProcess.FFmpegPath = ffmpegPath;
|
|
lock (ImageQueue)
|
|
{
|
|
ImageQueue.Clear();
|
|
}
|
|
PassCount = 0;
|
|
RenderMod.FrameCountForVideo = 0;
|
|
RenderMod.RenderCheck = true;
|
|
RenderMod.SkipCount = 0;
|
|
EndRequested = false;
|
|
EndSuccessful = false;
|
|
|
|
var targetFps = Configuration.TargetFPS;
|
|
var videoFps = Configuration.VideoFPS;
|
|
float bitrate = Configuration.Bitrate;
|
|
var codecStr = Configuration.Codec;
|
|
|
|
var selectedCodec = VideoFFmpeg.Codec.H264_Default;
|
|
if (Enum.TryParse<VideoFFmpeg.Codec>(codecStr, true, out var parsedCodec))
|
|
{
|
|
selectedCodec = parsedCodec;
|
|
}
|
|
|
|
var ext = ".mp4";
|
|
if (selectedCodec == VideoFFmpeg.Codec.AV1_Default ||
|
|
selectedCodec == VideoFFmpeg.Codec.AV1_NVIDIA ||
|
|
selectedCodec == VideoFFmpeg.Codec.AV1_QSV ||
|
|
selectedCodec == VideoFFmpeg.Codec.VP9_Default ||
|
|
selectedCodec == VideoFFmpeg.Codec.VP9_NVIDIA)
|
|
{
|
|
ext = ".webm";
|
|
}
|
|
|
|
var startTime = DateTime.Now;
|
|
var fileName = startTime.ToString("yyyy-MM-dd HH-mm-ss") + ext;
|
|
_currentOutPath = Path.Combine(MelonEnvironment.GameRootDirectory, "Render", fileName);
|
|
|
|
RenderThread = new Thread(() =>
|
|
{
|
|
try
|
|
{
|
|
var quotient = targetFps / videoFps;
|
|
var isMultiple = Math.Abs(quotient - Math.Round(quotient)) < 1e-9;
|
|
var workingFps = isMultiple ? videoFps : targetFps;
|
|
var videoFFmpeg = new VideoFFmpeg(Configuration.RenderWidth, Configuration.RenderHeight, workingFps, videoFps, selectedCodec);
|
|
|
|
SetupCaptureDirectory();
|
|
|
|
videoFFmpeg.Start(_currentOutPath, bitrate);
|
|
|
|
var num = isMultiple ? (int)Math.Round(quotient) : 1;
|
|
|
|
while (true)
|
|
{
|
|
byte[]? buffer = null;
|
|
lock (ImageQueue)
|
|
{
|
|
if (ImageQueue.Count > 0)
|
|
{
|
|
int firstKey = ImageQueue.Keys[0];
|
|
if (firstKey == PassCount)
|
|
{
|
|
buffer = ImageQueue[firstKey];
|
|
ImageQueue.RemoveAt(0);
|
|
PassCount += num;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (buffer != null)
|
|
{
|
|
if (buffer.Length > 0)
|
|
{
|
|
var bufferSize = Configuration.RenderWidth * Configuration.RenderHeight * 4;
|
|
videoFFmpeg.SendDataFromBuffer(buffer, bufferSize);
|
|
System.Buffers.ArrayPool<byte>.Shared.Return(buffer);
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (EndRequested)
|
|
{
|
|
bool hasMore;
|
|
lock (ImageQueue)
|
|
{
|
|
hasMore = ImageQueue.Count > 0;
|
|
}
|
|
if (!hasMore) break;
|
|
}
|
|
|
|
Thread.Sleep(1);
|
|
}
|
|
|
|
videoFFmpeg.Close();
|
|
EndSuccessful = true;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Melon<RenderMod>.Logger.Error($"FFmpeg 렌더링 스레드 오류: {ex}");
|
|
}
|
|
});
|
|
|
|
RenderThread.Start();
|
|
}
|
|
|
|
public void CancelRender()
|
|
{
|
|
RenderMod.IsRendering = RenderMod.IsCapturing = false;
|
|
StopFFmpeg(false);
|
|
EndRender();
|
|
}
|
|
|
|
private void StopFFmpeg(bool success)
|
|
{
|
|
EndRequested = true;
|
|
SkipPressToStart = false;
|
|
_renderSetup = false;
|
|
if (_escapeHandlerObj)
|
|
{
|
|
Destroy(_escapeHandlerObj);
|
|
_escapeHandlerObj = null;
|
|
}
|
|
if (success)
|
|
{
|
|
StartCoroutine(StopRenderCoroutine());
|
|
}
|
|
else
|
|
{
|
|
Melon<RenderMod>.Logger.Error("렌더링이 실패했거나 취소되었습니다. 상태 및 리소스를 안전하게 청소합니다.");
|
|
|
|
// 1. 메모리 누수 방지: 큐에 렌트된 임시 프레임 버퍼 풀 반환 및 클리어
|
|
lock (ImageQueue)
|
|
{
|
|
foreach (var kvp in ImageQueue)
|
|
{
|
|
if (kvp.Value != null && kvp.Value.Length > 0)
|
|
{
|
|
System.Buffers.ArrayPool<byte>.Shared.Return(kvp.Value);
|
|
}
|
|
}
|
|
ImageQueue.Clear();
|
|
}
|
|
|
|
// 디버그 모니터 UI 정리
|
|
if (_debugMonitorObj)
|
|
{
|
|
Destroy(_debugMonitorObj);
|
|
_debugMonitorObj = null;
|
|
}
|
|
|
|
// 2. 렌더링 스레드가 큐 비어 있음을 감지하고 break 하도록 대기
|
|
if (RenderThread is { IsAlive: true })
|
|
{
|
|
RenderThread.Join(2000);
|
|
}
|
|
|
|
// 3. 인게임 카메라 및 렌더링용 RTCam 상태 안전 원복
|
|
if (scrCamera.instance)
|
|
{
|
|
scrCamera.instance.forceRTCam = false;
|
|
scrCamera.instance.SetupRTCam(false);
|
|
|
|
var hook = scrCamera.instance.GetComponent<PostRenderHook>();
|
|
if (hook) Destroy(hook);
|
|
}
|
|
|
|
// 4. 인게임 실패 방지 플래그 원상 복구
|
|
if (scrController.instance)
|
|
{
|
|
scrController.instance.noFail = false;
|
|
scrController.instance.noFailInfiniteMargin = false;
|
|
}
|
|
|
|
// 5. 취소된 미완성 임시 동영상 파일 즉각 삭제
|
|
if (!string.IsNullOrEmpty(_currentOutPath) && File.Exists(_currentOutPath))
|
|
{
|
|
try
|
|
{
|
|
File.Delete(_currentOutPath);
|
|
Melon<RenderMod>.Logger.Msg($"[File Cleanup] 미완성 동영상 삭제 완료: {_currentOutPath}");
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Melon<RenderMod>.Logger.Error($"[File Cleanup] 임시 동영상 삭제 실패: {ex.Message}");
|
|
}
|
|
_currentOutPath = null;
|
|
}
|
|
}
|
|
}
|
|
|
|
private static IEnumerator StopRenderCoroutine()
|
|
{
|
|
yield return new WaitUntil(() => EndSuccessful);
|
|
EndRender();
|
|
}
|
|
|
|
private static void EndRender()
|
|
{
|
|
Time.timeScale = 1f;
|
|
if (_debugMonitorObj)
|
|
{
|
|
Destroy(_debugMonitorObj);
|
|
_debugMonitorObj = null;
|
|
}
|
|
|
|
RDC.auto = false;
|
|
scrConductor.currentPreset.inputOffset = _originalInputOffset;
|
|
GCS.customLevelPaths = null;
|
|
GCS.customLevelId = null;
|
|
|
|
ADOBase.loader.LoadSceneWithTransition(WipeDirection.StartsFromRight, "scnEditor");
|
|
}
|
|
}
|
|
} |