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

99 lines
4.9 KiB
C++

//
// Copyright Contributors to the MaterialX Project
// SPDX-License-Identifier: Apache-2.0
//
#include <PyMaterialX/PyMaterialX.h>
#include <MaterialXCore/Geom.h>
namespace py = pybind11;
namespace mx = MaterialX;
#define BIND_GEOMINFO_FUNC_INSTANCE(NAME, T) \
.def("_setGeomPropValue" #NAME, &mx::GeomInfo::setGeomPropValue<T>)
void bindPyGeom(py::module& mod)
{
py::class_<mx::GeomElement, mx::GeomElementPtr, mx::Element>(mod, "GeomElement")
.def("setGeom", &mx::GeomElement::setGeom)
.def("hasGeom", &mx::GeomElement::hasGeom)
.def("getGeom", &mx::GeomElement::getGeom)
.def("setCollectionString", &mx::GeomElement::setCollectionString)
.def("hasCollectionString", &mx::GeomElement::hasCollectionString)
.def("getCollectionString", &mx::GeomElement::getCollectionString)
.def("setCollection", &mx::GeomElement::setCollection)
.def("getCollection", &mx::GeomElement::getCollection);
py::class_<mx::GeomInfo, mx::GeomInfoPtr, mx::GeomElement>(mod, "GeomInfo")
.def("addGeomProp", &mx::GeomInfo::addGeomProp)
.def("getGeomProp", &mx::GeomInfo::getGeomProp)
.def("getGeomProps", &mx::GeomInfo::getGeomProps)
.def("removeGeomProp", &mx::GeomInfo::removeGeomProp)
.def("addToken", &mx::GeomInfo::addToken,
py::arg("name") = mx::DEFAULT_TYPE_STRING)
.def("getToken", &mx::GeomInfo::getToken)
.def("getTokens", &mx::GeomInfo::getTokens)
.def("removeToken", &mx::GeomInfo::removeToken)
.def("setTokenValue", &mx::GeomInfo::setTokenValue)
BIND_GEOMINFO_FUNC_INSTANCE(integer, int)
BIND_GEOMINFO_FUNC_INSTANCE(boolean, bool)
BIND_GEOMINFO_FUNC_INSTANCE(float, float)
BIND_GEOMINFO_FUNC_INSTANCE(color3, mx::Color3)
BIND_GEOMINFO_FUNC_INSTANCE(color4, mx::Color4)
BIND_GEOMINFO_FUNC_INSTANCE(vector2, mx::Vector2)
BIND_GEOMINFO_FUNC_INSTANCE(vector3, mx::Vector3)
BIND_GEOMINFO_FUNC_INSTANCE(vector4, mx::Vector4)
BIND_GEOMINFO_FUNC_INSTANCE(matrix33, mx::Matrix33)
BIND_GEOMINFO_FUNC_INSTANCE(matrix44, mx::Matrix44)
BIND_GEOMINFO_FUNC_INSTANCE(string, std::string)
BIND_GEOMINFO_FUNC_INSTANCE(integerarray, mx::IntVec)
BIND_GEOMINFO_FUNC_INSTANCE(booleanarray, mx::BoolVec)
BIND_GEOMINFO_FUNC_INSTANCE(floatarray, mx::FloatVec)
BIND_GEOMINFO_FUNC_INSTANCE(stringarray, mx::StringVec)
.def_readonly_static("CATEGORY", &mx::GeomInfo::CATEGORY);
py::class_<mx::GeomProp, mx::GeomPropPtr, mx::ValueElement>(mod, "GeomProp")
.def_readonly_static("CATEGORY", &mx::GeomProp::CATEGORY);
py::class_<mx::GeomPropDef, mx::GeomPropDefPtr, mx::TypedElement>(mod, "GeomPropDef")
.def("setGeomProp", &mx::GeomPropDef::setGeomProp)
.def("hasGeomProp", &mx::GeomPropDef::hasGeomProp)
.def("getGeomProp", &mx::GeomPropDef::getGeomProp)
.def("setSpace", &mx::GeomPropDef::setSpace)
.def("hasSpace", &mx::GeomPropDef::hasSpace)
.def("getSpace", &mx::GeomPropDef::getSpace)
.def("setIndex", &mx::GeomPropDef::setIndex)
.def("hasIndex", &mx::GeomPropDef::hasIndex)
.def("getIndex", &mx::GeomPropDef::getIndex)
.def("setGeomProp", &mx::GeomPropDef::setGeomProp)
.def("hasGeomProp", &mx::GeomPropDef::hasGeomProp)
.def("getGeomProp", &mx::GeomPropDef::getGeomProp)
.def_readonly_static("CATEGORY", &mx::GeomPropDef::CATEGORY);
py::class_<mx::Collection, mx::CollectionPtr, mx::Element>(mod, "Collection")
.def("setIncludeGeom", &mx::Collection::setIncludeGeom)
.def("hasIncludeGeom", &mx::Collection::hasIncludeGeom)
.def("getIncludeGeom", &mx::Collection::getIncludeGeom)
.def("setExcludeGeom", &mx::Collection::setExcludeGeom)
.def("hasExcludeGeom", &mx::Collection::hasExcludeGeom)
.def("getExcludeGeom", &mx::Collection::getExcludeGeom)
.def("setIncludeCollectionString", &mx::Collection::setIncludeCollectionString)
.def("hasIncludeCollectionString", &mx::Collection::hasIncludeCollectionString)
.def("getIncludeCollectionString", &mx::Collection::getIncludeCollectionString)
.def("setIncludeCollection", &mx::Collection::setIncludeCollection)
.def("setIncludeCollections", &mx::Collection::setIncludeCollections)
.def("getIncludeCollections", &mx::Collection::getIncludeCollections)
.def("hasIncludeCycle", &mx::Collection::hasIncludeCycle)
.def("matchesGeomString", &mx::Collection::matchesGeomString)
.def_readonly_static("CATEGORY", &mx::Collection::CATEGORY);
mod.def("geomStringsMatch", &mx::geomStringsMatch);
mod.attr("GEOM_PATH_SEPARATOR") = mx::GEOM_PATH_SEPARATOR;
mod.attr("UNIVERSAL_GEOM_NAME") = mx::UNIVERSAL_GEOM_NAME;
mod.attr("UDIM_TOKEN") = mx::UDIM_TOKEN;
mod.attr("UV_TILE_TOKEN") = mx::UV_TILE_TOKEN;
mod.attr("UDIM_SET_PROPERTY") = mx::UDIM_SET_PROPERTY;
}