Mercurial > hg > segmenter-vamp-plugin
diff armadillo-3.900.4/include/armadillo_bits/fn_kron.hpp @ 49:1ec0e2823891
Switch to using subrepo copies of qm-dsp, nnls-chroma, vamp-plugin-sdk; update Armadillo version; assume build without external BLAS/LAPACK
author | Chris Cannam |
---|---|
date | Thu, 13 Jun 2013 10:25:24 +0100 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/armadillo-3.900.4/include/armadillo_bits/fn_kron.hpp Thu Jun 13 10:25:24 2013 +0100 @@ -0,0 +1,88 @@ +// Copyright (C) 2009-2010 NICTA (www.nicta.com.au) +// Copyright (C) 2009-2010 Conrad Sanderson +// Copyright (C) 2009-2010 Dimitrios Bouzas +// +// This Source Code Form is subject to the terms of the Mozilla Public +// License, v. 2.0. If a copy of the MPL was not distributed with this +// file, You can obtain one at http://mozilla.org/MPL/2.0/. + + +//! \addtogroup fn_kron +//! @{ + + + +//! \brief +//! kronecker product of two matrices, +//! with the matrices having the same element type +template<typename T1, typename T2> +arma_inline +const Glue<T1,T2,glue_kron> +kron(const Base<typename T1::elem_type,T1>& A, const Base<typename T1::elem_type,T2>& B) + { + arma_extra_debug_sigprint(); + + return Glue<T1, T2, glue_kron>(A.get_ref(), B.get_ref()); + } + + + +//! \brief +//! kronecker product of two matrices, +//! with the matrices having different element types +template<typename T, typename T1, typename T2> +inline +Mat<typename eT_promoter<T1,T2>::eT> +kron(const Base<std::complex<T>,T1>& X, const Base<T,T2>& Y) + { + arma_extra_debug_sigprint(); + + typedef typename std::complex<T> eT1; + + promote_type<eT1,T>::check(); + + const unwrap<T1> tmp1(X.get_ref()); + const unwrap<T2> tmp2(Y.get_ref()); + + const Mat<eT1>& A = tmp1.M; + const Mat<T >& B = tmp2.M; + + Mat<eT1> out; + + glue_kron::direct_kron(out, A, B); + + return out; + } + + + +//! \brief +//! kronecker product of two matrices, +//! with the matrices having different element types +template<typename T, typename T1, typename T2> +inline +Mat<typename eT_promoter<T1,T2>::eT> +kron(const Base<T,T1>& X, const Base<std::complex<T>,T2>& Y) + { + arma_extra_debug_sigprint(); + + typedef typename std::complex<T> eT2; + + promote_type<T,eT2>::check(); + + const unwrap<T1> tmp1(X.get_ref()); + const unwrap<T2> tmp2(Y.get_ref()); + + const Mat<T >& A = tmp1.M; + const Mat<eT2>& B = tmp2.M; + + Mat<eT2> out; + + glue_kron::direct_kron(out, A, B); + + return out; + } + + + +//! @}