release/scripts/op/object.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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  16 #
  17 # ##### END GPL LICENSE BLOCK #####
  18
  19 # <pep8 compliant>
  20
  21 import bpy
  22 from bpy.props import *
  23
  24
  25 class SelectPattern(bpy.types.Operator):
  26     '''Select object matching a naming pattern.'''
  27     bl_idname = "object.select_pattern"
  28     bl_label = "Select Pattern"
  29     bl_register = True
  30     bl_undo = True
  31
  32     pattern = StringProperty(name="Pattern", description="Name filter using '*' and '?' wildcard chars", maxlen=32, default="*")
  33     case_sensitive = BoolProperty(name="Case Sensitive", description="Do a case sensitive compare", default=False)
  34     extend = BoolProperty(name="Extend", description="Extend the existing selection", default=True)
  35
  36     def execute(self, context):
  37
  38         import fnmatch
  39
  40         if self.properties.case_sensitive:
  41             pattern_match = fnmatch.fnmatchcase
  42         else:
  43             pattern_match = lambda a, b: fnmatch.fnmatchcase(a.upper(), b.upper())
  44
  45         obj = context.object
  46         if obj and obj.mode == 'POSE':
  47             items = obj.data.bones
  48         elif obj and obj.type == 'ARMATURE' and obj.mode == 'EDIT':
  49             items = obj.data.edit_bones
  50         else:
  51             items = context.visible_objects
  52
  53         # Can be pose bones or objects
  54         for item in items:
  55             if pattern_match(item.name, self.properties.pattern):
  56                 item.selected = True
  57             elif not self.properties.extend:
  58                 item.selected = False
  59
  60         return ('FINISHED',)
  61
  62     def invoke(self, context, event):
  63         wm = context.manager
  64         # return wm.invoke_props_popup(self, event)
  65         wm.invoke_props_popup(self, event)
  66         return ('RUNNING_MODAL',)
  67
  68     def draw(self, context):
  69         layout = self.layout
  70         props = self.properties
  71
  72         layout.prop(props, "pattern")
  73         row = layout.row()
  74         row.prop(props, "case_sensitive")
  75         row.prop(props, "extend")
  76
  77
  78 class SubdivisionSet(bpy.types.Operator):
  79     '''Sets a Subdivision Surface Level (1-5)'''
  80
  81     bl_idname = "object.subdivision_set"
  82     bl_label = "Subdivision Set"
  83     bl_register = True
  84     bl_undo = True
  85
  86     level = IntProperty(name="Level",
  87             default=1, min=0, max=100, soft_min=0, soft_max=6)
  88
  89     def poll(self, context):
  90         ob = context.active_object
  91         return (ob and ob.type == 'MESH')
  92
  93     def execute(self, context):
  94         level = self.properties.level
  95
  96         def set_object_subd(obj):
  97             for mod in obj.modifiers:
  98                 if mod.type == 'MULTIRES':
  99                     if level < mod.total_levels:
 100                         if obj.mode == 'SCULPT' and mod.sculpt_levels != level:
 101                             mod.sculpt_levels = level
 102                         elif obj.mode == 'OBJECT' and mod.levels != level:
 103                             mod.levels = level
 104                     return
 105                 elif mod.type == 'SUBSURF':
 106                     if mod.levels != level:
 107                         mod.levels = level
 108                     return
 109
 110             # adda new modifier
 111             mod = obj.modifiers.new("Subsurf", 'SUBSURF')
 112             mod.levels = level
 113
 114         for obj in context.selected_editable_objects:
 115             set_object_subd(obj)
 116
 117         return ('FINISHED',)
 118
 119
 120 class Retopo(bpy.types.Operator):
 121     '''TODO - doc'''
 122
 123     bl_idname = "object.retopology"
 124     bl_label = "Retopology from Grease Pencil"
 125     bl_register = True
 126     bl_undo = True
 127
 128     def execute(self, context):
 129         import retopo
 130         retopo.main()
 131         return ('FINISHED',)
 132
 133
 134 bpy.ops.add(SelectPattern)
 135 bpy.ops.add(SubdivisionSet)
 136 bpy.ops.add(Retopo)