sid/NazaraEngine
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Shader: Add export attribute for structs ([export])

Browse Source
This commit is contained in:
Jérôme Leclercq
2022-03-04 18:26:18 +01:00
parent a2f4f3c802
commit ca83f363a3
6 changed files with 40 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/Nazara/Shader/Ast/AstCloner.cpp
View File

@@ -148,6 +148,7 @@ namespace Nz::ShaderAst
{
auto clone = std::make_unique<DeclareStructStatement>();
clone->structIndex = node.structIndex;
clone->isExported = Clone(node.isExported);
clone->description.layout = Clone(node.description.layout);
clone->description.name = node.description.name;

3
src/Nazara/Shader/Ast/AstSerializer.cpp
View File

@@ -3,9 +3,9 @@
// For conditions of distribution and use, see copyright notice in Config.hpp
#include <Nazara/Shader/Ast/AstSerializer.hpp>
#include <Nazara/Shader/ShaderBuilder.hpp>
#include <Nazara/Shader/Ast/AstExpressionVisitor.hpp>
#include <Nazara/Shader/Ast/AstStatementVisitor.hpp>
#include <Nazara/Shader/ShaderBuilder.hpp>
#include <Nazara/Shader/Debug.hpp>
namespace Nz::ShaderAst
@@ -249,6 +249,7 @@ namespace Nz::ShaderAst
void AstSerializerBase::Serialize(DeclareStructStatement& node)
{
OptVal(node.structIndex);
ExprValue(node.isExported);
Value(node.description.name);
ExprValue(node.description.layout);

3
src/Nazara/Shader/Ast/SanitizeVisitor.cpp
View File

@@ -939,6 +939,9 @@ namespace Nz::ShaderAst
auto clone = static_unique_pointer_cast<DeclareStructStatement>(AstCloner::Clone(node));
if (clone->isExported.HasValue())
clone->isExported = ComputeExprValue(clone->isExported);
std::unordered_set<std::string> declaredMembers;
for (auto& member : clone->description.members)
{

32
src/Nazara/Shader/ShaderLangParser.cpp
View File

@@ -28,6 +28,7 @@ namespace Nz::ShaderLang
{ "depth_write", ShaderAst::AttributeType::DepthWrite },
{ "early_fragment_tests", ShaderAst::AttributeType::EarlyFragmentTests },
{ "entry", ShaderAst::AttributeType::Entry },
{ "export", ShaderAst::AttributeType::Export },
{ "layout", ShaderAst::AttributeType::Layout },
{ "location", ShaderAst::AttributeType::Location },
{ "nzsl_version", ShaderAst::AttributeType::LangVersion },
@@ -66,15 +67,27 @@ namespace Nz::ShaderLang
}
template<typename T>
void HandleUniqueAttribute(const std::string_view& attributeName, ShaderAst::ExpressionValue<T>& targetAttribute, ShaderAst::ExprValue::Param&& param, bool requireValue = true)
void HandleUniqueAttribute(const std::string_view& attributeName, ShaderAst::ExpressionValue<T>& targetAttribute, ShaderAst::ExprValue::Param&& param)
{
if (targetAttribute.HasValue())
throw AttributeError{ "attribute " + std::string(attributeName) + " must be present once" };
if (!param && requireValue)
if (param)
targetAttribute = std::move(*param);
else
throw AttributeError{ "attribute " + std::string(attributeName) + " requires a parameter" };
}
targetAttribute = std::move(*param);
template<typename T>
void HandleUniqueAttribute(const std::string_view& attributeName, ShaderAst::ExpressionValue<T>& targetAttribute, ShaderAst::ExprValue::Param&& param, T defaultValue)
{
if (targetAttribute.HasValue())
throw AttributeError{ "attribute " + std::string(attributeName) + " must be present once" };
if (param)
targetAttribute = std::move(*param);
else
targetAttribute = std::move(defaultValue);
}
template<typename T>
@@ -122,7 +135,7 @@ namespace Nz::ShaderLang
{
ShaderAst::StatementPtr statement = ParseRootStatement();
if (!m_context->module)
throw UnexpectedToken{ "unexpected token before module declaration" };
throw UnexpectedToken{}; //< "unexpected token before module declaration"
if (!statement)
break;
@@ -611,7 +624,7 @@ namespace Nz::ShaderLang
break;
case ShaderAst::AttributeType::EarlyFragmentTests:
HandleUniqueAttribute("early_fragment_tests", func->earlyFragmentTests, std::move(arg), false);
HandleUniqueAttribute("early_fragment_tests", func->earlyFragmentTests, std::move(arg), true);
break;
default:
@@ -836,6 +849,7 @@ namespace Nz::ShaderLang
description.name = ParseIdentifierAsName();
ShaderAst::ExpressionValue<bool> condition;
ShaderAst::ExpressionValue<bool> exported;
for (auto&& [attributeType, attributeParam] : attributes)
{
@@ -845,6 +859,10 @@ namespace Nz::ShaderLang
HandleUniqueAttribute("cond", condition, std::move(attributeParam));
break;
case ShaderAst::AttributeType::Export:
HandleUniqueAttribute("export", exported, std::move(attributeParam), true);
break;
case ShaderAst::AttributeType::Layout:
HandleUniqueStringAttribute("layout", s_layoutMapping, description.layout, std::move(attributeParam));
break;
@@ -917,9 +935,9 @@ namespace Nz::ShaderLang
Expect(Advance(), TokenType::ClosingCurlyBracket);
if (condition.HasValue())
return ShaderBuilder::ConditionalStatement(std::move(condition).GetExpression(), ShaderBuilder::DeclareStruct(std::move(description)));
return ShaderBuilder::ConditionalStatement(std::move(condition).GetExpression(), ShaderBuilder::DeclareStruct(std::move(description), std::move(exported)));
else
return ShaderBuilder::DeclareStruct(std::move(description));
return ShaderBuilder::DeclareStruct(std::move(description), std::move(exported));
}
ShaderAst::ExpressionPtr Parser::ParseVariableAssignation()