mirror of
https://github.com/MobileGL-Dev/MobileGL
synced 2026-09-11 21:58:31 +09:00
[Chore] (MG_Util/Program): auto generate ubo for plain GL uniforms
This commit is contained in:
@@ -10,56 +10,6 @@ class ProgramTest : public ::testing::Test {
|
|||||||
protected:
|
protected:
|
||||||
};
|
};
|
||||||
|
|
||||||
struct InterfaceVariable {
|
|
||||||
std::string name;
|
|
||||||
uint32_t location;
|
|
||||||
|
|
||||||
bool operator<(const InterfaceVariable& other) const {
|
|
||||||
return location < other.location;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool operator==(const InterfaceVariable& other) const {
|
|
||||||
return location == other.location && name == other.name;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
static std::vector<InterfaceVariable> GetShaderInterface(const std::vector<unsigned int>& spirv, spvc_resource_type resource_type) {
|
|
||||||
spvc_context context = nullptr;
|
|
||||||
spvc_context_create(&context);
|
|
||||||
|
|
||||||
spvc_parsed_ir ir = nullptr;
|
|
||||||
spvc_context_parse_spirv(context, spirv.data(), spirv.size(), &ir);
|
|
||||||
|
|
||||||
spvc_compiler compiler = nullptr;
|
|
||||||
spvc_context_create_compiler(context, SPVC_BACKEND_NONE, ir, SPVC_CAPTURE_MODE_TAKE_OWNERSHIP, &compiler);
|
|
||||||
|
|
||||||
spvc_resources resources = nullptr;
|
|
||||||
spvc_compiler_create_shader_resources(compiler, &resources);
|
|
||||||
|
|
||||||
const spvc_reflected_resource *list = nullptr;
|
|
||||||
size_t count = 0;
|
|
||||||
spvc_resources_get_resource_list_for_type(resources, resource_type, &list, &count);
|
|
||||||
|
|
||||||
std::vector<InterfaceVariable> variables;
|
|
||||||
for (size_t i = 0; i < count; ++i) {
|
|
||||||
unsigned int builtin;
|
|
||||||
if (spvc_compiler_has_decoration(compiler, list[i].id, SpvDecorationBuiltIn)) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
InterfaceVariable var;
|
|
||||||
var.name = list[i].name;
|
|
||||||
var.location = spvc_compiler_get_decoration(compiler, list[i].id, SpvDecorationLocation);
|
|
||||||
variables.push_back(var);
|
|
||||||
}
|
|
||||||
|
|
||||||
spvc_context_destroy(context);
|
|
||||||
|
|
||||||
std::sort(variables.begin(), variables.end());
|
|
||||||
|
|
||||||
return variables;
|
|
||||||
}
|
|
||||||
|
|
||||||
TEST_F(ProgramTest, Sanity) {
|
TEST_F(ProgramTest, Sanity) {
|
||||||
ASSERT_TRUE(true);
|
ASSERT_TRUE(true);
|
||||||
}
|
}
|
||||||
@@ -235,8 +185,8 @@ TEST_F(ProgramTest, DecompProgram) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// spirv link check
|
// spirv link check
|
||||||
auto vs_outputs = GetShaderInterface(spirvs[0], SPVC_RESOURCE_TYPE_STAGE_OUTPUT);
|
auto vs_outputs = sessions[0].GetShaderInterface(SPVC_RESOURCE_TYPE_STAGE_OUTPUT);
|
||||||
auto fs_inputs = GetShaderInterface(spirvs[1], SPVC_RESOURCE_TYPE_STAGE_INPUT);
|
auto fs_inputs = sessions[1].GetShaderInterface(SPVC_RESOURCE_TYPE_STAGE_INPUT);
|
||||||
|
|
||||||
ASSERT_EQ(vs_outputs.size(), fs_inputs.size());
|
ASSERT_EQ(vs_outputs.size(), fs_inputs.size());
|
||||||
|
|
||||||
@@ -244,8 +194,8 @@ TEST_F(ProgramTest, DecompProgram) {
|
|||||||
EXPECT_EQ(vs_outputs[i].location, fs_inputs[i].location);
|
EXPECT_EQ(vs_outputs[i].location, fs_inputs[i].location);
|
||||||
}
|
}
|
||||||
|
|
||||||
auto vs_uniforms = GetShaderInterface(spirvs[0], SPVC_RESOURCE_TYPE_GL_PLAIN_UNIFORM);
|
auto vs_uniforms = sessions[0].GetShaderInterface(SPVC_RESOURCE_TYPE_GL_PLAIN_UNIFORM);
|
||||||
auto fs_uniforms = GetShaderInterface(spirvs[1], SPVC_RESOURCE_TYPE_GL_PLAIN_UNIFORM);
|
auto fs_uniforms = sessions[1].GetShaderInterface(SPVC_RESOURCE_TYPE_GL_PLAIN_UNIFORM);
|
||||||
|
|
||||||
std::unordered_map<std::string, uint32_t> uniform_locations;
|
std::unordered_map<std::string, uint32_t> uniform_locations;
|
||||||
for (const auto& uniform : vs_uniforms) {
|
for (const auto& uniform : vs_uniforms) {
|
||||||
@@ -259,8 +209,8 @@ TEST_F(ProgramTest, DecompProgram) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
auto vs_samplers = GetShaderInterface(spirvs[0], SPVC_RESOURCE_TYPE_SAMPLED_IMAGE);
|
auto vs_samplers = sessions[0].GetShaderInterface(SPVC_RESOURCE_TYPE_SAMPLED_IMAGE);
|
||||||
auto fs_samplers = GetShaderInterface(spirvs[1], SPVC_RESOURCE_TYPE_SAMPLED_IMAGE);
|
auto fs_samplers = sessions[1].GetShaderInterface(SPVC_RESOURCE_TYPE_SAMPLED_IMAGE);
|
||||||
|
|
||||||
std::unordered_map<std::string, uint32_t> sampler_locations;
|
std::unordered_map<std::string, uint32_t> sampler_locations;
|
||||||
for (const auto& uniform : vs_uniforms) {
|
for (const auto& uniform : vs_uniforms) {
|
||||||
|
|||||||
@@ -136,12 +136,13 @@ namespace MobileGL {
|
|||||||
tshader->setStrings(src, 1);
|
tshader->setStrings(src, 1);
|
||||||
tshader->setInvertY(true);
|
tshader->setInvertY(true);
|
||||||
tshader->setEnvInput(glslang::EShSourceGlsl, lang, glslang::EShClientVulkan, 450);
|
tshader->setEnvInput(glslang::EShSourceGlsl, lang, glslang::EShClientVulkan, 450);
|
||||||
tshader->setEnvClient(glslang::EShClientOpenGL, glslang::EShTargetOpenGL_450);
|
tshader->setEnvClient(glslang::EShClientVulkan, glslang::EShTargetVulkan_1_3);
|
||||||
tshader->setEnvTarget(glslang::EShTargetSpv, glslang::EShTargetSpv_1_6);
|
tshader->setEnvTarget(glslang::EShTargetSpv, glslang::EShTargetSpv_1_6);
|
||||||
tshader->setAutoMapLocations(true);
|
tshader->setAutoMapLocations(true);
|
||||||
tshader->setAutoMapBindings(true);
|
tshader->setAutoMapBindings(true);
|
||||||
tshader->setEnvInputVulkanRulesRelaxed(); // using EXT_vulkan_glsl_relaxed for gl_VertexID and gl_InstanceID?
|
tshader->setEnvInputVulkanRulesRelaxed(); // using EXT_vulkan_glsl_relaxed for gl_VertexID and gl_InstanceID?
|
||||||
if (!tshader->parse(&GetTBuiltInResourceInstance(), 150, ECompatibilityProfile,
|
tshader->setGlobalUniformBlockName("MGL_GLOBAL_UBO");
|
||||||
|
if (!tshader->parse(&GetTBuiltInResourceInstance(), 150, ECoreProfile,
|
||||||
/*forceDefaultVersionAndProfile: */false,
|
/*forceDefaultVersionAndProfile: */false,
|
||||||
/*forwardCompatible: */true, EShMsgDefault)) {
|
/*forwardCompatible: */true, EShMsgDefault)) {
|
||||||
ResultInfo r;
|
ResultInfo r;
|
||||||
|
|||||||
@@ -28,17 +28,31 @@ namespace MobileGL {
|
|||||||
template <typename T>
|
template <typename T>
|
||||||
using Result = std::expected<T, ResultInfo>;
|
using Result = std::expected<T, ResultInfo>;
|
||||||
|
|
||||||
|
struct InterfaceVariable {
|
||||||
|
std::string name;
|
||||||
|
uint32_t location;
|
||||||
|
|
||||||
|
bool operator<(const InterfaceVariable& other) const {
|
||||||
|
return location < other.location;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool operator==(const InterfaceVariable& other) const {
|
||||||
|
return location == other.location && name == other.name;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
class SpvcSession {
|
class SpvcSession {
|
||||||
public:
|
public:
|
||||||
SpvcSession() {}
|
SpvcSession() {}
|
||||||
|
|
||||||
explicit SpvcSession(Vector<unsigned int> spirv) {
|
explicit SpvcSession(const Vector<unsigned int>& spirv) {
|
||||||
const SpvId *p_spirv = spirv.data();
|
const SpvId *p_spirv = spirv.data();
|
||||||
size_t word_count = spirv.size();
|
size_t word_count = spirv.size();
|
||||||
|
|
||||||
spvc_context_create(&context);
|
spvc_context_create(&context);
|
||||||
spvc_context_parse_spirv(context, p_spirv, word_count, &ir);
|
spvc_context_parse_spirv(context, p_spirv, word_count, &ir);
|
||||||
spvc_context_create_compiler(context, SPVC_BACKEND_GLSL, ir, SPVC_CAPTURE_MODE_TAKE_OWNERSHIP, &compiler);
|
spvc_context_create_compiler(context, SPVC_BACKEND_GLSL, ir, SPVC_CAPTURE_MODE_TAKE_OWNERSHIP, &compiler);
|
||||||
|
spvc_compiler_create_shader_resources(compiler, &resources);
|
||||||
}
|
}
|
||||||
|
|
||||||
SpvcSession(SpvcSession&) = delete;
|
SpvcSession(SpvcSession&) = delete;
|
||||||
@@ -48,6 +62,7 @@ namespace MobileGL {
|
|||||||
std::swap(this->compiler, that.compiler);
|
std::swap(this->compiler, that.compiler);
|
||||||
std::swap(this->ir, that.ir);
|
std::swap(this->ir, that.ir);
|
||||||
std::swap(this->compiler_options, that.compiler_options);
|
std::swap(this->compiler_options, that.compiler_options);
|
||||||
|
std::swap(this->resources, that.resources);
|
||||||
}
|
}
|
||||||
|
|
||||||
SpvcSession& operator=(SpvcSession& session) = delete;
|
SpvcSession& operator=(SpvcSession& session) = delete;
|
||||||
@@ -57,6 +72,7 @@ namespace MobileGL {
|
|||||||
std::swap(this->compiler, that.compiler);
|
std::swap(this->compiler, that.compiler);
|
||||||
std::swap(this->ir, that.ir);
|
std::swap(this->ir, that.ir);
|
||||||
std::swap(this->compiler_options, that.compiler_options);
|
std::swap(this->compiler_options, that.compiler_options);
|
||||||
|
std::swap(this->resources, that.resources);
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -69,11 +85,31 @@ namespace MobileGL {
|
|||||||
return spvc_compiler_install_compiler_options(compiler, options);
|
return spvc_compiler_install_compiler_options(compiler, options);
|
||||||
}
|
}
|
||||||
|
|
||||||
spvc_result Compile(const char** result) {
|
Vector<InterfaceVariable> GetShaderInterface(spvc_resource_type resource_type) const {
|
||||||
|
const spvc_reflected_resource *list = nullptr;
|
||||||
|
size_t count = 0;
|
||||||
|
spvc_resources_get_resource_list_for_type(resources, resource_type, &list, &count);
|
||||||
|
|
||||||
|
Vector<InterfaceVariable> variables;
|
||||||
|
for (size_t i = 0; i < count; ++i) {
|
||||||
|
if (spvc_compiler_has_decoration(compiler, list[i].id, SpvDecorationBuiltIn)) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
InterfaceVariable var;
|
||||||
|
var.name = list[i].name;
|
||||||
|
var.location = spvc_compiler_get_decoration(compiler, list[i].id, SpvDecorationLocation);
|
||||||
|
variables.push_back(var);
|
||||||
|
}
|
||||||
|
std::sort(variables.begin(), variables.end());
|
||||||
|
return variables;
|
||||||
|
}
|
||||||
|
|
||||||
|
spvc_result Compile(const char** result) const {
|
||||||
return spvc_compiler_compile(compiler, result);
|
return spvc_compiler_compile(compiler, result);
|
||||||
}
|
}
|
||||||
|
|
||||||
const char* GetLastErrorString() {
|
const char* GetLastErrorString() const {
|
||||||
return spvc_context_get_last_error_string(context);
|
return spvc_context_get_last_error_string(context);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -83,6 +119,7 @@ namespace MobileGL {
|
|||||||
spvc_parsed_ir ir = nullptr;
|
spvc_parsed_ir ir = nullptr;
|
||||||
spvc_compiler compiler = nullptr;
|
spvc_compiler compiler = nullptr;
|
||||||
spvc_compiler_options compiler_options = nullptr;
|
spvc_compiler_options compiler_options = nullptr;
|
||||||
|
spvc_resources resources = nullptr;
|
||||||
};
|
};
|
||||||
|
|
||||||
inline static EShLanguage GetEShLanguageByShaderType(GLenum shaderType) {
|
inline static EShLanguage GetEShLanguageByShaderType(GLenum shaderType) {
|
||||||
|
|||||||
Reference in New Issue
Block a user