max@0: // Copyright (C) 2008-2011 NICTA (www.nicta.com.au) max@0: // Copyright (C) 2008-2011 Conrad Sanderson max@0: // max@0: // This file is part of the Armadillo C++ library. max@0: // It is provided without any warranty of fitness max@0: // for any purpose. You can redistribute this file max@0: // and/or modify it under the terms of the GNU max@0: // Lesser General Public License (LGPL) as published max@0: // by the Free Software Foundation, either version 3 max@0: // of the License or (at your option) any later version. max@0: // (see http://www.opensource.org/licenses for more info) max@0: max@0: max@0: //! \addtogroup op_max max@0: //! @{ max@0: max@0: max@0: max@0: template max@0: arma_pure max@0: inline max@0: eT max@0: op_max::direct_max(const eT* const X, const uword n_elem) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: eT max_val = priv::most_neg(); max@0: max@0: uword i,j; max@0: max@0: for(i=0, j=1; j max_val) max@0: { max@0: max_val = X_i; max@0: } max@0: max@0: if(X_j > max_val) max@0: { max@0: max_val = X_j; max@0: } max@0: } max@0: max@0: max@0: if(i < n_elem) max@0: { max@0: const eT X_i = X[i]; max@0: max@0: if(X_i > max_val) max@0: { max@0: max_val = X_i; max@0: } max@0: } max@0: max@0: return max_val; max@0: } max@0: max@0: max@0: max@0: template max@0: inline max@0: eT max@0: op_max::direct_max(const eT* const X, const uword n_elem, uword& index_of_max_val) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: eT max_val = priv::most_neg(); max@0: max@0: uword best_index = 0; max@0: max@0: uword i,j; max@0: max@0: for(i=0, j=1; j max_val) max@0: { max@0: max_val = X_i; max@0: best_index = i; max@0: } max@0: max@0: if(X_j > max_val) max@0: { max@0: max_val = X_j; max@0: best_index = j; max@0: } max@0: } max@0: max@0: max@0: if(i < n_elem) max@0: { max@0: const eT X_i = X[i]; max@0: max@0: if(X_i > max_val) max@0: { max@0: max_val = X_i; max@0: best_index = i; max@0: } max@0: } max@0: max@0: index_of_max_val = best_index; max@0: max@0: return max_val; max@0: } max@0: max@0: max@0: max@0: template max@0: inline max@0: eT max@0: op_max::direct_max(const Mat& X, const uword row) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: const uword X_n_cols = X.n_cols; max@0: max@0: eT max_val = priv::most_neg(); max@0: max@0: for(uword col=0; col max_val) max@0: { max@0: max_val = tmp_val; max@0: } max@0: } max@0: max@0: return max_val; max@0: } max@0: max@0: max@0: max@0: template max@0: inline max@0: eT max@0: op_max::direct_max(const subview& X) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: const uword X_n_elem = X.n_elem; max@0: max@0: eT max_val = priv::most_neg(); max@0: max@0: for(uword i=0; i max_val) max@0: { max@0: max_val = tmp_val; max@0: } max@0: } max@0: max@0: return max_val; max@0: } max@0: max@0: max@0: max@0: template max@0: inline max@0: eT max@0: op_max::direct_max(const diagview& X) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: const uword X_n_elem = X.n_elem; max@0: max@0: eT max_val = priv::most_neg(); max@0: max@0: for(uword i=0; i max_val) max@0: { max@0: max_val = tmp_val; max@0: } max@0: } max@0: max@0: return max_val; max@0: } max@0: max@0: max@0: max@0: //! \brief max@0: //! For each row or for each column, find the maximum value. max@0: //! The result is stored in a dense matrix that has either one column or one row. max@0: //! The dimension, for which the maxima are found, is set via the max() function. max@0: template max@0: inline max@0: void max@0: op_max::apply(Mat& out, const Op& in) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: typedef typename T1::elem_type eT; max@0: max@0: const unwrap_check tmp(in.m, out); max@0: const Mat& X = tmp.M; max@0: max@0: const uword dim = in.aux_uword_a; max@0: arma_debug_check( (dim > 1), "max(): incorrect usage. dim must be 0 or 1"); max@0: max@0: const uword X_n_rows = X.n_rows; max@0: const uword X_n_cols = X.n_cols; max@0: max@0: if(dim == 0) max@0: { max@0: arma_extra_debug_print("op_max::apply(), dim = 0"); max@0: max@0: arma_debug_check( (X_n_rows == 0), "max(): given object has zero rows" ); max@0: max@0: out.set_size(1, X_n_cols); max@0: max@0: eT* out_mem = out.memptr(); max@0: max@0: for(uword col=0; col max@0: inline max@0: std::complex max@0: op_max::direct_max(const std::complex* const X, const uword n_elem) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: uword index = 0; max@0: T max_val = priv::most_neg(); max@0: max@0: for(uword i=0; i max_val) max@0: { max@0: max_val = tmp_val; max@0: index = i; max@0: } max@0: } max@0: max@0: return X[index]; max@0: } max@0: max@0: max@0: max@0: template max@0: inline max@0: std::complex max@0: op_max::direct_max(const std::complex* const X, const uword n_elem, uword& index_of_max_val) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: uword index = 0; max@0: T max_val = priv::most_neg(); max@0: max@0: for(uword i=0; i max_val) max@0: { max@0: max_val = tmp_val; max@0: index = i; max@0: } max@0: } max@0: max@0: index_of_max_val = index; max@0: max@0: return X[index]; max@0: } max@0: max@0: max@0: max@0: template max@0: inline max@0: std::complex max@0: op_max::direct_max(const Mat< std::complex >& X, const uword row) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: const uword X_n_cols = X.n_cols; max@0: max@0: uword index = 0; max@0: T max_val = priv::most_neg(); max@0: max@0: for(uword col=0; col max_val) max@0: { max@0: max_val = tmp_val; max@0: index = col; max@0: } max@0: } max@0: max@0: return X.at(row,index); max@0: } max@0: max@0: max@0: max@0: template max@0: inline max@0: std::complex max@0: op_max::direct_max(const subview< std::complex >& X) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: const uword X_n_elem = X.n_elem; max@0: max@0: uword index = 0; max@0: T max_val = priv::most_neg(); max@0: max@0: for(uword i=0; i max_val) max@0: { max@0: max_val = tmp_val; max@0: index = i; max@0: } max@0: } max@0: max@0: return X[index]; max@0: } max@0: max@0: max@0: max@0: template max@0: inline max@0: std::complex max@0: op_max::direct_max(const diagview< std::complex >& X) max@0: { max@0: arma_extra_debug_sigprint(); max@0: max@0: const uword X_n_elem = X.n_elem; max@0: max@0: uword index = 0; max@0: T max_val = priv::most_neg(); max@0: max@0: for(uword i=0; i max_val) max@0: { max@0: max_val = tmp_val; max@0: index = i; max@0: } max@0: } max@0: max@0: return X[index]; max@0: } max@0: max@0: max@0: max@0: //! @}