Glamourer/Glamourer/Gui/InterfaceActorSelector.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Numerics;
using Dalamud.Game.ClientState.Objects.Enums;
using Dalamud.Game.ClientState.Objects.Types;
using Dalamud.Interface;
using Dalamud.Logging;
using ImGuiNET;
using OtterGui;
using OtterGui.Raii;
using Penumbra.PlayerWatch;

namespace Glamourer.Gui;

internal partial class Interface
{
    public const int CharacterScreenIndex = 240;
    public const int ExamineScreenIndex   = 241;
    public const int FittingRoomIndex     = 242;
    public const int DyePreviewIndex      = 243;

    private          Character?                     _player;
    private          string                         _currentLabel      = string.Empty;
    private          string                         _playerFilter      = string.Empty;
    private          string                         _playerFilterLower = string.Empty;
    private readonly Dictionary<string, int>        _playerNames       = new(100);
    private readonly Dictionary<string, Character?> _gPoseActors       = new(CharacterScreenIndex - GPoseObjectId);

    private void DrawPlayerFilter()
    {
        using var style = ImRaii.PushStyle(ImGuiStyleVar.ItemSpacing, Vector2.Zero)
            .Push(ImGuiStyleVar.FrameRounding, 0);
        ImGui.SetNextItemWidth(SelectorWidth * ImGui.GetIO().FontGlobalScale);
        if (ImGui.InputTextWithHint("##playerFilter", "Filter Players...", ref _playerFilter, 32))
            _playerFilterLower = _playerFilter.ToLowerInvariant();
    }

    private void DrawGPoseSelectable(Character player)
    {
        var playerName = player.Name.ToString();
        if (!playerName.Any())
            return;

        _gPoseActors[playerName] = null;

        DrawSelectable(player, $"{playerName} (GPose)", true);
    }

    private static string GetLabel(Character player, string playerName, int num)
    {
        if (player.ObjectKind == ObjectKind.Player)
            return num == 1 ? playerName : $"{playerName} #{num}";

        if (player.ModelType() == 0)
            return num == 1 ? $"{playerName} (NPC)" : $"{playerName} #{num} (NPC)";

        return num == 1 ? $"{playerName} (Monster)" : $"{playerName} #{num} (Monster)";
    }

    private void DrawPlayerSelectable(Character player, int idx = 0)
    {
        var (playerName, modifiable) = idx switch
        {
            CharacterScreenIndex => ("Character Screen Actor", false),
            ExamineScreenIndex   => ("Examine Screen Actor", false),
            FittingRoomIndex     => ("Fitting Room Actor", false),
            DyePreviewIndex      => ("Dye Preview Actor", false),
            _                    => (player.Name.ToString(), true),
        };
        if (!playerName.Any())
            return;

        if (_playerNames.TryGetValue(playerName, out var num))
            _playerNames[playerName] = ++num;
        else
            _playerNames[playerName] = num = 1;

        if (_gPoseActors.ContainsKey(playerName))
        {
            _gPoseActors[playerName] = player;
            return;
        }

        var label = GetLabel(player, playerName, num);
        DrawSelectable(player, label, modifiable);
    }


    private void DrawSelectable(Character player, string label, bool modifiable)
    {
        if (!_playerFilterLower.Any() || label.ToLowerInvariant().Contains(_playerFilterLower))
            if (ImGui.Selectable(label, _currentLabel == label))
            {
                _currentLabel = label;
                _currentSave.LoadCharacter(player);
                _player                     = player;
                _currentSave.WriteProtected = !modifiable;
                return;
            }

        if (_currentLabel != label)
            return;

        try
        {
            _currentSave.LoadCharacter(player);
            _player                     = player;
            _currentSave.WriteProtected = !modifiable;
        }
        catch (Exception e)
        {
            PluginLog.Error($"Could not load character {player.Name}s information:\n{e}");
        }
    }

    private void DrawSelectionButtons()
    {
        using var style = ImRaii.PushStyle(ImGuiStyleVar.ItemSpacing, Vector2.Zero)
            .Push(ImGuiStyleVar.FrameRounding, 0);
        using var  font        = ImRaii.PushFont(UiBuilder.IconFont);
        Character? select      = null;
        var        buttonWidth = Vector2.UnitX * SelectorWidth / 2;
        if (ImGui.Button(FontAwesomeIcon.UserCircle.ToIconString(), buttonWidth))
            select = Dalamud.ClientState.LocalPlayer;
        font.Pop();
        ImGuiUtil.HoverTooltip("Select the local player character.");
        ImGui.SameLine();
        font.Push(UiBuilder.IconFont);
        if (_inGPose)
        {
            style.Push(ImGuiStyleVar.Alpha, 0.5f);
            ImGui.Button(FontAwesomeIcon.HandPointer.ToIconString(), buttonWidth);
            style.Pop();
        }
        else
        {
            if (ImGui.Button(FontAwesomeIcon.HandPointer.ToIconString(), buttonWidth))
                select = CharacterFactory.Convert(Dalamud.Targets.Target);
        }

        font.Pop();
        ImGuiUtil.HoverTooltip("Select the current target, if it is in the list.");

        if (select == null)
            return;

        try
        {
            _currentSave.LoadCharacter(select);
            _player                     = select;
            _currentLabel               = _player.Name.ToString();
            _currentSave.WriteProtected = false;
        }
        catch (Exception e)
        {
            PluginLog.Error($"Could not load character {select.Name}s information:\n{e}");
        }
    }

    private void DrawPlayerSelector()
    {
        ImGui.BeginGroup();
        DrawPlayerFilter();
        if (!ImGui.BeginChild("##playerSelector",
                new Vector2(SelectorWidth * ImGui.GetIO().FontGlobalScale, -ImGui.GetFrameHeight() - 1), true))
        {
            ImGui.EndChild();
            ImGui.EndGroup();
            return;
        }

        _playerNames.Clear();
        _gPoseActors.Clear();
        for (var i = GPoseObjectId; i < GPoseObjectId + 48; ++i)
        {
            var player = CharacterFactory.Convert(Dalamud.Objects[i]);
            if (player == null)
                break;

            DrawGPoseSelectable(player);
        }

        for (var i = 0; i < GPoseObjectId; ++i)
        {
            var player = CharacterFactory.Convert(Dalamud.Objects[i]);
            if (player != null)
                DrawPlayerSelectable(player);
        }

        for (var i = CharacterScreenIndex; i < Dalamud.Objects.Length; ++i)
        {
            var player = CharacterFactory.Convert(Dalamud.Objects[i]);
            if (player != null)
                DrawPlayerSelectable(player, i);
        }


        using (var _ = ImRaii.PushStyle(ImGuiStyleVar.ItemSpacing, Vector2.Zero))
        {
            ImGui.EndChild();
        }

        DrawSelectionButtons();
        ImGui.EndGroup();
    }

    private void DrawPlayerTab()
    {
        using var tab = ImRaii.TabItem("Current Players");
        _player = null;
        if (!tab)
            return;

        DrawPlayerSelector();

        if (_currentLabel.Length == 0)
            return;

        ImGui.SameLine();
        DrawActorPanel();
    }
}