ADOFAIRenderer/UI/RenderConfigUIManager.cs

using ADOFAIRenderer.Core;
using ADOFAIRenderer.Data;
using ADOFAIRenderer.Util.FFmpeg;
using MelonLoader;
using UnityEngine;
using UnityEngine.UIElements;

namespace ADOFAIRenderer.UI
{
    public class RenderConfigUIManager : MonoBehaviour
    {
        [SerializeField]
        private UIDocument uiDocument = null!;
        
        private VisualElement _root = null!;
        private IntegerInputValidator _intValidator = null!;
        private DoubleInputValidator _doubleValidator = null!;
        private EnumDropdownBinder<Codec> _codecBinder = null!;
        
        private Button _goRenderButton = null!;
        private Button _encodingConfigButton = null!;
        private VisualElement _encodingConfigContainer = null!;
        
        private void Awake()
        {
            _root = uiDocument.rootVisualElement;
            RenderManager.Instance.Configuration = new RenderConfig();
            
            _intValidator = new IntegerInputValidator(_root);
            _doubleValidator = new DoubleInputValidator(_root);
        }

        private void Start()
        {
            // Init Validators
            _intValidator
                .AddLimits("RenderWidth", new PositiveOnlyIntegerLimit(), new RangeIntegerLimit(1, 7680))
                .AddLimits("RenderHeight", new PositiveOnlyIntegerLimit(), new RangeIntegerLimit(1, 4320));

            _doubleValidator
                .AddLimits("RenderFPS", new PositiveOnlyDoubleLimit(), new RangeDoubleLimit(0, 9600))
                .AddLimits("VideoFPS", new PositiveOnlyDoubleLimit(), new RangeDoubleLimit(0, 4800))
                .AddLimits("DelayAfterLevelEnd", new PositiveOnlyDoubleLimit(), new RangeDoubleLimit(0, 86400));
            
            _intValidator.SetupValidation();
            _doubleValidator.SetupValidation();
            
            // Init Enum Binders & Event Listeners
            _codecBinder = new EnumDropdownBinder<Codec>(_root, "CodecDropdown", Codec.H264);
            _codecBinder.OnValueChanged(value => RenderManager.Instance.Configuration.Codec = value);
            
            // Attach Button Event Listener
            _goRenderButton = _root.Q<Button>("GoRenderButton");
            _goRenderButton?.RegisterCallback<ClickEvent>(_ => TryGoRender());

            _encodingConfigButton = _root.Q<Button>("EncodingSettingsButton");
            if (_encodingConfigButton != null)
            {
                Melon<RenderMod>.Logger.Msg($"[UI] EncodingSettingsButton 발견: pickingMode={_encodingConfigButton.pickingMode}, enabledSelf={_encodingConfigButton.enabledSelf}");
                _encodingConfigButton.RegisterCallback<ClickEvent>(_ => OpenEncodingConfig());
            }
            else
            {
                Melon<RenderMod>.Logger.Error("Button 'EncodingSettingsButton'을(를) 찾을 수 없습니다.");
            }
        }

        private void Update()
        {
            if (Input.GetKeyDown(KeyCode.Escape))
            {
                ADOBase.loader.LoadSceneWithTransition(WipeDirection.StartsFromRight, "scnEditor");
            }
        }

        private void OpenEncodingConfig()
        {
            var config = RenderManager.Instance.Configuration;
            config.EncodingConfig ??= GenerateEncodingConfig();
            
            var tempConfig = new EncodingConfig
            {
                RateControlMode = config.EncodingConfig.RateControlMode,
                Bitrate = config.EncodingConfig.Bitrate,
                MaxBitrate = config.EncodingConfig.MaxBitrate,
                QP = config.EncodingConfig.QP,
                KeyframeInterval = config.EncodingConfig.KeyframeInterval,
                ForceSoftwareEncoder = config.EncodingConfig.ForceSoftwareEncoder
            };

            if (_root.Contains(_encodingConfigContainer))
            {
                _root.Remove(_encodingConfigContainer);
            }

            // 1. 전체 화면 컨테이너 생성 및 배경 적용
            _encodingConfigContainer = new VisualElement { name = "EncodingSettingsContainer" };
            _encodingConfigContainer.style.position = Position.Absolute;
            _encodingConfigContainer.style.width = Length.Percent(100);
            _encodingConfigContainer.style.height = Length.Percent(100);
            _encodingConfigContainer.style.left = 0;
            _encodingConfigContainer.style.top = 0;
            _encodingConfigContainer.style.backgroundColor = new StyleColor(new Color(0.15f, 0.15f, 0.15f, 1.0f));
            _encodingConfigContainer.style.justifyContent = Justify.Center;
            _encodingConfigContainer.style.alignItems = Align.Center;

            // 2. 타이틀 로우
            var titleRow = CreateRow();
            var title = new Label("인코딩 설정")
            {
                style =
                {
                    fontSize = 36,
                    unityFontStyleAndWeight = FontStyle.Bold,
                    color = new StyleColor(Color.white),
                    unityTextAlign = TextAnchor.MiddleCenter,
                    width = 770, // 360 * 2 + 50 (여백)에 맞춤
                    height = StyleKeyword.Auto
                }
            };
            titleRow.Add(title);
            _encodingConfigContainer.Add(titleRow);

            // 3. 전송률 모드 로우 (라벨 + 드롭다운)
            var modeRow = CreateRow();
            var modeLabel = CreateFormLabel("전송률 제어 모드");
            var modeDropdown = new DropdownField { label = "" };
            foreach (RateControlMode modeVal in System.Enum.GetValues(typeof(RateControlMode)))
            {
                modeDropdown.choices.Add(modeVal.ToString());
            }
            modeDropdown.value = tempConfig.RateControlMode.ToString();
            
            modeRow.Add(modeLabel);
            modeRow.Add(modeDropdown);
            _encodingConfigContainer.Add(modeRow);

            // 4. Bitrate 로우 (라벨 + 입력)
            var bitrateRow = CreateRow();
            var bitrateLabel = CreateFormLabel("대상 비트레이트 (kbps)");
            var bitrateField = new IntegerField { name = "EncodingBitrate", label = "", value = tempConfig.Bitrate };
            bitrateRow.Add(bitrateLabel);
            bitrateRow.Add(bitrateField);
            _encodingConfigContainer.Add(bitrateRow);

            // 5. Max Bitrate 로우 (라벨 + 입력)
            var maxBitrateRow = CreateRow();
            var maxBitrateLabel = CreateFormLabel("최대 비트레이트 (kbps)");
            var maxBitrateField = new IntegerField { name = "EncodingMaxBitrate", label = "", value = tempConfig.MaxBitrate };
            maxBitrateRow.Add(maxBitrateLabel);
            maxBitrateRow.Add(maxBitrateField);
            _encodingConfigContainer.Add(maxBitrateRow);

            // 6. QP 로우 (라벨 + 입력)
            var qpRow = CreateRow();
            var qpLabel = CreateFormLabel("QP 품질 값 (0 ~ 51)");
            var qpField = new IntegerField { name = "EncodingQP", label = "", value = tempConfig.QP };
            qpRow.Add(qpLabel);
            qpRow.Add(qpField);
            _encodingConfigContainer.Add(qpRow);

            // 6.5. 키프레임 간격 로우 (라벨 + 입력)
            var keyframeRow = CreateRow();
            var keyframeLabel = CreateFormLabel("키프레임 간격 (초)");
            var keyframeField = new DoubleField { name = "EncodingKeyframeInterval", label = "", value = tempConfig.KeyframeInterval };
            keyframeRow.Add(keyframeLabel);
            keyframeRow.Add(keyframeField);
            _encodingConfigContainer.Add(keyframeRow);

            // 6.75. 강제 소프트웨어 인코더 로우 (라벨 + 토글)
            var forceSoftwareRow = CreateRow();
            var forceSoftwareLabel = CreateFormLabel("소프트웨어 인코더");
            var forceSoftwareToggle = new Toggle { name = "ForceSoftwareEncoder", label = "", value = tempConfig.ForceSoftwareEncoder };
            forceSoftwareRow.Add(forceSoftwareLabel);
            forceSoftwareRow.Add(forceSoftwareToggle);
            _encodingConfigContainer.Add(forceSoftwareRow);

            // 7. QP = 0 (무손실) 경고 라벨 로우 생성 (ui-row 클래스를 상속받아 NanumSquareNeo 폰트가 적용되도록 처리)
            var warningRow = CreateRow();
            warningRow.style.height = StyleKeyword.Auto;
            warningRow.style.maxHeight = StyleKeyword.Null;
            warningRow.style.minHeight = StyleKeyword.Null;

            var warningLabel = new Label("※ QP 0 (무손실)은 일부 하드웨어/코덱에서 지원되지 않을 수 있으며, 인코딩 실패 시 CPU 모드로 폴백됩니다.")
                {
                    style =
                    {
                        fontSize = 20, // 30px에서 20px로 오버라이드
                        color = new StyleColor(new Color(0.9f, 0.6f, 0.2f)),
                        whiteSpace = WhiteSpace.Normal,
                        width = 720, // 360px에서 720px로 오버라이드
                        height = StyleKeyword.Auto,
                        unityTextAlign = TextAnchor.MiddleCenter,
                        marginTop = 10,
                        marginBottom = 10
                    }
                };

            warningRow.Add(warningLabel);
            _encodingConfigContainer.Add(warningRow);

            // 7.5. 유효성 검사기 설정 (기존에 작성된 validator 재사용 및 이전 값 복원)
            var encodingValidator = new IntegerInputValidator(_encodingConfigContainer);
            encodingValidator
                .AddLimits("EncodingBitrate", new RangeIntegerLimit(100, 1000000))
                .AddLimits("EncodingMaxBitrate", new RangeIntegerLimit(100, 1000000))
                .AddLimits("EncodingQP", new RangeIntegerLimit(0, 51));
            encodingValidator.SetupValidation();

            var encodingDoubleValidator = new DoubleInputValidator(_encodingConfigContainer);
            encodingDoubleValidator
                .AddLimits("EncodingKeyframeInterval", new PositiveOnlyDoubleLimit(), new RangeDoubleLimit(0.01, 1000.0));
            encodingDoubleValidator.SetupValidation();

            modeDropdown.RegisterValueChangedCallback(evt => UpdateVisibility(evt.newValue));
            qpField.RegisterValueChangedCallback(evt =>
            {
                warningRow.style.display = (modeDropdown.value == "CQP" && evt.newValue == 0) ? DisplayStyle.Flex : DisplayStyle.None;
            });

            UpdateVisibility(modeDropdown.value);

            // 8. 제어 버튼 로우 (취소 / 저장)
            var buttonRow = CreateRow();
            var cancelBtn = new Button { text = "취소" };
            var saveBtn = new Button { text = "저장" };
            
            buttonRow.Add(cancelBtn);
            buttonRow.Add(saveBtn);
            _encodingConfigContainer.Add(buttonRow);

            cancelBtn.RegisterCallback<ClickEvent>(_ =>
            {
                ToggleScreens(false);
                _root.Remove(_encodingConfigContainer);
                _encodingConfigContainer = null!;
            });

            saveBtn.RegisterCallback<ClickEvent>(_ =>
            {
                if (System.Enum.TryParse<RateControlMode>(modeDropdown.value, out var mode))
                {
                    tempConfig.RateControlMode = mode;
                }
                tempConfig.Bitrate = bitrateField.value;
                tempConfig.MaxBitrate = maxBitrateField.value;
                tempConfig.QP = qpField.value;
                tempConfig.KeyframeInterval = keyframeField.value;
                tempConfig.ForceSoftwareEncoder = forceSoftwareToggle.value;

                if (tempConfig.RateControlMode == RateControlMode.VBR && tempConfig.MaxBitrate < tempConfig.Bitrate)
                {
                    tempConfig.MaxBitrate = tempConfig.Bitrate;
                }

                config.EncodingConfig = tempConfig;
                
                ToggleScreens(false);
                _root.Remove(_encodingConfigContainer);
                _encodingConfigContainer = null!;
            });

            _root.Add(_encodingConfigContainer);

            // 화면 전환 활성화
            ToggleScreens(true);
            return;

            // 화면 전환 제어
            void ToggleScreens(bool showEncoding)
            {
                foreach (var child in _root.Children())
                {
                    if (child == _encodingConfigContainer) continue;
                    child.style.display = showEncoding ? DisplayStyle.None : DisplayStyle.Flex;
                }
                _encodingConfigContainer.style.display = showEncoding ? DisplayStyle.Flex : DisplayStyle.None;
            }

            // 가시성 필터링 업데이트 (행 단위로 보임/숨김 제어)
            void UpdateVisibility(string selectedModeStr)
            {
                if (!System.Enum.TryParse<RateControlMode>(selectedModeStr, out var mode)) return;
                bitrateRow.style.display = mode is RateControlMode.CBR or RateControlMode.VBR ? DisplayStyle.Flex : DisplayStyle.None;
                maxBitrateRow.style.display = mode == RateControlMode.VBR ? DisplayStyle.Flex : DisplayStyle.None;
                qpRow.style.display = mode == RateControlMode.CQP ? DisplayStyle.Flex : DisplayStyle.None;
                warningRow.style.display = mode == RateControlMode.CQP && qpField.value == 0 ? DisplayStyle.Flex : DisplayStyle.None;
            }

            // 로우 생성 헬퍼 (UXML의 ui-row 속성 복제)
            VisualElement CreateRow()
            {
                var row = new VisualElement
                {
                    style =
                    {
                        flexDirection = FlexDirection.Row,
                        height = 130,
                        maxHeight = 130,
                        minHeight = 80,
                        justifyContent = Justify.Center,
                        alignItems = Align.Center
                    }
                };
                row.AddToClassList("ui-row");
                return row;
            }

            // 라벨 생성 헬퍼 (우측 정렬 적용하여 인풋과의 인접성 향상)
            Label CreateFormLabel(string text)
            {
                var label = new Label(text)
                {
                    style =
                    {
                        unityTextAlign = TextAnchor.MiddleRight,
                        color = new StyleColor(Color.white)
                    }
                };
                return label;
            }
        }

        private void TryGoRender()
        {
            var widthField = _root.Q<IntegerField>("RenderWidth");
            var heightField = _root.Q<IntegerField>("RenderHeight");
            var randomSeedField = _root.Q<IntegerField>("RandomSeed");
            var renderFpsField = _root.Q<DoubleField>("RenderFPS");
            var videoFpsField = _root.Q<DoubleField>("VideoFPS");
            var delayField = _root.Q<DoubleField>("DelayAfterLevelEnd");
            
            var showJudgementText = _root.Q<Toggle>("ShowJudgementText");
            var processAudio = _root.Q<Toggle>("ProcessAudioToggle");

            if (string.IsNullOrEmpty(widthField.text) ||
                string.IsNullOrEmpty(heightField.text) ||
                string.IsNullOrEmpty(renderFpsField.text) ||
                string.IsNullOrEmpty(videoFpsField.text) ||
                string.IsNullOrEmpty(delayField.text))
            {
                Melon<RenderMod>.Logger.Error("필수 입력 값이 비어 있습니다. 모든 설정을 입력해 주세요.");
                return;
            }

            // Copy validated values to Configuration manually
            var config = RenderManager.Instance.Configuration;
            config.RenderWidth = widthField.value;
            config.RenderHeight = heightField.value;
            config.RenderFPS = renderFpsField.value;
            config.VideoFPS = videoFpsField.value;
            config.DelayAfterLevelEnd = delayField.value;
            
            config.ShowJudgementText = showJudgementText.value;
            config.ProcessAudio = processAudio.value;
            
            if (!string.IsNullOrEmpty(randomSeedField.text))
                config.RandomSeed = randomSeedField.value;

            RenderManager.Instance.Configuration.EncodingConfig ??= GenerateEncodingConfig();
            
            RenderManager.Instance.SetupCaptureDirectory();
            RenderManager.Instance.SetupGameScene();
        }

        private EncodingConfig GenerateEncodingConfig()
        {
            return new EncodingConfig
            {
                RateControlMode = RateControlMode.CBR,
                Bitrate = 30000, MaxBitrate = 50000,
                QP = 18, KeyframeInterval = 2.0,
                ForceSoftwareEncoder = false
            };
        }
    }
}