From 9f89c3b2de80c49222c4a6b48e845894c4256a5d Mon Sep 17 00:00:00 2001
From: Yong He <yongh@outlook.com>
Date: Wed, 3 Jan 2018 12:46:23 -0800
Subject: Add API for querying TypeLayout from a Type

Added two API functions:
1. `spReflection_FindTypeByName`, which returns a DeclRefType to the struct type with the given name. The function finds from all loaded modules in a `CompileRequest` for a decl with the given name, construct a `Type` object and cache it in `CompileRequest::types` dictionary. The subsequent calls to `spReflection_FindTypeByName` with the same name will simply returned the cached Type objects.
2. `spReflection_GetTypeLayout`, which returns a `TypeLayout` for a given `Type`. This function creates and caches the `TypeLayout` in the `TargetRequest` object that is used to create the `ProgramLayout`.
---
 source/slang/reflection.cpp | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

(limited to 'source/slang/reflection.cpp')

diff --git a/source/slang/reflection.cpp b/source/slang/reflection.cpp
index 5270df8b4..0558ff773 100644
--- a/source/slang/reflection.cpp
+++ b/source/slang/reflection.cpp
@@ -428,6 +428,47 @@ SLANG_API char const* spReflectionType_GetName(SlangReflectionType* inType)
     return nullptr;
 }
 
+SLANG_API SlangReflectionType * spReflection_FindTypeByName(SlangReflection * reflection, char const * name)
+{
+    auto context = convert(reflection);
+    auto compileRequest = context->targetRequest->compileRequest;
+
+    RefPtr<Type> result;
+    if (compileRequest->types.TryGetValue(name, result))
+        return (SlangReflectionType*)result.Ptr();
+
+    Decl* resultDecl = nullptr;
+    for (auto module : compileRequest->loadedModulesList)
+    {
+        auto nameObj = compileRequest->getNamePool()->getName(name);
+        if (module->moduleDecl->memberDictionary.TryGetValue(nameObj, resultDecl))
+            break;
+    }
+    if (resultDecl)
+    {
+        RefPtr<DeclRefType> declRefType = new DeclRefType();
+        declRefType->declRef.decl = resultDecl;
+        compileRequest->types[name] = declRefType;
+        return (SlangReflectionType*)declRefType.Ptr();
+    }
+    return nullptr;
+}
+
+SLANG_API SlangReflectionTypeLayout* spReflection_GetTypeLayout(
+    SlangReflection* reflection,
+    SlangReflectionType* inType, 
+    SlangLayoutRules /*rules*/)
+{
+    auto context = convert(reflection);
+    auto type = convert(inType);
+    auto layoutContext = getInitialLayoutContextForTarget(context->targetRequest);
+    RefPtr<TypeLayout> result;
+    if (context->targetRequest->typeLayouts.TryGetValue(type, result))
+        return (SlangReflectionTypeLayout*)result.Ptr();
+    result = CreateTypeLayout(layoutContext, type);
+    context->targetRequest->typeLayouts[type] = result;
+    return (SlangReflectionTypeLayout*)result.Ptr();
+}
 
 SLANG_API SlangReflectionType* spReflectionType_GetResourceResultType(SlangReflectionType* inType)
 {
-- 
cgit v1.2.3