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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#include "slang-test-tool-util.h"
#include "../../slang-com-helper.h"
#include "slang-io.h"
#include "slang-string-util.h"
namespace Slang
{
/* static */ToolReturnCode TestToolUtil::getReturnCode(SlangResult res)
{
switch (res)
{
case SLANG_OK: return ToolReturnCode::Success;
case SLANG_E_INTERNAL_FAIL: return ToolReturnCode::CompilationFailed;
case SLANG_FAIL: return ToolReturnCode::Failed;
case SLANG_E_NOT_AVAILABLE: return ToolReturnCode::Ignored;
default:
{
return (SLANG_SUCCEEDED(res)) ? ToolReturnCode::Success : ToolReturnCode::Failed;
}
}
}
/* static */ToolReturnCode TestToolUtil::getReturnCodeFromInt(int code)
{
if (code >= int(ToolReturnCodeSpan::First) && code <= int(ToolReturnCodeSpan::Last))
{
return ToolReturnCode(code);
}
else
{
SLANG_ASSERT(!"Invalid integral code");
return ToolReturnCode::Failed;
}
}
static SlangResult _calcIncludePath(const String& parentPath, const char* path, String& outIncludePath)
{
String includePath;
SLANG_RETURN_ON_FAIL(Path::getCanonical(Path::combine(parentPath, path), includePath));
// Use forward slashes, to avoid escaping the path
includePath = StringUtil::calcCharReplaced(includePath, '\\', '/');
// It must exist!
if (!File::exists(includePath))
{
return SLANG_FAIL;
}
outIncludePath = includePath;
return SLANG_OK;
}
static SlangResult _addCPPPrelude(const String& parentPath, slang::IGlobalSession* session)
{
String includePath;
SLANG_RETURN_ON_FAIL(_calcIncludePath(parentPath, "../../../prelude/slang-cpp-prelude.h", includePath));
StringBuilder prelude;
prelude << "#include \"" << includePath << "\"\n\n";
const SlangPassThrough downstreamCompilers[] = {
SLANG_PASS_THROUGH_CLANG, ///< Clang C/C++ compiler
SLANG_PASS_THROUGH_VISUAL_STUDIO, ///< Visual studio C/C++ compiler
SLANG_PASS_THROUGH_GCC, ///< GCC C/C++ compiler
SLANG_PASS_THROUGH_GENERIC_C_CPP,
};
for (auto downstreamCompiler : downstreamCompilers)
{
session->setDownstreamCompilerPrelude(downstreamCompiler, prelude.getBuffer());
}
return SLANG_OK;
}
static SlangResult _addCUDAPrelude(const String& parentPath, slang::IGlobalSession* session)
{
String includePath;
SLANG_RETURN_ON_FAIL(_calcIncludePath(parentPath, "../../../prelude/slang-cuda-prelude.h", includePath));
StringBuilder prelude;
prelude << "#include \"" << includePath << "\"\n\n";
const SlangPassThrough downstreamCompilers[] = {
SLANG_PASS_THROUGH_NVRTC, ///< nvrtc CUDA compiler
};
for (auto downstreamCompiler : downstreamCompilers)
{
session->setDownstreamCompilerPrelude(downstreamCompiler, prelude.getBuffer());
}
return SLANG_OK;
}
/* static */SlangResult TestToolUtil::setSessionDefaultPrelude(const char* exePath, slang::IGlobalSession* session)
{
// Set the prelude to a path
String canonicalPath;
if (SLANG_SUCCEEDED(Path::getCanonical(exePath, canonicalPath)))
{
// Get the directory
String parentPath = Path::getParentDirectory(canonicalPath);
if (SLANG_FAILED(_addCPPPrelude(parentPath, session)))
{
SLANG_ASSERT(!"Couldn't find the C++ prelude relative to the executable");
}
if (SLANG_FAILED(_addCUDAPrelude(parentPath, session)))
{
SLANG_ASSERT(!"Couldn't find the CUDA prelude relative to the executable");
}
}
return SLANG_OK;
}
}
|
