package bjc.utils.funcutils;

import bjc.utils.funcdata.FunctionalList;
import bjc.utils.funcdata.IList;

import java.util.Random;
import java.util.function.Consumer;

/**
 * Utility methods on enums
 *
 * @author ben
 *
 */
public class EnumUtils {
	/**
	 * Do an action for a random number of enum values
	 *
	 * @param <E>
	 *                The type of the enum
	 * @param clasz
	 *                The enum class
	 * @param nValues
	 *                The number of values to execute the action on
	 * @param action
	 *                The action to perform on random values
	 * @param rnd
	 *                The source of randomness to use
	 */
	public static <E extends Enum<E>> void doForValues(Class<E> clasz, int nValues, Consumer<E> action,
			Random rnd) {
		E[] enumValues = clasz.getEnumConstants();

		IList<E> valueList = new FunctionalList<>(enumValues);

		int randomValueCount = enumValues.length - nValues;

		for(int i = 0; i <= randomValueCount; i++) {
			E rDir = valueList.randItem(rnd::nextInt);

			valueList.removeMatching(rDir);
		}

		valueList.forEach(action);
	}

	/**
	 * Get a random value from an enum
	 *
	 * @param <E>
	 *                The type of the enum
	 * @param clasz
	 *                The class of the enum
	 * @param rnd
	 *                The random source to use
	 * @return A random value from the specified enum
	 */
	public static <E extends Enum<E>> E getRandomValue(Class<E> clasz, Random rnd) {
		E[] enumValues = clasz.getEnumConstants();

		return new FunctionalList<>(enumValues).randItem(rnd::nextInt);
	}
}