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

annotate src/DLD-FUNCTIONS/find.cc @ 10840:89f4d7e294cc

Grammarcheck .cc files

author	Rik <octave@nomad.inbox5.com>
date	Sat, 31 Jul 2010 11:18:11 -0700
parents	9e6aed3c6704
children	fd0a3ac60b0e

rev	line source
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	1 /*
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	2
7017 a1dbe9d80eee [project @ 2007-10-12 21:27:11 by jwe] jwe parents: 7016 diff changeset	3 Copyright (C) 1996, 1997, 1999, 2000, 2002, 2003, 2004, 2005, 2006,
8920 eb63fbe60fab update copyright notices John W. Eaton <jwe@octave.org> parents: 8810 diff changeset	4 2007, 2008, 2009 John W. Eaton
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	5
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	6 This file is part of Octave.
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	7
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	8 Octave is free software; you can redistribute it and/or modify it
295f037b4b3e [project @ 1997-05-05 05:32:33 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: 6524 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: 6524 diff changeset	11 option) any later version.
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	12
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	13 Octave is distributed in the hope that it will be useful, but WITHOUT
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	16 for more details.
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	17
295f037b4b3e [project @ 1997-05-05 05:32:33 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: 6524 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: 6524 diff changeset	20 <http://www.gnu.org/licenses/>.
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	21
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	22 */
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	23
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	24 #ifdef HAVE_CONFIG_H
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	25 #include <config.h>
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	26 #endif
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	27
4153 6b96ce9f5743 [project @ 2002-11-06 20:38:49 by jwe] jwe parents: 3418 diff changeset	28 #include "quit.h"
6b96ce9f5743 [project @ 2002-11-06 20:38:49 by jwe] jwe parents: 3418 diff changeset	29
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	30 #include "defun-dld.h"
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	31 #include "error.h"
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	32 #include "gripes.h"
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	33 #include "oct-obj.h"
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	34
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	35 // Find at most N_TO_FIND nonzero elements in NDA. Search forward if
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	36 // DIRECTION is 1, backward if it is -1. NARGOUT is the number of
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	37 // output arguments. If N_TO_FIND is -1, find all nonzero elements.
4678 e1c2d8ca8bc0 [project @ 2003-12-17 00:36:36 by jwe] jwe parents: 4628 diff changeset	38
e1c2d8ca8bc0 [project @ 2003-12-17 00:36:36 by jwe] jwe parents: 4628 diff changeset	39 template <typename T>
e1c2d8ca8bc0 [project @ 2003-12-17 00:36:36 by jwe] jwe parents: 4628 diff changeset	40 octave_value_list
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	41 find_nonzero_elem_idx (const Array<T>& nda, int nargout,
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	42 octave_idx_type n_to_find, int direction)
4678 e1c2d8ca8bc0 [project @ 2003-12-17 00:36:36 by jwe] jwe parents: 4628 diff changeset	43 {
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	44 octave_value_list retval ((nargout == 0 ? 1 : nargout), Matrix ());
4678 e1c2d8ca8bc0 [project @ 2003-12-17 00:36:36 by jwe] jwe parents: 4628 diff changeset	45
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	46 Array<octave_idx_type> idx;
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	47 if (n_to_find >= 0)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	48 idx = nda.find (n_to_find, direction == -1);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	49 else
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	50 idx = nda.find ();
5130 3bf4c0cd775d [project @ 2005-02-08 03:01:13 by jwe] jwe parents: 5107 diff changeset	51
10285 22a7913bbeb5 optimize return values of find and sort Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	52 // The maximum element is always at the end.
22a7913bbeb5 optimize return values of find and sort Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	53 octave_idx_type iext = idx.is_empty () ? 0 : idx.xelem (idx.numel () - 1) + 1;
22a7913bbeb5 optimize return values of find and sort Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	54
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	55 switch (nargout)
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	56 {
6254 5b43c2332b69 [project @ 2007-01-29 17:35:15 by jwe] jwe parents: 6026 diff changeset	57 default:
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	58 case 3:
9732 b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops Jaroslav Hajek <highegg@gmail.com> parents: 9544 diff changeset	59 retval(2) = Array<T> (nda.index (idx_vector (idx)));
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	60 // Fall through!
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	61
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	62 case 2:
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	63 {
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	64 Array<octave_idx_type> jdx (idx.dims ());
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	65 octave_idx_type n = idx.length (), nr = nda.rows ();
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	66 for (octave_idx_type i = 0; i < n; i++)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	67 {
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	68 jdx.xelem (i) = idx.xelem (i) / nr;
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	69 idx.xelem (i) %= nr;
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	70 }
10285 22a7913bbeb5 optimize return values of find and sort Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	71 iext = -1;
22a7913bbeb5 optimize return values of find and sort Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	72 retval(1) = idx_vector (jdx, -1);
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	73 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	74 // Fall through!
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	75
6254 5b43c2332b69 [project @ 2007-01-29 17:35:15 by jwe] jwe parents: 6026 diff changeset	76 case 1:
5b43c2332b69 [project @ 2007-01-29 17:35:15 by jwe] jwe parents: 6026 diff changeset	77 case 0:
10285 22a7913bbeb5 optimize return values of find and sort Jaroslav Hajek <highegg@gmail.com> parents: 10155 diff changeset	78 retval(0) = idx_vector (idx, iext);
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	79 break;
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	80 }
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	81
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	82 return retval;
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	83 }
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	84
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	85 template <typename T>
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	86 octave_value_list
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	87 find_nonzero_elem_idx (const Sparse<T>& v, int nargout,
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	88 octave_idx_type n_to_find, int direction)
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	89 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	90 octave_value_list retval ((nargout == 0 ? 1 : nargout), Matrix ());
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	91
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	92
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	93 octave_idx_type nc = v.cols();
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	94 octave_idx_type nr = v.rows();
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	95 octave_idx_type nz = v.nnz();
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	96
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	97 // Search in the default range.
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	98 octave_idx_type start_nc = -1;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	99 octave_idx_type end_nc = -1;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	100 octave_idx_type count;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	101
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	102 // Search for the range to search
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	103 if (n_to_find < 0)
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	104 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	105 start_nc = 0;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	106 end_nc = nc;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	107 n_to_find = nz;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	108 count = nz;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	109 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	110 else if (direction > 0)
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	111 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	112 for (octave_idx_type j = 0; j < nc; j++)
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	113 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	114 OCTAVE_QUIT;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	115 if (v.cidx(j) == 0 && v.cidx(j+1) != 0)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	116 start_nc = j;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	117 if (v.cidx(j+1) >= n_to_find)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	118 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	119 end_nc = j + 1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	120 break;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	121 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	122 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	123 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	124 else
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	125 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	126 for (octave_idx_type j = nc; j > 0; j--)
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	127 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	128 OCTAVE_QUIT;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	129 if (v.cidx(j) == nz && v.cidx(j-1) != nz)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	130 end_nc = j;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	131 if (nz - v.cidx(j-1) >= n_to_find)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	132 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	133 start_nc = j - 1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	134 break;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	135 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	136 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	137 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	138
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	139 count = (n_to_find > v.cidx(end_nc) - v.cidx(start_nc) ?
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	140 v.cidx(end_nc) - v.cidx(start_nc) : n_to_find);
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	141
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	142 // If the original argument was a row vector, force a row vector of
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	143 // the overall indices to be returned. But see below for scalar
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	144 // case...
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	145
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	146 octave_idx_type result_nr = count;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	147 octave_idx_type result_nc = 1;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	148
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	149 bool scalar_arg = false;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	150
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	151 if (v.rows () == 1)
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	152 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	153 result_nr = 1;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	154 result_nc = count;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	155
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	156 scalar_arg = (v.columns () == 1);
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	157 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	158
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	159 Matrix idx (result_nr, result_nc);
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	160
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	161 Matrix i_idx (result_nr, result_nc);
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	162 Matrix j_idx (result_nr, result_nc);
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	163
9732 b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops Jaroslav Hajek <highegg@gmail.com> parents: 9544 diff changeset	164 Array<T> val (dim_vector (result_nr, result_nc));
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	165
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	166 if (count > 0)
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	167 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	168 // Search for elements to return. Only search the region where
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	169 // there are elements to be found using the count that we want
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	170 // to find.
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	171 for (octave_idx_type j = start_nc, cx = 0; j < end_nc; j++)
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	172 for (octave_idx_type i = v.cidx(j); i < v.cidx(j+1); i++ )
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	173 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	174 OCTAVE_QUIT;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	175 if (direction < 0 && i < nz - count)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	176 continue;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	177 i_idx(cx) = static_cast<double> (v.ridx(i) + 1);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	178 j_idx(cx) = static_cast<double> (j + 1);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	179 idx(cx) = j * nr + v.ridx(i) + 1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	180 val(cx) = v.data(i);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	181 cx++;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	182 if (cx == count)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	183 break;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	184 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	185 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	186 else if (scalar_arg)
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	187 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	188 idx.resize (0, 0);
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	189
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	190 i_idx.resize (0, 0);
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	191 j_idx.resize (0, 0);
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	192
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	193 val.resize (dim_vector (0, 0));
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	194 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	195
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	196 switch (nargout)
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	197 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	198 case 0:
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	199 case 1:
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	200 retval(0) = idx;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	201 break;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	202
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	203 case 5:
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	204 retval(4) = nc;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	205 // Fall through
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	206
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	207 case 4:
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	208 retval(3) = nr;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	209 // Fall through
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	210
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	211 case 3:
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	212 retval(2) = val;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	213 // Fall through!
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	214
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	215 case 2:
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	216 retval(1) = j_idx;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	217 retval(0) = i_idx;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	218 break;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	219
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	220 default:
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	221 panic_impossible ();
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	222 break;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	223 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	224
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	225 return retval;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	226 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	227
8955 6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	228 octave_value_list
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	229 find_nonzero_elem_idx (const PermMatrix& v, int nargout,
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	230 octave_idx_type n_to_find, int direction)
8955 6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	231 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	232 // There are far fewer special cases to handle for a PermMatrix.
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	233 octave_value_list retval ((nargout == 0 ? 1 : nargout), Matrix ());
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	234
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	235 octave_idx_type nc = v.cols();
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	236 octave_idx_type start_nc, end_nc, count;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	237
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	238 // Determine the range to search.
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	239 if (n_to_find < 0 \|\| n_to_find >= nc)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	240 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	241 start_nc = 0;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	242 end_nc = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	243 n_to_find = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	244 count = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	245 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	246 else if (direction > 0)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	247 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	248 start_nc = 0;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	249 end_nc = n_to_find;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	250 count = n_to_find;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	251 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	252 else
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	253 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	254 start_nc = nc - n_to_find;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	255 end_nc = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	256 count = n_to_find;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	257 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	258
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	259 bool scalar_arg = (v.rows () == 1 && v.cols () == 1);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	260
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	261 Matrix idx (count, 1);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	262 Matrix i_idx (count, 1);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	263 Matrix j_idx (count, 1);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	264 // Every value is 1.
9732 b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops Jaroslav Hajek <highegg@gmail.com> parents: 9544 diff changeset	265 Array<double> val (dim_vector (count, 1), 1.0);
8955 6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	266
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	267 if (count > 0)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	268 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	269 const octave_idx_type* p = v.data ();
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	270 if (v.is_col_perm ())
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	271 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	272 for (octave_idx_type k = 0; k < count; k++)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	273 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	274 OCTAVE_QUIT;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	275 const octave_idx_type j = start_nc + k;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	276 const octave_idx_type i = p[j];
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	277 i_idx(k) = static_cast<double> (1+i);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	278 j_idx(k) = static_cast<double> (1+j);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	279 idx(k) = j * nc + i + 1;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	280 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	281 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	282 else
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	283 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	284 for (octave_idx_type k = 0; k < count; k++)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	285 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	286 OCTAVE_QUIT;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	287 const octave_idx_type i = start_nc + k;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	288 const octave_idx_type j = p[i];
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	289 // Scatter into the index arrays according to
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	290 // j adjusted by the start point.
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	291 const octave_idx_type koff = j - start_nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	292 i_idx(koff) = static_cast<double> (1+i);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	293 j_idx(koff) = static_cast<double> (1+j);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	294 idx(koff) = j * nc + i + 1;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	295 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	296 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	297 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	298 else if (scalar_arg)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	299 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	300 // Same odd compatibility case as the other overrides.
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	301 idx.resize (0, 0);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	302 i_idx.resize (0, 0);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	303 j_idx.resize (0, 0);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	304 val.resize (dim_vector (0, 0));
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	305 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	306
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	307 switch (nargout)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	308 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	309 case 0:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	310 case 1:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	311 retval(0) = idx;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	312 break;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	313
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	314 case 5:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	315 retval(4) = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	316 // Fall through
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	317
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	318 case 4:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	319 retval(3) = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	320 // Fall through
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	321
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	322 case 3:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	323 retval(2) = val;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	324 // Fall through!
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	325
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	326 case 2:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	327 retval(1) = j_idx;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	328 retval(0) = i_idx;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	329 break;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	330
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	331 default:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	332 panic_impossible ();
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	333 break;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	334 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	335
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	336 return retval;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	337 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	338
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	339 DEFUN_DLD (find, args, nargout,
3369 f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	340 "-- texinfo --\n\
10840 89f4d7e294cc Grammarcheck .cc files Rik <octave@nomad.inbox5.com> parents: 10826 diff changeset	341 @deftypefn {Loadable Function} {} find (@var{x})\n\
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	342 @deftypefnx {Loadable Function} {} find (@var{x}, @var{n})\n\
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	343 @deftypefnx {Loadable Function} {} find (@var{x}, @var{n}, @var{direction})\n\
6524 03c89ade4070 [project @ 2007-04-13 12:19:30 by jwe] jwe parents: 6254 diff changeset	344 Return a vector of indices of nonzero elements of a matrix, as a row if\n\
03c89ade4070 [project @ 2007-04-13 12:19:30 by jwe] jwe parents: 6254 diff changeset	345 @var{x} is a row or as a column otherwise. To obtain a single index for\n\
03c89ade4070 [project @ 2007-04-13 12:19:30 by jwe] jwe parents: 6254 diff changeset	346 each matrix element, Octave pretends that the columns of a matrix form one\n\
10840 89f4d7e294cc Grammarcheck .cc files Rik <octave@nomad.inbox5.com> parents: 10826 diff changeset	347 long vector (like Fortran arrays are stored). For example:\n\
3369 f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	348 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	349 @example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	350 @group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	351 find (eye (2))\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	352 @result{} [ 1; 4 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	353 @end group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	354 @end example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	355 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	356 If two outputs are requested, @code{find} returns the row and column\n\
10840 89f4d7e294cc Grammarcheck .cc files Rik <octave@nomad.inbox5.com> parents: 10826 diff changeset	357 indices of nonzero elements of a matrix. For example:\n\
3369 f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	358 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	359 @example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	360 @group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	361 [i, j] = find (2 * eye (2))\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	362 @result{} i = [ 1; 2 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	363 @result{} j = [ 1; 2 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	364 @end group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	365 @end example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	366 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	367 If three outputs are requested, @code{find} also returns a vector\n\
10840 89f4d7e294cc Grammarcheck .cc files Rik <octave@nomad.inbox5.com> parents: 10826 diff changeset	368 containing the nonzero values. For example:\n\
3369 f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	369 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	370 @example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	371 @group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	372 [i, j, v] = find (3 * eye (2))\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	373 @result{} i = [ 1; 2 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	374 @result{} j = [ 1; 2 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	375 @result{} v = [ 3; 3 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	376 @end group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	377 @end example\n\
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	378 \n\
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	379 If two inputs are given, @var{n} indicates the maximum number of\n\
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	380 elements to find from the beginning of the matrix or vector.\n\
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	381 \n\
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	382 If three inputs are given, @var{direction} should be one of \"first\" or\n\
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	383 \"last\", requesting only the first or last @var{n} indices, respectively.\n\
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	384 However, the indices are always returned in ascending order.\n\
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	385 \n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	386 Note that this function is particularly useful for sparse matrices, as\n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	387 it extracts the non-zero elements as vectors, which can then be used to\n\
10840 89f4d7e294cc Grammarcheck .cc files Rik <octave@nomad.inbox5.com> parents: 10826 diff changeset	388 create the original matrix. For example:\n\
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	389 \n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	390 @example\n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	391 @group\n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	392 sz = size(a);\n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	393 [i, j, v] = find (a);\n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	394 b = sparse(i, j, v, sz(1), sz(2));\n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	395 @end group\n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	396 @end example\n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	397 @seealso{sparse}\n\
3369 f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	398 @end deftypefn")
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	399 {
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	400 octave_value_list retval;
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	401
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	402 int nargin = args.length ();
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	403
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	404 if (nargin > 3 \|\| nargin < 1)
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	405 {
5823 080c08b192d8 [project @ 2006-05-19 05:32:17 by jwe] jwe parents: 5307 diff changeset	406 print_usage ();
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	407 return retval;
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	408 }
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	409
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	410 // Setup the default options.
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	411 octave_idx_type n_to_find = -1;
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	412 if (nargin > 1)
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	413 {
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	414 double val = args(1).scalar_value ();
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	415
10338 21dd58bd683c fix unsafe conversion in find Jaroslav Hajek <highegg@gmail.com> parents: 10285 diff changeset	416 if (error_state \|\| (val < 0 \|\| (! xisinf (val) && val != xround (val))))
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	417 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	418 error ("find: expecting second argument to be a nonnegative integer");
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	419 return retval;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	420 }
10338 21dd58bd683c fix unsafe conversion in find Jaroslav Hajek <highegg@gmail.com> parents: 10285 diff changeset	421 else if (! xisinf (val))
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	422 n_to_find = val;
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	423 }
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	424
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	425 // Direction to do the searching (1 == forward, -1 == reverse).
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	426 int direction = 1;
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	427 if (nargin > 2)
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	428 {
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	429 direction = 0;
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	430
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	431 std::string s_arg = args(2).string_value ();
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	432
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	433 if (! error_state)
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	434 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	435 if (s_arg == "first")
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	436 direction = 1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	437 else if (s_arg == "last")
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	438 direction = -1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	439 }
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	440
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	441 if (direction == 0)
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	442 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	443 error ("find: expecting third argument to be \"first\" or \"last\"");
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	444 return retval;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	445 }
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	446 }
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	447
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	448 octave_value arg = args(0);
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	449
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	450 if (arg.is_bool_type ())
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	451 {
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	452 if (arg.is_sparse_type ())
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	453 {
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	454 SparseBoolMatrix v = arg.sparse_bool_matrix_value ();
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	455
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	456 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	457 retval = find_nonzero_elem_idx (v, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	458 n_to_find, direction);
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	459 }
9544 df0abc63c230 cache/use cached index vector on full find in logical masks Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	460 else if (nargout <= 1 && n_to_find == -1 && direction == 1)
df0abc63c230 cache/use cached index vector on full find in logical masks Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	461 {
df0abc63c230 cache/use cached index vector on full find in logical masks Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	462 // This case is equivalent to extracting indices from a logical
9894 83bd7f34f9da improve idx_vector->octave_value conversion Jaroslav Hajek <highegg@gmail.com> parents: 9732 diff changeset	463 // matrix. Try to reuse the possibly cached index vector.
83bd7f34f9da improve idx_vector->octave_value conversion Jaroslav Hajek <highegg@gmail.com> parents: 9732 diff changeset	464 retval(0) = arg.index_vector ().unmask ();
9544 df0abc63c230 cache/use cached index vector on full find in logical masks Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	465 }
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	466 else
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	467 {
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	468 boolNDArray v = arg.bool_array_value ();
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	469
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	470 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	471 retval = find_nonzero_elem_idx (v, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	472 n_to_find, direction);
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	473 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	474 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	475 else if (arg.is_integer_type ())
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	476 {
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	477 #define DO_INT_BRANCH(INTT) \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	478 else if (arg.is_ ## INTT ## _type ()) \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	479 { \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	480 INTT ## NDArray v = arg.INTT ## _array_value (); \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	481 \
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	482 if (! error_state) \
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	483 retval = find_nonzero_elem_idx (v, nargout, \
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	484 n_to_find, direction);\
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	485 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	486
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	487 if (false)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	488 ;
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	489 DO_INT_BRANCH (int8)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	490 DO_INT_BRANCH (int16)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	491 DO_INT_BRANCH (int32)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	492 DO_INT_BRANCH (int64)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	493 DO_INT_BRANCH (uint8)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	494 DO_INT_BRANCH (uint16)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	495 DO_INT_BRANCH (uint32)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	496 DO_INT_BRANCH (uint64)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	497 else
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	498 panic_impossible ();
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	499 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	500 else if (arg.is_sparse_type ())
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	501 {
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	502 if (arg.is_real_type ())
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	503 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	504 SparseMatrix v = arg.sparse_matrix_value ();
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	505
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	506 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	507 retval = find_nonzero_elem_idx (v, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	508 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	509 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	510 else if (arg.is_complex_type ())
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	511 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	512 SparseComplexMatrix v = arg.sparse_complex_matrix_value ();
5107 c891dd97b837 [project @ 2005-01-12 19:49:18 by jwe] jwe parents: 4826 diff changeset	513
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	514 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	515 retval = find_nonzero_elem_idx (v, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	516 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	517 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	518 else
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	519 gripe_wrong_type_arg ("find", arg);
5107 c891dd97b837 [project @ 2005-01-12 19:49:18 by jwe] jwe parents: 4826 diff changeset	520 }
10550 c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	521 else if (arg.is_perm_matrix ())
c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	522 {
c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	523 PermMatrix P = arg.perm_matrix_value ();
8955 6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	524
10550 c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	525 if (! error_state)
c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	526 retval = find_nonzero_elem_idx (P, nargout, n_to_find, direction);
c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	527 }
10826 9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	528 else if (arg.is_string ())
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	529 {
10826 9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	530 charNDArray chnda = arg.char_array_value ();
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	531
10826 9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	532 if (! error_state)
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	533 retval = find_nonzero_elem_idx (chnda, nargout, n_to_find, direction);
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	534 }
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	535 else if (arg.is_single_type ())
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	536 {
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	537 if (arg.is_real_type ())
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	538 {
10826 9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	539 FloatNDArray nda = arg.float_array_value ();
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7505 diff changeset	540
10826 9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	541 if (! error_state)
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	542 retval = find_nonzero_elem_idx (nda, nargout, n_to_find,
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	543 direction);
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	544 }
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	545 else if (arg.is_complex_type ())
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	546 {
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	547 FloatComplexNDArray cnda = arg.float_complex_array_value ();
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7505 diff changeset	548
10826 9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	549 if (! error_state)
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	550 retval = find_nonzero_elem_idx (cnda, nargout, n_to_find,
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	551 direction);
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	552 }
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	553 }
10826 9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	554 else if (arg.is_real_type ())
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	555 {
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	556 NDArray nda = arg.array_value ();
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	557
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	558 if (! error_state)
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	559 retval = find_nonzero_elem_idx (nda, nargout, n_to_find, direction);
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	560 }
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	561 else if (arg.is_complex_type ())
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	562 {
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	563 ComplexNDArray cnda = arg.complex_array_value ();
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	564
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	565 if (! error_state)
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	566 retval = find_nonzero_elem_idx (cnda, nargout, n_to_find, direction);
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	567 }
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	568 else
9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	569 gripe_wrong_type_arg ("find", arg);
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	570
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	571 return retval;
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	572 }
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	573
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	574 /*
10826 9e6aed3c6704 fix find for character arrays John W. Eaton <jwe@octave.org> parents: 10550 diff changeset	575 %!assert(find (char ([0, 97])), 2);
7814 87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	576 %!assert(find ([1, 0, 1, 0, 1]), [1, 3, 5]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	577 %!assert(find ([1; 0; 3; 0; 1]), [1; 3; 5]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	578 %!assert(find ([0, 0, 2; 0, 3, 0; -1, 0, 0]), [3; 5; 7]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	579
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	580 %!test
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	581 %! [i, j, v] = find ([0, 0, 2; 0, 3, 0; -1, 0, 0]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	582 %!
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	583 %! assert(i, [3; 2; 1]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	584 %! assert(j, [1; 2; 3]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	585 %! assert(v, [-1; 3; 2]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	586
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	587 %!assert(find (single([1, 0, 1, 0, 1])), [1, 3, 5]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	588 %!assert(find (single([1; 0; 3; 0; 1])), [1; 3; 5]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	589 %!assert(find (single([0, 0, 2; 0, 3, 0; -1, 0, 0])), [3; 5; 7]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	590
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	591 %!test
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	592 %! [i, j, v] = find (single([0, 0, 2; 0, 3, 0; -1, 0, 0]));
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	593 %!
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	594 %! assert(i, [3; 2; 1]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	595 %! assert(j, [1; 2; 3]);
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	596 %! assert(v, single([-1; 3; 2]));
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	597
8955 6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	598 %!test
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	599 %! pcol = [5 1 4 3 2];
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	600 %! P = eye (5) (:, pcol);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	601 %! [i, j, v] = find (P);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	602 %! [ifull, jfull, vfull] = find (full (P));
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	603 %! assert (i, ifull);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	604 %! assert (j, jfull);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	605 %! assert (all (v == 1));
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	606
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	607 %!test
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	608 %! prow = [5 1 4 3 2];
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	609 %! P = eye (5) (prow, :);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	610 %! [i, j, v] = find (P);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	611 %! [ifull, jfull, vfull] = find (full (P));
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	612 %! assert (i, ifull);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	613 %! assert (j, jfull);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	614 %! assert (all (v == 1));
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	615
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	616 %!assert (find ([2 0 1 0 5 0], 1), 1)
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	617 %!assert (find ([2 0 1 0 5 0], 2, "last"), [3, 5])
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	618
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	619 %!assert (find ([2 0 1 0 5 0], Inf), [1, 3, 5])
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	620 %!assert (find ([2 0 1 0 5 0], Inf, "last"), [1, 3, 5])
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	621
7814 87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	622 %!error <Invalid call to find.*> find ();
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	623
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	624 */

Mercurial > octave-nkf

annotate src/DLD-FUNCTIONS/find.cc @ 10840:89f4d7e294cc