VectorDefinition.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Numerics;
using System.Threading.Tasks;
using Waher.Runtime.Collections;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Abstraction.Sets;
using Waher.Script.Exceptions;
using Waher.Script.Model;
using Waher.Script.Objects;
using Waher.Script.Objects.VectorSpaces;
 
namespace Waher.Script.Operators.Vectors
{
    public class VectorDefinition : ElementList
    {
        public VectorDefinition(ScriptNode[] Elements, int Start, int Length, Expression Expression)
            : base(Elements, Start, Length, Expression)
        {
        }
 
        public override IElement Evaluate(Variables Variables)
        {
            ChunkedList<IElement> VectorElements = new ChunkedList<IElement>();
 
            foreach (ScriptNode Node in this.Elements)
            {
                if (Node is null)
                    VectorElements.Add(ObjectValue.Null);
                else
                    VectorElements.Add(Node.Evaluate(Variables));
            }
 
            return Encapsulate(VectorElements, true, this);
        }
 
        public override async Task<IElement> EvaluateAsync(Variables Variables)
        {
            if (!this.isAsync)
                return this.Evaluate(Variables);
 
            ChunkedList<IElement> VectorElements = new ChunkedList<IElement>();
 
            foreach (ScriptNode Node in this.Elements)
            {
                if (Node is null)
                    VectorElements.Add(ObjectValue.Null);
                else
                    VectorElements.Add(await Node.EvaluateAsync(Variables));
            }
 
            return Encapsulate(VectorElements, true, this);
        }
 
        public static IElement Encapsulate(Array Elements, bool CanEncapsulateAsMatrix, ScriptNode Node)
        {
            if (Elements is double[] dv)
                return new DoubleVector(dv);
            else if (Elements is bool[] bv)
                return new BooleanVector(bv);
            else if (Elements is Complex[] zv)
                return new ComplexVector(zv);
            else if (Elements is DateTime[] dtv)
                return new DateTimeVector(dtv);
            else if (Elements is byte[] Bin)
                return new ObjectValue(Bin);
            else
            {
                ChunkedList<IElement> Elements2 = new ChunkedList<IElement>();
 
                foreach (object Obj in Elements)
                    Elements2.Add(Expression.Encapsulate(Obj));
 
                return Encapsulate(Elements2, CanEncapsulateAsMatrix, Node);
            }
        }
 
        public static IElement Encapsulate(IEnumerable Elements, bool CanEncapsulateAsMatrix, ScriptNode Node)
        {
            ChunkedList<IElement> Elements2 = new ChunkedList<IElement>();
 
            foreach (object Obj in Elements)
                Elements2.Add(Expression.Encapsulate(Obj));
 
            return Encapsulate(Elements2, CanEncapsulateAsMatrix, Node);
        }
 
        public static IVector Encapsulate(IElement[] Elements, bool CanEncapsulateAsMatrix, ScriptNode Node)
        {
            return Encapsulate((ICollection<IElement>)Elements, CanEncapsulateAsMatrix, Node);
        }
 
        public static IVector Encapsulate(IEnumerable<object> Elements, bool CanEncapsulateAsMatrix, ScriptNode Node)
        {
            ChunkedList<IElement> Elements2 = new ChunkedList<IElement>();
 
            foreach (object Obj in Elements)
                Elements2.Add(Expression.Encapsulate(Obj));
 
            return Encapsulate(Elements2, CanEncapsulateAsMatrix, Node);
        }
 
        public static IVector Encapsulate(ICollection<IElement> Elements, bool CanEncapsulateAsMatrix, ScriptNode Node)
        {
            IElement SuperSetExample = null;
            IElement Element2;
            ISet CommonSuperSet = null;
            IVectorSpaceElement Vector;
            ISet Set;
            int? Columns = null;
            bool Upgraded = false;
            bool SameDimensions = true;
 
            foreach (IElement Element in Elements)
            {
                if (CanEncapsulateAsMatrix && SameDimensions)
                {
                    Vector = Element as IVectorSpaceElement;
                    if (Vector is null)
                        SameDimensions = false;
                    else
                    {
                        if (!Columns.HasValue)
                            Columns = Vector.Dimension;
                        else if (Columns.Value != Vector.Dimension)
                            SameDimensions = false;
                    }
                }
 
                if (CommonSuperSet is null)
                {
                    SuperSetExample = Element;
 
                    if (Element is null)
                        CommonSuperSet = new ObjectValues();
                    else
                        CommonSuperSet = Element.AssociatedSet;
                }
                else
                {
                    if (Element is null)
                        Set = new ObjectValues();
                    else
                        Set = Element.AssociatedSet;
 
                    if (!Set.Equals(CommonSuperSet))
                    {
                        Element2 = Element;
                        if (!Expression.UpgradeField(ref Element2, ref Set, ref SuperSetExample, ref CommonSuperSet))
                        {
                            CommonSuperSet = null;
                            break;
                        }
                        else
                            Upgraded = true;
                    }
                }
            }
 
            if (CanEncapsulateAsMatrix && SameDimensions && Columns.HasValue)
            {
                IMatrix M = Matrices.MatrixDefinition.Encapsulate(Elements, Node);
                if (M is IVector V)
                    return V;
                else
                    throw new ScriptRuntimeException("Unable to convert matrix to vector.", Node);
            }
 
            if (!(CommonSuperSet is null))
            {
                if (Upgraded)
                {
                    ChunkedList<IElement> SuperElements = new ChunkedList<IElement>();
 
                    foreach (IElement Element in Elements)
                    {
                        if (Element is null)
                            Set = new ObjectValues();
                        else
                            Set = Element.AssociatedSet;
 
                        if (Set.Equals(CommonSuperSet))
                            SuperElements.Add(Element);
                        else
                        {
                            Element2 = Element;
                            if (Expression.UpgradeField(ref Element2, ref Set, ref SuperSetExample, ref CommonSuperSet))
                                SuperElements.Add(Element2);
                            else
                            {
                                SuperElements = null;
                                CommonSuperSet = null;
                                break;
                            }
                        }
                    }
 
                    if (!(SuperElements is null))
                        Elements = SuperElements;
                }
 
                if (!(CommonSuperSet is null))
                {
                    if (CommonSuperSet is DoubleNumbers)
                        return new DoubleVector(Elements);
                    else if (CommonSuperSet is ComplexNumbers)
                        return new ComplexVector(Elements);
                    else if (CommonSuperSet is BooleanValues)
                        return new BooleanVector(Elements);
                    else if (CommonSuperSet is DateTimeValues)
                        return new DateTimeVector(Elements);
                }
            }
 
            return new ObjectVector(Elements);
        }
 
        public override PatternMatchResult PatternMatch(IElement CheckAgainst, Dictionary<string, IElement> AlreadyFound)
        {
            ScriptNode[] Elements = this.Elements;
 
            if (!(CheckAgainst is IVector Vector) || Vector.Dimension != Elements.Length)
                return PatternMatchResult.NoMatch;
 
            PatternMatchResult Result;
            int i = 0;
 
            foreach (IElement E in Vector.VectorElements)
            {
                Result = Elements[i++].PatternMatch(E, AlreadyFound);
                if (Result != PatternMatchResult.Match)
                    return Result;
            }
 
            return PatternMatchResult.Match;
        }
 
    }
}