octave-nkf: src/DLD-FUNCTIONS/sparse.cc annotate

annotate src/DLD-FUNCTIONS/sparse.cc @ 10513:c5005bc2b7a9

implement working spalloc

author	Jaroslav Hajek <highegg@gmail.com>
date	Tue, 13 Apr 2010 12:36:24 +0200
parents	ded9beac7582
children	4d1fc073fbb7

rev	line source
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	1 /*
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	2
8920 eb63fbe60fab update copyright notices John W. Eaton <jwe@octave.org> parents: 8910 diff changeset	3 Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009 David Bateman
7016 93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 7001 diff changeset	4 Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004 Andy Adler
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 7001 diff changeset	5
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 7001 diff changeset	6 This file is part of Octave.
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	7
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	8 Octave is free software; you can redistribute it and/or modify it
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	9 under the terms of the GNU General Public License as published by the
7016 93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 7001 diff changeset	10 Free Software Foundation; either version 3 of the License, or (at your
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 7001 diff changeset	11 option) any later version.
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	12
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	13 Octave is distributed in the hope that it will be useful, but WITHOUT
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	16 for more details.
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	17
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	18 You should have received a copy of the GNU General Public License
7016 93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 7001 diff changeset	19 along with Octave; see the file COPYING. If not, see
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe] jwe parents: 7001 diff changeset	20 <http://www.gnu.org/licenses/>.
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	21
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	22 */
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	23
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	24 #ifdef HAVE_CONFIG_H
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	25 #include <config.h>
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	26 #endif
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	27
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	28 #include <cstdlib>
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	29 #include <string>
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	30
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	31 #include "variables.h"
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	32 #include "utils.h"
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	33 #include "pager.h"
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	34 #include "defun-dld.h"
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	35 #include "gripes.h"
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	36 #include "quit.h"
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	37 #include "unwind-prot.h"
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	38
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	39 #include "ov-re-sparse.h"
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	40 #include "ov-cx-sparse.h"
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	41 #include "ov-bool-sparse.h"
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	42
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	43 DEFUN_DLD (issparse, args, ,
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	44 "-- texinfo --\n\
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	45 @deftypefn {Loadable Function} {} issparse (@var{expr})\n\
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	46 Return 1 if the value of the expression @var{expr} is a sparse matrix.\n\
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	47 @end deftypefn")
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	48 {
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	49 if (args.length() != 1)
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	50 {
5823 080c08b192d8 [project @ 2006-05-19 05:32:17 by jwe] jwe parents: 5760 diff changeset	51 print_usage ();
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	52 return octave_value ();
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	53 }
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	54 else
7515 f3c00dc0912b Eliminate the rest of the dispatched sparse functions David Bateman <dbateman@free.fr> parents: 7505 diff changeset	55 return octave_value (args(0).is_sparse_type ());
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	56 }
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	57
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	58 DEFUN_DLD (sparse, args, ,
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	59 "-- texinfo --\n\
6556 8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	60 @deftypefn {Loadable Function} {@var{s} =} sparse (@var{a})\n\
8106 8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	61 @deftypefnx {Loadable Function} {@var{s} =} sparse (@var{i}, @var{j}, @var{sv}, @var{m}, @var{n}, @var{nzmax})\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	62 @deftypefnx {Loadable Function} {@var{s} =} sparse (@var{i}, @var{j}, @var{sv})\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	63 @deftypefnx {Loadable Function} {@var{s} =} sparse (@var{i}, @var{j}, @var{s}, @var{m}, @var{n}, \"unique\")\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	64 @deftypefnx {Loadable Function} {@var{s} =} sparse (@var{m}, @var{n})\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	65 Create a sparse matrix from the full matrix or row, column, value triplets.\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	66 If @var{a} is a full matrix, convert it to a sparse matrix representation,\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	67 removing all zero values in the process.\n\
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	68 \n\
8106 8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	69 Given the integer index vectors @var{i} and @var{j}, a 1-by-@code{nnz} vector\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	70 of real of complex values @var{sv}, overall dimensions @var{m} and @var{n}\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	71 of the sparse matrix. The argument @code{nzmax} is ignored but accepted for\n\
9064 7c02ec148a3c Check grammar on all .cc files Rik <rdrider0-list@yahoo.com> parents: 8920 diff changeset	72 compatibility with @sc{matlab}. If @var{m} or @var{n} are not specified their\n\
8106 8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	73 values are derived from the maximum index in the vectors @var{i} and @var{j}\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	74 as given by @code{@var{m} = max (@var{i})}, @code{@var{n} = max (@var{j})}.\n\
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	75 \n\
6556 8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	76 @strong{Note}: if multiple values are specified with the same\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	77 @var{i}, @var{j} indices, the corresponding values in @var{s} will\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	78 be added.\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	79 \n\
6557 9883cb21e2aa [project @ 2007-04-20 19:18:13 by jwe] jwe parents: 6556 diff changeset	80 The following are all equivalent:\n\
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	81 \n\
6556 8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	82 @example\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	83 @group\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	84 s = sparse (i, j, s, m, n)\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	85 s = sparse (i, j, s, m, n, \"summation\")\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	86 s = sparse (i, j, s, m, n, \"sum\")\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	87 @end group\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	88 @end example\n\
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	89 \n\
8106 8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	90 Given the option \"unique\". if more than two values are specified for the\n\
6556 8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	91 same @var{i}, @var{j} indices, the last specified value will be used.\n\
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	92 \n\
8106 8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	93 @code{sparse(@var{m}, @var{n})} is equivalent to\n\
8a42498edb30 Clarify doc for sparse function David Bateman <dbateman@free.fr> parents: 7515 diff changeset	94 @code{sparse ([], [], [], @var{m}, @var{n}, 0)}\n\
6556 8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	95 \n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	96 If any of @var{sv}, @var{i} or @var{j} are scalars, they are expanded\n\
8810bbf321ce [project @ 2007-04-20 18:39:40 by jwe] jwe parents: 6547 diff changeset	97 to have a common size.\n\
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	98 @seealso{full}\n\
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	99 @end deftypefn")
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	100 {
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	101 octave_value retval;
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	102 int nargin = args.length ();
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	103
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	104 // Temporarily disable sparse_auto_mutate if set (it's obsolete anyway).
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	105 unwind_protect frame;
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	106 frame.protect_var (Vsparse_auto_mutate);
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	107 Vsparse_auto_mutate = false;
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	108
7287 3f29467c1667 [project @ 2007-12-11 17:03:32 by jwe] jwe parents: 7017 diff changeset	109 if (nargin == 1)
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	110 {
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	111 octave_value arg = args (0);
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	112 if (arg.is_bool_type ())
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	113 retval = arg.sparse_bool_matrix_value ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	114 else if (arg.is_complex_type ())
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	115 retval = arg.sparse_complex_matrix_value ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	116 else if (arg.is_numeric_type ())
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	117 retval = arg.sparse_matrix_value ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	118 else
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	119 gripe_wrong_type_arg ("sparse", arg);
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	120 }
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	121 else if (nargin == 2)
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	122 {
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	123 octave_idx_type m = 0, n = 0;
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	124 if (args(0).is_scalar_type () && args(1).is_scalar_type ())
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9064 diff changeset	125 {
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	126 m = args(0).idx_type_value ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	127 n = args(1).idx_type_value ();
8910 6e9f26506804 optimize diag -> sparse and perm -> sparse conversions Jaroslav Hajek <highegg@gmail.com> parents: 8458 diff changeset	128 }
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	129 else
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	130 error ("sparse: expecting scalar dimensions");
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	131
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	132 if (! error_state)
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9064 diff changeset	133 {
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	134 if (m >= 0 && n >= 0)
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	135 retval = SparseMatrix (m, n);
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	136 else
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	137 error ("sparse: dimensions must be nonnegative");
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9064 diff changeset	138 }
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	139 }
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	140 else if (nargin >= 3)
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	141 {
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	142 bool summation = true;
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	143 if (nargin > 3 && args(nargin-1).is_string ())
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9064 diff changeset	144 {
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	145 std::string opt = args(nargin-1).string_value ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	146 if (opt == "unique")
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	147 summation = false;
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	148 else if (opt == "sum" \|\| opt == "summation")
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	149 summation = true;
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9064 diff changeset	150 else
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	151 error ("sparse: invalid option: %s", opt.c_str ());
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	152
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	153 nargin -= 1;
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	154 }
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	155
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	156 if (! error_state)
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	157 {
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	158 octave_idx_type m = -1, n = -1;
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	159 if (nargin == 5)
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	160 {
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	161 if (args(3).is_scalar_type () && args(4).is_scalar_type ())
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	162 {
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	163 m = args(3).idx_type_value ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	164 n = args(4).idx_type_value ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	165 }
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	166 else
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	167 error ("sparse: expecting scalar dimensions");
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	168
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	169
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	170 if (! error_state && (m < 0 \|\| n < 0))
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	171 error ("sparse: dimensions must be nonnegative");
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	172 }
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	173 else if (nargin != 3)
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	174 print_usage ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	175
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	176 if (! error_state)
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	177 {
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	178 idx_vector i = args(0).index_vector ();
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	179 idx_vector j = args(1).index_vector ();
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	180
10479 ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	181 if (args(2).is_bool_type ())
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	182 retval = SparseBoolMatrix (args(2).bool_array_value (), i, j,
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	183 m, n, summation);
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	184 else if (args(2).is_complex_type ())
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	185 retval = SparseComplexMatrix (args(2).complex_array_value (), i, j,
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	186 m, n, summation);
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	187 else if (args(2).is_numeric_type ())
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	188 retval = SparseMatrix (args(2).array_value (), i, j,
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	189 m, n, summation);
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	190 else
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	191 gripe_wrong_type_arg ("sparse", args(2));
ded9beac7582 optimize sparse matrix assembly Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	192 }
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	193
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9064 diff changeset	194 }
5164 57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	195 }
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	196
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	197 return retval;
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe] jwe parents: diff changeset	198 }
10513 c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	199
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	200 DEFUN_DLD (spalloc, args, ,
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	201 "-- texinfo --\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	202 @deftypefn {Loadable Function} {@var{s} =} spalloc (@var{m}, @var{n}, @var{nz})\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	203 Creates a @var{m}-by-@var{n} sparse matrix with preallocated space for at most\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	204 @var{nz} nonzero elements. This is useful for building the matrix incrementally\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	205 by a sequence of indexed assignments. Subsequent indexed assignments will reuse\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	206 the pre-allocated memory, provided they are of one of the simple forms\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	207 \n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	208 @itemize\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	209 @item @code{@var{s}(I:J) = @var{x}}\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	210 @item @code{@var{s}(:,I:J) = @var{x}}\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	211 @item @code{@var{s}(K:L,I:J) = @var{x}}\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	212 @end itemize\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	213 \n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	214 @b{and} that the following conditions are met:\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	215 \n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	216 @itemize\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	217 @item the assignment does not decrease nnz(@var{S}).\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	218 @item after the assignment, nnz(@var{S}) does not exceed @var{nz}.\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	219 @item no index is out of bounds.\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	220 @end itemize\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	221 \n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	222 Partial movement of data may still occur, but in general the assignment will be more\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	223 memory and time-efficient under these circumstances. In particular, it is possible\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	224 to efficiently build a pre-allocated sparse matrix from contiguous block of columns.\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	225 \n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	226 The amount of preallocated memory for a given matrix may be queried using the function\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	227 @code{nzmax}.\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	228 @seealso{nzmax, sparse}\n\
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	229 @end deftypefn")
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	230 {
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	231 octave_value retval;
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	232 int nargin = args.length ();
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	233
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	234 if (nargin == 2 \|\| nargin == 3)
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	235 {
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	236 octave_idx_type m = args(0).idx_type_value ();
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	237 octave_idx_type n = args(1).idx_type_value ();
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	238 octave_idx_type nz = 0;
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	239 if (nargin == 3)
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	240 nz = args(2).idx_type_value ();
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	241 if (error_state)
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	242 ;
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	243 else if (m >= 0 && n >= 0 && nz >= 0)
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	244 retval = SparseMatrix (dim_vector (m, n), nz);
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	245 else
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	246 error ("spalloc: m,n,nz must be non-negative");
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	247 }
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	248 else
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	249 print_usage ();
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	250
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	251 return retval;
c5005bc2b7a9 implement working spalloc Jaroslav Hajek <highegg@gmail.com> parents: 10479 diff changeset	252 }

Mercurial > octave-nkf

annotate src/DLD-FUNCTIONS/sparse.cc @ 10513:c5005bc2b7a9