BinaryScalarOperator.cs

using System.Collections.Generic;
using System.Threading.Tasks;
using Waher.Runtime.Collections;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Abstraction.Sets;
using Waher.Script.Exceptions;
 
namespace Waher.Script.Model
{
    public enum UpgradeBehaviour
    {
        DifferentTypesOk,
 
        UpgradeIfPossble,
 
        SameTypeRequired
    }
 
    public abstract class BinaryScalarOperator : BinaryOperator
    {
        public BinaryScalarOperator(ScriptNode Left, ScriptNode Right, int Start, int Length, Expression Expression)
            : base(Left, Right, Start, Length, Expression)
        {
        }
 
        public override IElement Evaluate(Variables Variables)
        {
            IElement Left = this.left.Evaluate(Variables);
            IElement Right = this.right.Evaluate(Variables);
 
            return this.Evaluate(Left, Right, Variables);
        }
 
        public override async Task<IElement> EvaluateAsync(Variables Variables)
        {
            if (!this.isAsync)
                return this.Evaluate(Variables);
 
            IElement Left = await this.left.EvaluateAsync(Variables);
            IElement Right = await this.right.EvaluateAsync(Variables);
 
            return await this.EvaluateAsync(Left, Right, Variables);
        }
 
        public virtual IElement Evaluate(IElement Left, IElement Right, Variables Variables)
        {
            if (Left.IsScalar)
            {
                if (Right.IsScalar)
                {
                    ISet LeftSet = Left.AssociatedSet;
                    ISet RightSet = Right.AssociatedSet;
                    UpgradeBehaviour UpgradeBehaviour;
 
                    if (!LeftSet.Equals(RightSet) && (UpgradeBehaviour = this.ScalarUpgradeBehaviour) != UpgradeBehaviour.DifferentTypesOk)
                    {
                        if (!Expression.UpgradeField(ref Left, ref LeftSet, ref Right, ref RightSet))
                        {
                            if (UpgradeBehaviour == UpgradeBehaviour.SameTypeRequired)
                                throw new ScriptRuntimeException("Incompatible operands.", this);
                        }
                    }
 
                    return this.EvaluateScalar(Left, Right, Variables);
                }
                else
                {
                    ChunkedList<IElement> Result = new ChunkedList<IElement>();
 
                    foreach (IElement RightChild in Right.ChildElements)
                        Result.Add(this.Evaluate(Left, RightChild, Variables));
 
                    return Right.Encapsulate(Result, this);
                }
            }
            else
            {
                if (Right.IsScalar)
                {
                    ChunkedList<IElement> Result = new ChunkedList<IElement>();
 
                    foreach (IElement LeftChild in Left.ChildElements)
                        Result.Add(this.Evaluate(LeftChild, Right, Variables));
 
                    return Left.Encapsulate(Result, this);
                }
                else
                {
                    ICollection<IElement> LeftChildren = Left.ChildElements;
                    ICollection<IElement> RightChildren = Right.ChildElements;
 
                    if (LeftChildren.Count == RightChildren.Count)
                    {
                        ChunkedList<IElement> Result = new ChunkedList<IElement>();
                        IEnumerator<IElement> eLeft = LeftChildren.GetEnumerator();
                        IEnumerator<IElement> eRight = RightChildren.GetEnumerator();
 
                        try
                        {
                            while (eLeft.MoveNext() && eRight.MoveNext())
                                Result.Add(this.Evaluate(eLeft.Current, eRight.Current, Variables));
                        }
                        finally
                        {
                            eLeft.Dispose();
                            eRight.Dispose();
                        }
 
                        return Left.Encapsulate(Result, this);
                    }
                    else
                    {
                        ChunkedList<IElement> LeftResult = new ChunkedList<IElement>();
 
                        foreach (IElement LeftChild in LeftChildren)
                        {
                            ChunkedList<IElement> RightResult = new ChunkedList<IElement>();
 
                            foreach (IElement RightChild in RightChildren)
                                RightResult.Add(this.Evaluate(LeftChild, RightChild, Variables));
 
                            LeftResult.Add(Right.Encapsulate(RightResult, this));
                        }
 
                        return Left.Encapsulate(LeftResult, this);
                    }
                }
            }
        }
 
        public virtual async Task<IElement> EvaluateAsync(IElement Left, IElement Right, Variables Variables)
        {
            if (Left.IsScalar)
            {
                if (Right.IsScalar)
                {
                    ISet LeftSet = Left.AssociatedSet;
                    ISet RightSet = Right.AssociatedSet;
                    UpgradeBehaviour UpgradeBehaviour;
 
                    if (!LeftSet.Equals(RightSet) && (UpgradeBehaviour = this.ScalarUpgradeBehaviour) != UpgradeBehaviour.DifferentTypesOk)
                    {
                        if (!Expression.UpgradeField(ref Left, ref LeftSet, ref Right, ref RightSet))
                        {
                            if (UpgradeBehaviour == UpgradeBehaviour.SameTypeRequired)
                                throw new ScriptRuntimeException("Incompatible operands.", this);
                        }
                    }
 
                    return await this.EvaluateScalarAsync(Left, Right, Variables);
                }
                else
                {
                    ChunkedList<IElement> Result = new ChunkedList<IElement>();
 
                    foreach (IElement RightChild in Right.ChildElements)
                        Result.Add(await this.EvaluateAsync(Left, RightChild, Variables));
 
                    return Right.Encapsulate(Result, this);
                }
            }
            else
            {
                if (Right.IsScalar)
                {
                    ChunkedList<IElement> Result = new ChunkedList<IElement>();
 
                    foreach (IElement LeftChild in Left.ChildElements)
                        Result.Add(await this.EvaluateAsync(LeftChild, Right, Variables));
 
                    return Left.Encapsulate(Result, this);
                }
                else
                {
                    ICollection<IElement> LeftChildren = Left.ChildElements;
                    ICollection<IElement> RightChildren = Right.ChildElements;
 
                    if (LeftChildren.Count == RightChildren.Count)
                    {
                        ChunkedList<IElement> Result = new ChunkedList<IElement>();
                        IEnumerator<IElement> eLeft = LeftChildren.GetEnumerator();
                        IEnumerator<IElement> eRight = RightChildren.GetEnumerator();
 
                        try
                        {
                            while (eLeft.MoveNext() && eRight.MoveNext())
                                Result.Add(await this.EvaluateAsync(eLeft.Current, eRight.Current, Variables));
                        }
                        finally
                        {
                            eLeft.Dispose();
                            eRight.Dispose();
                        }
 
                        return Left.Encapsulate(Result, this);
                    }
                    else
                    {
                        ChunkedList<IElement> LeftResult = new ChunkedList<IElement>();
 
                        foreach (IElement LeftChild in LeftChildren)
                        {
                            ChunkedList<IElement> RightResult = new ChunkedList<IElement>();
 
                            foreach (IElement RightChild in RightChildren)
                                RightResult.Add(await this.EvaluateAsync(LeftChild, RightChild, Variables));
 
                            LeftResult.Add(Right.Encapsulate(RightResult, this));
                        }
 
                        return Left.Encapsulate(LeftResult, this);
                    }
                }
            }
        }
 
        public abstract IElement EvaluateScalar(IElement Left, IElement Right, Variables Variables);
 
        public virtual Task<IElement> EvaluateScalarAsync(IElement Left, IElement Right, Variables Variables)
        {
            return Task.FromResult<IElement>(this.EvaluateScalar(Left, Right, Variables));
        }
 
        public virtual UpgradeBehaviour ScalarUpgradeBehaviour => UpgradeBehaviour.SameTypeRequired;
    }
}