From 0f5f9f34c83a05f4e0850d4f24cfc87a4a966208 Mon Sep 17 00:00:00 2001
From: Adam Johnson <AdamJohnso@gmail.com>
Date: Wed, 4 Aug 2021 17:26:45 -0400
Subject: [PATCH] Regenerate lightmap UVs on ephemeral meshes.

---
 korman/exporter/etlight.py | 40 +++++++++++++++++++++++---------------
 1 file changed, 24 insertions(+), 16 deletions(-)

diff --git a/korman/exporter/etlight.py b/korman/exporter/etlight.py
index 2ee446b..9518f21 100644
--- a/korman/exporter/etlight.py
+++ b/korman/exporter/etlight.py
@@ -110,11 +110,11 @@ class LightBaker:
         """Bakes all static lighting for Plasma geometry"""
 
         self._report.msg("\nBaking Static Lighting...")
-        bake = self._harvest_bakable_objects(objs)
 
         with GoodNeighbor() as toggle:
             try:
                 # reduce the amount of indentation
+                bake = self._harvest_bakable_objects(objs, toggle)
                 result = self._bake_static_lighting(bake, toggle)
             finally:
                 # this stuff has been observed to be problematic with GoodNeighbor
@@ -238,7 +238,7 @@ class LightBaker:
                 return True
         return False
 
-    def _harvest_bakable_objects(self, objs):
+    def _harvest_bakable_objects(self, objs, toggle):
         # The goal here is to minimize the calls to bake_image, so we are going to collect everything
         # that needs to be baked and sort it out by configuration.
         default_layers = tuple((True,) * _NUM_RENDER_LAYERS)
@@ -255,7 +255,8 @@ class LightBaker:
                     if self.lightmap_uvtex_name in uv_texture_names:
                         self._report.msg("'{}': Skipping due to valid lightmap override", obj.name, indent=1)
                     else:
-                        self._report.msg("'{}': Have lightmap but UVs are missing???", obj.name, indent=1)
+                        self._report.warn("'{}': Have lightmap, but regenerating UVs", obj.name, indent=1)
+                        self._prep_for_lightmap_uvs(obj, mod.image, toggle)
                     return False
                 return True
             return False
@@ -360,6 +361,24 @@ class LightBaker:
             im = data_images.new(im_name, width=size, height=size)
         self._lightmap_images[bo.name] = im
 
+        self._prep_for_lightmap_uvs(bo, im, toggle)
+
+        # Now, set the new LIGHTMAPGEN uv layer as what we want to render to...
+        # NOTE that this will need to be reset by us to what the user had previously
+        # Not using toggle.track due to observed oddities
+        for i in uv_textures:
+            value = i.name == self.lightmap_uvtex_name
+            i.active = value
+            i.active_render = value
+
+        # Indicate we should bake
+        return True
+
+    def _prep_for_lightmap_uvs(self, bo, image, toggle):
+        mesh = bo.data
+        modifier = bo.plasma_modifiers.lightmap
+        uv_textures = mesh.uv_textures
+
         # If there is a cached LIGHTMAPGEN uvtexture, nuke it
         uvtex = uv_textures.get(self.lightmap_uvtex_name, None)
         if uvtex is not None:
@@ -392,7 +411,7 @@ class LightBaker:
             # if the artist hid any UVs, they will not be baked to... fix this now
             with self._set_mode("EDIT"):
                 bpy.ops.uv.reveal()
-            self._associate_image_with_uvtex(uv_textures.active, im)
+            self._associate_image_with_uvtex(uv_textures.active, image)
 
             # Meshes with modifiers need to have islands packed to prevent generated vertices
             # from sharing UVs. Sigh.
@@ -408,23 +427,12 @@ class LightBaker:
             # results in my tests. it will be good enough for quick exports.
             uvtex = uv_textures.new(self.lightmap_uvtex_name)
             uv_textures.active = uvtex
-            self._associate_image_with_uvtex(uvtex, im)
+            self._associate_image_with_uvtex(uvtex, image)
             with self._set_mode("EDIT"):
                 bpy.ops.mesh.select_all(action="SELECT")
                 bpy.ops.uv.select_all(action="SELECT")
                 bpy.ops.uv.smart_project(island_margin=0.05)
 
-        # Now, set the new LIGHTMAPGEN uv layer as what we want to render to...
-        # NOTE that this will need to be reset by us to what the user had previously
-        # Not using toggle.track due to observed oddities
-        for i in uv_textures:
-            value = i.name == self.lightmap_uvtex_name
-            i.active = value
-            i.active_render = value
-
-        # Indicate we should bake
-        return True
-
     def _prep_for_vcols(self, bo, toggle):
         mesh = bo.data
         modifier = bo.plasma_modifiers.lightmap