From eda5bc26f44ee9a6f83dcf8c91f17296d7fc509d Mon Sep 17 00:00:00 2001 From: Nao Pross Date: Mon, 12 Feb 2024 14:52:43 +0100 Subject: Move into version control --- .../include/armadillo_bits/spop_misc_meat.hpp | 596 +++++++++++++++++++++ 1 file changed, 596 insertions(+) create mode 100644 src/armadillo/include/armadillo_bits/spop_misc_meat.hpp (limited to 'src/armadillo/include/armadillo_bits/spop_misc_meat.hpp') diff --git a/src/armadillo/include/armadillo_bits/spop_misc_meat.hpp b/src/armadillo/include/armadillo_bits/spop_misc_meat.hpp new file mode 100644 index 0000000..1ef51cc --- /dev/null +++ b/src/armadillo/include/armadillo_bits/spop_misc_meat.hpp @@ -0,0 +1,596 @@ +// SPDX-License-Identifier: Apache-2.0 +// +// Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au) +// Copyright 2008-2016 National ICT Australia (NICTA) +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. +// ------------------------------------------------------------------------ + + +//! \addtogroup spop_misc +//! @{ + + + +namespace priv + { + template + struct functor_scalar_times + { + const eT k; + + functor_scalar_times(const eT in_k) : k(in_k) {} + + arma_inline eT operator()(const eT val) const { return val * k; } + }; + } + + + +template +inline +void +spop_scalar_times::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + + if(in.aux != eT(0)) + { + out.init_xform(in.m, priv::functor_scalar_times(in.aux)); + } + else + { + const SpProxy P(in.m); + + out.zeros( P.get_n_rows(), P.get_n_cols() ); + } + } + + + +namespace priv + { + template + struct functor_cx_scalar_times + { + typedef std::complex out_eT; + + const out_eT k; + + functor_cx_scalar_times(const out_eT in_k) : k(in_k) {} + + arma_inline out_eT operator()(const T val) const { return val * k; } + }; + } + + + +template +inline +void +spop_cx_scalar_times::apply(SpMat< std::complex >& out, const mtSpOp< std::complex, T1, spop_cx_scalar_times >& in) + { + arma_extra_debug_sigprint(); + + typedef typename T1::pod_type T; + typedef typename std::complex out_eT; + + if(in.aux_out_eT != out_eT(0)) + { + out.init_xform_mt(in.m, priv::functor_cx_scalar_times(in.aux_out_eT)); + } + else + { + const SpProxy P(in.m); + + out.zeros( P.get_n_rows(), P.get_n_cols() ); + } + } + + + +namespace priv + { + struct functor_square + { + template + arma_inline eT operator()(const eT val) const { return val*val; } + }; + } + + + +template +inline +void +spop_square::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_square()); + } + + + +namespace priv + { + struct functor_sqrt + { + template + arma_inline eT operator()(const eT val) const { return eop_aux::sqrt(val); } + }; + } + + + +template +inline +void +spop_sqrt::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_sqrt()); + } + + + +namespace priv + { + struct functor_abs + { + template + arma_inline eT operator()(const eT val) const { return eop_aux::arma_abs(val); } + }; + } + + + +template +inline +void +spop_abs::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_abs()); + } + + + +namespace priv + { + struct functor_cx_abs + { + template + arma_inline T operator()(const std::complex& val) const { return std::abs(val); } + }; + } + + + +template +inline +void +spop_cx_abs::apply(SpMat& out, const mtSpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform_mt(in.m, priv::functor_cx_abs()); + } + + + +namespace priv + { + struct functor_arg + { + template + arma_inline eT operator()(const eT val) const { return arma_arg::eval(val); } + }; + } + + + +template +inline +void +spop_arg::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_arg()); + } + + + +namespace priv + { + struct functor_cx_arg + { + template + arma_inline T operator()(const std::complex& val) const { return std::arg(val); } + }; + } + + + +template +inline +void +spop_cx_arg::apply(SpMat& out, const mtSpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform_mt(in.m, priv::functor_cx_arg()); + } + + + +namespace priv + { + struct functor_real + { + template + arma_inline T operator()(const std::complex& val) const { return val.real(); } + }; + } + + + +template +inline +void +spop_real::apply(SpMat& out, const mtSpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform_mt(in.m, priv::functor_real()); + } + + + +namespace priv + { + struct functor_imag + { + template + arma_inline T operator()(const std::complex& val) const { return val.imag(); } + }; + } + + + +template +inline +void +spop_imag::apply(SpMat& out, const mtSpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform_mt(in.m, priv::functor_imag()); + } + + + +namespace priv + { + struct functor_conj + { + template + arma_inline eT operator()(const eT val) const { return eop_aux::conj(val); } + }; + } + + + +template +inline +void +spop_conj::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_conj()); + } + + + +template +inline +void +spop_repelem::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + + const unwrap_spmat U(in.m); + const SpMat& X = U.M; + + const uword copies_per_row = in.aux_uword_a; + const uword copies_per_col = in.aux_uword_b; + + const uword out_n_rows = X.n_rows * copies_per_row; + const uword out_n_cols = X.n_cols * copies_per_col; + const uword out_nnz = X.n_nonzero * copies_per_row * copies_per_col; + + if( (out_n_rows > 0) && (out_n_cols > 0) && (out_nnz > 0) ) + { + umat locs(2, out_nnz, arma_nozeros_indicator()); + Col vals( out_nnz, arma_nozeros_indicator()); + + uword* locs_mem = locs.memptr(); + eT* vals_mem = vals.memptr(); + + typename SpMat::const_iterator X_it = X.begin(); + typename SpMat::const_iterator X_end = X.end(); + + for(; X_it != X_end; ++X_it) + { + const uword col_base = copies_per_col * X_it.col(); + const uword row_base = copies_per_row * X_it.row(); + + const eT X_val = (*X_it); + + for(uword cols = 0; cols < copies_per_col; cols++) + for(uword rows = 0; rows < copies_per_row; rows++) + { + (*locs_mem) = row_base + rows; ++locs_mem; + (*locs_mem) = col_base + cols; ++locs_mem; + + (*vals_mem) = X_val; ++vals_mem; + } + } + + out = SpMat(locs, vals, out_n_rows, out_n_cols); + } + else + { + out.zeros(out_n_rows, out_n_cols); + } + } + + + +template +inline +void +spop_reshape::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out = in.m; + + out.reshape(in.aux_uword_a, in.aux_uword_b); + } + + + +template +inline +void +spop_resize::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out = in.m; + + out.resize(in.aux_uword_a, in.aux_uword_b); + } + + + +namespace priv + { + struct functor_floor + { + template + arma_inline eT operator()(const eT val) const { return eop_aux::floor(val); } + }; + } + + + +template +inline +void +spop_floor::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_floor()); + } + + + +namespace priv + { + struct functor_ceil + { + template + arma_inline eT operator()(const eT val) const { return eop_aux::ceil(val); } + }; + } + + + +template +inline +void +spop_ceil::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_ceil()); + } + + + +namespace priv + { + struct functor_round + { + template + arma_inline eT operator()(const eT val) const { return eop_aux::round(val); } + }; + } + + + +template +inline +void +spop_round::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_round()); + } + + + +namespace priv + { + struct functor_trunc + { + template + arma_inline eT operator()(const eT val) const { return eop_aux::trunc(val); } + }; + } + + + +template +inline +void +spop_trunc::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_trunc()); + } + + + +namespace priv + { + struct functor_sign + { + template + arma_inline eT operator()(const eT val) const { return arma_sign(val); } + }; + } + + + +template +inline +void +spop_sign::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out.init_xform(in.m, priv::functor_sign()); + } + + + +template +inline +void +spop_diagvec::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + + const unwrap_spmat U(in.m); + + const SpMat& X = U.M; + + const uword a = in.aux_uword_a; + const uword b = in.aux_uword_b; + + const uword row_offset = (b > 0) ? a : 0; + const uword col_offset = (b == 0) ? a : 0; + + arma_debug_check_bounds + ( + ((row_offset > 0) && (row_offset >= X.n_rows)) || ((col_offset > 0) && (col_offset >= X.n_cols)), + "diagvec(): requested diagonal out of bounds" + ); + + const uword len = (std::min)(X.n_rows - row_offset, X.n_cols - col_offset); + + Col cache(len, arma_nozeros_indicator()); + eT* cache_mem = cache.memptr(); + + uword n_nonzero = 0; + + for(uword i=0; i < len; ++i) + { + const eT val = X.at(i + row_offset, i + col_offset); + + cache_mem[i] = val; + + n_nonzero += (val != eT(0)) ? uword(1) : uword(0); + } + + out.reserve(len, 1, n_nonzero); + + uword count = 0; + for(uword i=0; i < len; ++i) + { + const eT val = cache_mem[i]; + + if(val != eT(0)) + { + access::rw(out.row_indices[count]) = i; + access::rw(out.values[count]) = val; + ++count; + } + } + + access::rw(out.col_ptrs[0]) = 0; + access::rw(out.col_ptrs[1]) = n_nonzero; + } + + + +template +inline +void +spop_flipud::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out = reverse(in.m, 0); + } + + + +template +inline +void +spop_fliplr::apply(SpMat& out, const SpOp& in) + { + arma_extra_debug_sigprint(); + + out = reverse(in.m, 1); + } + + + +//! @} -- cgit v1.2.1