Add API to control interface specialization. (#1925)

1 files changed, 39 insertions, 0 deletions

diff --git a/tools/render-test/slang-support.cpp b/tools/render-test/slang-support.cpp
index 65f8f244b..f479218f7 100644
--- a/tools/render-test/slang-support.cpp
+++ b/tools/render-test/slang-support.cpp
@@ -235,6 +235,37 @@ void ShaderCompilerUtil::Output::reset()
         actualEntryPoints = request.entryPoints;
     }
 
+    if (request.typeConformances.getCount())
+    {
+        ComPtr<slang::ISession> session;
+        slangRequest->getSession(session.writeRef());
+        List<ComPtr<slang::ITypeConformance>> typeConformanceComponents;
+        List<slang::IComponentType*> componentsRawPtr;
+        componentsRawPtr.add(linkedSlangProgram.get());
+        auto reflection = slang::ProgramLayout::get(slangRequest);
+        ComPtr<ISlangBlob> outDiagnostic;
+        for (auto& conformance : request.typeConformances)
+        {
+            auto derivedType = reflection->findTypeByName(conformance.derivedTypeName.getBuffer());
+            auto baseType = reflection->findTypeByName(conformance.baseTypeName.getBuffer());
+            ComPtr<slang::ITypeConformance> conformanceComponentType;
+            session->createTypeConformanceComponentType(
+                derivedType,
+                baseType,
+                conformanceComponentType.writeRef(),
+                conformance.idOverride,
+                outDiagnostic.writeRef());
+            typeConformanceComponents.add(conformanceComponentType);
+            componentsRawPtr.add(conformanceComponentType);
+        }
+        ComPtr<slang::IComponentType> newProgram;
+        session->createCompositeComponentType(
+            componentsRawPtr.getBuffer(),
+            componentsRawPtr.getCount(),
+            newProgram.writeRef(),
+            outDiagnostic.writeRef());
+        linkedSlangProgram = newProgram;
+    }
     out.set(input.pipelineType, linkedSlangProgram);
     return SLANG_OK;
 }
@@ -415,6 +446,14 @@ void ShaderCompilerUtil::Output::reset()
     }
     compileRequest.globalSpecializationArgs = layout.globalSpecializationArgs;
     compileRequest.entryPointSpecializationArgs = layout.entryPointSpecializationArgs;
+    for (auto conformance : layout.typeConformances)
+    {
+        ShaderCompileRequest::TypeConformance c;
+        c.derivedTypeName = conformance.derivedTypeName;
+        c.baseTypeName = conformance.baseTypeName;
+        c.idOverride = conformance.idOverride;
+        compileRequest.typeConformances.add(c);
+    }
     return ShaderCompilerUtil::compileProgram(session, options, input, compileRequest, output.output);
 }