SHA256
284 lines
12 KiB
C#
284 lines
12 KiB
C#
using System.Linq;
|
|
using ADOFAIRenderer.Core;
|
|
using ADOFAIRenderer.Util.FFmpeg;
|
|
using MelonLoader;
|
|
using TMPro;
|
|
using UnityEngine;
|
|
using UnityEngine.UI;
|
|
|
|
namespace ADOFAIRenderer.UI
|
|
{
|
|
public class RenderButton : MonoBehaviour
|
|
{
|
|
[Header("Input Fields")]
|
|
[SerializeField] private TMP_InputField targetFPSInput = null!;
|
|
[SerializeField] private TMP_InputField videoFPSInput = null!;
|
|
[SerializeField] private TMP_InputField renderWidthInput = null!;
|
|
[SerializeField] private TMP_InputField renderHeightInput = null!;
|
|
[SerializeField] private TMP_InputField waitAfterEndInput = null!;
|
|
[SerializeField] private TMP_InputField randomSeedInput = null!;
|
|
|
|
[Header("Dropdown")]
|
|
[SerializeField] private TMP_Dropdown codecDropdown = null!;
|
|
[SerializeField] private TMP_Dropdown bitrateModeDropdown = null!;
|
|
|
|
[Header("Toggle")]
|
|
[SerializeField] private Toggle showHitTextToggle = null!;
|
|
|
|
[Header("Button")]
|
|
[SerializeField] private Button advancedSettingsButton = null!;
|
|
|
|
[Header("Bitrate Dynamic UI Container")]
|
|
[SerializeField] private RectTransform bitrateContainer = null!;
|
|
|
|
// Dynamically instantiated inputs
|
|
private TMP_InputField _bitrateInput = null!;
|
|
private TMP_InputField _maxBitrateInput = null!;
|
|
private TMP_InputField _qpInput = null!;
|
|
|
|
private Button _buttonComponent = null!;
|
|
|
|
private void Start()
|
|
{
|
|
_buttonComponent = GetComponent<Button>();
|
|
_buttonComponent.onClick.AddListener(OnClick);
|
|
advancedSettingsButton.onClick.AddListener(AdvancedSettingsClick);
|
|
|
|
RenderMod.FixEventSystem();
|
|
|
|
if (codecDropdown != null)
|
|
{
|
|
codecDropdown.ClearOptions();
|
|
var options = System.Enum.GetNames(typeof(Codec)).ToList();
|
|
codecDropdown.AddOptions(options);
|
|
}
|
|
|
|
// 1. Initialize Bitrate Mode Dropdown
|
|
if (bitrateModeDropdown != null)
|
|
{
|
|
bitrateModeDropdown.ClearOptions();
|
|
var modeOptions = System.Enum.GetNames(typeof(RateControlMode)).ToList();
|
|
bitrateModeDropdown.AddOptions(modeOptions);
|
|
bitrateModeDropdown.onValueChanged.AddListener(delegate { UpdateUIByMode(); });
|
|
}
|
|
|
|
// 2. Dynamically instantiate input fields using targetFPSInput as a styling template
|
|
if (bitrateContainer != null && targetFPSInput != null)
|
|
{
|
|
// Instantiate bitrateInput
|
|
var bitrateInputObj = Instantiate(targetFPSInput.gameObject, bitrateContainer);
|
|
bitrateInputObj.name = "Dynamic_BitrateInput";
|
|
_bitrateInput = bitrateInputObj.GetComponent<TMP_InputField>();
|
|
_bitrateInput.text = "15000";
|
|
SetPlaceholderText(_bitrateInput, "비트레이트 (kbps)");
|
|
|
|
// Instantiate maxBitrateInput
|
|
var maxBitrateInputObj = Instantiate(targetFPSInput.gameObject, bitrateContainer);
|
|
maxBitrateInputObj.name = "Dynamic_MaxBitrateInput";
|
|
_maxBitrateInput = maxBitrateInputObj.GetComponent<TMP_InputField>();
|
|
_maxBitrateInput.text = "30000";
|
|
SetPlaceholderText(_maxBitrateInput, "최대 비트레이트 (kbps)");
|
|
|
|
// Instantiate qpInput
|
|
var qpInputObj = Instantiate(targetFPSInput.gameObject, bitrateContainer);
|
|
qpInputObj.name = "Dynamic_QPInput";
|
|
_qpInput = qpInputObj.GetComponent<TMP_InputField>();
|
|
_qpInput.text = "18";
|
|
SetPlaceholderText(_qpInput, "QP (0-51) / CRF");
|
|
}
|
|
|
|
// 3. Perform initial UI state updates based on default mode
|
|
UpdateUIByMode();
|
|
}
|
|
|
|
private void SetPlaceholderText(TMP_InputField inputField, string text)
|
|
{
|
|
if (inputField.placeholder == null) return;
|
|
var tmpText = inputField.placeholder.GetComponent<TMP_Text>();
|
|
|
|
if (tmpText)
|
|
{
|
|
tmpText.text = text;
|
|
}
|
|
}
|
|
|
|
private void UpdateUIByMode()
|
|
{
|
|
if (bitrateModeDropdown == null) return;
|
|
if (_bitrateInput == null || _maxBitrateInput == null || _qpInput == null) return;
|
|
|
|
var selectedMode = (RateControlMode)bitrateModeDropdown.value;
|
|
|
|
switch (selectedMode)
|
|
{
|
|
case RateControlMode.CBR:
|
|
_bitrateInput.gameObject.SetActive(true);
|
|
_maxBitrateInput.gameObject.SetActive(false);
|
|
_qpInput.gameObject.SetActive(false);
|
|
break;
|
|
case RateControlMode.VBR:
|
|
_bitrateInput.gameObject.SetActive(true);
|
|
_maxBitrateInput.gameObject.SetActive(true);
|
|
_qpInput.gameObject.SetActive(false);
|
|
break;
|
|
case RateControlMode.CQP:
|
|
_bitrateInput.gameObject.SetActive(false);
|
|
_maxBitrateInput.gameObject.SetActive(false);
|
|
_qpInput.gameObject.SetActive(true);
|
|
break;
|
|
}
|
|
|
|
if (bitrateContainer != null)
|
|
{
|
|
Canvas.ForceUpdateCanvases();
|
|
LayoutRebuilder.ForceRebuildLayoutImmediate(bitrateContainer);
|
|
}
|
|
}
|
|
|
|
private void Update()
|
|
{
|
|
if (Input.GetKeyDown(KeyCode.Escape))
|
|
{
|
|
ADOBase.loader.LoadSceneWithTransition(WipeDirection.StartsFromRight, "scnEditor");
|
|
}
|
|
}
|
|
|
|
public void OnClick()
|
|
{
|
|
if (IsAnyFieldEmpty()) return;
|
|
if (RenderManager.IsActive) return;
|
|
|
|
if (!double.TryParse(targetFPSInput.text, out var targetFPS) || targetFPS <= 0)
|
|
{
|
|
Melon<RenderMod>.Logger.Error("올바르지 않은 Target FPS 값입니다. 0보다 큰 숫자여야 합니다.");
|
|
return;
|
|
}
|
|
if (!double.TryParse(videoFPSInput.text, out var videoFPS) || videoFPS <= 0)
|
|
{
|
|
Melon<RenderMod>.Logger.Error("올바르지 않은 Video FPS 값입니다. 0보다 큰 숫자여야 합니다.");
|
|
return;
|
|
}
|
|
if (!int.TryParse(renderWidthInput.text, out var renderWidth) || renderWidth <= 0 || renderWidth > SystemInfo.maxTextureSize)
|
|
{
|
|
Melon<RenderMod>.Logger.Error($"올바르지 않은 Render Width 값입니다. 1에서 {SystemInfo.maxTextureSize} 사이의 정수여야 합니다.");
|
|
return;
|
|
}
|
|
if (!int.TryParse(renderHeightInput.text, out var renderHeight) || renderHeight <= 0 || renderHeight > SystemInfo.maxTextureSize)
|
|
{
|
|
Melon<RenderMod>.Logger.Error($"올바르지 않은 Render Height 값입니다. 1에서 {SystemInfo.maxTextureSize} 사이의 정수여야 합니다.");
|
|
return;
|
|
}
|
|
|
|
var bitrateMode = RateControlMode.CBR;
|
|
if (bitrateModeDropdown != null)
|
|
{
|
|
bitrateMode = (RateControlMode)bitrateModeDropdown.value;
|
|
}
|
|
|
|
var bitrate = 0;
|
|
var maxBitrate = 0;
|
|
var qp = 18;
|
|
|
|
if (bitrateMode == RateControlMode.CBR || bitrateMode == RateControlMode.VBR)
|
|
{
|
|
if (!int.TryParse(_bitrateInput.text, out bitrate) || bitrate <= 0 || bitrate > 1_000_000)
|
|
{
|
|
Melon<RenderMod>.Logger.Error("올바르지 않은 Bitrate 값입니다. 1에서 1,000,000 (kbps) 사이의 정수여야 합니다.");
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (bitrateMode == RateControlMode.VBR)
|
|
{
|
|
if (!int.TryParse(_maxBitrateInput.text, out maxBitrate) || maxBitrate <= 0 || maxBitrate > 1_000_000)
|
|
{
|
|
Melon<RenderMod>.Logger.Error("올바르지 않은 Max Bitrate 값입니다. 1에서 1,000,000 (kbps) 사이의 정수여야 합니다.");
|
|
return;
|
|
}
|
|
if (maxBitrate < bitrate)
|
|
{
|
|
Melon<RenderMod>.Logger.Error("Max Bitrate는 Target Bitrate보다 크거나 같아야 합니다.");
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (bitrateMode == RateControlMode.CQP)
|
|
{
|
|
if (!int.TryParse(_qpInput.text, out qp) || qp < 0 || qp > 51)
|
|
{
|
|
Melon<RenderMod>.Logger.Error("올바르지 않은 QP 값입니다. 0에서 51 사이의 정수여야 합니다.");
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (!double.TryParse(waitAfterEndInput.text, out var waitAfterEnd) || waitAfterEnd < 0)
|
|
{
|
|
Melon<RenderMod>.Logger.Error("올바르지 않은 Wait After End 값입니다. 0 이상의 숫자여야 합니다.");
|
|
return;
|
|
}
|
|
|
|
int? parsedRandomSeed = null;
|
|
var randomSeedText = randomSeedInput.text.Truncate(int.MaxValue);
|
|
if (!randomSeedText.IsNullOrEmpty())
|
|
{
|
|
if (!int.TryParse(randomSeedText, out var seedVal))
|
|
{
|
|
Melon<RenderMod>.Logger.Error("올바르지 않은 Random Seed 값입니다. 비어 있거나 정수여야 합니다.");
|
|
return;
|
|
}
|
|
parsedRandomSeed = seedVal;
|
|
}
|
|
|
|
if (RenderManager.Instance)
|
|
{
|
|
Melon<RenderMod>.Logger.Msg($"레벨 로드: {RenderManager.Instance.levelPath}");
|
|
}
|
|
|
|
var selectedCodecIndex = codecDropdown.value;
|
|
var selectedCodecName = codecDropdown.options[selectedCodecIndex].text;
|
|
|
|
var showHitText = showHitTextToggle != null && showHitTextToggle.isOn;
|
|
|
|
if (RenderManager.Instance)
|
|
{
|
|
RenderManager.Instance.Configuration = new RenderManager.RenderSettings(
|
|
targetFPS,
|
|
videoFPS,
|
|
renderWidth,
|
|
renderHeight,
|
|
bitrateMode,
|
|
bitrate,
|
|
maxBitrate,
|
|
qp,
|
|
selectedCodecName,
|
|
waitAfterEnd,
|
|
showHitText,
|
|
parsedRandomSeed
|
|
);
|
|
}
|
|
|
|
RenderManager.Instance!.SetupCaptureDirectory();
|
|
RenderManager.Instance.SetupGameScene();
|
|
}
|
|
|
|
public void AdvancedSettingsClick()
|
|
{
|
|
|
|
}
|
|
|
|
private bool IsAnyFieldEmpty()
|
|
{
|
|
if (string.IsNullOrWhiteSpace(targetFPSInput.text)) return true;
|
|
if (string.IsNullOrWhiteSpace(videoFPSInput.text)) return true;
|
|
if (string.IsNullOrWhiteSpace(renderWidthInput.text)) return true;
|
|
if (string.IsNullOrWhiteSpace(renderHeightInput.text)) return true;
|
|
if (string.IsNullOrWhiteSpace(waitAfterEndInput.text)) return true;
|
|
|
|
if (_bitrateInput != null && _bitrateInput.gameObject.activeSelf && string.IsNullOrWhiteSpace(_bitrateInput.text)) return true;
|
|
if (_maxBitrateInput != null && _maxBitrateInput.gameObject.activeSelf && string.IsNullOrWhiteSpace(_maxBitrateInput.text)) return true;
|
|
if (_qpInput != null && _qpInput.gameObject.activeSelf && string.IsNullOrWhiteSpace(_qpInput.text)) return true;
|
|
|
|
return codecDropdown.value < 0 || (bitrateModeDropdown != null && bitrateModeDropdown.value < 0);
|
|
}
|
|
}
|
|
} |