Chris@49: // Copyright (C) 2008-2013 NICTA (www.nicta.com.au)
Chris@49: // Copyright (C) 2008-2013 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_cx_scalar
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_times::apply
Chris@49:   (
Chris@49:         Mat< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_times>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const Proxy<T1> A(X.m);
Chris@49:   
Chris@49:   const uword n_rows = A.get_n_rows();
Chris@49:   const uword n_cols = A.get_n_cols();
Chris@49:   
Chris@49:   out.set_size(n_rows, n_cols);
Chris@49:   
Chris@49:   const eT  k       = X.aux_out_eT;
Chris@49:         eT* out_mem = out.memptr();
Chris@49:   
Chris@49:   if(Proxy<T1>::prefer_at_accessor == false)
Chris@49:     {
Chris@49:     const uword n_elem = A.get_n_elem();
Chris@49:   
Chris@49:     for(uword i=0; i<n_elem; ++i)
Chris@49:       {
Chris@49:       out_mem[i] = A[i] * k;
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     for(uword col=0; col < n_cols; ++col)
Chris@49:     for(uword row=0; row < n_rows; ++row)
Chris@49:       {
Chris@49:       *out_mem = A.at(row,col) * k;  ++out_mem;
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_plus::apply
Chris@49:   (
Chris@49:         Mat< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_plus>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const Proxy<T1> A(X.m);
Chris@49:   
Chris@49:   const uword n_rows = A.get_n_rows();
Chris@49:   const uword n_cols = A.get_n_cols();
Chris@49:   
Chris@49:   out.set_size(n_rows, n_cols);
Chris@49:   
Chris@49:   const eT  k       = X.aux_out_eT;
Chris@49:         eT* out_mem = out.memptr();
Chris@49:   
Chris@49:   if(Proxy<T1>::prefer_at_accessor == false)
Chris@49:     {
Chris@49:     const uword n_elem = A.get_n_elem();
Chris@49:   
Chris@49:     for(uword i=0; i<n_elem; ++i)
Chris@49:       {
Chris@49:       out_mem[i] = A[i] + k;
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     for(uword col=0; col < n_cols; ++col)
Chris@49:     for(uword row=0; row < n_rows; ++row)
Chris@49:       {
Chris@49:       *out_mem = A.at(row,col) + k;  ++out_mem;
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_minus_pre::apply
Chris@49:   (
Chris@49:         Mat< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_minus_pre>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const Proxy<T1> A(X.m);
Chris@49:   
Chris@49:   const uword n_rows = A.get_n_rows();
Chris@49:   const uword n_cols = A.get_n_cols();
Chris@49:   
Chris@49:   out.set_size(n_rows, n_cols);
Chris@49:   
Chris@49:   const eT  k       = X.aux_out_eT;
Chris@49:         eT* out_mem = out.memptr();
Chris@49:   
Chris@49:   if(Proxy<T1>::prefer_at_accessor == false)
Chris@49:     {
Chris@49:     const uword n_elem = A.get_n_elem();
Chris@49:   
Chris@49:     for(uword i=0; i<n_elem; ++i)
Chris@49:       {
Chris@49:       out_mem[i] = k - A[i];
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     for(uword col=0; col < n_cols; ++col)
Chris@49:     for(uword row=0; row < n_rows; ++row)
Chris@49:       {
Chris@49:       *out_mem = k - A.at(row,col);  ++out_mem;
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_minus_post::apply
Chris@49:   (
Chris@49:         Mat< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_minus_post>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const Proxy<T1> A(X.m);
Chris@49:   
Chris@49:   const uword n_rows = A.get_n_rows();
Chris@49:   const uword n_cols = A.get_n_cols();
Chris@49:   
Chris@49:   out.set_size(n_rows, n_cols);
Chris@49:   
Chris@49:   const eT  k       = X.aux_out_eT;
Chris@49:         eT* out_mem = out.memptr();
Chris@49:   
Chris@49:   if(Proxy<T1>::prefer_at_accessor == false)
Chris@49:     {
Chris@49:     const uword n_elem = A.get_n_elem();
Chris@49:   
Chris@49:     for(uword i=0; i<n_elem; ++i)
Chris@49:       {
Chris@49:       out_mem[i] = A[i] - k;
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     for(uword col=0; col < n_cols; ++col)
Chris@49:     for(uword row=0; row < n_rows; ++row)
Chris@49:       {
Chris@49:       *out_mem = A.at(row,col) - k;  ++out_mem;
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_div_pre::apply
Chris@49:   (
Chris@49:         Mat< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_div_pre>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const Proxy<T1> A(X.m);
Chris@49:   
Chris@49:   const uword n_rows = A.get_n_rows();
Chris@49:   const uword n_cols = A.get_n_cols();
Chris@49:   
Chris@49:   out.set_size(n_rows, n_cols);
Chris@49:   
Chris@49:   const eT  k       = X.aux_out_eT;
Chris@49:         eT* out_mem = out.memptr();
Chris@49:   
Chris@49:   if(Proxy<T1>::prefer_at_accessor == false)
Chris@49:     {
Chris@49:     const uword n_elem = A.get_n_elem();
Chris@49:   
Chris@49:     for(uword i=0; i<n_elem; ++i)
Chris@49:       {
Chris@49:       out_mem[i] = k / A[i];
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     for(uword col=0; col < n_cols; ++col)
Chris@49:     for(uword row=0; row < n_rows; ++row)
Chris@49:       {
Chris@49:       *out_mem = k / A.at(row,col);  ++out_mem;
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_div_post::apply
Chris@49:   (
Chris@49:         Mat< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOp<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_div_post>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const Proxy<T1> A(X.m);
Chris@49:   
Chris@49:   const uword n_rows = A.get_n_rows();
Chris@49:   const uword n_cols = A.get_n_cols();
Chris@49:   
Chris@49:   out.set_size(n_rows, n_cols);
Chris@49:   
Chris@49:   const eT  k       = X.aux_out_eT;
Chris@49:         eT* out_mem = out.memptr();
Chris@49:   
Chris@49:   if(Proxy<T1>::prefer_at_accessor == false)
Chris@49:     {
Chris@49:     const uword n_elem = A.get_n_elem();
Chris@49:   
Chris@49:     for(uword i=0; i<n_elem; ++i)
Chris@49:       {
Chris@49:       out_mem[i] = A[i] / k;
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     for(uword col=0; col < n_cols; ++col)
Chris@49:     for(uword row=0; row < n_rows; ++row)
Chris@49:       {
Chris@49:       *out_mem = A.at(row,col) / k;  ++out_mem;
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //
Chris@49: //
Chris@49: //
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_times::apply
Chris@49:   (
Chris@49:            Cube< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOpCube<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_times>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const ProxyCube<T1> A(X.m);
Chris@49:   
Chris@49:   out.set_size(A.get_n_rows(), A.get_n_cols(), A.get_n_slices());
Chris@49:   
Chris@49:   const eT    k       = X.aux_out_eT;
Chris@49:   const uword n_elem  = out.n_elem;
Chris@49:         eT*   out_mem = out.memptr();
Chris@49:   
Chris@49:   // TODO: implement handling for ProxyCube<T1>::prefer_at_accessor == true
Chris@49:   for(uword i=0; i<n_elem; ++i)
Chris@49:     {
Chris@49:     out_mem[i] = A[i] * k;
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_plus::apply
Chris@49:   (
Chris@49:            Cube< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOpCube<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_plus>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const ProxyCube<T1> A(X.m);
Chris@49:   
Chris@49:   out.set_size(A.get_n_rows(), A.get_n_cols(), A.get_n_slices());
Chris@49:   
Chris@49:   const eT    k       = X.aux_out_eT;
Chris@49:   const uword n_elem  = out.n_elem;
Chris@49:         eT*   out_mem = out.memptr();
Chris@49:   
Chris@49:   for(uword i=0; i<n_elem; ++i)
Chris@49:     {
Chris@49:     out_mem[i] = A[i] + k;
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_minus_pre::apply
Chris@49:   (
Chris@49:            Cube< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOpCube<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_minus_pre>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const ProxyCube<T1> A(X.m);
Chris@49:   
Chris@49:   out.set_size(A.get_n_rows(), A.get_n_cols(), A.get_n_slices());
Chris@49:   
Chris@49:   const eT    k       = X.aux_out_eT;
Chris@49:   const uword n_elem  = out.n_elem;
Chris@49:         eT*   out_mem = out.memptr();
Chris@49:   
Chris@49:   for(uword i=0; i<n_elem; ++i)
Chris@49:     {
Chris@49:     out_mem[i] = k - A[i];
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_minus_post::apply
Chris@49:   (
Chris@49:            Cube< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOpCube<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_minus_post>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const ProxyCube<T1> A(X.m);
Chris@49:   
Chris@49:   out.set_size(A.get_n_rows(), A.get_n_cols(), A.get_n_slices());
Chris@49:   
Chris@49:   const eT    k       = X.aux_out_eT;
Chris@49:   const uword n_elem  = out.n_elem;
Chris@49:         eT*   out_mem = out.memptr();
Chris@49:   
Chris@49:   for(uword i=0; i<n_elem; ++i)
Chris@49:     {
Chris@49:     out_mem[i] = A[i] - k;
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_div_pre::apply
Chris@49:   (
Chris@49:            Cube< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOpCube<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_div_pre>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const ProxyCube<T1> A(X.m);
Chris@49:   
Chris@49:   out.set_size(A.get_n_rows(), A.get_n_cols(), A.get_n_slices());
Chris@49:   
Chris@49:   const eT    k       = X.aux_out_eT;
Chris@49:   const uword n_elem  = out.n_elem;
Chris@49:         eT*   out_mem = out.memptr();
Chris@49:   
Chris@49:   for(uword i=0; i<n_elem; ++i)
Chris@49:     {
Chris@49:     out_mem[i] = k / A[i];
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: op_cx_scalar_div_post::apply
Chris@49:   (
Chris@49:            Cube< typename std::complex<typename T1::pod_type> >& out,
Chris@49:   const mtOpCube<typename std::complex<typename T1::pod_type>, T1, op_cx_scalar_div_post>& X
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename std::complex<typename T1::pod_type> eT;
Chris@49:   
Chris@49:   const ProxyCube<T1> A(X.m);
Chris@49:   
Chris@49:   out.set_size(A.get_n_rows(), A.get_n_cols(), A.get_n_slices());
Chris@49:   
Chris@49:   const eT    k       = X.aux_out_eT;
Chris@49:   const uword n_elem  = out.n_elem;
Chris@49:         eT*   out_mem = out.memptr();
Chris@49:   
Chris@49:   for(uword i=0; i<n_elem; ++i)
Chris@49:     {
Chris@49:     out_mem[i] = A[i] / k;
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}