VerticalBars.cs

using System.Collections.Generic;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Model;
 
namespace Waher.Script.Graphs.Functions.Plots
{
    public class VerticalBars : FunctionMultiVariate
    {
        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 VerticalBars(ScriptNode Labels, ScriptNode Values, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Labels, Values }, argumentTypes2Parameters, Start, Length, Expression)
        {
        }
 
        public VerticalBars(ScriptNode Labels, ScriptNode Values, ScriptNode Color, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Labels, Values, Color }, argumentTypes3Parameters, Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(VerticalBars);
 
        public override string[] DefaultArgumentNames
        {
            get { return new string[] { "labels", "values", "color" }; }
        }
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            if (!(Arguments[0] is IVector Labels))
                throw new ScriptRuntimeException("Expected vector for Labels argument.", this);
 
            if (!(Arguments[1] is IVector Values))
                throw new ScriptRuntimeException("Expected vector for Values argument.", this);
 
            int Dimension = Labels.Dimension;
            if (Values.Dimension != Dimension)
                throw new ScriptRuntimeException("Vector size mismatch.", this);
 
            IElement Color = Arguments.Length <= 2 ? null : Arguments[2];
 
            return new Graph2D(Variables, Labels, Values, new VerticalBarsPainter(), false, true, this,
                Color?.AssociatedObjectValue ?? Graph.DefaultColor);
        }
    }
}