#pragma once

#include "./core/omp_config.h"
#include "detail/mul_serial.h"


namespace numerics{

	// ---------------- Scalar ----------------
	template <typename T>
	inline void inplace_mul(utils::Matrix<T>& A, const T b) {
		detail::inplace_mul_scalar_serial(A,b);
	}

	template <typename T>
	inline utils::Matrix<T> mul(const utils::Matrix<T>& A, const T b) {
	    utils::Matrix<T> out = A;
	    inplace_mul(out, b);
	    return out;
	}

	template <typename T>
	inline void inplace_mul(utils::Vector<T>& v, const T b) {
		detail::inplace_mul_scalar_serial(v,b);
	}

	template <typename T>
	inline utils::Vector<T> mul(const utils::Vector<T>& v, const T b) {
	    utils::Vector<T> out = v;
	    inplace_mul(out, b);
	    return out;
	}
	// ---------------- Elementwise ----------------
	template <typename T>
	inline void inplace_mul(utils::Matrix<T>& A, const utils::Matrix<T>& B) {
		detail::inplace_mul_elementwise_serial(A,B);
	}

	template <typename T>
	inline utils::Matrix<T> mul(const utils::Matrix<T>& A, const utils::Matrix<T>& B) {
	    utils::Matrix<T> out = A;
	    inplace_mul(out, B);
	    return out;
	}

	template <typename T>
	inline void inplace_mul(utils::Vector<T>& v, const utils::Vector<T>& p) {
		detail::inplace_mul_elementwise_serial(v,p);
	}

	template <typename T>
	inline utils::Vector<T> mul(const utils::Vector<T>& v, const utils::Vector<T>& p) {
	    utils::Vector<T> out = v;
	    inplace_mul(out, p);
	    return out;
	}
	// ---------------- Rowwise ----------------
	template <typename T>
	inline void inplace_mul_rowwise(utils::Matrix<T>& A, const utils::Vector<T>& v) {
		detail::inplace_mul_rowwise_serial(A,v);
	}

	template <typename T>
	inline utils::Matrix<T> mul_rowwise(const utils::Matrix<T>& A, const utils::Vector<T>& v) {
	    utils::Matrix<T> out = A;
	    inplace_mul_rowwise(out, v);
	    return out;
	}

	// ---------------- Colwise ----------------
	template <typename T>
	inline void inplace_mul_colwise(utils::Matrix<T>& A, const utils::Vector<T>& v) {
		detail::inplace_mul_colwise_serial(A,v);
	}

	template <typename T>
	inline utils::Matrix<T> mul_colwise(const utils::Matrix<T>& A, const utils::Vector<T>& v) {
	    utils::Matrix<T> out = A;
	    inplace_mul_colwise(out, v);
	    return out;
	}

}