Mercurial > hg > octave-lyh
annotate src/ov-cx-mat.cc @ 7503:8c32f95c2639
convert mapper functions to new format
| author | David Bateman <dbateman@free.fr> |
|---|---|
| date | Wed, 20 Feb 2008 04:22:50 -0500 |
| parents | a1dbe9d80eee |
| children | 99c410f7f0b0 |
| rev | line source |
|---|---|
| 2376 | 1 /* |
| 2 | |
| 7017 | 3 Copyright (C) 1996, 1997, 1998, 2000, 2001, 2002, 2003, 2004, 2005, |
| 4 2006, 2007 John W. Eaton | |
| 2376 | 5 |
| 6 This file is part of Octave. | |
| 7 | |
| 8 Octave is free software; you can redistribute it and/or modify it | |
| 9 under the terms of the GNU General Public License as published by the | |
| 7016 | 10 Free Software Foundation; either version 3 of the License, or (at your |
| 11 option) any later version. | |
| 2376 | 12 |
| 13 Octave is distributed in the hope that it will be useful, but WITHOUT | |
| 14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
| 15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
| 16 for more details. | |
| 17 | |
| 18 You should have received a copy of the GNU General Public License | |
| 7016 | 19 along with Octave; see the file COPYING. If not, see |
| 20 <http://www.gnu.org/licenses/>. | |
| 2376 | 21 |
| 22 */ | |
| 23 | |
| 24 #ifdef HAVE_CONFIG_H | |
| 25 #include <config.h> | |
| 26 #endif | |
| 27 | |
| 3503 | 28 #include <iostream> |
| 4726 | 29 #include <vector> |
| 2901 | 30 |
| 4944 | 31 #include "data-conv.h" |
| 2376 | 32 #include "lo-ieee.h" |
|
7503
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
33 #include "lo-specfun.h" |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
34 #include "lo-mappers.h" |
| 2376 | 35 #include "mx-base.h" |
| 4944 | 36 #include "mach-info.h" |
| 2376 | 37 |
| 38 #include "gripes.h" | |
| 39 #include "oct-obj.h" | |
| 4944 | 40 #include "oct-stream.h" |
| 2410 | 41 #include "ops.h" |
| 3219 | 42 #include "ov-base.h" |
| 43 #include "ov-base-mat.h" | |
| 44 #include "ov-base-mat.cc" | |
| 2410 | 45 #include "ov-complex.h" |
| 2376 | 46 #include "ov-cx-mat.h" |
| 2410 | 47 #include "ov-re-mat.h" |
| 48 #include "ov-scalar.h" | |
| 2376 | 49 #include "pr-output.h" |
| 50 | |
| 4687 | 51 #include "byte-swap.h" |
| 52 #include "ls-oct-ascii.h" | |
| 53 #include "ls-hdf5.h" | |
| 54 #include "ls-utils.h" | |
| 55 | |
| 4513 | 56 template class octave_base_matrix<ComplexNDArray>; |
| 2376 | 57 |
| 3219 | 58 DEFINE_OCTAVE_ALLOCATOR (octave_complex_matrix); |
| 2477 | 59 |
| 4612 | 60 DEFINE_OV_TYPEID_FUNCTIONS_AND_DATA (octave_complex_matrix, |
| 61 "complex matrix", "double"); | |
| 2376 | 62 |
| 5759 | 63 octave_base_value * |
| 2410 | 64 octave_complex_matrix::try_narrowing_conversion (void) |
| 65 { | |
| 5759 | 66 octave_base_value *retval = 0; |
| 2410 | 67 |
| 4513 | 68 if (matrix.ndims () == 2) |
| 69 { | |
| 70 ComplexMatrix cm = matrix.matrix_value (); | |
| 2410 | 71 |
| 5275 | 72 octave_idx_type nr = cm.rows (); |
| 73 octave_idx_type nc = cm.cols (); | |
| 4513 | 74 |
| 75 if (nr == 1 && nc == 1) | |
| 76 { | |
| 77 Complex c = matrix (0, 0); | |
| 2410 | 78 |
| 5450 | 79 double im = std::imag (c); |
| 80 | |
| 81 if (im == 0.0 && ! lo_ieee_signbit (im)) | |
| 4513 | 82 retval = new octave_scalar (std::real (c)); |
| 83 else | |
| 84 retval = new octave_complex (c); | |
| 85 } | |
| 86 else if (nr == 0 || nc == 0) | |
| 87 retval = new octave_matrix (Matrix (nr, nc)); | |
| 88 else if (cm.all_elements_are_real ()) | |
| 89 retval = new octave_matrix (::real (cm)); | |
| 2410 | 90 } |
| 4923 | 91 else if (matrix.all_elements_are_real ()) |
| 92 retval = new octave_matrix (::real (matrix)); | |
| 2410 | 93 |
| 94 return retval; | |
| 95 } | |
| 2376 | 96 |
| 97 void | |
| 98 octave_complex_matrix::assign (const octave_value_list& idx, | |
| 4686 | 99 const ComplexNDArray& rhs) |
| 4418 | 100 { |
| 4513 | 101 octave_base_matrix<ComplexNDArray>::assign (idx, rhs); |
| 4418 | 102 } |
| 103 | |
| 104 void | |
| 105 octave_complex_matrix::assign (const octave_value_list& idx, | |
| 4686 | 106 const NDArray& rhs) |
| 2376 | 107 { |
| 5275 | 108 octave_idx_type len = idx.length (); |
| 2376 | 109 |
| 5275 | 110 for (octave_idx_type i = 0; i < len; i++) |
| 4513 | 111 matrix.set_index (idx(i).index_vector ()); |
| 2376 | 112 |
| 4513 | 113 ::assign (matrix, rhs); |
| 2376 | 114 } |
| 115 | |
| 116 bool | |
| 117 octave_complex_matrix::valid_as_scalar_index (void) const | |
| 118 { | |
| 5775 | 119 // FIXME |
| 2376 | 120 return false; |
| 121 } | |
| 122 | |
| 123 double | |
| 124 octave_complex_matrix::double_value (bool force_conversion) const | |
| 125 { | |
| 4102 | 126 double retval = lo_ieee_nan_value (); |
| 2376 | 127 |
| 5781 | 128 if (! force_conversion) |
| 129 gripe_implicit_conversion ("Octave:imag-to-real", | |
| 130 "complex matrix", "real scalar"); | |
| 2376 | 131 |
| 4455 | 132 if (rows () > 0 && columns () > 0) |
| 133 { | |
| 5781 | 134 gripe_implicit_conversion ("Octave:array-as-scalar", |
| 135 "complex matrix", "real scalar"); | |
| 4455 | 136 |
| 137 retval = std::real (matrix (0, 0)); | |
| 138 } | |
| 2376 | 139 else |
| 140 gripe_invalid_conversion ("complex matrix", "real scalar"); | |
| 141 | |
| 142 return retval; | |
| 143 } | |
| 144 | |
| 145 Matrix | |
| 146 octave_complex_matrix::matrix_value (bool force_conversion) const | |
| 147 { | |
| 148 Matrix retval; | |
| 149 | |
| 5781 | 150 if (! force_conversion) |
| 151 gripe_implicit_conversion ("Octave:imag-to-real", | |
| 152 "complex matrix", "real matrix"); | |
| 2376 | 153 |
| 4513 | 154 retval = ::real (matrix.matrix_value ()); |
| 2376 | 155 |
| 156 return retval; | |
| 157 } | |
| 158 | |
| 159 Complex | |
| 160 octave_complex_matrix::complex_value (bool) const | |
| 161 { | |
| 4102 | 162 double tmp = lo_ieee_nan_value (); |
| 163 | |
| 164 Complex retval (tmp, tmp); | |
| 2376 | 165 |
| 4455 | 166 if (rows () > 0 && columns () > 0) |
| 167 { | |
| 5781 | 168 gripe_implicit_conversion ("Octave:array-as-scalar", |
| 169 "complex matrix", "complex scalar"); | |
| 4455 | 170 |
| 171 retval = matrix (0, 0); | |
| 172 } | |
| 2376 | 173 else |
| 174 gripe_invalid_conversion ("complex matrix", "complex scalar"); | |
| 175 | |
| 176 return retval; | |
| 177 } | |
| 178 | |
| 179 ComplexMatrix | |
| 180 octave_complex_matrix::complex_matrix_value (bool) const | |
| 181 { | |
| 4513 | 182 return matrix.matrix_value (); |
| 2376 | 183 } |
| 184 | |
|
7503
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
185 charNDArray |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
186 octave_complex_matrix::char_array_value (bool frc_str_conv) const |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
187 { |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
188 charNDArray retval; |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
189 |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
190 if (! frc_str_conv) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
191 gripe_implicit_conversion ("Octave:num-to-str", |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
192 "complex matrix", "string"); |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
193 else |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
194 { |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
195 retval = charNDArray (dims ()); |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
196 octave_idx_type nel = numel (); |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
197 |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
198 for (octave_idx_type i = 0; i < nel; i++) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
199 retval.elem (i) = static_cast<char>(std::real (matrix.elem (i))); |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
200 } |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
201 |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
202 return retval; |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
203 } |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
204 |
| 5164 | 205 SparseMatrix |
| 206 octave_complex_matrix::sparse_matrix_value (bool force_conversion) const | |
| 207 { | |
| 208 SparseMatrix retval; | |
| 209 | |
| 5781 | 210 if (! force_conversion) |
| 211 gripe_implicit_conversion ("Octave:imag-to-real", | |
| 212 "complex matrix", "real matrix"); | |
| 5164 | 213 |
| 214 retval = SparseMatrix (::real (matrix.matrix_value ())); | |
| 215 | |
| 216 return retval; | |
| 217 } | |
| 218 | |
| 219 SparseComplexMatrix | |
| 220 octave_complex_matrix::sparse_complex_matrix_value (bool) const | |
| 221 { | |
| 222 return SparseComplexMatrix (matrix.matrix_value ()); | |
| 223 } | |
| 224 | |
| 4687 | 225 bool |
| 6974 | 226 octave_complex_matrix::save_ascii (std::ostream& os) |
| 4687 | 227 { |
| 228 dim_vector d = dims (); | |
| 229 if (d.length () > 2) | |
| 230 { | |
| 231 ComplexNDArray tmp = complex_array_value (); | |
| 232 | |
| 233 os << "# ndims: " << d.length () << "\n"; | |
| 234 | |
| 5275 | 235 for (int i = 0; i < d.length (); i++) |
| 4687 | 236 os << " " << d (i); |
| 237 | |
| 238 os << "\n" << tmp; | |
| 239 } | |
| 240 else | |
| 241 { | |
| 242 // Keep this case, rather than use generic code above for backward | |
| 243 // compatiability. Makes load_ascii much more complex!! | |
| 244 os << "# rows: " << rows () << "\n" | |
| 245 << "# columns: " << columns () << "\n"; | |
| 246 | |
| 5958 | 247 os << complex_matrix_value (); |
| 4687 | 248 } |
| 249 | |
| 250 return true; | |
| 251 } | |
| 252 | |
| 253 bool | |
| 254 octave_complex_matrix::load_ascii (std::istream& is) | |
| 255 { | |
| 256 bool success = true; | |
| 5099 | 257 |
| 258 string_vector keywords(2); | |
| 259 | |
| 260 keywords[0] = "ndims"; | |
| 261 keywords[1] = "rows"; | |
| 4687 | 262 |
| 5099 | 263 std::string kw; |
| 5275 | 264 octave_idx_type val = 0; |
| 5099 | 265 |
| 266 if (extract_keyword (is, keywords, kw, val, true)) | |
| 4687 | 267 { |
| 5099 | 268 if (kw == "ndims") |
| 4687 | 269 { |
| 5275 | 270 int mdims = static_cast<int> (val); |
| 4687 | 271 |
| 5099 | 272 if (mdims >= 0) |
| 4687 | 273 { |
| 5099 | 274 dim_vector dv; |
| 275 dv.resize (mdims); | |
| 4687 | 276 |
| 5099 | 277 for (int i = 0; i < mdims; i++) |
| 278 is >> dv(i); | |
| 4687 | 279 |
| 6717 | 280 if (is) |
| 281 { | |
| 282 ComplexNDArray tmp(dv); | |
| 283 | |
| 284 if (tmp.is_empty ()) | |
| 285 matrix = tmp; | |
| 286 else | |
| 287 { | |
| 288 is >> tmp; | |
| 5099 | 289 |
| 6717 | 290 if (is) |
| 291 matrix = tmp; | |
| 292 else | |
| 293 { | |
| 294 error ("load: failed to load matrix constant"); | |
| 295 success = false; | |
| 296 } | |
| 297 } | |
| 298 } | |
| 299 else | |
| 4687 | 300 { |
| 6717 | 301 error ("load: failed to read dimensions"); |
| 4687 | 302 success = false; |
| 303 } | |
| 304 } | |
| 305 else | |
| 5099 | 306 { |
| 6717 | 307 error ("load: failed to extract number of dimensions"); |
| 5099 | 308 success = false; |
| 309 } | |
| 4687 | 310 } |
| 5099 | 311 else if (kw == "rows") |
| 4687 | 312 { |
| 5275 | 313 octave_idx_type nr = val; |
| 314 octave_idx_type nc = 0; | |
| 5099 | 315 |
| 316 if (nr >= 0 && extract_keyword (is, "columns", nc) && nc >= 0) | |
| 317 { | |
| 318 if (nr > 0 && nc > 0) | |
| 319 { | |
| 320 ComplexMatrix tmp (nr, nc); | |
| 321 is >> tmp; | |
| 6717 | 322 if (is) |
| 323 matrix = tmp; | |
| 324 else | |
| 5099 | 325 { |
| 326 error ("load: failed to load matrix constant"); | |
| 327 success = false; | |
| 328 } | |
| 329 } | |
| 330 else if (nr == 0 || nc == 0) | |
| 331 matrix = ComplexMatrix (nr, nc); | |
| 332 else | |
| 333 panic_impossible (); | |
| 334 } | |
| 335 else | |
| 336 { | |
| 337 error ("load: failed to extract number of rows and columns"); | |
| 338 success = false; | |
| 339 } | |
| 4687 | 340 } |
| 5099 | 341 else |
| 342 panic_impossible (); | |
| 343 } | |
| 344 else | |
| 345 { | |
| 346 error ("load: failed to extract number of rows and columns"); | |
| 347 success = false; | |
| 4687 | 348 } |
| 349 | |
| 350 return success; | |
| 351 } | |
| 352 | |
| 353 bool | |
| 354 octave_complex_matrix::save_binary (std::ostream& os, bool& save_as_floats) | |
| 355 { | |
| 356 dim_vector d = dims (); | |
| 357 if (d.length() < 1) | |
| 358 return false; | |
| 359 | |
| 360 // Use negative value for ndims to differentiate with old format!! | |
| 5828 | 361 int32_t tmp = - d.length(); |
| 5760 | 362 os.write (reinterpret_cast<char *> (&tmp), 4); |
| 5275 | 363 for (int i = 0; i < d.length (); i++) |
| 4687 | 364 { |
| 365 tmp = d(i); | |
| 5760 | 366 os.write (reinterpret_cast<char *> (&tmp), 4); |
| 4687 | 367 } |
| 368 | |
| 369 ComplexNDArray m = complex_array_value (); | |
| 370 save_type st = LS_DOUBLE; | |
| 371 if (save_as_floats) | |
| 372 { | |
| 373 if (m.too_large_for_float ()) | |
| 374 { | |
| 375 warning ("save: some values too large to save as floats --"); | |
| 376 warning ("save: saving as doubles instead"); | |
| 377 } | |
| 378 else | |
| 379 st = LS_FLOAT; | |
| 380 } | |
| 5775 | 381 else if (d.numel () > 4096) // FIXME -- make this configurable. |
| 4687 | 382 { |
| 383 double max_val, min_val; | |
| 384 if (m.all_integers (max_val, min_val)) | |
| 385 st = get_save_type (max_val, min_val); | |
| 386 } | |
| 387 | |
| 388 | |
| 389 const Complex *mtmp = m.data (); | |
| 5760 | 390 write_doubles (os, reinterpret_cast<const double *> (mtmp), st, 2 * d.numel ()); |
| 4687 | 391 |
| 392 return true; | |
| 393 } | |
| 394 | |
| 395 bool | |
| 396 octave_complex_matrix::load_binary (std::istream& is, bool swap, | |
| 397 oct_mach_info::float_format fmt) | |
| 398 { | |
| 399 char tmp; | |
| 5828 | 400 int32_t mdims; |
| 5760 | 401 if (! is.read (reinterpret_cast<char *> (&mdims), 4)) |
| 4687 | 402 return false; |
| 403 if (swap) | |
| 4944 | 404 swap_bytes<4> (&mdims); |
| 4687 | 405 if (mdims < 0) |
| 406 { | |
| 407 mdims = - mdims; | |
| 5828 | 408 int32_t di; |
| 4687 | 409 dim_vector dv; |
| 410 dv.resize (mdims); | |
| 411 | |
| 412 for (int i = 0; i < mdims; i++) | |
| 413 { | |
| 5760 | 414 if (! is.read (reinterpret_cast<char *> (&di), 4)) |
| 4687 | 415 return false; |
| 416 if (swap) | |
| 4944 | 417 swap_bytes<4> (&di); |
| 4687 | 418 dv(i) = di; |
| 419 } | |
| 420 | |
| 5157 | 421 // Convert an array with a single dimension to be a row vector. |
| 422 // Octave should never write files like this, other software | |
| 423 // might. | |
| 424 | |
| 425 if (mdims == 1) | |
| 426 { | |
| 427 mdims = 2; | |
| 428 dv.resize (mdims); | |
| 429 dv(1) = dv(0); | |
| 430 dv(0) = 1; | |
| 431 } | |
| 432 | |
| 5760 | 433 if (! is.read (reinterpret_cast<char *> (&tmp), 1)) |
| 4687 | 434 return false; |
| 435 | |
| 436 ComplexNDArray m(dv); | |
| 437 Complex *im = m.fortran_vec (); | |
| 5760 | 438 read_doubles (is, reinterpret_cast<double *> (im), |
| 439 static_cast<save_type> (tmp), 2 * dv.numel (), swap, fmt); | |
| 4687 | 440 if (error_state || ! is) |
| 441 return false; | |
| 442 matrix = m; | |
| 443 } | |
| 444 else | |
| 445 { | |
| 5828 | 446 int32_t nr, nc; |
| 4687 | 447 nr = mdims; |
| 5760 | 448 if (! is.read (reinterpret_cast<char *> (&nc), 4)) |
| 4687 | 449 return false; |
| 450 if (swap) | |
| 4944 | 451 swap_bytes<4> (&nc); |
| 5760 | 452 if (! is.read (reinterpret_cast<char *> (&tmp), 1)) |
| 4687 | 453 return false; |
| 454 ComplexMatrix m (nr, nc); | |
| 455 Complex *im = m.fortran_vec (); | |
| 5275 | 456 octave_idx_type len = nr * nc; |
| 5760 | 457 read_doubles (is, reinterpret_cast<double *> (im), |
| 458 static_cast<save_type> (tmp), 2*len, swap, fmt); | |
| 4687 | 459 if (error_state || ! is) |
| 460 return false; | |
| 461 matrix = m; | |
| 462 } | |
| 463 return true; | |
| 464 } | |
| 465 | |
| 466 #if defined (HAVE_HDF5) | |
| 4944 | 467 |
| 4687 | 468 bool |
| 469 octave_complex_matrix::save_hdf5 (hid_t loc_id, const char *name, | |
| 470 bool save_as_floats) | |
| 471 { | |
| 4837 | 472 dim_vector dv = dims (); |
| 473 int empty = save_hdf5_empty (loc_id, name, dv); | |
| 474 if (empty) | |
| 4805 | 475 return (empty > 0); |
| 476 | |
| 4837 | 477 int rank = dv.length (); |
| 4805 | 478 hid_t space_hid = -1, data_hid = -1, type_hid = -1; |
| 4687 | 479 bool retval = true; |
| 480 ComplexNDArray m = complex_array_value (); | |
| 481 | |
| 4805 | 482 OCTAVE_LOCAL_BUFFER (hsize_t, hdims, rank); |
| 483 | |
| 4687 | 484 // Octave uses column-major, while HDF5 uses row-major ordering |
| 4805 | 485 for (int i = 0; i < rank; i++) |
| 4837 | 486 hdims[i] = dv (rank-i-1); |
| 4805 | 487 |
| 4815 | 488 space_hid = H5Screate_simple (rank, hdims, 0); |
| 4687 | 489 if (space_hid < 0) return false; |
| 490 | |
| 491 hid_t save_type_hid = H5T_NATIVE_DOUBLE; | |
| 492 | |
| 493 if (save_as_floats) | |
| 494 { | |
| 495 if (m.too_large_for_float ()) | |
| 496 { | |
| 497 warning ("save: some values too large to save as floats --"); | |
| 498 warning ("save: saving as doubles instead"); | |
| 499 } | |
| 500 else | |
| 501 save_type_hid = H5T_NATIVE_FLOAT; | |
| 502 } | |
| 503 #if HAVE_HDF5_INT2FLOAT_CONVERSIONS | |
| 504 // hdf5 currently doesn't support float/integer conversions | |
| 505 else | |
| 506 { | |
| 507 double max_val, min_val; | |
| 508 | |
| 509 if (m.all_integers (max_val, min_val)) | |
| 510 save_type_hid | |
| 511 = save_type_to_hdf5 (get_save_type (max_val, min_val)); | |
| 512 } | |
| 513 #endif /* HAVE_HDF5_INT2FLOAT_CONVERSIONS */ | |
| 514 | |
| 515 type_hid = hdf5_make_complex_type (save_type_hid); | |
| 516 if (type_hid < 0) | |
| 517 { | |
| 518 H5Sclose (space_hid); | |
| 519 return false; | |
| 520 } | |
| 521 | |
| 522 data_hid = H5Dcreate (loc_id, name, type_hid, space_hid, H5P_DEFAULT); | |
| 523 if (data_hid < 0) | |
| 524 { | |
| 525 H5Sclose (space_hid); | |
| 526 H5Tclose (type_hid); | |
| 527 return false; | |
| 528 } | |
| 529 | |
| 530 hid_t complex_type_hid = hdf5_make_complex_type (H5T_NATIVE_DOUBLE); | |
| 531 if (complex_type_hid < 0) retval = false; | |
| 532 | |
| 533 if (retval) | |
| 534 { | |
| 535 Complex *mtmp = m.fortran_vec (); | |
| 536 if (H5Dwrite (data_hid, complex_type_hid, H5S_ALL, H5S_ALL, H5P_DEFAULT, | |
| 4815 | 537 mtmp) < 0) |
| 4687 | 538 { |
| 539 H5Tclose (complex_type_hid); | |
| 540 retval = false; | |
| 541 } | |
| 542 } | |
| 543 | |
| 544 H5Tclose (complex_type_hid); | |
| 545 H5Dclose (data_hid); | |
| 546 H5Tclose (type_hid); | |
| 547 H5Sclose (space_hid); | |
| 4837 | 548 |
| 4687 | 549 return retval; |
| 550 } | |
| 551 | |
| 552 bool | |
| 553 octave_complex_matrix::load_hdf5 (hid_t loc_id, const char *name, | |
| 554 bool /* have_h5giterate_bug */) | |
| 555 { | |
| 4837 | 556 bool retval = false; |
| 557 | |
| 4805 | 558 dim_vector dv; |
| 559 int empty = load_hdf5_empty (loc_id, name, dv); | |
| 560 if (empty > 0) | |
| 561 matrix.resize(dv); | |
| 4837 | 562 if (empty) |
| 4805 | 563 return (empty > 0); |
| 564 | |
| 4687 | 565 hid_t data_hid = H5Dopen (loc_id, name); |
| 566 hid_t type_hid = H5Dget_type (data_hid); | |
| 567 | |
| 568 hid_t complex_type = hdf5_make_complex_type (H5T_NATIVE_DOUBLE); | |
| 569 | |
| 570 if (! hdf5_types_compatible (type_hid, complex_type)) | |
| 571 { | |
| 4837 | 572 H5Tclose (complex_type); |
| 4687 | 573 H5Dclose (data_hid); |
| 574 return false; | |
| 575 } | |
| 576 | |
| 577 hid_t space_id = H5Dget_space (data_hid); | |
| 578 | |
| 579 hsize_t rank = H5Sget_simple_extent_ndims (space_id); | |
| 580 | |
| 581 if (rank < 1) | |
| 582 { | |
| 4837 | 583 H5Tclose (complex_type); |
| 4687 | 584 H5Sclose (space_id); |
| 585 H5Dclose (data_hid); | |
| 586 return false; | |
| 587 } | |
| 588 | |
| 589 OCTAVE_LOCAL_BUFFER (hsize_t, hdims, rank); | |
| 590 OCTAVE_LOCAL_BUFFER (hsize_t, maxdims, rank); | |
| 591 | |
| 592 H5Sget_simple_extent_dims (space_id, hdims, maxdims); | |
| 593 | |
| 594 // Octave uses column-major, while HDF5 uses row-major ordering | |
| 595 if (rank == 1) | |
| 596 { | |
| 597 dv.resize (2); | |
| 598 dv(0) = 1; | |
| 599 dv(1) = hdims[0]; | |
| 600 } | |
| 601 else | |
| 602 { | |
| 603 dv.resize (rank); | |
| 4815 | 604 for (hsize_t i = 0, j = rank - 1; i < rank; i++, j--) |
| 4687 | 605 dv(j) = hdims[i]; |
| 606 } | |
| 607 | |
| 608 ComplexNDArray m (dv); | |
| 609 Complex *reim = m.fortran_vec (); | |
| 4815 | 610 if (H5Dread (data_hid, complex_type, H5S_ALL, H5S_ALL, H5P_DEFAULT, |
| 611 reim) >= 0) | |
| 4687 | 612 { |
| 613 retval = true; | |
| 614 matrix = m; | |
| 615 } | |
| 616 | |
| 4837 | 617 H5Tclose (complex_type); |
| 4687 | 618 H5Sclose (space_id); |
| 619 H5Dclose (data_hid); | |
| 4837 | 620 |
| 4687 | 621 return retval; |
| 622 } | |
| 4944 | 623 |
| 4687 | 624 #endif |
| 625 | |
| 4643 | 626 void |
| 627 octave_complex_matrix::print_raw (std::ostream& os, | |
| 628 bool pr_as_read_syntax) const | |
| 629 { | |
| 630 octave_print_internal (os, matrix, pr_as_read_syntax, | |
| 631 current_print_indent_level ()); | |
| 632 } | |
| 633 | |
| 5900 | 634 mxArray * |
| 635 octave_complex_matrix::as_mxArray (void) const | |
| 636 { | |
| 637 mxArray *retval = new mxArray (mxDOUBLE_CLASS, dims (), mxCOMPLEX); | |
| 638 | |
| 639 double *pr = static_cast<double *> (retval->get_data ()); | |
| 640 double *pi = static_cast<double *> (retval->get_imag_data ()); | |
| 641 | |
| 6686 | 642 mwSize nel = numel (); |
| 5900 | 643 |
| 644 const Complex *p = matrix.data (); | |
| 645 | |
| 6686 | 646 for (mwIndex i = 0; i < nel; i++) |
| 5900 | 647 { |
|
7503
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
648 pr[i] = std::real (p[i]); |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
649 pi[i] = std::imag (p[i]); |
| 5900 | 650 } |
| 651 | |
| 652 return retval; | |
| 653 } | |
| 654 | |
|
7503
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
655 static double |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
656 xabs (const Complex& x) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
657 { |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
658 return (xisinf (x.real ()) || xisinf (x.imag ())) ? octave_Inf : abs (x); |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
659 } |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
660 |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
661 static double |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
662 ximag (const Complex& x) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
663 { |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
664 return x.imag (); |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
665 } |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
666 |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
667 static double |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
668 xreal (const Complex& x) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
669 { |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
670 return x.real (); |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
671 } |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
672 |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
673 #define ARRAY_MAPPER(MAP, AMAP, FCN) \ |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
674 octave_value \ |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
675 octave_complex_matrix::MAP (void) const \ |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
676 { \ |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
677 static AMAP cmap = FCN; \ |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
678 return matrix.map (cmap); \ |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
679 } |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
680 |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
681 ARRAY_MAPPER (abs, ComplexNDArray::dmapper, xabs) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
682 ARRAY_MAPPER (acos, ComplexNDArray::cmapper, ::acos) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
683 ARRAY_MAPPER (acosh, ComplexNDArray::cmapper, ::acosh) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
684 ARRAY_MAPPER (angle, ComplexNDArray::dmapper, std::arg) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
685 ARRAY_MAPPER (arg, ComplexNDArray::dmapper, std::arg) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
686 ARRAY_MAPPER (asin, ComplexNDArray::cmapper, ::asin) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
687 ARRAY_MAPPER (asinh, ComplexNDArray::cmapper, ::asinh) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
688 ARRAY_MAPPER (atan, ComplexNDArray::cmapper, ::atan) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
689 ARRAY_MAPPER (atanh, ComplexNDArray::cmapper, ::atanh) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
690 ARRAY_MAPPER (ceil, ComplexNDArray::cmapper, ::ceil) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
691 ARRAY_MAPPER (conj, ComplexNDArray::cmapper, std::conj) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
692 ARRAY_MAPPER (cos, ComplexNDArray::cmapper, std::cos) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
693 ARRAY_MAPPER (cosh, ComplexNDArray::cmapper, std::cosh) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
694 ARRAY_MAPPER (exp, ComplexNDArray::cmapper, std::exp) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
695 ARRAY_MAPPER (fix, ComplexNDArray::cmapper, ::fix) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
696 ARRAY_MAPPER (floor, ComplexNDArray::cmapper, ::floor) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
697 ARRAY_MAPPER (imag, ComplexNDArray::dmapper, ximag) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
698 ARRAY_MAPPER (log, ComplexNDArray::cmapper, std::log) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
699 ARRAY_MAPPER (log10, ComplexNDArray::cmapper, std::log10) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
700 ARRAY_MAPPER (real, ComplexNDArray::dmapper, xreal) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
701 ARRAY_MAPPER (round, ComplexNDArray::cmapper, xround) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
702 ARRAY_MAPPER (signum, ComplexNDArray::cmapper, ::signum) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
703 ARRAY_MAPPER (sin, ComplexNDArray::cmapper, std::sin) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
704 ARRAY_MAPPER (sinh, ComplexNDArray::cmapper, std::sinh) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
705 ARRAY_MAPPER (sqrt, ComplexNDArray::cmapper, std::sqrt) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
706 ARRAY_MAPPER (tan, ComplexNDArray::cmapper, std::tan) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
707 ARRAY_MAPPER (tanh, ComplexNDArray::cmapper, std::tanh) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
708 ARRAY_MAPPER (finite, ComplexNDArray::bmapper, xfinite) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
709 ARRAY_MAPPER (isinf, ComplexNDArray::bmapper, xisinf) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
710 ARRAY_MAPPER (isna, ComplexNDArray::bmapper, octave_is_NA) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
711 ARRAY_MAPPER (isnan, ComplexNDArray::bmapper, xisnan) |
|
8c32f95c2639
convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
712 |
| 2376 | 713 /* |
| 714 ;;; Local Variables: *** | |
| 715 ;;; mode: C++ *** | |
| 716 ;;; End: *** | |
| 717 */ |