max@0: // Copyright (C) 2009-2010 NICTA (www.nicta.com.au) max@0: // Copyright (C) 2009-2010 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_median max@0: //! @{ max@0: max@0: max@0: template max@0: struct arma_cx_median_packet max@0: { max@0: T val; max@0: uword index; max@0: }; max@0: max@0: max@0: max@0: template max@0: inline max@0: bool max@0: operator< (const arma_cx_median_packet& A, const arma_cx_median_packet& B) max@0: { max@0: return A.val < B.val; max@0: } max@0: max@0: max@0: max@0: //! Class for finding median values of a matrix max@0: class op_median max@0: { max@0: public: max@0: max@0: template max@0: arma_inline static eT robust_mean(const eT A, const eT B); max@0: max@0: template max@0: inline static eT direct_median(std::vector& X); max@0: max@0: template max@0: inline static eT direct_median(const eT* X, const uword n_elem); max@0: max@0: template max@0: inline static eT direct_median(const subview& X); max@0: max@0: template max@0: inline static eT direct_median(const diagview& X); max@0: max@0: template max@0: inline static void apply(Mat& out, const Op& in); max@0: max@0: max@0: // max@0: // for complex numbers max@0: max@0: template max@0: arma_inline static std::complex robust_mean(const std::complex& A, const std::complex& B); max@0: max@0: template max@0: inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, std::vector< arma_cx_median_packet >& X); max@0: max@0: template max@0: inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, const std::complex* X, const uword n_elem); max@0: max@0: template max@0: inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, const subview< std::complex >& X); max@0: max@0: template max@0: inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, const diagview< std::complex >& X); max@0: max@0: template max@0: inline static void apply(Mat< std::complex >& out, const Op& in); max@0: max@0: max@0: }; max@0: max@0: //! @}