MatrixColumnAssignment.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 MatrixColumnAssignment : TernaryOperator
    {
        public MatrixColumnAssignment(ColumnVector MatrixColumn, ScriptNode Operand, int Start, int Length, Expression Expression)
            : base(MatrixColumn.LeftOperand, MatrixColumn.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 column vector assignment can only be performed on matrices.", this);
 
            IElement Index = this.middle.Evaluate(Variables);
 
            if (!(Index.AssociatedObjectValue is double d) || d < 0 || d > int.MaxValue || d != Math.Truncate(d))
                throw new ScriptRuntimeException("Index must be a non-negative integer.", this);
 
            IElement Value = this.right.Evaluate(Variables);
            if (!(Value is IVector V))
                throw new ScriptRuntimeException("Matrix columns must be vectors.", this);
 
            if (M.Rows != V.Dimension)
                throw new ScriptRuntimeException("Vector dimension does not match number of rows in matrix.", this);
 
            M.SetColumn((int)d, V);
 
            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 column vector assignment can only be performed on matrices.", this);
 
            IElement Index = await this.middle.EvaluateAsync(Variables);
 
            if (!(Index.AssociatedObjectValue is double d) || d < 0 || d > int.MaxValue || d != Math.Truncate(d))
                throw new ScriptRuntimeException("Index must be a non-negative integer.", this);
 
            IElement Value = await this.right.EvaluateAsync(Variables);
            if (!(Value is IVector V))
                throw new ScriptRuntimeException("Matrix columns must be vectors.", this);
 
            if (M.Rows != V.Dimension)
                throw new ScriptRuntimeException("Vector dimension does not match number of rows in matrix.", this);
 
            M.SetColumn((int)d, V);
 
            return Value;
        }
 
    }
}