1 files changed, 136 insertions, 0 deletions
diff --git a/source/blender/compositor/operations/COM_TrackPositionOperation.cc b/source/blender/compositor/operations/COM_TrackPositionOperation.cc
new file mode 100644
index 00000000000..ddabfb7cf6c
--- /dev/null
+++ b/source/blender/compositor/operations/COM_TrackPositionOperation.cc
@@ -0,0 +1,136 @@
+/*
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ *
+ * Copyright 2012, Blender Foundation.
+ */
+
+#include "COM_TrackPositionOperation.h"
+
+#include "MEM_guardedalloc.h"
+
+#include "BLI_listbase.h"
+#include "BLI_math.h"
+#include "BLI_math_color.h"
+
+#include "BKE_movieclip.h"
+#include "BKE_node.h"
+#include "BKE_tracking.h"
+
+TrackPositionOperation::TrackPositionOperation()
+{
+  this->addOutputSocket(COM_DT_VALUE);
+  this->m_movieClip = nullptr;
+  this->m_framenumber = 0;
+  this->m_trackingObjectName[0] = 0;
+  this->m_trackName[0] = 0;
+  this->m_axis = 0;
+  this->m_position = CMP_TRACKPOS_ABSOLUTE;
+  this->m_relativeFrame = 0;
+  this->m_speed_output = false;
+}
+
+void TrackPositionOperation::initExecution()
+{
+  MovieTracking *tracking = nullptr;
+  MovieClipUser user = {0};
+  MovieTrackingObject *object;
+
+  zero_v2(this->m_markerPos);
+  zero_v2(this->m_relativePos);
+
+  if (!this->m_movieClip) {
+    return;
+  }
+
+  tracking = &this->m_movieClip->tracking;
+
+  BKE_movieclip_user_set_frame(&user, this->m_framenumber);
+  BKE_movieclip_get_size(this->m_movieClip, &user, &this->m_width, &this->m_height);
+
+  object = BKE_tracking_object_get_named(tracking, this->m_trackingObjectName);
+  if (object) {
+    MovieTrackingTrack *track;
+
+    track = BKE_tracking_track_get_named(tracking, object, this->m_trackName);
+
+    if (track) {
+      MovieTrackingMarker *marker;
+      int clip_framenr = BKE_movieclip_remap_scene_to_clip_frame(this->m_movieClip,
+                                                                 this->m_framenumber);
+
+      marker = BKE_tracking_marker_get(track, clip_framenr);
+
+      copy_v2_v2(this->m_markerPos, marker->pos);
+
+      if (this->m_speed_output) {
+        int relative_clip_framenr = BKE_movieclip_remap_scene_to_clip_frame(this->m_movieClip,
+                                                                            this->m_relativeFrame);
+
+        marker = BKE_tracking_marker_get_exact(track, relative_clip_framenr);
+        if (marker != nullptr && (marker->flag & MARKER_DISABLED) == 0) {
+          copy_v2_v2(this->m_relativePos, marker->pos);
+        }
+        else {
+          copy_v2_v2(this->m_relativePos, this->m_markerPos);
+        }
+        if (this->m_relativeFrame < this->m_framenumber) {
+          swap_v2_v2(this->m_relativePos, this->m_markerPos);
+        }
+      }
+      else if (this->m_position == CMP_TRACKPOS_RELATIVE_START) {
+        int i;
+
+        for (i = 0; i < track->markersnr; i++) {
+          marker = &track->markers[i];
+
+          if ((marker->flag & MARKER_DISABLED) == 0) {
+            copy_v2_v2(this->m_relativePos, marker->pos);
+
+            break;
+          }
+        }
+      }
+      else if (this->m_position == CMP_TRACKPOS_RELATIVE_FRAME) {
+        int relative_clip_framenr = BKE_movieclip_remap_scene_to_clip_frame(this->m_movieClip,
+                                                                            this->m_relativeFrame);
+
+        marker = BKE_tracking_marker_get(track, relative_clip_framenr);
+        copy_v2_v2(this->m_relativePos, marker->pos);
+      }
+    }
+  }
+}
+
+void TrackPositionOperation::executePixelSampled(float output[4],
+                                                 float /*x*/,
+                                                 float /*y*/,
+                                                 PixelSampler /*sampler*/)
+{
+  output[0] = this->m_markerPos[this->m_axis] - this->m_relativePos[this->m_axis];
+
+  if (this->m_axis == 0) {
+    output[0] *= this->m_width;
+  }
+  else {
+    output[0] *= this->m_height;
+  }
+}
+
+void TrackPositionOperation::determineResolution(unsigned int resolution[2],
+                                                 unsigned int preferredResolution[2])
+{
+  resolution[0] = preferredResolution[0];
+  resolution[1] = preferredResolution[1];
+}