// 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 <typename eT>
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; }
};
}  // namespace priv

template <typename T1>
inline void spop_scalar_times::apply(SpMat<typename T1::elem_type>& out,
                                     const SpOp<T1, spop_scalar_times>& 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<eT>(in.aux));
  } else {
    const SpProxy<T1> P(in.m);

    out.zeros(P.get_n_rows(), P.get_n_cols());
  }
}

namespace priv {
template <typename T>
struct functor_cx_scalar_times {
  typedef std::complex<T> 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; }
};
}  // namespace priv

template <typename T1>
inline void spop_cx_scalar_times::apply(
    SpMat<std::complex<typename T1::pod_type> >& out,
    const mtSpOp<std::complex<typename T1::pod_type>, T1, spop_cx_scalar_times>& in) {
  arma_extra_debug_sigprint();

  typedef typename T1::pod_type T;
  typedef typename std::complex<T> out_eT;

  if (in.aux_out_eT != out_eT(0)) {
    out.init_xform_mt(in.m, priv::functor_cx_scalar_times<T>(in.aux_out_eT));
  } else {
    const SpProxy<T1> P(in.m);

    out.zeros(P.get_n_rows(), P.get_n_cols());
  }
}

namespace priv {
struct functor_square {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return val * val;
  }
};
}  // namespace priv

template <typename T1>
inline void spop_square::apply(SpMat<typename T1::elem_type>& out,
                               const SpOp<T1, spop_square>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_square());
}

namespace priv {
struct functor_sqrt {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return eop_aux::sqrt(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_sqrt::apply(SpMat<typename T1::elem_type>& out,
                             const SpOp<T1, spop_sqrt>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_sqrt());
}

namespace priv {
struct functor_abs {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return eop_aux::arma_abs(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_abs::apply(SpMat<typename T1::elem_type>& out,
                            const SpOp<T1, spop_abs>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_abs());
}

namespace priv {
struct functor_cx_abs {
  template <typename T>
  arma_inline T operator()(const std::complex<T>& val) const {
    return std::abs(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_cx_abs::apply(SpMat<typename T1::pod_type>& out,
                               const mtSpOp<typename T1::pod_type, T1, spop_cx_abs>& in) {
  arma_extra_debug_sigprint();

  out.init_xform_mt(in.m, priv::functor_cx_abs());
}

namespace priv {
struct functor_arg {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return arma_arg<eT>::eval(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_arg::apply(SpMat<typename T1::elem_type>& out,
                            const SpOp<T1, spop_arg>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_arg());
}

namespace priv {
struct functor_cx_arg {
  template <typename T>
  arma_inline T operator()(const std::complex<T>& val) const {
    return std::arg(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_cx_arg::apply(SpMat<typename T1::pod_type>& out,
                               const mtSpOp<typename T1::pod_type, T1, spop_cx_arg>& in) {
  arma_extra_debug_sigprint();

  out.init_xform_mt(in.m, priv::functor_cx_arg());
}

namespace priv {
struct functor_real {
  template <typename T>
  arma_inline T operator()(const std::complex<T>& val) const {
    return val.real();
  }
};
}  // namespace priv

template <typename T1>
inline void spop_real::apply(SpMat<typename T1::pod_type>& out,
                             const mtSpOp<typename T1::pod_type, T1, spop_real>& in) {
  arma_extra_debug_sigprint();

  out.init_xform_mt(in.m, priv::functor_real());
}

namespace priv {
struct functor_imag {
  template <typename T>
  arma_inline T operator()(const std::complex<T>& val) const {
    return val.imag();
  }
};
}  // namespace priv

template <typename T1>
inline void spop_imag::apply(SpMat<typename T1::pod_type>& out,
                             const mtSpOp<typename T1::pod_type, T1, spop_imag>& in) {
  arma_extra_debug_sigprint();

  out.init_xform_mt(in.m, priv::functor_imag());
}

namespace priv {
struct functor_conj {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return eop_aux::conj(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_conj::apply(SpMat<typename T1::elem_type>& out,
                             const SpOp<T1, spop_conj>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_conj());
}

template <typename T1>
inline void spop_repelem::apply(SpMat<typename T1::elem_type>& out,
                                const SpOp<T1, spop_repelem>& in) {
  arma_extra_debug_sigprint();

  typedef typename T1::elem_type eT;

  const unwrap_spmat<T1> U(in.m);
  const SpMat<eT>& 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<eT> vals(out_nnz, arma_nozeros_indicator());

    uword* locs_mem = locs.memptr();
    eT* vals_mem = vals.memptr();

    typename SpMat<eT>::const_iterator X_it = X.begin();
    typename SpMat<eT>::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<eT>(locs, vals, out_n_rows, out_n_cols);
  } else {
    out.zeros(out_n_rows, out_n_cols);
  }
}

template <typename T1>
inline void spop_reshape::apply(SpMat<typename T1::elem_type>& out,
                                const SpOp<T1, spop_reshape>& in) {
  arma_extra_debug_sigprint();

  out = in.m;

  out.reshape(in.aux_uword_a, in.aux_uword_b);
}

template <typename T1>
inline void spop_resize::apply(SpMat<typename T1::elem_type>& out,
                               const SpOp<T1, spop_resize>& in) {
  arma_extra_debug_sigprint();

  out = in.m;

  out.resize(in.aux_uword_a, in.aux_uword_b);
}

namespace priv {
struct functor_floor {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return eop_aux::floor(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_floor::apply(SpMat<typename T1::elem_type>& out,
                              const SpOp<T1, spop_floor>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_floor());
}

namespace priv {
struct functor_ceil {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return eop_aux::ceil(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_ceil::apply(SpMat<typename T1::elem_type>& out,
                             const SpOp<T1, spop_ceil>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_ceil());
}

namespace priv {
struct functor_round {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return eop_aux::round(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_round::apply(SpMat<typename T1::elem_type>& out,
                              const SpOp<T1, spop_round>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_round());
}

namespace priv {
struct functor_trunc {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return eop_aux::trunc(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_trunc::apply(SpMat<typename T1::elem_type>& out,
                              const SpOp<T1, spop_trunc>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_trunc());
}

namespace priv {
struct functor_sign {
  template <typename eT>
  arma_inline eT operator()(const eT val) const {
    return arma_sign(val);
  }
};
}  // namespace priv

template <typename T1>
inline void spop_sign::apply(SpMat<typename T1::elem_type>& out,
                             const SpOp<T1, spop_sign>& in) {
  arma_extra_debug_sigprint();

  out.init_xform(in.m, priv::functor_sign());
}

template <typename T1>
inline void spop_diagvec::apply(SpMat<typename T1::elem_type>& out,
                                const SpOp<T1, spop_diagvec>& in) {
  arma_extra_debug_sigprint();

  typedef typename T1::elem_type eT;

  const unwrap_spmat<T1> U(in.m);

  const SpMat<eT>& 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<eT> 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 <typename T1>
inline void spop_flipud::apply(SpMat<typename T1::elem_type>& out,
                               const SpOp<T1, spop_flipud>& in) {
  arma_extra_debug_sigprint();

  out = reverse(in.m, 0);
}

template <typename T1>
inline void spop_fliplr::apply(SpMat<typename T1::elem_type>& out,
                               const SpOp<T1, spop_fliplr>& in) {
  arma_extra_debug_sigprint();

  out = reverse(in.m, 1);
}

//! @}
