// Copyright (C) 2020 Jérôme Leclercq
// This file is part of the "Nazara Engine - Shader generator"
// For conditions of distribution and use, see copyright notice in Config.hpp

#pragma once

#ifndef NAZARA_SPIRVSECTION_HPP
#define NAZARA_SPIRVSECTION_HPP

#include <Nazara/Prerequisites.hpp>
#include <Nazara/Shader/ShaderEnums.hpp>
#include <Nazara/Shader/SpirvData.hpp>
#include <string>
#include <vector>

namespace Nz
{
	class NAZARA_SHADER_API SpirvSection
	{
		public:
			struct OpSize;
			struct Raw;

			SpirvSection() = default;
			SpirvSection(const SpirvSection& cache) = default;
			SpirvSection(SpirvSection&& cache) = default;
			~SpirvSection() = default;

			inline std::size_t Append(const char* str);
			inline std::size_t Append(const std::string_view& str);
			inline std::size_t Append(const std::string& str);
			inline std::size_t Append(UInt32 value);
			inline std::size_t Append(SpirvOp opcode, const OpSize& wordCount);
			std::size_t Append(const Raw& raw);
			inline std::size_t Append(std::initializer_list<UInt32> codepoints);
			template<typename... Args> std::size_t Append(SpirvOp opcode, const Args&... args);
			template<typename F> std::size_t AppendVariadic(SpirvOp opcode, F&& callback);
			template<typename T> std::size_t Append(T value);

			inline unsigned int CountWord(const char* str);
			inline unsigned int CountWord(const std::string_view& str);
			inline unsigned int CountWord(const std::string& str);
			inline unsigned int CountWord(const Raw& raw);
			template<typename T> unsigned int CountWord(const T& value);
			template<typename T1, typename T2, typename... Args> unsigned int CountWord(const T1& value, const T2& value2, const Args&... rest);

			inline const std::vector<UInt32>& GetBytecode() const;
			inline std::size_t GetOutputOffset() const;

			SpirvSection& operator=(const SpirvSection& cache) = delete;
			SpirvSection& operator=(SpirvSection&& cache) = default;

			struct OpSize
			{
				unsigned int wc;
			};

			struct Raw
			{
				const void* ptr;
				std::size_t size;
			};

			static inline UInt32 BuildOpcode(SpirvOp opcode, unsigned int wordCount);

		private:
			std::vector<UInt32> m_bytecode;
	};
}

#include <Nazara/Shader/SpirvSection.inl>

#endif