Commit current work

include/Nazara/Renderer/GlslWriter.hpp

@@ -0,0 +1,73 @@
+// Copyright (C) 2016 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_GLSLWRITER_HPP
+#define NAZARA_GLSLWRITER_HPP
+
+#include <Nazara/Prerequesites.hpp>
+#include <Nazara/Core/StringStream.hpp>
+#include <Nazara/Renderer/Config.hpp>
+#include <Nazara/Renderer/ShaderWriter.hpp>
+#include <map>
+#include <set>
+#include <unordered_map>
+
+namespace Nz
+{
+	class NAZARA_RENDERER_API GlslWriter : public ShaderWriter
+	{
+		public:
+			GlslWriter();
+			GlslWriter(const GlslWriter&) = delete;
+			GlslWriter(GlslWriter&&) = delete;
+			~GlslWriter() = default;
+
+			Nz::String Generate(const ShaderAst::StatementPtr& node) override;
+
+			void RegisterFunction(const String& name, ShaderAst::StatementPtr statement, std::initializer_list<ShaderAst::VariablePtr> parameters, ShaderAst::ExpressionType ret) override;
+			void RegisterVariable(const String& name, ShaderAst::ExpressionType type) override;
+
+			void Write(const ShaderAst::AssignOp& node) override;
+			void Write(const ShaderAst::Branch& node) override;
+			void Write(const ShaderAst::BinaryOp& node) override;
+			void Write(const ShaderAst::Constant& node) override;
+			void Write(const ShaderAst::ExpressionStatement& node) override;
+			void Write(const ShaderAst::NodePtr& node) override;
+			void Write(const ShaderAst::StatementBlock& node) override;
+			void Write(const ShaderAst::Variable& node) override;
+
+		private:
+			struct Function;
+
+			void Append(ShaderAst::ExpressionType type);
+			void Append(const String& txt);
+			void AppendFunction(const Function& func);
+			void AppendLine(const Nz::String& txt = Nz::String());
+
+			void EnterScope();
+			void LeaveScope();
+
+			struct Function
+			{
+				std::vector<ShaderAst::VariablePtr> parameters;
+				ShaderAst::ExpressionType retType;
+				ShaderAst::StatementPtr node;
+				String name;
+			};
+
+			struct State
+			{
+				std::set<std::pair<ShaderAst::ExpressionType, String>> m_variables;
+				StringStream stream;
+				unsigned int indentLevel = 0;
+			};
+
+			std::unordered_map<String, Function> m_functions;
+			State* m_currentState;
+	};
+}
+
+#endif // NAZARA_GLSLWRITER_HPP