yum-mirror/slang

Making it easier to work with shaders

git clone https://git.yummers.dev/yum-mirror/slang

tareksanderReflection compiler option (#5507)7e51180ed

master
branchesmasterall refs →
2.5 KiB91 linesraw
1// slang-reflection-test-main.cpp
2
3#include "../../source/compiler-core/slang-pretty-writer.h"
4#include "../../source/core/slang-char-util.h"
5#include "../../source/core/slang-string-escape-util.h"
6#include "../../source/core/slang-string-util.h"
7#include "../../source/core/slang-test-tool-util.h"
8#include "slang-com-helper.h"
9#include "slang.h"
10
11#include <assert.h>
12#include <stdio.h>
13#include <stdlib.h>
14#include <string.h>
15
16using namespace Slang;
17
18void emitReflectionJSON(SlangCompileRequest* request, SlangReflection* reflection)
19{
20    ComPtr<ISlangBlob> b;
21
22    spReflection_ToJson(reflection, request, b.writeRef());
23
24    // Output the writer content to out stream
25    StdWriters::getOut().write((const char*)b->getBufferPointer(), b->getBufferSize());
26}
27
28static SlangResult maybeDumpDiagnostic(SlangResult res, SlangCompileRequest* request)
29{
30    const char* diagnostic;
31    if (SLANG_FAILED(res) && (diagnostic = spGetDiagnosticOutput(request)))
32    {
33        StdWriters::getError().put(diagnostic);
34    }
35    return res;
36}
37
38SlangResult performCompilationAndReflection(
39    SlangCompileRequest* request,
40    int argc,
41    const char* const* argv)
42{
43    SLANG_RETURN_ON_FAIL(
44        maybeDumpDiagnostic(spProcessCommandLineArguments(request, &argv[1], argc - 1), request));
45    SLANG_RETURN_ON_FAIL(maybeDumpDiagnostic(spCompile(request), request));
46
47    // Okay, let's go through and emit reflection info on whatever
48    // we have.
49
50    SlangReflection* reflection = spGetReflection(request);
51    emitReflectionJSON(request, reflection);
52
53    return SLANG_OK;
54}
55
56SLANG_TEST_TOOL_API SlangResult
57innerMain(Slang::StdWriters* stdWriters, SlangSession* session, int argc, const char* const* argv)
58{
59    Slang::StdWriters::setSingleton(stdWriters);
60
61    SlangCompileRequest* request = spCreateCompileRequest(session);
62    for (int i = 0; i < SLANG_WRITER_CHANNEL_COUNT_OF; ++i)
63    {
64        const auto channel = SlangWriterChannel(i);
65        spSetWriter(request, channel, stdWriters->getWriter(channel));
66    }
67
68    char const* appName = "slang-reflection-test";
69    if (argc > 0)
70        appName = argv[0];
71
72    SlangResult res = performCompilationAndReflection(request, argc, argv);
73
74    spDestroyCompileRequest(request);
75
76    return res;
77}
78
79int main(int argc, char** argv)
80{
81    using namespace Slang;
82
83    SlangSession* session = spCreateSession(nullptr);
84
85    auto stdWriters = StdWriters::initDefaultSingleton();
86
87    SlangResult res = innerMain(stdWriters, session, argc, argv);
88    spDestroySession(session);
89    slang::shutdown();
90    return SLANG_FAILED(res) ? 1 : 0;
91}