Add @shader attribute for functions (enable pseudocode)

Const vectors less than four values are padded with nan instead of 0
gideros · Oct 30, 2024 · 1fd3297 · 1fd3297
1 parent 4df1eaf
commit 1fd3297
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 4 deletions.
diff --git a/Ast/include/Luau/Ast.h b/Ast/include/Luau/Ast.h
@@ -208,6 +208,7 @@ class AstAttr : public AstNode
     {
         Checked,
         Native,
+        Shader,
     };
 
     AstAttr(const Location& location, Type type);

diff --git a/Ast/include/Luau/Lexer.h b/Ast/include/Luau/Lexer.h
@@ -79,7 +79,6 @@ struct Lexeme
         PowAssign,
         ConcatAssign,
 //GIDEROS ADDED
-        DivInt,
         MaxOf,
         MinOf,
         AngToRad,

diff --git a/Ast/src/Parser.cpp b/Ast/src/Parser.cpp
@@ -34,7 +34,11 @@ struct AttributeEntry
     AstAttr::Type type;
 };
 
-AttributeEntry kAttributeEntries[] = {{"@checked", AstAttr::Type::Checked}, {"@native", AstAttr::Type::Native}, {nullptr, AstAttr::Type::Checked}};
+AttributeEntry kAttributeEntries[] = {
+    {"@checked", AstAttr::Type::Checked},
+    {"@native", AstAttr::Type::Native},
+    {"@shader", AstAttr::Type::Shader},
+    {nullptr, AstAttr::Type::Checked}};
 
 ParseError::ParseError(const Location& location, const std::string& message)
     : location(location)

diff --git a/Compiler/src/Compiler.cpp b/Compiler/src/Compiler.cpp
@@ -214,11 +214,17 @@ struct Compiler
         bool self = func->self != 0;
         uint32_t fid = bytecode.beginFunction(uint8_t(self + func->args.size), func->vararg);
 
-        if (func->returnAnnotation) {
+        bool isShader=false;
+        for (auto it:func->attributes) {
+            if (it->type==AstAttr::Shader) isShader=true;
+        }
+        if ((!isShader)&&(func->returnAnnotation)) {
             AstTypeReference *returnType=(AstTypeReference *)func->returnAnnotation->types.data[0];
             if (returnType->name=="Shader")
-                bytecode.setPseudoCode(generatePseudoCode(func));
+                isShader=true;
         }
+        if (isShader)
+            bytecode.setPseudoCode(generatePseudoCode(func));
 
         setDebugLine(func);