Go to the documentation of this file.00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026 #if !defined (octave_DiagArray2_h)
00027 #define octave_DiagArray2_h 1
00028
00029 #include <cassert>
00030 #include <cstdlib>
00031
00032 #include "Array.h"
00033
00034
00035
00036
00037 template <class T>
00038 class
00039 DiagArray2 : protected Array<T>
00040 {
00041 protected:
00042 octave_idx_type d1, d2;
00043
00044 public:
00045
00046 using Array<T>::element_type;
00047
00048 DiagArray2 (void)
00049 : Array<T> (), d1 (0), d2 (0) { }
00050
00051 DiagArray2 (octave_idx_type r, octave_idx_type c)
00052 : Array<T> (dim_vector (std::min (r, c), 1)), d1 (r), d2 (c) { }
00053
00054 DiagArray2 (octave_idx_type r, octave_idx_type c, const T& val)
00055 : Array<T> (dim_vector (std::min (r, c), 1), val), d1 (r), d2 (c) { }
00056
00057 explicit DiagArray2 (const Array<T>& a)
00058 : Array<T> (a.as_column ()), d1 (a.numel ()), d2 (a.numel ()) { }
00059
00060 DiagArray2 (const Array<T>& a, octave_idx_type r, octave_idx_type c);
00061
00062 DiagArray2 (const DiagArray2<T>& a)
00063 : Array<T> (a), d1 (a.d1), d2 (a.d2) { }
00064
00065 template <class U>
00066 DiagArray2 (const DiagArray2<U>& a)
00067 : Array<T> (a.diag ()), d1 (a.dim1 ()), d2 (a.dim2 ()) { }
00068
00069 ~DiagArray2 (void) { }
00070
00071 DiagArray2<T>& operator = (const DiagArray2<T>& a)
00072 {
00073 if (this != &a)
00074 {
00075 Array<T>::operator = (a);
00076 d1 = a.d1;
00077 d2 = a.d2;
00078 }
00079
00080 return *this;
00081 }
00082
00083 octave_idx_type dim1 (void) const { return d1; }
00084 octave_idx_type dim2 (void) const { return d2; }
00085
00086 octave_idx_type rows (void) const { return dim1 (); }
00087 octave_idx_type cols (void) const { return dim2 (); }
00088 octave_idx_type columns (void) const { return dim2 (); }
00089
00090 octave_idx_type diag_length (void) const { return Array<T>::length (); }
00091
00092 octave_idx_type length (void) const { return Array<T>::length (); }
00093 octave_idx_type nelem (void) const { return dim1 () * dim2 (); }
00094 octave_idx_type numel (void) const { return nelem (); }
00095
00096 size_t byte_size (void) const { return Array<T>::byte_size (); }
00097
00098 dim_vector dims (void) const { return dim_vector (d1, d2); }
00099
00100 Array<T> diag (octave_idx_type k = 0) const;
00101
00102
00103
00104
00105 T elem (octave_idx_type r, octave_idx_type c) const
00106 {
00107 return (r == c) ? Array<T>::elem (r) : T (0);
00108 }
00109
00110 T& elem (octave_idx_type r, octave_idx_type c)
00111 {
00112 static T zero (0);
00113 return (r == c) ? Array<T>::elem (r) : zero;
00114 }
00115
00116 T dgelem (octave_idx_type i) const
00117 { return Array<T>::elem (i); }
00118
00119 T& dgelem (octave_idx_type i)
00120 { return Array<T>::elem (i); }
00121
00122 T checkelem (octave_idx_type r, octave_idx_type c) const;
00123
00124 T operator () (octave_idx_type r, octave_idx_type c) const
00125 {
00126 #if defined (BOUNDS_CHECKING)
00127 return checkelem (r, c);
00128 #else
00129 return elem (r, c);
00130 #endif
00131 }
00132
00133
00134
00135 T xelem (octave_idx_type r, octave_idx_type c) const
00136 {
00137 return (r == c) ? Array<T>::xelem (r) : T (0);
00138 }
00139
00140 T& dgxelem (octave_idx_type i)
00141 { return Array<T>::xelem (i); }
00142
00143 T dgxelem (octave_idx_type i) const
00144 { return Array<T>::xelem (i); }
00145
00146 void resize (octave_idx_type n, octave_idx_type m,
00147 const T& rfv = Array<T>::resize_fill_value ());
00148
00149 DiagArray2<T> transpose (void) const;
00150 DiagArray2<T> hermitian (T (*fcn) (const T&) = 0) const;
00151
00152 Array<T> array_value (void) const;
00153
00154 const T *data (void) const { return Array<T>::data (); }
00155
00156 const T *fortran_vec (void) const { return Array<T>::fortran_vec (); }
00157
00158 T *fortran_vec (void) { return Array<T>::fortran_vec (); }
00159
00160 void print_info (std::ostream& os, const std::string& prefix) const
00161 { Array<T>::print_info (os, prefix); }
00162 };
00163
00164 #endif