PKHeX/PKHeX.Core/PKM/Shared/IContestStats.cs

namespace PKHeX.Core
{
    public interface IContestStats
    {
        byte CNT_Cool { get; }
        byte CNT_Beauty { get; }
        byte CNT_Cute { get; }
        byte CNT_Smart { get; }
        byte CNT_Tough { get; }
        byte CNT_Sheen { get; }
    }

    public interface IContestStatsMutable
    {
        byte CNT_Cool { set; }
        byte CNT_Beauty { set; }
        byte CNT_Cute { set; }
        byte CNT_Smart { set; }
        byte CNT_Tough { set; }
        byte CNT_Sheen { set; }
    }

    public static partial class Extensions
    {
        public static byte[] GetContestStats(this IContestStats stats) => new[]
        {
            stats.CNT_Cool,
            stats.CNT_Beauty,
            stats.CNT_Cute,
            stats.CNT_Smart,
            stats.CNT_Tough,
            stats.CNT_Sheen,
        };

        /// <summary>
        /// Checks if any contest stat value is nonzero.
        /// </summary>
        /// <param name="stats">Object containing contest stat data.</param>
        /// <returns>True if has any nonzero contest stat, false if all are zero.</returns>
        public static bool HasContestStats(this IContestStats stats)
        {
            if (stats.CNT_Cool != 0)
                return true;
            if (stats.CNT_Beauty != 0)
                return true;
            if (stats.CNT_Cute != 0)
                return true;
            if (stats.CNT_Smart != 0)
                return true;
            if (stats.CNT_Tough != 0)
                return true;
            if (stats.CNT_Sheen != 0)
                return true;
            return false;
        }

        public static bool IsContestBelow(this IContestStats current, IContestStats initial) => !current.IsContestAboveOrEqual(initial);

        public static bool IsContestAboveOrEqual(this IContestStats current, IContestStats initial)
        {
            if (current.CNT_Cool   < initial.CNT_Cool)
                return false;
            if (current.CNT_Beauty < initial.CNT_Beauty)
                return false;
            if (current.CNT_Cute   < initial.CNT_Cute)
                return false;
            if (current.CNT_Smart  < initial.CNT_Smart)
                return false;
            if (current.CNT_Tough  < initial.CNT_Tough)
                return false;
            if (current.CNT_Sheen  < initial.CNT_Sheen)
                return false;
            return true;
        }

        public static bool IsContestEqual(this IContestStats current, IContestStats initial)
        {
            if (current.CNT_Cool != initial.CNT_Cool)
                return false;
            if (current.CNT_Beauty != initial.CNT_Beauty)
                return false;
            if (current.CNT_Cute != initial.CNT_Cute)
                return false;
            if (current.CNT_Smart != initial.CNT_Smart)
                return false;
            if (current.CNT_Tough != initial.CNT_Tough)
                return false;
            if (current.CNT_Sheen != initial.CNT_Sheen)
                return false;
            return true;
        }

        public static void CopyContestStatsTo(this IContestStats source, IContestStatsMutable dest)
        {
            dest.CNT_Cool = source.CNT_Cool;
            dest.CNT_Beauty = source.CNT_Beauty;
            dest.CNT_Cute = source.CNT_Cute;
            dest.CNT_Smart = source.CNT_Smart;
            dest.CNT_Tough = source.CNT_Tough;
            dest.CNT_Sheen = source.CNT_Sheen;
        }

        public static void SetAllContestStatsTo(this IContestStatsMutable dest, byte value, byte sheen)
        {
            dest.CNT_Cool = value;
            dest.CNT_Beauty = value;
            dest.CNT_Cute = value;
            dest.CNT_Smart = value;
            dest.CNT_Tough = value;
            dest.CNT_Sheen = sheen;
        }

        private const byte CONTEST_MAX = 255;

        /// <summary>
        /// Check if any contest stat besides <see cref="IContestStats.CNT_Sheen"/> is equal to <see cref="CONTEST_MAX"/>.
        /// </summary>
        /// <param name="s">Entity to check</param>
        /// <returns>True if any equals <see cref="CONTEST_MAX"/></returns>
        public static bool IsAnyContestStatMax(this IContestStats s) => CONTEST_MAX == s.CNT_Cool 
                                                                     || CONTEST_MAX == s.CNT_Beauty 
                                                                     || CONTEST_MAX == s.CNT_Cute
                                                                     || CONTEST_MAX == s.CNT_Smart 
                                                                     || CONTEST_MAX == s.CNT_Tough;
    }
}