Add adjust exposure option

Naxela · May 31, 2022 · 5eb694f · 5eb694f
1 parent 71956fe
commit 5eb694f
Show file tree

Hide file tree

Showing 7 changed files with 85 additions and 4 deletions.
diff --git a/addon/operators/__init__.py b/addon/operators/__init__.py
@@ -38,7 +38,8 @@
     imagetools.TLM_ImageUpscale,
     imagetools.TLM_ImageDownscale,
     tlm.TLM_AddGLTFNode,
-    tlm.TLM_ShiftMultiplyLinks
+    tlm.TLM_ShiftMultiplyLinks,
+    tlm.TLM_AdjustExpore
 
 ]
 

diff --git a/addon/operators/tlm.py b/addon/operators/tlm.py
@@ -1385,6 +1385,77 @@ def execute(self, context):
 
         return{'FINISHED'}
 
+class TLM_AdjustExpore(bpy.types.Operator):
+    bl_idname = "tlm.adjust_exposure"
+    bl_label = "Adjust Exposure"
+    bl_description = "Value override in found exposure nodes"
+    bl_options = {'REGISTER', 'UNDO'}
+
+    def execute(self, context):
+
+        if bpy.context.scene.TLM_SceneProperties.tlm_utility_set == "Scene":
+            for obj in bpy.context.scene.objects:
+                if obj.type == "MESH":
+
+                    for slots in obj.material_slots:
+
+                        mat = slots.material
+
+                        nodetree = mat.node_tree
+
+                        if nodetree:
+
+                            for node in nodetree.nodes:
+
+                                if node.name == "Lightmap_Exposure":
+
+                                    print(node.name)
+
+                                    node.inputs[1].default_value = bpy.context.scene.TLM_SceneProperties.tlm_adjust_exposure
+
+        elif bpy.context.scene.TLM_SceneProperties.tlm_utility_set == "Selection":
+            for obj in bpy.context.selected_objects:
+                if obj.type == "MESH":
+
+                    for slots in obj.material_slots:
+
+                        mat = slots.material
+
+                        nodetree = mat.node_tree
+
+                        if nodetree:
+
+                            for node in nodetree.nodes:
+
+                                if node.name == "Lightmap_Exposure":
+
+                                    print(node.name)
+
+                                    node.inputs[1].default_value = bpy.context.scene.TLM_SceneProperties.tlm_adjust_exposure
+
+        else: #Enabled
+            for obj in bpy.context.scene.objects:
+                if obj.type == "MESH":
+                    if obj.TLM_ObjectProperties.tlm_mesh_lightmap_use:
+
+                        for slots in obj.material_slots:
+
+                            mat = slots.material
+
+                            nodetree = mat.node_tree
+
+                            if nodetree:
+
+                                for node in nodetree.nodes:
+
+                                    if node.name == "Lightmap_Exposure":
+
+                                        print(node.name)
+
+                                        node.inputs[1].default_value = bpy.context.scene.TLM_SceneProperties.tlm_adjust_exposure
+
+        return{'FINISHED'}
+
 class TLM_PostAtlasSpecialsMenu(bpy.types.Menu):
     bl_label = "Lightmap"
     bl_idname = "TLM_MT_PostAtlasListSpecials"

diff --git a/addon/panels/scene.py b/addon/panels/scene.py
@@ -523,6 +523,10 @@ def draw(self, context):
             row = layout.row(align=True)
             row.prop(sceneProperties, "tlm_isolate_lightmap_uv")
             row = layout.row(align=True)
+            row.operator("tlm.adjust_exposure")
+            row = layout.row(align=True)
+            row.prop(sceneProperties, "tlm_adjust_exposure")
+            row = layout.row(align=True) 
 
         elif sceneProperties.tlm_utility_context == "NetworkRender":
 

diff --git a/addon/properties/scene.py b/addon/properties/scene.py
@@ -541,6 +541,11 @@ class TLM_SceneProperties(bpy.types.PropertyGroup):
         description="Deletes other UV channels than the lightmap channel", 
         default=False)
 
+    tlm_adjust_exposure : FloatProperty(
+        name="Adjusted exposure", 
+        default=0,
+        description="Value override in found exposure nodes")
+
     tlm_utility_context : EnumProperty(
         items = [('SetBatching', 'Set Batching', 'Set batching options. Allows to set lightmap options for multiple objects.'),
                  ('EnvironmentProbes', 'Environment Probes', 'Options for rendering environment probes. Cubemaps and panoramic HDRs for external engines'),

diff --git a/addon/utility/denoiser/oidn.py b/addon/utility/denoiser/oidn.py
@@ -25,7 +25,7 @@ def check_binary(self):
 
         if oidnPath != "":
 
-            file = os.path.basename(os.path.realpath(oidnPath))
+            file = oidnPath
             filename, file_extension = os.path.splitext(file)
 
 

diff --git a/addon/utility/denoiser/optix.py b/addon/utility/denoiser/optix.py
@@ -24,7 +24,7 @@ def check_binary(self):
 
         if optixPath != "":
 
-            file = os.path.basename(os.path.realpath(optixPath))
+            file = optixPath
             filename, file_extension = os.path.splitext(file)
 
             if(file_extension == ".exe"):

diff --git a/addon/utility/utility.py b/addon/utility/utility.py
@@ -434,7 +434,7 @@ def Unwrap_Lightmap_Group_Xatlas_2_headless_call(obj):
 
     #get the path to xatlas
     #file_path = os.path.dirname(os.path.abspath(__file__))
-    scriptsDir = bpy.utils.user_resource('SCRIPTS', "addons")
+    scriptsDir = bpy.utils.user_resource('SCRIPTS', path="addons")
     file_path = os.path.join(scriptsDir, "blender_xatlas")
     if platform.system() == "Windows":
         xatlas_path = os.path.join(file_path, "xatlas", "xatlas-blender.exe")