Mercurial > hg > segmenter-vamp-plugin

annotate armadillo-2.4.4/include/armadillo_bits/glue_kron_meat.hpp @ 0:8b6102e2a9b0

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Armadillo Library
author maxzanoni76 <max.zanoni@eecs.qmul.ac.uk>
date Wed, 11 Apr 2012 09:27:06 +0100
parents
children
rev   line source
max@0 1 // Copyright (C) 2009-2010 NICTA (www.nicta.com.au)
max@0 2 // Copyright (C) 2009-2010 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 //! \addtogroup glue_kron
max@0 16 //! @{
max@0 17
max@0 18
max@0 19
max@0 20 //! \brief
max@0 21 //! both input matrices have the same element type
max@0 22 template<typename eT>
max@0 23 inline
max@0 24 void
max@0 25 glue_kron::direct_kron(Mat<eT>& out, const Mat<eT>& A, const Mat<eT>& B)
max@0 26 {
max@0 27 arma_extra_debug_sigprint();
max@0 28
max@0 29 const uword A_rows = A.n_rows;
max@0 30 const uword A_cols = A.n_cols;
max@0 31 const uword B_rows = B.n_rows;
max@0 32 const uword B_cols = B.n_cols;
max@0 33
max@0 34 out.set_size(A_rows*B_rows, A_cols*B_cols);
max@0 35
max@0 36 for(uword i = 0; i < A_rows; i++)
max@0 37 {
max@0 38 for(uword j = 0; j < A_cols; j++)
max@0 39 {
max@0 40 out.submat(i*B_rows, j*B_cols, (i+1)*B_rows-1, (j+1)*B_cols-1) = A(i,j) * B;
max@0 41 }
max@0 42 }
max@0 43 }
max@0 44
max@0 45
max@0 46
max@0 47 //! \brief
max@0 48 //! different types of input matrices
max@0 49 //! A -> complex, B -> basic element type
max@0 50 template<typename T>
max@0 51 inline
max@0 52 void
max@0 53 glue_kron::direct_kron(Mat< std::complex<T> >& out, const Mat< std::complex<T> >& A, const Mat<T>& B)
max@0 54 {
max@0 55 arma_extra_debug_sigprint();
max@0 56
max@0 57 typedef typename std::complex<T> eT;
max@0 58
max@0 59 const uword A_rows = A.n_rows;
max@0 60 const uword A_cols = A.n_cols;
max@0 61 const uword B_rows = B.n_rows;
max@0 62 const uword B_cols = B.n_cols;
max@0 63
max@0 64 out.set_size(A_rows*B_rows, A_cols*B_cols);
max@0 65
max@0 66 Mat<eT> tmp_B = conv_to< Mat<eT> >::from(B);
max@0 67
max@0 68 for(uword i = 0; i < A_rows; i++)
max@0 69 {
max@0 70 for(uword j = 0; j < A_cols; j++)
max@0 71 {
max@0 72 out.submat(i*B_rows, j*B_cols, (i+1)*B_rows-1, (j+1)*B_cols-1) = A(i,j) * tmp_B;
max@0 73 }
max@0 74 }
max@0 75 }
max@0 76
max@0 77
max@0 78
max@0 79 //! \brief
max@0 80 //! different types of input matrices
max@0 81 //! A -> basic element type, B -> complex
max@0 82 template<typename T>
max@0 83 inline
max@0 84 void
max@0 85 glue_kron::direct_kron(Mat< std::complex<T> >& out, const Mat<T>& A, const Mat< std::complex<T> >& B)
max@0 86 {
max@0 87 arma_extra_debug_sigprint();
max@0 88
max@0 89 const uword A_rows = A.n_rows;
max@0 90 const uword A_cols = A.n_cols;
max@0 91 const uword B_rows = B.n_rows;
max@0 92 const uword B_cols = B.n_cols;
max@0 93
max@0 94 out.set_size(A_rows*B_rows, A_cols*B_cols);
max@0 95
max@0 96 for(uword i = 0; i < A_rows; i++)
max@0 97 {
max@0 98 for(uword j = 0; j < A_cols; j++)
max@0 99 {
max@0 100 out.submat(i*B_rows, j*B_cols, (i+1)*B_rows-1, (j+1)*B_cols-1) = A(i,j) * B;
max@0 101 }
max@0 102 }
max@0 103 }
max@0 104
max@0 105
max@0 106
max@0 107 //! \brief
max@0 108 //! apply Kronecker product for two objects with same element type
max@0 109 template<typename T1, typename T2>
max@0 110 inline
max@0 111 void
max@0 112 glue_kron::apply(Mat<typename T1::elem_type>& out, const Glue<T1,T2,glue_kron>& X)
max@0 113 {
max@0 114 arma_extra_debug_sigprint();
max@0 115
max@0 116 typedef typename T1::elem_type eT;
max@0 117
max@0 118 const unwrap_check<T1> A_tmp(X.A, out);
max@0 119 const unwrap_check<T2> B_tmp(X.B, out);
max@0 120
max@0 121 const Mat<eT>& A = A_tmp.M;
max@0 122 const Mat<eT>& B = B_tmp.M;
max@0 123
max@0 124 glue_kron::direct_kron(out, A, B);
max@0 125 }
max@0 126
max@0 127
max@0 128
max@0 129 //! @}