updates for tracker page

1 files changed, 58 insertions, 66 deletions

diff --git a/io_export_after_effects.py b/io_export_after_effects.py
index 997faec1..a6ecd781 100644
--- a/io_export_after_effects.py
+++ b/io_export_after_effects.py
@@ -21,13 +21,16 @@ bl_info = {
     'name': 'Export: Adobe After Effects (.jsx)',
     'description': 'Export selected cameras, objects & bundles to Adobe After Effects CS3 and above',
     'author': 'Bartek Skorupa',
-    'version': (0, 59),
+    'version': (0, 5, 9),
     'blender': (2, 6, 1),
     'api': 43253,
     'location': 'File > Export > Adobe After Effects (.jsx)',
-    'category': 'Import-Export',
     "warning": "",
-    "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.5/Py/Scripts/Import-Export/Adobe_After_Effects"
+    "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.5/Py/"\
+        "Scripts/Import-Export/Adobe_After_Effects",
+    "tracker_url": "https://projects.blender.org/tracker/index.php?"\
+        "func=detail&aid=29858",
+    'category': 'Import-Export',
     }
 
 
@@ -59,7 +62,7 @@ def get_comp_data(context):
 
 # create managable list of selected objects
 # (only selected objects will be analyzed and exported)
-def get_selected(context):
+def get_selected(context, prefix):
     cameras = []  # list of selected cameras
     cams_names = []  # list of selected cameras' names (prevent from calling "ConvertName(ob)" function too many times)
     nulls = []  # list of all selected objects exept cameras (will be used to create nulls in AE)
@@ -69,17 +72,11 @@ def get_selected(context):
     for ob in obs:
         if ob.type == 'CAMERA':
             cameras.append(ob)
-            cams_names.append(convert_name(False, ob))
+            cams_names.append(convert_name(False, ob, prefix))
         else:
             nulls.append(ob)
-            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))
-	
+            nulls_names.append(convert_name(False, ob, prefix))
+
     selection = {
         'cameras': cameras,
         'cams_names': cams_names,
@@ -90,13 +87,13 @@ def get_selected(context):
     return selection
 
 
-# convert names of objects to avoid errors in AE.
-def convert_name(is_comp, ob):
+# convert names of objects to avoid errors in AE. Add user specified prefix
+def convert_name(is_comp, ob, prefix):
     if is_comp:
-        ob_name = ob
+        ob_name = prefix + ob
         ob_name = ob_name.replace('"', "_")
     else:
-        ob_name = "_" + ob.name
+        ob_name = prefix + "_" + ob.name
 
         if ob_name[0].isdigit():
             ob_name = "_" + ob_name
@@ -109,12 +106,11 @@ def convert_name(is_comp, ob):
 
 # get object's blender's location and rotation and return AE's Position and Rotation/Orientation
 # 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_pos_rot_matrix(matrix, width, height, aspect, x_rot_correction=False):
 
     # get blender location for ob
     b_loc_x, b_loc_y, b_loc_z = matrix.to_translation()
     b_rot_x, b_rot_y, b_rot_z = matrix.to_euler()
-    b_scale_x, b_scale_y, b_scale_z = matrix.to_scale()
 
     # get blender rotation for ob
     if x_rot_correction:
@@ -129,20 +125,17 @@ def convert_transform_matrix(matrix, width, height, aspect, x_rot_correction=Fal
     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
     z = b_loc_y * 100.0  # calculate AE's Z position
-    # Using AE's rotation combined with AE's orientation allows to compensate for different euler rotation order.
     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
     rz = b_rot_y  # calculate AE's Z rotation. Will become AE's OrentationZ property
-    sx = b_scale_x * 100.0
-    sy = b_scale_z * 100.0
-    sz = b_scale_y * 100.0
+    # Using AE's rotation combined with AE's orientation allows to compensate for different euler rotation order.
 
-    return x, y, z, rx, ry, rz, sx, sy, sz
+    return x, y, z, rx, ry, rz
 
 
-def convert_transform(obj, width, height, aspect, x_rot_correction=False):
+def convert_pos_rot(obj, width, height, aspect, x_rot_correction=False):
     matrix = obj.matrix_world.copy()
-    return convert_transform_matrix(matrix, width, height, aspect, x_rot_correction)
+    return convert_pos_rot_matrix(matrix, width, height, aspect, x_rot_correction)
 
 
 # get camera's lens and convert to AE's "zoom" value in pixels
@@ -191,11 +184,15 @@ def convert_transform(obj, width, height, aspect, x_rot_correction=False):
 #    zoom = lens * dimension / sensor * aspect
 #
 def convert_lens(camera, width, height, aspect):
-    if camera.data.sensor_fit == 'VERTICAL':
-        sensor = camera.data.sensor_height
-        dimension = height
+    if hasattr(camera.data, "sensor_width"):  # Preserve compatibility with versions not supporting camera sensor.
+        if camera.data.sensor_fit == 'VERTICAL':
+            sensor = camera.data.sensor_height
+            dimension = height
+        else:
+            sensor = camera.data.sensor_width
+            dimension = width
     else:
-        sensor = camera.data.sensor_width
+        sensor = 32  # standard blender's sensor size
         dimension = width
     
     zoom = camera.data.lens * dimension / sensor * aspect
@@ -204,7 +201,7 @@ def convert_lens(camera, width, height, aspect):
 
 
 # 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, export_bundles, comp_name, prefix):
     from mathutils import Matrix
 
     print("\n---------------------------\n- Export to After Effects -\n---------------------------")
@@ -235,7 +232,6 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
             'position': '',
             'orientation': '',
             'rotationX': '',
-            'scale': '',
             }
 
     # get all keyframes for each objects and store into dico
@@ -251,14 +247,14 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
             #get cam name
             name_ae = selection['cams_names'][i]
             #convert cam position to AE space
-            ae_transform = convert_transform(cam, data['width'], data['height'], data['aspect'], x_rot_correction=True)
+            ae_pos_rot = convert_pos_rot(cam, data['width'], data['height'], data['aspect'], x_rot_correction=True)
             #convert Blender's cam zoom to AE's
             zoom = convert_lens(cam, data['width'], data['height'], data['aspect'])
             #store all the value into dico
-            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]['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]['position'] += '[%f,%f,%f],' % (ae_pos_rot[0], ae_pos_rot[1], ae_pos_rot[2])
+            js_data['cameras'][name_ae]['pointOfInterest'] += '[%f,%f,%f],' % (ae_pos_rot[0], ae_pos_rot[1], ae_pos_rot[2])
+            js_data['cameras'][name_ae]['orientation'] += '[%f,%f,%f],' % (0, ae_pos_rot[4], ae_pos_rot[5])
+            js_data['cameras'][name_ae]['rotationX'] += '%f ,' % (ae_pos_rot[3])
             js_data['cameras'][name_ae]['zoom'] += '[%f],' % (zoom)
 
         #keyframes for all nulls
@@ -266,14 +262,11 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
             #get object name
             name_ae = selection['nulls_names'][i]
             #convert ob position to AE space
-            ae_transform = convert_transform(ob, data['width'], data['height'], data['aspect'], x_rot_correction=False)
+            ae_pos_rot = convert_pos_rot(ob, data['width'], data['height'], data['aspect'], x_rot_correction=False)
             #store all datas into dico
-            js_data['objects'][name_ae]['position'] += '[%f,%f,%f],' % (ae_transform[0], ae_transform[1], ae_transform[2])
-            if include_rotation:
-                js_data['objects'][name_ae]['orientation'] += '[%f,%f,%f],' % (0, ae_transform[4], ae_transform[5])
-                js_data['objects'][name_ae]['rotationX'] += '%f ,' % (ae_transform[3])
-            if include_scale:
-                js_data['objects'][name_ae]['scale'] += '[%f,%f,%f],' % (ae_transform[6], ae_transform[7], ae_transform[8])
+            js_data['objects'][name_ae]['position'] += '[%f,%f,%f],' % (ae_pos_rot[0], ae_pos_rot[1], ae_pos_rot[2])
+            js_data['objects'][name_ae]['orientation'] += '[%f,%f,%f],' % (0, ae_pos_rot[4], ae_pos_rot[5])
+            js_data['objects'][name_ae]['rotationX'] += '%f ,' % (ae_pos_rot[3])
 
     # ---- write JSX file
     jsx_file = open(file, 'w')
@@ -292,8 +285,7 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
     #wrap in function
     jsx_file.write("function compFromBlender(){\n")
     # 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('if (compName){')
+    jsx_file.write('\nvar compName = "%s";' % (comp_name))
     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']))
 
@@ -322,7 +314,6 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
         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:
@@ -344,22 +335,22 @@ def write_jsx_file(file, data, selection, export_bundles, include_rotation, incl
                     else:
                         mc = constrain.clip
 
-                    #go through each tracking point
+                    #go throuhg 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)
+                            ae_pos_rot = convert_pos_rot_matrix(matrix, data['width'], data['height'], data['aspect'], x_rot_correction=False)
                             #get the name of the tracker
-                            name_ae = convert_name(False, track)
+                            name_ae = convert_name(False, track, prefix)
                             #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('%s.property("position").setValue([%f,%f,%f]);\n\n\n' % (name_ae, ae_pos_rot[0], ae_pos_rot[1], ae_pos_rot[2]))
+
     jsx_file.write("}\n\n\n")
     jsx_file.write('app.beginUndoGroup("Import Blender animation data");\n')
     jsx_file.write('compFromBlender();\n')
@@ -373,10 +364,11 @@ 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, export_bundles, comp_name, prefix):
     data = get_comp_data(context)
-    selection = get_selected(context)
-    write_jsx_file(file, data, selection, export_bundles, include_rotation, include_scale)
+    selection = get_selected(context, prefix)
+    comp_name = convert_name(True, comp_name, "")
+    write_jsx_file(file, data, selection, export_bundles, comp_name, prefix)
     print ("\nExport to After Effects Completed")
     return {'FINISHED'}
 
@@ -394,29 +386,29 @@ class ExportJsx(bpy.types.Operator, ExportHelper):
     bl_label = "Export to Adobe After Effects"
     filename_ext = ".jsx"
     filter_glob = StringProperty(default="*.jsx", options={'HIDDEN'})
-    include_rotation = BoolProperty(
-            name="Include Rotation",
-            description="Include export of selected objects' rotation",
-            default=True,
+
+    comp_name = StringProperty(
+            name="Comp Name",
+            description="Name of composition to be created in After Effects",
+            default="BlendComp"
             )
-    include_scale = BoolProperty(
-            name="Include Scale",
-            description="Include export of selected objects' scale",
-            default=True,
+    prefix = StringProperty(
+            name="Layer's Prefix",
+            description="Prefix to use before AE layer's name",
+            #default="bl_"
             )
     export_bundles = BoolProperty(
             name="Export Bundles",
             description="Export 3D Tracking points of a selected camera",
-            default=True,
+            default=False,
             )
 
-
     @classmethod
     def poll(cls, context):
         return context.active_object is not None
 
     def execute(self, context):
-        return main(self.filepath, context, self.export_bundles, self.include_rotation, self.include_scale)
+        return main(self.filepath, context, self.export_bundles, self.comp_name, self.prefix)
 
 
 def menu_func(self, context):