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
129
130
|
// main.cpp
#include "../../slang.h"
SLANG_API void spSetCommandLineCompilerMode(SlangCompileRequest* request);
#include "../core/slang-io.h"
using namespace Slang;
#include <assert.h>
// Try to read an argument for a command-line option.
static void diagnosticCallback(
char const* message,
void* /*userData*/)
{
fputs(message, stderr);
fflush(stderr);
}
#ifdef _WIN32
#define MAIN slangc_main
#else
#define MAIN main
#endif
// Used to identify that compilation was the failure - with a unique 'internal' code
#define SLANG_E_INTERNAL_COMPILE_FAILED SLANG_MAKE_ERROR(SLANG_FACILITY_INTERNAL, 0x7fab)
static SlangResult innerMain(int argc, char** argv)
{
// Parse any command-line options
SlangSession* session = spCreateSession(nullptr);
SlangCompileRequest* compileRequest = spCreateCompileRequest(session);
spSetDiagnosticCallback(
compileRequest,
&diagnosticCallback,
nullptr);
spSetCommandLineCompilerMode(compileRequest);
char const* appName = "slangc";
if (argc > 0) appName = argv[0];
{
const SlangResult res = spProcessCommandLineArguments(compileRequest, &argv[1], argc - 1);
if (SLANG_FAILED(res))
{
// TODO: print usage message
return res;
}
}
#ifndef _DEBUG
try
#endif
{
// Run the compiler (this will produce any diagnostics through
// our callback above).
if (SLANG_FAILED(spCompile(compileRequest)))
{
// If the compilation failed, then get out of here...
// Turn into an internal Result -> such that return code can be used to vary result to match previous behavior
return SLANG_E_INTERNAL_COMPILE_FAILED;
}
// Now that we are done, clean up after ourselves
spDestroyCompileRequest(compileRequest);
spDestroySession(session);
}
#ifndef _DEBUG
catch (Exception & e)
{
printf("internal compiler error: %S\n", e.Message.ToWString().begin());
return SLANG_FAIL;
}
#endif
return SLANG_OK;
}
int MAIN(int argc, char** argv)
{
SlangResult res = innerMain(argc, argv);
if (SLANG_SUCCEEDED(res))
{
return 0;
}
else if (res == SLANG_E_INTERNAL_COMPILE_FAILED)
{
return -1;
}
return 1;
}
#ifdef _WIN32
int wmain(int argc, wchar_t** argv)
{
int result = 0;
{
// Conver the wide-character Unicode arguments to UTF-8,
// since that is what Slang expects on the API side.
List<String> args;
for(int ii = 0; ii < argc; ++ii)
{
args.Add(String::FromWString(argv[ii]));
}
List<char const*> argBuffers;
for(int ii = 0; ii < argc; ++ii)
{
argBuffers.Add(args[ii].Buffer());
}
result = MAIN(argc, (char**) &argBuffers[0]);
}
#ifdef _MSC_VER
_CrtDumpMemoryLeaks();
#endif
return result;
}
#endif
|
