#pragma once

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


namespace numerics{

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

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

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

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

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

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

	template <typename T>
	inline utils::Vector<T> add(const utils::Vector<T>& v, const utils::Vector<T>& p) {
	    utils::Vector<T> out = v;
	    inplace_add(out, p);
	    return out;
	}
	// ---------------- Rowwise ----------------
	template <typename T>
	inline void inplace_add_rowwise(utils::Matrix<T>& A, const utils::Vector<T>& v) {
		detail::inplace_add_rowwise_serial(A,v);
	}

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

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

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

}