Chris@49
|
1 // Copyright (C) 2010-2013 NICTA (www.nicta.com.au)
|
Chris@49
|
2 // Copyright (C) 2010-2013 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
|
Chris@49
|
10 //! \addtogroup glue_cross
|
Chris@49
|
11 //! @{
|
Chris@49
|
12
|
Chris@49
|
13
|
Chris@49
|
14
|
Chris@49
|
15 template<typename T1, typename T2>
|
Chris@49
|
16 inline
|
Chris@49
|
17 void
|
Chris@49
|
18 glue_cross::apply(Mat<typename T1::elem_type>& out, const Glue<T1, T2, glue_cross>& X)
|
Chris@49
|
19 {
|
Chris@49
|
20 arma_extra_debug_sigprint();
|
Chris@49
|
21
|
Chris@49
|
22 typedef typename T1::elem_type eT;
|
Chris@49
|
23
|
Chris@49
|
24 const Proxy<T1> PA(X.A);
|
Chris@49
|
25 const Proxy<T2> PB(X.B);
|
Chris@49
|
26
|
Chris@49
|
27 arma_debug_check( ((PA.get_n_elem() != 3) || (PB.get_n_elem() != 3)), "cross(): input vectors must have 3 elements" );
|
Chris@49
|
28
|
Chris@49
|
29 const uword PA_n_rows = Proxy<T1>::is_row ? 1 : PA.get_n_rows();
|
Chris@49
|
30 const uword PA_n_cols = Proxy<T1>::is_col ? 1 : PA.get_n_cols();
|
Chris@49
|
31
|
Chris@49
|
32 out.set_size(PA_n_rows, PA_n_cols);
|
Chris@49
|
33
|
Chris@49
|
34 eT* out_mem = out.memptr();
|
Chris@49
|
35
|
Chris@49
|
36 if( (Proxy<T1>::prefer_at_accessor == false) && (Proxy<T2>::prefer_at_accessor == false) )
|
Chris@49
|
37 {
|
Chris@49
|
38 typename Proxy<T1>::ea_type A = PA.get_ea();
|
Chris@49
|
39 typename Proxy<T2>::ea_type B = PB.get_ea();
|
Chris@49
|
40
|
Chris@49
|
41 const eT ax = A[0];
|
Chris@49
|
42 const eT ay = A[1];
|
Chris@49
|
43 const eT az = A[2];
|
Chris@49
|
44
|
Chris@49
|
45 const eT bx = B[0];
|
Chris@49
|
46 const eT by = B[1];
|
Chris@49
|
47 const eT bz = B[2];
|
Chris@49
|
48
|
Chris@49
|
49 out_mem[0] = ay*bz - az*by;
|
Chris@49
|
50 out_mem[1] = az*bx - ax*bz;
|
Chris@49
|
51 out_mem[2] = ax*by - ay*bx;
|
Chris@49
|
52 }
|
Chris@49
|
53 else
|
Chris@49
|
54 {
|
Chris@49
|
55 const bool PA_is_col = Proxy<T1>::is_col ? true : (PA_n_cols == 1);
|
Chris@49
|
56 const bool PB_is_col = Proxy<T2>::is_col ? true : (PB.get_n_cols() == 1);
|
Chris@49
|
57
|
Chris@49
|
58 const eT ax = PA.at(0,0);
|
Chris@49
|
59 const eT ay = PA_is_col ? PA.at(1,0) : PA.at(0,1);
|
Chris@49
|
60 const eT az = PA_is_col ? PA.at(2,0) : PA.at(0,2);
|
Chris@49
|
61
|
Chris@49
|
62 const eT bx = PB.at(0,0);
|
Chris@49
|
63 const eT by = PB_is_col ? PB.at(1,0) : PB.at(0,1);
|
Chris@49
|
64 const eT bz = PB_is_col ? PB.at(2,0) : PB.at(0,2);
|
Chris@49
|
65
|
Chris@49
|
66 out_mem[0] = ay*bz - az*by;
|
Chris@49
|
67 out_mem[1] = az*bx - ax*bz;
|
Chris@49
|
68 out_mem[2] = ax*by - ay*bx;
|
Chris@49
|
69 }
|
Chris@49
|
70 }
|
Chris@49
|
71
|
Chris@49
|
72
|
Chris@49
|
73
|
Chris@49
|
74 //! @}
|