matrix_mul_scalar.hpp

Go to the documentation of this file.

# /*
#  * Created by YongGyu Lee on 2020/02/05.
#  */
#
# ifndef VCCC_MATH_MATRIX_MATRIX_MUL_SCALAR_HPP
# define VCCC_MATH_MATRIX_MATRIX_MUL_SCALAR_HPP
#
# include "vccc/__math/matrix/matrix_base.hpp"
# include "vccc/__math/matrix/type_helper.hpp"
# include <type_traits>
# include <cassert>
 
namespace vccc {
 
namespace internal {
namespace math {
 
template<typename LhsType, typename RhsType>
struct traits<MatrixMulScalar<LhsType, RhsType>> {
  enum : int {
    rows = traits<LhsType>::rows,
    cols = traits<LhsType>::cols,
    size = rows * cols,
  };
 
  enum : int {
    option = static_cast<int>(traits<LhsType>::option) | static_cast<int>(Flag::kReferenceUnsafe)
  };
  using value_type = typename LhsType::value_type;
};
 
} // namespace math
} // namespace internal
 
 
template<typename LhsType, typename RhsType>
class MatrixMulScalar : public MatrixBase<MatrixMulScalar<LhsType, RhsType>> {
 public:
  using value_type = typename LhsType::value_type;
  using lhs_type = internal::math::hold_type_selector_t<LhsType>;
  using rhs_type = const std::remove_reference_t<RhsType>;
 
  constexpr MatrixMulScalar(const LhsType& lhs, const RhsType& value) : lhs(lhs), value(value) {
    static_assert(!is_matrix<RhsType>::value, "");
  }
 
  constexpr value_type operator() (std::size_t i) const { return lhs(i) * value; }
  constexpr value_type operator() (std::size_t i, std::size_t j) const { return lhs(i, j) * value; }
  constexpr value_type operator[] (std::size_t i) const { return lhs[i] * value; }
 
 private:
  lhs_type lhs;
  rhs_type value;
};
 
template<typename LhsType, typename RhsType, std::enable_if_t<!is_matrix<RhsType>::value, int> = 0>
constexpr inline
MatrixMulScalar<LhsType, RhsType>
operator * (const MatrixBase<LhsType>& lhs, const RhsType& value) {
  return MatrixMulScalar<LhsType, RhsType>(*static_cast<const LhsType*>(&lhs), value);
}
 
 
} // namespace vccc
 
# endif // VCCC_MATH_MATRIX_MATRIX_MUL_SCALAR_HPP