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, 1),
  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": "https://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.types import (
  39         Operator,
  40         Panel,
  41         AddonPreferences,
  42         )
  43 from bpy.props import (
  44         BoolProperty,
  45         StringProperty,
  46         )
  47
  48
  49 # Property Definitions
  50
  51 bpy.types.WindowManager.key_shape = BoolProperty(
  52     name="Shape",
  53     description="Insert keyframes on active Shape Key layer",
  54     default=False
  55     )
  56 bpy.types.WindowManager.key_uvs = BoolProperty(
  57     name="UVs",
  58     description="Insert keyframes on active UV coordinates",
  59     default=False
  60     )
  61 bpy.types.WindowManager.key_ebevel = BoolProperty(
  62     name="E-Bevel",
  63     description="Insert keyframes on edge bevel weight",
  64     default=False
  65     )
  66 bpy.types.WindowManager.key_vbevel = BoolProperty(
  67     name="V-Bevel",
  68     description="Insert keyframes on vertex bevel weight",
  69     default=False
  70     )
  71 bpy.types.WindowManager.key_crease = BoolProperty(
  72     name="Crease",
  73     description="Insert keyframes on edge creases",
  74     default=False
  75     )
  76 bpy.types.WindowManager.key_vcols = BoolProperty(
  77     name="V-Cols",
  78     description="Insert keyframes on active Vertex Color values",
  79     default=False
  80     )
  81 bpy.types.WindowManager.key_vgroups = BoolProperty(
  82     name="V-Groups",
  83     description="Insert keyframes on active Vertex Group values",
  84     default=False
  85     )
  86 bpy.types.WindowManager.key_points = BoolProperty(
  87     name="Points",
  88     description="Insert keyframes on point locations",
  89     default=False
  90     )
  91 bpy.types.WindowManager.key_radius = BoolProperty(
  92     name="Radius",
  93     description="Insert keyframes on point radius (Shrink/Fatten)",
  94     default=False
  95     )
  96 bpy.types.WindowManager.key_tilt = BoolProperty(
  97     name="Tilt",
  98     description="Insert keyframes on point tilt",
  99     default=False
 100     )
 101
 102
 103 # Utility functions
 104
 105 def refresh_ui_keyframes():
 106     try:
 107         for area in bpy.context.screen.areas:
 108             if area.type in ('TIMELINE', 'GRAPH_EDITOR', 'DOPESHEET_EDITOR'):
 109                 area.tag_redraw()
 110     except:
 111         pass
 112
 113
 114 def insert_key(data, key):
 115     try:
 116         data.keyframe_insert(key)
 117     except:
 118         pass
 119
 120
 121 def delete_key(data, key):
 122     try:
 123         data.keyframe_delete(key)
 124     except:
 125         pass
 126
 127
 128 # GUI (Panel)
 129
 130 class VIEW3D_PT_animall(Panel):
 131     bl_space_type = 'VIEW_3D'
 132     bl_region_type = 'TOOLS'
 133     bl_category = "Animation"
 134     bl_label = 'AnimAll'
 135
 136     @classmethod
 137     def poll(self, context):
 138         if context.active_object and context.active_object.type in {'MESH', 'LATTICE', 'CURVE', 'SURFACE'}:
 139             return context.active_object.type
 140
 141     def draw(self, context):
 142         Obj = context.active_object
 143
 144         layout = self.layout
 145         col = layout.column(align=True)
 146         row = col.row()
 147
 148         if Obj.type == 'LATTICE':
 149             row.prop(context.window_manager, "key_points")
 150             row.prop(context.window_manager, "key_shape")
 151
 152         elif Obj.type == 'MESH':
 153             row.prop(context.window_manager, "key_points")
 154             row.prop(context.window_manager, "key_shape")
 155             row = col.row()
 156             row.prop(context.window_manager, "key_ebevel")
 157             row.prop(context.window_manager, "key_vbevel")
 158             row = col.row()
 159             row.prop(context.window_manager, "key_crease")
 160             row.prop(context.window_manager, "key_uvs")
 161             row = col.row()
 162             row.prop(context.window_manager, "key_vcols")
 163             row.prop(context.window_manager, "key_vgroups")
 164
 165         elif Obj.type == 'CURVE':
 166             row.prop(context.window_manager, "key_points")
 167             row.prop(context.window_manager, "key_shape")
 168             row = col.row()
 169             row.prop(context.window_manager, "key_radius")
 170             row.prop(context.window_manager, "key_tilt")
 171
 172         elif Obj.type == 'SURFACE':
 173             row.prop(context.window_manager, "key_points")
 174             row.prop(context.window_manager, "key_shape")
 175             row = col.row()
 176             row.prop(context.window_manager, "key_radius")
 177             row.prop(context.window_manager, "key_tilt")
 178
 179         layout.separator()
 180         row = layout.row(align=True)
 181         row.operator("anim.insert_keyframe_animall", icon="KEY_HLT")
 182         row.operator("anim.delete_keyframe_animall", icon="KEY_DEHLT")
 183         row = layout.row()
 184         row.operator("anim.clear_animation_animall", icon="X")
 185
 186         if context.window_manager.key_shape:
 187             ShapeKey = Obj.active_shape_key
 188             ShapeKeyIndex = Obj.active_shape_key_index
 189
 190             split = layout.split()
 191             row = split.row()
 192
 193             if ShapeKeyIndex > 0:
 194                 row.label(ShapeKey.name, icon="SHAPEKEY_DATA")
 195                 row.prop(ShapeKey, "value", text="")
 196                 row.prop(Obj, "show_only_shape_key", text="")
 197                 if ShapeKey.value < 1:
 198                     row = layout.row()
 199                     row.label('Maybe set "%s" to 1.0?' % ShapeKey.name, icon="INFO")
 200             elif ShapeKey:
 201                 row.label("Can not key on Basis Shape", icon="ERROR")
 202             else:
 203                 row.label("No active Shape Key", icon="ERROR")
 204
 205         if context.window_manager.key_points and context.window_manager.key_shape:
 206             row = layout.row()
 207             row.label('"Points" and "Shape" are redundant?', icon="INFO")
 208
 209
 210 class ANIM_OT_insert_keyframe_animall(Operator):
 211     bl_label = "Insert"
 212     bl_idname = "anim.insert_keyframe_animall"
 213     bl_description = "Insert a Keyframe"
 214     bl_options = {'REGISTER', 'UNDO'}
 215
 216     def invoke(self, context, event):
 217         self.execute(context)
 218
 219         return {'FINISHED'}
 220
 221     def execute(op, context):
 222         Obj = context.active_object
 223
 224         if Obj.type == 'MESH':
 225             Mode = False
 226             if context.mode == 'EDIT_MESH':
 227                 Mode = not Mode
 228                 bpy.ops.object.editmode_toggle()
 229
 230             Data = Obj.data
 231
 232             if context.window_manager.key_shape:
 233                 if Obj.active_shape_key_index > 0:
 234                     for Vert in Obj.active_shape_key.data:
 235                         insert_key(Vert, 'co')
 236
 237             if context.window_manager.key_points:
 238                 for Vert in Data.vertices:
 239                     insert_key(Vert, 'co')
 240
 241             if context.window_manager.key_ebevel:
 242                 for Edge in Data.edges:
 243                     insert_key(Edge, 'bevel_weight')
 244
 245             if context.window_manager.key_vbevel:
 246                 for Vert in Data.vertices:
 247                     insert_key(Vert, 'bevel_weight')
 248
 249             if context.window_manager.key_crease:
 250                 for Edge in Data.edges:
 251                     insert_key(Edge, 'crease')
 252
 253             if context.window_manager.key_vgroups:
 254                 for Vert in Data.vertices:
 255                     for Group in Vert.groups:
 256                         insert_key(Group, 'weight')
 257
 258             if context.window_manager.key_uvs:
 259                 for UV in Data.uv_layers.active.data:
 260                     insert_key(UV, 'uv')
 261
 262             if context.window_manager.key_vcols:
 263                 for VColLayer in Data.vertex_colors:
 264                     if VColLayer.active:  # only insert in active VCol layer
 265                         for Data in VColLayer.data:
 266                             insert_key(Data, 'color')
 267
 268             if Mode:
 269                 bpy.ops.object.editmode_toggle()
 270
 271         if Obj.type == 'LATTICE':
 272             Mode = False
 273             if context.mode != 'OBJECT':
 274                 Mode = not Mode
 275                 bpy.ops.object.editmode_toggle()
 276
 277             Data = Obj.data
 278
 279             if context.window_manager.key_shape:
 280                 if Obj.active_shape_key_index > 0:
 281                     for Point in Obj.active_shape_key.data:
 282                         insert_key(Point, 'co')
 283
 284             if context.window_manager.key_points:
 285                 for Point in Data.points:
 286                     insert_key(Point, 'co_deform')
 287
 288             if Mode:
 289                 bpy.ops.object.editmode_toggle()
 290
 291         if Obj.type in {'CURVE', 'SURFACE'}:
 292             Mode = False
 293             if context.mode != 'OBJECT':
 294                 Mode = not Mode
 295                 bpy.ops.object.editmode_toggle()
 296
 297             Data = Obj.data
 298
 299             # run this outside the splines loop (only once)
 300             if context.window_manager.key_shape:
 301                 if Obj.active_shape_key_index > 0:
 302                     for CV in Obj.active_shape_key.data:
 303                         insert_key(CV, 'co')
 304                         insert_key(CV, 'handle_left')
 305                         insert_key(CV, 'handle_right')
 306
 307             for Spline in Data.splines:
 308                 if Spline.type == 'BEZIER':
 309
 310                     for CV in Spline.bezier_points:
 311                         if context.window_manager.key_points:
 312                             insert_key(CV, 'co')
 313                             insert_key(CV, 'handle_left')
 314                             insert_key(CV, 'handle_right')
 315
 316                         if context.window_manager.key_radius:
 317                             insert_key(CV, 'radius')
 318
 319                         if context.window_manager.key_tilt:
 320                             insert_key(CV, 'tilt')
 321
 322                 elif Spline.type == 'NURBS':
 323                     for CV in Spline.points:
 324                         if context.window_manager.key_points:
 325                             insert_key(CV, 'co')
 326
 327                         if context.window_manager.key_radius:
 328                             insert_key(CV, 'radius')
 329
 330                         if context.window_manager.key_tilt:
 331                             insert_key(CV, 'tilt')
 332
 333             if Mode:
 334                 bpy.ops.object.editmode_toggle()
 335
 336         refresh_ui_keyframes()
 337
 338         return {'FINISHED'}
 339
 340
 341 class ANIM_OT_delete_keyframe_animall(Operator):
 342     bl_label = "Delete"
 343     bl_idname = "anim.delete_keyframe_animall"
 344     bl_description = "Delete a Keyframe"
 345     bl_options = {'REGISTER', 'UNDO'}
 346
 347     def invoke(self, context, event):
 348         self.execute(context)
 349
 350         return {'FINISHED'}
 351
 352     def execute(op, context):
 353         Obj = context.active_object
 354
 355         if Obj.type == 'MESH':
 356             Mode = False
 357             if context.mode == 'EDIT_MESH':
 358                 Mode = not Mode
 359                 bpy.ops.object.editmode_toggle()
 360
 361             Data = Obj.data
 362
 363             if context.window_manager.key_shape:
 364                 if Obj.active_shape_key:
 365                     for Vert in Obj.active_shape_key.data:
 366                         delete_key(Vert, 'co')
 367
 368             if context.window_manager.key_points:
 369                 for Vert in Data.vertices:
 370                     delete_key(Vert, 'co')
 371
 372             if context.window_manager.key_ebevel:
 373                 for Edge in Data.edges:
 374                     delete_key(Edge, 'bevel_weight')
 375
 376             if context.window_manager.key_vbevel:
 377                 for Vert in Data.vertices:
 378                     delete_key(Vert, 'bevel_weight')
 379
 380             if context.window_manager.key_crease:
 381                 for Edge in Data.edges:
 382                     delete_key(Edge, 'crease')
 383
 384             if context.window_manager.key_vgroups:
 385                 for Vert in Data.vertices:
 386                     for Group in Vert.groups:
 387                         delete_key(Group, 'weight')
 388
 389             if context.window_manager.key_uvs:
 390                 for UV in Data.uv_layers.active.data:
 391                     delete_key(UV, 'uv')
 392
 393             if context.window_manager.key_vcols:
 394                 for VColLayer in Data.vertex_colors:
 395                     if VColLayer.active:  # only delete in active VCol layer
 396                         for Data in VColLayer.data:
 397                             delete_key(Data, 'color')
 398
 399             if Mode:
 400                 bpy.ops.object.editmode_toggle()
 401
 402         if Obj.type == 'LATTICE':
 403             Mode = False
 404             if context.mode != 'OBJECT':
 405                 Mode = not Mode
 406                 bpy.ops.object.editmode_toggle()
 407
 408             Data = Obj.data
 409
 410             if context.window_manager.key_shape:
 411                 if Obj.active_shape_key:
 412                     for Point in Obj.active_shape_key.data:
 413                         delete_key(Point, 'co')
 414
 415             if context.window_manager.key_points:
 416                 for Point in Data.points:
 417                     delete_key(Point, 'co_deform')
 418
 419             if Mode:
 420                 bpy.ops.object.editmode_toggle()
 421
 422         if Obj.type in {'CURVE', 'SURFACE'}:
 423             Mode = False
 424             if context.mode != 'OBJECT':
 425                 Mode = not Mode
 426                 bpy.ops.object.editmode_toggle()
 427
 428             Data = Obj.data
 429
 430             # run this outside the splines loop (only once)
 431             if context.window_manager.key_shape:
 432                 if Obj.active_shape_key_index > 0:
 433                     for CV in Obj.active_shape_key.data:
 434                         delete_key(CV, 'co')
 435                         delete_key(CV, 'handle_left')
 436                         delete_key(CV, 'handle_right')
 437
 438             for Spline in Data.splines:
 439                 if Spline.type == 'BEZIER':
 440                     for CV in Spline.bezier_points:
 441                         if context.window_manager.key_points:
 442                             delete_key(CV, 'co')
 443                             delete_key(CV, 'handle_left')
 444                             delete_key(CV, 'handle_right')
 445                         if context.window_manager.key_radius:
 446                             delete_key(CV, 'radius')
 447                         if context.window_manager.key_tilt:
 448                             delete_key(CV, 'tilt')
 449
 450                 elif Spline.type == 'NURBS':
 451                     for CV in Spline.points:
 452                         if context.window_manager.key_points:
 453                             delete_key(CV, 'co')
 454                         if context.window_manager.key_radius:
 455                             delete_key(CV, 'radius')
 456                         if context.window_manager.key_tilt:
 457                             delete_key(CV, 'tilt')
 458
 459             if Mode:
 460                 bpy.ops.object.editmode_toggle()
 461
 462         refresh_ui_keyframes()
 463
 464         return {'FINISHED'}
 465
 466
 467 class ANIM_OT_clear_animation_animall(Operator):
 468     bl_label = "Clear Animation"
 469     bl_idname = "anim.clear_animation_animall"
 470     bl_description = ("Delete all keyframes for this object\n"
 471                       "If in a specific case it doesn't work\n"
 472                       "try to delete the keys manually")
 473     bl_options = {'REGISTER', 'UNDO'}
 474
 475     def invoke(self, context, event):
 476         wm = context.window_manager
 477         return wm.invoke_confirm(self, event)
 478
 479     def execute(self, context):
 480         try:
 481             Data = context.active_object.data
 482             Data.animation_data_clear()
 483         except:
 484             self.report({'WARNING'}, "Clear Animation could not be performed")
 485             return {'CANCELLED'}
 486
 487         refresh_ui_keyframes()
 488
 489         return {'FINISHED'}
 490
 491
 492 # Add-ons Preferences Update Panel
 493
 494 # Define Panel classes for updating
 495 panels = [
 496         VIEW3D_PT_animall
 497         ]
 498
 499
 500 def update_panel(self, context):
 501     message = "AnimAll: Updating Panel locations has failed"
 502     try:
 503         for panel in panels:
 504             if "bl_rna" in panel.__dict__:
 505                 bpy.utils.unregister_class(panel)
 506
 507         for panel in panels:
 508             panel.bl_category = context.user_preferences.addons[__name__].preferences.category
 509             bpy.utils.register_class(panel)
 510
 511     except Exception as e:
 512         print("\n[{}]\n{}\n\nError:\n{}".format(__name__, message, e))
 513         pass
 514
 515
 516 class AnimallAddonPreferences(AddonPreferences):
 517     # this must match the addon name, use '__package__'
 518     # when defining this in a submodule of a python package.
 519     bl_idname = __name__
 520
 521     category = StringProperty(
 522             name="Tab Category",
 523             description="Choose a name for the category of the panel",
 524             default="Animation",
 525             update=update_panel
 526             )
 527
 528     def draw(self, context):
 529         layout = self.layout
 530         row = layout.row()
 531         col = row.column()
 532
 533         col.label(text="Tab Category:")
 534         col.prop(self, "category", text="")
 535
 536
 537 def register():
 538     bpy.utils.register_module(__name__)
 539     update_panel(None, bpy.context)
 540     pass
 541
 542
 543 def unregister():
 544     bpy.utils.unregister_module(__name__)
 545     pass
 546
 547
 548 if __name__ == "__main__":
 549     register()