use object filtering in smoke data node

LuxCoreRender · Nov 27, 2018 · 4d382ff · 4d382ff
1 parent 021d26d
commit 4d382ff
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nodes/textures/smoke.py b/nodes/textures/smoke.py
@@ -14,8 +14,12 @@
 class LuxCoreNodeTexSmoke(LuxCoreNodeTexture):
     bl_label = "Smoke"
     bl_width_default = 200
+
+    def poll_domain(self, obj):
+        # Only allow objects with a smoke modifier in domain mode to be picked
+        return utils.find_smoke_domain_modifier(obj)
 
-    domain = PointerProperty(name="Domain", type=bpy.types.Object)
+    domain = PointerProperty(name="Domain", type=bpy.types.Object, poll=poll_domain)
 
     def update_source(self, context):
         value_output = self.outputs["Value"]