Fix headless mode for atlas mapping

LASTEXILE-CH · Mar 16, 2021 · 8308f58 · 8308f58
1 parent 349e8c9
commit 8308f58
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/addon/utility/build.py b/addon/utility/build.py
@@ -725,9 +725,21 @@ def manage_build(background_pass=False):
             for obj in bpy.context.scene.objects:
                 if obj.type == 'MESH' and obj.name in bpy.context.view_layer.objects:
                     if obj.TLM_ObjectProperties.tlm_mesh_lightmap_use:
-                        img_name = obj.name + '_baked'
-                        Lightmapimage = bpy.data.images[img_name]
-                        obj["Lightmap"] = Lightmapimage.filepath_raw
+
+                        if obj.TLM_ObjectProperties.tlm_mesh_lightmap_unwrap_mode == "AtlasGroupA":
+                            atlasName = obj.TLM_ObjectProperties.tlm_atlas_pointer
+                            img_name = atlasName + '_baked'
+                            Lightmapimage = bpy.data.images[img_name]
+                            obj["Lightmap"] = Lightmapimage.filepath_raw
+                        elif obj.TLM_ObjectProperties.tlm_postpack_object:
+                            atlasName = obj.TLM_ObjectProperties.tlm_postatlas_pointer
+                            img_name = atlasName + '_baked' + ".hdr"
+                            Lightmapimage = bpy.data.images[img_name]
+                            obj["Lightmap"] = Lightmapimage.filepath_raw
+                        else:
+                            img_name = obj.name + '_baked'
+                            Lightmapimage = bpy.data.images[img_name]
+                            obj["Lightmap"] = Lightmapimage.filepath_raw
 
             for image in bpy.data.images:
                 if image.name.endswith("_baked"):