00001 // Copyright (C) 2010 NICTA and the authors listed below 00002 // http://nicta.com.au 00003 // 00004 // Authors: 00005 // - Conrad Sanderson (conradsand at ieee dot org) 00006 // 00007 // This file is part of the Armadillo C++ library. 00008 // It is provided without any warranty of fitness 00009 // for any purpose. You can redistribute this file 00010 // and/or modify it under the terms of the GNU 00011 // Lesser General Public License (LGPL) as published 00012 // by the Free Software Foundation, either version 3 00013 // of the License or (at your option) any later version. 00014 // (see http://www.opensource.org/licenses for more info) 00015 00016 00017 //! \addtogroup op_diagvec 00018 //! @{ 00019 00020 00021 00022 template<typename T1> 00023 inline 00024 void 00025 op_diagvec::apply(Mat<typename T1::elem_type>& out, const Op<T1, op_diagvec>& X) 00026 { 00027 arma_extra_debug_sigprint(); 00028 00029 typedef typename T1::elem_type eT; 00030 00031 const s32 id = (X.aux_u32_b > 0) ? -s32(X.aux_u32_a) : s32(X.aux_u32_a); 00032 00033 const unwrap_check<T1> tmp(X.m, out); 00034 const Mat<eT>& A = tmp.M; 00035 00036 out = A.diag(id); 00037 } 00038 00039 00040 00041 //! @}