mesh_relax.py

   1 # mesh_relax.py Copyright (C) 2010, Fabian Fricke
   2 #
   3 # Relaxes selected vertices while retaining the shape as much as possible
   4 #
   5 # ***** BEGIN GPL LICENSE BLOCK *****
   6 #
   7 #
   8 # This program is free software; you can redistribute it and/or
   9 # modify it under the terms of the GNU General Public License
  10 # as published by the Free Software Foundation; either version 2
  11 # of the License, or (at your option) any later version.
  12 #
  13 # This program is distributed in the hope that it will be useful,
  14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 # GNU General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with this program; if not, write to the Free Software Foundation,
  20 # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  21 #
  22 # ***** END GPL LICENCE BLOCK *****
  23
  24 bl_info = {
  25     "name": "Relax",
  26     "author": "Fabian Fricke",
  27     "version": (1, 1),
  28     "blender": (2, 57, 0),
  29     "location": "View3D > Specials > Relax ",
  30     "description": "Relax the selected verts while retaining the shape",
  31     "warning": "",
  32     "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"
  33                 "Scripts/Modeling/Relax",
  34     "category": "Mesh",
  35 }
  36
  37 """
  38 Usage:
  39
  40 Launch from "W-menu" or from "Mesh -> Vertices -> Relax"
  41
  42
  43 Additional links:
  44     Author Site: http://frigi.designdevil.de
  45     e-mail: frigi.f {at} gmail {dot} com
  46 """
  47
  48
  49 import bpy
  50 from bpy.props import IntProperty
  51
  52 def relax_mesh(context):
  53
  54     # deselect everything that's not related
  55     for obj in context.selected_objects:
  56         obj.select = False
  57
  58     # get active object
  59     obj = context.active_object
  60
  61     # duplicate the object so it can be used for the shrinkwrap modifier
  62     obj.select = True # make sure the object is selected!
  63     bpy.ops.object.mode_set(mode='OBJECT')
  64     bpy.ops.object.duplicate()
  65     target = context.active_object
  66
  67     # remove all other modifiers from the target
  68     for m in range(0, len(target.modifiers)):
  69         target.modifiers.remove(target.modifiers[0])
  70
  71     context.scene.objects.active = obj
  72
  73     sw = obj.modifiers.new(type='SHRINKWRAP', name='relax_target')
  74     sw.target = target
  75
  76     # run smooth operator to relax the mesh
  77     bpy.ops.object.mode_set(mode='EDIT')
  78     bpy.ops.mesh.vertices_smooth()
  79     bpy.ops.object.mode_set(mode='OBJECT')
  80
  81     # apply the modifier
  82     bpy.ops.object.modifier_apply(modifier='relax_target')
  83
  84     # delete the target object
  85     obj.select = False
  86     target.select = True
  87     bpy.ops.object.delete()
  88
  89     # go back to initial state
  90     obj.select = True
  91     bpy.ops.object.mode_set(mode='EDIT')
  92
  93 class Relax(bpy.types.Operator):
  94     """Relaxes selected vertices while retaining the shape """ \
  95     """as much as possible"""
  96     bl_idname = 'mesh.relax'
  97     bl_label = 'Relax'
  98     bl_options = {'REGISTER', 'UNDO'}
  99
 100     iterations = IntProperty(name="Relax iterations",
 101                 default=1, min=0, max=100, soft_min=0, soft_max=10)
 102
 103     @classmethod
 104     def poll(cls, context):
 105         obj = context.active_object
 106         return (obj and obj.type == 'MESH')
 107
 108     def execute(self, context):
 109         for i in range(0,self.iterations):
 110             relax_mesh(context)
 111         return {'FINISHED'}
 112
 113
 114 def menu_func(self, context):
 115     self.layout.operator(Relax.bl_idname, text="Relax")
 116
 117
 118 def register():
 119     bpy.utils.register_module(__name__)
 120
 121     bpy.types.VIEW3D_MT_edit_mesh_specials.append(menu_func)
 122     bpy.types.VIEW3D_MT_edit_mesh_vertices.append(menu_func)
 123
 124 def unregister():
 125     bpy.utils.unregister_module(__name__)
 126
 127     bpy.types.VIEW3D_MT_edit_mesh_specials.remove(menu_func)
 128     bpy.types.VIEW3D_MT_edit_mesh_vertices.remove(menu_func)
 129
 130 if __name__ == "__main__":
 131     register()