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

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

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()