- Added transpose method to matrix class

2 files changed, 43 insertions, 0 deletions

diff --git a/src/math/matrix.php b/src/math/matrix.php
index dae2f64..f64d7e4 100644
--- a/src/math/matrix.php
+++ b/src/math/matrix.php
@@ -228,6 +228,31 @@ class ezcGraphMatrix
         }
     }
 
+    /**
+     * Transpose matrix
+     * 
+     * @return ezcGraphMatrix Transposed matrix
+     */
+    public function transpose()
+    {
+        $matrix = clone $this;
+
+        $this->rows = $matrix->columns();
+        $this->columns = $matrix->rows();
+
+        $this->matrix = array();
+
+        for ( $i = 0; $i < $this->columns; ++$i )
+        {
+            for ( $j = 0; $j < $this->rows; ++$j )
+            {
+                $this->matrix[$i][$j] = $matrix->get( $j, $i );
+            }
+        }
+
+        return $this;
+    }
+
 	/**
      * Multiplies two matrices
      *
diff --git a/tests/matrix_test.php b/tests/matrix_test.php
index 0afd631..79e8afc 100644
--- a/tests/matrix_test.php
+++ b/tests/matrix_test.php
@@ -258,6 +258,24 @@ class ezcGraphMatrixTest extends ezcTestCase
 
         $this->fail( 'Expected ezcGraphMatrixInvalidDimensionsException.' );
     }
+
+    public function testTransposeMatrix()
+    {
+        $matrix = new ezcGraphMatrix( 2, 3, array(
+            array( 1, 2, 3 ),
+            array( 4, 5, 6 ),
+        ) );
+        $matrix->transpose();
+
+        $this->assertEquals(
+            array(
+                array( 1, 4 ),
+                array( 2, 5 ),
+                array( 3, 6 ),
+            ),
+            $this->getAttribute( $matrix, 'matrix' )
+        );
+    }
 }
 
 ?>