camera_turnaround.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": "Turnaround Camera",
  21     "author": "Antonio Vazquez (antonioya)",
  22     "version": (0, 2, 5),
  23     "blender": (2, 68, 0),
  24     "location": "View3D > Toolshelf > Animation Tab > Turnaround Camera",
  25     "description": "Add a camera rotation around selected object",
  26     "wiki_url": "https://wiki.blender.org/index.php/Extensions:2.6/Py/"
  27                 "Scripts/Animation/TurnaroundCamera",
  28     "category": "Camera"}
  29
  30
  31 import bpy
  32 from math import pi
  33 from bpy.props import (
  34         BoolProperty,
  35         EnumProperty,
  36         FloatProperty,
  37         PointerProperty,
  38         )
  39 from bpy.types import (
  40         Operator,
  41         Panel,
  42         PropertyGroup,
  43         )
  44
  45
  46 # ------------------------------------------------------
  47 # Action class
  48 # ------------------------------------------------------
  49 class RunAction(Operator):
  50     bl_idname = "object.rotate_around"
  51     bl_label = "Turnaround"
  52     bl_description = "Create camera rotation around selected object"
  53
  54     def execute(self, context):
  55         # ----------------------
  56         # Save old data
  57         # ----------------------
  58         scene = context.scene
  59         turn_camera = scene.turn_camera
  60         selectobject = context.active_object
  61         camera = context.scene.camera
  62         savedcursor = bpy.context.scene.cursor_location.copy()  # cursor position
  63         savedframe = scene.frame_current
  64         if turn_camera.use_cursor is False:
  65             bpy.ops.view3d.snap_cursor_to_selected()
  66
  67         # -------------------------
  68         # Create empty and parent
  69         # -------------------------
  70         bpy.ops.object.empty_add(type='PLAIN_AXES')
  71         myempty = context.active_object
  72
  73         myempty.location = selectobject.location
  74         savedstate = myempty.matrix_world
  75         myempty.parent = selectobject
  76         myempty.name = 'MCH_Rotation_target'
  77         myempty.matrix_world = savedstate
  78
  79         # -------------------------
  80         # Parent camera to empty
  81         # -------------------------
  82         savedstate = camera.matrix_world
  83         camera.parent = myempty
  84         camera.matrix_world = savedstate
  85
  86         # -------------------------
  87         # Now add revolutions
  88         # (make empty active object)
  89         # -------------------------
  90         bpy.ops.object.select_all(False)
  91         myempty.select = True
  92         context.scene.objects.active = myempty
  93         # save current configuration
  94         savedinterpolation = context.user_preferences.edit.keyframe_new_interpolation_type
  95         # change interpolation mode
  96         context.user_preferences.edit.keyframe_new_interpolation_type = 'LINEAR'
  97         # create first frame
  98         myempty.rotation_euler = (0, 0, 0)
  99         myempty.empty_display_size = 0.1
 100         context.scene.frame_set(scene.frame_start)
 101         myempty.keyframe_insert(data_path='rotation_euler', frame=scene.frame_start)
 102
 103         # Clear the Camera Animations if the option is checked
 104         if turn_camera.reset_cam_anim:
 105             try:
 106                 if bpy.data.cameras[camera.name].animation_data:
 107                     bpy.data.cameras[camera.name].animation_data_clear()
 108             except Exception as e:
 109                 print("\n[Camera Turnaround]\nWarning: {}\n".format(e))
 110
 111         # Dolly zoom
 112         if turn_camera.dolly_zoom != "0":
 113             bpy.data.cameras[camera.name].lens = turn_camera.camera_from_lens
 114             bpy.data.cameras[camera.name].keyframe_insert('lens', frame=scene.frame_start)
 115
 116         # Calculate rotation XYZ
 117         ix = -1 if turn_camera.inverse_x else 1
 118         iy = -1 if turn_camera.inverse_y else 1
 119         iz = -1 if turn_camera.inverse_z else 1
 120
 121         xrot = (pi * 2) * turn_camera.camera_revol_x * ix
 122         yrot = (pi * 2) * turn_camera.camera_revol_y * iy
 123         zrot = (pi * 2) * turn_camera.camera_revol_z * iz
 124
 125         # create middle frame
 126         if turn_camera.back_forw is True:
 127             myempty.rotation_euler = (xrot, yrot, zrot)
 128             myempty.keyframe_insert(
 129                         data_path='rotation_euler',
 130                         frame=((scene.frame_end - scene.frame_start) / 2)
 131                         )
 132             # reverse
 133             xrot *= -1
 134             yrot *= -1
 135             zrot = 0
 136
 137         # Dolly zoom
 138         if turn_camera.dolly_zoom == "2":
 139             bpy.data.cameras[camera.name].lens = turn_camera.camera_to_lens
 140             bpy.data.cameras[camera.name].keyframe_insert(
 141                                             'lens',
 142                                             frame=((scene.frame_end - scene.frame_start) / 2)
 143                                             )
 144
 145         # create last frame
 146         myempty.rotation_euler = (xrot, yrot, zrot)
 147         myempty.keyframe_insert(data_path='rotation_euler', frame=scene.frame_end)
 148         # Dolly zoom
 149         if turn_camera.dolly_zoom != "0":
 150             if turn_camera.dolly_zoom == "1":
 151                 bpy.data.cameras[camera.name].lens = turn_camera.camera_to_lens  # final
 152             else:
 153                 bpy.data.cameras[camera.name].lens = turn_camera.camera_from_lens  # back to init
 154
 155             bpy.data.cameras[camera.name].keyframe_insert(
 156                                             'lens', frame=scene.frame_end
 157                                             )
 158
 159         # Track constraint
 160         if turn_camera.track is True:
 161             bpy.context.scene.objects.active = camera
 162             bpy.ops.object.constraint_add(type='TRACK_TO')
 163             bpy.context.object.constraints[-1].track_axis = 'TRACK_NEGATIVE_Z'
 164             bpy.context.object.constraints[-1].up_axis = 'UP_Y'
 165             bpy.context.object.constraints[-1].target = bpy.data.objects[myempty.name]
 166
 167         # back previous configuration
 168         context.user_preferences.edit.keyframe_new_interpolation_type = savedinterpolation
 169         bpy.context.scene.cursor_location = savedcursor
 170
 171         # -------------------------
 172         # Back to old selection
 173         # -------------------------
 174         bpy.ops.object.select_all(False)
 175         selectobject.select = True
 176         bpy.context.scene.objects.active = selectobject
 177         bpy.context.scene.frame_set(savedframe)
 178
 179         return {'FINISHED'}
 180
 181
 182 # ------------------------------------------------------
 183 # Define Properties
 184 # ------------------------------------------------------
 185 class CameraTurnProps(PropertyGroup):
 186
 187     camera_revol_x = FloatProperty(
 188             name='X', min=0, max=25,
 189             default=0, precision=2,
 190             description='Number total of revolutions in X axis'
 191             )
 192     camera_revol_y = FloatProperty(
 193             name='Y', min=0, max=25,
 194             default=0, precision=2,
 195             description='Number total of revolutions in Y axis'
 196             )
 197     camera_revol_z = FloatProperty(
 198             name='Z', min=0, max=25,
 199             default=1, precision=2,
 200             description='Number total of revolutions in Z axis'
 201             )
 202     inverse_x = BoolProperty(
 203             name="-X",
 204             description="Inverse rotation",
 205             default=False
 206             )
 207     inverse_y = BoolProperty(
 208             name="-Y",
 209             description="Inverse rotation",
 210             default=False
 211             )
 212     inverse_z = BoolProperty(
 213             name="-Z",
 214             description="Inverse rotation",
 215             default=False
 216             )
 217     use_cursor = BoolProperty(
 218             name="Use cursor position",
 219             description="Use cursor position instead of object origin",
 220             default=False
 221             )
 222     back_forw = BoolProperty(
 223             name="Back and forward",
 224             description="Create back and forward animation",
 225             default=False
 226             )
 227     dolly_zoom = EnumProperty(
 228             items=(
 229                 ('0', "None", ""),
 230                 ('1', "Dolly zoom", ""),
 231                 ('2', "Dolly zoom B/F", "")
 232                 ),
 233             name="Lens Effects",
 234             description="Create a camera lens movement"
 235             )
 236     camera_from_lens = FloatProperty(
 237             name="From",
 238             min=1, max=500, default=35,
 239             precision=3,
 240             description="Start lens value"
 241             )
 242     camera_to_lens = FloatProperty(
 243             name="To",
 244             min=1, max=500,
 245             default=35, precision=3,
 246             description="End lens value"
 247             )
 248     track = BoolProperty(
 249             name="Create track constraint",
 250             description="Add a track constraint to the camera",
 251             default=False
 252             )
 253     reset_cam_anim = BoolProperty(
 254             name="Clear Camera",
 255             description="Clear previous camera animations if there are any\n"
 256                         "(For instance, previous Dolly Zoom)",
 257             default=False
 258             )
 259
 260
 261 # ------------------------------------------------------
 262 # UI Class
 263 # ------------------------------------------------------
 264 class PanelUI(Panel):
 265     bl_idname = "CAMERA_TURN_PT_main"
 266     bl_label = "Turnaround Camera"
 267     bl_space_type = "VIEW_3D"
 268     bl_region_type = "TOOLS"
 269     bl_category = "Animation"
 270
 271     def draw(self, context):
 272         layout = self.layout
 273         scene = context.scene
 274         turn_camera = scene.turn_camera
 275
 276         try:
 277             bpy.context.scene.camera.name
 278         except AttributeError:
 279             row = layout.row(align=False)
 280             row.label("No defined camera for scene", icon="INFO")
 281             return
 282
 283         if context.active_object is not None:
 284             if context.active_object.type != 'CAMERA':
 285                 buf = context.active_object.name
 286                 row = layout.row(align=True)
 287                 row.operator("object.rotate_around", icon='OUTLINER_DATA_CAMERA')
 288                 box = row.box()
 289                 box.scale_y = 0.5
 290                 box.label(buf, icon='MESH_DATA')
 291                 row = layout.row(align=False)
 292                 row.prop(scene, "camera")
 293
 294                 layout.label("Rotation:")
 295                 row = layout.row(align=True)
 296                 row.prop(scene, "frame_start")
 297                 row.prop(scene, "frame_end")
 298
 299                 col = layout.column(align=True)
 300                 split = col.split(percentage=0.85, align=True)
 301                 split.prop(turn_camera, "camera_revol_x")
 302                 split.prop(turn_camera, "inverse_x", toggle=True)
 303                 split = col.split(percentage=0.85, align=True)
 304                 split.prop(turn_camera, "camera_revol_y")
 305                 split.prop(turn_camera, "inverse_y", toggle=True)
 306                 split = col.split(percentage=0.85, align=True)
 307                 split.prop(turn_camera, "camera_revol_z")
 308                 split.prop(turn_camera, "inverse_z", toggle=True)
 309
 310                 col = layout.column(align=True)
 311                 col.label("Options:")
 312                 row = col.row(align=True)
 313                 row.prop(turn_camera, "back_forw", toggle=True)
 314                 row.prop(turn_camera, "reset_cam_anim", toggle=True)
 315                 col.prop(turn_camera, "track", toggle=True)
 316                 col.prop(turn_camera, "use_cursor", toggle=True)
 317
 318                 row = layout.row()
 319                 row.prop(turn_camera, "dolly_zoom")
 320                 if turn_camera.dolly_zoom != "0":
 321                     row = layout.row(align=True)
 322                     row.prop(turn_camera, "camera_from_lens")
 323                     row.prop(turn_camera, "camera_to_lens")
 324
 325             else:
 326                 buf = "No valid object selected"
 327                 layout.label(buf, icon='MESH_DATA')
 328
 329
 330 # ------------------------------------------------------
 331 # Registration
 332 # ------------------------------------------------------
 333 def register():
 334     bpy.utils.register_class(RunAction)
 335     bpy.utils.register_class(PanelUI)
 336     bpy.utils.register_class(CameraTurnProps)
 337     bpy.types.Scene.turn_camera = PointerProperty(type=CameraTurnProps)
 338
 339
 340 def unregister():
 341     bpy.utils.unregister_class(RunAction)
 342     bpy.utils.unregister_class(PanelUI)
 343     bpy.utils.unregister_class(CameraTurnProps)
 344     del bpy.types.Scene.turn_camera
 345
 346
 347 if __name__ == "__main__":
 348     register()