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
121
122
123
124
125
126
127
128
|
#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 TestToolUtil::getIncludePath(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& rootPath, slang::IGlobalSession* session)
{
String includePath;
SLANG_RETURN_ON_FAIL(TestToolUtil::getIncludePath(rootPath, "prelude/slang-cpp-prelude.h", includePath));
StringBuilder prelude;
prelude << "#include \"" << includePath << "\"\n\n";
session->setLanguagePrelude(SLANG_SOURCE_LANGUAGE_CPP, prelude.getBuffer());
return SLANG_OK;
}
static SlangResult _addCUDAPrelude(const String& rootPath, slang::IGlobalSession* session)
{
String includePath;
SLANG_RETURN_ON_FAIL(TestToolUtil::getIncludePath(rootPath, "prelude/slang-cuda-prelude.h", includePath));
StringBuilder prelude;
prelude << "#include \"" << includePath << "\"\n\n";
session->setLanguagePrelude(SLANG_SOURCE_LANGUAGE_CUDA, prelude.getBuffer());
return SLANG_OK;
}
/* static */SlangResult TestToolUtil::getExeDirectoryPath(const char* exePath, String& outExeDirectoryPath)
{
String canonicalPath;
SLANG_RETURN_ON_FAIL(Path::getCanonical(exePath, canonicalPath));
// Get the directory
outExeDirectoryPath = Path::getParentDirectory(canonicalPath);
return SLANG_OK;
}
/* static */SlangResult TestToolUtil::getRootPath(const char* inExePath, String& outExePath)
{
// Get the directory holding the exe
String parentPath;
SLANG_RETURN_ON_FAIL(getExeDirectoryPath(inExePath, parentPath));
// From directory to the root is ../../..
// Work out the relative path to the root
String rootRelPath = Path::combine(parentPath, "../../../");
// We want the absolute path to the root
SLANG_RETURN_ON_FAIL(Path::getCanonical(rootRelPath, outExePath));
return SLANG_OK;
}
/* static */SlangResult TestToolUtil::setSessionDefaultPreludeFromExePath(const char* inExePath, slang::IGlobalSession* session)
{
String rootPath;
SLANG_RETURN_ON_FAIL(getRootPath(inExePath, rootPath));
SLANG_RETURN_ON_FAIL(setSessionDefaultPreludeFromRootPath(rootPath, session));
return SLANG_OK;
}
/* static */SlangResult TestToolUtil::setSessionDefaultPreludeFromRootPath(const String& rootPath, slang::IGlobalSession* session)
{
// Set the prelude to a path
if (SLANG_FAILED(_addCPPPrelude(rootPath, session)))
{
SLANG_ASSERT(!"Couldn't find the C++ prelude relative to the executable");
}
if (SLANG_FAILED(_addCUDAPrelude(rootPath, session)))
{
SLANG_ASSERT(!"Couldn't find the CUDA prelude relative to the executable");
}
return SLANG_OK;
}
}
|
