yum-mirror/glslang

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

Arcady Goldmints-OrlovAdd SPIRV/SpvTools.h back to the public headers7e896697

main
branchesmainall refs →
4.2 KiB122 linesraw
1/**
2    This code is based on the glslang_c_interface implementation by Viktor Latypov
3**/
4
5/**
6BSD 2-Clause License
7
8Copyright (c) 2019, Viktor Latypov
9All rights reserved.
10
11Redistribution and use in source and binary forms, with or without
12modification, are permitted provided that the following conditions are met:
13
141. Redistributions of source code must retain the above copyright notice, this
15   list of conditions and the following disclaimer.
16
172. Redistributions in binary form must reproduce the above copyright notice,
18   this list of conditions and the following disclaimer in the documentation
19   and/or other materials provided with the distribution.
20
21THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
24DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
25FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
27SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
28CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
29OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31**/
32
33#include "glslang/Include/glslang_c_interface.h"
34
35#include <cstring>
36#include "glslang/Public/ShaderLang.h"
37#include "SPIRV/GlslangToSpv.h"
38#include "SPIRV/Logger.h"
39#include "SPIRV/SpvTools.h"
40
41static_assert(sizeof(glslang_spv_options_t) == sizeof(glslang::SpvOptions), "");
42
43typedef struct glslang_program_s {
44    glslang::TProgram* program;
45    std::vector<unsigned int> spirv;
46    std::string loggerMessages;
47} glslang_program_t;
48
49static EShLanguage c_shader_stage(glslang_stage_t stage)
50{
51    switch (stage) {
52    case GLSLANG_STAGE_VERTEX:
53        return EShLangVertex;
54    case GLSLANG_STAGE_TESSCONTROL:
55        return EShLangTessControl;
56    case GLSLANG_STAGE_TESSEVALUATION:
57        return EShLangTessEvaluation;
58    case GLSLANG_STAGE_GEOMETRY:
59        return EShLangGeometry;
60    case GLSLANG_STAGE_FRAGMENT:
61        return EShLangFragment;
62    case GLSLANG_STAGE_COMPUTE:
63        return EShLangCompute;
64    case GLSLANG_STAGE_RAYGEN:
65        return EShLangRayGen;
66    case GLSLANG_STAGE_INTERSECT:
67        return EShLangIntersect;
68    case GLSLANG_STAGE_ANYHIT:
69        return EShLangAnyHit;
70    case GLSLANG_STAGE_CLOSESTHIT:
71        return EShLangClosestHit;
72    case GLSLANG_STAGE_MISS:
73        return EShLangMiss;
74    case GLSLANG_STAGE_CALLABLE:
75        return EShLangCallable;
76    case GLSLANG_STAGE_TASK:
77        return EShLangTask;
78    case GLSLANG_STAGE_MESH:
79        return EShLangMesh;
80    default:
81        break;
82    }
83    return EShLangCount;
84}
85
86GLSLANG_EXPORT void glslang_program_SPIRV_generate(glslang_program_t* program, glslang_stage_t stage)
87{
88    glslang_spv_options_t spv_options {};
89    spv_options.disable_optimizer = true;
90    spv_options.validate = true;
91
92    glslang_program_SPIRV_generate_with_options(program, stage, &spv_options);
93}
94
95GLSLANG_EXPORT void glslang_program_SPIRV_generate_with_options(glslang_program_t* program, glslang_stage_t stage, glslang_spv_options_t* spv_options) {
96    spv::SpvBuildLogger logger;
97
98    const glslang::TIntermediate* intermediate = program->program->getIntermediate(c_shader_stage(stage));
99
100    program->spirv.clear();
101
102    glslang::GlslangToSpv(*intermediate, program->spirv, &logger, reinterpret_cast<glslang::SpvOptions*>(spv_options));
103
104    program->loggerMessages = logger.getAllMessages();
105}
106
107GLSLANG_EXPORT size_t glslang_program_SPIRV_get_size(glslang_program_t* program) { return program->spirv.size(); }
108
109GLSLANG_EXPORT void glslang_program_SPIRV_get(glslang_program_t* program, unsigned int* out)
110{
111    memcpy(out, program->spirv.data(), program->spirv.size() * sizeof(unsigned int));
112}
113
114GLSLANG_EXPORT unsigned int* glslang_program_SPIRV_get_ptr(glslang_program_t* program)
115{
116    return program->spirv.data();
117}
118
119GLSLANG_EXPORT const char* glslang_program_SPIRV_get_messages(glslang_program_t* program)
120{
121    return program->loggerMessages.empty() ? nullptr : program->loggerMessages.c_str();
122}