Chris@49: // Copyright (C) 2009-2012 NICTA (www.nicta.com.au)
Chris@49: // Copyright (C) 2009-2012 Conrad Sanderson
Chris@49: // 
Chris@49: // This Source Code Form is subject to the terms of the Mozilla Public
Chris@49: // License, v. 2.0. If a copy of the MPL was not distributed with this
Chris@49: // file, You can obtain one at http://mozilla.org/MPL/2.0/.
Chris@49: 
Chris@49: 
Chris@49: //! \addtogroup op_median
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: template<typename T>
Chris@49: struct arma_cx_median_packet
Chris@49:   {
Chris@49:   T     val;
Chris@49:   uword index;
Chris@49:   };
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T>
Chris@49: arma_inline
Chris@49: bool
Chris@49: operator< (const arma_cx_median_packet<T>& A, const arma_cx_median_packet<T>& B)
Chris@49:   {
Chris@49:   return A.val < B.val;
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! Class for finding median values of a matrix
Chris@49: class op_median
Chris@49:   {
Chris@49:   public:
Chris@49:   
Chris@49:   template<typename T1>
Chris@49:   inline static void apply(Mat<typename T1::elem_type>& out, const Op<T1,op_median>& in);
Chris@49:   
Chris@49:   template<typename T, typename T1>
Chris@49:   inline static void apply(Mat< std::complex<T> >& out, const Op<T1,op_median>& in);
Chris@49:   
Chris@49:   //
Chris@49:   //
Chris@49:   
Chris@49:   template<typename T1>
Chris@49:   inline static typename T1::elem_type median_vec(const T1& X, const typename arma_not_cx<typename T1::elem_type>::result* junk = 0);
Chris@49:   
Chris@49:   template<typename T1>
Chris@49:   inline static typename T1::elem_type median_vec(const T1& X, const typename arma_cx_only<typename T1::elem_type>::result* junk = 0);
Chris@49:   
Chris@49:   //
Chris@49:   //
Chris@49:   
Chris@49:   template<typename eT>
Chris@49:   inline static eT direct_median(std::vector<eT>& X);
Chris@49:   
Chris@49:   template<typename T>
Chris@49:   inline static void direct_cx_median_index(uword& out_index1, uword& out_index2, std::vector< arma_cx_median_packet<T> >& X);
Chris@49:   };
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}