operator_cube_schur.hpp

Go to the documentation of this file.
00001 // Copyright (C) 2010 NICTA and the authors listed below
00002 // http://nicta.com.au
00003 // 
00004 // Authors:
00005 // - Conrad Sanderson (conradsand at ieee dot org)
00006 // 
00007 // This file is part of the Armadillo C++ library.
00008 // It is provided without any warranty of fitness
00009 // for any purpose. You can redistribute this file
00010 // and/or modify it under the terms of the GNU
00011 // Lesser General Public License (LGPL) as published
00012 // by the Free Software Foundation, either version 3
00013 // of the License or (at your option) any later version.
00014 // (see http://www.opensource.org/licenses for more info)
00015 
00016 
00017 //! \addtogroup operator_cube_schur
00018 //! @{
00019 
00020 
00021 // operator %, which we define it to do a schur product (element-wise multiplication)
00022 
00023 
00024 //! element-wise multiplication of BaseCube objects with same element type
00025 template<typename T1, typename T2>
00026 arma_inline
00027 const eGlueCube<T1, T2, eglue_cube_schur>
00028 operator%
00029   (
00030   const BaseCube<typename T1::elem_type,T1>& X,
00031   const BaseCube<typename T1::elem_type,T2>& Y
00032   )
00033   {
00034   arma_extra_debug_sigprint();
00035   
00036   return eGlueCube<T1, T2, eglue_cube_schur>(X.get_ref(), Y.get_ref());
00037   }
00038 
00039 
00040 
00041 //! element-wise multiplication of BaseCube objects with different element types
00042 template<typename T1, typename T2>
00043 arma_inline
00044 Cube<typename promote_type<typename T1::elem_type, typename T2::elem_type>::result>
00045 operator%
00046   (
00047   const BaseCube< typename force_different_type<typename T1::elem_type, typename T2::elem_type>::T1_result, T1>& X,
00048   const BaseCube< typename force_different_type<typename T1::elem_type, typename T2::elem_type>::T2_result, T2>& Y
00049   )
00050   {
00051   arma_extra_debug_sigprint();
00052   
00053   typedef typename T1::elem_type eT1;
00054   typedef typename T2::elem_type eT2;
00055   
00056   typedef typename promote_type<eT1,eT2>::result out_eT;
00057   
00058   promote_type<eT1,eT2>::check();
00059   
00060   const ProxyCube<T1> A(X.get_ref());
00061   const ProxyCube<T2> B(Y.get_ref());
00062   
00063   arma_debug_assert_same_size(A, B, "element-wise cube multiplication");
00064   
00065   Cube<out_eT> out(A.n_rows, A.n_cols, A.n_slices);
00066 
00067         out_eT* out_mem = out.memptr();
00068   const u32     n_elem  = out.n_elem;
00069   
00070   for(u32 i=0; i<n_elem; ++i)
00071     {
00072     out_mem[i] = upgrade_val<eT1,eT2>::apply(A[i]) * upgrade_val<eT1,eT2>::apply(B[i]);
00073     }
00074   
00075   return out;
00076   }
00077 
00078 
00079 
00080 //! @}