generated from stilobique/BlenderTemplate
Rename the main folder
This commit is contained in:
@@ -0,0 +1,90 @@
|
||||
import bpy
|
||||
|
||||
from ..properties.models import Outline
|
||||
|
||||
|
||||
class ConfigBlendScene(bpy.types.Operator):
|
||||
"""
|
||||
This operator init a news blend file, or update it.
|
||||
|
||||
Prepare a set of collection (from the models in `properties/models/Outline`.
|
||||
|
||||
Hierarchy design give that:
|
||||
- Placeholder
|
||||
| ${StaticMesh}
|
||||
| Socket (Optional)
|
||||
- Icon
|
||||
- Game Object
|
||||
"""
|
||||
bl_idname = 'graou.build_scene'
|
||||
bl_label = 'Config or update the outline'
|
||||
|
||||
def __init__(self):
|
||||
self._outline = Outline()
|
||||
|
||||
# Main property
|
||||
self._settings = bpy.context.scene.graou_props
|
||||
self._socket = self._settings.socket_collection
|
||||
|
||||
def execute(self, context):
|
||||
for key, value in self._outline.collections.items():
|
||||
collection = bpy.data.collections.get(key)
|
||||
|
||||
# Make the collection if this item not exist
|
||||
if collection is None:
|
||||
collection = bpy.data.collections.new(value.name)
|
||||
context.scene.collection.children.link(collection)
|
||||
|
||||
# Check if the color are correctly setup, if not update-it
|
||||
if collection.color_tag is not value.color:
|
||||
collection.color_tag = value.color
|
||||
|
||||
# Check child collection color
|
||||
if collection.children:
|
||||
for child in collection.children:
|
||||
if isinstance(child, bpy.types.Collection):
|
||||
child.color_tag = value.color
|
||||
|
||||
# Set or update socket collection
|
||||
if self._socket:
|
||||
self.set_socket_collection()
|
||||
else:
|
||||
self.del_socket_collection()
|
||||
|
||||
return {'FINISHED'}
|
||||
|
||||
def set_socket_collection(self):
|
||||
"""Make or update the socket collection"""
|
||||
col_socket = bpy.data.collections.get(self._outline.socket.name)
|
||||
col_placeholder = self._outline.get_placeholder_collection
|
||||
col_game_object = self._outline.get_game_object_collection
|
||||
|
||||
if col_socket is None:
|
||||
col_socket = bpy.data.collections.new(self._outline.socket.name)
|
||||
|
||||
# Attach to Placeholder and Game Icon if they collection exist
|
||||
if col_placeholder.children.get(self._outline.socket.name) is None:
|
||||
col_placeholder.children.link(col_socket)
|
||||
|
||||
if col_game_object.children.get(self._outline.socket.name) is None:
|
||||
col_game_object.children.link(col_socket)
|
||||
|
||||
# Set the COLOR Tag
|
||||
col_socket.color_tag = self._outline.socket.color
|
||||
|
||||
def del_socket_collection(self):
|
||||
"""Remove the socket collection, however, if the collection doesn't exist, terminate the function"""
|
||||
socket = bpy.data.collections.get(self._outline.socket.name)
|
||||
if socket is not None:
|
||||
bpy.data.collections.remove(socket)
|
||||
|
||||
|
||||
class SetCollectionSocket(bpy.types.Operator):
|
||||
"""Configure a/the collection with socket param"""
|
||||
bl_idname = 'graou.collection.socket_setup'
|
||||
bl_label = 'Set param to be socket functional'
|
||||
|
||||
def __init__(self):
|
||||
self.socket = bpy.data.collections.get('Socket')
|
||||
def execute(self, context):
|
||||
return {'FINISHED'}
|
||||
Reference in New Issue
Block a user