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 subview_cube
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: subview_cube<eT>::~subview_cube()
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: arma_inline
Chris@49: subview_cube<eT>::subview_cube
Chris@49:   (
Chris@49:   const Cube<eT>& in_m,
Chris@49:   const uword       in_row1,
Chris@49:   const uword       in_col1,
Chris@49:   const uword       in_slice1,
Chris@49:   const uword       in_n_rows,
Chris@49:   const uword       in_n_cols,
Chris@49:   const uword       in_n_slices
Chris@49:   )
Chris@49:   : m           (in_m)
Chris@49:   , aux_row1    (in_row1)
Chris@49:   , aux_col1    (in_col1)
Chris@49:   , aux_slice1  (in_slice1)
Chris@49:   , n_rows      (in_n_rows)
Chris@49:   , n_cols      (in_n_cols)
Chris@49:   , n_elem_slice(in_n_rows * in_n_cols)
Chris@49:   , n_slices    (in_n_slices)
Chris@49:   , n_elem      (n_elem_slice * in_n_slices)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator+= (const eT val)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const uword local_n_rows   = n_rows;
Chris@49:   const uword local_n_cols   = n_cols;
Chris@49:   const uword local_n_slices = n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < local_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < local_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_plus( slice_colptr(slice,col), val, local_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator-= (const eT val)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const uword local_n_rows   = n_rows;
Chris@49:   const uword local_n_cols   = n_cols;
Chris@49:   const uword local_n_slices = n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < local_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < local_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_minus( slice_colptr(slice,col), val, local_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator*= (const eT val)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const uword local_n_rows   = n_rows;
Chris@49:   const uword local_n_cols   = n_cols;
Chris@49:   const uword local_n_slices = n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < local_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < local_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_mul( slice_colptr(slice,col), val, local_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator/= (const eT val)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const uword local_n_rows   = n_rows;
Chris@49:   const uword local_n_cols   = n_cols;
Chris@49:   const uword local_n_slices = n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < local_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < local_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_div( slice_colptr(slice,col), val, local_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator= (const BaseCube<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap_cube<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Cube<eT>&         x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "copy into subcube");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::copy( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator+= (const BaseCube<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap_cube<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Cube<eT>&         x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "addition");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_plus( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator-= (const BaseCube<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap_cube<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Cube<eT>&         x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "subtraction");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_minus( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator%= (const BaseCube<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap_cube<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Cube<eT>&         x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "element-wise multiplication");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_mul( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator/= (const BaseCube<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap_cube<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Cube<eT>&         x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "element-wise division");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_div( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! x.subcube(...) = y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator= (const subview_cube<eT>& x_in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const bool overlap = check_overlap(x_in);
Chris@49:   
Chris@49:         Cube<eT>*         tmp_cube         = overlap ? new Cube<eT>(x_in.m) : 0;
Chris@49:   const subview_cube<eT>* tmp_subview_cube = overlap ? new subview_cube<eT>(*tmp_cube, x_in.aux_row1, x_in.aux_col1, x_in.aux_slice1, x_in.n_rows, x_in.n_cols, x_in.n_slices) : 0;
Chris@49:   const subview_cube<eT>& x                = overlap ? (*tmp_subview_cube) : x_in;
Chris@49:   
Chris@49:   subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "copy into subcube");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::copy( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:     
Chris@49:   if(overlap)
Chris@49:     {
Chris@49:     delete tmp_subview_cube;
Chris@49:     delete tmp_cube;
Chris@49:     }
Chris@49:   
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator+= (const subview_cube<eT>& x_in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const bool overlap = check_overlap(x_in);
Chris@49:   
Chris@49:         Cube<eT>*         tmp_cube         = overlap ? new Cube<eT>(x_in.m) : 0;
Chris@49:   const subview_cube<eT>* tmp_subview_cube = overlap ? new subview_cube<eT>(*tmp_cube, x_in.aux_row1, x_in.aux_col1, x_in.aux_slice1, x_in.n_rows, x_in.n_cols, x_in.n_slices) : 0;
Chris@49:   const subview_cube<eT>& x                = overlap ? (*tmp_subview_cube) : x_in;
Chris@49:   
Chris@49:   subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "addition");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_plus( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   
Chris@49:   if(overlap)
Chris@49:     {
Chris@49:     delete tmp_subview_cube;
Chris@49:     delete tmp_cube;
Chris@49:     }
Chris@49:   
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator-= (const subview_cube<eT>& x_in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const bool overlap = check_overlap(x_in);
Chris@49:   
Chris@49:         Cube<eT>*         tmp_cube         = overlap ? new Cube<eT>(x_in.m) : 0;
Chris@49:   const subview_cube<eT>* tmp_subview_cube = overlap ? new subview_cube<eT>(*tmp_cube, x_in.aux_row1, x_in.aux_col1, x_in.aux_slice1, x_in.n_rows, x_in.n_cols, x_in.n_slices) : 0;
Chris@49:   const subview_cube<eT>& x                = overlap ? (*tmp_subview_cube) : x_in;
Chris@49:   
Chris@49:   subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "subtraction");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_minus( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   
Chris@49:   if(overlap)
Chris@49:     {
Chris@49:     delete tmp_subview_cube;
Chris@49:     delete tmp_cube;
Chris@49:     }
Chris@49:     
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator%= (const subview_cube<eT>& x_in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const bool overlap = check_overlap(x_in);
Chris@49:   
Chris@49:         Cube<eT>*         tmp_cube         = overlap ? new Cube<eT>(x_in.m) : 0;
Chris@49:   const subview_cube<eT>* tmp_subview_cube = overlap ? new subview_cube<eT>(*tmp_cube, x_in.aux_row1, x_in.aux_col1, x_in.aux_slice1, x_in.n_rows, x_in.n_cols, x_in.n_slices) : 0;
Chris@49:   const subview_cube<eT>& x                = overlap ? (*tmp_subview_cube) : x_in;
Chris@49:   
Chris@49:   subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "element-wise multiplication");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_mul( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   
Chris@49:   if(overlap)
Chris@49:     {
Chris@49:     delete tmp_subview_cube;
Chris@49:     delete tmp_cube;
Chris@49:     }
Chris@49:   
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator/= (const subview_cube<eT>& x_in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const bool overlap = check_overlap(x_in);
Chris@49:   
Chris@49:         Cube<eT>*         tmp_cube         = overlap ? new Cube<eT>(x_in.m) : 0;
Chris@49:   const subview_cube<eT>* tmp_subview_cube = overlap ? new subview_cube<eT>(*tmp_cube, x_in.aux_row1, x_in.aux_col1, x_in.aux_slice1, x_in.n_rows, x_in.n_cols, x_in.n_slices) : 0;
Chris@49:   const subview_cube<eT>& x                = overlap ? (*tmp_subview_cube) : x_in;
Chris@49:   
Chris@49:   subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   arma_debug_assert_same_size(t, x, "element-wise division");
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < t_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_div( t.slice_colptr(slice,col), x.slice_colptr(slice,col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   
Chris@49:   if(overlap)
Chris@49:     {
Chris@49:     delete tmp_subview_cube;
Chris@49:     delete tmp_cube;
Chris@49:     }
Chris@49:   
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator= (const Base<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Mat<eT>&          x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   const uword x_n_rows   = x.n_rows;
Chris@49:   const uword x_n_cols   = x.n_cols;
Chris@49:   
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == x_n_cols) && (t_n_slices == 1) )
Chris@49:     {
Chris@49:     // interpret the matrix as a cube with one slice
Chris@49:     
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::copy( t.slice_colptr(0, col), x.colptr(col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == 1) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     for(uword i=0; i < t_n_slices; ++i)
Chris@49:       {
Chris@49:       arrayops::copy( t.slice_colptr(i, 0), x.colptr(i), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == 1) && (t_n_cols == x_n_rows) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     Cube<eT>& Q = const_cast< Cube<eT>& >(t.m);
Chris@49:     
Chris@49:     const uword t_aux_row1   = t.aux_row1;
Chris@49:     const uword t_aux_col1   = t.aux_col1;
Chris@49:     const uword t_aux_slice1 = t.aux_slice1;
Chris@49:     
Chris@49:     for(uword slice=0; slice < t_n_slices; ++slice)
Chris@49:       {
Chris@49:       const uword mod_slice = t_aux_slice1 + slice;
Chris@49:       
Chris@49:       const eT* x_colptr = x.colptr(slice);
Chris@49:       
Chris@49:       uword i,j;
Chris@49:       for(i=0, j=1; j < t_n_cols; i+=2, j+=2)
Chris@49:         {
Chris@49:         const eT tmp_i = x_colptr[i];
Chris@49:         const eT tmp_j = x_colptr[j];
Chris@49:         
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) = tmp_i;
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + j, mod_slice) = tmp_j;
Chris@49:         }
Chris@49:       
Chris@49:       if(i < t_n_cols)
Chris@49:         {
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) = x_colptr[i];
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(arma_config::debug == true)
Chris@49:       {
Chris@49:       arma_stop( arma_incompat_size_string(t, x, "copy into subcube") );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator+= (const Base<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Mat<eT>&          x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   const uword x_n_rows   = x.n_rows;
Chris@49:   const uword x_n_cols   = x.n_cols;
Chris@49:   
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == x_n_cols) && (t_n_slices == 1) )
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_plus( t.slice_colptr(0, col), x.colptr(col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == 1) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     for(uword i=0; i < t_n_slices; ++i)
Chris@49:       {
Chris@49:       arrayops::inplace_plus( t.slice_colptr(i, 0), x.colptr(i), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == 1) && (t_n_cols == x_n_rows) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     Cube<eT>& Q = const_cast< Cube<eT>& >(t.m);
Chris@49:     
Chris@49:     const uword t_aux_row1   = t.aux_row1;
Chris@49:     const uword t_aux_col1   = t.aux_col1;
Chris@49:     const uword t_aux_slice1 = t.aux_slice1;
Chris@49:     
Chris@49:     for(uword slice=0; slice < t_n_slices; ++slice)
Chris@49:       {
Chris@49:       const uword mod_slice = t_aux_slice1 + slice;
Chris@49:       
Chris@49:       const eT* x_colptr = x.colptr(slice);
Chris@49:       
Chris@49:       uword i,j;
Chris@49:       for(i=0, j=1; j < t_n_cols; i+=2, j+=2)
Chris@49:         {
Chris@49:         const eT tmp_i = x_colptr[i];
Chris@49:         const eT tmp_j = x_colptr[j];
Chris@49:         
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) += tmp_i;
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + j, mod_slice) += tmp_j;
Chris@49:         }
Chris@49:       
Chris@49:       if(i < t_n_cols)
Chris@49:         {
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) += x_colptr[i];
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(arma_config::debug == true)
Chris@49:       {
Chris@49:       arma_stop( arma_incompat_size_string(t, x, "addition") );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator-= (const Base<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Mat<eT>&          x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   const uword x_n_rows   = x.n_rows;
Chris@49:   const uword x_n_cols   = x.n_cols;
Chris@49:   
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == x_n_cols) && (t_n_slices == 1) )
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_minus( t.slice_colptr(0, col), x.colptr(col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == 1) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     for(uword i=0; i < t_n_slices; ++i)
Chris@49:       {
Chris@49:       arrayops::inplace_minus( t.slice_colptr(i, 0), x.colptr(i), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == 1) && (t_n_cols == x_n_rows) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     Cube<eT>& Q = const_cast< Cube<eT>& >(t.m);
Chris@49:     
Chris@49:     const uword t_aux_row1   = t.aux_row1;
Chris@49:     const uword t_aux_col1   = t.aux_col1;
Chris@49:     const uword t_aux_slice1 = t.aux_slice1;
Chris@49:     
Chris@49:     for(uword slice=0; slice < t_n_slices; ++slice)
Chris@49:       {
Chris@49:       const uword mod_slice = t_aux_slice1 + slice;
Chris@49:       
Chris@49:       const eT* x_colptr = x.colptr(slice);
Chris@49:       
Chris@49:       uword i,j;
Chris@49:       for(i=0, j=1; j < t_n_cols; i+=2, j+=2)
Chris@49:         {
Chris@49:         const eT tmp_i = x_colptr[i];
Chris@49:         const eT tmp_j = x_colptr[j];
Chris@49:         
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) -= tmp_i;
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + j, mod_slice) -= tmp_j;
Chris@49:         }
Chris@49:       
Chris@49:       if(i < t_n_cols)
Chris@49:         {
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) -= x_colptr[i];
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(arma_config::debug == true)
Chris@49:       {
Chris@49:       arma_stop( arma_incompat_size_string(t, x, "subtraction") );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator%= (const Base<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Mat<eT>&          x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   const uword x_n_rows   = x.n_rows;
Chris@49:   const uword x_n_cols   = x.n_cols;
Chris@49:   
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == x_n_cols) && (t_n_slices == 1) )
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_mul( t.slice_colptr(0, col), x.colptr(col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == 1) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     for(uword i=0; i < t_n_slices; ++i)
Chris@49:       {
Chris@49:       arrayops::inplace_mul( t.slice_colptr(i, 0), x.colptr(i), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == 1) && (t_n_cols == x_n_rows) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     Cube<eT>& Q = const_cast< Cube<eT>& >(t.m);
Chris@49:     
Chris@49:     const uword t_aux_row1   = t.aux_row1;
Chris@49:     const uword t_aux_col1   = t.aux_col1;
Chris@49:     const uword t_aux_slice1 = t.aux_slice1;
Chris@49:     
Chris@49:     for(uword slice=0; slice < t_n_slices; ++slice)
Chris@49:       {
Chris@49:       const uword mod_slice = t_aux_slice1 + slice;
Chris@49:       
Chris@49:       const eT* x_colptr = x.colptr(slice);
Chris@49:       
Chris@49:       uword i,j;
Chris@49:       for(i=0, j=1; j < t_n_cols; i+=2, j+=2)
Chris@49:         {
Chris@49:         const eT tmp_i = x_colptr[i];
Chris@49:         const eT tmp_j = x_colptr[j];
Chris@49:         
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) *= tmp_i;
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + j, mod_slice) *= tmp_j;
Chris@49:         }
Chris@49:       
Chris@49:       if(i < t_n_cols)
Chris@49:         {
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) *= x_colptr[i];
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(arma_config::debug == true)
Chris@49:       {
Chris@49:       arma_stop( arma_incompat_size_string(t, x, "element-wise multiplication") );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::operator/= (const Base<eT,T1>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const unwrap<T1> tmp(in.get_ref());
Chris@49:   
Chris@49:   const Mat<eT>&          x = tmp.M;
Chris@49:         subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   const uword t_n_rows   = t.n_rows;
Chris@49:   const uword t_n_cols   = t.n_cols;
Chris@49:   const uword t_n_slices = t.n_slices;
Chris@49:   
Chris@49:   const uword x_n_rows   = x.n_rows;
Chris@49:   const uword x_n_cols   = x.n_cols;
Chris@49:   
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == x_n_cols) && (t_n_slices == 1) )
Chris@49:     {
Chris@49:     for(uword col = 0; col < t_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_div( t.slice_colptr(0, col), x.colptr(col), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == x_n_rows) && (t_n_cols == 1) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     for(uword i=0; i < t_n_slices; ++i)
Chris@49:       {
Chris@49:       arrayops::inplace_div( t.slice_colptr(i, 0), x.colptr(i), t_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:   if( (t_n_rows == 1) && (t_n_cols == x_n_rows) && (t_n_slices == x_n_cols) )
Chris@49:     {
Chris@49:     Cube<eT>& Q = const_cast< Cube<eT>& >(t.m);
Chris@49:     
Chris@49:     const uword t_aux_row1   = t.aux_row1;
Chris@49:     const uword t_aux_col1   = t.aux_col1;
Chris@49:     const uword t_aux_slice1 = t.aux_slice1;
Chris@49:     
Chris@49:     for(uword slice=0; slice < t_n_slices; ++slice)
Chris@49:       {
Chris@49:       const uword mod_slice = t_aux_slice1 + slice;
Chris@49:       
Chris@49:       const eT* x_colptr = x.colptr(slice);
Chris@49:       
Chris@49:       uword i,j;
Chris@49:       for(i=0, j=1; j < t_n_cols; i+=2, j+=2)
Chris@49:         {
Chris@49:         const eT tmp_i = x_colptr[i];
Chris@49:         const eT tmp_j = x_colptr[j];
Chris@49:         
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) /= tmp_i;
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + j, mod_slice) /= tmp_j;
Chris@49:         }
Chris@49:       
Chris@49:       if(i < t_n_cols)
Chris@49:         {
Chris@49:         Q.at(t_aux_row1, t_aux_col1 + i, mod_slice) /= x_colptr[i];
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(arma_config::debug == true)
Chris@49:       {
Chris@49:       arma_stop( arma_incompat_size_string(t, x, "element-wise division") );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! transform each element in the subview using a functor
Chris@49: template<typename eT>
Chris@49: template<typename functor>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::transform(functor F)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   Cube<eT>& Q = const_cast< Cube<eT>& >(m);
Chris@49:   
Chris@49:   const uword start_col   = aux_col1;
Chris@49:   const uword start_row   = aux_row1;
Chris@49:   const uword start_slice = aux_slice1;
Chris@49:   
Chris@49:   const uword end_col_plus1   = start_col   + n_cols;
Chris@49:   const uword end_row_plus1   = start_row   + n_rows;
Chris@49:   const uword end_slice_plus1 = start_slice + n_slices;
Chris@49:   
Chris@49:   for(uword uslice = start_slice; uslice < end_slice_plus1; ++uslice)
Chris@49:   for(uword ucol   = start_col;     ucol < end_col_plus1;   ++ucol  )
Chris@49:   for(uword urow   = start_row;     urow < end_row_plus1;   ++urow  )
Chris@49:     {
Chris@49:     Q.at(urow, ucol, uslice) = eT( F( Q.at(urow, ucol, uslice) ) );
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! imbue (fill) the subview with values provided by a functor
Chris@49: template<typename eT>
Chris@49: template<typename functor>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::imbue(functor F)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   Cube<eT>& Q = const_cast< Cube<eT>& >(m);
Chris@49:   
Chris@49:   const uword start_col   = aux_col1;
Chris@49:   const uword start_row   = aux_row1;
Chris@49:   const uword start_slice = aux_slice1;
Chris@49:   
Chris@49:   const uword end_col_plus1   = start_col   + n_cols;
Chris@49:   const uword end_row_plus1   = start_row   + n_rows;
Chris@49:   const uword end_slice_plus1 = start_slice + n_slices;
Chris@49:   
Chris@49:   for(uword uslice = start_slice; uslice < end_slice_plus1; ++uslice)
Chris@49:   for(uword ucol   = start_col;     ucol < end_col_plus1;   ++ucol  )
Chris@49:   for(uword urow   = start_row;     urow < end_row_plus1;   ++urow  )
Chris@49:     {
Chris@49:     Q.at(urow, ucol, uslice) = eT( F() );
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::fill(const eT val)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49: 
Chris@49:   const uword local_n_rows   = n_rows;
Chris@49:   const uword local_n_cols   = n_cols;
Chris@49:   const uword local_n_slices = n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice < local_n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < local_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_set( slice_colptr(slice,col), val, local_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::zeros()
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   fill(eT(0));
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::ones()
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   fill(eT(1));
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT
Chris@49: subview_cube<eT>::at_alt(const uword i) const
Chris@49:   {
Chris@49:   return operator[](i);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT&
Chris@49: subview_cube<eT>::operator[](const uword i)
Chris@49:   {
Chris@49:   const uword in_slice = i / n_elem_slice;
Chris@49:   const uword offset   = in_slice * n_elem_slice;
Chris@49:   const uword j        = i - offset;
Chris@49:   
Chris@49:   const uword in_col   = j / n_rows;
Chris@49:   const uword in_row   = j % n_rows;
Chris@49: 
Chris@49:   const uword index = (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
Chris@49:   
Chris@49:   return access::rw( (const_cast< Cube<eT>& >(m)).mem[index] );
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT
Chris@49: subview_cube<eT>::operator[](const uword i) const
Chris@49:   {
Chris@49:   const uword in_slice = i / n_elem_slice;
Chris@49:   const uword offset   = in_slice * n_elem_slice;
Chris@49:   const uword j        = i - offset;
Chris@49:   
Chris@49:   const uword in_col   = j / n_rows;
Chris@49:   const uword in_row   = j % n_rows;
Chris@49: 
Chris@49:   const uword index = (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
Chris@49:   
Chris@49:   return m.mem[index];
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT&
Chris@49: subview_cube<eT>::operator()(const uword i)
Chris@49:   {
Chris@49:   arma_debug_check( (i >= n_elem), "subview_cube::operator(): index out of bounds");
Chris@49:   
Chris@49:   const uword in_slice = i / n_elem_slice;
Chris@49:   const uword offset   = in_slice * n_elem_slice;
Chris@49:   const uword j        = i - offset;
Chris@49:   
Chris@49:   const uword in_col   = j / n_rows;
Chris@49:   const uword in_row   = j % n_rows;
Chris@49: 
Chris@49:   const uword index = (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
Chris@49:   
Chris@49:   return access::rw( (const_cast< Cube<eT>& >(m)).mem[index] );
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: eT
Chris@49: subview_cube<eT>::operator()(const uword i) const
Chris@49:   {
Chris@49:   arma_debug_check( (i >= n_elem), "subview_cube::operator(): index out of bounds");
Chris@49:   
Chris@49:   const uword in_slice = i / n_elem_slice;
Chris@49:   const uword offset   = in_slice * n_elem_slice;
Chris@49:   const uword j        = i - offset;
Chris@49:   
Chris@49:   const uword in_col   = j / n_rows;
Chris@49:   const uword in_row   = j % n_rows;
Chris@49: 
Chris@49:   const uword index = (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
Chris@49:   
Chris@49:   return m.mem[index];
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: arma_inline
Chris@49: eT&
Chris@49: subview_cube<eT>::operator()(const uword in_row, const uword in_col, const uword in_slice)
Chris@49:   {
Chris@49:   arma_debug_check( ( (in_row >= n_rows) || (in_col >= n_cols) || (in_slice >= n_slices) ), "subview_cube::operator(): location out of bounds");
Chris@49:   
Chris@49:   const uword index = (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
Chris@49:   
Chris@49:   return access::rw( (const_cast< Cube<eT>& >(m)).mem[index] );
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: arma_inline
Chris@49: eT
Chris@49: subview_cube<eT>::operator()(const uword in_row, const uword in_col, const uword in_slice) const
Chris@49:   {
Chris@49:   arma_debug_check( ( (in_row >= n_rows) || (in_col >= n_cols) || (in_slice >= n_slices) ), "subview_cube::operator(): location out of bounds");
Chris@49:   
Chris@49:   const uword index = (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
Chris@49:   
Chris@49:   return m.mem[index];
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: arma_inline
Chris@49: eT&
Chris@49: subview_cube<eT>::at(const uword in_row, const uword in_col, const uword in_slice)
Chris@49:   {
Chris@49:   const uword index = (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
Chris@49:   
Chris@49:   return access::rw( (const_cast< Cube<eT>& >(m)).mem[index] );
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: arma_inline
Chris@49: eT
Chris@49: subview_cube<eT>::at(const uword in_row, const uword in_col, const uword in_slice) const
Chris@49:   {
Chris@49:   const uword index = (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 + in_row;
Chris@49:   
Chris@49:   return m.mem[index];
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: arma_inline
Chris@49: eT*
Chris@49: subview_cube<eT>::slice_colptr(const uword in_slice, const uword in_col)
Chris@49:   {
Chris@49:   return & access::rw((const_cast< Cube<eT>& >(m)).mem[  (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1  ]);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: arma_inline
Chris@49: const eT*
Chris@49: subview_cube<eT>::slice_colptr(const uword in_slice, const uword in_col) const
Chris@49:   {
Chris@49:   return & m.mem[ (in_slice + aux_slice1)*m.n_elem_slice + (in_col + aux_col1)*m.n_rows + aux_row1 ];
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: bool
Chris@49: subview_cube<eT>::check_overlap(const subview_cube<eT>& x) const
Chris@49:   {
Chris@49:   const subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   if(&t.m != &x.m)
Chris@49:     {
Chris@49:     return false;
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if( (t.n_elem == 0) || (x.n_elem == 0) )
Chris@49:       {
Chris@49:       return false;
Chris@49:       }
Chris@49:     else
Chris@49:       {
Chris@49:       const uword t_row_start  = t.aux_row1;
Chris@49:       const uword t_row_end_p1 = t_row_start + t.n_rows;
Chris@49:       
Chris@49:       const uword t_col_start  = t.aux_col1;
Chris@49:       const uword t_col_end_p1 = t_col_start + t.n_cols;
Chris@49:       
Chris@49:       const uword t_slice_start  = t.aux_slice1;
Chris@49:       const uword t_slice_end_p1 = t_slice_start + t.n_slices;
Chris@49:       
Chris@49:       
Chris@49:       const uword x_row_start  = x.aux_row1;
Chris@49:       const uword x_row_end_p1 = x_row_start + x.n_rows;
Chris@49:       
Chris@49:       const uword x_col_start  = x.aux_col1;
Chris@49:       const uword x_col_end_p1 = x_col_start + x.n_cols;
Chris@49:       
Chris@49:       const uword x_slice_start  = x.aux_slice1;
Chris@49:       const uword x_slice_end_p1 = x_slice_start + x.n_slices;
Chris@49:       
Chris@49:       
Chris@49:       const bool outside_rows   = ( (x_row_start   >= t_row_end_p1  ) || (t_row_start   >= x_row_end_p1  ) );
Chris@49:       const bool outside_cols   = ( (x_col_start   >= t_col_end_p1  ) || (t_col_start   >= x_col_end_p1  ) );
Chris@49:       const bool outside_slices = ( (x_slice_start >= t_slice_end_p1) || (t_slice_start >= x_slice_end_p1) );
Chris@49:       
Chris@49:       return ( (outside_rows == false) && (outside_cols == false) && (outside_slices == false) );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: bool
Chris@49: subview_cube<eT>::check_overlap(const Mat<eT>& x) const
Chris@49:   {
Chris@49:   const subview_cube<eT>& t = *this;
Chris@49:   
Chris@49:   const uword t_aux_slice1        = t.aux_slice1;
Chris@49:   const uword t_aux_slice2_plus_1 = t_aux_slice1 + t.n_slices;
Chris@49:   
Chris@49:   for(uword slice = t_aux_slice1; slice < t_aux_slice2_plus_1; ++slice)
Chris@49:     {
Chris@49:     const Mat<eT>& y = *(t.m.mat_ptrs[slice]);
Chris@49:   
Chris@49:     if( x.memptr() == y.memptr() )
Chris@49:       {
Chris@49:       return true;
Chris@49:       }
Chris@49:     }
Chris@49:   
Chris@49:   return false;
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! cube X = Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::extract(Cube<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49: 
Chris@49:   // NOTE: we're assuming that the cube has already been set to the correct size and there is no aliasing;
Chris@49:   // size setting and alias checking is done by either the Cube contructor or operator=()
Chris@49:   
Chris@49:   const uword n_rows   = in.n_rows;
Chris@49:   const uword n_cols   = in.n_cols;
Chris@49:   const uword n_slices = in.n_slices;
Chris@49:   
Chris@49:   arma_extra_debug_print(arma_boost::format("out.n_rows = %d   out.n_cols = %d    out.n_slices = %d    in.m.n_rows = %d   in.m.n_cols = %d   in.m.n_slices = %d") % out.n_rows % out.n_cols % out.n_slices % in.m.n_rows % in.m.n_cols % in.m.n_slices);
Chris@49:   
Chris@49:   
Chris@49:   for(uword slice = 0; slice < n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col < n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::copy( out.slice_colptr(slice,col), in.slice_colptr(slice,col), n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! cube X += Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::plus_inplace(Cube<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_same_size(out, in, "addition");
Chris@49:   
Chris@49:   const uword n_rows   = out.n_rows;
Chris@49:   const uword n_cols   = out.n_cols;
Chris@49:   const uword n_slices = out.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice<n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col<n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_plus( out.slice_colptr(slice,col), in.slice_colptr(slice,col), n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! cube X -= Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::minus_inplace(Cube<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_same_size(out, in, "subtraction");
Chris@49:   
Chris@49:   const uword n_rows   = out.n_rows;
Chris@49:   const uword n_cols   = out.n_cols;
Chris@49:   const uword n_slices = out.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice<n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col<n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_minus( out.slice_colptr(slice,col), in.slice_colptr(slice,col), n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! cube X %= Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::schur_inplace(Cube<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_same_size(out, in, "element-wise multiplication");
Chris@49:   
Chris@49:   const uword n_rows   = out.n_rows;
Chris@49:   const uword n_cols   = out.n_cols;
Chris@49:   const uword n_slices = out.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice<n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col<n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_mul( out.slice_colptr(slice,col), in.slice_colptr(slice,col), n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! cube X /= Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::div_inplace(Cube<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_same_size(out, in, "element-wise division");
Chris@49:   
Chris@49:   const uword n_rows   = out.n_rows;
Chris@49:   const uword n_cols   = out.n_cols;
Chris@49:   const uword n_slices = out.n_slices;
Chris@49:   
Chris@49:   for(uword slice = 0; slice<n_slices; ++slice)
Chris@49:     {
Chris@49:     for(uword col = 0; col<n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_div( out.slice_colptr(slice,col), in.slice_colptr(slice,col), n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! mat X = Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::extract(Mat<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_cube_as_mat(out, in, "copy into matrix", false);
Chris@49:   
Chris@49:   const uword in_n_rows   = in.n_rows;
Chris@49:   const uword in_n_cols   = in.n_cols;
Chris@49:   const uword in_n_slices = in.n_slices;
Chris@49:   
Chris@49:   const uword out_vec_state = out.vec_state;
Chris@49:   
Chris@49:   if(in_n_slices == 1)
Chris@49:     {
Chris@49:     out.set_size(in_n_rows, in_n_cols);
Chris@49:     
Chris@49:     for(uword col=0; col < in_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::copy( out.colptr(col), in.slice_colptr(0, col), in_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(out_vec_state == 0)
Chris@49:       {
Chris@49:       if(in_n_cols == 1)
Chris@49:         {
Chris@49:         out.set_size(in_n_rows, in_n_slices);
Chris@49:         
Chris@49:         for(uword i=0; i < in_n_slices; ++i)
Chris@49:           {
Chris@49:           arrayops::copy( out.colptr(i), in.slice_colptr(i, 0), in_n_rows );
Chris@49:           }
Chris@49:         }
Chris@49:       else
Chris@49:       if(in_n_rows == 1)
Chris@49:         {
Chris@49:         const Cube<eT>& Q = in.m;
Chris@49:         
Chris@49:         const uword in_aux_row1   = in.aux_row1;
Chris@49:         const uword in_aux_col1   = in.aux_col1;
Chris@49:         const uword in_aux_slice1 = in.aux_slice1;
Chris@49:         
Chris@49:         out.set_size(in_n_cols, in_n_slices);
Chris@49:         
Chris@49:         for(uword slice=0; slice < in_n_slices; ++slice)
Chris@49:           {
Chris@49:           const uword mod_slice = in_aux_slice1 + slice;
Chris@49:           
Chris@49:           eT* out_colptr = out.colptr(slice);
Chris@49:           
Chris@49:           uword i,j;
Chris@49:           for(i=0, j=1; j < in_n_cols; i+=2, j+=2)
Chris@49:             {
Chris@49:             const eT tmp_i = Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             const eT tmp_j = Q.at(in_aux_row1, in_aux_col1 + j, mod_slice);
Chris@49:             
Chris@49:             out_colptr[i] = tmp_i;
Chris@49:             out_colptr[j] = tmp_j;
Chris@49:             }
Chris@49:           
Chris@49:           if(i < in_n_cols)
Chris@49:             {
Chris@49:             out_colptr[i] = Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             }
Chris@49:           }
Chris@49:         }
Chris@49:       }
Chris@49:     else
Chris@49:       {
Chris@49:       out.set_size(in_n_slices);
Chris@49:       
Chris@49:       eT* out_mem = out.memptr();
Chris@49:       
Chris@49:       const Cube<eT>& Q = in.m;
Chris@49:       
Chris@49:       const uword in_aux_row1   = in.aux_row1;
Chris@49:       const uword in_aux_col1   = in.aux_col1;
Chris@49:       const uword in_aux_slice1 = in.aux_slice1;
Chris@49:       
Chris@49:       for(uword i=0; i<in_n_slices; ++i)
Chris@49:         {
Chris@49:         out_mem[i] = Q.at(in_aux_row1, in_aux_col1, in_aux_slice1 + i);
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! mat X += Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::plus_inplace(Mat<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_cube_as_mat(out, in, "addition", true);
Chris@49:   
Chris@49:   const uword in_n_rows   = in.n_rows;
Chris@49:   const uword in_n_cols   = in.n_cols;
Chris@49:   const uword in_n_slices = in.n_slices;
Chris@49:   
Chris@49:   const uword out_n_rows    = out.n_rows;
Chris@49:   const uword out_n_cols    = out.n_cols;
Chris@49:   const uword out_vec_state = out.vec_state;
Chris@49:   
Chris@49:   if(in_n_slices == 1)
Chris@49:     {
Chris@49:     for(uword col=0; col < in_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_plus( out.colptr(col), in.slice_colptr(0, col), in_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(out_vec_state == 0)
Chris@49:       {
Chris@49:       if( (in_n_rows == out_n_rows) && (in_n_cols == 1) && (in_n_slices == out_n_cols) )
Chris@49:         {
Chris@49:         for(uword i=0; i < in_n_slices; ++i)
Chris@49:           {
Chris@49:           arrayops::inplace_plus( out.colptr(i), in.slice_colptr(i, 0), in_n_rows );
Chris@49:           }
Chris@49:         }
Chris@49:       else
Chris@49:       if( (in_n_rows == 1) && (in_n_cols == out_n_rows) && (in_n_slices == out_n_cols) )
Chris@49:         {
Chris@49:         const Cube<eT>& Q = in.m;
Chris@49:         
Chris@49:         const uword in_aux_row1   = in.aux_row1;
Chris@49:         const uword in_aux_col1   = in.aux_col1;
Chris@49:         const uword in_aux_slice1 = in.aux_slice1;
Chris@49:         
Chris@49:         for(uword slice=0; slice < in_n_slices; ++slice)
Chris@49:           {
Chris@49:           const uword mod_slice = in_aux_slice1 + slice;
Chris@49:           
Chris@49:           eT* out_colptr = out.colptr(slice);
Chris@49:           
Chris@49:           uword i,j;
Chris@49:           for(i=0, j=1; j < in_n_cols; i+=2, j+=2)
Chris@49:             {
Chris@49:             const eT tmp_i = Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             const eT tmp_j = Q.at(in_aux_row1, in_aux_col1 + j, mod_slice);
Chris@49:             
Chris@49:             out_colptr[i] += tmp_i;
Chris@49:             out_colptr[j] += tmp_j;
Chris@49:             }
Chris@49:           
Chris@49:           if(i < in_n_cols)
Chris@49:             {
Chris@49:             out_colptr[i] += Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             }
Chris@49:           }
Chris@49:         }
Chris@49:       }
Chris@49:     else
Chris@49:       {
Chris@49:       eT* out_mem = out.memptr();
Chris@49:       
Chris@49:       const Cube<eT>& Q = in.m;
Chris@49:       
Chris@49:       const uword in_aux_row1   = in.aux_row1;
Chris@49:       const uword in_aux_col1   = in.aux_col1;
Chris@49:       const uword in_aux_slice1 = in.aux_slice1;
Chris@49:       
Chris@49:       for(uword i=0; i<in_n_slices; ++i)
Chris@49:         {
Chris@49:         out_mem[i] += Q.at(in_aux_row1, in_aux_col1, in_aux_slice1 + i);
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! mat X -= Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::minus_inplace(Mat<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_cube_as_mat(out, in, "subtraction", true);
Chris@49:   
Chris@49:   const uword in_n_rows   = in.n_rows;
Chris@49:   const uword in_n_cols   = in.n_cols;
Chris@49:   const uword in_n_slices = in.n_slices;
Chris@49:   
Chris@49:   const uword out_n_rows    = out.n_rows;
Chris@49:   const uword out_n_cols    = out.n_cols;
Chris@49:   const uword out_vec_state = out.vec_state;
Chris@49:   
Chris@49:   if(in_n_slices == 1)
Chris@49:     {
Chris@49:     for(uword col=0; col < in_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_minus( out.colptr(col), in.slice_colptr(0, col), in_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(out_vec_state == 0)
Chris@49:       {
Chris@49:       if( (in_n_rows == out_n_rows) && (in_n_cols == 1) && (in_n_slices == out_n_cols) )
Chris@49:         {
Chris@49:         for(uword i=0; i < in_n_slices; ++i)
Chris@49:           {
Chris@49:           arrayops::inplace_minus( out.colptr(i), in.slice_colptr(i, 0), in_n_rows );
Chris@49:           }
Chris@49:         }
Chris@49:       else
Chris@49:       if( (in_n_rows == 1) && (in_n_cols == out_n_rows) && (in_n_slices == out_n_cols) )
Chris@49:         {
Chris@49:         const Cube<eT>& Q = in.m;
Chris@49:         
Chris@49:         const uword in_aux_row1   = in.aux_row1;
Chris@49:         const uword in_aux_col1   = in.aux_col1;
Chris@49:         const uword in_aux_slice1 = in.aux_slice1;
Chris@49:         
Chris@49:         for(uword slice=0; slice < in_n_slices; ++slice)
Chris@49:           {
Chris@49:           const uword mod_slice = in_aux_slice1 + slice;
Chris@49:           
Chris@49:           eT* out_colptr = out.colptr(slice);
Chris@49:           
Chris@49:           uword i,j;
Chris@49:           for(i=0, j=1; j < in_n_cols; i+=2, j+=2)
Chris@49:             {
Chris@49:             const eT tmp_i = Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             const eT tmp_j = Q.at(in_aux_row1, in_aux_col1 + j, mod_slice);
Chris@49:             
Chris@49:             out_colptr[i] -= tmp_i;
Chris@49:             out_colptr[j] -= tmp_j;
Chris@49:             }
Chris@49:           
Chris@49:           if(i < in_n_cols)
Chris@49:             {
Chris@49:             out_colptr[i] -= Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             }
Chris@49:           }
Chris@49:         }
Chris@49:       }
Chris@49:     else
Chris@49:       {
Chris@49:       eT* out_mem = out.memptr();
Chris@49:       
Chris@49:       const Cube<eT>& Q = in.m;
Chris@49:       
Chris@49:       const uword in_aux_row1   = in.aux_row1;
Chris@49:       const uword in_aux_col1   = in.aux_col1;
Chris@49:       const uword in_aux_slice1 = in.aux_slice1;
Chris@49:       
Chris@49:       for(uword i=0; i<in_n_slices; ++i)
Chris@49:         {
Chris@49:         out_mem[i] -= Q.at(in_aux_row1, in_aux_col1, in_aux_slice1 + i);
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! mat X %= Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::schur_inplace(Mat<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_cube_as_mat(out, in, "element-wise multiplication", true);
Chris@49:   
Chris@49:   const uword in_n_rows   = in.n_rows;
Chris@49:   const uword in_n_cols   = in.n_cols;
Chris@49:   const uword in_n_slices = in.n_slices;
Chris@49:   
Chris@49:   const uword out_n_rows    = out.n_rows;
Chris@49:   const uword out_n_cols    = out.n_cols;
Chris@49:   const uword out_vec_state = out.vec_state;
Chris@49:   
Chris@49:   if(in_n_slices == 1)
Chris@49:     {
Chris@49:     for(uword col=0; col < in_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_mul( out.colptr(col), in.slice_colptr(0, col), in_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(out_vec_state == 0)
Chris@49:       {
Chris@49:       if( (in_n_rows == out_n_rows) && (in_n_cols == 1) && (in_n_slices == out_n_cols) )
Chris@49:         {
Chris@49:         for(uword i=0; i < in_n_slices; ++i)
Chris@49:           {
Chris@49:           arrayops::inplace_mul( out.colptr(i), in.slice_colptr(i, 0), in_n_rows );
Chris@49:           }
Chris@49:         }
Chris@49:       else
Chris@49:       if( (in_n_rows == 1) && (in_n_cols == out_n_rows) && (in_n_slices == out_n_cols) )
Chris@49:         {
Chris@49:         const Cube<eT>& Q = in.m;
Chris@49:         
Chris@49:         const uword in_aux_row1   = in.aux_row1;
Chris@49:         const uword in_aux_col1   = in.aux_col1;
Chris@49:         const uword in_aux_slice1 = in.aux_slice1;
Chris@49:         
Chris@49:         for(uword slice=0; slice < in_n_slices; ++slice)
Chris@49:           {
Chris@49:           const uword mod_slice = in_aux_slice1 + slice;
Chris@49:           
Chris@49:           eT* out_colptr = out.colptr(slice);
Chris@49:           
Chris@49:           uword i,j;
Chris@49:           for(i=0, j=1; j < in_n_cols; i+=2, j+=2)
Chris@49:             {
Chris@49:             const eT tmp_i = Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             const eT tmp_j = Q.at(in_aux_row1, in_aux_col1 + j, mod_slice);
Chris@49:             
Chris@49:             out_colptr[i] *= tmp_i;
Chris@49:             out_colptr[j] *= tmp_j;
Chris@49:             }
Chris@49:           
Chris@49:           if(i < in_n_cols)
Chris@49:             {
Chris@49:             out_colptr[i] *= Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             }
Chris@49:           }
Chris@49:         }
Chris@49:       }
Chris@49:     else
Chris@49:       {
Chris@49:       eT* out_mem = out.memptr();
Chris@49:       
Chris@49:       const Cube<eT>& Q = in.m;
Chris@49:       
Chris@49:       const uword in_aux_row1   = in.aux_row1;
Chris@49:       const uword in_aux_col1   = in.aux_col1;
Chris@49:       const uword in_aux_slice1 = in.aux_slice1;
Chris@49:       
Chris@49:       for(uword i=0; i<in_n_slices; ++i)
Chris@49:         {
Chris@49:         out_mem[i] *= Q.at(in_aux_row1, in_aux_col1, in_aux_slice1 + i);
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! mat X /= Y.subcube(...)
Chris@49: template<typename eT>
Chris@49: inline
Chris@49: void
Chris@49: subview_cube<eT>::div_inplace(Mat<eT>& out, const subview_cube<eT>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   arma_debug_assert_cube_as_mat(out, in, "element-wise division", true);
Chris@49:   
Chris@49:   const uword in_n_rows   = in.n_rows;
Chris@49:   const uword in_n_cols   = in.n_cols;
Chris@49:   const uword in_n_slices = in.n_slices;
Chris@49:   
Chris@49:   const uword out_n_rows    = out.n_rows;
Chris@49:   const uword out_n_cols    = out.n_cols;
Chris@49:   const uword out_vec_state = out.vec_state;
Chris@49:   
Chris@49:   if(in_n_slices == 1)
Chris@49:     {
Chris@49:     for(uword col=0; col < in_n_cols; ++col)
Chris@49:       {
Chris@49:       arrayops::inplace_div( out.colptr(col), in.slice_colptr(0, col), in_n_rows );
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     if(out_vec_state == 0)
Chris@49:       {
Chris@49:       if( (in_n_rows == out_n_rows) && (in_n_cols == 1) && (in_n_slices == out_n_cols) )
Chris@49:         {
Chris@49:         for(uword i=0; i < in_n_slices; ++i)
Chris@49:           {
Chris@49:           arrayops::inplace_div( out.colptr(i), in.slice_colptr(i, 0), in_n_rows );
Chris@49:           }
Chris@49:         }
Chris@49:       else
Chris@49:       if( (in_n_rows == 1) && (in_n_cols == out_n_rows) && (in_n_slices == out_n_cols) )
Chris@49:         {
Chris@49:         const Cube<eT>& Q = in.m;
Chris@49:         
Chris@49:         const uword in_aux_row1   = in.aux_row1;
Chris@49:         const uword in_aux_col1   = in.aux_col1;
Chris@49:         const uword in_aux_slice1 = in.aux_slice1;
Chris@49:         
Chris@49:         for(uword slice=0; slice < in_n_slices; ++slice)
Chris@49:           {
Chris@49:           const uword mod_slice = in_aux_slice1 + slice;
Chris@49:           
Chris@49:           eT* out_colptr = out.colptr(slice);
Chris@49:           
Chris@49:           uword i,j;
Chris@49:           for(i=0, j=1; j < in_n_cols; i+=2, j+=2)
Chris@49:             {
Chris@49:             const eT tmp_i = Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             const eT tmp_j = Q.at(in_aux_row1, in_aux_col1 + j, mod_slice);
Chris@49:             
Chris@49:             out_colptr[i] /= tmp_i;
Chris@49:             out_colptr[j] /= tmp_j;
Chris@49:             }
Chris@49:           
Chris@49:           if(i < in_n_cols)
Chris@49:             {
Chris@49:             out_colptr[i] /= Q.at(in_aux_row1, in_aux_col1 + i, mod_slice);
Chris@49:             }
Chris@49:           }
Chris@49:         }
Chris@49:       }
Chris@49:     else
Chris@49:       {
Chris@49:       eT* out_mem = out.memptr();
Chris@49:       
Chris@49:       const Cube<eT>& Q = in.m;
Chris@49:       
Chris@49:       const uword in_aux_row1   = in.aux_row1;
Chris@49:       const uword in_aux_col1   = in.aux_col1;
Chris@49:       const uword in_aux_slice1 = in.aux_slice1;
Chris@49:       
Chris@49:       for(uword i=0; i<in_n_slices; ++i)
Chris@49:         {
Chris@49:         out_mem[i] /= Q.at(in_aux_row1, in_aux_col1, in_aux_slice1 + i);
Chris@49:         }
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}