blob: c88ca77608f245e43cbd057c8bd31c06d117fdd5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// unit-test-function-lookup-resolution.cpp
#include "../../source/core/slang-io.h"
#include "../../source/core/slang-process.h"
#include "slang-com-ptr.h"
#include "slang.h"
#include "unit-test/slang-unit-test.h"
#include <stdio.h>
#include <stdlib.h>
using namespace Slang;
static String getTypeFullName(slang::TypeReflection* type)
{
ComPtr<ISlangBlob> blob;
type->getFullName(blob.writeRef());
return String((const char*)blob->getBufferPointer());
}
// Test that the reflection API provides correctly resolved lookup results.
SLANG_UNIT_TEST(functionLookupResolution)
{
// Source for a module that contains an undecorated entrypoint.
const char* userSourceBody = R"(
public interface IBase
{
public void step(inout float f);
public void method(int x) {}
}
public struct Impl : IBase
{
public void step(inout float f)
{
f += 1.0f;
}
public override void method(int x) {}
}
public extension<T : IBase> T {
public void method(int x) {}
}
)";
auto moduleName = "moduleG" + String(Process::getId());
String userSource = "import " + moduleName + ";\n" + userSourceBody;
ComPtr<slang::IGlobalSession> globalSession;
SLANG_CHECK(slang_createGlobalSession(SLANG_API_VERSION, globalSession.writeRef()) == SLANG_OK);
slang::TargetDesc targetDesc = {};
targetDesc.format = SLANG_HLSL;
targetDesc.profile = globalSession->findProfile("sm_5_0");
slang::SessionDesc sessionDesc = {};
sessionDesc.targetCount = 1;
sessionDesc.targets = &targetDesc;
ComPtr<slang::ISession> session;
SLANG_CHECK(globalSession->createSession(sessionDesc, session.writeRef()) == SLANG_OK);
ComPtr<slang::IBlob> diagnosticBlob;
auto module = session->loadModuleFromSourceString(
"m",
"m.slang",
userSourceBody,
diagnosticBlob.writeRef());
SLANG_CHECK(module != nullptr);
auto layout = module->getLayout();
auto type = layout->findTypeByName("Impl");
SLANG_CHECK_ABORT(type != nullptr);
auto func = layout->findFunctionByNameInType(type, "step");
SLANG_CHECK_ABORT(func && !func->isOverloaded());
auto func1 = layout->findFunctionByNameInType(type, "method");
SLANG_CHECK_ABORT(func1->isOverloaded());
SLANG_CHECK(func1->getOverloadCount() == 3);
// Test that overloaded function containers return the correct name
SLANG_CHECK(func1->getName() != nullptr);
SLANG_CHECK(String(func1->getName()) == "method");
if (func1->isOverloaded())
{
List<slang::FunctionReflection*> candidates;
for (uint32_t i = 0; i < func1->getOverloadCount(); i++)
{
candidates.add(func1->getOverload(i));
}
func1 = layout->tryResolveOverloadedFunction(
(uint32_t)candidates.getCount(),
candidates.getBuffer());
}
SLANG_CHECK(!func1->isOverloaded());
SLANG_CHECK(String(func1->getName()) == "method");
}
|
