Set material and color properties of Blob objects.
[blender-addons-contrib.git] / mesh_select_vertex_groups.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 bl_info = {
20     "name": "Select Vertex Groups",
21     "author": "Martin Ellison",
22     "version": (1, 0),
23     "blender": (2, 71, 0),
24     "location": "Toolbox",
25     "description": "Finds all the vertex groups that chosen verts are in, & any verts that are not in any group",
26     "warning": "Buggy", # used for warning icon and text in addons panel
27     "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"
28                 "Scripts/Modeling/Select_Vertex_Groups",
29     "tracker_url": "https://developer.blender.org/T22025",
30     "category": "Mesh"}
31
32 """
33 This script finds all the vertex groups that chosen vertexes are in, and any vertexes that are not in any vertex group.
34
35 This is useful for cleaning up a mesh if vertex groups under animation go off to the wrong place (because they are in a vertex group that they should not be in, or not in the vertex group that they should be in).
36
37 How to use:
38 1. select a mesh and get into edit mode.
39 2. by default it will use all vertexes; alternatively, select vertexes of interest and click on 'use selected vertexes'. Note that subsequent selections and deselections of vertexes will not change the set of vertexes to be used, you need to click on these buttons again to do that.
40 3. click on 'select' and 'deselect' buttons for listed vertex groups, and for no vertex group, to select and deselect vertexes.
41
42 This only lists vertex groups that have used vertexes.
43
44 You may want to use the mesh select/deselect all (keyboard A) operator to start.
45
46 Once you have the right vertexes selected, you can use the standard vertex groups property editor to add them to or remove them from the desired vertex groups.
47 """
48
49
50 import bpy
51 from bpy.props import *
52
53 global use_selected_only, used_vertexes, the_mesh, vertex_usage
54 use_selected_only = False
55 used_vertexes = set()
56 the_mesh = None
57 vertex_usage = ''
58
59 class UseAll(bpy.types.Operator):
60     bl_idname = "mesh.primitive_fvg_useall"
61     bl_label = "Use all Vertexes"
62     bl_register = True
63     bl_undo = True
64     #limit = FloatProperty(name="limit", description="Ignore weights under this limit.", default= 0.01, min = 0.0, max = 1.0, soft_min=0.0, soft_max=1.0)
65
66     def execute(self, context):
67         global use_selected_only
68         use_selected_only = False
69         bpy.ops.object.editmode_toggle()
70         set_used()
71         bpy.ops.object.editmode_toggle()
72         return {'FINISHED'}
73
74 class UseSelected(bpy.types.Operator):
75     bl_idname = "mesh.primitive_fvg_useselected"
76     bl_label = "Use Selected Vertexes"
77     bl_register = True
78     bl_undo = True
79
80     def execute(self, context):
81         global use_selected_only
82         use_selected_only = True
83         bpy.ops.object.editmode_toggle()
84         set_used()
85         bpy.ops.object.editmode_toggle()
86         return {'FINISHED'}
87
88 class SelectFound(bpy.types.Operator):
89     bl_idname = "mesh.primitive_fvg_selfound"
90     bl_label = "Select"
91     bl_register = True
92     bl_undo = True
93     vertexgroup = bpy.props.StringProperty(name = 'vertexgroup', description = 'vertexgroup', default = '', options = set())
94
95     def execute(self, context):
96         global the_mesh
97         bpy.ops.object.editmode_toggle()
98         vertexgroup = self.properties.vertexgroup
99         fv = found_verts(vertexgroup)
100         for v in fv: v.select = True
101         bpy.ops.object.editmode_toggle()
102         return {'FINISHED'}
103
104 class DeselectFound(bpy.types.Operator):
105     bl_idname = "mesh.primitive_fvg_deselfound"
106     bl_label = "Deselect"
107     bl_register = True
108     bl_undo = True
109     vertexgroup = bpy.props.StringProperty(name = 'vertexgroup', description = 'vertexgroup', default = '', options = set())
110
111     def execute(self, context):
112         global the_mesh
113         bpy.ops.object.editmode_toggle()
114         vertexgroup = self.properties.vertexgroup
115         fv = found_verts(vertexgroup)
116         for v in fv: v.select = False
117         bpy.ops.object.editmode_toggle()
118         return {'FINISHED'}
119
120 def set_used():
121     global use_selected_only, used_vertexes, the_mesh, vertex_usage
122     obj = bpy.context.active_object
123     used_vertexes = set()
124     if use_selected_only:
125         for v in obj.data.vertices:
126             if v.select: used_vertexes.add(v.index)
127     else:
128         for v in obj.data.vertices: used_vertexes.add(v.index)
129     the_mesh = obj
130     vertex_usage = '%d vertexes used' % (len(used_vertexes))
131
132
133 def make_groups(limit):
134     global used_vertexes
135     vgp = []
136     vgdict = {}
137     vgused = {}
138     obj = bpy.context.active_object
139     all_in_group = True
140     for vg in obj.vertex_groups:
141         vgdict[vg.index] = vg.name
142     for v in obj.data.vertices:
143         in_group = False
144         if v.index in used_vertexes:
145             for g in v.groups:
146                 gr = g.group
147                 w = g.weight
148                 if w > limit:
149                     if not gr in vgused: vgused[gr] = 0
150                     vgused[gr] += 1
151                     in_group = True
152         if not in_group: all_in_group = False
153     if not all_in_group:
154         vgp.append(("no group", "(No group)"))
155     for gn in vgused.keys():
156         name = vgdict[gn]
157         vgp.append((name, '%s has %d vertexes' % (name, vgused[gn]) ))
158     print("%d groups found\n" % len(vgp))
159     return vgp
160
161 def found_verts(vertex_group):
162     global used_vertexes
163     vgfound = []
164     obj = bpy.context.active_object
165     if vertex_group == 'no group':
166         for v in obj.data.vertices:
167             if v.index in used_vertexes and (not v.groups):
168                 vgfound.append(v)
169     else:
170         vgnum = obj.vertex_groups.find(vertex_group)
171         for v in obj.data.vertices:
172             if v.index in used_vertexes:
173                 for g in v.groups:
174                     if g.group == vgnum:
175                         vgfound.append(v)
176                         break
177
178     print('%d vertexes found for %s' % (len(vgfound), vertex_group))
179     return vgfound
180
181
182 class VIEW3D_PT_FixVertexGroups(bpy.types.Panel):
183     bl_space_type = "VIEW_3D"
184     bl_region_type = "TOOLS"
185     bl_label = "Select Vertex Groups"
186     bl_category = 'Tools'
187     bl_options = {'DEFAULT_CLOSED'}
188
189     @classmethod
190     def poll(self, context):
191         if bpy.context.active_object:
192             obj = bpy.context.active_object
193             if obj.type == 'MESH' and obj.mode == 'EDIT': return True
194         return False
195
196     def draw(self, context):
197         global use_selected_only, used_vertexes, the_mesh, vertex_usage
198
199         if bpy.context.active_object:
200             obj = bpy.context.active_object
201             if obj.type == 'MESH' and obj.mode == 'EDIT':
202                 layout = self.layout
203                 use_all = layout.operator("mesh.primitive_fvg_useall", "Use all vertexes")
204                 layout.operator("mesh.primitive_fvg_useselected", "Use selected vertexes")
205                 if use_selected_only:
206                     layout.label(text = 'Using selected vertexes.')
207                 else:
208                     layout.label(text = 'Using all vertexes.')
209                 layout.label(vertex_usage)
210                 if len(used_vertexes) == 0 or obj is not the_mesh: set_used()
211                 #layout.prop(use_all, 'limit', slider = True)
212                 #groups = make_groups(use_all.limitval)
213                 groups = make_groups(0.01)
214                 for gp in groups:
215                     layout.label(text = gp[1])
216                     row = layout.row()
217                     sel_op = row.operator("mesh.primitive_fvg_selfound", "Select")
218                     sel_op.vertexgroup = gp[0]
219                     desel_op = row.operator("mesh.primitive_fvg_deselfound", "Deselect")
220                     desel_op.vertexgroup = gp[0]
221
222 classes = [UseAll, UseSelected, SelectFound, DeselectFound]
223
224 def register():
225     bpy.utils.register_module(__name__)
226     pass
227
228 def unregister():
229     bpy.utils.unregister_module(__name__)
230     pass
231
232 if __name__ == "__main__":
233     print('------ executing --------')
234     register()