Mercurial > octave-dspies
view liboctave/util/min-with-nnz.h @ 19006:2e0613dadfee draft
All calls to "find" use the same generic implementation (bug #42408, 42421)
* find.cc: Rewrite.
Move generic "find" logic to find.h
(Ffind) : Changed calls to find_nonzero_elem_idx to find_templated
Added unit test for bug #42421
* Array.cc (and .h) (Array::find): Deleted function. Replaced with find::find(Array)
from find.h
* Array.h: Added typedef for array_iterator (in nz-iterators.h) as
Array::iter_type
* DiagArray2.h: Added typedef for diag_iterator (in nz-iterators.h) as
DiagArray2::iter_type
* PermMatrix.h: Added typedef for perm_iterator (in nz-iterators.h) as
PermMatrix::iter_type
Also added typedef for bool as PermMatrix::element_type
(not octave_idx_type)
Added an nnz() function (which is an alias for perm_length) and a
perm_elem(i) function for retrieving the ith element of the permutation
* Sparse.h: Added typedef for sparse_iterator (in nz-iterators.h) as
Sparse::iter_type
Added a short comment documenting the the argument to the numel
function
* idx-vector.cc (idx_vector::idx_mask_rep::as_array): Changed Array.find to
find::find(Array) (in find.h)
* (new file) find.h
* (new file) interp-idx.h: Simple methods for converting between interpreter
index type and internal octave_idx_type/row-col pair
* (new file) min-with-nnz.h: Fast methods for taking an arbitrary matrix M and
an octave_idx_type n and finding min(M.nnz(), n)
* (new file) nz-iterators.h: Iterators for traversing (in column-major order)
the nonzero elements of any array or matrix backwards or forwards
* (new file) direction.h: Generic methods for simplifying code has to deal with
a "backwards or forwards" template argument
* build-sparse-tests.sh: Removed 5-return-value calls to "find" in unit-tests;
Admittedly this commit breaks this "feature" which was undocumented and only
partially supported to begin with (ie never worked for full matrices,
permutation matrices, or diagonal matrices)
author | David Spies <dnspies@gmail.com> |
---|---|
date | Tue, 17 Jun 2014 16:41:11 -0600 |
parents | |
children |
line wrap: on
line source
/* Copyright (C) 2014 David Spies This file is part of Octave. Octave is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 3 of the License, or (at your option) any later version. Octave is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with Octave; see the file COPYING. If not, see <http://www.gnu.org/licenses/>. */ #if !defined (octave_min_with_nnz_h) #define octave_min_with_nnz_h 1 #include "Sparse.h" #include "direction.h" // Generic efficient methods for finding the equivalent of // min(arr.nnz(), minwith). In case arr.nnz() happens to run // in O(arr.numel()) time, this operation will short-circuit and return // as soon as it counts minwith nonzero elements. // But if arr has a constant-time nnz() operation, then this will use // that instead and run in constant-time. template<typename M> octave_idx_type min_with_nnz (M arr, octave_idx_type minwith) { octave_idx_type count; typename M::iter_type iter (arr); dir_handler<FORWARD> dirc; for (iter.begin (dirc), count = 0; !iter.finished (dirc) && count < minwith; iter.step (dirc), ++count); return count; } template<typename T> octave_idx_type min_with_nnz (Sparse<T> arr, octave_idx_type minwith) { return std::min (arr.nnz (), minwith); } // FIXME: I want to add one for PermMatrix, but for some reason C++ doesn't // like it when you overload templated functions with non-templated ones. // Except that PermMatrix is not templated. Anyone know how to handle that? // Should I add a dummy template parameter? (that seems like quite an ugly hack) #endif