summaryrefslogtreecommitdiffstats
path: root/tools/slang-unit-test/unit-test-geometry-shader.cpp
blob: e1126673ed754bc52d5073fe67a6e0bc972aa1f9 (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
// unit-test-geometry-shader.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;

// Test the compilation API for compiling geometry shaders to DXIL.

#if SLANG_WINDOWS_FAMILY

SLANG_UNIT_TEST(geometryShader)
{
    const char* userSourceBody = R"(
        struct GS_INPUT
        {
            float4 PosSS : TEXTURE0;     // [Screen Space] Position
        };

        struct PS_INPUT
        {
            float4 PosSS : SV_POSITION;  // [Screen Space] Position
        };

        [maxvertexcount(3)] 
        void main(triangle GS_INPUT input[3], inout TriangleStream<PS_INPUT> outStream)
        {
            PS_INPUT output;

            output.PosSS = input[0].PosSS;
            outStream.Append(output);
     
            output.PosSS = input[1].PosSS;
            outStream.Append(output);

            output.PosSS = input[2].PosSS;
            outStream.Append(output);

            outStream.RestartStrip();
        }
        )";
    ComPtr<slang::IGlobalSession> globalSession;
    SlangGlobalSessionDesc globalDesc = {};
    globalDesc.enableGLSL = true;
    SLANG_CHECK(slang_createGlobalSession2(&globalDesc, globalSession.writeRef()) == SLANG_OK);
    slang::TargetDesc targetDesc = {};
    targetDesc.format = SLANG_DXIL;
    targetDesc.profile = globalSession->findProfile("sm_6_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);

    ComPtr<slang::IEntryPoint> entryPoint;
    module->findAndCheckEntryPoint(
        "main",
        SLANG_STAGE_GEOMETRY,
        entryPoint.writeRef(),
        diagnosticBlob.writeRef());

    slang::IComponentType* componentTypes[2] = {module, entryPoint.get()};
    ComPtr<slang::IComponentType> composedProgram;
    session->createCompositeComponentType(
        componentTypes,
        2,
        composedProgram.writeRef(),
        diagnosticBlob.writeRef());

    ComPtr<slang::IComponentType> linkedProgram;
    composedProgram->link(linkedProgram.writeRef(), diagnosticBlob.writeRef());

    ComPtr<slang::IBlob> code;
    linkedProgram->getEntryPointCode(0, 0, code.writeRef(), diagnosticBlob.writeRef());

    SLANG_CHECK(code != nullptr);
}

#endif