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 //! @}
|