Add Python bindings for generation and rendering (#1386)

Most of these wrapper updates are related to exposing missing APIs in Python for code gen and rendering.
AcademySoftwareFoundation · Jun 22, 2023 · da444d5 · da444d5
1 parent a8dffbd
commit da444d5
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 0 deletions.
diff --git a/source/PyMaterialX/PyMaterialXCore/PyDefinition.cpp b/source/PyMaterialX/PyMaterialXCore/PyDefinition.cpp
@@ -43,6 +43,9 @@ void bindPyDefinition(py::module& mod)
         .def("getFunction", &mx::Implementation::getFunction)
         .def("setNodeDef", &mx::Implementation::setNodeDef)
         .def("getNodeDef", &mx::Implementation::getNodeDef)
+        .def("setNodeGraph", &mx::Implementation::setNodeGraph)
+        .def("hasNodeGraph", &mx::Implementation::hasNodeGraph)        
+        .def("getNodeGraph", &mx::Implementation::getNodeGraph)
         .def_readonly_static("CATEGORY", &mx::Implementation::CATEGORY)
         .def_readonly_static("FILE_ATTRIBUTE", &mx::Implementation::FILE_ATTRIBUTE)
         .def_readonly_static("FUNCTION_ATTRIBUTE", &mx::Implementation::FUNCTION_ATTRIBUTE);

diff --git a/source/PyMaterialX/PyMaterialXGenShader/PyGenOptions.cpp b/source/PyMaterialX/PyMaterialXGenShader/PyGenOptions.cpp
@@ -27,12 +27,18 @@ void bindPyGenOptions(py::module& mod)
         .def_readwrite("shaderInterfaceType", &mx::GenOptions::shaderInterfaceType)
         .def_readwrite("fileTextureVerticalFlip", &mx::GenOptions::fileTextureVerticalFlip)
         .def_readwrite("targetColorSpaceOverride", &mx::GenOptions::targetColorSpaceOverride)
+        .def_readwrite("addUpstreamDependencies", &mx::GenOptions::addUpstreamDependencies)
+        .def_readwrite("libraryPrefix", &mx::GenOptions::libraryPrefix)        
         .def_readwrite("targetDistanceUnit", &mx::GenOptions::targetDistanceUnit)
         .def_readwrite("hwTransparency", &mx::GenOptions::hwTransparency)
         .def_readwrite("hwSpecularEnvironmentMethod", &mx::GenOptions::hwSpecularEnvironmentMethod)
+        .def_readwrite("hwWriteDepthMoments", &mx::GenOptions::hwWriteDepthMoments)
+        .def_readwrite("hwShadowMap", &mx::GenOptions::hwShadowMap)
         .def_readwrite("hwMaxActiveLightSources", &mx::GenOptions::hwMaxActiveLightSources)
         .def_readwrite("hwNormalizeUdimTexCoords", &mx::GenOptions::hwNormalizeUdimTexCoords)
+        .def_readwrite("hwAmbientOcclusion", &mx::GenOptions::hwAmbientOcclusion)        
         .def_readwrite("hwWriteAlbedoTable", &mx::GenOptions::hwWriteAlbedoTable)
+        .def_readwrite("hwImplicitBitangents", &mx::GenOptions::hwImplicitBitangents)
         .def_readwrite("emitColorTransforms", &mx::GenOptions::emitColorTransforms)
         .def(py::init<>());
 }
diff --git a/source/PyMaterialX/PyMaterialXRender/PyCgltfLoader.cpp b/source/PyMaterialX/PyMaterialXRender/PyCgltfLoader.cpp
@@ -0,0 +1,18 @@
+//
+// Copyright Contributors to the MaterialX Project
+// SPDX-License-Identifier: Apache-2.0
+//
+
+#include <PyMaterialX/PyMaterialX.h>
+#include <MaterialXRender/CgltfLoader.h>
+
+namespace py = pybind11;
+namespace mx = MaterialX;
+
+void bindPyCgltfLoader(py::module& mod)
+{
+    py::class_<mx::CgltfLoader, mx::CgltfLoaderPtr, mx::GeometryLoader>(mod, "CgltfLoader")
+        .def_static("create", &mx::CgltfLoader::create)
+        .def(py::init<>())
+        .def("load", &mx::CgltfLoader::load);
+}
diff --git a/source/PyMaterialX/PyMaterialXRender/PyLightHandler.cpp b/source/PyMaterialX/PyMaterialXRender/PyLightHandler.cpp
@@ -31,6 +31,8 @@ void bindPyLightHandler(py::module& mod)
         .def("getAlbedoTable", &mx::LightHandler::getAlbedoTable)
         .def("setEnvSampleCount", &mx::LightHandler::setEnvSampleCount)
         .def("getEnvSampleCount", &mx::LightHandler::getEnvSampleCount)
+        .def("setRefractionTwoSided", &mx::LightHandler::setRefractionTwoSided)
+        .def("getRefractionTwoSided", &mx::LightHandler::getRefractionTwoSided)        
         .def("addLightSource", &mx::LightHandler::addLightSource)
         .def("setLightSources", &mx::LightHandler::setLightSources)
         .def("getLightSources", &mx::LightHandler::getLightSources)