MatrixIndexAssignment.cs

using System;
using System.Threading.Tasks;
using Waher.Script.Abstraction.Elements;
using Waher.Script.Exceptions;
using Waher.Script.Model;
using Waher.Script.Objects;
using Waher.Script.Operators.Matrices;
 
namespace Waher.Script.Operators.Assignments
{
    public class MatrixIndexAssignment : QuaternaryOperator
    {
        public MatrixIndexAssignment(MatrixIndex MatrixIndex, ScriptNode Operand, int Start, int Length, Expression Expression)
            : base(MatrixIndex.LeftOperand, MatrixIndex.MiddleOperand, MatrixIndex.RightOperand, Operand, Start, Length, Expression)
        {
        }
 
        public override IElement Evaluate(Variables Variables)
        {
            IElement Left = this.left.Evaluate(Variables);
            if (!(Left is IMatrix M))
                throw new ScriptRuntimeException("Matrix element assignment can only be performed on matrices.", this);
 
            IElement ColIndex = this.middle.Evaluate(Variables);
            IElement RowIndex = this.middle2.Evaluate(Variables);
 
            if (!(ColIndex.AssociatedObjectValue is double x) || x < 0 || x > int.MaxValue || x != Math.Truncate(x) ||
                !(RowIndex.AssociatedObjectValue is double y) || y < 0 || y > int.MaxValue || y != Math.Truncate(y))
            {
                throw new ScriptRuntimeException("Indices must be non-negative integers.", this);
            }
 
            IElement Value = this.right.Evaluate(Variables);
 
            M.SetElement((int)x, (int)y, Value);
 
            return Value;
        }
 
        public override async Task<IElement> EvaluateAsync(Variables Variables)
        {
            if (!this.isAsync)
                return this.Evaluate(Variables);
 
            IElement Left = await this.left.EvaluateAsync(Variables);
            if (!(Left is IMatrix M))
                throw new ScriptRuntimeException("Matrix element assignment can only be performed on matrices.", this);
 
            IElement ColIndex = await this.middle.EvaluateAsync(Variables);
            IElement RowIndex = await this.middle2.EvaluateAsync(Variables);
 
            if (!(ColIndex.AssociatedObjectValue is double x) || x < 0 || x > int.MaxValue || x != Math.Truncate(x) ||
                !(RowIndex.AssociatedObjectValue is double y) || y < 0 || y > int.MaxValue || y != Math.Truncate(y))
            {
                throw new ScriptRuntimeException("Indices must be non-negative integers.", this);
            }
 
            IElement Value = await this.right.EvaluateAsync(Variables);
 
            M.SetElement((int)x, (int)y, Value);
 
            return Value;
        }
 
    }
}