#5585: Expose light flag setters

codereader · Apr 11, 2021 · 50653db · 50653db
1 parent 434aa35
commit 50653db
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 0 deletions.
diff --git a/install/scripts/materialtest.py b/install/scripts/materialtest.py
@@ -61,6 +61,8 @@
 newMaterial.setSurfaceFlag(dr.Material.SurfaceFlag.NONSOLID)
 newMaterial.setSurfaceType(dr.Material.SurfaceType.WOOD)
 newMaterial.setSpectrum(5)
+newMaterial.setIsFogLight(1)
+newMaterial.setIsBlendLight(0)
 
 print('Full Material definition:\n{0}\n{{{1}}}'.format(newMaterial.getName(), newMaterial.getDefinition()))
 

diff --git a/plugins/script/interfaces/ShaderSystemInterface.cpp b/plugins/script/interfaces/ShaderSystemInterface.cpp
@@ -218,6 +218,11 @@ void ShaderSystemInterface::registerInterface(py::module& scope, py::dict& globa
 	material.def("isAmbientLight", &ScriptMaterial::isAmbientLight);
 	material.def("isBlendLight", &ScriptMaterial::isBlendLight);
 	material.def("isFogLight", &ScriptMaterial::isFogLight);
+	material.def("isCubicLight", &ScriptMaterial::isCubicLight);
+	material.def("setIsAmbientLight", &ScriptMaterial::setIsAmbientLight);
+	material.def("setIsBlendLight", &ScriptMaterial::setIsBlendLight);
+	material.def("setIsFogLight", &ScriptMaterial::setIsFogLight);
+	material.def("setIsCubicLight", &ScriptMaterial::setIsCubicLight);
 	material.def("isNull", &ScriptMaterial::isNull);
 	material.def("getEditorImageExpressionString", &ScriptMaterial::getEditorImageExpressionString);
 	material.def("setEditorImageExpressionFromString", &ScriptMaterial::setEditorImageExpressionFromString);

diff --git a/plugins/script/interfaces/ShaderSystemInterface.h b/plugins/script/interfaces/ShaderSystemInterface.h
@@ -63,6 +63,34 @@ class ScriptMaterial
 		return _material ? _material->isFogLight() : false;
 	}
 
+    bool isCubicLight() {
+        return _material ? _material->isCubicLight() : false;
+    }
+
+    void setIsAmbientLight(bool newValue)
+    {
+        throwIfMaterialCannotBeModified();
+        _material->setIsAmbientLight(newValue);
+    }
+
+    void setIsBlendLight(bool newValue)
+    {
+        throwIfMaterialCannotBeModified();
+        _material->setIsBlendLight(newValue);
+    }
+
+    void setIsFogLight(bool newValue)
+    {
+        throwIfMaterialCannotBeModified();
+        _material->setIsFogLight(newValue);
+    }
+
+    void setIsCubicLight(bool newValue)
+    {
+        throwIfMaterialCannotBeModified();
+        _material->setIsCubicLight(newValue);
+    }
+
 	bool isNull() const {
 		return _material == nullptr;
 	}