#pragma once

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


namespace numerics{

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

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

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

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

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

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

	template <typename T>
	inline utils::Vector<T> sub(const utils::Vector<T>& v, const utils::Vector<T>& p) {
	    utils::Vector<T> out = v;
	    inplace_sub(out, p);
	    return out;
	}
	// ---------------- Rowwise ----------------
	template <typename T>
	inline void inplace_sub_rowwise(utils::Matrix<T>& A, const utils::Vector<T>& v) {
		detail::inplace_sub_rowwise_serial(A,v);
	}

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

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

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

}