Mercurial > hg > segmenter-vamp-plugin
diff armadillo-3.900.4/include/armadillo_bits/diagmat_proxy.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/diagmat_proxy.hpp Thu Jun 13 10:25:24 2013 +0100 @@ -0,0 +1,578 @@ +// Copyright (C) 2008-2012 NICTA (www.nicta.com.au) +// Copyright (C) 2008-2012 Conrad Sanderson +// +// 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 diagmat_proxy +//! @{ + + + +template<typename T1> +class diagmat_proxy_default + { + public: + + typedef typename T1::elem_type elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy_default(const T1& X) + : P ( X ) + , P_is_vec( (resolves_to_vector<T1>::value) || (P.get_n_rows() == 1) || (P.get_n_cols() == 1) ) + , P_is_col( T1::is_col || (P.get_n_cols() == 1) ) + , n_elem ( P_is_vec ? P.get_n_elem() : (std::min)(P.get_n_elem(), P.get_n_rows()) ) + { + arma_extra_debug_sigprint(); + + arma_debug_check + ( + (P_is_vec == false) && (P.get_n_rows() != P.get_n_cols()), + "diagmat(): only vectors and square matrices are accepted" + ); + } + + + arma_inline + elem_type + operator[](const uword i) const + { + if(Proxy<T1>::prefer_at_accessor == false) + { + return P_is_vec ? P[i] : P.at(i,i); + } + else + { + if(P_is_vec) + { + return (P_is_col) ? P.at(i,0) : P.at(0,i); + } + else + { + return P.at(i,i); + } + } + } + + + arma_inline + elem_type + at(const uword row, const uword col) const + { + if(row == col) + { + if(Proxy<T1>::prefer_at_accessor == false) + { + return (P_is_vec) ? P[row] : P.at(row,row); + } + else + { + if(P_is_vec) + { + return (P_is_col) ? P.at(row,0) : P.at(0,row); + } + else + { + return P.at(row,row); + } + } + } + else + { + return elem_type(0); + } + } + + + const Proxy<T1> P; + const bool P_is_vec; + const bool P_is_col; + const uword n_elem; + }; + + + +template<typename T1> +class diagmat_proxy_fixed + { + public: + + typedef typename T1::elem_type elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy_fixed(const T1& X) + : P(X) + { + arma_extra_debug_sigprint(); + + arma_debug_check + ( + (P_is_vec == false) && (T1::n_rows != T1::n_cols), + "diagmat(): only vectors and square matrices are accepted" + ); + } + + + arma_inline + elem_type + operator[](const uword i) const + { + return (P_is_vec) ? P[i] : P.at(i,i); + } + + + arma_inline + elem_type + at(const uword row, const uword col) const + { + if(row == col) + { + return (P_is_vec) ? P[row] : P.at(row,row); + } + else + { + return elem_type(0); + } + } + + const T1& P; + + static const bool P_is_vec = (T1::n_rows == 1) || (T1::n_cols == 1); + static const uword n_elem = P_is_vec ? T1::n_elem : ( (T1::n_elem < T1::n_rows) ? T1::n_elem : T1::n_rows ); + }; + + + +template<typename T1, bool condition> +struct diagmat_proxy_redirect {}; + +template<typename T1> +struct diagmat_proxy_redirect<T1, false> { typedef diagmat_proxy_default<T1> result; }; + +template<typename T1> +struct diagmat_proxy_redirect<T1, true> { typedef diagmat_proxy_fixed<T1> result; }; + + +template<typename T1> +class diagmat_proxy : public diagmat_proxy_redirect<T1, is_Mat_fixed<T1>::value >::result + { + public: + inline diagmat_proxy(const T1& X) + : diagmat_proxy_redirect< T1, is_Mat_fixed<T1>::value >::result(X) + { + } + }; + + + +template<typename eT> +class diagmat_proxy< Mat<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy(const Mat<eT>& X) + : P ( X ) + , P_is_vec( (X.n_rows == 1) || (X.n_cols == 1) ) + , n_elem ( P_is_vec ? X.n_elem : (std::min)(X.n_elem, X.n_rows) ) + { + arma_extra_debug_sigprint(); + + arma_debug_check + ( + (P_is_vec == false) && (P.n_rows != P.n_cols), + "diagmat(): only vectors and square matrices are accepted" + ); + } + + arma_inline elem_type operator[] (const uword i) const { return P_is_vec ? P[i] : P.at(i,i); } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? ( P_is_vec ? P[row] : P.at(row,row) ) : elem_type(0); } + + const Mat<eT>& P; + const bool P_is_vec; + const uword n_elem; + }; + + + +template<typename eT> +class diagmat_proxy< Row<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + + inline + diagmat_proxy(const Row<eT>& X) + : P(X) + , n_elem(X.n_elem) + { + arma_extra_debug_sigprint(); + } + + arma_inline elem_type operator[] (const uword i) const { return P[i]; } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? P[row] : elem_type(0); } + + static const bool P_is_vec = true; + + const Row<eT>& P; + const uword n_elem; + }; + + + +template<typename eT> +class diagmat_proxy< Col<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + + inline + diagmat_proxy(const Col<eT>& X) + : P(X) + , n_elem(X.n_elem) + { + arma_extra_debug_sigprint(); + } + + arma_inline elem_type operator[] (const uword i) const { return P[i]; } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? P[row] : elem_type(0); } + + static const bool P_is_vec = true; + + const Col<eT>& P; + const uword n_elem; + }; + + + +template<typename eT> +class diagmat_proxy< subview_row<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + + inline + diagmat_proxy(const subview_row<eT>& X) + : P(X) + , n_elem(X.n_elem) + { + arma_extra_debug_sigprint(); + } + + arma_inline elem_type operator[] (const uword i) const { return P[i]; } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? P[row] : elem_type(0); } + + static const bool P_is_vec = true; + + const subview_row<eT>& P; + const uword n_elem; + }; + + + +template<typename eT> +class diagmat_proxy< subview_col<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + + inline + diagmat_proxy(const subview_col<eT>& X) + : P(X) + , n_elem(X.n_elem) + { + arma_extra_debug_sigprint(); + } + + arma_inline elem_type operator[] (const uword i) const { return P[i]; } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? P[row] : elem_type(0); } + + static const bool P_is_vec = true; + + const subview_col<eT>& P; + const uword n_elem; + }; + + + +// +// +// + + + +template<typename T1> +class diagmat_proxy_check_default + { + public: + + typedef typename T1::elem_type elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy_check_default(const T1& X, const Mat<typename T1::elem_type>&) + : P(X) + , P_is_vec( (resolves_to_vector<T1>::value) || (P.n_rows == 1) || (P.n_cols == 1) ) + , n_elem( P_is_vec ? P.n_elem : (std::min)(P.n_elem, P.n_rows) ) + { + arma_extra_debug_sigprint(); + + arma_debug_check + ( + (P_is_vec == false) && (P.n_rows != P.n_cols), + "diagmat(): only vectors and square matrices are accepted" + ); + } + + arma_inline elem_type operator[] (const uword i) const { return P_is_vec ? P[i] : P.at(i,i); } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? ( P_is_vec ? P[row] : P.at(row,row) ) : elem_type(0); } + + const Mat<elem_type> P; + const bool P_is_vec; + const uword n_elem; + }; + + + +template<typename T1> +class diagmat_proxy_check_fixed + { + public: + + typedef typename T1::elem_type eT; + typedef typename T1::elem_type elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy_check_fixed(const T1& X, const Mat<eT>& out) + : P( const_cast<eT*>(X.memptr()), T1::n_rows, T1::n_cols, (&X == &out), false ) + { + arma_extra_debug_sigprint(); + + arma_debug_check + ( + (P_is_vec == false) && (T1::n_rows != T1::n_cols), + "diagmat(): only vectors and square matrices are accepted" + ); + } + + + arma_inline eT operator[] (const uword i) const { return P_is_vec ? P[i] : P.at(i,i); } + arma_inline eT at (const uword row, const uword col) const { return (row == col) ? ( P_is_vec ? P[row] : P.at(row,row) ) : elem_type(0); } + + const Mat<eT> P; // TODO: why not just store X directly as T1& ? test with fixed size vectors and matrices + + static const bool P_is_vec = (T1::n_rows == 1) || (T1::n_cols == 1); + static const uword n_elem = P_is_vec ? T1::n_elem : ( (T1::n_elem < T1::n_rows) ? T1::n_elem : T1::n_rows ); + }; + + + +template<typename T1, bool condition> +struct diagmat_proxy_check_redirect {}; + +template<typename T1> +struct diagmat_proxy_check_redirect<T1, false> { typedef diagmat_proxy_check_default<T1> result; }; + +template<typename T1> +struct diagmat_proxy_check_redirect<T1, true> { typedef diagmat_proxy_check_fixed<T1> result; }; + + +template<typename T1> +class diagmat_proxy_check : public diagmat_proxy_check_redirect<T1, is_Mat_fixed<T1>::value >::result + { + public: + inline diagmat_proxy_check(const T1& X, const Mat<typename T1::elem_type>& out) + : diagmat_proxy_check_redirect< T1, is_Mat_fixed<T1>::value >::result(X, out) + { + } + }; + + + +template<typename eT> +class diagmat_proxy_check< Mat<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + + inline + diagmat_proxy_check(const Mat<eT>& X, const Mat<eT>& out) + : P_local ( (&X == &out) ? new Mat<eT>(X) : 0 ) + , P ( (&X == &out) ? (*P_local) : X ) + , P_is_vec( (P.n_rows == 1) || (P.n_cols == 1) ) + , n_elem ( P_is_vec ? P.n_elem : (std::min)(P.n_elem, P.n_rows) ) + { + arma_extra_debug_sigprint(); + + arma_debug_check + ( + (P_is_vec == false) && (P.n_rows != P.n_cols), + "diagmat(): only vectors and square matrices are accepted" + ); + } + + inline ~diagmat_proxy_check() + { + if(P_local) { delete P_local; } + } + + arma_inline elem_type operator[] (const uword i) const { return P_is_vec ? P[i] : P.at(i,i); } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? ( P_is_vec ? P[row] : P.at(row,row) ) : elem_type(0); } + + const Mat<eT>* P_local; + const Mat<eT>& P; + const bool P_is_vec; + const uword n_elem; + }; + + + +template<typename eT> +class diagmat_proxy_check< Row<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy_check(const Row<eT>& X, const Mat<eT>& out) + : P_local ( (&X == reinterpret_cast<const Row<eT>*>(&out)) ? new Row<eT>(X) : 0 ) + , P ( (&X == reinterpret_cast<const Row<eT>*>(&out)) ? (*P_local) : X ) + , n_elem (X.n_elem) + { + arma_extra_debug_sigprint(); + } + + inline ~diagmat_proxy_check() + { + if(P_local) { delete P_local; } + } + + arma_inline elem_type operator[] (const uword i) const { return P[i]; } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? P[row] : elem_type(0); } + + static const bool P_is_vec = true; + + const Row<eT>* P_local; + const Row<eT>& P; + const uword n_elem; + }; + + + +template<typename eT> +class diagmat_proxy_check< Col<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy_check(const Col<eT>& X, const Mat<eT>& out) + : P_local ( (&X == reinterpret_cast<const Col<eT>*>(&out)) ? new Col<eT>(X) : 0 ) + , P ( (&X == reinterpret_cast<const Col<eT>*>(&out)) ? (*P_local) : X ) + , n_elem (X.n_elem) + { + arma_extra_debug_sigprint(); + } + + inline ~diagmat_proxy_check() + { + if(P_local) { delete P_local; } + } + + arma_inline elem_type operator[] (const uword i) const { return P[i]; } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? P[row] : elem_type(0); } + + static const bool P_is_vec = true; + + const Col<eT>* P_local; + const Col<eT>& P; + const uword n_elem; + }; + + + +template<typename eT> +class diagmat_proxy_check< subview_row<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy_check(const subview_row<eT>& X, const Mat<eT>&) + : P ( X ) + , n_elem ( X.n_elem ) + { + arma_extra_debug_sigprint(); + } + + arma_inline elem_type operator[] (const uword i) const { return P[i]; } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? P[row] : elem_type(0); } + + static const bool P_is_vec = true; + + const Row<eT> P; + const uword n_elem; + }; + + + +template<typename eT> +class diagmat_proxy_check< subview_col<eT> > + { + public: + + typedef eT elem_type; + typedef typename get_pod_type<elem_type>::result pod_type; + + inline + diagmat_proxy_check(const subview_col<eT>& X, const Mat<eT>& out) + : P ( const_cast<eT*>(X.colptr(0)), X.n_rows, (&(X.m) == &out), false ) + , n_elem( X.n_elem ) + //, X_ref ( X ) + { + arma_extra_debug_sigprint(); + } + + arma_inline elem_type operator[] (const uword i) const { return P[i]; } + arma_inline elem_type at (const uword row, const uword col) const { return (row == col) ? P[row] : elem_type(0); } + + static const bool P_is_vec = true; + + const Col<eT> P; + const uword n_elem; + + //const subview_col<eT>& X_ref; // prevents the compiler from potentially deleting X before we're done with it + }; + + + +//! @}