Mercurial > hg > segmenter-vamp-plugin
diff armadillo-3.900.4/include/armadillo_bits/op_repmat_meat.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/op_repmat_meat.hpp Thu Jun 13 10:25:24 2013 +0100 @@ -0,0 +1,93 @@ +// Copyright (C) 2009-2012 NICTA (www.nicta.com.au) +// Copyright (C) 2009-2012 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 op_repmat +//! @{ + + + +//! \brief +//! implementation of the 'repeat matrix' operation, used for constructing matrices +template<typename T1> +inline +void +op_repmat::apply(Mat<typename T1::elem_type>& out, const Op<T1,op_repmat>& in) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + + const unwrap_check<T1> tmp(in.m, out); + const Mat<eT>& X = tmp.M; + + const uword copies_per_row = in.aux_uword_a; + const uword copies_per_col = in.aux_uword_b; + + const uword X_n_rows = X.n_rows; + const uword X_n_cols = X.n_cols; + + out.set_size(X_n_rows * copies_per_row, X_n_cols * copies_per_col); + + const uword out_n_rows = out.n_rows; + const uword out_n_cols = out.n_cols; + + // if( (out_n_rows > 0) && (out_n_cols > 0) ) + // { + // for(uword col = 0; col < out_n_cols; col += X_n_cols) + // for(uword row = 0; row < out_n_rows; row += X_n_rows) + // { + // out.submat(row, col, row+X_n_rows-1, col+X_n_cols-1) = X; + // } + // } + + if( (out_n_rows > 0) && (out_n_cols > 0) ) + { + if(copies_per_row != 1) + { + for(uword col_copy=0; col_copy < copies_per_col; ++col_copy) + { + const uword out_col_offset = X_n_cols * col_copy; + + for(uword col=0; col < X_n_cols; ++col) + { + eT* out_colptr = out.colptr(col + out_col_offset); + const eT* X_colptr = X.colptr(col); + + for(uword row_copy=0; row_copy < copies_per_row; ++row_copy) + { + const uword out_row_offset = X_n_rows * row_copy; + + arrayops::copy( &out_colptr[out_row_offset], X_colptr, X_n_rows ); + } + } + } + } + else + { + for(uword col_copy=0; col_copy < copies_per_col; ++col_copy) + { + const uword out_col_offset = X_n_cols * col_copy; + + for(uword col=0; col < X_n_cols; ++col) + { + eT* out_colptr = out.colptr(col + out_col_offset); + const eT* X_colptr = X.colptr(col); + + arrayops::copy( out_colptr, X_colptr, X_n_rows ); + } + } + } + } + + } + + + +//! @}