00001 // Copyright (C) 2008-2010 NICTA (www.nicta.com.au)00002 // Copyright (C) 2008-2010 Conrad Sanderson00003 // 00004 // This file is part of the Armadillo C++ library.00005 // It is provided without any warranty of fitness00006 // for any purpose. You can redistribute this file00007 // and/or modify it under the terms of the GNU00008 // Lesser General Public License (LGPL) as published00009 // by the Free Software Foundation, either version 300010 // of the License or (at your option) any later version.00011 // (see http://www.opensource.org/licenses for more info)00012
00013
00016
00017
00018
00019 template<typename T1>
00020 inline00021 void00022op_diagvec::apply(Mat<typename T1::elem_type>& out, constOp<T1, op_diagvec>& X)
00023 {
00024 arma_extra_debug_sigprint();
00025
00026 typedeftypename T1::elem_type eT;
00027
00028 constswordid = (X.aux_uword_b > 0) ? -sword(X.aux_uword_a) : sword(X.aux_uword_a);
00029
00030 constunwrap_check<T1> tmp(X.m, out);
00031 constMat<eT>& A = tmp.M;
00032
00033 out = A.diag(id);
00034 }
00035
00036
00037