IndexOf.cs

using System;
using System.Collections.Generic;
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.Functions.Vectors
{
    public class IndexOf : FunctionMultiVariate
    {
        public IndexOf(ScriptNode Vector, ScriptNode Item, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Vector, Item }, new ArgumentType[] { ArgumentType.Normal, ArgumentType.Scalar },
                  Start, Length, Expression)
        {
        }
 
        public IndexOf(ScriptNode Vector, ScriptNode Item, ScriptNode From, int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Vector, Item, From }, new ArgumentType[] { ArgumentType.Normal, ArgumentType.Scalar, ArgumentType.Scalar },
                  Start, Length, Expression)
        {
        }
 
        public IndexOf(ScriptNode Matrix, ScriptNode Item, ScriptNode FromColumn, ScriptNode FromRow,
            int Start, int Length, Expression Expression)
            : base(new ScriptNode[] { Matrix, Item, FromColumn, FromRow }, new ArgumentType[] { ArgumentType.Matrix, ArgumentType.Scalar, ArgumentType.Scalar, ArgumentType.Scalar },
                  Start, Length, Expression)
        {
        }
 
        public override string FunctionName => nameof(IndexOf);
 
        public override string[] DefaultArgumentNames
        {
            get { return new string[] { "Vector", "Item" }; }
        }
 
        public override IElement Evaluate(IElement[] Arguments, Variables Variables)
        {
            IElement Vector = Arguments[0];
            IElement Item = Arguments[1];
            IElement From = Arguments.Length > 2 ? Arguments[2] : null;
            IElement From2 = Arguments.Length > 3 ? Arguments[3] : null;
            int FromIndex, FromIndex2;
 
            if (From is null)
                FromIndex = 0;
            else
                FromIndex = (int)Expression.ToDouble(From.AssociatedObjectValue);
 
            if (Vector.AssociatedObjectValue is string s1 && Item.AssociatedObjectValue is string s2)
            {
                if (From is null)
                    return new DoubleNumber(s1.IndexOf(s2));
 
                return new DoubleNumber(s1.IndexOf(s2, FromIndex));
            }
            else if (Vector is IMatrix M)
            {
                if (From is null || From2 is null)
                {
                    if (M.TryFind(Item, out int Column, out int Row))
                        return new DoubleVector(Column, Row);
                    else
                        return new DoubleVector(-1, -1);
                }
                else
                {
                    FromIndex2 = (int)Expression.ToDouble(From2.AssociatedObjectValue);
 
                    if (M.TryFind(Item, FromIndex, FromIndex2, out int Column, out int Row))
                        return new DoubleVector(Column, Row);
                    else
                        return new DoubleVector(-1, -1);
                }
            }
            else
            {
                ICollection<IElement> ChildElements;
 
                if (Vector is IVector V)
                    ChildElements = V.VectorElements;
                else if (Vector is ISet S)
                    ChildElements = S.ChildElements;
                else
                    throw new ScriptRuntimeException("Expected string arguments, or the first argument to be a vector or a set.", this);
 
                if (ChildElements is IElement[] V2)
                    return new DoubleNumber(Array.IndexOf(V2, Item));
 
                int i = 0;
 
                foreach (IElement Element in ChildElements)
                {
                    if (i >= FromIndex)
                    {
                        if (Item.Equals(Element))
                            return new DoubleNumber(i);
                    }
 
                    i++;
                }
 
                return new DoubleNumber(-1);
            }
        }
 
    }
}