From 71e90a7ba78d0566e3b7da54df48f9af598e4cbb Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Mon, 9 Dec 2024 04:47:35 -0800
Subject: Fix reflection for pointer element types. (#5797)

* Fix reflection for pointer element types.

* Fix.

---------

Co-authored-by: Ellie Hermaszewska <ellieh@nvidia.com>
---
 tools/slang-unit-test/unit-test-ptr-layout.cpp | 54 ++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 tools/slang-unit-test/unit-test-ptr-layout.cpp

(limited to 'tools/slang-unit-test/unit-test-ptr-layout.cpp')

diff --git a/tools/slang-unit-test/unit-test-ptr-layout.cpp b/tools/slang-unit-test/unit-test-ptr-layout.cpp
new file mode 100644
index 000000000..3eb8de10f
--- /dev/null
+++ b/tools/slang-unit-test/unit-test-ptr-layout.cpp
@@ -0,0 +1,54 @@
+// unit-test-ptr-layout.cpp
+
+#include "slang-com-ptr.h"
+#include "slang.h"
+#include "unit-test/slang-unit-test.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+
+using namespace Slang;
+
+SLANG_UNIT_TEST(pointerTypeLayout)
+{
+    const char* testSource = "struct TestStruct {"
+                             "   int3 member0;"
+                             "   float member1;"
+                             "};";
+
+    ComPtr<slang::IGlobalSession> globalSession;
+    SLANG_CHECK(slang_createGlobalSession(SLANG_API_VERSION, globalSession.writeRef()) == SLANG_OK);
+    slang::TargetDesc targetDesc = {};
+    targetDesc.format = SLANG_SPIRV;
+    targetDesc.profile = globalSession->findProfile("spirv_1_5");
+    slang::SessionDesc sessionDesc = {};
+    sessionDesc.targetCount = 1;
+    sessionDesc.targets = &targetDesc;
+    sessionDesc.compilerOptionEntryCount = 1;
+    slang::CompilerOptionEntry compilerOptionEntry = {};
+    compilerOptionEntry.name = slang::CompilerOptionName::EmitSpirvViaGLSL;
+    compilerOptionEntry.value.kind = slang::CompilerOptionValueKind::Int;
+    compilerOptionEntry.value.intValue0 = 1;
+    sessionDesc.compilerOptionEntries = &compilerOptionEntry;
+
+    ComPtr<slang::ISession> session;
+    SLANG_CHECK(globalSession->createSession(sessionDesc, session.writeRef()) == SLANG_OK);
+
+    ComPtr<slang::IBlob> diagnosticBlob;
+    auto module =
+        session->loadModuleFromSourceString("m", "m.slang", testSource, diagnosticBlob.writeRef());
+    SLANG_CHECK(module != nullptr);
+
+
+    auto testBody = [&]()
+    {
+        auto reflection = module->getLayout();
+        auto testStruct = reflection->findTypeByName("Ptr<TestStruct>");
+        auto ptrLayout = reflection->getTypeLayout(testStruct);
+        auto valueLayout = ptrLayout->getElementTypeLayout();
+        SLANG_CHECK_ABORT(valueLayout->getFieldCount() == 2);
+        SLANG_CHECK_ABORT(valueLayout->getFieldByIndex(1)->getOffset() == 12);
+    };
+
+    testBody();
+}
-- 
cgit v1.2.3