Ui-Info-Suite/SDVModTest/UIElements/ShowAccurateHearts.cs

using Microsoft.Xna.Framework;
using StardewModdingAPI;
using StardewModdingAPI.Events;
using StardewValley;
using StardewValley.Menus;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;

namespace UIInfoSuite.UIElements
{
    class ShowAccurateHearts : IDisposable
    {
        private String[] _friendNames;
        private SocialPage _socialPage;
        private IModEvents _events;

        private readonly int[][] _numArray = new int[][]
        {
            new int[] { 1, 1, 0, 1, 1 },
            new int[] { 1, 1, 1, 1, 1 },
            new int[] { 0, 1, 1, 1, 0 },
            new int[] { 0, 0, 1, 0, 0 }
        };

        public ShowAccurateHearts(IModEvents events)
        {
            _events = events;
        }

        public void ToggleOption(bool showAccurateHearts)
        {
            _events.Display.MenuChanged -= OnMenuChanged;
            _events.Display.RenderedActiveMenu -= OnRenderedActiveMenu;

            if (showAccurateHearts)
            {
                _events.Display.MenuChanged += OnMenuChanged;
                _events.Display.RenderedActiveMenu += OnRenderedActiveMenu;
            }
        }

        public void Dispose()
        {
            ToggleOption(false);
        }

        /// <summary>When a menu is open (<see cref="Game1.activeClickableMenu"/> isn't null), raised after that menu is drawn to the sprite batch but before it's rendered to the screen.</summary>
        /// <param name="sender">The event sender.</param>
        /// <param name="e">The event arguments.</param>
        private void OnRenderedActiveMenu(object sender, RenderedActiveMenuEventArgs e)
        {
            // draw heart fills
            if (Game1.activeClickableMenu is GameMenu gameMenu)
            {
                if (gameMenu.currentTab == 2)
                {
                    if (_socialPage != null)
                    {
                        int slotPosition = (int)typeof(SocialPage)
                            .GetField(
                                "slotPosition",
                                BindingFlags.Instance | BindingFlags.NonPublic)
                                .GetValue(_socialPage);
                        int yOffset = 0;

                        if (Constants.TargetPlatform != GamePlatform.Android)
                        {
                            for (int i = slotPosition; i < slotPosition + 5 && i < _friendNames.Length; ++i)
                            {
                                int yPosition = Game1.activeClickableMenu.yPositionOnScreen + 130 + yOffset;
                                yOffset += 112;
                                Friendship friendshipValues;
                                String nextName = _friendNames[i];
                                if (Game1.player.friendshipData.TryGetValue(nextName, out friendshipValues))
                                {
                                    int friendshipRawValue = friendshipValues.Points;

                                    if (friendshipRawValue > 0)
                                    {
                                        int pointsToNextHeart = friendshipRawValue % 250;
                                        int numHearts = friendshipRawValue / 250;

                                        if (friendshipRawValue < 3000 &&
                                            _friendNames[i] == Game1.player.spouse ||
                                            friendshipRawValue < 2500)
                                        {
                                            DrawEachIndividualSquare(numHearts, pointsToNextHeart, yPosition);
                                            //if (!Game1.options.hardwareCursor)
                                            //    Game1.spriteBatch.Draw(
                                            //        Game1.mouseCursors,
                                            //        new Vector2(Game1.getMouseX(), Game1.getMouseY()),
                                            //        Game1.getSourceRectForStandardTileSheet(
                                            //            Game1.mouseCursors, Game1.mouseCursor,
                                            //            16,
                                            //            16),
                                            //        Color.White,
                                            //        0.0f,
                                            //        Vector2.Zero,
                                            //        Game1.pixelZoom + (float)(Game1.dialogueButtonScale / 150.0),
                                            //        SpriteEffects.None,
                                            //        1f);
                                        }
                                    }
                                }
                            }
                        }
                        else
                        {
                            List<ClickableTextureComponent> sprites = (List<ClickableTextureComponent>)_socialPage.GetType().GetField("sprites", BindingFlags.Instance | BindingFlags.NonPublic).GetValue(_socialPage);
                            for (int i = 0; i < _friendNames.Length; ++i)
                            {
                                if(sprites[i].bounds.Y < 40)
                                {
                                    continue;
                                }
                                if(sprites[i].bounds.Y > _socialPage.height)
                                {
                                    break;
                                }
                                Friendship friendshipValues;
                                String nextName = _friendNames[i];
                                if (Game1.player.friendshipData.TryGetValue(nextName, out friendshipValues))
                                {
                                    int friendshipRawValue = friendshipValues.Points;

                                    if (friendshipRawValue > 0)
                                    {
                                        int pointsToNextHeart = friendshipRawValue % 250;
                                        int numHearts = friendshipRawValue / 250;

                                        if (friendshipRawValue < 3000 &&
                                            _friendNames[i] == Game1.player.spouse ||
                                            friendshipRawValue < 2500)
                                        {
                                            DrawEachIndividualSquare(numHearts, pointsToNextHeart, sprites[i].bounds.Y + 40);
                                            //if (!Game1.options.hardwareCursor)
                                            //    Game1.spriteBatch.Draw(
                                            //        Game1.mouseCursors,
                                            //        new Vector2(Game1.getMouseX(), Game1.getMouseY()),
                                            //        Game1.getSourceRectForStandardTileSheet(
                                            //            Game1.mouseCursors, Game1.mouseCursor,
                                            //            16,
                                            //            16),
                                            //        Color.White,
                                            //        0.0f,
                                            //        Vector2.Zero,
                                            //        Game1.pixelZoom + (float)(Game1.dialogueButtonScale / 150.0),
                                            //        SpriteEffects.None,
                                            //        1f);
                                        }
                                    }
                                }
                            }
                        }
                        String hoverText = gameMenu.hoverText;
                        IClickableMenu.drawHoverText(
                            Game1.spriteBatch,
                            hoverText,
                            Game1.smallFont);
                    }
                    else
                    {
                        ExtendMenuIfNeeded();
                    }
                }
            }
        }

        /// <summary>Raised after a game menu is opened, closed, or replaced.</summary>
        /// <param name="sender">The event sender.</param>
        /// <param name="e">The event arguments.</param>
        private void OnMenuChanged(object sender, MenuChangedEventArgs e)
        {
            ExtendMenuIfNeeded();
        }

        private void ExtendMenuIfNeeded()
        {
            if (Game1.activeClickableMenu is GameMenu gameMenu)
            {
                foreach (var menu in gameMenu.pages)
                {
                    if (menu is SocialPage page)
                    {
                        _socialPage = page;
                        _friendNames = _socialPage.names
                            .Select(name => name.ToString())
                            .ToArray();
                        break;
                    }
                }
            }
        }

        private void DrawEachIndividualSquare(int friendshipLevel, int friendshipPoints, int yPosition)
        {
            int numberOfPointsToDraw = (int)(((double)friendshipPoints) / 12.5);
            int num2;

            if (friendshipLevel > 10)
            {
                num2 = 32 * (friendshipLevel - 10);
                yPosition += 28;
            }
            else
            {
                num2 = 32 * friendshipLevel;
            }

            for (int i = 3; i >= 0 && numberOfPointsToDraw > 0; --i)
            {
                for (int j = 0; j < 5 && numberOfPointsToDraw > 0; ++j, --numberOfPointsToDraw)
                {
                    if (_numArray[i][j] == 1)
                    {
                        if(Constants.TargetPlatform != GamePlatform.Android)
                        {
                            Game1.spriteBatch.Draw(
                                Game1.staminaRect,
                                new Rectangle(
                                    Game1.activeClickableMenu.xPositionOnScreen + 316 + num2 + j * 4,
                                    yPosition + 14 + i * 4,
                                    4,
                                    4),
                                Color.Crimson);
                        }
                        else
                        {
                            int heartsX = (int)_socialPage.GetType().GetField("heartsX", BindingFlags.NonPublic | BindingFlags.Instance).GetValue(_socialPage);
                            float widthMod = (float)_socialPage.GetType().GetField("widthMod", BindingFlags.NonPublic | BindingFlags.Instance).GetValue(_socialPage);
                            Game1.spriteBatch.Draw(
                                Game1.staminaRect,
                                new Rectangle(
                                    Game1.activeClickableMenu.xPositionOnScreen + heartsX + (int)(num2 * widthMod) + 24 + j * 4,
                                    yPosition + i * 4,
                                    4,
                                    4),
                                Color.Crimson);
                        }
                    }
                }
            }
        }
    }
}