HSUI/Interface/GeneralElements/PrimaryResourceHud.cs

using HSUI.Config;
using HSUI.Helpers;
using Dalamud.Bindings.ImGui;
using System.Collections.Generic;
using System.Numerics;
using Dalamud.Game.ClientState.Objects.Types;
using System;
using Dalamud.Game.ClientState.Objects.SubKinds;
using HSUI.Enums;
using HSUI.Interface.Bars;
using HSUI.Interface.Party;

namespace HSUI.Interface.GeneralElements
{
    public class PrimaryResourceHud : ParentAnchoredDraggableHudElement, IHudElementWithActor, IHudElementWithAnchorableParent, IHudElementWithVisibilityConfig
    {
        private PrimaryResourceConfig Config => (PrimaryResourceConfig)_config;
        public VisibilityConfig? VisibilityConfig => Config is UnitFramePrimaryResourceConfig config ? config.VisibilityConfig : null;

        public PrimaryResourceTypes ResourceType = PrimaryResourceTypes.MP;

        private IGameObject? _actor;
        public IGameObject? Actor
        {
            get => _actor;
            set
            {
                if (value is IPlayerCharacter chara)
                {
                    _actor = value;

                    JobRoles role = JobsHelper.RoleForJob(chara.ClassJob.RowId);
                    ResourceType = JobsHelper.PrimaryResourceTypesByRole[role];
                }
                else
                {
                    _actor = null;
                    ResourceType = PrimaryResourceTypes.None;
                }
            }
        }

        public IPartyFramesMember? PartyMember;

        protected override bool AnchorToParent => Config is UnitFramePrimaryResourceConfig config ? config.AnchorToUnitFrame : false;
        protected override DrawAnchor ParentAnchor => Config is UnitFramePrimaryResourceConfig config ? config.UnitFrameAnchor : DrawAnchor.Center;

        public PrimaryResourceHud(PrimaryResourceConfig config, string? displayName = null) : base(config, displayName)
        {
        }

        protected override (List<Vector2>, List<Vector2>) ChildrenPositionsAndSizes()
        {
            return (new List<Vector2>() { Config.Position }, new List<Vector2>() { Config.Size });
        }

        public override void DrawChildren(Vector2 origin)
        {
            if (!Config.Enabled)
            {
                return;
            }

            if (PartyMember == null && (ResourceType == PrimaryResourceTypes.None || Actor == null || Actor is not IPlayerCharacter))
            {
                return;
            }

            ICharacter? chara = Actor != null ? (ICharacter)Actor : null;
            uint current = chara == null ? PartyMember?.MP ?? 0 : 0;
            uint max = chara == null ? PartyMember?.MaxMP ?? 0 : 0;

            if (chara != null)
            {
                GetResources(ref current, ref max, chara);
            }

            if (Config.HidePrimaryResourceWhenFull && current == max)
            {
                return;
            }

            BarHud bar = BarUtilities.GetProgressBar(
                Config,
                Config.ThresholdConfig,
                new LabelConfig[] { Config.ValueLabel },
                current,
                max,
                0,
                chara,
                GetColor()
            );

            Vector2 pos = origin + ParentPos();
            AddDrawActions(bar.GetDrawActions(pos, Config.StrataLevel));
        }

        private void GetResources(ref uint current, ref uint max, ICharacter actor)
        {
            switch (ResourceType)
            {
                case PrimaryResourceTypes.MP:
                    current = actor.CurrentMp;
                    max = actor.MaxMp;
                    break;

                case PrimaryResourceTypes.CP:
                    current = actor.CurrentCp;
                    max = actor.MaxCp;
                    break;

                case PrimaryResourceTypes.GP:
                    current = actor.CurrentGp;
                    max = actor.MaxGp;
                    break;
            }
        }

        public virtual PluginConfigColor GetColor()
        {
            if (!Config.UseJobColor)
            {
                return Config.FillColor;
            }

            if (PartyMember != null)
            {
                return GlobalColors.Instance.SafeColorForJobId(PartyMember.JobId);
            }

            return ColorUtils.ColorForActor(Actor);
        }
    }
}