namespace SPTarkov.Server.Core.Extensions
{
    public static class MathExtensions
    {
        /// <summary>
        ///     Helper to create the cumulative sum of all enumerable elements
        ///     [1, 2, 3, 4].CumulativeSum() = [1, 3, 6, 10]
        /// </summary>
        /// <param name="values">The enumerable with numbers of which to calculate the cumulative sum</param>
        /// <returns>cumulative sum of values</returns>
        public static IEnumerable<double> CumulativeSum(this IEnumerable<double> values)
        {
            double sum = 0;
            foreach (var value in values)
            {
                sum += value;
                yield return sum;
            }
        }

        /// <summary>
        ///     Helper to create the cumulative sum of all enumerable elements
        ///     [1, 2, 3, 4].CumulativeSum() = [1, 3, 6, 10]
        /// </summary>
        /// <param name="values">The enumerable with numbers of which to calculate the cumulative sum</param>
        /// <returns>cumulative sum of values</returns>
        public static IEnumerable<float> CumulativeSum(this IEnumerable<float> values)
        {
            float sum = 0;
            foreach (var value in values)
            {
                sum += value;
                yield return sum;
            }
        }

        /// <summary>
        ///     Helper to create the product of each element times factor
        /// </summary>
        /// <param name="values">The enumerable of numbers which shall be multiplied by the factor</param>
        /// <param name="factor">Number to multiply each element by</param>
        /// <returns>An enumerable of elements all multiplied by the factor</returns>
        public static IEnumerable<double> Product(this IEnumerable<double> values, double factor)
        {
            return values.Select(v => v * factor);
        }

        /// <summary>
        ///     Helper to create the product of each element times factor
        /// </summary>
        /// <param name="values">The enumerable of numbers which shall be multiplied by the factor</param>
        /// <param name="factor">Number to multiply each element by</param>
        /// <returns>An enumerable of elements all multiplied by the factor</returns>
        public static IEnumerable<float> Product(this IEnumerable<float> values, float factor)
        {
            return values.Select(v => v * factor);
        }

        /// <summary>
        ///     Helper to determine if one double is approx equal to another double
        /// </summary>
        /// <param name="value">Value to check</param>
        /// <param name="target">Target value</param>
        /// <param name="error">Error value</param>
        /// <returns>True if value is approx target within the error range</returns>
        public static bool Approx(this double value, double target, double error = 0.001d)
        {
            return Math.Abs(value - target) <= error;
        }

        /// <summary>
        ///     Helper to determine if one float is approx equal to another float
        /// </summary>
        /// <param name="value">Value to check</param>
        /// <param name="target">Target value</param>
        /// <param name="error">Error value</param>
        /// <returns>True if value is approx target within the error range</returns>
        public static bool Approx(this float value, float target, float error = 0.001f)
        {
            return Math.Abs(value - target) <= error;
        }
    }
}