Cielonos/Assets/Scripts/MainGame/UI/PlayerUI/MainWeapon/MainWeaponFunctionIcon.cs

using Cielonos.MainGame.Inventory;
using SLSFramework.UI;
using UnityEngine;
using UnityEngine.Serialization;
using UnityEngine.UI;

namespace Cielonos.UI
{
    public class MainWeaponFunctionIcon : UIElementBase
    {
        public RuntimeFunctionUnit functionUnit;
        [FormerlySerializedAs("timerFillImage")] public Image frame;
        public Image iconImage;
        public Image timer;

        public void Initialize(RuntimeFunctionUnit functionUnit)
        {
            this.functionUnit = functionUnit;
            iconImage.sprite = functionUnit.data.icon != null ? functionUnit.data.icon : null;
        }

        public override void UpdateUI()
        {
            float fillAmount;
            
            if (functionUnit.maxCooldown <= 0f)
            {
                fillAmount = 1f;
            }
            else
            {
                fillAmount = 1f - functionUnit.currentCooldown / functionUnit.maxCooldown;
            }
            
            frame.fillAmount = fillAmount;
        }
    }
}