HorizontalBars.cs

using System;
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 HorizontalBars : 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 HorizontalBars(ScriptNode Labels, ScriptNode Values, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Labels, Values }, argumentTypes2Parameters, Start, Length, Expression)
        {
        }
 
        public HorizontalBars(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(HorizontalBars);
 
        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];
 
            LinkedList<IElement> ReverseLabelOrder = new LinkedList<IElement>();
            LinkedList<IElement> ReverseValueOrder = new LinkedList<IElement>();
 
            foreach (IElement Label in Labels.VectorElements)
                ReverseLabelOrder.AddFirst(Label);
 
            foreach (IElement Value in Values.VectorElements)
                ReverseValueOrder.AddFirst(Value);
 
            return new Graph2D(Variables, (IVector)Values.Encapsulate(ReverseValueOrder, this), (IVector)Labels.Encapsulate(ReverseLabelOrder, 
                this), new HorizontalBarPainter(), true, false, this, Color?.AssociatedObjectValue ?? Graph.DefaultColor);
        }
    }
}