Chi2.cs

using System;
using System.Collections.Generic;
using System.Security.Cryptography;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Model;
using Waher.Script.Objects;
using Waher.Script.Objects.VectorSpaces;
 
namespace Waher.Script.Statistics.Functions.RandomNumbers
{
    public class Chi2 : FunctionMultiVariate
    {
        private static readonly ArgumentType[] argumentTypes2Parameters = new ArgumentType[] { ArgumentType.Scalar, ArgumentType.Scalar };
        private static readonly ArgumentType[] argumentTypes1Parameters = new ArgumentType[] { ArgumentType.Scalar, };
 
        public Chi2(ScriptNode Degrees, ScriptNode N, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Degrees, N }, argumentTypes2Parameters, Start, Length, Expression)
        {
        }
 
        public Chi2(ScriptNode Degrees, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Degrees }, argumentTypes1Parameters, Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(Chi2);
 
        public override string[] DefaultArgumentNames
        {
            get { return new string[] { "degrees", "N" }; }
        }
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            switch (Arguments.Length)
            {
                case 1:
                    double Degrees = Expression.ToDouble(Arguments[0].AssociatedObjectValue);
 
                    return new DoubleNumber(NextDouble(Degrees));
 
                case 2:
                default:
                    Degrees = Expression.ToDouble(Arguments[0].AssociatedObjectValue);
                    double d = Expression.ToDouble(Arguments[1].AssociatedObjectValue);
                    int N = (int)Math.Round(d);
                    if (N < 0 || N != d)
                        throw new ScriptRuntimeException("N must be a non-negative integer.", this);
 
                    double[] v = new double[N];
                    int i;
 
                    for (i = 0; i < N; i++)
                        v[i] = NextDouble(Degrees);
 
                    return new DoubleVector(v);
            }
        }
 
        public static double NextDouble(double Degrees)
        {
            return Gamma.NextDouble(Degrees / 2, 2);
        }
 
    }
 
}