fix tracker url
[blender-addons-contrib.git] / mesh_face_info_select.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": "Face info / select by type",
21     "author": "CoDEmanX",
22     "version": (0, 0, 3),
23     "blender": (2, 6, 2),
24     "location": "Properties > Object data > Face info / select",
25     "description": "Displays triangle, quad and ngon count of the active object. Allows to select faces by these types.",
26     "warning": "",
27     "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"\
28         "Scripts/Mesh/Face_Info_Select",
29     "tracker_url": "http://projects.blender.org/tracker/index.php?"\
30         "func=detail&aid=31926",
31     "support": 'TESTING',
32     "category": "Mesh"
33 }
34
35 # Written by CoDEmanX for frivus
36
37
38 import bpy
39 from bpy.props import EnumProperty
40
41 class DATA_OP_facetype_select(bpy.types.Operator):
42     """Select all faces of a certain type"""
43     bl_idname = "data.facetype_select"
44     bl_label = "Select by face type"
45     bl_options = {'REGISTER', 'UNDO'}
46
47     face_type = EnumProperty(name="Select faces:",
48                              items = (("3","Triangles","Faces made up of 3 vertices"),
49                                       ("4","Quads","Faces made up of 4 vertices"),
50                                       ("5","Ngons","Faces made up of 5 and more vertices")),
51                              default = "5")
52     @classmethod
53     def poll(cls, context):
54         return context.active_object is not None and context.active_object.type == 'MESH'
55
56     def execute(self, context):
57
58         bpy.ops.object.mode_set(mode='EDIT')
59         bpy.ops.mesh.select_all(action='DESELECT')
60         context.tool_settings.mesh_select_mode=(False, False, True)
61
62         if self.face_type == "3":
63             bpy.ops.mesh.select_by_number_vertices(number=3, type='EQUAL')
64         elif self.face_type == "4":
65             bpy.ops.mesh.select_by_number_vertices(number=4, type='EQUAL')
66         else:
67             bpy.ops.mesh.select_by_number_vertices(number=4, type='GREATER')
68
69         return {'FINISHED'}
70         
71
72 class DATA_PT_info_panel(bpy.types.Panel):
73     """Creates a face info / select panel in the Object properties window"""
74     bl_label = "Face info / select"
75     bl_idname = "DATA_PT_face_info"
76     bl_space_type = "PROPERTIES"
77     bl_region_type = "WINDOW"
78     bl_context = "data"
79     bl_options = {'DEFAULT_CLOSED'}
80
81     @classmethod
82     def poll(self, context):
83         return context.active_object is not None and context.active_object.type == 'MESH'
84
85     def draw(self, context):
86         layout = self.layout
87
88         ob = context.active_object
89         
90         info_str = ""
91         tris = quads = ngons = 0
92
93         for p in ob.data.polygons:
94             count = p.loop_total
95             if count == 3:
96                 tris += 1
97             elif count == 4:
98                 quads += 1
99             else:
100                 ngons += 1
101
102         info_str = "  Ngons: %i  Quads: %i  Tris: %i" % (ngons, quads, tris)
103         
104         col = layout.column()
105         col.label(info_str, icon='MESH_DATA')
106
107         col = layout.column()
108         col.label("Select faces by type:")
109
110         row = layout.row()
111         row.operator("data.facetype_select", text="Ngons").face_type = "5"
112         row.operator("data.facetype_select", text="Quads").face_type = "4"
113         row.operator("data.facetype_select", text="Tris").face_type = "3"
114
115
116 def register():
117     bpy.utils.register_class(DATA_PT_info_panel)
118     bpy.utils.register_class(DATA_OP_facetype_select)
119
120
121 def unregister():
122     bpy.utils.unregister_class(DATA_PT_info_panel)
123     bpy.utils.unregister_class(DATA_OP_facetype_select)
124
125
126 if __name__ == "__main__":
127     register()