2025-07-01

2026-03-17 14:30:01 -06:00
parent f9a22056dd
commit 62b5978595
4579 changed files with 1257472 additions and 0 deletions
@@ -0,0 +1,151 @@
+"""
+Copyright (C) 2023 Adobe.
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+"""
+
+# file: ops/shader.py
+# brief: Shader Operators
+# author Adobe - 3D & Immersive
+# copyright 2023 Adobe Inc. All rights reserved.
+
+
+import bpy
+import os
+
+from ..common import ADDON_ROOT, ADDON_PACKAGE, Code_Response
+from ..api import SUBSTANCE_Api
+from ..utils import SUBSTANCE_Utils
+from ..shader.shader import ShaderPreset
+
+
+class SUBSTANCE_OT_LoadShaderPresets(bpy.types.Operator):
+    bl_idname = 'substance.load_shader_presets'
+    bl_label = 'Initialize shader preset'
+    bl_description = "Initialize the shader presets"
+
+    def execute(self, context):
+        _addon_prefs = context.preferences.addons[ADDON_PACKAGE].preferences
+        _addon_prefs.shaders.clear()
+
+        _shaders_dir = os.path.join(ADDON_ROOT, "_presets/default").replace('\\', '/')
+        for _filepath in os.listdir(_shaders_dir):
+            # Load the shader presets
+            _result = SUBSTANCE_Api.shader_load(_filepath)
+            if _result[0] != Code_Response.success:
+                SUBSTANCE_Utils.log_data(
+                    "ERROR",
+                    "Substance file [{}] could not be loaded".format(
+                        _filepath),
+                    display=True)
+                continue
+            _shader_preset = ShaderPreset(_filepath, _result[1])
+            _new_shader = _addon_prefs.shaders.add()
+            _new_shader.init(_shader_preset)
+            SUBSTANCE_Api.shader_register(_shader_preset)
+            SUBSTANCE_Utils.log_data("INFO", "Shader Presets [{}] initialized...".format(_filepath))
+        return {'FINISHED'}
+
+
+class SUBSTANCE_OT_RemoveShaderPresets(bpy.types.Operator):
+    bl_idname = 'substance.remove_shader_presets'
+    bl_label = 'Remove shader preset'
+    bl_description = "Remove the shader presets"
+
+    def execute(self, context):
+        _addon_prefs = context.preferences.addons[ADDON_PACKAGE].preferences
+
+        _result = SUBSTANCE_Api.shader_presets_remove(_addon_prefs.shaders)
+        if _result != Code_Response.success:
+            SUBSTANCE_Utils.log_data("ERROR", "Shader Presets could not be removed...")
+            _addon_prefs.shader_presets.clear()
+            return {'FINISHED'}
+
+        SUBSTANCE_Utils.log_data("INFO", "Shader Presets fully removed...")
+        _addon_prefs.shaders.clear()
+        return {'FINISHED'}
+
+
+class SUBSTANCE_OT_SaveShaderPresets(bpy.types.Operator):
+    bl_idname = 'substance.save_shader_presets'
+    bl_label = 'Save shader preset'
+    bl_description = "Save the shader presets"
+
+    def execute(self, context):
+        _addon_prefs = context.preferences.addons[ADDON_PACKAGE].preferences
+
+        for _shader_preset in _addon_prefs.shaders:
+            if _shader_preset.modified:
+                _inputs = getattr(context.scene, _shader_preset.inputs_class_name)
+
+                _outputs = {}
+                for _output in _shader_preset.outputs:
+                    _outputs[_output.id] = _output.to_json()
+
+                _obj = {
+                    "label": _shader_preset.label,
+                    "filename": _shader_preset.filename,
+                    "inputs": _inputs.get(),
+                    "outputs": _outputs,
+                }
+
+                _result = SUBSTANCE_Api.shader_presets_save(_obj)
+                if _result != Code_Response.success:
+                    SUBSTANCE_Utils.log_data(
+                        "ERROR",
+                        "Shader Presets [{}] could not be saved...[{}]".format(_obj["label"], _result))
+                SUBSTANCE_Utils.log_data("INFO", "Shader Presets [{}] saved...".format(_obj["label"]))
+
+        return {'FINISHED'}
+
+
+class SUBSTANCE_OT_ResetShaderPreset(bpy.types.Operator):
+    bl_idname = 'substance.reset_shader_preset'
+    bl_label = 'Reset shader preset'
+    bl_description = "Reset the shader preset"
+
+    @classmethod
+    def poll(cls, context):
+        _addons_prefs, _shader_idx, _shader = SUBSTANCE_Utils.get_selected_shader(context)
+        return _shader.modified
+
+    def execute(self, context):
+        _addons_prefs, _shader_idx, _shader = SUBSTANCE_Utils.get_selected_shader(context)
+
+        _inputs = getattr(context.scene, _shader.inputs_class_name)
+        _inputs.reset()
+
+        # Load the shader presets
+        _result = SUBSTANCE_Api.shader_load(_shader.filename)
+        if _result[0] != Code_Response.success:
+            SUBSTANCE_Utils.log_data(
+                "ERROR",
+                "Substance file [{}] could not be loaded".format(
+                    _shader.filename),
+                display=True)
+            return {'FINISHED'}
+
+        _outputs = _result[1]["outputs"]
+        for _output in _shader.outputs:
+            _output.enabled = _outputs[_output.id]["enabled"]
+            _output.colorspace = _outputs[_output.id]["colorspace"]
+            _output.format = _outputs[_output.id]["format"]
+            _output.bitdepth = _outputs[_output.id]["bitdepth"]
+
+        _shader.modified = False
+
+        SUBSTANCE_Utils.log_data(
+            "INFO",
+            "Shader Presets [{}] is back to default values".format(_shader.label),
+            display=True)
+        return {'FINISHED'}