lib/export/cosa.rb in tracksperanto-2.11.2 vs lib/export/cosa.rb in tracksperanto-2.11.3

@@ -2,17 +2,23 @@
 # Exports trackers as an AfterEffects script creating nulls
 class Tracksperanto::Export::AE < Tracksperanto::Export::Base
 
   PREAMBLE = '// Run this script from the Script Editor
 function convertFrameToSeconds(layerWithFootage, frameValue)
-  {
-  		var comp = layerWithFootage.containingComp;
-  		var rate = 1.0 / comp.frameDuration;
-  		// Frames in AE are 0-based by default
-  		return (frameValue) / rate;
-  }'
+{
+    var comp = layerWithFootage.containingComp;
+    var rate = 1.0 / comp.frameDuration;
+    // Frames in AE are 0-based by default
+    return (frameValue) / rate;
+}
 
+var activeComp = app.project.activeItem;
+if(!activeComp) {
+    // Totally fail
+    alert("You need to be focused on a comp for the tracker creation to work");
+} else {'
+
   def self.desc_and_extension
     "createNulls.jsx"
   end
   
   def self.human_name
@@ -26,11 +32,11 @@
     @height = h
   end
   
   def start_tracker_segment(tracker_name)
     @io.puts("")
-    @io.puts('var layer%d = app.project.activeItem.layers.addNull();' % @count)
+    @io.puts('var layer%d = activeComp.layers.addNull();' % @count)
     @io.puts( 'layer%d.name = %s;' % [@count, tracker_name.inspect])
     @io.puts("")
     @io.puts('var pos = layer%d.property("Transform").property("Position");' % @count)
   end
   
@@ -38,8 +44,12 @@
     @io.puts('pos.setValueAtTime(convertFrameToSeconds(layer%d, %d), [%0.5f,%0.5f]);' % [@count, frame, abs_float_x, @height - abs_float_y])
   end
   
   def end_tracker_segment
     @count += 1
+  end
+  
+  def end_export
+    @io.puts('}') # Finishes the export clause
   end
   
 end