Mercurial > hg > segmenter-vamp-plugin
diff armadillo-2.4.4/include/armadillo_bits/op_median_bones.hpp @ 0:8b6102e2a9b0
Armadillo Library
author | maxzanoni76 <max.zanoni@eecs.qmul.ac.uk> |
---|---|
date | Wed, 11 Apr 2012 09:27:06 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/armadillo-2.4.4/include/armadillo_bits/op_median_bones.hpp Wed Apr 11 09:27:06 2012 +0100 @@ -0,0 +1,85 @@ +// Copyright (C) 2009-2010 NICTA (www.nicta.com.au) +// Copyright (C) 2009-2010 Conrad Sanderson +// +// This file is part of the Armadillo C++ library. +// It is provided without any warranty of fitness +// for any purpose. You can redistribute this file +// and/or modify it under the terms of the GNU +// Lesser General Public License (LGPL) as published +// by the Free Software Foundation, either version 3 +// of the License or (at your option) any later version. +// (see http://www.opensource.org/licenses for more info) + + +//! \addtogroup op_median +//! @{ + + +template<typename T> +struct arma_cx_median_packet + { + T val; + uword index; + }; + + + +template<typename T> +inline +bool +operator< (const arma_cx_median_packet<T>& A, const arma_cx_median_packet<T>& B) + { + return A.val < B.val; + } + + + +//! Class for finding median values of a matrix +class op_median + { + public: + + template<typename eT> + arma_inline static eT robust_mean(const eT A, const eT B); + + template<typename eT> + inline static eT direct_median(std::vector<eT>& X); + + template<typename eT> + inline static eT direct_median(const eT* X, const uword n_elem); + + template<typename eT> + inline static eT direct_median(const subview<eT>& X); + + template<typename eT> + inline static eT direct_median(const diagview<eT>& X); + + template<typename T1> + inline static void apply(Mat<typename T1::elem_type>& out, const Op<T1,op_median>& in); + + + // + // for complex numbers + + template<typename T> + arma_inline static std::complex<T> robust_mean(const std::complex<T>& A, const std::complex<T>& B); + + template<typename T> + inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, std::vector< arma_cx_median_packet<T> >& X); + + template<typename T> + inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, const std::complex<T>* X, const uword n_elem); + + template<typename T> + inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, const subview< std::complex<T> >& X); + + template<typename T> + inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, const diagview< std::complex<T> >& X); + + template<typename T, typename T1> + inline static void apply(Mat< std::complex<T> >& out, const Op<T1,op_median>& in); + + + }; + +//! @}