137 lines
4.3 KiB
Python
137 lines
4.3 KiB
Python
import shutil
|
|
import bpy
|
|
import os
|
|
import json
|
|
from bpy.types import Menu, Panel, UIList
|
|
from bpy.props import *
|
|
|
|
class ArmLibraryListItem(bpy.types.PropertyGroup):
|
|
name = bpy.props.StringProperty(
|
|
name="Name",
|
|
description="A name for this item",
|
|
default="Untitled")
|
|
|
|
enabled_prop = bpy.props.BoolProperty(
|
|
name="",
|
|
description="A name for this item",
|
|
default=True)
|
|
|
|
class ArmLibraryList(bpy.types.UIList):
|
|
def draw_item(self, context, layout, data, item, icon, active_data, active_propname, index):
|
|
# We could write some code to decide which icon to use here...
|
|
custom_icon = 'OBJECT_DATAMODE'
|
|
|
|
# Make sure your code supports all 3 layout types
|
|
if self.layout_type in {'DEFAULT', 'COMPACT'}:
|
|
layout.prop(item, "enabled_prop")
|
|
#layout.label(item.name, icon = custom_icon)
|
|
layout.prop(item, "name", text="", emboss=False, icon=custom_icon)
|
|
|
|
elif self.layout_type in {'GRID'}:
|
|
layout.alignment = 'CENTER'
|
|
layout.label("", icon = custom_icon)
|
|
|
|
class ArmLibraryListNewItem(bpy.types.Operator):
|
|
# Add a new item to the list
|
|
bl_idname = "arm_librarylist.new_item"
|
|
bl_label = "Add a new item"
|
|
|
|
def execute(self, context):
|
|
trait = bpy.data.worlds['Arm']
|
|
trait.arm_librarylist.add()
|
|
trait.arm_librarylist_index = len(trait.arm_librarylist) - 1
|
|
return{'FINISHED'}
|
|
|
|
|
|
class ArmLibraryListDeleteItem(bpy.types.Operator):
|
|
# Delete the selected item from the list
|
|
bl_idname = "arm_librarylist.delete_item"
|
|
bl_label = "Deletes an item"
|
|
|
|
@classmethod
|
|
def poll(self, context):
|
|
""" Enable if there's something in the list """
|
|
trait = bpy.data.worlds['Arm']
|
|
return len(trait.arm_librarylist) > 0
|
|
|
|
def execute(self, context):
|
|
trait = bpy.data.worlds['Arm']
|
|
list = trait.arm_librarylist
|
|
index = trait.arm_librarylist_index
|
|
|
|
list.remove(index)
|
|
|
|
if index > 0:
|
|
index = index - 1
|
|
|
|
trait.arm_librarylist_index = index
|
|
return{'FINISHED'}
|
|
|
|
|
|
class ArmLibraryListMoveItem(bpy.types.Operator):
|
|
# Move an item in the list
|
|
bl_idname = "arm_librarylist.move_item"
|
|
bl_label = "Move an item in the list"
|
|
direction = bpy.props.EnumProperty(
|
|
items=(
|
|
('UP', 'Up', ""),
|
|
('DOWN', 'Down', ""),))
|
|
|
|
@classmethod
|
|
def poll(self, context):
|
|
""" Enable if there's something in the list. """
|
|
trait = bpy.data.worlds['Arm']
|
|
return len(trait.arm_librarylist) > 0
|
|
|
|
|
|
def move_index(self):
|
|
# Move index of an item render queue while clamping it
|
|
trait = bpy.data.worlds['Arm']
|
|
index = trait.arm_librarylist_index
|
|
list_length = len(trait.arm_librarylist) - 1
|
|
new_index = 0
|
|
|
|
if self.direction == 'UP':
|
|
new_index = index - 1
|
|
elif self.direction == 'DOWN':
|
|
new_index = index + 1
|
|
|
|
new_index = max(0, min(new_index, list_length))
|
|
index = new_index
|
|
|
|
|
|
def execute(self, context):
|
|
trait = bpy.data.worlds['Arm']
|
|
list = trait.arm_librarylist
|
|
index = trait.arm_librarylist_index
|
|
|
|
if self.direction == 'DOWN':
|
|
neighbor = index + 1
|
|
#queue.move(index,neighbor)
|
|
self.move_index()
|
|
|
|
elif self.direction == 'UP':
|
|
neighbor = index - 1
|
|
#queue.move(neighbor, index)
|
|
self.move_index()
|
|
else:
|
|
return{'CANCELLED'}
|
|
return{'FINISHED'}
|
|
|
|
def register():
|
|
bpy.utils.register_class(ArmLibraryListItem)
|
|
bpy.utils.register_class(ArmLibraryList)
|
|
bpy.utils.register_class(ArmLibraryListNewItem)
|
|
bpy.utils.register_class(ArmLibraryListDeleteItem)
|
|
bpy.utils.register_class(ArmLibraryListMoveItem)
|
|
|
|
bpy.types.World.arm_librarylist = bpy.props.CollectionProperty(type=ArmLibraryListItem)
|
|
bpy.types.World.arm_librarylist_index = bpy.props.IntProperty(name="Library index", default=0)
|
|
|
|
def unregister():
|
|
bpy.utils.unregister_class(ArmLibraryListItem)
|
|
bpy.utils.unregister_class(ArmLibraryList)
|
|
bpy.utils.unregister_class(ArmLibraryListNewItem)
|
|
bpy.utils.unregister_class(ArmLibraryListDeleteItem)
|
|
bpy.utils.unregister_class(ArmLibraryListMoveItem)
|