Beta.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 Beta : FunctionMultiVariate
    {
        private static readonly ArgumentType[] argumentTypes3Parameters = new ArgumentType[] { ArgumentType.Scalar, ArgumentType.Scalar, ArgumentType.Scalar };
        private static readonly ArgumentType[] argumentTypes2Parameters = new ArgumentType[] { ArgumentType.Scalar, ArgumentType.Scalar, };
 
        public Beta(ScriptNode Alpha, ScriptNode Beta, ScriptNode N, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Alpha, Beta, N }, argumentTypes3Parameters, Start, Length, Expression)
        {
        }
 
        public Beta(ScriptNode Alpha, ScriptNode Beta, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Alpha, Beta }, argumentTypes2Parameters, Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(Beta);
 
        public override string[] DefaultArgumentNames
        {
            get { return new string[] { "alpha", "beta", "N" }; }
        }
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            switch (Arguments.Length)
            {
                case 2:
                    double Alpha = Expression.ToDouble(Arguments[0].AssociatedObjectValue);
                    double Beta = Expression.ToDouble(Arguments[1].AssociatedObjectValue);
 
                    return new DoubleNumber(NextDouble(Alpha, Beta));
 
                case 3:
                default:
                    Alpha = Expression.ToDouble(Arguments[0].AssociatedObjectValue);
                    Beta = Expression.ToDouble(Arguments[1].AssociatedObjectValue);
                    double d = Expression.ToDouble(Arguments[2].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(Alpha, Beta);
 
                    return new DoubleVector(v);
            }
        }
 
        public static double NextDouble(double Alpha, double Beta)
        {
            if (Alpha <= 0)
                throw new ArgumentOutOfRangeException("Must be positive.", nameof(Alpha));
 
            if (Beta <= 0)
                throw new ArgumentOutOfRangeException("Must be positive.", nameof(Beta));
 
            double A = Gamma.NextDouble(Alpha, 1);
            double B = Gamma.NextDouble(Beta, 1);
 
            return A / (A + B);
        }
 
    }
 
}