Skip to content
Snippets Groups Projects
Commit a7fb1b31 authored by Bartek Skorupa's avatar Bartek Skorupa
Browse files

1. Fix for active camera export - when cameras bound to markers - option to... 

1. Fix for active camera export - when cameras bound to markers - option to export as one camera.    2. poll fix - changed from active object to active camera or selected objects    3. Export of active camera is now optional    4. Export of selected cameras is now optional    5. Naming fix - Changed tracking data export naming for user from BUNDLES to 3D MARKERS    6. Toolbar fix - wrapped in box, options added    7. Code cleanup - Prepared structure for future options of exporting lamps as AE lights, planes as AE solids and object tracking bundles, additional conditions added to avoid AE errors in some rare cases
parent eeb3fc50
Branches
Tags
No related merge requests found
Hide whitespace changes
Inline Side-by-side
io_export_after_effects.py
+ 331
− 150
View file @ a7fb1b31
...@@ -19,9 +19,9 @@ ...@@ -19,9 +19,9 @@
# #
bl_info = { bl_info = {
'name': 'Export: Adobe After Effects (.jsx)', 'name': 'Export: Adobe After Effects (.jsx)',
'description': 'Export selected cameras, objects & bundles to Adobe After Effects CS3 and above', 'description': 'Export cameras, selected objects & camera solution 3D Markers to Adobe After Effects CS3 and above',
'author': 'Bartek Skorupa', 'author': 'Bartek Skorupa',
'version': (0, 5, 9), 'version': (0, 6, 0),
'blender': (2, 6, 1), 'blender': (2, 6, 1),
'location': 'File > Export > Adobe After Effects (.jsx)', 'location': 'File > Export > Adobe After Effects (.jsx)',
"warning": "", "warning": "",
...@@ -33,9 +33,10 @@ bl_info = { ...@@ -33,9 +33,10 @@ bl_info = {
} }
from math import pi
import bpy import bpy
import datetime import datetime
from math import pi
from mathutils import Matrix
# create list of static blender's data # create list of static blender's data
...@@ -44,6 +45,9 @@ def get_comp_data(context): ...@@ -44,6 +45,9 @@ def get_comp_data(context):
aspect_x = scene.render.pixel_aspect_x aspect_x = scene.render.pixel_aspect_x
aspect_y = scene.render.pixel_aspect_y aspect_y = scene.render.pixel_aspect_y
aspect = aspect_x / aspect_y aspect = aspect_x / aspect_y
start = scene.frame_start
end = scene.frame_end
active_cam_frames = get_active_cam_for_each_frame(scene, start, end)
fps = scene.render.fps fps = scene.render.fps
return { return {
...@@ -52,64 +56,94 @@ def get_comp_data(context): ...@@ -52,64 +56,94 @@ def get_comp_data(context):
'height': scene.render.resolution_y, 'height': scene.render.resolution_y,
'aspect': aspect, 'aspect': aspect,
'fps': fps, 'fps': fps,
'start': scene.frame_start, 'start': start,
'end': scene.frame_end, 'end': end,
'duration': (scene.frame_end - scene.frame_start + 1.0) / fps, 'duration': (end - start + 1.0) / fps,
'active_cam_frames': active_cam_frames,
'curframe': scene.frame_current, 'curframe': scene.frame_current,
} }
# create list of active camera for each frame in case active camera is set by markers
def get_active_cam_for_each_frame(scene, start, end):
active_cam_frames = []
sorted_markers = []
markers = scene.timeline_markers
if markers:
for marker in markers:
if marker.camera:
sorted_markers.append([marker.frame, marker])
sorted_markers = sorted(sorted_markers)
for i, marker in enumerate (sorted_markers):
cam = marker[1].camera
if i is 0 and marker[0] > start:
start_range = start
else:
start_range = sorted_markers[i][0]
if len(sorted_markers) > i + 1:
end_range = sorted_markers[i+1][0] - 1
else:
end_range = end
for i in range(start_range, end_range + 1):
active_cam_frames.append(cam)
if not active_cam_frames:
if scene.camera:
# in this case active_cam_frames array will have legth of 1. This will indicate that there is only one active cam in all frames
active_cam_frames.append(scene.camera)
return(active_cam_frames)
# create managable list of selected objects # create managable list of selected objects
# (only selected objects will be analyzed and exported)
def get_selected(context): def get_selected(context):
cameras = [] # list of selected cameras cameras = [] # list of selected cameras
cams_names = [] # list of selected cameras' names (prevent from calling "ConvertName(ob)" function too many times) solids = [] # list of all selected meshes that can be exported as AE's solids
lights = [] # list of all selected lamps that can be exported as AE's lights
nulls = [] # list of all selected objects exept cameras (will be used to create nulls in AE) nulls = [] # list of all selected objects exept cameras (will be used to create nulls in AE)
nulls_names = [] # list of above objects names (prevent from calling "ConvertName(ob)" function too many times)
obs = context.selected_objects obs = context.selected_objects
for ob in obs: for ob in obs:
if ob.type == 'CAMERA': if ob.type == 'CAMERA':
cameras.append(ob) cameras.append([ob, convert_name(ob.name)])
cams_names.append(convert_name(False, ob))
elif is_plane(ob):
# not ready yet. is_plane(object) returns False in all cases. This is temporary
solids.append([ob, convert_name(ob.name)])
elif ob.type == 'LAMP':
# not ready yet. Lamps will be exported as nulls. This is temporary
nulls.append([ob, convert_name(ob.name)])
else: else:
nulls.append(ob) nulls.append([ob, convert_name(ob.name)])
nulls_names.append(convert_name(False, ob))
# If no camera is selected - export at least scene's camera if exists
if not cameras:
cam = context.scene.camera
if cam:
cameras.append(cam)
cams_names.append(convert_name(False, cam))
selection = { selection = {
'cameras': cameras, 'cameras': cameras,
'cams_names': cams_names, 'solids': solids,
'lights': lights,
'nulls': nulls, 'nulls': nulls,
'nulls_names': nulls_names,
} }
return selection return selection
# check if object is plane and can be exported as AE's solid
def is_plane(object):
# work in progress. Not ready yet
return False
# convert names of objects to avoid errors in AE. # convert names of objects to avoid errors in AE.
def convert_name(is_comp, ob): def convert_name(name):
if is_comp: name = "_" + name
ob_name = ob
ob_name = ob_name.replace('"', "_")
else:
ob_name = "_" + ob.name
if ob_name[0].isdigit(): if name[0].isdigit():
ob_name = "_" + ob_name name = "_" + name
ob_name = bpy.path.clean_name(ob_name) name = bpy.path.clean_name(name)
ob_name = ob_name.replace("-", "_") name = name.replace("-", "_")
return ob_name return name
# get object's blender's location and rotation and return AE's Position and Rotation/Orientation # get object's blender's location rotation and scale and return AE's Position, Rotation/Orientation and scale
# this function will be called for every object for every frame # this function will be called for every object for every frame
def convert_transform_matrix(matrix, width, height, aspect, x_rot_correction=False): def convert_transform_matrix(matrix, width, height, aspect, x_rot_correction=False):
...@@ -126,7 +160,7 @@ def convert_transform_matrix(matrix, width, height, aspect, x_rot_correction=Fal ...@@ -126,7 +160,7 @@ def convert_transform_matrix(matrix, width, height, aspect, x_rot_correction=Fal
b_rot_y = b_rot_y / pi * 180.0 b_rot_y = b_rot_y / pi * 180.0
b_rot_z = b_rot_z / pi * 180.0 b_rot_z = b_rot_z / pi * 180.0
# convert to AE Position and Rotation # convert to AE Position Rotation and Scale
# Axes in AE are different. AE's X is blender's X, AE's Y is negative Blender's Z, AE's Z is Blender's Y # Axes in AE are different. AE's X is blender's X, AE's Y is negative Blender's Z, AE's Z is Blender's Y
x = (b_loc_x * 100.0) / aspect + width / 2.0 # calculate AE's X position x = (b_loc_x * 100.0) / aspect + width / 2.0 # calculate AE's X position
y = (-b_loc_z * 100.0) + (height / 2.0) # calculate AE's Y position y = (-b_loc_z * 100.0) + (height / 2.0) # calculate AE's Y position
...@@ -135,18 +169,12 @@ def convert_transform_matrix(matrix, width, height, aspect, x_rot_correction=Fal ...@@ -135,18 +169,12 @@ def convert_transform_matrix(matrix, width, height, aspect, x_rot_correction=Fal
rx = b_rot_x # calculate AE's X rotation. Will become AE's RotationX property rx = b_rot_x # calculate AE's X rotation. Will become AE's RotationX property
ry = -b_rot_z # calculate AE's Y rotation. Will become AE's OrientationY property ry = -b_rot_z # calculate AE's Y rotation. Will become AE's OrientationY property
rz = b_rot_y # calculate AE's Z rotation. Will become AE's OrentationZ property rz = b_rot_y # calculate AE's Z rotation. Will become AE's OrentationZ property
sx = b_scale_x * 100.0 sx = b_scale_x * 100.0 # scale of 1.0 is 100% in AE
sy = b_scale_z * 100.0 sy = b_scale_z * 100.0 # scale of 1.0 is 100% in AE
sz = b_scale_y * 100.0 sz = b_scale_y * 100.0 # scale of 1.0 is 100% in AE
return x, y, z, rx, ry, rz, sx, sy, sz return x, y, z, rx, ry, rz, sx, sy, sz
def convert_transform(obj, width, height, aspect, x_rot_correction=False):
matrix = obj.matrix_world.copy()
return convert_transform_matrix(matrix, width, height, aspect, x_rot_correction)
# get camera's lens and convert to AE's "zoom" value in pixels # get camera's lens and convert to AE's "zoom" value in pixels
# this function will be called for every camera for every frame # this function will be called for every camera for every frame
# #
...@@ -204,10 +232,14 @@ def convert_lens(camera, width, height, aspect): ...@@ -204,10 +232,14 @@ def convert_lens(camera, width, height, aspect):
return zoom return zoom
# convert object bundle's matrix. Not ready yet. Temporarily not active
#def get_ob_bundle_matrix_world(cam_matrix_world, bundle_matrix):
# matrix = cam_matrix_basis
# return matrix
# jsx script for AE creation # jsx script for AE creation
def write_jsx_file(file, data, selection, export_bundles, include_rotation, include_scale): def write_jsx_file(file, data, selection, include_active_cam, include_selected_cams, include_selected_objects, include_cam_bundles, include_rotation, include_scale):
from mathutils import Matrix
print("\n---------------------------\n- Export to After Effects -\n---------------------------") print("\n---------------------------\n- Export to After Effects -\n---------------------------")
#store the current frame to restore it at the enf of export #store the current frame to restore it at the enf of export
...@@ -216,12 +248,22 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl ...@@ -216,12 +248,22 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
js_data = { js_data = {
'times': '', 'times': '',
'cameras': {}, 'cameras': {},
'objects': {}, 'solids': {},
'lights': {},
'nulls': {},
'bundles_cam': {},
'bundles_ob': {}, # not ready yet
} }
# create camera structure # create structure for active camera/cameras
for i, cam in enumerate(selection['cameras']): # more than one camera can be selected active_cam_name = ''
name_ae = selection['cams_names'][i] if include_active_cam and data['active_cam_frames'] != []:
# check if more that one active cam exist (true if active cams set by markers)
if len(data['active_cam_frames']) is 1:
name_ae = convert_name(data['active_cam_frames'][0].name) # take name of the only active camera in scene
else:
name_ae = 'Active_Camera'
active_cam_name = name_ae # store name to be used when creating keyframes for active cam.
js_data['cameras'][name_ae] = { js_data['cameras'][name_ae] = {
'position': '', 'position': '',
'pointOfInterest': '', 'pointOfInterest': '',
...@@ -230,52 +272,170 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl ...@@ -230,52 +272,170 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
'zoom': '', 'zoom': '',
} }
# create object structure # create camera structure for selected cameras
for i, obj in enumerate(selection['nulls']): # nulls representing blender's obs except cameras if include_selected_cams:
name_ae = selection['nulls_names'][i] for i, cam in enumerate(selection['cameras']): # more than one camera can be selected
js_data['objects'][name_ae] = { if cam[1] != active_cam_name:
'position': '', name_ae = selection['cameras'][i][1]
'orientation': '', js_data['cameras'][name_ae] = {
'rotationX': '', 'position': '',
'scale': '', 'pointOfInterest': '',
} 'orientation': '',
'rotationX': '',
'zoom': '',
}
# create structure for solids. Not ready yet. Temporarily not active
# for i, obj in enumerate(selection['solids']):
# name_ae = selection['solids'][i][1]
# js_data['solids'][name_ae] = {
# 'position': '',
# 'orientation': '',
# 'rotationX': '',
# 'scale': '',
# }
# create structure for lights. Not ready yet. Temporarily not active
# for i, obj in enumerate(selection['lights']):
# name_ae = selection['lights'][i][1]
# js_data['nulls'][name_ae] = {
# 'position': '',
# 'orientation': '',
# 'rotationX': '',
# 'scale': '',
# }
# create structure for nulls
for i, obj in enumerate(selection['nulls']): # nulls representing blender's obs except cameras, lamps and solids
if include_selected_objects:
name_ae = selection['nulls'][i][1]
js_data['nulls'][name_ae] = {
'position': '',
'orientation': '',
'rotationX': '',
'scale': '',
}
# create structure for cam bundles including positions (cam bundles don't move)
if include_cam_bundles:
# go through each selected Camera and active cameras
selected_cams = []
active_cams = []
if include_active_cam:
active_cams = data['active_cam_frames']
if include_selected_cams:
for cam in selection['cameras']:
selected_cams.append(cam[0])
# list of cameras that will be checked for 'CAMERA SOLVER'
cams = list(set.union(set(selected_cams), set(active_cams)))
for cam in cams:
# go through each constraints of this camera
for constraint in cam.constraints:
# does the camera have a Camera Solver constraint
if constraint.type == 'CAMERA_SOLVER':
# Which movie clip does it use ?
if constraint.use_active_clip:
clip = data['scn'].active_clip
else:
clip = constraint.clip
# get all keyframes for each objects and store into dico # go through each tracking point
for track in clip.tracking.tracks:
# Does this tracking point have a bundle (has its 3D position been solved)
if track.has_bundle:
# get the name of the tracker
name_ae = convert_name(str(cam.name) + '__' + str(track.name))
js_data['bundles_cam'][name_ae] = {
'position': '',
}
# bundles are in camera space. Transpose to world space
matrix = Matrix.Translation(cam.matrix_basis.copy() * track.bundle)
# convert the position into AE space
ae_transform = convert_transform_matrix(matrix, data['width'], data['height'], data['aspect'], x_rot_correction=False)
js_data['bundles_cam'][name_ae]['position'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2])
# get all keyframes for each object and store in dico
for frame in range(data['start'], data['end'] + 1): for frame in range(data['start'], data['end'] + 1):
print("working on frame: " + str(frame)) print("working on frame: " + str(frame))
data['scn'].frame_set(frame) data['scn'].frame_set(frame)
#get time for this loop # get time for this loop
js_data['times'] += '%f ,' % ((frame - data['start']) / data['fps']) js_data['times'] += '%f ,' % ((frame - data['start']) / data['fps'])
# keyframes for all cameras # keyframes for active camera/cameras
for i, cam in enumerate(selection['cameras']): if include_active_cam and data['active_cam_frames'] != []:
#get cam name if len(data['active_cam_frames']) == 1:
name_ae = selection['cams_names'][i] cur_cam_index = 0
#convert cam position to AE space else:
ae_transform = convert_transform(cam, data['width'], data['height'], data['aspect'], x_rot_correction=True) cur_cam_index = frame - data['start']
#convert Blender's cam zoom to AE's active_cam = data['active_cam_frames'][cur_cam_index]
zoom = convert_lens(cam, data['width'], data['height'], data['aspect']) # get cam name
#store all the value into dico name_ae = active_cam_name
# convert cam transform properties to AE space
ae_transform = convert_transform_matrix(active_cam.matrix_world.copy(), data['width'], data['height'], data['aspect'], x_rot_correction=True)
# convert Blender's lens to AE's zoom in pixels
zoom = convert_lens(active_cam, data['width'], data['height'], data['aspect'])
# store all values in dico
js_data['cameras'][name_ae]['position'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2]) js_data['cameras'][name_ae]['position'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2])
js_data['cameras'][name_ae]['pointOfInterest'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2]) js_data['cameras'][name_ae]['pointOfInterest'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2])
js_data['cameras'][name_ae]['orientation'] += '[%f,%f,%f],' % (0, ae_transform[4], ae_transform[5]) js_data['cameras'][name_ae]['orientation'] += '[%f,%f,%f],' % (0, ae_transform[4], ae_transform[5])
js_data['cameras'][name_ae]['rotationX'] += '%f ,' % (ae_transform[3]) js_data['cameras'][name_ae]['rotationX'] += '%f ,' % (ae_transform[3])
js_data['cameras'][name_ae]['zoom'] += '[%f],' % (zoom) js_data['cameras'][name_ae]['zoom'] += '[%f],' % (zoom)
#keyframes for all nulls # keyframes for selected cameras
for i, ob in enumerate(selection['nulls']): if include_selected_cams:
#get object name for i, cam in enumerate(selection['cameras']):
name_ae = selection['nulls_names'][i] if cam[1] != active_cam_name:
#convert ob position to AE space # get cam name
ae_transform = convert_transform(ob, data['width'], data['height'], data['aspect'], x_rot_correction=False) name_ae = selection['cameras'][i][1]
#store all datas into dico # convert cam transform properties to AE space
js_data['objects'][name_ae]['position'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2]) ae_transform = convert_transform_matrix(cam[0].matrix_world.copy(), data['width'], data['height'], data['aspect'], x_rot_correction=True)
if include_rotation: # convert Blender's lens to AE's zoom in pixels
js_data['objects'][name_ae]['orientation'] += '[%f,%f,%f],' % (0, ae_transform[4], ae_transform[5]) zoom = convert_lens(cam[0], data['width'], data['height'], data['aspect'])
js_data['objects'][name_ae]['rotationX'] += '%f ,' % (ae_transform[3]) # store all values in dico
if include_scale: js_data['cameras'][name_ae]['position'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2])
js_data['objects'][name_ae]['scale'] += '[%f,%f,%f],' % (ae_transform[6], ae_transform[7], ae_transform[8]) js_data['cameras'][name_ae]['pointOfInterest'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2])
js_data['cameras'][name_ae]['orientation'] += '[%f,%f,%f],' % (0, ae_transform[4], ae_transform[5])
js_data['cameras'][name_ae]['rotationX'] += '%f ,' % (ae_transform[3])
js_data['cameras'][name_ae]['zoom'] += '[%f],' % (zoom)
# keyframes for all solids. Not ready yet. Temporarily not active
# for i, ob in enumerate(selection['solids']):
# #get object name
# name_ae = selection['solids'][i][1]
# #convert ob position to AE space
# keyframes for all lights. Not ready yet. Temporarily not active
# for i, ob in enumerate(selection['lights']):
# #get object name
# name_ae = selection['lights'][i][1]
# #convert ob position to AE space
# keyframes for all nulls
if include_selected_objects:
for i, ob in enumerate(selection['nulls']):
# get object name
name_ae = selection['nulls'][i][1]
# convert ob transform properties to AE space
ae_transform = convert_transform_matrix(ob[0].matrix_world.copy(), data['width'], data['height'], data['aspect'], x_rot_correction=False)
# store all values in dico
js_data['nulls'][name_ae]['position'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2])
if include_rotation:
js_data['nulls'][name_ae]['orientation'] += '[%f,%f,%f],' % (0, ae_transform[4], ae_transform[5])
js_data['nulls'][name_ae]['rotationX'] += '%f ,' % (ae_transform[3])
if include_scale:
js_data['nulls'][name_ae]['scale'] += '[%f,%f,%f],' % (ae_transform[6], ae_transform[7], ae_transform[8])
# keyframes for all object bundles. Not ready yet.
#
#
#
# ---- write JSX file # ---- write JSX file
jsx_file = open(file, 'w') jsx_file = open(file, 'w')
...@@ -291,7 +451,7 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl ...@@ -291,7 +451,7 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
jsx_file.write('Exported with io_export_after_effects.py\n') jsx_file.write('Exported with io_export_after_effects.py\n')
jsx_file.write('**************************************/\n\n\n\n') jsx_file.write('**************************************/\n\n\n\n')
#wrap in function # wrap in function
jsx_file.write("function compFromBlender(){\n") jsx_file.write("function compFromBlender(){\n")
# create new comp # create new comp
jsx_file.write('\nvar compName = prompt("Blender Comp\'s Name \\nEnter Name of newly created Composition","BlendComp","Composition\'s Name");') jsx_file.write('\nvar compName = prompt("Blender Comp\'s Name \\nEnter Name of newly created Composition","BlendComp","Composition\'s Name");')
...@@ -299,6 +459,37 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl ...@@ -299,6 +459,37 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
jsx_file.write('\nvar newComp = app.project.items.addComp(compName, %i, %i, %f, %f, %i);\n\n\n' % jsx_file.write('\nvar newComp = app.project.items.addComp(compName, %i, %i, %f, %f, %i);\n\n\n' %
(data['width'], data['height'], data['aspect'], data['duration'], data['fps'])) (data['width'], data['height'], data['aspect'], data['duration'], data['fps']))
# create camera bundles (nulls)
jsx_file.write('// ************** CAMERA 3D MARKERS **************\n\n\n')
for i, obj in enumerate(js_data['bundles_cam']):
name_ae = obj
jsx_file.write('var %s = newComp.layers.addNull();\n' % (name_ae))
jsx_file.write('%s.threeDLayer = true;\n' % name_ae)
jsx_file.write('%s.source.name = "%s";\n' % (name_ae, name_ae))
jsx_file.write('%s.property("position").setValue(%s);\n\n\n' % (name_ae, js_data['bundles_cam'][obj]['position']))
# create object bundles (not ready yet)
# create objects (nulls)
jsx_file.write('// ************** OBJECTS **************\n\n\n')
for i, obj in enumerate(js_data['nulls']):
name_ae = obj
jsx_file.write('var %s = newComp.layers.addNull();\n' % (name_ae))
jsx_file.write('%s.threeDLayer = true;\n' % name_ae)
jsx_file.write('%s.source.name = "%s";\n' % (name_ae, name_ae))
jsx_file.write('%s.property("position").setValuesAtTimes([%s],[%s]);\n' % (name_ae, js_data['times'], js_data['nulls'][obj]['position']))
if include_rotation:
jsx_file.write('%s.property("orientation").setValuesAtTimes([%s],[%s]);\n' % (name_ae, js_data['times'], js_data['nulls'][obj]['orientation']))
jsx_file.write('%s.property("rotationX").setValuesAtTimes([%s],[%s]);\n' % (name_ae, js_data['times'], js_data['nulls'][obj]['rotationX']))
jsx_file.write('%s.property("rotationY").setValue(0);\n' % name_ae)
jsx_file.write('%s.property("rotationZ").setValue(0);\n\n\n' % name_ae)
if include_scale:
jsx_file.write('%s.property("scale").setValuesAtTimes([%s],[%s]);\n\n\n' % (name_ae, js_data['times'], js_data['nulls'][obj]['scale']))
# create solids (not ready yet)
# create lights (not ready yet)
# create cameras # create cameras
jsx_file.write('// ************** CAMERAS **************\n\n\n') jsx_file.write('// ************** CAMERAS **************\n\n\n')
for i, cam in enumerate(js_data['cameras']): # more than one camera can be selected for i, cam in enumerate(js_data['cameras']): # more than one camera can be selected
...@@ -312,55 +503,6 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl ...@@ -312,55 +503,6 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
jsx_file.write('%s.property("rotationZ").setValue(0);\n' % name_ae) jsx_file.write('%s.property("rotationZ").setValue(0);\n' % name_ae)
jsx_file.write('%s.property("zoom").setValuesAtTimes([%s],[%s]);\n\n\n' % (name_ae, js_data['times'], js_data['cameras'][cam]['zoom'])) jsx_file.write('%s.property("zoom").setValuesAtTimes([%s],[%s]);\n\n\n' % (name_ae, js_data['times'], js_data['cameras'][cam]['zoom']))
# create objects
jsx_file.write('// ************** OBJECTS **************\n\n\n')
for i, obj in enumerate(js_data['objects']): # more than one camera can be selected
name_ae = obj
jsx_file.write('var %s = newComp.layers.addNull();\n' % (name_ae))
jsx_file.write('%s.threeDLayer = true;\n' % name_ae)
jsx_file.write('%s.source.name = "%s";\n' % (name_ae, name_ae))
jsx_file.write('%s.property("position").setValuesAtTimes([%s],[%s]);\n' % (name_ae, js_data['times'], js_data['objects'][obj]['position']))
jsx_file.write('%s.property("orientation").setValuesAtTimes([%s],[%s]);\n' % (name_ae, js_data['times'], js_data['objects'][obj]['orientation']))
jsx_file.write('%s.property("rotationX").setValuesAtTimes([%s],[%s]);\n' % (name_ae, js_data['times'], js_data['objects'][obj]['rotationX']))
jsx_file.write('%s.property("rotationY").setValue(0);\n' % name_ae)
jsx_file.write('%s.property("rotationZ").setValue(0);\n\n\n' % name_ae)
jsx_file.write('%s.property("scale").setValuesAtTimes([%s],[%s]);\n' % (name_ae, js_data['times'], js_data['objects'][obj]['scale']))
# create Bundles
if export_bundles:
jsx_file.write('// ************** BUNDLES (3d tracks) **************\n\n\n')
#Bundles are linked to MovieClip, so we have to find which MC is linked to our selected camera (if any?)
mc = ''
#go through each selected Cameras
for cam in selection['cameras']:
#go through each constrains of this camera
for constrain in cam.constraints:
#does the camera have a Camera Solver constrain
if constrain.type == 'CAMERA_SOLVER':
#Which movie clip does it use ?
if constrain.use_active_clip:
mc = data['scn'].active_clip
else:
mc = constrain.clip
#go through each tracking point
for track in mc.tracking.tracks:
#is this tracking point has a Bundles (does it's 3D position has been solved)
if track.has_bundle:
# bundle are in camera space, so transpose it to world space
matrix = Matrix.Translation(cam.matrix_basis * track.bundle)
#convert the position into AE space
ae_transform = convert_transform_matrix(matrix, data['width'], data['height'], data['aspect'], x_rot_correction=False)
#get the name of the tracker
name_ae = convert_name(False, track)
#write JS script for this Bundle
jsx_file.write('var %s = newComp.layers.addNull();\n' % name_ae)
jsx_file.write('%s.threeDLayer = true;\n' % name_ae)
jsx_file.write('%s.source.name = "%s";\n' % (name_ae, name_ae))
jsx_file.write('%s.property("position").setValue([%f,%f,%f]);\n\n\n' % (name_ae, ae_transform[0], ae_transform[1], ae_transform[2]))
jsx_file.write('\n}else{alert ("Exit Import Blender animation data \\nNo Comp\'s name has been chosen","EXIT")};') jsx_file.write('\n}else{alert ("Exit Import Blender animation data \\nNo Comp\'s name has been chosen","EXIT")};')
jsx_file.write("}\n\n\n") jsx_file.write("}\n\n\n")
jsx_file.write('app.beginUndoGroup("Import Blender animation data");\n') jsx_file.write('app.beginUndoGroup("Import Blender animation data");\n')
...@@ -375,10 +517,10 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl ...@@ -375,10 +517,10 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
########################################## ##########################################
def main(file, context, export_bundles, include_rotation, include_scale): def main(file, context, include_active_cam, include_selected_cams, include_selected_objects, include_cam_bundles, include_rotation, include_scale):
data = get_comp_data(context) data = get_comp_data(context)
selection = get_selected(context) selection = get_selected(context)
write_jsx_file(file, data, selection, export_bundles, include_rotation, include_scale) write_jsx_file(file, data, selection, include_active_cam, include_selected_cams, include_selected_objects, include_cam_bundles, include_rotation, include_scale)
print ("\nExport to After Effects Completed") print ("\nExport to After Effects Completed")
return {'FINISHED'} return {'FINISHED'}
...@@ -396,29 +538,68 @@ class ExportJsx(bpy.types.Operator, ExportHelper): ...@@ -396,29 +538,68 @@ class ExportJsx(bpy.types.Operator, ExportHelper):
bl_label = "Export to Adobe After Effects" bl_label = "Export to Adobe After Effects"
filename_ext = ".jsx" filename_ext = ".jsx"
filter_glob = StringProperty(default="*.jsx", options={'HIDDEN'}) filter_glob = StringProperty(default="*.jsx", options={'HIDDEN'})
include_active_cam = BoolProperty(
name = "Active Camera",
description = "Include Active Camera Data",
default = True,
)
include_selected_cams = BoolProperty(
name = "Selected Cameras",
description = "Add Selected Cameras Data",
default = True,
)
include_selected_objects = BoolProperty(
name = "Selected Objects",
description = "Add Selected Objects Data",
default = True,
)
include_rotation = BoolProperty( include_rotation = BoolProperty(
name="Include Rotation", name = "Rotation",
description="Include export of selected objects' rotation", description ="Include rotation of selected objects",
default=True, default = True,
) )
include_scale = BoolProperty( include_scale = BoolProperty(
name="Include Scale", name = "Scale",
description="Include export of selected objects' scale", description = "Include scale of selected object",
default=True, default = True,
) )
export_bundles = BoolProperty( include_cam_bundles = BoolProperty(
name="Export Bundles", name = "Camera 3D Markers",
description="Export 3D Tracking points of a selected camera", description = "Include 3D Markers of Camera Motion Solution for selected cameras",
default=True, default = True,
) )
# include_ob_bundles = BoolProperty(
# name = "Objects 3D Markers",
# description = "Include 3D Markers of Object Motion Solution for selected cameras",
# default = True,
# )
def draw(self, context):
layout = self.layout
box = layout.box()
box.label('Include Cameras and Objects:')
box.prop(self, 'include_active_cam')
box.prop(self, 'include_selected_cams')
box.prop(self, 'include_selected_objects')
box.label("Include Objects' Properties:")
box.prop(self, 'include_rotation')
box.prop(self, 'include_scale')
box.label("Include Tracking Data:")
box.prop(self, 'include_cam_bundles')
# box.prop(self, 'include_ob_bundles')
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
return context.active_object is not None active = context.active_object
selected = context.selected_objects
camera = context.scene.camera
ok = selected or camera
return ok
def execute(self, context): def execute(self, context):
return main(self.filepath, context, self.export_bundles, self.include_rotation, self.include_scale) return main(self.filepath, context, self.include_active_cam, self.include_selected_cams, self.include_selected_objects, self.include_cam_bundles, self.include_rotation, self.include_scale)
def menu_func(self, context): def menu_func(self, context):
...@@ -435,4 +616,4 @@ def unregister(): ...@@ -435,4 +616,4 @@ def unregister():
bpy.types.INFO_MT_file_export.remove(menu_func) bpy.types.INFO_MT_file_export.remove(menu_func)
if __name__ == "__main__": if __name__ == "__main__":
register() register()
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment