Run array specialization in a sperate pass. (#1449)

* Run array specialization in a sperate pass.

* rename specializeFunctionCall->specializeFunctionCalls

Co-authored-by: Tim Foley <tfoleyNV@users.noreply.github.com>

1 files changed, 53 insertions, 0 deletions

diff --git a/source/slang/slang-ir-specialize-arrays.cpp b/source/slang/slang-ir-specialize-arrays.cpp
new file mode 100644
index 000000000..53e317b81
--- /dev/null
+++ b/source/slang/slang-ir-specialize-arrays.cpp
@@ -0,0 +1,53 @@
+// slang-ir-specialize-arrays.cpp
+#include "slang-ir-specialize-arrays.h"
+
+#include "slang-ir-specialize-function-call.h"
+#include "slang-ir.h"
+#include "slang-ir-insts.h"
+
+namespace Slang
+{
+
+struct ArrayParameterSpecializationCondition : FunctionCallSpecializeCondition
+{
+    // This pass is intended to specialize functions
+    // with struct parameters that has array fields
+    // to avoid performance problems for GLSL targets.
+
+    // Returns true if `type` is an `IRStructType` with array-typed fields.
+    bool isStructTypeWithArray(IRType* type)
+    {
+        if (auto structType = as<IRStructType>(type))
+        {
+            for (auto field : structType->getFields())
+            {
+                if (auto arrayType = as<IRArrayType>(field->getFieldType()))
+                {
+                    return true;
+                }
+                if (auto subStructType = as<IRStructType>(field->getFieldType()))
+                {
+                    if (isStructTypeWithArray(subStructType))
+                        return true;
+                }
+            }
+        }
+        return false;
+    }
+
+    bool doesParamNeedSpecialization(IRParam* param)
+    {
+        return isStructTypeWithArray(param->getDataType());
+    }
+};
+
+void specializeArrayParameters(
+    BackEndCompileRequest* compileRequest,
+    TargetRequest*  targetRequest,
+    IRModule*       module)
+{
+    ArrayParameterSpecializationCondition condition;
+    specializeFunctionCalls(compileRequest, targetRequest, module, &condition);
+}
+
+} // namesapce Slang