#pragma once

#include "schema.hpp"
#include "schema_meta.hpp"

namespace saw {
namespace schema {

/**
 * Tensor Schema. I think I need this due to the amount of mathematical operators being able to be represented by this one construct.
 */
template<typename Inner, uint64_t... D>
struct Tensor {
	static constexpr uint64_t Rank = sizeof...(D);
	static constexpr std::array<uint64_t,Rank> Dimension{D...};
	static constexpr string_literal name = "Tensor";
};

template<typename Inner>
using Scalar = Tensor<Inner>;

template<typename Inner, uint64_t D>
using Vector = Tensor<Inner,D>;

template<typename Inner, uint64_t D1, uint64_t D2>
using Matrix = Tensor<Inner,D1,D2>;

}

template<typename Inner, uint64_t... D>
struct meta_schema<schema::Tensor<Inner, D...>> {
	using Schema = schema::Tensor<Inner, D...>;
	using MetaSchema = schema::FixedArray<schema::UInt64, sizeof...(D)>;
};
}