VerticalBars3D.cs

using System;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Graphs;
using Waher.Script.Model;
 
namespace Waher.Script.Graphs3D.Functions.Plots
{
    public class VerticalBars3D : FunctionMultiVariate
    {
        public VerticalBars3D(ScriptNode X, ScriptNode Y, ScriptNode Z, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { X, Y, Z },
                  new ArgumentType[] { ArgumentType.Matrix, ArgumentType.Matrix, ArgumentType.Matrix },
                  Start, Length, Expression)
        {
        }
 
        public VerticalBars3D(ScriptNode X, ScriptNode Y, ScriptNode Z, ScriptNode Color, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { X, Y, Z, Color },
                  new ArgumentType[] { ArgumentType.Matrix, ArgumentType.Matrix, ArgumentType.Matrix, ArgumentType.Scalar },
                  Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(VerticalBars3D);
 
        public override string[] DefaultArgumentNames
        {
            get { return new string[] { "x", "y", "z", "color", "twoSided" }; }
        }
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            if (!(Arguments[0] is IMatrix X))
                throw new ScriptRuntimeException("Expected matrix for X argument.", this);
 
            if (!(Arguments[1] is IMatrix Y))
                throw new ScriptRuntimeException("Expected matrix for Y argument.", this);
 
            if (!(Arguments[2] is IMatrix Z))
                throw new ScriptRuntimeException("Expected matrix for Z argument.", this);
 
            int Rows = X.Rows;
            int Columns = X.Columns;
            if (Y.Rows != Rows || Y.Columns != Columns || Z.Rows != Rows || Z.Columns != Columns)
                throw new ScriptRuntimeException("Matrix dimension mismatch.", this);
 
            IElement Color = Arguments.Length <= 3 ? null : Arguments[3];
 
            return new Graph3D(Variables, X, Y, Z, null, new VerticalBars3DPainter(), false, true, false, this,
                Color?.AssociatedObjectValue ?? Graph.DefaultColor);
        }
    }
}