2025-12-01

2026-03-17 14:58:51 -06:00
parent 183e865f8b
commit 4b82b57113
6846 changed files with 954887 additions and 162606 deletions
@@ -0,0 +1,172 @@
+# SPDX-FileCopyrightText: 2025 Blender Studio Tools Authors
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+import bpy
+import json
+from . import categories
+from . import utils
+
+class LOR_OT_find_settings(bpy.types.Operator):
+    """
+    """
+    bl_idname = "lighting_overrider.find_settings"
+    bl_label = "Find JSON"
+    bl_description = "Find settings datablock for the current sequence or shot as it is referenced in the scene properties"
+    bl_options = {"REGISTER", "UNDO"}
+    
+    def execute(self, context):
+        meta_settings = context.scene.LOR_Settings
+        settings = utils.get_settings(meta_settings)
+        
+        sequence_settings_db, shot_settings_db = utils.find_settings(bpy.context)
+        
+        if meta_settings.settings_toggle == 'SEQUENCE':
+            meta_settings.sequence_settings.text_datablock = sequence_settings_db
+        elif meta_settings.settings_toggle == 'SHOT':
+            meta_settings.shot_settings.text_datablock = shot_settings_db
+        
+        return {'FINISHED'}
+        
+class LOR_OT_read_settings(bpy.types.Operator):
+    """
+    """
+    bl_idname = "lighting_overrider.read_settings"
+    bl_label = "Read JSON"
+    bl_description = "Read settings from specified text datablock"
+    bl_options = {"REGISTER", "UNDO"}
+    
+    @classmethod
+    def poll(cls, context):
+        meta_settings = context.scene.LOR_Settings
+        settings = utils.get_settings(meta_settings)
+        return bool(settings.text_datablock)
+    
+    def execute(self, context):
+        meta_settings = context.scene.LOR_Settings
+        settings = utils.get_settings(meta_settings)
+        
+        text = settings.text_datablock
+        
+        if not text.is_in_memory:
+            override = context.copy()
+            area_type = override['area'].type
+            override['area'].type = 'TEXT_EDITOR'
+            override['region'] = override['area'].regions[0]
+            override['edit_text'] = text
+            with context.temp_override(**override):
+               bpy.ops.text.reload()
+            override['area'].type = area_type
+        
+        data = read_data_from_json(text)
+        
+        unpack_settings_data(settings, data)
+        settings.is_dirty = False
+        
+        return {'FINISHED'}
+
+class LOR_OT_write_settings(bpy.types.Operator):
+    """
+    """
+    bl_idname = "lighting_overrider.write_settings"
+    bl_label = "Write JSON"
+    bl_description = "Write settings to specified text datablock"
+    bl_options = {"REGISTER", "UNDO"}
+    
+    @classmethod
+    def poll(cls, context):
+        meta_settings = context.scene.LOR_Settings
+        settings = utils.get_settings(meta_settings)
+        return bool(settings.text_datablock)
+    
+    def execute(self, context):
+        meta_settings = context.scene.LOR_Settings
+        settings = utils.get_settings(meta_settings)
+        
+        text = settings.text_datablock
+    
+        data = pack_settings_data(settings)
+        
+        write_data_to_json( text, data)
+        
+        if not text.is_in_memory:
+            override = context.copy()
+            area_type = override['area'].type
+            override['area'].type = 'TEXT_EDITOR'
+            override['region'] = override['area'].regions[0]
+            override['edit_text'] = text
+            with context.temp_override(**override):
+                bpy.ops.text.save()
+            override['area'].type = area_type
+        settings.is_dirty = False
+        
+        return {'FINISHED'}
+
+def pack_settings_data(settings) -> dict:
+    data = {}
+
+    for cat in categories.modules:
+        cat_name = cat.__name__.split(".")[-1]
+        data[cat_name] = cat.settings_as_dict(settings)
+    
+    return data
+
+def unpack_settings_data(settings, data):
+    for cat in categories.modules:
+        cat_name = cat.__name__.split(".")[-1]
+        if not cat_name in data.keys():
+            continue
+        cat.load_settings(settings, data[cat_name])
+    
+    return
+
+def cleanup_json(string):
+    
+    counter = 0
+    open_brackets = 0
+    remove = []
+    for i, char in enumerate(string):
+        #if char == '{':
+        #    counter += 1
+        #    continue
+        #if counter < 3:
+        #    continue
+        if char == '[':
+            open_brackets += 1
+        elif char == ']':
+            open_brackets -= 1
+        if open_brackets:
+            if char == '\n':
+                remove += [i]
+            elif char == ' ' and string[i+1] == ' ':
+                remove += [i]
+    counter = 0
+    for i in remove:
+        i -= counter
+        string = string[:i] + string[i+1:]
+        counter += 1
+    
+    return string
+
+def write_data_to_json(text: bpy.types.Text, data: dict, partial=None):
+    text.clear()
+    text.write(cleanup_json(json.dumps(data, separators=(',', ':'), indent=4)))
+    return
+
+def read_data_from_json(text: bpy.types.Text):
+    return json.loads(text.as_string()) if text.as_string() else json.loads('{}')
+
+
+classes = (
+    LOR_OT_find_settings,
+    LOR_OT_read_settings,
+    LOR_OT_write_settings,
+)
+
+def register():
+    for c in classes:
+        bpy.utils.register_class(c)
+
+def unregister():
+    for c in classes:
+        bpy.utils.unregister_class(c)