improve-PrimFillAttr

zenustech · Jan 23, 2025 · f71523e · f71523e
1 parent b4ed874
commit f71523e
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/zeno/src/nodes/neo/PrimAttrOps.cpp b/zeno/src/nodes/neo/PrimAttrOps.cpp
@@ -44,9 +44,23 @@ struct PrimFillAttr : INode {
                 std::fill(arr.begin(), arr.end(), val);
             }
         }, enum_variant<std::variant<
-            float, vec3f, int
+            float
+            , vec2f
+            , vec3f
+            , vec4f
+            , int
+            , vec2i
+            , vec3i
+            , vec4i
         >>(array_index({
-            "float", "vec3f", "int"
+            "float"
+            , "vec2f"
+            , "vec3f"
+            , "vec4f"
+            , "int"
+            , "vec2i"
+            , "vec3i"
+            , "vec4i"
         }, type)));
         set_output("prim", std::move(prim));
     }
@@ -57,7 +71,7 @@ ZENDEFNODE(PrimFillAttr, {
         {"PrimitiveObject", "prim"},
         {"enum vert tri loop poly line", "scope", "vert"},
         {"string", "attr", "rad"},
-        {"enum float vec3f int", "type", "float"},
+        {"enum float vec2f vec3f vec4f int vec2i vec3i vec4i", "type", "float"},
         {"float", "value", "0"},
     },
     {