Mercurial > hg > octave-lyh
diff src/ov-bool-sparse.cc @ 5164:57077d0ddc8e
[project @ 2005-02-25 19:55:24 by jwe]
author | jwe |
---|---|
date | Fri, 25 Feb 2005 19:55:28 +0000 |
parents | |
children | fd90c6df7d52 |
line wrap: on
line diff
new file mode 100644 --- /dev/null +++ b/src/ov-bool-sparse.cc @@ -0,0 +1,696 @@ +/* + +Copyright (C) 2004 David Bateman +Copyright (C) 1998-2004 Andy Adler + +Octave is free software; you can redistribute it and/or modify it +under the terms of the GNU General Public License as published by the +Free Software Foundation; either version 2, or (at your option) any +later version. + +Octave is distributed in the hope that it will be useful, but WITHOUT +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or +FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License +for more details. + +You should have received a copy of the GNU General Public License +along with this program; see the file COPYING. If not, write to the Free +Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + +*/ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include <climits> + +#include <iostream> +#include <vector> + +#include "ov-base.h" +#include "ov-scalar.h" +#include "ov-bool.h" +#include "ov-bool-mat.h" +#include "gripes.h" +#include "ops.h" + +#include "ov-re-sparse.h" +#include "ov-cx-sparse.h" +#include "ov-bool-sparse.h" + +#include "ov-base-sparse.h" +#include "ov-base-sparse.cc" + +template class octave_base_sparse<SparseBoolMatrix>; + +DEFINE_OCTAVE_ALLOCATOR (octave_sparse_bool_matrix); + +DEFINE_OV_TYPEID_FUNCTIONS_AND_DATA (octave_sparse_bool_matrix, "sparse bool matrix", "sparse"); + +static octave_value * +default_numeric_conversion_function (const octave_value& a) +{ + CAST_CONV_ARG (const octave_sparse_bool_matrix&); + + return new octave_sparse_matrix (SparseMatrix (v.sparse_bool_matrix_value ())); +} + +type_conv_fcn +octave_sparse_bool_matrix::numeric_conversion_function (void) const +{ + return default_numeric_conversion_function; +} + +octave_value * +octave_sparse_bool_matrix::try_narrowing_conversion (void) +{ + octave_value *retval = 0; + + // Don't use numel, since it can overflow for very large matrices + // Note that for the second test, this means it becomes approximative + // since it involves a cast to double to avoid issues of overflow + if (matrix.rows () == 1 && matrix.cols () == 1) + { + // Const copy of the matrix, so the right version of () operator used + const SparseBoolMatrix tmp (matrix); + + retval = new octave_bool (tmp (0)); + } + else if (matrix.cols () > 0 && matrix.rows () > 0 && + double (matrix.byte_size ()) > double (matrix.rows ()) * + double (matrix.cols ()) * sizeof (bool)) + retval = new octave_bool_matrix (matrix.matrix_value ()); + + return retval; +} + +bool +octave_sparse_bool_matrix::valid_as_scalar_index (void) const +{ + // XXX FIXME XXX + return false; +} + +double +octave_sparse_bool_matrix::double_value (bool) const +{ + double retval = lo_ieee_nan_value (); + + if (numel () > 0) + { + // XXX FIXME XXX -- is warn_fortran_indexing the right variable here? + if (Vwarn_fortran_indexing) + gripe_implicit_conversion ("bool sparse matrix", "real scalar"); + + retval = matrix (0, 0); + } + else + gripe_invalid_conversion ("bool sparse matrix", "real scalar"); + + return retval; +} + +Complex +octave_sparse_bool_matrix::complex_value (bool) const +{ + double tmp = lo_ieee_nan_value (); + + Complex retval (tmp, tmp); + + // XXX FIXME XXX -- maybe this should be a function, valid_as_scalar() + if (rows () > 0 && columns () > 0) + { + // XXX FIXME XXX -- is warn_fortran_indexing the right variable here? + if (Vwarn_fortran_indexing) + gripe_implicit_conversion ("bool sparse matrix", "complex scalar"); + + retval = matrix (0, 0); + } + else + gripe_invalid_conversion ("bool sparse matrix", "complex scalar"); + + return retval; +} + +octave_value +octave_sparse_bool_matrix::convert_to_str_internal (bool pad, bool force) const +{ + octave_value tmp = octave_value (array_value ()); + return tmp.convert_to_str (pad, force); +} + +// XXX FIXME XXX These are inefficient ways of creating full matrices + +Matrix +octave_sparse_bool_matrix::matrix_value (bool) const +{ + return Matrix (matrix.matrix_value ()); +} + +ComplexMatrix +octave_sparse_bool_matrix::complex_matrix_value (bool) const +{ + return ComplexMatrix (matrix.matrix_value ()); +} + +ComplexNDArray +octave_sparse_bool_matrix::complex_array_value (bool) const +{ + return ComplexNDArray (ComplexMatrix (matrix.matrix_value ())); +} + +NDArray +octave_sparse_bool_matrix::array_value (bool) const +{ + return NDArray (Matrix(matrix.matrix_value ())); +} + +boolMatrix +octave_sparse_bool_matrix::bool_matrix_value (void) const +{ + return matrix.matrix_value (); +} + +boolNDArray +octave_sparse_bool_matrix::bool_array_value (void) const +{ + return boolNDArray (matrix.matrix_value ()); +} + + +SparseMatrix +octave_sparse_bool_matrix::sparse_matrix_value (void) const +{ + return SparseMatrix (this->matrix); +} + +SparseComplexMatrix +octave_sparse_bool_matrix::sparse_complex_matrix_value (void) const +{ + return SparseComplexMatrix (this->matrix); +} + +bool +octave_sparse_bool_matrix::save_binary (std::ostream& os, bool&) +{ + dim_vector d = this->dims (); + if (d.length() < 1) + return false; + + // Ensure that additional memory is deallocated + matrix.maybe_compress (); + + int nr = d(0); + int nc = d(1); + int nz = nnz (); + + FOUR_BYTE_INT itmp; + // Use negative value for ndims to be consistent with other formats + itmp= -2; + os.write (X_CAST (char *, &itmp), 4); + + itmp= nr; + os.write (X_CAST (char *, &itmp), 4); + + itmp= nc; + os.write (X_CAST (char *, &itmp), 4); + + itmp= nz; + os.write (X_CAST (char *, &itmp), 4); + + // add one to the printed indices to go from + // zero-based to one-based arrays + for (int i = 0; i < nc+1; i++) + { + OCTAVE_QUIT; + itmp = matrix.cidx(i); + os.write (X_CAST (char *, &itmp), 4); + } + + for (int i = 0; i < nz; i++) + { + OCTAVE_QUIT; + itmp = matrix.ridx(i); + os.write (X_CAST (char *, &itmp), 4); + } + + OCTAVE_LOCAL_BUFFER (char, htmp, nz); + + for (int i = 0; i < nz; i++) + htmp[i] = (matrix.data (i) ? 1 : 0); + + os.write (htmp, nz); + + return true; +} + +bool +octave_sparse_bool_matrix::load_binary (std::istream& is, bool swap, + oct_mach_info::float_format fmt) +{ + FOUR_BYTE_INT nz, nc, nr, tmp; + if (! is.read (X_CAST (char *, &tmp), 4)) + return false; + + if (swap) + swap_bytes<4> (&tmp); + + if (tmp != -2) { + error("load: only 2D sparse matrices are supported"); + return false; + } + + if (! is.read (X_CAST (char *, &nr), 4)) + return false; + if (! is.read (X_CAST (char *, &nc), 4)) + return false; + if (! is.read (X_CAST (char *, &nz), 4)) + return false; + + if (swap) + { + swap_bytes<4> (&nr); + swap_bytes<4> (&nc); + swap_bytes<4> (&nz); + } + + SparseBoolMatrix m (nr, nc, nz); + + for (int i = 0; i < nc+1; i++) + { + OCTAVE_QUIT; + if (! is.read (X_CAST (char *, &tmp), 4)) + return false; + if (swap) + swap_bytes<4> (&tmp); + m.cidx(i) = tmp; + } + + for (int i = 0; i < nz; i++) + { + OCTAVE_QUIT; + if (! is.read (X_CAST (char *, &tmp), 4)) + return false; + if (swap) + swap_bytes<4> (&tmp); + m.ridx(i) = tmp; + } + + if (error_state || ! is) + return false; + + OCTAVE_LOCAL_BUFFER (char, htmp, nz); + + if (! is.read (htmp, nz)) + return false; + + for (int i = 0; i < nz; i++) + m.data(i) = (htmp[i] ? 1 : 0); + + matrix = m; + + return true; +} + +#if defined (HAVE_HDF5) +bool +octave_sparse_bool_matrix::save_hdf5 (hid_t loc_id, const char *name, bool) +{ + dim_vector dv = dims (); + int empty = save_hdf5_empty (loc_id, name, dv); + if (empty) + return (empty > 0); + + // Ensure that additional memory is deallocated + matrix.maybe_compress (); + + hid_t group_hid = H5Gcreate (loc_id, name, 0); + if (group_hid < 0) + return false; + + hid_t space_hid = -1, data_hid = -1; + bool retval = true; + SparseBoolMatrix m = sparse_bool_matrix_value (); + int tmp; + hsize_t hdims[2]; + + space_hid = H5Screate_simple (0, hdims, (hsize_t*) 0); + if (space_hid < 0) + { + H5Gclose (group_hid); + return false; + } + + data_hid = H5Dcreate (group_hid, "nr", H5T_NATIVE_INT, space_hid, + H5P_DEFAULT); + if (data_hid < 0) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + tmp = m.rows (); + retval = H5Dwrite (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, + (void*) &tmp) >= 0; + H5Dclose (data_hid); + if (!retval) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + data_hid = H5Dcreate (group_hid, "nc", H5T_NATIVE_INT, space_hid, + H5P_DEFAULT); + if (data_hid < 0) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + tmp = m.cols (); + retval = H5Dwrite (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, + (void*) &tmp) >= 0; + H5Dclose (data_hid); + if (!retval) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + data_hid = H5Dcreate (group_hid, "nz", H5T_NATIVE_INT, space_hid, + H5P_DEFAULT); + if (data_hid < 0) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + tmp = m.nnz (); + retval = H5Dwrite (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, + (void*) &tmp) >= 0; + H5Dclose (data_hid); + if (!retval) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + H5Sclose (space_hid); + + hdims[0] = m.cols() + 1; + hdims[1] = 1; + + space_hid = H5Screate_simple (2, hdims, 0); + + if (space_hid < 0) + { + H5Gclose (group_hid); + return false; + } + + data_hid = H5Dcreate (group_hid, "cidx", H5T_NATIVE_INT, space_hid, + H5P_DEFAULT); + if (data_hid < 0) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + int * itmp = m.xcidx (); + retval = H5Dwrite (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, + (void*) itmp) >= 0; + H5Dclose (data_hid); + if (!retval) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + H5Sclose (space_hid); + + hdims[0] = m.nnz(); + hdims[1] = 1; + + space_hid = H5Screate_simple (2, hdims, 0); + + if (space_hid < 0) + { + H5Gclose (group_hid); + return false; + } + + data_hid = H5Dcreate (group_hid, "ridx", H5T_NATIVE_INT, space_hid, + H5P_DEFAULT); + if (data_hid < 0) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + itmp = m.xridx (); + retval = H5Dwrite (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, H5P_DEFAULT, + (void*) itmp) >= 0; + H5Dclose (data_hid); + if (!retval) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + data_hid = H5Dcreate (group_hid, "data", H5T_NATIVE_HBOOL, space_hid, + H5P_DEFAULT); + if (data_hid < 0) + { + H5Sclose (space_hid); + H5Gclose (group_hid); + return false; + } + + hbool_t htmp[m.nnz ()]; + for (int i = 0; i < m.nnz (); i++) + htmp[i] = m.xdata(i); + + retval = H5Dwrite (data_hid, H5T_NATIVE_HBOOL, H5S_ALL, H5S_ALL, + H5P_DEFAULT, (void*) htmp) >= 0; + H5Dclose (data_hid); + H5Sclose (space_hid); + H5Gclose (group_hid); + + return retval; +} + +bool +octave_sparse_bool_matrix::load_hdf5 (hid_t loc_id, const char *name, + bool /* have_h5giterate_bug */) +{ + int nr, nc, nz; + hid_t group_hid, data_hid, space_hid; + hsize_t rank; + + dim_vector dv; + int empty = load_hdf5_empty (loc_id, name, dv); + if (empty > 0) + matrix.resize(dv); + if (empty) + return (empty > 0); + + group_hid = H5Gopen (loc_id, name); + if (group_hid < 0 ) return false; + + data_hid = H5Dopen (group_hid, "nr"); + space_hid = H5Dget_space (data_hid); + rank = H5Sget_simple_extent_ndims (space_hid); + + if (rank != 0) + { + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + if (H5Dread (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, + H5P_DEFAULT, (void *) &nr) < 0) + { + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + H5Dclose (data_hid); + + data_hid = H5Dopen (group_hid, "nc"); + space_hid = H5Dget_space (data_hid); + rank = H5Sget_simple_extent_ndims (space_hid); + + if (rank != 0) + { + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + if (H5Dread (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, + H5P_DEFAULT, (void *) &nc) < 0) + { + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + H5Dclose (data_hid); + + data_hid = H5Dopen (group_hid, "nz"); + space_hid = H5Dget_space (data_hid); + rank = H5Sget_simple_extent_ndims (space_hid); + + if (rank != 0) + { + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + if (H5Dread (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, + H5P_DEFAULT, (void *) &nz) < 0) + { + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + H5Dclose (data_hid); + + SparseBoolMatrix m (nr, nc, nz); + + data_hid = H5Dopen (group_hid, "cidx"); + space_hid = H5Dget_space (data_hid); + rank = H5Sget_simple_extent_ndims (space_hid); + + if (rank != 2) + { + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + OCTAVE_LOCAL_BUFFER (hsize_t, hdims, rank); + OCTAVE_LOCAL_BUFFER (hsize_t, maxdims, rank); + + H5Sget_simple_extent_dims (space_hid, hdims, maxdims); + + if (hdims[0] != nc + 1 || hdims[1] != 1) + { + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + int *itmp = m.xcidx (); + if (H5Dread (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, + H5P_DEFAULT, (void *) itmp) < 0) + { + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + H5Sclose (space_hid); + H5Dclose (data_hid); + + data_hid = H5Dopen (group_hid, "ridx"); + space_hid = H5Dget_space (data_hid); + rank = H5Sget_simple_extent_ndims (space_hid); + + if (rank != 2) + { + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + H5Sget_simple_extent_dims (space_hid, hdims, maxdims); + + if (hdims[0] != nz || hdims[1] != 1) + { + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + itmp = m.xridx (); + if (H5Dread (data_hid, H5T_NATIVE_INT, H5S_ALL, H5S_ALL, + H5P_DEFAULT, (void *) itmp) < 0) + { + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + H5Sclose (space_hid); + H5Dclose (data_hid); + + data_hid = H5Dopen (group_hid, "data"); + space_hid = H5Dget_space (data_hid); + rank = H5Sget_simple_extent_ndims (space_hid); + + if (rank != 2) + { + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + H5Sget_simple_extent_dims (space_hid, hdims, maxdims); + + if (hdims[0] != nz || hdims[1] != 1) + { + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + return false; + } + + hbool_t htmp[nz]; + bool retval = false; + if (H5Dread (data_hid, H5T_NATIVE_HBOOL, H5S_ALL, H5S_ALL, + H5P_DEFAULT, (void *) htmp) >= 0) + { + retval = true; + + for (int i = 0; i < nz; i++) + m.xdata(i) = htmp[i]; + + matrix = m; + } + + H5Sclose (space_hid); + H5Dclose (data_hid); + H5Gclose (group_hid); + + return retval; +} +#endif + +/* +;;; Local Variables: *** +;;; mode: C++ *** +;;; End: *** +*/