Chris@49: // Copyright (C) 2012 Conrad Sanderson Chris@49: // Chris@49: // This Source Code Form is subject to the terms of the Mozilla Public Chris@49: // License, v. 2.0. If a copy of the MPL was not distributed with this Chris@49: // file, You can obtain one at http://mozilla.org/MPL/2.0/. Chris@49: Chris@49: Chris@49: //! \addtogroup spop_misc Chris@49: //! @{ Chris@49: Chris@49: Chris@49: Chris@49: namespace priv Chris@49: { Chris@49: template Chris@49: struct functor_scalar_times Chris@49: { Chris@49: const eT k; Chris@49: Chris@49: functor_scalar_times(const eT in_k) : k(in_k) {} Chris@49: Chris@49: arma_inline eT operator()(const eT val) const { return val * k; } Chris@49: }; Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: spop_scalar_times::apply(SpMat& out, const SpOp& in) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: typedef typename T1::elem_type eT; Chris@49: Chris@49: out.init_xform(in.m, priv::functor_scalar_times(in.aux)); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: namespace priv Chris@49: { Chris@49: struct functor_square Chris@49: { Chris@49: template Chris@49: arma_inline eT operator()(const eT val) const { return val*val; } Chris@49: }; Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: spop_square::apply(SpMat& out, const SpOp& in) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: out.init_xform(in.m, priv::functor_square()); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: namespace priv Chris@49: { Chris@49: struct functor_sqrt Chris@49: { Chris@49: template Chris@49: arma_inline eT operator()(const eT val) const { return eop_aux::sqrt(val); } Chris@49: }; Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: spop_sqrt::apply(SpMat& out, const SpOp& in) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: out.init_xform(in.m, priv::functor_sqrt()); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: namespace priv Chris@49: { Chris@49: struct functor_abs Chris@49: { Chris@49: template Chris@49: arma_inline eT operator()(const eT val) const { return eop_aux::arma_abs(val); } Chris@49: }; Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: spop_abs::apply(SpMat& out, const SpOp& in) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: out.init_xform(in.m, priv::functor_abs()); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: namespace priv Chris@49: { Chris@49: struct functor_cx_abs Chris@49: { Chris@49: template Chris@49: arma_inline T operator()(const std::complex& val) const { return std::abs(val); } Chris@49: }; Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: template Chris@49: inline Chris@49: void Chris@49: spop_cx_abs::apply(SpMat& out, const mtSpOp& in) Chris@49: { Chris@49: arma_extra_debug_sigprint(); Chris@49: Chris@49: out.init_xform_mt(in.m, priv::functor_cx_abs()); Chris@49: } Chris@49: Chris@49: Chris@49: Chris@49: //! @}