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,113 @@
+"""
+Copyright (C) 2023 Adobe.
+This program is free software: you can redistribute it and/or modify
+it under theshad 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: shader/manager.py
+# brief: Shader presets operation manager
+# author Adobe - 3D & Immersive
+# copyright 2023 Adobe Inc. All rights reserved.
+
+
+import os
+import traceback
+
+from .shader import ShaderPreset
+from ..utils import SUBSTANCE_Utils
+from ..factory.shader import SUBSTANCE_ShaderFactory
+from ..common import (
+    Code_Response,
+    ADDON_ROOT
+)
+
+
+class SUBSTANCE_ShaderManager():
+    def __init__(self):
+        self.shaders = {}
+
+    def initialize(self, addon_prefs):
+        _shaders_dir = os.path.join(ADDON_ROOT, "_presets/default").replace('\\', '/')
+        for _filepath in os.listdir(_shaders_dir):
+            # Load the shader presets
+            _result = self.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)
+            self.register(_shader_preset)
+            SUBSTANCE_Utils.log_data("INFO", "Shader Presets [{}] initialized...".format(_filepath))
+        return (Code_Response.success, None)
+
+    def load(self, filename):
+        try:
+            _path = SUBSTANCE_Utils.get_shader_file(filename)
+            _data = SUBSTANCE_Utils.get_json(_path)
+
+            return (Code_Response.success, _data)
+        except Exception:
+            return (Code_Response.shader_preset_load_error, None)
+
+    def register(self, shader):
+        self.shaders[shader.filename] = shader
+        _result = SUBSTANCE_ShaderFactory.register_class(shader)
+        return _result
+
+    def remove_presets(self, shader_presets):
+        try:
+
+            return Code_Response.success
+        except Exception:
+            SUBSTANCE_Utils.log_data("ERROR", "Exception - Shader Preset removal error:")
+            SUBSTANCE_Utils.log_traceback(traceback.format_exc())
+            return Code_Response.shader_preset_remove_error
+
+    def save_presets(self, shader_preset):
+        try:
+            _default_shader_file = os.path.join(
+                ADDON_ROOT,
+                "_presets/default/" + shader_preset["filename"]).replace('\\', '/')
+            _custom_shader_file = os.path.join(
+                ADDON_ROOT,
+                "_presets/custom/" + shader_preset["filename"]).replace('\\', '/')
+            _custom_shader_path = os.path.join(
+                ADDON_ROOT,
+                "_presets/custom").replace('\\', '/')
+
+            # Read Original Shader Preset
+            if not os.path.exists(_default_shader_file):
+                return Code_Response.shader_preset_default_not_exist_error
+
+            _data = SUBSTANCE_Utils.get_json(_default_shader_file)
+
+            _data["inputs"] = shader_preset["inputs"]
+            _data["outputs"] = shader_preset["outputs"]
+
+            # Write Custom Shader Preset
+            if not os.path.exists(_custom_shader_path):
+                os.mkdir(_custom_shader_path)
+
+            SUBSTANCE_Utils.set_json(_custom_shader_file, _data)
+
+            return Code_Response.success
+
+        except Exception:
+            SUBSTANCE_Utils.log_data("ERROR", "Exception - Shader Preset saving error:")
+            SUBSTANCE_Utils.log_traceback(traceback.format_exc())
+            return Code_Response.shader_preset_save_error