annotate armadillo-3.900.4/include/armadillo_bits/spop_htrans_meat.hpp @ 84:55a047986812 tip

Update library URI so as not to be document-local
author Chris Cannam
date Wed, 22 Apr 2020 14:21:57 +0100
parents 1ec0e2823891
children
rev   line source
Chris@49 1 // Copyright (C) 2012 Ryan Curtin
Chris@49 2 // Copyright (C) 2012 Conrad Sanderson
Chris@49 3 //
Chris@49 4 // This Source Code Form is subject to the terms of the Mozilla Public
Chris@49 5 // License, v. 2.0. If a copy of the MPL was not distributed with this
Chris@49 6 // file, You can obtain one at http://mozilla.org/MPL/2.0/.
Chris@49 7
Chris@49 8
Chris@49 9 //! \addtogroup spop_htrans
Chris@49 10 //! @{
Chris@49 11
Chris@49 12
Chris@49 13
Chris@49 14 template<typename T1>
Chris@49 15 arma_hot
Chris@49 16 inline
Chris@49 17 void
Chris@49 18 spop_htrans::apply(SpMat<typename T1::elem_type>& out, const SpOp<T1,spop_htrans>& in, const typename arma_not_cx<typename T1::elem_type>::result* junk)
Chris@49 19 {
Chris@49 20 arma_extra_debug_sigprint();
Chris@49 21 arma_ignore(junk);
Chris@49 22
Chris@49 23 spop_strans::apply_proxy(out, in.m);
Chris@49 24 }
Chris@49 25
Chris@49 26
Chris@49 27
Chris@49 28 template<typename T1>
Chris@49 29 arma_hot
Chris@49 30 inline
Chris@49 31 void
Chris@49 32 spop_htrans::apply(SpMat<typename T1::elem_type>& out, const SpOp<T1,spop_htrans>& in, const typename arma_cx_only<typename T1::elem_type>::result* junk)
Chris@49 33 {
Chris@49 34 arma_extra_debug_sigprint();
Chris@49 35 arma_ignore(junk);
Chris@49 36
Chris@49 37 const SpProxy<T1> p(in.m);
Chris@49 38
Chris@49 39 if(p.is_alias(out) == false)
Chris@49 40 {
Chris@49 41 out.set_size( p.get_n_cols(), p.get_n_rows() );
Chris@49 42
Chris@49 43 out.mem_resize(p.get_n_nonzero());
Chris@49 44
Chris@49 45 typename SpProxy<T1>::const_row_iterator_type it = p.begin_row();
Chris@49 46
Chris@49 47 while(it.pos() < p.get_n_nonzero())
Chris@49 48 {
Chris@49 49 access::rw(out.values[it.pos()]) = std::conj(*it);
Chris@49 50 access::rw(out.row_indices[it.pos()]) = it.col(); // transpose
Chris@49 51 ++access::rw(out.col_ptrs[it.row() + 1]);
Chris@49 52
Chris@49 53 ++it;
Chris@49 54 }
Chris@49 55
Chris@49 56 // Fix column pointers.
Chris@49 57 const uword out_n_cols = out.n_cols;
Chris@49 58
Chris@49 59 for(uword c = 1; c <= out_n_cols; ++c)
Chris@49 60 {
Chris@49 61 access::rw(out.col_ptrs[c]) += out.col_ptrs[c - 1];
Chris@49 62 }
Chris@49 63 }
Chris@49 64 else
Chris@49 65 {
Chris@49 66 SpMat<typename T1::elem_type> result( p.get_n_cols(), p.get_n_rows() );
Chris@49 67
Chris@49 68 result.mem_resize(p.get_n_nonzero());
Chris@49 69
Chris@49 70 typename SpProxy<T1>::const_row_iterator_type it = p.begin_row();
Chris@49 71
Chris@49 72 while(it.pos() < p.get_n_nonzero())
Chris@49 73 {
Chris@49 74 access::rw(result.values[it.pos()]) = std::conj(*it);
Chris@49 75 access::rw(result.row_indices[it.pos()]) = it.col(); // transpose
Chris@49 76 ++access::rw(result.col_ptrs[it.row() + 1]);
Chris@49 77
Chris@49 78 ++it;
Chris@49 79 }
Chris@49 80
Chris@49 81 // Fix column pointers.
Chris@49 82 const uword result_n_cols = result.n_cols;
Chris@49 83
Chris@49 84 for(uword c = 1; c <= result_n_cols; ++c)
Chris@49 85 {
Chris@49 86 access::rw(result.col_ptrs[c]) += result.col_ptrs[c - 1];
Chris@49 87 }
Chris@49 88
Chris@49 89 out.steal_mem(result);
Chris@49 90 }
Chris@49 91 }
Chris@49 92
Chris@49 93
Chris@49 94
Chris@49 95 //! @}