Files
SDL3CPlusPlus/MaterialX/source/PyMaterialX/PyMaterialXGenGlsl/PyGlslShaderGenerator.cpp
2026-01-06 13:25:49 +00:00

92 lines
3.3 KiB
C++

//
// Copyright Contributors to the MaterialX Project
// SPDX-License-Identifier: Apache-2.0
//
#include <PyMaterialX/PyMaterialX.h>
#include <MaterialXGenGlsl/GlslShaderGenerator.h>
#include <MaterialXGenGlsl/GlslResourceBindingContext.h>
#include <MaterialXGenGlsl/EsslShaderGenerator.h>
#include <MaterialXGenGlsl/VkShaderGenerator.h>
#include <MaterialXGenGlsl/WgslShaderGenerator.h>
#include <MaterialXGenShader/Shader.h>
#include <MaterialXGenShader/ShaderGenerator.h>
namespace py = pybind11;
namespace mx = MaterialX;
// GLSL shader generator bindings
namespace
{
// Creator wrappers to avoid having to expose the TypeSystem class in python
mx::ShaderGeneratorPtr GlslShaderGenerator_create()
{
return mx::GlslShaderGenerator::create();
}
mx::ShaderGeneratorPtr EsslShaderGenerator_create()
{
return mx::EsslShaderGenerator::create();
}
mx::ShaderGeneratorPtr VkShaderGenerator_create()
{
return mx::VkShaderGenerator::create();
}
mx::ShaderGeneratorPtr WgslShaderGenerator_create()
{
return mx::WgslShaderGenerator::create();
}
}
void bindPyGlslShaderGenerator(py::module& mod)
{
py::class_<mx::GlslShaderGenerator, mx::HwShaderGenerator, mx::GlslShaderGeneratorPtr>(mod, "GlslShaderGenerator")
.def_static("create", &GlslShaderGenerator_create)
.def("generate", &mx::GlslShaderGenerator::generate)
.def("getTarget", &mx::GlslShaderGenerator::getTarget)
.def("getVersion", &mx::GlslShaderGenerator::getVersion);
}
void bindPyGlslResourceBindingContext(py::module &mod)
{
py::class_<mx::GlslResourceBindingContext, mx::HwResourceBindingContext, mx::GlslResourceBindingContextPtr>(mod, "GlslResourceBindingContext")
.def_static("create", &mx::GlslResourceBindingContext::create)
.def(py::init<size_t, size_t>())
.def("emitDirectives", &mx::GlslResourceBindingContext::emitDirectives)
.def("emitResourceBindings", &mx::GlslResourceBindingContext::emitResourceBindings);
}
// Essl shader generator bindings
void bindPyEsslShaderGenerator(py::module& mod)
{
py::class_<mx::EsslShaderGenerator, mx::GlslShaderGenerator, mx::EsslShaderGeneratorPtr>(mod, "EsslShaderGenerator")
.def_static("create", &EsslShaderGenerator_create)
.def("generate", &mx::EsslShaderGenerator::generate)
.def("getTarget", &mx::EsslShaderGenerator::getTarget)
.def("getVersion", &mx::EsslShaderGenerator::getVersion);
}
// Glsl Vulkan shader generator bindings
void bindPyVkShaderGenerator(py::module& mod)
{
py::class_<mx::VkShaderGenerator, mx::GlslShaderGenerator, mx::VkShaderGeneratorPtr>(mod, "VkShaderGenerator")
.def_static("create", &VkShaderGenerator_create)
.def("generate", &mx::VkShaderGenerator::generate)
.def("getTarget", &mx::VkShaderGenerator::getTarget)
.def("getVersion", &mx::VkShaderGenerator::getVersion);
}
// Glsl Wgsl shader generator bindings
void bindPyWgslShaderGenerator(py::module& mod)
{
py::class_<mx::WgslShaderGenerator, mx::GlslShaderGenerator, mx::WgslShaderGeneratorPtr>(mod, "WgslShaderGenerator")
.def_static("create", &WgslShaderGenerator_create)
.def("generate", &mx::WgslShaderGenerator::generate)
.def("getTarget", &mx::WgslShaderGenerator::getTarget)
.def("getVersion", &mx::WgslShaderGenerator::getVersion);
}