This commit is contained in:
Lynix
2020-08-02 20:42:51 +02:00
parent 10860ed562
commit 50bd150345
8 changed files with 3626 additions and 113 deletions

View File

@@ -13,13 +13,13 @@
#include <Nazara/Renderer/ShaderVarVisitor.hpp>
#include <Nazara/Renderer/ShaderVisitor.hpp>
#include <Nazara/Renderer/ShaderWriter.hpp>
#include <set>
#include <string>
#include <string_view>
#include <unordered_map>
namespace Nz
{
class NAZARA_RENDERER_API SpirvWriter : public ShaderVarVisitor, public ShaderVisitor
class NAZARA_RENDERER_API SpirvWriter : public ShaderVisitor
{
public:
struct Environment;
@@ -35,48 +35,54 @@ namespace Nz
struct Environment
{
UInt32 spvMajorVersion = 1;
UInt32 spvMinorVersion = 0;
};
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);
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);
std::size_t Append(UInt32 value);
std::size_t Append(const Opcode& opcode, unsigned int wordCount);
inline std::size_t Append(std::initializer_list<UInt32> codepoints);
template<typename... Args> std::size_t Append(Opcode opcode, const Args&... args);
template<typename T> std::size_t Append(T value);
UInt32 AllocateResultId();
void AppendHeader();
void AppendTypes();
inline unsigned int CountWord(const char* str);
unsigned int CountWord(const std::string_view& str);
inline unsigned int CountWord(const std::string& 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;
std::size_t GetOutputOffset() const;
UInt32 ProcessType(ShaderExpressionType type);
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;
static void MergeBlocks(std::vector<UInt32>& output, const std::vector<UInt32>& from);
struct Context
{
@@ -84,10 +90,7 @@ namespace Nz
const ShaderAst::Function* currentFunction = nullptr;
};
struct State
{
std::vector<UInt32> output;
};
struct State;
Context m_context;
Environment m_environment;