WIP SpirvWriter

include/Nazara/Renderer/ShaderEnums.hpp

@@ -81,6 +81,15 @@ namespace Nz::ShaderNodes
 		StatementBlock
 	};
 
+	enum class SsaInstruction
+	{
+		OpAdd,
+		OpDiv,
+		OpMul,
+		OpSub,
+		OpSample
+	};
+
 	enum class SwizzleComponent
 	{
 		First,

include/Nazara/Renderer/SpirvWriter.hpp

@@ -0,0 +1,100 @@
+// Copyright (C) 2020 Jérôme Leclercq
+// This file is part of the "Nazara Engine - Renderer module"
+// For conditions of distribution and use, see copyright notice in Config.hpp
+
+#pragma once
+
+#ifndef NAZARA_SPIRVWRITER_HPP
+#define NAZARA_SPIRVWRITER_HPP
+
+#include <Nazara/Prerequisites.hpp>
+#include <Nazara/Renderer/Config.hpp>
+#include <Nazara/Renderer/ShaderAst.hpp>
+#include <Nazara/Renderer/ShaderVarVisitor.hpp>
+#include <Nazara/Renderer/ShaderVisitor.hpp>
+#include <Nazara/Renderer/ShaderWriter.hpp>
+#include <set>
+#include <string_view>
+#include <unordered_map>
+
+namespace Nz
+{
+	class NAZARA_RENDERER_API SpirvWriter : public ShaderVarVisitor, public ShaderVisitor
+	{
+		public:
+			struct Environment;
+
+			SpirvWriter();
+			SpirvWriter(const SpirvWriter&) = delete;
+			SpirvWriter(SpirvWriter&&) = delete;
+			~SpirvWriter() = default;
+
+			std::vector<UInt32> Generate(const ShaderAst& shader);
+
+			void SetEnv(Environment environment);
+
+			struct Environment
+			{
+			};
+
+		private:
+			struct Opcode;
+
+			inline void Append(const char* str);
+			void Append(const std::string_view& str);
+			void Append(const Opcode& opcode, unsigned int wordCount);
+			void Append(UInt32 codepoint);
+			void Append(std::initializer_list<UInt32> codepoints);
+			template<typename... Args> void Append(Opcode opcode, const Args&... args);
+			template<typename T> void Append(T value);
+
+			void AppendHeader();
+
+			inline unsigned int CountWord(const char* str);
+			unsigned int CountWord(const std::string_view& str);
+			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);
+
+			using ShaderVarVisitor::Visit;
+			using ShaderVisitor::Visit;
+			void Visit(const ShaderNodes::ExpressionPtr& expr, bool encloseIfRequired = false);
+			void Visit(const ShaderNodes::AccessMember& node) override;
+			void Visit(const ShaderNodes::AssignOp& node) override;
+			void Visit(const ShaderNodes::Branch& node) override;
+			void Visit(const ShaderNodes::BinaryOp& node) override;
+			void Visit(const ShaderNodes::BuiltinVariable& var) override;
+			void Visit(const ShaderNodes::Cast& node) override;
+			void Visit(const ShaderNodes::Constant& node) override;
+			void Visit(const ShaderNodes::DeclareVariable& node) override;
+			void Visit(const ShaderNodes::ExpressionStatement& node) override;
+			void Visit(const ShaderNodes::Identifier& node) override;
+			void Visit(const ShaderNodes::InputVariable& var) override;
+			void Visit(const ShaderNodes::IntrinsicCall& node) override;
+			void Visit(const ShaderNodes::LocalVariable& var) override;
+			void Visit(const ShaderNodes::ParameterVariable& var) override;
+			void Visit(const ShaderNodes::OutputVariable& var) override;
+			void Visit(const ShaderNodes::Sample2D& node) override;
+			void Visit(const ShaderNodes::StatementBlock& node) override;
+			void Visit(const ShaderNodes::SwizzleOp& node) override;
+			void Visit(const ShaderNodes::UniformVariable& var) override;
+
+			struct Context
+			{
+				const ShaderAst* shader = nullptr;
+				const ShaderAst::Function* currentFunction = nullptr;
+			};
+
+			struct State
+			{
+				std::vector<UInt32> output;
+			};
+
+			Context m_context;
+			Environment m_environment;
+			State* m_currentState;
+	};
+}
+
+#include <Nazara/Renderer/SpirvWriter.inl>
+
+#endif

include/Nazara/Renderer/SpirvWriter.inl

@@ -0,0 +1,56 @@
+// Copyright (C) 2020 Jérôme Leclercq
+// This file is part of the "Nazara Engine - Renderer module"
+// For conditions of distribution and use, see copyright notice in Config.hpp
+
+#include <Nazara/Renderer/SpirvWriter.hpp>
+#include <cassert>
+#include <type_traits>
+#include <Nazara/Renderer/Debug.hpp>
+
+namespace Nz
+{
+	inline void SpirvWriter::Append(const char* str)
+	{
+		return Append(std::string_view(str));
+	}
+
+	template<typename T>
+	void SpirvWriter::Append(T value)
+	{
+		assert(m_currentState);
+		m_currentState->output.push_back(static_cast<UInt32>(value));
+	}
+
+	template<typename ...Args>
+	inline void SpirvWriter::Append(Opcode opcode, const Args&... args)
+	{
+		unsigned int wordCount = 1 + (CountWord(args) + ... + 0);
+		Append(opcode, wordCount);
+		if constexpr (sizeof...(args) > 0)
+			(Append(args), ...);
+	}
+
+	template<typename T>
+	inline unsigned int SpirvWriter::CountWord(const T& value)
+	{
+		return 1;
+	}
+
+	template<typename T1, typename T2, typename ...Args>
+	unsigned int SpirvWriter::CountWord(const T1& value, const T2& value2, const Args&... rest)
+	{
+		return CountWord(value) + CountWord(value2) + (CountWord(rest) + ...);
+	}
+
+	inline unsigned int SpirvWriter::CountWord(const char* str)
+	{
+		return CountWord(std::string_view(str));
+	}
+
+	inline unsigned int SpirvWriter::CountWord(const std::string_view& str)
+	{
+		return (str.size() + 1 + 4 - 1) / 4; //< + 1 for null character
+	}
+}
+
+#include <Nazara/Renderer/DebugOff.hpp>