Chris@49: // Copyright (C) 2012 Ryan Curtin
Chris@49: // Copyright (C) 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 spop_strans
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: arma_hot
Chris@49: inline
Chris@49: void
Chris@49: spop_strans::apply_proxy(SpMat<typename T1::elem_type>& out, const T1& X)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   const SpProxy<T1> p(X);
Chris@49:   
Chris@49:   if(p.is_alias(out) == false)
Chris@49:     {
Chris@49:     out.set_size( p.get_n_cols(), p.get_n_rows() );
Chris@49:     
Chris@49:     out.mem_resize(p.get_n_nonzero());
Chris@49:     
Chris@49:     typename SpProxy<T1>::const_row_iterator_type it = p.begin_row();
Chris@49:     
Chris@49:     while(it.pos() < p.get_n_nonzero())
Chris@49:       {
Chris@49:       access::rw(out.values[it.pos()]) = (*it);
Chris@49:       access::rw(out.row_indices[it.pos()]) = it.col(); // transpose
Chris@49:       ++access::rw(out.col_ptrs[it.row() + 1]);
Chris@49:       
Chris@49:       ++it;
Chris@49:       }
Chris@49:     
Chris@49:     // Fix column pointers.
Chris@49:     const uword out_n_cols = out.n_cols;
Chris@49:     
Chris@49:     for(uword c = 1; c <= out_n_cols; ++c)
Chris@49:       {
Chris@49:       access::rw(out.col_ptrs[c]) += out.col_ptrs[c - 1];
Chris@49:       }
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     SpMat<typename T1::elem_type> result( p.get_n_cols(), p.get_n_rows() );
Chris@49:     
Chris@49:     result.mem_resize(p.get_n_nonzero());
Chris@49:     
Chris@49:     typename SpProxy<T1>::const_row_iterator_type it = p.begin_row();
Chris@49:     
Chris@49:     while(it.pos() < p.get_n_nonzero())
Chris@49:       {
Chris@49:       access::rw(result.values[it.pos()]) = (*it);
Chris@49:       access::rw(result.row_indices[it.pos()]) = it.col(); // transpose
Chris@49:       ++access::rw(result.col_ptrs[it.row() + 1]);
Chris@49:       
Chris@49:       ++it;
Chris@49:       }
Chris@49:     
Chris@49:     // Fix column pointers.
Chris@49:     const uword result_n_cols = result.n_cols;
Chris@49:     
Chris@49:     for(uword c = 1; c <= result_n_cols; ++c)
Chris@49:       {
Chris@49:       access::rw(result.col_ptrs[c]) += result.col_ptrs[c - 1];
Chris@49:       }
Chris@49:     
Chris@49:     out.steal_mem(result);
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: arma_hot
Chris@49: inline
Chris@49: void
Chris@49: spop_strans::apply(SpMat<typename T1::elem_type>& out, const SpOp<T1,spop_strans>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   spop_strans::apply_proxy(out, in.m);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}