115 lines
4.3 KiB
Python
115 lines
4.3 KiB
Python
import bpy
|
|
#import collections
|
|
#import inspect
|
|
from .utils import get_operator_from_id, get_mat
|
|
|
|
import re
|
|
|
|
|
|
|
|
|
|
class RP_PT_picker_maker_panel(bpy.types.Panel):
|
|
bl_label = 'Rig UI'
|
|
bl_category = 'RIG Tools'
|
|
bl_space_type = 'VIEW_3D'
|
|
bl_region_type = 'UI'
|
|
|
|
@classmethod
|
|
def poll(cls, context):
|
|
return context.object
|
|
|
|
def draw(self, context):
|
|
ob = context.object
|
|
scn = context.scene
|
|
|
|
layout = self.layout
|
|
col = layout.column(align=False)
|
|
col.prop_search(scn.rig_picker, 'rig', scn, 'objects', text='Rig ')
|
|
col.prop_search(scn.rig_picker, 'canvas', scn, 'objects', text='canvas ')
|
|
col.prop_search(scn.rig_picker, 'symmetry', scn, 'objects', text='Symmetry ')
|
|
|
|
if ob.type == 'ARMATURE':
|
|
row = layout.row(align=True)
|
|
is_packed = ('picker' in ob.data.rig_picker.keys())
|
|
sub_row = row.row(align=True)
|
|
sub_row.prop(ob.data.rig_picker, 'source', text='Picker')
|
|
sub_row.enabled = not is_packed
|
|
row.operator('rigpicker.pack_picker', icon='PACKAGE' if is_packed else 'UGLYPACKAGE', text='')
|
|
|
|
|
|
col = layout.column(align=True)
|
|
row = col.row(align=True)
|
|
row.operator('rigpicker.create_shape', icon='MESH_DATA', text='Create shape')
|
|
row.operator('rigpicker.mirror_shape', icon='ARROW_LEFTRIGHT', text='Mirror shape')
|
|
col.operator('rigpicker.name_from_bone', icon='SORTALPHA' , text='Name from bones')
|
|
|
|
|
|
if ob.type !='ARMATURE':
|
|
box = layout.box()
|
|
col = box.column(align=False)
|
|
|
|
material_row = col.row(align=True)
|
|
material_row.operator('rigpicker.remove_mat', icon='REMOVE', text='')
|
|
material_row.operator('rigpicker.add_mat', icon='ADD', text='')
|
|
mat = False
|
|
if ob.type in ('MESH', 'CURVE', 'FONT') and ob.data.materials:
|
|
mat = get_mat(ob)
|
|
if mat and mat.node_tree:
|
|
emission_nodes = [n for n in mat.node_tree.nodes if n.type =='EMISSION']
|
|
if emission_nodes:
|
|
material_row.prop(emission_nodes[0].inputs[0], 'default_value', text='')
|
|
mat = True
|
|
if not mat:
|
|
material_row.label(text='No Material')
|
|
material_row.operator('rigpicker.eyedropper_mat', icon='EYEDROPPER', text='')
|
|
|
|
shape_type_row = col.row(align=True)
|
|
shape_type_row.prop(ob.rig_picker,'shape_type',expand = True)
|
|
shape_type_row.operator('rigpicker.select_shape_type', text='', icon='RESTRICT_SELECT_OFF')
|
|
|
|
|
|
if ob.rig_picker.shape_type == 'OPERATOR':
|
|
|
|
op_row = col.row(align=True)
|
|
op_row.prop(ob.rig_picker, 'operator', text='')
|
|
op_row.operator('rigpicker.operator_selector', text='', icon='COLLAPSEMENU')
|
|
|
|
if ob.rig_picker.operator:
|
|
col.prop(ob.rig_picker, 'name', text='Tooltip')
|
|
col.prop(ob.rig_picker,'shortcut', text='Shortcut')
|
|
|
|
else:
|
|
col.prop(ob.rig_picker, 'name', text='Tooltip')
|
|
|
|
'''
|
|
if ob.rig_picker.operator:
|
|
op = get_operator_from_id(ob.rig_picker.idname)
|
|
if op:
|
|
doc = re.findall(r'\(([^\)]+)\)', op.__doc__)[0]
|
|
else:
|
|
doc = 'Operator not found'
|
|
|
|
col.prop(ob.rig_picker,'name', text='Tooltip')
|
|
if op:
|
|
col.prop(ob.rig_picker,'shortcut', text='Shortcut')
|
|
col.prop(ob.rig_picker,'arguments', text='')
|
|
col.label(text=doc)
|
|
else:
|
|
col.label(text=doc)
|
|
'''
|
|
|
|
|
|
if ob.rig_picker.shape_type == 'BONE':
|
|
if scn.rig_picker.rig:
|
|
col.prop_search(ob.rig_picker, 'name', scn.rig_picker.rig.pose, 'bones', text='Bone')
|
|
|
|
#layout.separator()
|
|
layout.prop(scn.rig_picker, 'destination', text='Filepath')
|
|
layout.operator('rigpicker.save_picker', icon='PASTEDOWN', text='Save Picker')
|
|
|
|
|
|
classes = (
|
|
RP_PT_picker_maker_panel,
|
|
)
|
|
|
|
register, unregister = bpy.utils.register_classes_factory(classes) |