pk3DS/pk3DS.Core/Randomizers/EggMoveRandomizer.cs

using pk3DS.Core.Structures;

namespace pk3DS.Core.Randomizers
{
    public class EggMoveRandomizer : IRandomizer
    {
        /*
         * 3111 Egg Moves Learned by 290 Species (10.73 avg)
         * 18 is the most
         * 1000 moves learned were STAB (32.1%)
         */
        private readonly MoveRandomizer moverand;
        private readonly GameConfig Config;
        private readonly EggMoves[] Sets;

        public EggMoveRandomizer(GameConfig config, EggMoves[] sets)
        {
            Config = config;
            Sets = sets;
            moverand = new MoveRandomizer(config);
        }

        public bool Expand = true;
        public int ExpandTo = 18;
        public bool STAB { set => moverand.rSTAB = value; }
        public int[] BannedMoves { set => moverand.BannedMoves = value; }
        public decimal rSTABPercent { set => moverand.rSTABPercent = value; }

        public void Execute()
        {
            if (Sets[0] is EggMoves6)
                for (int i = 0; i < Sets.Length; i++)
                    Randomize(Sets[i], i);
            else if (Sets[0] is EggMoves7)
            {
                for (int i = 0; i <= Config.MaxSpeciesID; i++)
                {
                    Randomize(Sets[i], i);
                    int formoff = ((EggMoves7) Sets[i]).FormTableIndex;
                    int count = Config.Personal[i].FormeCount;
                    for (int j = 1; j < count; j++)
                        Randomize(Sets[formoff + j - 1], Config.Personal.getFormeIndex(i, j));
                }
            }
        }

        private void Randomize(EggMoves eggMoves, int index)
        {
            int count = Expand ? ExpandTo : eggMoves.Count;
            eggMoves.Moves = GetRandomMoves(count, index);
        }

        private int[] GetRandomMoves(int count, int index)
        {
            count = Expand ? ExpandTo : count;
            int[] moves = new int[count];
            moverand.GetRandomLearnset(index, count).CopyTo(moves, 0);
            return moves;
        }
    }
}