HUD.cs

using System;
using CitizenFX.Core;
using CitizenFX.Core.UI;
using CitizenFX.Core.Native;
using static CitizenFX.Core.Native.API;
using System.Drawing;
using TinyTween;

namespace FRFuel
{
    public class HUD
    {
        protected Scaleform buttons = new Scaleform("instructional_buttons");

        protected float fuelBarWidth() { return GetBarWidth(); }
        protected float fuelBarHeight = 6f;

        protected Color fuelBarColourNormal;
        protected Color fuelBarColourWarning;

        protected CitizenFX.Core.UI.Rectangle fuelBarBackdrop;
        protected CitizenFX.Core.UI.Rectangle fuelBarBack;
        protected CitizenFX.Core.UI.Rectangle fuelBar;

        protected Tween<float> fuelBarColorTween = new FloatTween();
        protected bool fuelBarAnimationDir = true;
        protected PointF basePosition = new PointF(0f, 584f);

        protected SizeF fuelBarBackdropSize;
        protected SizeF fuelBarBackSize;
        protected SizeF fuelBarSize;

        public PointF Position
        {
            set
            {
                fuelBarBackdrop.Position = value;
                fuelBarBack.Position = new PointF(value.X, value.Y + 3f);
                fuelBar.Position = fuelBarBack.Position;
            }
        }

        public HUD()
        {
            PointF fuelBarBackdropPosition = basePosition;
            PointF fuelBarBackPosition = new PointF(fuelBarBackdropPosition.X, fuelBarBackdropPosition.Y + 3f);
            PointF fuelBarPosition = fuelBarBackPosition;

            fuelBarBackdropSize = new SizeF(fuelBarWidth(), 12f);
            fuelBarBackSize = new SizeF(fuelBarWidth(), fuelBarHeight);
            fuelBarSize = fuelBarBackSize;

            Color fuelBarBackdropColour = Color.FromArgb(100, 0, 0, 0);
            Color fuelBarBackColour = Color.FromArgb(50, 255, 179, 0);

            /// Default colors will be changed in the <see cref="FRFuel.LoadConfig"/> function, by calling the <see cref="UpdateBarColors"/> function.
            fuelBarColourNormal = Color.FromArgb(150, 255, 179, 0);
            fuelBarColourWarning = Color.FromArgb(255, 255, 245, 220);

            fuelBarBackdrop = new CitizenFX.Core.UI.Rectangle(fuelBarBackdropPosition, fuelBarBackdropSize, fuelBarBackdropColour);
            fuelBarBack = new CitizenFX.Core.UI.Rectangle(fuelBarBackPosition, fuelBarBackSize, fuelBarBackColour);
            fuelBar = new CitizenFX.Core.UI.Rectangle(fuelBarPosition, fuelBarSize, fuelBarColourNormal);
        }

        /// <summary>
        /// Updates the <see cref="fuelBarColourNormal"/> and <see cref="fuelBarColourWarning"/> colors.
        /// </summary>
        /// <param name="red"></param>
        /// <param name="green"></param>
        /// <param name="blue"></param>
        /// <param name="warningRed"></param>
        /// <param name="warningGreen"></param>
        /// <param name="warningBlue"></param>
        public void UpdateBarColors(int red, int green, int blue, int warningRed, int warningGreen, int warningBlue)
        {
            fuelBarColourNormal = Color.FromArgb(150, red, green, blue);
            fuelBarColourWarning = Color.FromArgb(255, warningRed, warningGreen, warningBlue);
        }

        /// <summary>
        /// Reloads scaleform movie to ensure that it will be rendered
        /// Workaround for bug
        /// Looks safe to span on every tick /shrug
        /// </summary>
        public void ReloadScaleformMovie()
        {
            buttons = new Scaleform("instructional_buttons");
        }

        /// <summary>
        /// Renders fuel bar
        /// </summary>
        /// <param name="currentFuelLevel"></param>
        /// <param name="maxFuelLevel"></param>
        public void RenderBar(float currentFuelLevel, float maxFuelLevel)
        {
            float fuelLevelPercentage = (100f / maxFuelLevel) * currentFuelLevel;
            PointF safeZone = GetSafezoneBounds();

            if (IsBigmapActive())
            {
                Position = new PointF(basePosition.X + safeZone.X, basePosition.Y - safeZone.Y - 180f);
            }
            else
            {
                Position = new PointF(basePosition.X + safeZone.X, basePosition.Y - safeZone.Y);
            }

            fuelBarBackdropSize = new SizeF(fuelBarWidth(), 12f);
            fuelBarSize = new SizeF((fuelBarWidth() / 100f) * fuelLevelPercentage, fuelBarHeight);
            fuelBarBackSize = fuelBarSize;

            fuelBar.Size = fuelBarSize;
            fuelBarBackdrop.Size = fuelBarBackdropSize;
            fuelBarBack.Size = fuelBarBackSize;



            if (maxFuelLevel > 0 && currentFuelLevel < 9f)
            {
                if (fuelBarColorTween.State == TweenState.Stopped)
                {
                    fuelBarAnimationDir = !fuelBarAnimationDir;

                    fuelBarColorTween.Start(
                      fuelBarAnimationDir ? 100f : 255f,
                      fuelBarAnimationDir ? 255f : 100f,
                      .5f, // seconds
                      ScaleFuncs.QuarticEaseOut
                    );
                }

                fuelBarColorTween.Update(Game.LastFrameTime);

                fuelBar.Color = Color.FromArgb((int)Math.Floor(fuelBarColorTween.CurrentValue), fuelBarColourWarning);
            }
            else
            {
                fuelBar.Color = fuelBarColourNormal;

                if (fuelBarColorTween.State != TweenState.Stopped)
                {
                    fuelBarColorTween.Stop(StopBehavior.ForceComplete);
                }
            }

            fuelBarBackdrop.Draw();
            fuelBarBack.Draw();
            fuelBar.Draw();
        }

        /// <summary>
        /// Returns user-configured screen safe zone offset
        /// </summary>
        /// <returns></returns>
        public static PointF GetSafezoneBounds()
        {
            float t = GetSafeZoneSize();
            float w = Screen.Width;
            float h = Screen.Height;

            return new PointF(
              (int)Math.Round((w - (w * t)) / 2 + 1),
              (int)Math.Round((h - (h * t)) / 2 - 2)
            );
        }

        /// <summary>
        /// Change instructions for engine cut off
        /// </summary>
        public void InstructTurnOffEngine()
        {
            buttons.CallFunction("CLEAR_ALL");
            buttons.CallFunction("TOGGLE_MOUSE_BUTTONS", 0);
            buttons.CallFunction("CREATE_CONTAINER");

            buttons.CallFunction("SET_DATA_SLOT", 0, Function.Call<string>((Hash)0x0499D7B09FC9B407, 2, (int)FRFuel.engineToggleControl, false), "Turn engine off");

            buttons.CallFunction("DRAW_INSTRUCTIONAL_BUTTONS", -1);
        }

        /// <summary>
        /// Returns resolution specified bar width
        /// </summary>
        /// <returns></returns>
        public static float GetBarWidth()
        {
            float width;
            double aspect = Screen.AspectRatio;
            bool bigMap = IsBigmapActive();

            switch (aspect)
            {
                case (float)1.5: // 3:2
                    width = bigMap ? 336f : 212f;
                    break;
                case (float)1.33333337306976: // 4:3
                    width = bigMap ? 378f : 240f;
                    break;
                case (float)1.66666662693024: // 5:3
                    width = bigMap ? 302f : 191f;
                    break;
                case (float)1.25: // 5:4
                    width = bigMap ? 405f : 255f;
                    break;
                case (float)1.60000002384186: // 16:10
                    width = bigMap ? 316f : 200f;
                    break;
                default:
                    width = bigMap ? 285f : 180f; // 16:9
                    break;
            }
            return width;
        }

        /// <summary>
        /// Change instructions for refueling and engine spin up
        /// </summary>
        public void InstructRefuelOrTurnOnEngine()
        {
            buttons.CallFunction("CLEAR_ALL");
            buttons.CallFunction("TOGGLE_MOUSE_BUTTONS", 0);
            buttons.CallFunction("CREATE_CONTAINER");

            buttons.CallFunction("SET_DATA_SLOT", 0, Function.Call<string>((Hash)0x0499D7B09FC9B407, 2, (int)Control.Jump, false), "Refuel");
            buttons.CallFunction("SET_DATA_SLOT", 1, Function.Call<string>((Hash)0x0499D7B09FC9B407, 2, (int)FRFuel.engineToggleControl, 0), "Turn engine on");

            buttons.CallFunction("DRAW_INSTRUCTIONAL_BUTTONS", -1);
        }

        /// <summary>
        /// Change instructions for manual refueling
        /// </summary>
        /// <param name="label"></param>
        public void InstructManualRefuel(string label)
        {
            buttons.CallFunction("CLEAR_ALL");
            buttons.CallFunction("TOGGLE_MOUSE_BUTTONS", 0);
            buttons.CallFunction("CREATE_CONTAINER");

            buttons.CallFunction("SET_DATA_SLOT", 0, Function.Call<string>((Hash)0x0499D7B09FC9B407, 2, (int)Control.Attack, false), label);

            buttons.CallFunction("DRAW_INSTRUCTIONAL_BUTTONS", -1);
        }

        /// <summary>
        /// Renders instruction
        /// </summary>
        public void RenderInstructions()
        {
            buttons.Render2D();
        }
    }
}