|
max@0
|
1 // Copyright (C) 2009-2011 NICTA (www.nicta.com.au)
|
|
max@0
|
2 // Copyright (C) 2009-2011 Conrad Sanderson
|
|
max@0
|
3 // Copyright (C) 2009-2010 Dimitrios Bouzas
|
|
max@0
|
4 //
|
|
max@0
|
5 // This file is part of the Armadillo C++ library.
|
|
max@0
|
6 // It is provided without any warranty of fitness
|
|
max@0
|
7 // for any purpose. You can redistribute this file
|
|
max@0
|
8 // and/or modify it under the terms of the GNU
|
|
max@0
|
9 // Lesser General Public License (LGPL) as published
|
|
max@0
|
10 // by the Free Software Foundation, either version 3
|
|
max@0
|
11 // of the License or (at your option) any later version.
|
|
max@0
|
12 // (see http://www.opensource.org/licenses for more info)
|
|
max@0
|
13
|
|
max@0
|
14
|
|
max@0
|
15
|
|
max@0
|
16 //! \addtogroup op_cor
|
|
max@0
|
17 //! @{
|
|
max@0
|
18
|
|
max@0
|
19
|
|
max@0
|
20
|
|
max@0
|
21 template<typename eT>
|
|
max@0
|
22 inline
|
|
max@0
|
23 void
|
|
max@0
|
24 op_cor::direct_cor(Mat<eT>& out, const Mat<eT>& A, const uword norm_type)
|
|
max@0
|
25 {
|
|
max@0
|
26 arma_extra_debug_sigprint();
|
|
max@0
|
27
|
|
max@0
|
28 if(A.is_empty())
|
|
max@0
|
29 {
|
|
max@0
|
30 out.reset();
|
|
max@0
|
31 return;
|
|
max@0
|
32 }
|
|
max@0
|
33
|
|
max@0
|
34 if(A.is_vec())
|
|
max@0
|
35 {
|
|
max@0
|
36 out.set_size(1,1);
|
|
max@0
|
37 out[0] = eT(1);
|
|
max@0
|
38 }
|
|
max@0
|
39 else
|
|
max@0
|
40 {
|
|
max@0
|
41 const uword N = A.n_rows;
|
|
max@0
|
42 const eT norm_val = (norm_type == 0) ? ( (N > 1) ? eT(N-1) : eT(1) ) : eT(N);
|
|
max@0
|
43
|
|
max@0
|
44 const Row<eT> acc = sum(A);
|
|
max@0
|
45 const Row<eT> sd = stddev(A);
|
|
max@0
|
46
|
|
max@0
|
47 out = (trans(A) * A);
|
|
max@0
|
48 out -= (trans(acc) * acc)/eT(N);
|
|
max@0
|
49 out /= norm_val;
|
|
max@0
|
50 out /= trans(sd) * sd;
|
|
max@0
|
51 }
|
|
max@0
|
52 }
|
|
max@0
|
53
|
|
max@0
|
54
|
|
max@0
|
55
|
|
max@0
|
56 template<typename T>
|
|
max@0
|
57 inline
|
|
max@0
|
58 void
|
|
max@0
|
59 op_cor::direct_cor(Mat< std::complex<T> >& out, const Mat< std::complex<T> >& A, const uword norm_type)
|
|
max@0
|
60 {
|
|
max@0
|
61 arma_extra_debug_sigprint();
|
|
max@0
|
62
|
|
max@0
|
63 typedef typename std::complex<T> eT;
|
|
max@0
|
64
|
|
max@0
|
65 if(A.is_empty())
|
|
max@0
|
66 {
|
|
max@0
|
67 out.reset();
|
|
max@0
|
68 return;
|
|
max@0
|
69 }
|
|
max@0
|
70
|
|
max@0
|
71 if(A.is_vec())
|
|
max@0
|
72 {
|
|
max@0
|
73 out.set_size(1,1);
|
|
max@0
|
74 out[0] = eT(1);
|
|
max@0
|
75 }
|
|
max@0
|
76 else
|
|
max@0
|
77 {
|
|
max@0
|
78 const uword N = A.n_rows;
|
|
max@0
|
79 const eT norm_val = (norm_type == 0) ? ( (N > 1) ? eT(N-1) : eT(1) ) : eT(N);
|
|
max@0
|
80
|
|
max@0
|
81 const Row<eT> acc = sum(A);
|
|
max@0
|
82 const Row<T> sd = stddev(A);
|
|
max@0
|
83
|
|
max@0
|
84 out = trans(A) * A; // out = strans(conj(A)) * A;
|
|
max@0
|
85 out -= (trans(acc) * acc)/eT(N); // out -= (strans(conj(acc)) * acc)/eT(N);
|
|
max@0
|
86 out /= norm_val;
|
|
max@0
|
87
|
|
max@0
|
88 //out = out / (trans(sd) * sd);
|
|
max@0
|
89 out /= conv_to< Mat<eT> >::from(trans(sd) * sd);
|
|
max@0
|
90 }
|
|
max@0
|
91 }
|
|
max@0
|
92
|
|
max@0
|
93
|
|
max@0
|
94
|
|
max@0
|
95 template<typename T1>
|
|
max@0
|
96 inline
|
|
max@0
|
97 void
|
|
max@0
|
98 op_cor::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_cor>& in)
|
|
max@0
|
99 {
|
|
max@0
|
100 arma_extra_debug_sigprint();
|
|
max@0
|
101
|
|
max@0
|
102 typedef typename T1::elem_type eT;
|
|
max@0
|
103
|
|
max@0
|
104 const unwrap_check<T1> tmp(in.m, out);
|
|
max@0
|
105 const Mat<eT>& A = tmp.M;
|
|
max@0
|
106
|
|
max@0
|
107 const uword norm_type = in.aux_uword_a;
|
|
max@0
|
108
|
|
max@0
|
109 op_cor::direct_cor(out, A, norm_type);
|
|
max@0
|
110 }
|
|
max@0
|
111
|
|
max@0
|
112
|
|
max@0
|
113
|
|
max@0
|
114 //! @}
|
