Mercurial > hg > octave-nkf

diff src/ov-cx-mat.cc @ 8366:8b1a2555c4e2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
implement diagonal matrix objects * * *
author Jaroslav Hajek <highegg@gmail.com>
date Wed, 03 Dec 2008 13:32:57 +0100
parents c777f3ce02d8
children 25bc2d31e1bf
line wrap: on
line diff
--- a/src/ov-cx-mat.cc
+++ b/src/ov-cx-mat.cc
@@ -308,6 +308,19 @@
   return SparseComplexMatrix (matrix.matrix_value ());
 }
 
+octave_value
+octave_complex_matrix::diag (octave_idx_type k) const
+{
+  octave_value retval;
+  if (k == 0 && matrix.ndims () == 2 
+      && (matrix.rows () == 1 || matrix.columns () == 1))
+    retval = ComplexDiagMatrix (DiagArray2<Complex> (matrix));
+  else
+    retval = octave_base_matrix<ComplexNDArray>::diag (k);
+
+  return retval;
+}
+
 bool 
 octave_complex_matrix::save_ascii (std::ostream& os)
 {