136 lines
4.5 KiB
Python
136 lines
4.5 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: props/sbsar.py
|
|
# brief: Substance Property Groups
|
|
# author Adobe - 3D & Immersive
|
|
# copyright 2023 Adobe Inc. All rights reserved.
|
|
|
|
|
|
import bpy
|
|
from .sbsar_graph import SUBSTANCE_PG_SbsarGraph
|
|
|
|
from ..common import (
|
|
Code_SbsarLoadSuffix,
|
|
ADDON_PACKAGE
|
|
)
|
|
|
|
|
|
def get_graph_items(self, context):
|
|
_graphs = []
|
|
for _idx, _graph in enumerate(self.graphs):
|
|
_item = (_graph.index, _graph.label, "{}:{}:{}".format(_graph.index, _graph.label, _graph.uid))
|
|
_graphs.append(_item)
|
|
if len(_graphs) == 0:
|
|
return [("0", "NONE", "NONE:NONE:NONE")]
|
|
return _graphs
|
|
|
|
|
|
def on_graph_update(self, context):
|
|
pass
|
|
|
|
|
|
def on_suffix_update(self, context):
|
|
if self.suffix == Code_SbsarLoadSuffix.success.value[0]:
|
|
self.loading = Code_SbsarLoadSuffix.success.value[1]
|
|
self.load_success = True
|
|
elif self.suffix == Code_SbsarLoadSuffix.error.value[0]:
|
|
self.loading = Code_SbsarLoadSuffix.error.value[1]
|
|
self.load_success = False
|
|
|
|
|
|
class SUBSTANCE_PG_Sbsar(bpy.types.PropertyGroup):
|
|
uuid: bpy.props.StringProperty(
|
|
name="uuid",
|
|
description="The ID of the Substance 3D *.sbsar file") # noqa
|
|
version: bpy.props.IntProperty(
|
|
name="version",
|
|
description="The Version of the Substance Engine used by the .sbsar file") # noqa
|
|
name: bpy.props.StringProperty(
|
|
name="name",
|
|
description="The name of the Substance *.sbsar file") # noqa
|
|
filename: bpy.props.StringProperty(
|
|
name='filename',
|
|
description='The name of the *.sbsar file') # noqa
|
|
filepath: bpy.props.StringProperty(
|
|
name='filepath',
|
|
description='The path to the *.sbsar file') # noqa
|
|
|
|
graphs: bpy.props.CollectionProperty(type=SUBSTANCE_PG_SbsarGraph)
|
|
graphs_list: bpy.props.EnumProperty(
|
|
name="graphs",
|
|
description="The available graphs of the Substance 3D material", # noqa
|
|
items=get_graph_items,
|
|
update=on_graph_update)
|
|
|
|
suffix: bpy.props.StringProperty(
|
|
name="suffix",
|
|
default=Code_SbsarLoadSuffix.loading.value[0],
|
|
update=on_suffix_update)
|
|
icon: bpy.props.StringProperty(
|
|
name="icon",
|
|
default=Code_SbsarLoadSuffix.loading.value[1],
|
|
update=on_suffix_update)
|
|
|
|
loading: bpy.props.StringProperty(name="loading", default="TEMP") # noqa
|
|
load_success: bpy.props.BoolProperty(default=False)
|
|
|
|
def init(self, sbsar):
|
|
self.uuid = sbsar.uuid
|
|
self.version = sbsar.version
|
|
self.name = sbsar.name
|
|
self.filename = sbsar.filename
|
|
self.filepath = sbsar.filepath
|
|
|
|
_addon_prefs = bpy.context.preferences.addons[ADDON_PACKAGE].preferences
|
|
for _graph in sbsar.graphs:
|
|
_new_graph = self.graphs.add()
|
|
_new_graph.init(_graph, _addon_prefs)
|
|
|
|
def init_shader(self, shader_idx, shader, default_output):
|
|
for _graph in self.graphs:
|
|
_graph.init_shader(shader_idx, shader, default_output)
|
|
|
|
def init_tiling(self, sbsar):
|
|
for _idx, _graph in enumerate(self.graphs):
|
|
_graph.init_tiling(sbsar.graphs[_idx].tiling.get(), sbsar.graphs[_idx].tiling.linked)
|
|
|
|
def init_shader_duplicate(self, sbsar):
|
|
for _idx, _graph in enumerate(self.graphs):
|
|
_graph.init_shader_duplicate(sbsar.graphs[_idx].shaders_list)
|
|
|
|
def init_outputs(self, sbsar):
|
|
for _idx, _graph in enumerate(self.graphs):
|
|
_graph.init_outputs(sbsar.graphs[_idx].outputs)
|
|
|
|
def init_presets(self, sbsar):
|
|
for _idx, _graph in enumerate(self.graphs):
|
|
_graph.init_preset(sbsar.graphs[_idx].presets_list)
|
|
|
|
def get(self):
|
|
_obj = {
|
|
"uuid": self.uuid,
|
|
"version": self.version,
|
|
"name": self.name,
|
|
"filename": self.filename,
|
|
"filepath": self.filepath,
|
|
"graphs": []
|
|
}
|
|
for _graph in self.graphs:
|
|
_obj["graphs"].append(_graph.get())
|
|
|
|
return _obj
|