annotate armadillo-3.900.4/include/armadillo_bits/glue_cross_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) 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 //! @}