evaluator: Added fixed_size attribute

WerWolv · Jul 9, 2024 · 8909b96 · 8909b96
1 parent 6da1493
commit 8909b96
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lib/source/pl/core/ast/ast_node_attribute.cpp b/lib/source/pl/core/ast/ast_node_attribute.cpp
@@ -297,6 +297,18 @@ namespace pl::core::ast {
             pattern->setSealed(true);
         }
 
+        if (const auto &arguments = attributable->getAttributeArguments("fixed_size"); arguments.size() == 1) {
+            auto requestedSize = getAttributeValueAsInteger(arguments.front(), evaluator);
+            auto actualSize = pattern->getSize();
+            if (actualSize < requestedSize) {
+                pattern->setSize(requestedSize);
+                evaluator->setReadOffset(evaluator->getReadOffset() + (requestedSize - actualSize));
+            }
+            else if (actualSize > requestedSize)
+                err::E0004.throwError("Type size larger than expected", fmt::format("Pattern of type {} is larger than expected. Expected size {}, got {}", pattern->getTypeName(), requestedSize, actualSize), node->getLocation());
+
+        }
+
         if (!pattern->hasOverriddenColor()) {
             if (const auto &arguments = attributable->getAttributeArguments("color"); arguments.size() == 1) {
                 auto colorString = getAttributeValueAsString(arguments.front(), evaluator);