Scatter2D.cs

using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Model;
 
namespace Waher.Script.Graphs.Functions.Plots
{
    public class Scatter2D : FunctionMultiVariate
    {
        private static readonly ArgumentType[] argumentTypes4Parameters = new ArgumentType[] { ArgumentType.Vector, ArgumentType.Vector, ArgumentType.Scalar, ArgumentType.Scalar };
        private static readonly ArgumentType[] argumentTypes3Parameters = new ArgumentType[] { ArgumentType.Vector, ArgumentType.Vector, ArgumentType.Scalar };
        private static readonly ArgumentType[] argumentTypes2Parameters = new ArgumentType[] { ArgumentType.Vector, ArgumentType.Vector };
 
        public Scatter2D(ScriptNode X, ScriptNode Y, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { X, Y }, argumentTypes2Parameters, Start, Length, Expression)
        {
        }
 
        public Scatter2D(ScriptNode X, ScriptNode Y, ScriptNode Color, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { X, Y, Color }, argumentTypes3Parameters, Start, Length, Expression)
        {
        }
 
        public Scatter2D(ScriptNode X, ScriptNode Y, ScriptNode Color, ScriptNode Size, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { X, Y, Color, Size }, argumentTypes4Parameters, Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(Scatter2D);
 
        public override string[] DefaultArgumentNames
        {
            get { return new string[] { "x", "y", "color", "size" }; }
        }
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            if (!(Arguments[0] is IVector X))
                throw new ScriptRuntimeException("Expected vector for X argument.", this);
 
            if (!(Arguments[1] is IVector Y))
                throw new ScriptRuntimeException("Expected vector for Y argument.", this);
 
            int Dimension = X.Dimension;
            if (Y.Dimension != Dimension)
                throw new ScriptRuntimeException("Vector size mismatch.", this);
 
            IElement Color = Arguments.Length <= 2 ? null : Arguments[2];
            IElement Size = Arguments.Length <= 3 ? null : Arguments[3];
 
            return new Graph2D(Variables, X, Y, new Scatter2DPainter(), false, false, this,
                Color?.AssociatedObjectValue ?? Graph.DefaultColor, Size?.AssociatedObjectValue ?? 5.0);
        }
    }
}