octave-lyh: src/DLD-FUNCTIONS/lookup.cc annotate

annotate src/DLD-FUNCTIONS/lookup.cc @ 11560:05b5bcdb09c8

validate opts in lookup

author	Jaroslav Hajek <highegg@gmail.com>
date	Mon, 17 Jan 2011 22:19:29 +0100
parents	01f703952eff
children	57632dea2446

rev	line source
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	1 /*
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	2
11523 fd0a3ac60b0e update copyright notices John W. Eaton <jwe@octave.org> parents: 11346 diff changeset	3 Copyright (C) 2008-2011 VZLU Prague a.s., Czech Republic
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	4
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	5 This file is part of Octave.
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	6
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	7 Octave is free software; you can redistribute it and/or modify it
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	8 under the terms of the GNU General Public License as published by
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	9 the Free Software Foundation; either version 3 of the License, or (at
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	10 your option) any later version.
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	11
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	12 Octave is distributed in the hope that it will be useful, but
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	13 WITHOUT ANY WARRANTY; without even the implied warranty of
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	15 General Public License for more details.
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	16
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	17 You should have received a copy of the GNU General Public License
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	18 along with Octave; see the file COPYING. If not, see
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	19 <http://www.gnu.org/licenses/>.
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	20
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	21 */
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	22
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	23 // Author: Jaroslav Hajek <highegg@gmail.com>
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	24
10039 1e306aa51d6c Include config.h before another header file. Thomas Treichl parents: 9923 diff changeset	25 #ifdef HAVE_CONFIG_H
1e306aa51d6c Include config.h before another header file. Thomas Treichl parents: 9923 diff changeset	26 #include <config.h>
1e306aa51d6c Include config.h before another header file. Thomas Treichl parents: 9923 diff changeset	27 #endif
1e306aa51d6c Include config.h before another header file. Thomas Treichl parents: 9923 diff changeset	28
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	29 #include <cctype>
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	30 #include <functional>
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	31 #include <algorithm>
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	32
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	33 #include "dNDArray.h"
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	34 #include "CNDArray.h"
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	35
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	36 #include "Cell.h"
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	37 #include "defun-dld.h"
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	38 #include "error.h"
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	39 #include "gripes.h"
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	40 #include "oct-obj.h"
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	41 #include "ov.h"
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	42
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	43 static
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	44 bool
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	45 contains_char (const std::string& str, char c)
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	46 {
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	47 return (str.find (c) != std::string::npos
10154 40dfc0c99116 DLD-FUNCTIONS/.cc: untabify John W. Eaton <jwe@octave.org>* parents: 10076 diff changeset	48 \|\| str.find (std::toupper (c)) != std::string::npos);
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	49 }
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	50
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	51 // case-insensitive character comparison functors
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	52 struct icmp_char_lt : public std::binary_function<char, char, bool>
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	53 {
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	54 bool operator () (char x, char y) const
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	55 { return std::toupper (x) < std::toupper (y); }
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	56 };
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	57
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	58 struct icmp_char_gt : public std::binary_function<char, char, bool>
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	59 {
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	60 bool operator () (char x, char y) const
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	61 { return std::toupper (x) > std::toupper (y); }
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	62 };
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	63
10076 4b270d1540f7 avoid various GCC warnings John W. Eaton <jwe@octave.org> parents: 10039 diff changeset	64 // FIXME -- maybe these should go elsewhere?
4b270d1540f7 avoid various GCC warnings John W. Eaton <jwe@octave.org> parents: 10039 diff changeset	65 // FIXME -- are they even needed now?
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	66 // case-insensitive ascending comparator
10076 4b270d1540f7 avoid various GCC warnings John W. Eaton <jwe@octave.org> parents: 10039 diff changeset	67 #if 0
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	68 static bool
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	69 stri_comp_lt (const std::string& a, const std::string& b)
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	70 {
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	71 return std::lexicographical_compare (a.begin (), a.end (),
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	72 b.begin (), b.end (),
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	73 icmp_char_lt());
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	74 }
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	75
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	76 // case-insensitive descending comparator
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	77 static bool
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	78 stri_comp_gt (const std::string& a, const std::string& b)
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	79 {
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	80 return std::lexicographical_compare (a.begin (), a.end (),
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	81 b.begin (), b.end (),
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	82 icmp_char_gt());
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	83 }
10076 4b270d1540f7 avoid various GCC warnings John W. Eaton <jwe@octave.org> parents: 10039 diff changeset	84 #endif
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	85
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	86 template <class T>
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	87 inline sortmode
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	88 get_sort_mode (const Array<T>& array,
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	89 typename octave_sort<T>::compare_fcn_type desc_comp
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	90 = octave_sort<T>::descending_compare)
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	91 {
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	92 octave_idx_type n = array.numel ();
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	93 if (n > 1 && desc_comp (array (0), array (n-1)))
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	94 return DESCENDING;
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	95 else
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	96 return ASCENDING;
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	97 }
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	98
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	99 // FIXME: perhaps there should be octave_value::lookup?
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	100 // The question is, how should it behave w.r.t. the second argument's type.
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	101 // We'd need a dispatch on two arguments. Hmmm...
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	102
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	103 #define INT_ARRAY_LOOKUP(TYPE) \
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	104 (table.is_ ## TYPE ## _type () && y.is_ ## TYPE ## _type ()) \
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	105 retval = do_numeric_lookup (table.TYPE ## _array_value (), \
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	106 y.TYPE ## _array_value (), \
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	107 left_inf, right_inf, \
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	108 match_idx, match_bool);
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	109 template <class ArrayT>
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	110 static octave_value
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	111 do_numeric_lookup (const ArrayT& array, const ArrayT& values,
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	112 bool left_inf, bool right_inf,
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	113 bool match_idx, bool match_bool)
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	114 {
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	115 octave_value retval;
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	116
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	117 Array<octave_idx_type> idx = array.lookup (values);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	118 octave_idx_type n = array.numel (), nval = values.numel ();
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	119
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	120 // Post-process.
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	121 if (match_bool)
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	122 {
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	123 boolNDArray match (idx.dims ());
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	124 for (octave_idx_type i = 0; i < nval; i++)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	125 {
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	126 octave_idx_type j = idx.xelem (i);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	127 match.xelem (i) = j != 0 && values(i) == array(j-1);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	128 }
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	129
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	130 retval = match;
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	131 }
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	132 else if (match_idx \|\| left_inf \|\| right_inf)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	133 {
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	134 if (match_idx)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	135 {
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	136 NDArray ridx (idx.dims ());
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	137
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	138 for (octave_idx_type i = 0; i < nval; i++)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	139 {
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	140 octave_idx_type j = idx.xelem (i);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	141 ridx.xelem (i) = (j != 0 && values(i) == array(j-1)) ? j : 0;
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	142 }
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	143
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	144 retval = ridx;
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	145 }
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	146 else if (left_inf && right_inf)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	147 {
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	148 // Results in valid indices. Optimize using lazy index.
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	149 octave_idx_type zero = 0;
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	150 for (octave_idx_type i = 0; i < nval; i++)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	151 {
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	152 octave_idx_type j = idx.xelem (i) - 1;
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	153 idx.xelem (i) = std::max (zero, std::min (j, n-2));
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	154 }
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	155
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	156 retval = idx_vector (idx);
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	157 }
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	158 else if (left_inf)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	159 {
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	160 // Results in valid indices. Optimize using lazy index.
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	161 octave_idx_type zero = 0;
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	162 for (octave_idx_type i = 0; i < nval; i++)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	163 {
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	164 octave_idx_type j = idx.xelem (i) - 1;
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	165 idx.xelem (i) = std::max (zero, j);
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	166 }
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	167
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	168 retval = idx_vector (idx);
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	169 }
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	170 else if (right_inf)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	171 {
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	172 NDArray ridx (idx.dims ());
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	173
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	174 for (octave_idx_type i = 0; i < nval; i++)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	175 {
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	176 octave_idx_type j = idx.xelem (i);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	177 ridx.xelem (i) = std::min (j, n-1);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	178 }
10327 76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	179
76cf6dd20f1a take advantage of lazy index optimization in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10258 diff changeset	180 retval = ridx;
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	181 }
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	182 }
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	183 else
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	184 retval = idx;
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	185
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	186 return retval;
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	187 }
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	188
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	189 DEFUN_DLD (lookup, args, ,
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	190 "-- texinfo --\n\
8617 4b26e9fbbb37 fix lookup's docstring Jaroslav Hajek <highegg@gmail.com> parents: 7924 diff changeset	191 @deftypefn {Loadable Function} {@var{idx} =} lookup (@var{table}, @var{y}, @var{opt})\n\
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	192 Lookup values in a sorted table. Usually used as a prelude to\n\
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	193 interpolation.\n\
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	194 \n\
9366 25b52645de34 improve lookup's inline docs Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	195 If table is increasing and @code{idx = lookup (table, y)}, then\n\
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	196 @code{table(idx(i)) <= y(i) < table(idx(i+1))} for all @code{y(i)}\n\
8617 4b26e9fbbb37 fix lookup's docstring Jaroslav Hajek <highegg@gmail.com> parents: 7924 diff changeset	197 within the table. If @code{y(i) < table (1)} then\n\
9730 dd4aa19c3053 update docstring of lookup Jaroslav Hajek <highegg@gmail.com> parents: 9366 diff changeset	198 @code{idx(i)} is 0. If @code{y(i) >= table(end)} or @code{isnan (y(i))} then\n\
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	199 @code{idx(i)} is @code{n}.\n\
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	200 \n\
9366 25b52645de34 improve lookup's inline docs Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	201 If the table is decreasing, then the tests are reversed.\n\
25b52645de34 improve lookup's inline docs Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	202 For non-strictly monotonic tables, empty intervals are always skipped.\n\
25b52645de34 improve lookup's inline docs Jaroslav Hajek <highegg@gmail.com> parents: 9351 diff changeset	203 The result is undefined if @var{table} is not monotonic, or if\n\
9730 dd4aa19c3053 update docstring of lookup Jaroslav Hajek <highegg@gmail.com> parents: 9366 diff changeset	204 @var{table} contains a NaN.\n\
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	205 \n\
10846 a4f482e66b65 Grammarcheck more of the documentation. Rik <octave@nomad.inbox5.com> parents: 10840 diff changeset	206 The complexity of the lookup is O(M*log(N)) where N is the size of\n\
a4f482e66b65 Grammarcheck more of the documentation. Rik <octave@nomad.inbox5.com> parents: 10840 diff changeset	207 @var{table} and M is the size of @var{y}. In the special case when @var{y}\n\
a4f482e66b65 Grammarcheck more of the documentation. Rik <octave@nomad.inbox5.com> parents: 10840 diff changeset	208 is also sorted, the complexity is O(min(M*log(N),M+N)).\n\
8617 4b26e9fbbb37 fix lookup's docstring Jaroslav Hajek <highegg@gmail.com> parents: 7924 diff changeset	209 \n\
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	210 @var{table} and @var{y} can also be cell arrays of strings\n\
9064 7c02ec148a3c Check grammar on all .cc files Rik <rdrider0-list@yahoo.com> parents: 8920 diff changeset	211 (or @var{y} can be a single string). In this case, string lookup\n\
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	212 is performed using lexicographical comparison.\n\
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	213 \n\
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	214 If @var{opts} is specified, it shall be a string with letters indicating\n\
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	215 additional options.\n\
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	216 \n\
10687 a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	217 @table @code\n\
a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	218 @item m\n\
a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	219 @code{table(idx(i)) == val(i)} if @code{val(i)}\n\
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	220 occurs in table; otherwise, @code{idx(i)} is zero.\n\
10840 89f4d7e294cc Grammarcheck .cc files Rik <octave@nomad.inbox5.com> parents: 10711 diff changeset	221 \n\
10687 a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	222 @item b\n\
a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	223 @code{idx(i)} is a logical 1 or 0, indicating whether\n\
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	224 @code{val(i)} is contained in table or not.\n\
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	225 \n\
10687 a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	226 @item l\n\
a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	227 For numeric lookups\n\
9064 7c02ec148a3c Check grammar on all .cc files Rik <rdrider0-list@yahoo.com> parents: 8920 diff changeset	228 the leftmost subinterval shall be extended to infinity (i.e., all indices\n\
10687 a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	229 at least 1)\n\
10840 89f4d7e294cc Grammarcheck .cc files Rik <octave@nomad.inbox5.com> parents: 10711 diff changeset	230 \n\
10687 a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	231 @item r\n\
a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	232 For numeric lookups\n\
a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	233 the rightmost subinterval shall be extended to infinity (i.e., all indices\n\
a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	234 at most n-1).\n\
a8ce6bdecce5 Improve documentation strings. Rik <octave@nomad.inbox5.com> parents: 10407 diff changeset	235 @end table\n\
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	236 @end deftypefn")
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	237 {
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	238 octave_value retval;
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	239
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	240 int nargin = args.length ();
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	241
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	242 if (nargin < 2 \|\| nargin > 3 \|\| (nargin == 3 && ! args(2).is_string ()))
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	243 {
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	244 print_usage ();
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	245 return retval;
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	246 }
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	247
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	248 octave_value table = args(0), y = args(1);
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	249 if (table.ndims () > 2 \|\| (table.columns () > 1 && table.rows () > 1))
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	250 warning ("lookup: table is not a vector");
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	251
9344 0c4e6a3d6e3e support char arrays in lookup Jaroslav Hajek <highegg@gmail.com> parents: 9341 diff changeset	252 bool num_case = ((table.is_numeric_type () && y.is_numeric_type ())
0c4e6a3d6e3e support char arrays in lookup Jaroslav Hajek <highegg@gmail.com> parents: 9341 diff changeset	253 \|\| (table.is_char_matrix () && y.is_char_matrix ()));
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	254 bool str_case = table.is_cellstr () && (y.is_string () \|\| y.is_cellstr ());
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	255 bool left_inf = false;
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	256 bool right_inf = false;
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	257 bool match_idx = false;
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	258 bool match_bool = false;
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	259
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	260 if (nargin == 3)
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	261 {
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	262 std::string opt = args(2).string_value ();
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	263 left_inf = contains_char (opt, 'l');
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	264 right_inf = contains_char (opt, 'r');
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	265 match_idx = contains_char (opt, 'm');
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	266 match_bool = contains_char (opt, 'b');
11560 05b5bcdb09c8 validate opts in lookup Jaroslav Hajek <highegg@gmail.com> parents: 11553 diff changeset	267 if (opt.find_first_not_of ("lrmb") != std::string::npos)
05b5bcdb09c8 validate opts in lookup Jaroslav Hajek <highegg@gmail.com> parents: 11553 diff changeset	268 {
05b5bcdb09c8 validate opts in lookup Jaroslav Hajek <highegg@gmail.com> parents: 11553 diff changeset	269 error ("lookup: unrecognized option: %c",
05b5bcdb09c8 validate opts in lookup Jaroslav Hajek <highegg@gmail.com> parents: 11553 diff changeset	270 opt[opt.find_first_not_of ("lrmb")]);
05b5bcdb09c8 validate opts in lookup Jaroslav Hajek <highegg@gmail.com> parents: 11553 diff changeset	271 return retval;
05b5bcdb09c8 validate opts in lookup Jaroslav Hajek <highegg@gmail.com> parents: 11553 diff changeset	272 }
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	273 }
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	274
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	275 if ((match_idx \|\| match_bool) && (left_inf \|\| right_inf))
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	276 error ("lookup: m, b cannot be specified with l or r");
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	277 else if (match_idx && match_bool)
11553 01f703952eff Improve docstrings for functions in DLD-FUNCTIONS directory. Rik <octave@nomad.inbox5.com> parents: 11523 diff changeset	278 error ("lookup: only one of m or b can be specified");
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	279 else if (str_case && (left_inf \|\| right_inf))
11553 01f703952eff Improve docstrings for functions in DLD-FUNCTIONS directory. Rik <octave@nomad.inbox5.com> parents: 11523 diff changeset	280 error ("lookup: l, r are not recognized for string lookups");
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	281
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	282 if (error_state)
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	283 return retval;
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	284
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	285 if (num_case)
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	286 {
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	287
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	288 // In the case of a complex array, absolute values will be used for compatibility
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	289 // (though it's not too meaningful).
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	290
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	291 if (table.is_complex_type ())
9813 8fa32b527d9a improve & partially revert previous change Jaroslav Hajek <highegg@gmail.com> parents: 9812 diff changeset	292 table = table.abs ();
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	293
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	294 if (y.is_complex_type ())
9813 8fa32b527d9a improve & partially revert previous change Jaroslav Hajek <highegg@gmail.com> parents: 9812 diff changeset	295 y = y.abs ();
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	296
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	297 Array<octave_idx_type> idx;
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	298
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	299 // PS: I learned this from data.cc
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	300 if INT_ARRAY_LOOKUP (int8)
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	301 else if INT_ARRAY_LOOKUP (int16)
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	302 else if INT_ARRAY_LOOKUP (int32)
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	303 else if INT_ARRAY_LOOKUP (int64)
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	304 else if INT_ARRAY_LOOKUP (uint8)
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	305 else if INT_ARRAY_LOOKUP (uint16)
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	306 else if INT_ARRAY_LOOKUP (uint32)
de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	307 else if INT_ARRAY_LOOKUP (uint64)
9344 0c4e6a3d6e3e support char arrays in lookup Jaroslav Hajek <highegg@gmail.com> parents: 9341 diff changeset	308 else if (table.is_char_matrix () && y.is_char_matrix ())
0c4e6a3d6e3e support char arrays in lookup Jaroslav Hajek <highegg@gmail.com> parents: 9341 diff changeset	309 retval = do_numeric_lookup (table.char_array_value (),
0c4e6a3d6e3e support char arrays in lookup Jaroslav Hajek <highegg@gmail.com> parents: 9341 diff changeset	310 y.char_array_value (),
0c4e6a3d6e3e support char arrays in lookup Jaroslav Hajek <highegg@gmail.com> parents: 9341 diff changeset	311 left_inf, right_inf,
0c4e6a3d6e3e support char arrays in lookup Jaroslav Hajek <highegg@gmail.com> parents: 9341 diff changeset	312 match_idx, match_bool);
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	313 else if (table.is_single_type () \|\| y.is_single_type ())
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	314 retval = do_numeric_lookup (table.float_array_value (),
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	315 y.float_array_value (),
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	316 left_inf, right_inf,
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	317 match_idx, match_bool);
7789 82be108cc558 First attempt at single precision tyeps David Bateman <dbateman@free.fr> parents: 7671 diff changeset	318 else
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	319 retval = do_numeric_lookup (table.array_value (),
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	320 y.array_value (),
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	321 left_inf, right_inf,
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	322 match_idx, match_bool);
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	323
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	324 }
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	325 else if (str_case)
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	326 {
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	327 Array<std::string> str_table = table.cellstr_value ();
10407 2516ca5763e9 fix deprecated ctor in lookup Jaroslav Hajek <highegg@gmail.com> parents: 10327 diff changeset	328 Array<std::string> str_y (1, 1);
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	329
8814 de16ebeef93d improve lookup, provide Array<T>::lookup Jaroslav Hajek <highegg@gmail.com> parents: 8646 diff changeset	330 if (y.is_cellstr ())
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	331 str_y = y.cellstr_value ();
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	332 else
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	333 str_y(0) = y.string_value ();
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	334
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	335 Array<octave_idx_type> idx = str_table.lookup (str_y);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	336 octave_idx_type nval = str_y.numel ();
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	337
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	338 // Post-process.
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	339 if (match_bool)
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	340 {
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	341 boolNDArray match (idx.dims ());
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	342 for (octave_idx_type i = 0; i < nval; i++)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	343 {
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	344 octave_idx_type j = idx.xelem (i);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	345 match.xelem (i) = j != 0 && str_y(i) == str_table(j-1);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	346 }
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	347
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	348 retval = match;
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	349 }
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	350 else if (match_idx)
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	351 {
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	352 NDArray ridx (idx.dims ());
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	353 if (match_idx)
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	354 {
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	355 for (octave_idx_type i = 0; i < nval; i++)
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	356 {
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	357 octave_idx_type j = idx.xelem (i);
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	358 ridx.xelem (i) = (j != 0 && str_y(i) == str_table(j-1)) ? j : 0;
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	359 }
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	360 }
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	361
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	362 retval = ridx;
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	363 }
9921 7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	364 else
7c8392a034e6 fix & improve lookup API Jaroslav Hajek <highegg@gmail.com> parents: 9813 diff changeset	365 retval = idx;
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	366 }
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	367 else
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	368 print_usage ();
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	369
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	370 return retval;
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	371
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	372 }
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	373
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	374 /*
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	375 %!assert (lookup(1:3, 0.5), 0) # value before table
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	376 %!assert (lookup(1:3, 3.5), 3) # value after table error
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	377 %!assert (lookup(1:3, 1.5), 1) # value within table error
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	378 %!assert (lookup(1:3, [3,2,1]), [3,2,1])
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	379 %!assert (lookup([1:4]', [1.2, 3.5]'), [1, 3]');
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	380 %!assert (lookup([1:4], [1.2, 3.5]'), [1, 3]');
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	381 %!assert (lookup([1:4]', [1.2, 3.5]), [1, 3]);
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	382 %!assert (lookup([1:4], [1.2, 3.5]), [1, 3]);
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	383 %!assert (lookup(1:3, [3, 2, 1]), [3, 2, 1]);
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	384 %!assert (lookup([3:-1:1], [3.5, 3, 1.2, 2.5, 2.5]), [0, 1, 2, 1, 1])
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	385 %!assert (isempty(lookup([1:3], [])))
4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	386 %!assert (isempty(lookup([1:3]', [])))
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	387 %!assert (lookup(1:3, [1, 2; 3, 0.5]), [1, 2; 3, 0]);
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	388 %!assert (lookup(1:4, [1, 1.2; 3, 2.5], "m"), [1, 0; 3, 0]);
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	389 %!assert (lookup(4:-1:1, [1, 1.2; 3, 2.5], "m"), [4, 0; 2, 0]);
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	390 %!assert (lookup(1:4, [1, 1.2; 3, 2.5], "b"), logical ([1, 0; 3, 0]));
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	391 %!assert (lookup(4:-1:1, [1, 1.2; 3, 2.5], "b"), logical ([4, 0; 2, 0]));
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	392 %!
9341 9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	393 %!assert (lookup({"apple","lemon","orange"}, {"banana","kiwi"; "ananas","mango"}), [1,1;0,2])
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	394 %!assert (lookup({"apple","lemon","orange"}, "potato"), 3)
9fd5c56ce57a extend lookup capabilities Jaroslav Hajek <highegg@gmail.com> parents: 9064 diff changeset	395 %!assert (lookup({"orange","lemon","apple"}, "potato"), 0)
7671 4fbaba9abec1 implement compiled binary lookup Jaroslav Hajek <highegg@gmail.com> parents: diff changeset	396 */

Mercurial > hg > octave-lyh

annotate src/DLD-FUNCTIONS/lookup.cc @ 11560:05b5bcdb09c8