animation_animall.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": "AnimAll",
  21     "author": "Daniel Salazar <zanqdo@gmail.com>",
  22     "version": (0, 7),
  23     "blender": (2, 69, 7),
  24     "location": "Tool bar > Animation tab > AnimAll",
  25     "description": "Allows animation of mesh, lattice, curve and surface data",
  26     "warning": "",
  27     "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"
  28                 "Scripts/Animation/AnimAll",
  29     "category": "Animation",
  30 }
  31
  32 """-------------------------------------------------------------------------
  33 Thanks to Campbell Barton and Joshua Leung for hes API additions and fixes
  34 Daniel 'ZanQdo' Salazar
  35 -------------------------------------------------------------------------"""
  36
  37 import bpy
  38 from bpy.props import *
  39
  40
  41 #
  42 # Property Definitions
  43 #
  44 bpy.types.WindowManager.key_shape = BoolProperty(
  45     name="Shape",
  46     description="Insert keyframes on active Shape Key layer",
  47     default=False)
  48
  49 bpy.types.WindowManager.key_uvs = BoolProperty(
  50     name="UVs",
  51     description="Insert keyframes on active UV coordinates",
  52     default=False)
  53
  54 bpy.types.WindowManager.key_bevel = BoolProperty(
  55     name="Bevel",
  56     description="Insert keyframes on edge bevel weight",
  57     default=False)
  58
  59 bpy.types.WindowManager.key_crease = BoolProperty(
  60     name="Crease",
  61     description="Insert keyframes on edge creases",
  62     default=False)
  63
  64 bpy.types.WindowManager.key_vcols = BoolProperty(
  65     name="VCols",
  66     description="Insert keyframes on active Vertex Color values",
  67     default=False)
  68
  69 bpy.types.WindowManager.key_vgroups = BoolProperty(
  70     name="VGroups",
  71     description="Insert keyframes on active Vertex Group values",
  72     default=False)
  73
  74 bpy.types.WindowManager.key_points = BoolProperty(
  75     name="Points",
  76     description="Insert keyframes on point locations",
  77     default=False)
  78
  79 bpy.types.WindowManager.key_radius = BoolProperty(
  80     name="Radius",
  81     description="Insert keyframes on point radius (Shrink/Fatten)",
  82     default=False)
  83
  84 bpy.types.WindowManager.key_tilt = BoolProperty(
  85     name="Tilt",
  86     description="Insert keyframes on point tilt",
  87     default=False)
  88
  89 #
  90 # GUI (Panel)
  91 #
  92 class VIEW3D_PT_animall(bpy.types.Panel):
  93
  94     bl_space_type = 'VIEW_3D'
  95     bl_region_type = 'TOOLS'
  96     bl_category = "Animation"
  97     bl_label = 'AnimAll'
  98     #bl_options = {'DEFAULT_CLOSED'}
  99     @classmethod
 100     def poll(self, context):
 101         if context.active_object and context.active_object.type in {'MESH', 'LATTICE', 'CURVE', 'SURFACE'}:
 102             return context.active_object.type
 103
 104     # draw the gui
 105     def draw(self, context):
 106
 107         Obj = context.active_object
 108
 109         layout = self.layout
 110         col = layout.column(align=True)
 111         row = col.row()
 112
 113         if Obj.type == 'LATTICE':
 114             row.prop(context.window_manager, "key_points")
 115             row.prop(context.window_manager, "key_shape")
 116
 117         elif Obj.type == 'MESH':
 118             row.prop(context.window_manager, "key_points")
 119             row.prop(context.window_manager, "key_shape")
 120             row = col.row()
 121             row.prop(context.window_manager, "key_bevel")
 122             row.prop(context.window_manager, "key_crease")
 123             row = col.row()
 124             row.prop(context.window_manager, "key_vcols")
 125             row.prop(context.window_manager, "key_vgroups")
 126             row = col.row()
 127             row.prop(context.window_manager, "key_uvs")
 128
 129         elif Obj.type == 'CURVE':
 130             row.prop(context.window_manager, "key_points")
 131             row.prop(context.window_manager, "key_shape")
 132             row = col.row()
 133             row.prop(context.window_manager, "key_radius")
 134             row.prop(context.window_manager, "key_tilt")
 135
 136         elif Obj.type == 'SURFACE':
 137             row.prop(context.window_manager, "key_points")
 138             row.prop(context.window_manager, "key_shape")
 139             row = col.row()
 140             row.prop(context.window_manager, "key_radius")
 141             row.prop(context.window_manager, "key_tilt")
 142
 143         row = col.row()
 144         row.operator('anim.insert_keyframe_animall', icon='KEY_HLT')
 145         row.operator('anim.delete_keyframe_animall', icon='KEY_DEHLT')
 146         row = layout.row()
 147         row.operator('anim.clear_animation_animall', icon='X')
 148
 149         if context.window_manager.key_shape:
 150
 151             ShapeKey = Obj.active_shape_key
 152             ShapeKeyIndex = Obj.active_shape_key_index
 153
 154             split = layout.split()
 155             row = split.row()
 156
 157             if ShapeKeyIndex > 0:
 158                 row.label(ShapeKey.name, icon='SHAPEKEY_DATA')
 159                 row.prop(ShapeKey, "value", text="")
 160                 row.prop(Obj, "show_only_shape_key", text="")
 161                 if ShapeKey.value < 1:
 162                     row = layout.row()
 163                     row.label('Maybe set "%s" to 1.0?' % ShapeKey.name, icon='INFO')
 164             elif ShapeKey:
 165                 row.label('Can not key on Basis Shape', icon='ERROR')
 166             else:
 167                 row.label('No active Shape Key', icon='ERROR')
 168
 169         if context.window_manager.key_points and context.window_manager.key_shape:
 170             row = layout.row()
 171             row.label('"Points" and "Shape" are redundant?', icon='INFO')
 172
 173
 174 class ANIM_OT_insert_keyframe_animall(bpy.types.Operator):
 175     bl_label = 'Insert'
 176     bl_idname = 'anim.insert_keyframe_animall'
 177     bl_description = 'Insert a Keyframe'
 178     bl_options = {'REGISTER', 'UNDO'}
 179
 180
 181     # on mouse up:
 182     def invoke(self, context, event):
 183
 184         self.execute(context)
 185
 186         return {'FINISHED'}
 187
 188
 189     def execute(op, context):
 190
 191         Obj = context.active_object
 192
 193         if Obj.type == 'MESH':
 194             Mode = False
 195             if context.mode == 'EDIT_MESH':
 196                 Mode = not Mode
 197                 bpy.ops.object.editmode_toggle()
 198
 199             Data = Obj.data
 200
 201             if context.window_manager.key_shape:
 202                 if Obj.active_shape_key_index > 0:
 203                     for Vert in Obj.active_shape_key.data:
 204                         Vert.keyframe_insert('co')
 205
 206             if context.window_manager.key_points:
 207                 for Vert in Data.vertices:
 208                     Vert.keyframe_insert('co')
 209
 210             if context.window_manager.key_bevel:
 211                 for Edge in Data.edges:
 212                     Edge.keyframe_insert('bevel_weight')
 213
 214             if context.window_manager.key_crease:
 215                 for Edge in Data.edges:
 216                     Edge.keyframe_insert('crease')
 217
 218             if context.window_manager.key_vgroups:
 219                 for Vert in Data.vertices:
 220                     for Group in Vert.groups:
 221                         Group.keyframe_insert('weight')
 222
 223             if context.window_manager.key_uvs:
 224                 for UV in Data.uv_layers.active.data:
 225                     UV.keyframe_insert('uv')
 226
 227             if context.window_manager.key_vcols:
 228                 for VColLayer in Data.vertex_colors:
 229                     if VColLayer.active: # only insert in active VCol layer
 230                         for Data in VColLayer.data:
 231                             Data.keyframe_insert('color')
 232
 233             if Mode:
 234                 bpy.ops.object.editmode_toggle()
 235
 236         if Obj.type == 'LATTICE':
 237             Mode = False
 238             if context.mode != 'OBJECT':
 239                 Mode = not Mode
 240                 bpy.ops.object.editmode_toggle()
 241
 242             Data = Obj.data
 243
 244             if context.window_manager.key_shape:
 245                 if Obj.active_shape_key_index > 0:
 246                     for Point in Obj.active_shape_key.data:
 247                         Point.keyframe_insert('co')
 248
 249             if context.window_manager.key_points:
 250                 for Point in Data.points:
 251                     Point.keyframe_insert('co_deform')
 252
 253             if Mode:
 254                 bpy.ops.object.editmode_toggle()
 255
 256         if Obj.type in {'CURVE', 'SURFACE'}:
 257             Mode = False
 258             if context.mode != 'OBJECT':
 259                 Mode = not Mode
 260                 bpy.ops.object.editmode_toggle()
 261
 262             Data = Obj.data
 263
 264             # run this outside the splines loop (only once)
 265             if context.window_manager.key_shape:
 266                 if Obj.active_shape_key_index > 0:
 267                     for CV in Obj.active_shape_key.data:
 268                         CV.keyframe_insert('co')
 269                         try: # in case spline has no handles
 270                             CV.keyframe_insert('handle_left')
 271                             CV.keyframe_insert('handle_right')
 272                         except: pass
 273
 274             for Spline in Data.splines:
 275                 if Spline.type == 'BEZIER':
 276
 277                     for CV in Spline.bezier_points:
 278
 279                         if context.window_manager.key_points:
 280                             CV.keyframe_insert('co')
 281                             CV.keyframe_insert('handle_left')
 282                             CV.keyframe_insert('handle_right')
 283
 284                         if context.window_manager.key_radius:
 285                             CV.keyframe_insert('radius')
 286
 287                         if context.window_manager.key_tilt:
 288                             CV.keyframe_insert('tilt')
 289
 290                 elif Spline.type == 'NURBS':
 291
 292                     for CV in Spline.points:
 293
 294                         if context.window_manager.key_points:
 295                             CV.keyframe_insert('co')
 296
 297                         if context.window_manager.key_radius:
 298                             CV.keyframe_insert('radius')
 299
 300                         if context.window_manager.key_tilt:
 301                             CV.keyframe_insert('tilt')
 302
 303             if Mode:
 304                 bpy.ops.object.editmode_toggle()
 305
 306
 307         return {'FINISHED'}
 308
 309
 310 class ANIM_OT_delete_keyframe_animall(bpy.types.Operator):
 311     bl_label = 'Delete'
 312     bl_idname = 'anim.delete_keyframe_animall'
 313     bl_description = 'Delete a Keyframe'
 314     bl_options = {'REGISTER', 'UNDO'}
 315
 316
 317     # on mouse up:
 318     def invoke(self, context, event):
 319
 320         self.execute(context)
 321
 322         return {'FINISHED'}
 323
 324
 325     def execute(op, context):
 326
 327         Obj = context.active_object
 328
 329         if Obj.type == 'MESH':
 330             Mode = False
 331             if context.mode == 'EDIT_MESH':
 332                 Mode = not Mode
 333                 bpy.ops.object.editmode_toggle()
 334
 335             Data = Obj.data
 336
 337             if context.window_manager.key_shape:
 338                 if Obj.active_shape_key:
 339                     for Vert in Obj.active_shape_key.data:
 340                         Vert.keyframe_delete('co')
 341
 342             if context.window_manager.key_points:
 343                 for Vert in Data.vertices:
 344                     Vert.keyframe_delete('co')
 345
 346             if context.window_manager.key_bevel:
 347                 for Edge in Data.edges:
 348                     Edge.keyframe_delete('bevel_weight')
 349
 350             if context.window_manager.key_crease:
 351                 for Edge in Data.edges:
 352                     Edge.keyframe_delete('crease')
 353
 354             if context.window_manager.key_vgroups:
 355                 for Vert in Data.vertices:
 356                     for Group in Vert.groups:
 357                         Group.keyframe_delete('weight')
 358
 359             if context.window_manager.key_uvs:
 360                 for UV in Data.uv_layers.active.data:
 361                     UV.keyframe_delete('uv')
 362
 363             if context.window_manager.key_vcols:
 364                 for VColLayer in Data.vertex_colors:
 365                     if VColLayer.active: # only delete in active VCol layer
 366                         for Data in VColLayer.data:
 367                             Data.keyframe_delete('color')
 368
 369             if Mode:
 370                 bpy.ops.object.editmode_toggle()
 371
 372         if Obj.type == 'LATTICE':
 373
 374             Mode = False
 375             if context.mode != 'OBJECT':
 376                 Mode = not Mode
 377                 bpy.ops.object.editmode_toggle()
 378
 379             Data = Obj.data
 380
 381             if context.window_manager.key_shape:
 382                 if Obj.active_shape_key:
 383                     for Point in Obj.active_shape_key.data:
 384                         Point.keyframe_delete('co')
 385
 386             if context.window_manager.key_points:
 387                 for Point in Data.points:
 388                     Point.keyframe_delete('co_deform')
 389
 390             if Mode:
 391                 bpy.ops.object.editmode_toggle()
 392
 393         if Obj.type in {'CURVE', 'SURFACE'}:
 394             Mode = False
 395             if context.mode != 'OBJECT':
 396                 Mode = not Mode
 397                 bpy.ops.object.editmode_toggle()
 398
 399             Data = Obj.data
 400
 401             # run this outside the splines loop (only once)
 402             if context.window_manager.key_shape:
 403                 if Obj.active_shape_key_index > 0:
 404                     for CV in Obj.active_shape_key.data:
 405                         CV.keyframe_delete('co')
 406                         try: # in case spline has no handles
 407                             CV.keyframe_delete('handle_left')
 408                             CV.keyframe_delete('handle_right')
 409                         except: pass
 410
 411             for Spline in Data.splines:
 412                 if Spline.type == 'BEZIER':
 413                     for CV in Spline.bezier_points:
 414                         if context.window_manager.key_points:
 415                             CV.keyframe_delete('co')
 416                             CV.keyframe_delete('handle_left')
 417                             CV.keyframe_delete('handle_right')
 418                         if context.window_manager.key_radius:
 419                             CV.keyframe_delete('radius')
 420                         if context.window_manager.key_tilt:
 421                             CV.keyframe_delete('tilt')
 422
 423                 elif Spline.type == 'NURBS':
 424                     for CV in Spline.points:
 425                         if context.window_manager.key_points:
 426                             CV.keyframe_delete('co')
 427                         if context.window_manager.key_radius:
 428                             CV.keyframe_delete('radius')
 429                         if context.window_manager.key_tilt:
 430                             CV.keyframe_delete('tilt')
 431
 432             if Mode:
 433                 bpy.ops.object.editmode_toggle()
 434
 435
 436         return {'FINISHED'}
 437
 438
 439 class ANIM_OT_clear_animation_animall(bpy.types.Operator):
 440     bl_label = 'Clear Animation'
 441     bl_idname = 'anim.clear_animation_animall'
 442     bl_description = 'Delete all keyframes for this object'
 443     bl_options = {'REGISTER', 'UNDO'}
 444
 445     # on mouse up:
 446     def invoke(self, context, event):
 447
 448         wm = context.window_manager
 449         return wm.invoke_confirm(self, event)
 450
 451
 452     def execute(op, context):
 453
 454         Data = context.active_object.data
 455         Data.animation_data_clear()
 456
 457         return {'FINISHED'}
 458
 459
 460 def register():
 461     bpy.utils.register_module(__name__)
 462
 463     pass
 464
 465 def unregister():
 466     bpy.utils.unregister_module(__name__)
 467
 468     pass
 469
 470 if __name__ == "__main__":
 471     register()