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