-
Daniel Salazar authoredDaniel Salazar authored
ui_layer_manager.py 27.20 KiB
# ##### BEGIN GPL LICENSE BLOCK #####
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# ##### END GPL LICENSE BLOCK #####
# <pep8 compliant>
#
bl_info = {
"name": "Layer Management",
"author": "Alfonso Annarumma",
"version": (1, 5,1),
"blender": (2, 70, 0),
"location": "View3D > Properties panel > Layer Management",
"warning": "",
"description": "Display and Edit Layer Name",
"wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"
"Scripts/3D_interaction/layer_manager",
"tracker_url": "http://projects.blender.org/tracker/index.php?"
"func=detail&aid=32472",
"category": "3D View"}
import bpy
from bpy.types import Menu, Panel, UIList
from bpy.props import StringProperty, BoolProperty, IntProperty, CollectionProperty, BoolVectorProperty
EDIT = ["EDIT_MESH", "EDIT_CURVE", "EDIT_SURFACE", "EDIT_METABALL", "EDIT_TEXT", "EDIT_ARMATURE"]
class LayerGroups(bpy.types.PropertyGroup):
toggle = BoolProperty(name="", default=False)
wire = BoolProperty(name="", default=False)
lock = BoolProperty(name="", default=False)
layer_groups = BoolVectorProperty(name="Layer Groups", default = ([False]*20), size=20, subtype='LAYER')
# A list of identifiers (colon-separated strings) which property’s controls should be displayed
# in a template_list.
# Note that the order is respected.
#template_list_controls = StringProperty(default="toggle", options={"HIDDEN"})
bpy.utils.register_class(LayerGroups)
bpy.types.Scene.layergroups = CollectionProperty(type=LayerGroups)
# Unused, but this is needed for the TemplateList to work…
bpy.types.Scene.layergroups_index = IntProperty(default=-1)
class RemoveLayerGroup(bpy.types.Operator):
'''Tooltip'''
bl_idname = "object.layergroup_remove"
bl_label = "Remove select Layer Group"
index_group = bpy.props.IntProperty()
@classmethod
def poll(cls, context):
return context.scene is not None
def execute(self, context):
scene = context.scene
index_group = self.index_group
scene.layergroups.remove(index_group)
if index_group > 0:
scene.layergroups_index = index_group - 1
return {'FINISHED'}
class AddLayerGroup(bpy.types.Operator):
'''Tooltip'''
bl_idname = "object.layergroup_add"
bl_label = "Add select Layer group"
index = bpy.props.IntProperty()
layer = layer = BoolVectorProperty(name="Layer", default = ([False]*20), size=20)
@classmethod
def poll(cls, context):
return context.scene is not None
def execute(self, context):
scene = context.scene
layergroups = scene.layergroups
index = self.index
layer = self.layer
item = layergroups.add()
index_number= str(index)
if len(index_number)==2:
index_number = "0"+index_number
if len(index_number)==3:
index_number = index_number
else:
index_number = "00"+index_number
item.name= 'LayerGroup.'+index_number
#item.use=True
scene.layergroups_index = index
scene.layergroups[index].layer_groups = layer
return {'FINISHED'}
class LayerToggle(bpy.types.Operator):
'''Visualize this Layer, Shift-Click to select multiple layers'''
bl_idname = "object.layertoggle"
bl_label = "Visualize this layer"
#prop definition
#layer number
layerN = IntProperty()
spacecheck = BoolProperty()
index_group = IntProperty()
@classmethod
def poll(cls, context):
return context.scene
def execute(self, context):
spacecheck = self.spacecheck
scene = context.scene
layerN = self.layerN
if spacecheck:
space = context.area.spaces.active
else:
space = context.scene
if layerN == -1:
index = self.index_group
groups = scene.layergroups[index].layer_groups
layergroups = scene.layergroups[index]
layers = space.layers
union = [False]*20
if not layergroups.toggle:
for i in range(0,20):
union[i] = groups[i] or layers[i]
space.layers = union
layergroups.toggle = True
else:
for i in range(0,20):
union[i] = not groups[i] and layers[i]
space.layers=union
layergroups.toggle=False
else:
if self.shift:
if space.layers[layerN]:
toggle = False
else:
toggle = True
space.layers[layerN] = toggle
else:
layer = [False]*20
layer[layerN] = True
space.layers = layer
if space.layers[layerN]:
toggle = False
return {'FINISHED'}
def invoke(self, context, event):
self.shift = event.shift
return self.execute(context)
class MergeSelected(bpy.types.Operator):
'''Move Selected Objects in this Layer Shift-Click to select multiple layers'''
bl_idname = "object.mergeselected"
bl_label = "Merge Selected object in this layer"
#prop definition
#layer number
layerN = IntProperty()
@classmethod
def poll(cls, context):
return context.scene
def execute(self, context):
layerN = self.layerN
scene = context.scene
#cyecle all object in the layer
for obj in scene.objects:
if obj.select:
visible = False
for i in range(0,20):
if obj.layers[i] and scene.layers[i]:
visible = True
break
if visible:
if self.shift:
if obj.layers[layerN]:
toggle = False
else:
toggle= True
obj.layers[layerN]=toggle
else:
layer = [False]*20
layer[layerN] = True
obj.layers = layer
if obj.layers[layerN]:
toggle = False
return {'FINISHED'}
def invoke(self, context, event):
self.shift = event.shift
return self.execute(context)
class MakeWireLayer(bpy.types.Operator):
'''Make Wire All Objects on this Layer'''
bl_idname = "object.wirelayer"
bl_label = "Make Wire Draw the object on this layer"
#prop definition
#layer number
layerN = IntProperty()
#lock status
wire = BoolProperty()
index_group = IntProperty()
@classmethod
def poll(cls, context):
return context.scene
def execute(self, context):
scene = context.scene
layerN = self.layerN
wire = self.wire
view_3d = context.area.spaces.active
#check if layer have some thing
if view_3d.layers_used[layerN] or layerN == -1:
if wire:
display = 'WIRE'
else:
display = 'TEXTURED'
#cyecle all object in the layer
for obj in context.scene.objects:
if layerN == -1:
index = self.index_group
groups = scene.layergroups[index].layer_groups
layers = obj.layers
layergroup = [False]*20
for i in range (0,20):
layergroup[i] = layers[i] and groups[i]
if True in layergroup:
obj.draw_type = display
scene.layergroups[index].wire = not wire
else:
if obj.layers[layerN]:
obj.draw_type = display
scene.WireLayer[layerN] = not wire
return {'FINISHED'}
class LockSelected(bpy.types.Operator):
'''Lock All Objects on this Layer'''
bl_idname = "object.lockselected"
bl_label = "Hide Select of Selected"
#prop definition
#layer number
layerN = IntProperty()
#lock status
lock = BoolProperty()
index_group = IntProperty()
@classmethod
def poll(cls, context):
return context.scene
def execute(self, context):
scene = context.scene
layerN = self.layerN
lock = self.lock
view_3d = context.area.spaces.active
#check if layer have some thing
if view_3d.layers_used[layerN] or layerN == -1:
#cyecle all object in the layer
for obj in context.scene.objects:
if layerN == -1:
index = self.index_group
groups = scene.layergroups[index].layer_groups
layers = obj.layers
layergroup = [False]*20
for i in range (0,20):
layergroup[i] = layers[i] and groups[i]
if True in layergroup:
obj.hide_select = not lock
obj.select = False
scene.layergroups[index].lock = not lock
else:
if obj.layers[layerN]:
obj.hide_select = not lock
obj.select = False
scene.LockLayer[layerN] = not lock
return {'FINISHED'}
class SelectObjectsLayer(bpy.types.Operator):
'''Select All the Objects on this Layer - Hold SHIFT for multy selection - Hold CTRL to make active the last selected object'''
bl_idname = "object.selectobjectslayer"
bl_label = "Select objects in Layer"
select_obj = BoolProperty()
layerN = IntProperty()
@classmethod
def poll(cls, context):
return context.scene
def execute(self, context):
view_3d = context.area.spaces.active
select_obj = self.select_obj
layerN = self.layerN
scene = context.scene
i = 0
s = 0
#check if layer have some thing
if view_3d.layers_used[layerN]:
for obj in context.scene.objects:
if obj.layers[layerN]:
i += 1
if self.ctrl:
context.scene.objects.active = obj
if obj.layers[layerN] and obj.select:
s += 1
#context.scene.objects.active = obj
if s == i:
for obj in context.scene.objects:
#context.scene.objects.active = obj
if obj.layers[layerN]:
obj.select = False
else:
shift = self.shift
bpy.ops.object.select_by_layer(extend=shift, layers=layerN+1)
return {'FINISHED'}
def invoke(self, context, event):
self.shift = event.shift
self.ctrl = event.ctrl
return self.execute(context)
class AllLayersSelect(bpy.types.Operator):
'''Active all Layer in scene'''
bl_idname = "scene.layersselect"
bl_label = "Select All Layer"
vis = BoolProperty()
@classmethod
def poll(cls, context):
return context.scene
def execute(self, context):
scene = context.scene
vis = self.vis
#store the active layer
active = scene.active_layer
view_3d = context.area.spaces.active
#check for lock camera and layer is active
if view_3d.lock_camera_and_layers is True:
space= scene
else:
space = view_3d
if not vis:
for i in range (0,20):
#keep selection of active layer
if active == i:
space.layers[i]= True
#deselect the other...
else:
space.layers[i]= False
else:
for i in range (0,20):
#select all layer
space.layers[i]=True
#after the cycle, deselect and select the previus active layer
space.layers[active]=False
space.layers[active]=True
return {'FINISHED'}
class LayerName(bpy.types.Panel):
bl_space_type = 'VIEW_3D'
bl_region_type = "TOOLS"
bl_label = "Layer Management"
bl_idname = "_PT_rig_layers"
bl_options = {'DEFAULT_CLOSED'}
bl_category = "Layer"
# layer name prop definition
bpy.types.Scene.LayerName1 = StringProperty(name="Layer Name", default='layer1')
bpy.types.Scene.LayerName2 = StringProperty(name="Layer Name", default='layer2')
bpy.types.Scene.LayerName3 = StringProperty(name="Layer Name", default='layer3')
bpy.types.Scene.LayerName4 = StringProperty(name="Layer Name", default='layer4')
bpy.types.Scene.LayerName5 = StringProperty(name="Layer Name", default='layer5')
bpy.types.Scene.LayerName6 = StringProperty(name="Layer Name", default='layer6')
bpy.types.Scene.LayerName7 = StringProperty(name="Layer Name", default='layer7')
bpy.types.Scene.LayerName8 = StringProperty(name="Layer Name", default='layer8')
bpy.types.Scene.LayerName9 = StringProperty(name="Layer Name", default='layer9')
bpy.types.Scene.LayerName10 = StringProperty(name="Layer Name", default='layer10')
bpy.types.Scene.LayerName11 = StringProperty(name="Layer Name", default='layer11')
bpy.types.Scene.LayerName12 = StringProperty(name="Layer Name", default='layer12')
bpy.types.Scene.LayerName13 = StringProperty(name="Layer Name", default='layer13')
bpy.types.Scene.LayerName14 = StringProperty(name="Layer Name", default='layer14')
bpy.types.Scene.LayerName15 = StringProperty(name="Layer Name", default='layer15')
bpy.types.Scene.LayerName16 = StringProperty(name="Layer Name", default='layer16')
bpy.types.Scene.LayerName17 = StringProperty(name="Layer Name", default='layer17')
bpy.types.Scene.LayerName18 = StringProperty(name="Layer Name", default='layer18')
bpy.types.Scene.LayerName19 = StringProperty(name="Layer Name", default='layer19')
bpy.types.Scene.LayerName20 = StringProperty(name="Layer Name", default='layer20')
#prop for hide empty
bpy.types.Scene.LayerVisibility = BoolProperty(name="Hide empty Layer", default=False)
#prop for extra option
bpy.types.Scene.ExtraOptions = BoolProperty(name="Show extra options", default=True)
#lock layer status
bpy.types.Scene.LockLayer = BoolVectorProperty(name="Lock Layer", default = ([False]*20), size=20)
#prop for show index
bpy.types.Scene.LayerIndex = BoolProperty(name="Show Index", default=False)
#prop for show classic
bpy.types.Scene.Classic = BoolProperty(name="Classic", default=False,description="Use a classic layer selection visibility")
#Select object Status
bpy.types.Scene.ObjectSelect = BoolVectorProperty(name="Object Select", default = ([True]*20), size=20)
#wire set up
bpy.types.Scene.WireLayer = BoolVectorProperty(name="Wire Layer", default = ([True]*20), size=20)
#toggle for merge
#bpy.types.Scene.MergeToggle = bpy.props.BoolVectorProperty(name="Merge Toggle", default = (False,False,False,False,False,False,False,False,False,False,False,False,False,False,False,False,False,False,False,False), size=20)
@classmethod
def poll(self, context):
try:
return (context.mode not in EDIT)
except (AttributeError, KeyError, TypeError):
return False
def draw(self, context):
scene = context.scene
view_3d = context.area.spaces.active
#check for lock camera and layer is active
if view_3d.lock_camera_and_layers is True:
space = scene
spacecheck = False
else:
space = view_3d
spacecheck = True
#row = layout.row(align=True)
vis = False
allIcon = 'RESTRICT_VIEW_OFF'
allText = "Isolate active"
#check if there is a layer off
for layer in space.layers:
if not layer:
vis = True
allIcon = 'RESTRICT_VIEW_ON'
allText = "All Visible"
layout = self.layout
column = layout.column()
row = column.row()
col2 = row.column()
#lock camera and layers button
col2.prop(view_3d, "lock_camera_and_layers", text="")
#select all
allView = col2.operator("scene.layersselect", emboss=False, icon=allIcon, text="")
allView.vis = vis
col = row.column()
col.alignment = 'RIGHT'
#classic
col.prop(scene, "Classic", text="Classic")
#show extra option checkbox
col.alignment = 'RIGHT'
col.prop(scene, "ExtraOptions", text="Options")
col1 = row.column()
#show index
col1.prop(scene, "LayerIndex", text="Index")
# hide empty
if context.object:
col1.alignment = 'RIGHT'
col1.prop(scene, "LayerVisibility", toggle=False, text="Hide Empty")
##########
#list the layer
for i in range(0,20):
#inizializing the icon
#lockIcon='UNLOCKED'
iconUsed = 'RADIOBUT_OFF'
iconAc = 'NONE'
iconLayer = 'NONE'
#selectIcon ='RESTRICT_SELECT_OFF'
#inizializing the bool value
noitem = False
active = False
layer = 20
scene = context.scene
extra = scene.ExtraOptions
layer_used = view_3d.layers_used[i]
#check the hide empty value
if scene.LayerVisibility:
#find the empty layer
if layer_used:
#print (i)
layer = i
'Add an object to see the layer'
else:
layer = i
#the layer number
#set icon for wire layer
wire = scene.WireLayer[i]
wireIcon = 'POTATO'
if not wire:
wireIcon = 'WIRE'
else:
wireIcon = 'POTATO'
#set icon for lock layer
lock = scene.LockLayer[i]
if lock:
lockIcon = 'LOCKED'
else:
lockIcon = 'UNLOCKED'
#check if there are Selected obj in the layer
#check if layer have some thing
if layer_used:
iconUsed = 'LAYER_USED'
active_object = context.object
if active_object:
if context.object.layers[i]:
active = True
else:
scene.ObjectSelect[i] = True
if layer == i:
#check for active layer and set the icon
if view_3d.lock_camera_and_layers:
if scene.active_layer == layer:
iconAc = 'FILE_TICK'
noitem = True
if active:
iconUsed = 'LAYER_ACTIVE'
#set icon for layer view
if view_3d.layers[layer]:
viewIcon = 'RESTRICT_VIEW_OFF'
noitem = True
else:
viewIcon = 'RESTRICT_VIEW_ON'
noitem = False
row2 = column.row(align=True)
# if space==scene:
#
# colb1= row2.column()
# #layer visibility operator
# tog = colb1.operator("view3d.layers", text="",icon=viewIcon, emboss=False)
# tog.nr=layer+1
# tog.toggle=True
# viewemboss = True
iconLayer = viewIcon
# layer index
if scene.LayerIndex:
col6 = row2.column()
col6.scale_x = 0.14
col6.label(text=str(i+1)+".")
# visualization
classic = scene.Classic
if not classic:
colb2 = row2.column()
colb2.prop(space, "layers", index=layer, emboss=True, icon=iconLayer, toggle=True, text="")
else:
colb6 = row2.column()
used = colb6.operator("object.layertoggle", text="", icon= iconLayer, emboss=True)
used.layerN = i
used.spacecheck=spacecheck
#text prop
s = "LayerName"+str(i+1)
colb3 = row2.column()
colb3.prop(scene, s, text="",icon=iconAc)
if extra:
#Select by type operator
colb4 = row2.column()
select = colb4.operator("object.selectobjectslayer", icon='RESTRICT_SELECT_OFF',text="", emboss=True )
select.layerN = i
#select.select_obj= selobj
#lock operator
colb5 = row2.column()
lk = colb5.operator("object.lockselected", text="", emboss=True, icon=lockIcon)
lk.layerN = i
lk.lock = lock
# #occuped layer
# colb6.label(text="", icon=iconUsed)
#merge layer
colb7 = row2.column()
merge = colb7.operator("object.mergeselected", text="", emboss=True, icon=iconUsed)
merge.layerN = i
#wire view:
colb8 = row2.column()
wirecol = colb8.operator("object.wirelayer", text="", emboss=True, icon=wireIcon)
wirecol.layerN = i
wirecol.wire = wire
if not scene.LayerVisibility:
if i == 4 or i == 9 or i == 14 or i == 19:
row3 = column.row()
row3.separator()
if len(scene.objects) == 0:
layout.label(text='No objects in scene')
class UI_UL_layergroups(UIList):
def draw_item(self, context, layout, data, item, icon, active_data, active_propname, index):
# assert(isinstance(item, bpy.types.VertexGroup)
lgroup = item
scene = context.scene
view_3d = context.area.spaces.active
#check for lock camera and layer is active
if view_3d.lock_camera_and_layers is True:
space = scene
spacecheck = False
else:
space = view_3d
spacecheck = True
#######################
if self.layout_type in {'DEFAULT', 'COMPACT'}:
layout.prop(lgroup, "name", text="", emboss=False)
#set icon for view
if lgroup.toggle:
iconLayer = 'RESTRICT_VIEW_OFF'
else:
iconLayer = 'RESTRICT_VIEW_ON'
#set icon for lock
if lgroup.lock:
iconLock = 'LOCKED'
else:
iconLock = 'UNLOCKED'
#set icon for wire layer
wireIcon = 'POTATO'
if not lgroup.wire:
wireIcon = 'WIRE'
else:
wireIcon = 'POTATO'
#lock operato
lk = layout.operator("object.lockselected", text="", emboss=False, icon= iconLock)
lk.index_group = index
lk.lock = lgroup.lock
lk.layerN = -1
#viwe operator
view = layout.operator("object.layertoggle", text="", emboss=False, icon= iconLayer)
view.index_group = index
view.layerN = -1
view.spacecheck = spacecheck
#wire operator
wirecol = layout.operator("object.wirelayer", text="", emboss=False, icon= wireIcon)
wirecol.index_group = index
wirecol.wire = lgroup.wire
wirecol.layerN = -1
elif self.layout_type in {'GRID'}:
layout.alignment = 'CENTER'
class LayerGroupsUI(bpy.types.Panel):
bl_space_type = 'VIEW_3D'
bl_region_type = "TOOLS"
bl_category = "Layer"
bl_label = "Layer Groups"
bl_idname = "_PT_layer_group"
bl_options = {'DEFAULT_CLOSED'}
@classmethod
def poll(self, context):
try:
return (context.mode not in EDIT)
except (AttributeError, KeyError, TypeError):
return False
def draw(self, context):
scene = context.scene
index_group = scene.layergroups_index
items = len(scene.layergroups)
viewIcon = 'RESTRICT_VIEW_OFF'
layout = self.layout
row = layout.row()
row.template_list("UI_UL_layergroups", "", context.scene, "layergroups", context.scene, "layergroups_index", rows=items)
col = row.column(align =True)
add = col.operator("object.layergroup_add", icon='ZOOMIN', text="")
add.index = items
add.layer = scene.layers
remove = col.operator("object.layergroup_remove", icon='ZOOMOUT', text="")
remove.index_group = index_group
if items > 0:
layout.prop(scene.layergroups[index_group], "layer_groups", text="", toggle=True)
layout.prop(scene.layergroups[index_group], "name", text="Name:")
def register():
bpy.utils.register_module(__name__)
'''bpy.utils.register_class(LayerGroupsUI)
bpy.utils.register_class(UI_UL_layergroups)
bpy.utils.register_class(LayerName)
bpy.utils.register_class(AllLayersSelect)
bpy.utils.register_class(SelectObjectsLayer)
bpy.utils.register_class(MakeWireLayer)
bpy.utils.register_class(LockSelected)
bpy.utils.register_class(MergeSelected)
bpy.utils.register_class(LayerToggle)
bpy.utils.register_class(AddLayerGroup)
bpy.utils.register_class(RemoveLayerGroup)
bpy.utils.register_class(LayerGroup)'''
def unregister():
bpy.utils.unregister_module(__name__)
'''bpy.utils.unregister_class(LayerGroupsUI)
bpy.utils.unregister_class(UI_UL_layergroups)
bpy.utils.unregister_class(LayerName)
bpy.utils.unregister_class(AllLayersSelect)
bpy.utils.unregister_class(SelectObjectsLayer)
bpy.utils.unregister_class(MakeWireLayer)
bpy.utils.unregister_class(LockSelected)
bpy.utils.unregister_class(MergeSelected)
bpy.utils.unregister_class(LayerToggle)
bpy.utils.unregister_class(AddLayerGroup)
bpy.utils.unregister_class(RemoveLayerGroup)
bpy.utils.unregister_class(LayerGroup)'''
if __name__ == "__main__":
register()