152 lines
5.6 KiB
Python
152 lines
5.6 KiB
Python
"""
|
|
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'}
|