armory/blender/arm/props_tilesheet.py
Moritz Brückner 4f25af45ba Cleanup
2021-01-11 19:56:40 +01:00

283 lines
9.3 KiB
Python

import bpy
from bpy.props import *
class ArmTilesheetActionListItem(bpy.types.PropertyGroup):
name: StringProperty(
name="Name",
description="A name for this item",
default="Untitled")
start_prop: IntProperty(
name="Start",
description="A name for this item",
default=0)
end_prop: IntProperty(
name="End",
description="A name for this item",
default=0)
loop_prop: BoolProperty(
name="Loop",
description="A name for this item",
default=True)
class ARM_UL_TilesheetActionList(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, "name", text="", emboss=False, icon=custom_icon)
elif self.layout_type in {'GRID'}:
layout.alignment = 'CENTER'
layout.label(text="", icon = custom_icon)
class ArmTilesheetActionListNewItem(bpy.types.Operator):
# Add a new item to the list
bl_idname = "arm_tilesheetactionlist.new_item"
bl_label = "Add a new item"
def execute(self, context):
wrd = bpy.data.worlds['Arm']
trait = wrd.arm_tilesheetlist[wrd.arm_tilesheetlist_index]
trait.arm_tilesheetactionlist.add()
trait.arm_tilesheetactionlist_index = len(trait.arm_tilesheetactionlist) - 1
return{'FINISHED'}
class ArmTilesheetActionListDeleteItem(bpy.types.Operator):
"""Delete the selected item from the list"""
bl_idname = "arm_tilesheetactionlist.delete_item"
bl_label = "Deletes an item"
@classmethod
def poll(self, context):
"""Enable if there's something in the list"""
wrd = bpy.data.worlds['Arm']
if len(wrd.arm_tilesheetlist) == 0:
return False
trait = wrd.arm_tilesheetlist[wrd.arm_tilesheetlist_index]
return len(trait.arm_tilesheetactionlist) > 0
def execute(self, context):
wrd = bpy.data.worlds['Arm']
trait = wrd.arm_tilesheetlist[wrd.arm_tilesheetlist_index]
list = trait.arm_tilesheetactionlist
index = trait.arm_tilesheetactionlist_index
list.remove(index)
if index > 0:
index = index - 1
trait.arm_tilesheetactionlist_index = index
return{'FINISHED'}
class ArmTilesheetActionListMoveItem(bpy.types.Operator):
"""Move an item in the list"""
bl_idname = "arm_tilesheetactionlist.move_item"
bl_label = "Move an item in the list"
bl_options = {'INTERNAL'}
direction: EnumProperty(
items=(
('UP', 'Up', ""),
('DOWN', 'Down', "")
))
@classmethod
def poll(self, context):
"""Enable if there's something in the list"""
wrd = bpy.data.worlds['Arm']
if len(wrd.arm_tilesheetlist) == 0:
return False
trait = wrd.arm_tilesheetlist[wrd.arm_tilesheetlist_index]
return len(trait.arm_tilesheetactionlist) > 0
def move_index(self):
# Move index of an item render queue while clamping it
wrd = bpy.data.worlds['Arm']
trait = wrd.arm_tilesheetlist[wrd.arm_tilesheetlist_index]
index = trait.arm_tilesheetactionlist_index
list_length = len(trait.arm_tilesheetactionlist) - 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))
trait.arm_tilesheetactionlist.move(index, new_index)
trait.arm_tilesheetactionlist_index = new_index
def execute(self, context):
wrd = bpy.data.worlds['Arm']
trait = wrd.arm_tilesheetlist[wrd.arm_tilesheetlist_index]
list = trait.arm_tilesheetactionlist
index = trait.arm_tilesheetactionlist_index
if self.direction == 'DOWN':
neighbor = index + 1
self.move_index()
elif self.direction == 'UP':
neighbor = index - 1
self.move_index()
else:
return{'CANCELLED'}
return{'FINISHED'}
class ArmTilesheetListItem(bpy.types.PropertyGroup):
name: StringProperty(
name="Name",
description="A name for this item",
default="Untitled")
tilesx_prop: IntProperty(
name="Tiles X",
description="A name for this item",
default=0)
tilesy_prop: IntProperty(
name="Tiles Y",
description="A name for this item",
default=0)
framerate_prop: FloatProperty(
name="Frame Rate",
description="A name for this item",
default=4.0)
arm_tilesheetactionlist: CollectionProperty(type=ArmTilesheetActionListItem)
arm_tilesheetactionlist_index: IntProperty(name="Index for arm_tilesheetactionlist", default=0)
class ARM_UL_TilesheetList(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, "name", text="", emboss=False, icon=custom_icon)
elif self.layout_type in {'GRID'}:
layout.alignment = 'CENTER'
layout.label(text="", icon=custom_icon)
class ArmTilesheetListNewItem(bpy.types.Operator):
"""Add a new item to the list"""
bl_idname = "arm_tilesheetlist.new_item"
bl_label = "Add a new item"
def execute(self, context):
wrd = bpy.data.worlds['Arm']
wrd.arm_tilesheetlist.add()
wrd.arm_tilesheetlist_index = len(wrd.arm_tilesheetlist) - 1
return{'FINISHED'}
class ArmTilesheetListDeleteItem(bpy.types.Operator):
"""Delete the selected item from the list"""
bl_idname = "arm_tilesheetlist.delete_item"
bl_label = "Deletes an item"
@classmethod
def poll(self, context):
""" Enable if there's something in the list """
wrd = bpy.data.worlds['Arm']
return len(wrd.arm_tilesheetlist) > 0
def execute(self, context):
wrd = bpy.data.worlds['Arm']
list = wrd.arm_tilesheetlist
index = wrd.arm_tilesheetlist_index
list.remove(index)
if index > 0:
index = index - 1
wrd.arm_tilesheetlist_index = index
return{'FINISHED'}
class ArmTilesheetListMoveItem(bpy.types.Operator):
"""Move an item in the list"""
bl_idname = "arm_tilesheetlist.move_item"
bl_label = "Move an item in the list"
bl_options = {'INTERNAL'}
direction: EnumProperty(
items=(
('UP', 'Up', ""),
('DOWN', 'Down', "")
))
@classmethod
def poll(self, context):
""" Enable if there's something in the list. """
wrd = bpy.data.worlds['Arm']
return len(wrd.arm_tilesheetlist) > 0
def move_index(self):
# Move index of an item render queue while clamping it
wrd = bpy.data.worlds['Arm']
index = wrd.arm_tilesheetlist_index
list_length = len(wrd.arm_tilesheetlist) - 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))
wrd.arm_tilesheetlist.move(index, new_index)
wrd.arm_tilesheetlist_index = new_index
def execute(self, context):
wrd = bpy.data.worlds['Arm']
list = wrd.arm_tilesheetlist
index = wrd.arm_tilesheetlist_index
if self.direction == 'DOWN':
neighbor = index + 1
self.move_index()
elif self.direction == 'UP':
neighbor = index - 1
self.move_index()
else:
return{'CANCELLED'}
return{'FINISHED'}
def register():
bpy.utils.register_class(ArmTilesheetActionListItem)
bpy.utils.register_class(ARM_UL_TilesheetActionList)
bpy.utils.register_class(ArmTilesheetActionListNewItem)
bpy.utils.register_class(ArmTilesheetActionListDeleteItem)
bpy.utils.register_class(ArmTilesheetActionListMoveItem)
bpy.utils.register_class(ArmTilesheetListItem)
bpy.utils.register_class(ARM_UL_TilesheetList)
bpy.utils.register_class(ArmTilesheetListNewItem)
bpy.utils.register_class(ArmTilesheetListDeleteItem)
bpy.utils.register_class(ArmTilesheetListMoveItem)
bpy.types.World.arm_tilesheetlist = CollectionProperty(type=ArmTilesheetListItem)
bpy.types.World.arm_tilesheetlist_index = IntProperty(name="Index for arm_tilesheetlist", default=0)
def unregister():
bpy.utils.unregister_class(ArmTilesheetListItem)
bpy.utils.unregister_class(ARM_UL_TilesheetList)
bpy.utils.unregister_class(ArmTilesheetListNewItem)
bpy.utils.unregister_class(ArmTilesheetListDeleteItem)
bpy.utils.unregister_class(ArmTilesheetListMoveItem)
bpy.utils.unregister_class(ArmTilesheetActionListItem)
bpy.utils.unregister_class(ARM_UL_TilesheetActionList)
bpy.utils.unregister_class(ArmTilesheetActionListNewItem)
bpy.utils.unregister_class(ArmTilesheetActionListDeleteItem)
bpy.utils.unregister_class(ArmTilesheetActionListMoveItem)