armory/blender/traits.py

import shutil
import bpy
import os
import json
from traits_animation import ListAnimationTraitItem
from bpy.types import Menu, Panel, UIList
from bpy.props import *

class ListTraitItem(bpy.types.PropertyGroup):
    # Group of properties representing an item in the list
    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)

    type_prop = bpy.props.EnumProperty(
        items = [('Script', 'Script', 'Script'),
                 ('Nodes', 'Nodes', 'Nodes'),
                 ('Scene Instance', 'Scene Instance', 'Scene Instance'),
                 ('Animation', 'Animation', 'Animation'),
                 ('Camera', 'Camera', 'Camera'),
                 ('Light', 'Light', 'Light'),
                 ('Rigid Body', 'Rigid Body', 'Rigid Body'),
                 
                 ('Mesh Renderer', 'Mesh Renderer', 'Mesh Renderer'),
                 ('Billboard Renderer', 'Billboard Renderer', 'Billboard Renderer'),
                 ('Particles Renderer', 'Particles Renderer', 'Particles Renderer'),
                 ('Skydome Renderer', 'Skydome Renderer', 'Skydome Renderer'),
                 ('Custom Renderer', 'Custom Renderer', 'Custom Renderer')
                 ],
        name = "Type")
        
    default_material_prop = bpy.props.BoolProperty(
           name="Default Material",
           description="A name for this item",
           default=True)

    material_prop = bpy.props.StringProperty(
           name="Material",
           description="A name for this item",
           default="")

    lighting_prop = bpy.props.BoolProperty(
            name="Lighting",
            default=True)

    cast_shadow_prop = bpy.props.BoolProperty(
            name="Cast Shadow",
            default=True)

    receive_shadow_prop = bpy.props.BoolProperty(
            name="Receive Shadow",
            default=True)

    shader_prop = bpy.props.StringProperty(
           name="Shader",
           description="A name for this item",
           default="")

    data_prop = bpy.props.StringProperty(
           name="Data",
           description="A name for this item",
           default="")

    camera_link_prop = bpy.props.StringProperty(
           name="Camera",
           description="A name for this item",
           default="Camera")

    light_link_prop = bpy.props.StringProperty(
           name="Light",
           description="A name for this item",
           default="Lamp")

    default_body_prop = bpy.props.BoolProperty(
           name="Default Body",
           description="A name for this item",
           default=True)

    body_link_prop = bpy.props.StringProperty(
           name="Body",
           description="A name for this item",
           default="")

    custom_shape_prop = bpy.props.BoolProperty(
           name="Custom Shape",
           description="A name for this item",
           default=False)

    custom_shape_type_prop = bpy.props.EnumProperty(
        items = [('Terrain', 'Terrain', 'Terrain'),
                 ('Static Mesh', 'Static Mesh', 'Static Mesh')
                 ],
        name = "Shape")

    shape_size_scale_prop = bpy.props.FloatVectorProperty(
           name = "Size Scale",
           default=[1,1,1],
           size=3, min=0, max=10)

    scene_prop = bpy.props.StringProperty(
           name="Scene",
           description="A name for this item",
           default="")

    class_name_prop = bpy.props.StringProperty(
           name="Class",
           description="A name for this item",
           default="Untitled")

    nodes_name_prop = bpy.props.StringProperty(
           name="Nodes",
           description="A name for this item",
           default="")

    start_track_name_prop = bpy.props.StringProperty(
           name="Start Track",
           description="A name for this item",
           default="")

bpy.utils.register_class(ListTraitItem)


class MY_UL_TraitList(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)

        elif self.layout_type in {'GRID'}:
            layout.alignment = 'CENTER'
            layout.label("", icon = custom_icon)
bpy.utils.register_class(MY_UL_TraitList)

    
def initObjectProperties():
    bpy.types.Object.my_traitlist = bpy.props.CollectionProperty(type = ListTraitItem)
    bpy.types.Object.traitlist_index = bpy.props.IntProperty(name = "Index for my_list", default = 0)
initObjectProperties()


class LIST_OT_TraitNewItem(bpy.types.Operator):
    # Add a new item to the list
    bl_idname = "my_traitlist.new_item"
    bl_label = "Add a new item"

    def execute(self, context):
        bpy.context.object.my_traitlist.add()
        bpy.context.object.traitlist_index += 1
        return{'FINISHED'}


class LIST_OT_TraitDeleteItem(bpy.types.Operator):
    # Delete the selected item from the list
    bl_idname = "my_traitlist.delete_item"
    bl_label = "Deletes an item"

    @classmethod
    def poll(self, context):
        """ Enable if there's something in the list """
        return len(bpy.context.object.my_traitlist) > 0

    def execute(self, context):
        list = bpy.context.object.my_traitlist
        index = bpy.context.object.traitlist_index

        list.remove(index)

        if index > 0:
            index = index - 1

        bpy.context.object.traitlist_index = index
        return{'FINISHED'}


class LIST_OT_TraitMoveItem(bpy.types.Operator):
    # Move an item in the list
    bl_idname = "my_traitlist.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. """
        return len(bpy.context.object.my_traitlist) > 0


    def move_index(self):
        # Move index of an item render queue while clamping it
        index = bpy.context.object.traitlist_index
        list_length = len(bpy.context.object.my_traitlist) - 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):
        list = bpy.context.object.my_traitlist
        index = bpy.context.object.traitlist_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'}


# Menu in tools region
class ToolsTraitsPanel(bpy.types.Panel):
    bl_label = "cycles_traits"
    bl_space_type = "PROPERTIES"
    bl_region_type = "WINDOW"
    bl_context = "object"
 
    def draw(self, context):
        layout = self.layout
        
        obj = bpy.context.object

        rows = 2
        if len(obj.my_traitlist) > 1:
            rows = 4
        
        row = layout.row()
        row.template_list("MY_UL_TraitList", "The_List", obj, "my_traitlist", obj, "traitlist_index", rows=rows)
        
        col = row.column(align=True)
        col.operator("my_traitlist.new_item", icon='ZOOMIN', text="")
        col.operator("my_traitlist.delete_item", icon='ZOOMOUT', text="")#.all = False

        if len(obj.my_traitlist) > 1:
            col.separator()
            col.operator("my_traitlist.move_item", icon='TRIA_UP', text="").direction = 'UP'
            col.operator("my_traitlist.move_item", icon='TRIA_DOWN', text="").direction = 'DOWN'

        if obj.traitlist_index >= 0 and len(obj.my_traitlist) > 0:
            item = obj.my_traitlist[obj.traitlist_index]         
            # Default props
            row = layout.row()
            row.prop(item, "type_prop")

            # Script
            if item.type_prop =='Script':
                item.name = item.class_name_prop
                row = layout.row()
                row.prop(item, "class_name_prop")
            # Nodes
            elif item.type_prop =='Nodes':
                item.name = item.nodes_name_prop
                row = layout.row()
                row.prop_search(item, "nodes_name_prop", bpy.data, "node_groups", "Tree")
            # Camera
            elif item.type_prop == 'Camera':
                item.name = item.type_prop
                row = layout.row()
                row.prop_search(item, "camera_link_prop", bpy.data, "cameras", "Camera")

            # Light
            elif item.type_prop == 'Light':
                item.name = item.type_prop
                row = layout.row()
                row.prop_search(item, "light_link_prop", bpy.data, "lamps", "Light")

            # Rigid body
            elif item.type_prop == 'Rigid Body':
                item.name = item.type_prop
                row = layout.row()
                row.prop(item, "default_body_prop")
                if item.default_body_prop == False:
                    row.prop_search(item, "body_link_prop", bpy.context.scene.rigidbody_world.group, "objects", "")
                row = layout.row()
                row.prop(item, "custom_shape_prop")
                if item.custom_shape_prop == True:
                    row.prop(item, "custom_shape_type_prop")
                row = layout.row()
                row.prop(item, "shape_size_scale_prop")

            # Scene instance
            elif item.type_prop == 'Scene Instance':
                item.name = item.type_prop
                row = layout.row()
                row.prop_search(item, "scene_prop", bpy.data, "scenes", "Scene")
                #row.prop(item, "scene_prop")

            # Animation
            elif item.type_prop == 'Animation':
                item.name = item.type_prop
                row = layout.row()
                row.prop_search(item, "start_track_name_prop", obj, "my_animationtraitlist", "Start Track")
                # List
                animrow = layout.row()
                animrows = 2
                if len(obj.my_animationtraitlist) > 1:
                    animrows = 4
                
                row = layout.row()
                row.template_list("MY_UL_AnimationTraitList", "The_List", obj, "my_animationtraitlist", obj, "animationtraitlist_index", rows=animrows)

                col = row.column(align=True)
                col.operator("my_animationtraitlist.new_item", icon='ZOOMIN', text="")
                col.operator("my_animationtraitlist.delete_item", icon='ZOOMOUT', text="")

                if len(obj.my_animationtraitlist) > 1:
                    col.separator()
                    col.operator("my_animationtraitlist.move_item", icon='TRIA_UP', text="").direction = 'UP'
                    col.operator("my_animationtraitlist.move_item", icon='TRIA_DOWN', text="").direction = 'DOWN'

                if obj.animationtraitlist_index >= 0 and len(obj.my_animationtraitlist) > 0:
                    item = obj.my_animationtraitlist[obj.animationtraitlist_index]         
                    
                    row = layout.row()
                    row.prop(item, "start_prop")
                    row.prop(item, "end_prop")
            # Animation end
            # Mesh renderer
            elif item.type_prop == 'Mesh Renderer':
                item.name = item.type_prop
                row = layout.row()
                row.prop(item, "default_material_prop")
                if item.default_material_prop == False:
                    row.prop_search(item, "material_prop", bpy.data, "materials", "")
                row = layout.row()
                row.prop(item, "lighting_prop")
                row = layout.row()
                row.prop(item, "cast_shadow_prop")
                row = layout.row()
                row.prop(item, "receive_shadow_prop")

            # Custom renderer
            elif item.type_prop == 'Custom Renderer':
                item.name = item.class_name_prop
                row = layout.row()
                row.prop(item, "class_name_prop")
                
                row = layout.row()
                row.prop(item, "default_material_prop")
                if item.default_material_prop == False:
                    row.prop_search(item, "material_prop", bpy.data, "materials", "")
                row = layout.row()
                row.prop(item, "shader_prop")
                row = layout.row()
                row.prop(item, "data_prop")

            # Billboard renderer
            elif item.type_prop == 'Billboard Renderer':
                item.name = item.type_prop
                row = layout.row()
                row.prop(item, "default_material_prop")
                if item.default_material_prop == False:
                    row.prop_search(item, "material_prop", bpy.data, "materials", "")

            # Particles renderer
            elif item.type_prop == 'Particles Renderer':
                item.name = item.type_prop
                row = layout.row()
                row.prop(item, "default_material_prop")
                if item.default_material_prop == False:
                    row.prop_search(item, "material_prop", bpy.data, "materials", "")
                row = layout.row()
                row.prop(item, "data_prop")

            # Skydome renderer
            elif item.type_prop == 'Skydome Renderer':
                item.name = item.type_prop
                row = layout.row()
                row.prop(item, "default_material_prop")
                if item.default_material_prop == False:
                    row.prop_search(item, "material_prop", bpy.data, "materials", "")

# Registration
bpy.utils.register_module(__name__)