From ff55f1cc8b2b8fd657571c005cd33220c2ddc4fb Mon Sep 17 00:00:00 2001
From: SimonNordon4 <simon.nordon@protonmail.com>
Date: Wed, 11 Sep 2024 20:20:49 +1000
Subject: [PATCH] stable but it's doing multiple bakes for each object

---
 lightmapper/lightmapper_operators.py | 54 ++++++++++++++++++----------
 1 file changed, 36 insertions(+), 18 deletions(-)

diff --git a/lightmapper/lightmapper_operators.py b/lightmapper/lightmapper_operators.py
index b2546d1..6e39771 100644
--- a/lightmapper/lightmapper_operators.py
+++ b/lightmapper/lightmapper_operators.py
@@ -170,26 +170,44 @@ def _prepare_objects_for_baking(self, mesh_objects, bake_object):
         bpy.context.view_layer.objects.active = bake_object
     
     def _create_bake_material(self, bake_object):
-        # Create a new material
-        bake_material_name = "PreviewBakeMaterial"
-        
-        # check if the material already exists
-        
-        existing_material = bpy.data.materials.get(bake_material_name)
-        if existing_material:
-            bake_material = existing_material
+        """Create a bake material for the bake object."""
+        # Ensure the object has an active material
+        if not bake_object.data.materials:
+            material = bpy.data.materials.new(name="BakeMaterial")
+            bake_object.data.materials.append(material)
         else:
-            bake_material = bpy.data.materials.new(name=bake_material_name)
-        
-        bake_object.data.materials.append(bake_material)
-        
-        bake_material.use_nodes = True
-        node_tree = bake_material.node_tree
-        nodes = node_tree.nodes
-        image_node = nodes.new(type='ShaderNodeTexImage')
-        image_node.image = self.bake_image
-        node_tree.nodes.active = image_node
+            material = bake_object.data.materials[0]
+
+        # Ensure the material uses nodes
+        if not material.use_nodes:
+            material.use_nodes = True
+
+        # Access the material's node tree
+        node_tree = material.node_tree
 
+        # Clear existing nodes
+        node_tree.nodes.clear()
+
+        # Create new nodes
+        bsdf_node = node_tree.nodes.new(type='ShaderNodeBsdfPrincipled')
+        output_node = node_tree.nodes.new(type='ShaderNodeOutputMaterial')
+        image_node = node_tree.nodes.new(type='ShaderNodeTexImage')
+
+        # Position nodes
+        bsdf_node.location = (0, 0)
+        output_node.location = (200, 0)
+        image_node.location = (-200, 0)
+
+        # Link nodes
+        node_tree.links.new(image_node.outputs['Color'], bsdf_node.inputs['Base Color'])
+        node_tree.links.new(bsdf_node.outputs['BSDF'], output_node.inputs['Surface'])
+
+        # Create a new image for baking
+        bake_image = bpy.data.images.new(name="BakeImage", width=1024, height=1024)
+        image_node.image = bake_image
+
+        return bake_image
+    
     def _setup_bake_settings(self):
         """ Set up bake settings for diffuse lightmap baking. """
         bpy.context.scene.render.engine = 'CYCLES'