ComplexMatrix.cs

using System;
using System.Collections.Generic;
using System.Numerics;
using System.Text;
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.VectorSpaces;
using Waher.Script.Operators.Matrices;
 
namespace Waher.Script.Objects.Matrices
{
    public sealed class ComplexMatrix : RingElement, IMatrix
    {
        private Complex[,] values;
        private IElement[,] matrixElements;
        private ICollection<IElement> elements;
        private readonly int rows;
        private readonly int columns;
 
        public ComplexMatrix(Complex[,] Values)
        {
            this.values = Values;
            this.elements = null;
            this.matrixElements = null;
            this.rows = Values.GetLength(0);
            this.columns = Values.GetLength(1);
        }
 
        public ComplexMatrix(int Rows, int Columns, ICollection<IElement> Elements)
        {
            this.values = null;
            this.elements = Elements;
            this.matrixElements = null;
            this.rows = Rows;
            this.columns = Columns;
        }
 
        public Complex[,] Values
        {
            get
            {
                if (this.values is null)
                {
                    Complex[,] v = new Complex[this.rows, this.columns];
                    int x = 0;
                    int y = 0;
 
                    if (this.elements is ChunkedList<IElement> Values)
                    {
                        ChunkNode<IElement> Loop = Values.FirstChunk;
                        int i, c;
 
                        while (!(Loop is null))
                        {
                            for (i = Loop.Start, c = Loop.Pos; i < c; i++)
                            {
                                if (!(Loop[i].AssociatedObjectValue is Complex z))
                                    z = 0;
 
                                v[y, x++] = z;
                                if (x >= this.columns)
                                {
                                    y++;
                                    x = 0;
                                }
                            }
 
                            Loop = Loop.Next;
                        }
                    }
                    else
                    {
                        foreach (IElement E in this.elements)
                        {
                            if (!(E.AssociatedObjectValue is Complex z))
                                z = 0;
 
                            v[y, x++] = z;
                            if (x >= this.columns)
                            {
                                y++;
                                x = 0;
                            }
                        }
                    }
 
                    this.values = v;
                }
 
                return this.values;
            }
        }
 
        public ICollection<IElement> Elements
        {
            get
            {
                if (this.elements is null)
                {
                    int x, y, i = 0;
                    IElement[] v = new IElement[this.rows * this.columns];
 
                    for (y = 0; y < this.rows; y++)
                    {
                        for (x = 0; x < this.columns; x++)
                            v[i++] = new ComplexNumber(this.values[y, x]);
                    }
 
                    this.elements = v;
                }
 
                return this.elements;
            }
        }
 
        public IElement[,] MatrixElements
        {
            get
            {
                if (this.matrixElements is null)
                {
                    IElement[,] v = new IElement[this.rows, this.columns];
                    int x = 0;
                    int y = 0;
 
                    foreach (IElement E in this.Elements)
                    {
                        v[y, x++] = E;
                        if (x >= this.columns)
                        {
                            y++;
                            x = 0;
                        }
                    }
 
                    this.matrixElements = v;
                }
 
                return this.matrixElements;
            }
        }
 
        public int Rows => this.rows;
 
        public int Columns => this.columns;
 
        public override string ToString()
        {
            Complex[,] v = this.Values;
            StringBuilder sb = null;
            bool First;
            int x, y;
 
            for (y = 0; y < this.rows; y++)
            {
                if (sb is null)
                    sb = new StringBuilder("[[");
                else
                    sb.Append(",\r\n [");
 
                First = true;
                for (x = 0; x < this.columns; x++)
                {
                    if (First)
                        First = false;
                    else
                        sb.Append(", ");
 
                    sb.Append(Expression.ToString(v[y, x]));
                }
 
                sb.Append(']');
            }
 
            if (sb is null)
                sb = new StringBuilder("[[]]");
            else
                sb.Append(']');
 
            return sb.ToString();
        }
 
        public override IRing AssociatedRing
        {
            get
            {
                if (this.associatedMatrixSpace is null)
                    this.associatedMatrixSpace = new ComplexMatrices(this.rows, this.columns);
 
                return this.associatedMatrixSpace;
            }
        }
 
        private ComplexMatrices associatedMatrixSpace = null;
 
        public override object AssociatedObjectValue => this;
 
        public override IRingElement MultiplyLeft(IRingElement Element)
        {
            Complex[,] v;
            int x, y, z;
 
            if (Element.AssociatedObjectValue is Complex n)
            {
                Complex[,] Values = this.Values;
 
                v = new Complex[this.rows, this.columns];
 
                for (y = 0; y < this.rows; y++)
                {
                    for (x = 0; x < this.columns; x++)
                        v[y, x] = n * Values[y, x];
                }
 
                return new ComplexMatrix(v);
            }
            else if (Element is ComplexMatrix Matrix)
            {
                if (Matrix.columns != this.rows)
                    return null;
 
                Complex[,] Values = this.Values;
                Complex[,] Values2 = Matrix.Values;
 
                v = new Complex[Matrix.rows, this.columns];
                for (y = 0; y < Matrix.rows; y++)
                {
                    for (x = 0; x < this.columns; x++)
                    {
                        n = 0;
 
                        for (z = 0; z < this.rows; z++)
                            n += Values2[y, z] * Values[z, x];
 
                        v[y, x] = n;
                    }
                }
 
                return new ComplexMatrix(v);
            }
            else if (Element is IMatrix)
                return new ObjectMatrix(this.MatrixElements).MultiplyLeft(Element);
            else
                return null;
        }
 
        public override IRingElement MultiplyRight(IRingElement Element)
        {
            Complex[,] v;
            int x, y, z;
 
            if (Element.AssociatedObjectValue is Complex n)
            {
                Complex[,] Values = this.Values;
 
                v = new Complex[this.rows, this.columns];
 
                for (y = 0; y < this.rows; y++)
                {
                    for (x = 0; x < this.columns; x++)
                        v[y, x] = n * Values[y, x];
                }
 
                return new ComplexMatrix(v);
            }
            else if (Element is ComplexMatrix Matrix)
            {
                if (this.columns != Matrix.rows)
                    return null;
 
                Complex[,] Values = this.Values;
                Complex[,] Values2 = Matrix.Values;
 
                v = new Complex[this.rows, Matrix.columns];
                for (y = 0; y < this.rows; y++)
                {
                    for (x = 0; x < Matrix.columns; x++)
                    {
                        n = 0;
 
                        for (z = 0; z < this.columns; z++)
                            n += Values[y, z] * Values2[z, x];
 
                        v[y, x] = n;
                    }
                }
 
                return new ComplexMatrix(v);
            }
            else if (Element is IMatrix)
                return new ObjectMatrix(this.MatrixElements).MultiplyRight(Element);
            else
                return null;
        }
 
        public override IRingElement Invert()
        {
            if (this.rows != this.columns)
                return null;
 
            Complex[,] Values = this.Values;
            int c2 = this.columns << 1;
            Complex[,] v = new Complex[this.rows, c2];
            int x, y;
 
            for (y = 0; y < this.rows; y++)
            {
                for (x = 0; x < this.columns; x++)
                {
                    v[y, x] = Values[y, x];
                    v[y, x + this.columns] = (x == y ? 1 : 0);
                }
            }
 
            if (Reduce(v, true, true, out _) < 0)
                return null;
 
            Complex[,] v2 = new Complex[this.rows, this.columns];
 
            for (y = 0; y < this.rows; y++)
            {
                for (x = 0; x < this.columns; x++)
                    v2[y, x] = v[y, x + this.columns];
            }
 
            return new ComplexMatrix(v2);
        }
 
        public IMatrix Reduce(bool Eliminate, bool BreakIfZero, out int Rank, out ICommutativeRingWithIdentityElement Factor)
        {
            Complex[,] M = (Complex[,])this.Values.Clone();
            Rank = Reduce(M, Eliminate, BreakIfZero, out Complex c);
            Factor = new ComplexNumber(c);
            return new ComplexMatrix(M);
        }
 
        public static int Reduce(Complex[,] Matrix, bool Eliminate, bool BreakIfZero, 
            out Complex Factor)
        {
            int x, y, u, z;
            int Rows = Matrix.GetLength(0);
            int Columns = Matrix.GetLength(1);
            int MinCount = Math.Min(Rows, Columns);
            double a, b;
            Complex w;
            int Rank = 0;
 
            Factor = 1;
 
            for (x = 0; x < MinCount; x++)
            {
                a = Matrix[x, x].Magnitude;
                z = x;
                for (y = x + 1; y < Rows; y++)
                {
                    b = Matrix[y, x].Magnitude;
                    if (b > a)
                    {
                        a = b;
                        z = y;
                    }
                }
 
                if (z != x)
                {
                    for (u = x; u < Columns; u++)
                    {
                        w = Matrix[x, u];
                        Matrix[x, u] = Matrix[z, u];
                        Matrix[z, u] = w;
                    }
 
                    Factor = -Factor;
                }
 
                w = Matrix[x, x];
                if (w == 0)
                {
                    if (BreakIfZero)
                        return -1;
                }
                else
                {
                    Rank++;
 
                    if (w != 1)
                    {
                        for (u = x; u < Columns; u++)
                            Matrix[x, u] /= w;
 
                        Factor *= w;
                    }
 
                    for (y = Eliminate ? 0 : x + 1; y < Rows; y++)
                    {
                        if (y != x && (w = Matrix[y, x]) != 0)
                        {
                            for (u = x; u < Columns; u++)
                                Matrix[y, u] -= w * Matrix[x, u];
                        }
                    }
                }
            }
 
            return Rank;
        }
 
        public override IAbelianGroupElement Add(IAbelianGroupElement Element)
        {
            Complex[,] v;
            int x, y;
 
            if (Element.AssociatedObjectValue is Complex n)
            {
                Complex[,] Values = this.Values;
 
                v = new Complex[this.rows, this.columns];
 
                for (y = 0; y < this.rows; y++)
                {
                    for (x = 0; x < this.columns; x++)
                        v[y, x] = n + Values[y, x];
                }
 
                return new ComplexMatrix(v);
            }
            else if (Element is ComplexMatrix Matrix)
            {
                if (this.columns != Matrix.columns || this.rows != Matrix.rows)
                    return null;
 
                Complex[,] Values = this.Values;
                Complex[,] Values2 = Matrix.Values;
 
                v = new Complex[this.rows, this.columns];
                for (y = 0; y < this.rows; y++)
                {
                    for (x = 0; x < this.columns; x++)
                        v[y, x] = Values[y, x] + Values2[y, x];
                }
 
                return new ComplexMatrix(v);
            }
            else if (Element is IMatrix)
                return new ObjectMatrix(this.MatrixElements).Add(Element);
            else
                return null;
        }
 
        public override IGroupElement Negate()
        {
            Complex[,] Values = this.Values;
            Complex[,] v = new Complex[this.rows, this.columns];
            int x, y;
 
            for (y = 0; y < this.rows; y++)
            {
                for (x = 0; x < this.columns; x++)
                    v[y, x] = -Values[y, x];
            }
 
            return new ComplexMatrix(v);
        }
 
        public override bool Equals(object obj)
        {
            if (obj is ComplexMatrix Matrix)
            {
                if (this.columns != Matrix.columns || this.rows != Matrix.rows)
                    return false;
 
                Complex[,] V1 = this.Values;
                Complex[,] V2 = Matrix.Values;
                int x, y;
 
                for (y = 0; y < this.rows; y++)
                {
                    for (x = 0; x < this.columns; x++)
                    {
                        if (V1[y, x] != V2[y, x])
                            return false;
                    }
                }
 
                return true;
            }
            else if (obj is IMatrix)
                return new ObjectMatrix(this.MatrixElements).Equals(obj);
            else
                return false;
        }
 
        public override int GetHashCode()
        {
            Complex[,] Values = this.Values;
            int Result = 0;
            int x, y;
 
            for (y = 0; y < this.rows; y++)
            {
                for (x = 0; x < this.columns; x++)
                    Result ^= Values[y, x].GetHashCode();
            }
 
            return Result;
        }
 
        public override bool IsScalar => false;
 
        public override ICollection<IElement> ChildElements => this.Elements;
 
        public override IElement Encapsulate(ChunkedList<IElement> Elements, ScriptNode Node)
        {
            return MatrixDefinition.Encapsulate(Elements, this.rows, this.columns, Node);
        }
 
        public override IElement Encapsulate(ICollection<IElement> Elements, ScriptNode Node)
        {
            return MatrixDefinition.Encapsulate(Elements, this.rows, this.columns, Node);
        }
 
        public override IAbelianGroupElement Zero
        {
            get
            {
                if (this.zero is null)
                    this.zero = new ComplexMatrix(new Complex[this.rows, this.columns]);
 
                return this.zero;
            }
        }
 
        private ComplexMatrix zero = null;
 
        public int Dimension => this.rows;
 
        public ICollection<IElement> VectorElements
        {
            get
            {
                if (!(this.rowVectors is null))
                    return this.rowVectors;
 
                Complex[,] v = this.Values;
                ChunkedList<IElement> Rows = new ChunkedList<IElement>();
                int x, y;
                Complex[] r;
 
                for (y = 0; y < this.rows; y++)
                {
                    r = new Complex[this.columns];
 
                    for (x = 0; x < this.columns; x++)
                        r[x] = v[y, x];
 
                    Rows.Add(new ComplexVector(r));
                }
 
                this.rowVectors = Rows;
                return Rows;
            }
        }
 
        private ChunkedList<IElement> rowVectors = null;
 
        public IMatrix Transpose()
        {
            Complex[,] v = new Complex[this.columns, this.rows];
            Complex[,] Values = this.Values;
            int x, y;
 
            for (y = 0; y < this.rows; y++)
            {
                for (x = 0; x < this.columns; x++)
                    v[x, y] = Values[y, x];
            }
 
            return new ComplexMatrix(v);
        }
 
        public IMatrix ConjugateTranspose()
        {
            Complex[,] v = new Complex[this.columns, this.rows];
            Complex[,] Values = this.Values;
            int x, y;
 
            for (y = 0; y < this.rows; y++)
            {
                for (x = 0; x < this.columns; x++)
                    v[x, y] = Complex.Conjugate(Values[y, x]);
            }
 
            return new ComplexMatrix(v);
        }
 
        public IElement GetElement(int Index)
        {
            if (Index < 0 || Index >= this.rows)
                throw new ScriptException("Index out of bounds.");
 
            Complex[,] M = this.Values;
            Complex[] V = new Complex[this.columns];
            int i;
 
            for (i = 0; i < this.columns; i++)
                V[i] = M[Index, i];
 
            return new ComplexVector(V);
        }
 
        public void SetElement(int Index, IElement Value)
        {
            if (Index < 0 || Index >= this.rows)
                throw new ScriptException("Index out of bounds.");
 
            if (!(Value is ComplexVector V))
                throw new ScriptException("Row vectors in a Complex matrix are required to be Complex vectors.");
 
            if (V.Dimension != this.columns)
                throw new ScriptException("Dimension mismatch.");
 
            Complex[] V2 = V.Values;
            Complex[,] M = this.Values;
            this.elements = null;
 
            int i;
 
            for (i = 0; i < this.columns; i++)
                M[Index, i] = V2[i];
        }
 
        public IElement GetElement(int Column, int Row)
        {
            if (Column < 0 || Column >= this.columns || Row < 0 || Row >= this.rows)
                throw new ScriptException("Index out of bounds.");
 
            return new ComplexNumber(this.Values[Row, Column]);
        }
 
        public void SetElement(int Column, int Row, IElement Value)
        {
            if (Column < 0 || Column >= this.columns || Row < 0 || Row >= this.rows)
                throw new ScriptException("Index out of bounds.");
 
            if (!(Value.AssociatedObjectValue is Complex V))
                throw new ScriptException("Elements in a Complex matrix must be Complex values.");
 
            Complex[,] M = this.Values;
            this.elements = null;
 
            M[Row, Column] = V;
        }
 
        public IVector GetRow(int Row)
        {
            if (Row < 0 || Row >= this.rows)
                throw new ScriptException("Index out of bounds.");
 
            Complex[,] M = this.Values;
            Complex[] V = new Complex[this.columns];
            int i;
 
            for (i = 0; i < this.columns; i++)
                V[i] = M[Row, i];
 
            return new ComplexVector(V);
        }
 
        public IVector GetColumn(int Column)
        {
            if (Column < 0 || Column >= this.columns)
                throw new ScriptException("Index out of bounds.");
 
            Complex[,] M = this.Values;
            Complex[] V = new Complex[this.rows];
            int i;
 
            for (i = 0; i < this.rows; i++)
                V[i] = M[i, Column];
 
            return new ComplexVector(V);
        }
 
        public void SetRow(int Row, IVector Vector)
        {
            if (Row < 0 || Row >= this.rows)
                throw new ScriptException("Index out of bounds.");
 
            if (Vector.Dimension != this.columns)
                throw new ScriptException("Vector dimension does not match number of columns");
 
            if (!(Vector is ComplexVector V))
                throw new ScriptException("Row vectors in a Complex matrix must be Complex vectors.");
 
            Complex[] V2 = V.Values;
            Complex[,] M = this.Values;
            this.elements = null;
            int i;
 
            for (i = 0; i < this.columns; i++)
                M[Row, i] = V2[i];
        }
 
        public void SetColumn(int Column, IVector Vector)
        {
            if (Column < 0 || Column >= this.columns)
                throw new ScriptException("Index out of bounds.");
 
            if (Vector.Dimension != this.rows)
                throw new ScriptException("Vector dimension does not match number of rows");
 
            if (!(Vector is ComplexVector V))
                throw new ScriptException("Column vectors in a Complex matrix must be Complex vectors.");
 
            Complex[] V2 = V.Values;
            Complex[,] M = this.Values;
            this.elements = null;
            int i;
 
            for (i = 0; i < this.rows; i++)
                M[i, Column] = V2[i];
        }
 
        public bool TryFind(IElement Element, out int Column, out int Row)
        {
            return this.TryFind(Element, 0, 0, out Column, out Row);
        }
 
        public bool TryFind(IElement Element, int FromColumn, int FromRow, out int Column, out int Row)
        {
            if (Element is ComplexNumber Z)
                return this.TryFind(Z.Value, FromColumn, FromRow, out Column, out Row);
            else
            {
                Column = -1;
                Row = -1;
                return false;
            }
        }
 
        public bool TryFind(Complex Element, int FromColumn, int FromRow, out int Column, out int Row)
        {
            Complex[,] Values = this.Values;
 
            while (FromRow < this.rows)
            {
                while (FromColumn < this.columns)
                {
                    if (Values[FromRow, FromColumn] == Element)
                    {
                        Column = FromColumn;
                        Row = FromRow;
 
                        return true;
                    }
 
                    FromColumn++;
                }
 
                FromColumn = 0;
                FromRow++;
            }
 
            Column = -1;
            Row = -1;
 
            return false;
        }
 
        public bool TryFindLast(IElement Element, out int Column, out int Row)
        {
            return this.TryFindLast(Element, this.columns - 1, this.rows - 1, out Column, out Row);
        }
 
        public bool TryFindLast(IElement Element, int FromColumn, int FromRow, out int Column, out int Row)
        {
            if (Element is ComplexNumber Z)
                return this.TryFindLast(Z.Value, FromColumn, FromRow, out Column, out Row);
            else
            {
                Column = -1;
                Row = -1;
                return false;
            }
        }
 
        public bool TryFindLast(Complex Element, int FromColumn, int FromRow, out int Column, out int Row)
        {
            Complex[,] Values = this.Values;
 
            while (FromRow >= 0)
            {
                while (FromColumn >= 0)
                {
                    if (Values[FromRow, FromColumn] == Element)
                    {
                        Column = FromColumn;
                        Row = FromRow;
 
                        return true;
                    }
 
                    FromColumn--;
                }
 
                FromColumn = this.columns - 1;
                FromRow--;
            }
 
            Column = -1;
            Row = -1;
 
            return false;
        }
 
    }
}