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

annotate src/DLD-FUNCTIONS/find.cc @ 10550:c48b7048e720

style fixes

author	John W. Eaton <jwe@octave.org>
date	Fri, 23 Apr 2010 14:43:41 -0400
parents	21dd58bd683c
children	9e6aed3c6704

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
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	85 #define INSTANTIATE_FIND_ARRAY(T) \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	86 template octave_value_list find_nonzero_elem_idx (const Array<T>&, int, \
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	87 octave_idx_type, int)
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	88
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	89 INSTANTIATE_FIND_ARRAY(double);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	90 INSTANTIATE_FIND_ARRAY(float);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	91 INSTANTIATE_FIND_ARRAY(Complex);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	92 INSTANTIATE_FIND_ARRAY(FloatComplex);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	93 INSTANTIATE_FIND_ARRAY(bool);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	94 INSTANTIATE_FIND_ARRAY(octave_int8);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	95 INSTANTIATE_FIND_ARRAY(octave_int16);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	96 INSTANTIATE_FIND_ARRAY(octave_int32);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	97 INSTANTIATE_FIND_ARRAY(octave_int64);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	98 INSTANTIATE_FIND_ARRAY(octave_uint8);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	99 INSTANTIATE_FIND_ARRAY(octave_uint16);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	100 INSTANTIATE_FIND_ARRAY(octave_uint32);
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	101 INSTANTIATE_FIND_ARRAY(octave_uint64);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7505 diff changeset	102
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	103 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	104 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	105 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	106 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	107 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	108 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	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
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	111 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	112 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	113 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	114
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	115 // 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	116 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	117 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	118 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	119
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	120 // 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	121 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	122 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	123 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	124 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	125 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	126 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	127 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	128 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	129 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	130 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	131 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	132 OCTAVE_QUIT;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	133 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	134 start_nc = j;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	135 if (v.cidx(j+1) >= n_to_find)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	136 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	137 end_nc = j + 1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	138 break;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	139 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	140 }
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 else
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	143 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	144 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	145 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	146 OCTAVE_QUIT;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	147 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	148 end_nc = j;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	149 if (nz - v.cidx(j-1) >= n_to_find)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	150 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	151 start_nc = j - 1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	152 break;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	153 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	154 }
7505 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
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	157 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	158 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	159
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	160 // 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	161 // 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	162 // case...
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	163
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	164 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	165 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	166
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	167 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	168
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	169 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	170 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	171 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	172 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	173
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	174 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	175 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	176
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	177 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	178
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	179 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	180 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	181
9732 b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops Jaroslav Hajek <highegg@gmail.com> parents: 9544 diff changeset	182 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	183
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	184 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	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 // 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	187 // 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	188 // 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	189 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	190 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	191 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	192 OCTAVE_QUIT;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	193 if (direction < 0 && i < nz - count)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	194 continue;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	195 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	196 j_idx(cx) = static_cast<double> (j + 1);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	197 idx(cx) = j * nr + v.ridx(i) + 1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	198 val(cx) = v.data(i);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	199 cx++;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	200 if (cx == count)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	201 break;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	202 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	203 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	204 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	205 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	206 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	207
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	208 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	209 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	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 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	212 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	213
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	214 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	215 {
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	216 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	217 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	218 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	219 break;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	220
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	221 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	222 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	223 // 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	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 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	226 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	227 // 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	228
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	229 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	230 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	231 // 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	232
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	233 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	234 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	235 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	236 break;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	237
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	238 default:
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	239 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	240 break;
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	241 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	242
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	243 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	244 }
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	245
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	246 template octave_value_list find_nonzero_elem_idx (const Sparse<double>&, int,
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	247 octave_idx_type, int);
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	248
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	249 template octave_value_list find_nonzero_elem_idx (const Sparse<Complex>&, int,
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	250 octave_idx_type, int);
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	251
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	252 template octave_value_list find_nonzero_elem_idx (const Sparse<bool>&, int,
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	253 octave_idx_type, int);
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	254
8955 6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	255 octave_value_list
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	256 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	257 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	258 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	259 // 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	260 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	261
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	262 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	263 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	264
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	265 // 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	266 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	267 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	268 start_nc = 0;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	269 end_nc = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	270 n_to_find = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	271 count = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	272 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	273 else if (direction > 0)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	274 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	275 start_nc = 0;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	276 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	277 count = n_to_find;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	278 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	279 else
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 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	282 end_nc = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	283 count = n_to_find;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	284 }
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 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	287
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	288 Matrix idx (count, 1);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	289 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	290 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	291 // Every value is 1.
9732 b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops Jaroslav Hajek <highegg@gmail.com> parents: 9544 diff changeset	292 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	293
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	294 if (count > 0)
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 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	297 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	298 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	299 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	300 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	301 OCTAVE_QUIT;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	302 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	303 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	304 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	305 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	306 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	307 }
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 else
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	310 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	311 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	312 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	313 OCTAVE_QUIT;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	314 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	315 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	316 // 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	317 // 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	318 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	319 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	320 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	321 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	322 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	323 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	324 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	325 else if (scalar_arg)
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	326 {
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	327 // 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	328 idx.resize (0, 0);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	329 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	330 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	331 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	332 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	333
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	334 switch (nargout)
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 case 0:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	337 case 1:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	338 retval(0) = idx;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	339 break;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	340
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	341 case 5:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	342 retval(4) = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	343 // Fall through
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	344
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	345 case 4:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	346 retval(3) = nc;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	347 // Fall through
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	348
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	349 case 3:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	350 retval(2) = val;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	351 // Fall through!
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	352
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	353 case 2:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	354 retval(1) = j_idx;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	355 retval(0) = i_idx;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	356 break;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	357
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	358 default:
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	359 panic_impossible ();
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	360 break;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	361 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	362
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	363 return retval;
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	364 }
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	365
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	366 DEFUN_DLD (find, args, nargout,
3369 f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	367 "-- texinfo --\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	368 @deftypefn {Loadable Function} {} find (@var{x})\n\
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	369 @deftypefnx {Loadable Function} {} find (@var{x}, @var{n})\n\
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	370 @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	371 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	372 @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	373 each matrix element, Octave pretends that the columns of a matrix form one\n\
03c89ade4070 [project @ 2007-04-13 12:19:30 by jwe] jwe parents: 6254 diff changeset	374 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	375 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	376 @example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	377 @group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	378 find (eye (2))\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	379 @result{} [ 1; 4 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	380 @end group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	381 @end example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	382 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	383 If two outputs are requested, @code{find} returns the row and column\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	384 indices of nonzero elements of a matrix. For example,\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	385 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	386 @example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	387 @group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	388 [i, j] = find (2 * eye (2))\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	389 @result{} i = [ 1; 2 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	390 @result{} j = [ 1; 2 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	391 @end group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	392 @end example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	393 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	394 If three outputs are requested, @code{find} also returns a vector\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	395 containing the nonzero values. For example,\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	396 \n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	397 @example\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	398 @group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	399 [i, j, v] = find (3 * eye (2))\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	400 @result{} i = [ 1; 2 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	401 @result{} j = [ 1; 2 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	402 @result{} v = [ 3; 3 ]\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	403 @end group\n\
f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	404 @end example\n\
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	405 \n\
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	406 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	407 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	408 \n\
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	409 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	410 \"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	411 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	412 \n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	413 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	414 it extracts the non-zero elements as vectors, which can then be used to\n\
9041 853f96e8008f Cleanup documentation file matrix.texi Rik <rdrider0-list@yahoo.com> parents: 8955 diff changeset	415 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	416 \n\
f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	417 @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	418 @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	419 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	420 [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	421 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	422 @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	423 @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	424 @seealso{sparse}\n\
3369 f37ca3017116 [project @ 1999-11-21 16:26:02 by jwe] jwe parents: 3014 diff changeset	425 @end deftypefn")
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	426 {
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	427 octave_value_list retval;
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	428
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	429 int nargin = args.length ();
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	430
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	431 if (nargin > 3 \|\| nargin < 1)
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	432 {
5823 080c08b192d8 [project @ 2006-05-19 05:32:17 by jwe] jwe parents: 5307 diff changeset	433 print_usage ();
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	434 return retval;
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	435 }
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	436
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	437 // Setup the default options.
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	438 octave_idx_type n_to_find = -1;
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	439 if (nargin > 1)
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	440 {
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	441 double val = args(1).scalar_value ();
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	442
10338 21dd58bd683c fix unsafe conversion in find Jaroslav Hajek <highegg@gmail.com> parents: 10285 diff changeset	443 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	444 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	445 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	446 return retval;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	447 }
10338 21dd58bd683c fix unsafe conversion in find Jaroslav Hajek <highegg@gmail.com> parents: 10285 diff changeset	448 else if (! xisinf (val))
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	449 n_to_find = val;
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	450 }
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	451
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	452 // Direction to do the searching (1 == forward, -1 == reverse).
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	453 int direction = 1;
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	454 if (nargin > 2)
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	455 {
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	456 direction = 0;
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	457
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	458 std::string s_arg = args(2).string_value ();
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	459
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	460 if (! error_state)
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	461 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	462 if (s_arg == "first")
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	463 direction = 1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	464 else if (s_arg == "last")
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	465 direction = -1;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	466 }
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	467
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	468 if (direction == 0)
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	469 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	470 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	471 return retval;
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	472 }
6002 c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	473 }
c5d0490aad55 [project @ 2006-09-26 19:45:42 by jwe] jwe parents: 5823 diff changeset	474
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	475 octave_value arg = args(0);
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	476
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	477 if (arg.is_bool_type ())
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	478 {
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	479 if (arg.is_sparse_type ())
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	480 {
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	481 SparseBoolMatrix v = arg.sparse_bool_matrix_value ();
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	482
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	483 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	484 retval = find_nonzero_elem_idx (v, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	485 n_to_find, direction);
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	486 }
9544 df0abc63c230 cache/use cached index vector on full find in logical masks Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	487 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	488 {
df0abc63c230 cache/use cached index vector on full find in logical masks Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	489 // 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	490 // 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	491 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	492 }
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	493 else
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	494 {
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	495 boolNDArray v = arg.bool_array_value ();
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	496
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	497 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	498 retval = find_nonzero_elem_idx (v, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	499 n_to_find, direction);
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	500 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	501 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	502 else if (arg.is_integer_type ())
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	503 {
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	504 #define DO_INT_BRANCH(INTT) \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	505 else if (arg.is_ ## INTT ## _type ()) \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	506 { \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	507 INTT ## NDArray v = arg.INTT ## _array_value (); \
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	508 \
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	509 if (! error_state) \
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	510 retval = find_nonzero_elem_idx (v, nargout, \
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	511 n_to_find, direction);\
9028 e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	512 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	513
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	514 if (false)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	515 ;
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	516 DO_INT_BRANCH (int8)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	517 DO_INT_BRANCH (int16)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	518 DO_INT_BRANCH (int32)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	519 DO_INT_BRANCH (int64)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	520 DO_INT_BRANCH (uint8)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	521 DO_INT_BRANCH (uint16)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	522 DO_INT_BRANCH (uint32)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	523 DO_INT_BRANCH (uint64)
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	524 else
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	525 panic_impossible ();
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	526 }
e67dc11ed6e8 use Array<T>::find in find Jaroslav Hajek <highegg@gmail.com> parents: 9019 diff changeset	527 else if (arg.is_sparse_type ())
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	528 {
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	529 if (arg.is_real_type ())
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	530 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	531 SparseMatrix v = arg.sparse_matrix_value ();
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	532
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	533 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	534 retval = find_nonzero_elem_idx (v, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	535 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	536 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	537 else if (arg.is_complex_type ())
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	538 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	539 SparseComplexMatrix v = arg.sparse_complex_matrix_value ();
5107 c891dd97b837 [project @ 2005-01-12 19:49:18 by jwe] jwe parents: 4826 diff changeset	540
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	541 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	542 retval = find_nonzero_elem_idx (v, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	543 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	544 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	545 else
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	546 gripe_wrong_type_arg ("find", arg);
5107 c891dd97b837 [project @ 2005-01-12 19:49:18 by jwe] jwe parents: 4826 diff changeset	547 }
10550 c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	548 else if (arg.is_perm_matrix ())
c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	549 {
c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	550 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	551
10550 c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	552 if (! error_state)
c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	553 retval = find_nonzero_elem_idx (P, nargout, n_to_find, direction);
c48b7048e720 style fixes John W. Eaton <jwe@octave.org> parents: 10338 diff changeset	554 }
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	555 else
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	556 {
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7505 diff changeset	557 if (arg.is_single_type ())
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	558 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	559 if (arg.is_real_type ())
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	560 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	561 FloatNDArray nda = arg.float_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	562
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	563 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	564 retval = find_nonzero_elem_idx (nda, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	565 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	566 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	567 else if (arg.is_complex_type ())
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	568 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	569 FloatComplexNDArray cnda = arg.float_complex_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	570
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	571 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	572 retval = find_nonzero_elem_idx (cnda, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	573 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	574 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	575 }
7505 f5005d9510f4 Remove dispatched sparse functions and treat in the generic versions of the functions David Bateman <dbateman@free.fr> parents: 7017 diff changeset	576 else
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	577 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	578 if (arg.is_real_type ())
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	579 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	580 NDArray nda = arg.array_value ();
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7505 diff changeset	581
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	582 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	583 retval = find_nonzero_elem_idx (nda, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	584 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	585 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	586 else if (arg.is_complex_type ())
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	587 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	588 ComplexNDArray cnda = arg.complex_array_value ();
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7505 diff changeset	589
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	590 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	591 retval = find_nonzero_elem_idx (cnda, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	592 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	593 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	594 else if (arg.is_string ())
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	595 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	596 charNDArray cnda = arg.char_array_value ();
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7505 diff changeset	597
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	598 if (! error_state)
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	599 retval = find_nonzero_elem_idx (cnda, nargout,
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	600 n_to_find, direction);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	601 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	602 else
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	603 {
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	604 gripe_wrong_type_arg ("find", arg);
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	605 }
40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 9894 diff changeset	606 }
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	607 }
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	608
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	609 return retval;
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	610 }
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	611
295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	612 /*
7814 87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	613 %!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	614 %!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	615 %!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	616
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	617 %!test
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	618 %! [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	619 %!
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	620 %! 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	621 %! 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	622 %! 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	623
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	624 %!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	625 %!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	626 %!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	627
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	628 %!test
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	629 %! [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	630 %!
87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	631 %! 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	632 %! 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	633 %! 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	634
8955 6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	635 %!test
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	636 %! 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	637 %! P = eye (5) (:, pcol);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	638 %! [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	639 %! [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	640 %! assert (i, ifull);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	641 %! assert (j, jfull);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	642 %! assert (all (v == 1));
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	643
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	644 %!test
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	645 %! 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	646 %! P = eye (5) (prow, :);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	647 %! [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	648 %! [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	649 %! assert (i, ifull);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	650 %! assert (j, jfull);
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	651 %! assert (all (v == 1));
6d3fcbf89267 Add an override to Octave's find() for permutation matrices. Jason Riedy <jason@acm.org> parents: 8920 diff changeset	652
9310 256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	653 %!assert (find ([2 0 1 0 5 0], 1), 1)
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	654 %!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	655
256c0db275b6 fix behavior of find Jaroslav Hajek <highegg@gmail.com> parents: 9046 diff changeset	656 %!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	657 %!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	658
7814 87865ed7405f Second set of single precision test code and fix of resulting bugs David Bateman <dbateman@free.fr> parents: 7789 diff changeset	659 %!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	660
2928 295f037b4b3e [project @ 1997-05-05 05:32:33 by jwe] jwe parents: diff changeset	661 */

Mercurial > octave

annotate src/DLD-FUNCTIONS/find.cc @ 10550:c48b7048e720