bcc817d0aa140b2c5e75a18cc957013cf12be123
[blender.git] / source / blender / makesrna / intern / rna_tracking.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * Contributor(s): Blender Foundation,
19  *                 Sergey Sharybin
20  *
21  * ***** END GPL LICENSE BLOCK *****
22  */
23
24 /** \file blender/makesrna/intern/rna_tracking.c
25  *  \ingroup RNA
26  */
27
28
29 #include <stdlib.h>
30 #include <limits.h>
31
32 #include "MEM_guardedalloc.h"
33
34 #include "BLI_math.h"
35 #include "BKE_movieclip.h"
36 #include "BKE_tracking.h"
37
38 #include "RNA_define.h"
39
40 #include "rna_internal.h"
41
42 #include "DNA_movieclip_types.h"
43 #include "DNA_object_types.h"   /* SELECT */
44 #include "DNA_scene_types.h"
45
46 #include "WM_types.h"
47
48 #ifdef RNA_RUNTIME
49
50 #include "BKE_depsgraph.h"
51 #include "BKE_node.h"
52
53 #include "IMB_imbuf.h"
54
55 #include "WM_api.h"
56
57 static void rna_tracking_defaultSettings_levelsUpdate(Main *UNUSED(bmain), Scene *scene, PointerRNA *ptr)
58 {
59         MovieClip *clip= (MovieClip*)ptr->id.data;
60         MovieTracking *tracking= &clip->tracking;
61         MovieTrackingSettings *settings= &tracking->settings;
62
63         if(settings->default_tracker==TRACKER_KLT) {
64                 int max_pyramid_level_factor= 1 << (settings->default_pyramid_levels - 1);
65                 float search_ratio= 2.3f * max_pyramid_level_factor;
66
67                 settings->default_search_size= settings->default_pattern_size*search_ratio;
68         }
69 }
70
71 static void rna_tracking_defaultSettings_patternUpdate(Main *UNUSED(bmain), Scene *scene, PointerRNA *ptr)
72 {
73         MovieClip *clip= (MovieClip*)ptr->id.data;
74         MovieTracking *tracking= &clip->tracking;
75         MovieTrackingSettings *settings= &tracking->settings;
76
77         if(settings->default_search_size<settings->default_pattern_size)
78                 settings->default_search_size= settings->default_pattern_size;
79 }
80
81 static void rna_tracking_defaultSettings_searchUpdate(Main *UNUSED(bmain), Scene *scene, PointerRNA *ptr)
82 {
83         MovieClip *clip= (MovieClip*)ptr->id.data;
84         MovieTracking *tracking= &clip->tracking;
85         MovieTrackingSettings *settings= &tracking->settings;
86
87         if(settings->default_pattern_size>settings->default_search_size)
88                 settings->default_pattern_size= settings->default_search_size;
89 }
90
91 static void rna_tracking_tracks_begin(CollectionPropertyIterator *iter, PointerRNA *ptr)
92 {
93         MovieClip *clip= (MovieClip*)ptr->id.data;
94
95         rna_iterator_listbase_begin(iter, &clip->tracking.tracks, NULL);
96 }
97
98 static void rna_tracking_objects_begin(CollectionPropertyIterator *iter, PointerRNA *ptr)
99 {
100         MovieClip *clip= (MovieClip*)ptr->id.data;
101
102         rna_iterator_listbase_begin(iter, &clip->tracking.objects, NULL);
103 }
104
105 static int rna_tracking_active_object_index_get(PointerRNA *ptr)
106 {
107         MovieClip *clip= (MovieClip*)ptr->id.data;
108
109         return clip->tracking.objectnr;
110 }
111
112 static void rna_tracking_active_object_index_set(PointerRNA *ptr, int value)
113 {
114         MovieClip *clip= (MovieClip*)ptr->id.data;
115
116         clip->tracking.objectnr= value;
117 }
118
119 static void rna_tracking_active_object_index_range(PointerRNA *ptr, int *min, int *max)
120 {
121         MovieClip *clip= (MovieClip*)ptr->id.data;
122
123         *min= 0;
124         *max= clip->tracking.tot_object-1;
125         *max= MAX2(0, *max);
126 }
127
128 static PointerRNA rna_tracking_active_track_get(PointerRNA *ptr)
129 {
130         MovieClip *clip= (MovieClip*)ptr->id.data;
131         MovieTrackingTrack *act_track= BKE_tracking_active_track(&clip->tracking);
132
133         return rna_pointer_inherit_refine(ptr, &RNA_MovieTrackingTrack, act_track);
134 }
135
136 static void rna_tracking_active_track_set(PointerRNA *ptr, PointerRNA value)
137 {
138         MovieClip *clip= (MovieClip*)ptr->id.data;
139         MovieTrackingTrack *track= (MovieTrackingTrack *)value.data;
140         ListBase *tracksbase= BKE_tracking_get_tracks(&clip->tracking);
141         int index= BLI_findindex(tracksbase, track);
142
143         if(index>=0)
144                 clip->tracking.act_track= track;
145         else
146                 clip->tracking.act_track= NULL;
147 }
148
149 void rna_trackingTrack_name_set(PointerRNA *ptr, const char *value)
150 {
151         MovieClip *clip= (MovieClip *)ptr->id.data;
152         MovieTracking *tracking= &clip->tracking;
153         MovieTrackingTrack *track= (MovieTrackingTrack *)ptr->data;
154         ListBase *tracksbase= &tracking->tracks;
155
156         BLI_strncpy(track->name, value, sizeof(track->name));
157
158         /* TODO: it's a bit difficult to find list track came from knowing just
159                  movie clip ID and MovieTracking structure, so keep this naive
160                          search for a while */
161         if(BLI_findindex(tracksbase, track) < 0) {
162                 MovieTrackingObject *object= tracking->objects.first;
163
164                 while(object) {
165                         if(BLI_findindex(&object->tracks, track)) {
166                                 tracksbase= &object->tracks;
167                                 break;
168                         }
169
170                         object= object->next;
171                 }
172         }
173
174         BKE_track_unique_name(tracksbase, track);
175 }
176
177 static int rna_trackingTrack_select_get(PointerRNA *ptr)
178 {
179         MovieTrackingTrack *track= (MovieTrackingTrack *)ptr->data;
180
181         return TRACK_SELECTED(track);
182 }
183
184 static void rna_trackingTrack_select_set(PointerRNA *ptr, int value)
185 {
186         MovieTrackingTrack *track= (MovieTrackingTrack *)ptr->data;
187
188         if(value) {
189                 track->flag|= SELECT;
190                 track->pat_flag|= SELECT;
191                 track->search_flag|= SELECT;
192         }
193         else {
194                 track->flag&= ~SELECT;
195                 track->pat_flag&= ~SELECT;
196                 track->search_flag&= ~SELECT;
197         }
198 }
199
200 static void rna_tracking_trackerPattern_update(Main *UNUSED(bmain), Scene *UNUSED(scene), PointerRNA *ptr)
201 {
202         MovieTrackingTrack *track= (MovieTrackingTrack *)ptr->data;
203
204         BKE_tracking_clamp_track(track, CLAMP_PAT_DIM);
205 }
206
207 static void rna_tracking_trackerSearch_update(Main *UNUSED(bmain), Scene *UNUSED(scene), PointerRNA *ptr)
208 {
209         MovieTrackingTrack *track= (MovieTrackingTrack *)ptr->data;
210
211         BKE_tracking_clamp_track(track, CLAMP_SEARCH_DIM);
212 }
213
214 static void rna_tracking_trackerAlgorithm_update(Main *UNUSED(bmain), Scene *UNUSED(scene), PointerRNA *ptr)
215 {
216         MovieTrackingTrack *track= (MovieTrackingTrack *)ptr->data;
217
218         if(track->tracker==TRACKER_KLT)
219           BKE_tracking_clamp_track(track, CLAMP_PYRAMID_LEVELS);
220         else
221           BKE_tracking_clamp_track(track, CLAMP_SEARCH_DIM);
222 }
223
224 static void rna_tracking_trackerPyramid_update(Main *UNUSED(bmain), Scene *UNUSED(scene), PointerRNA *ptr)
225 {
226         MovieTrackingTrack *track= (MovieTrackingTrack *)ptr->data;
227
228         BKE_tracking_clamp_track(track, CLAMP_PYRAMID_LEVELS);
229 }
230
231 static float rna_trackingCamera_focal_mm_get(PointerRNA *ptr)
232 {
233         MovieClip *clip= (MovieClip*)ptr->id.data;
234         MovieTrackingCamera *camera= &clip->tracking.camera;
235         float val= camera->focal;
236
237         if(clip->lastsize[0])
238                 val= val*camera->sensor_width/(float)clip->lastsize[0];
239
240         return val;
241 }
242
243 static void rna_trackingCamera_focal_mm_set(PointerRNA *ptr, float value)
244 {
245         MovieClip *clip= (MovieClip*)ptr->id.data;
246         MovieTrackingCamera *camera= &clip->tracking.camera;
247
248         if(clip->lastsize[0])
249                 value= clip->lastsize[0]*value/camera->sensor_width;
250
251         if(value>=0.0001)
252                 camera->focal= value;
253 }
254
255 static int rna_track_2d_stabilization(CollectionPropertyIterator *UNUSED(iter), void *data)
256 {
257         MovieTrackingTrack *track= (MovieTrackingTrack*)data;
258
259         if((track->flag&TRACK_USE_2D_STAB)==0)
260                 return 1;
261
262         return 0;
263 }
264
265 static void rna_tracking_stabTracks_begin(CollectionPropertyIterator *iter, PointerRNA *ptr)
266 {
267         MovieClip *clip= (MovieClip*)ptr->id.data;
268         rna_iterator_listbase_begin(iter, &clip->tracking.tracks, rna_track_2d_stabilization);
269 }
270
271 static int rna_tracking_stabTracks_active_index_get(PointerRNA *ptr)
272 {
273         MovieClip *clip= (MovieClip*)ptr->id.data;
274         return clip->tracking.stabilization.act_track;
275 }
276
277 static void rna_tracking_stabTracks_active_index_set(PointerRNA *ptr, int value)
278 {
279         MovieClip *clip= (MovieClip*)ptr->id.data;
280         clip->tracking.stabilization.act_track= value;
281 }
282
283 static void rna_tracking_stabTracks_active_index_range(PointerRNA *ptr, int *min, int *max)
284 {
285         MovieClip *clip= (MovieClip*)ptr->id.data;
286
287         *min= 0;
288         *max= clip->tracking.stabilization.tot_track-1;
289         *max= MAX2(0, *max);
290 }
291
292 static void rna_tracking_flushUpdate(Main *UNUSED(bmain), Scene *scene, PointerRNA *ptr)
293 {
294         MovieClip *clip= (MovieClip*)ptr->id.data;
295         MovieTrackingStabilization *stab= &clip->tracking.stabilization;
296
297         stab->ok= 0;
298
299         nodeUpdateID(scene->nodetree, &clip->id);
300
301         WM_main_add_notifier(NC_SCENE|ND_NODES, NULL);
302         DAG_id_tag_update(&clip->id, 0);
303 }
304
305 static void rna_trackingObject_tracks_begin(CollectionPropertyIterator *iter, PointerRNA *ptr)
306 {
307         MovieTrackingObject *object= (MovieTrackingObject* )ptr->data;
308
309         if(object->flag&TRACKING_OBJECT_CAMERA) {
310                 MovieClip *clip= (MovieClip*)ptr->id.data;
311
312                 rna_iterator_listbase_begin(iter, &clip->tracking.tracks, NULL);
313         }
314         else {
315                 rna_iterator_listbase_begin(iter, &object->tracks, NULL);
316         }
317 }
318
319 static PointerRNA rna_tracking_active_object_get(PointerRNA *ptr)
320 {
321         MovieClip *clip= (MovieClip*)ptr->id.data;
322         MovieTrackingObject *object= BLI_findlink(&clip->tracking.objects, clip->tracking.objectnr);
323
324         return rna_pointer_inherit_refine(ptr, &RNA_MovieTrackingObject, object);
325 }
326
327 static void rna_tracking_active_object_set(PointerRNA *ptr, PointerRNA value)
328 {
329         MovieClip *clip= (MovieClip*)ptr->id.data;
330         MovieTrackingObject *object= (MovieTrackingObject *)value.data;
331         int index= BLI_findindex(&clip->tracking.objects, object);
332
333         if(index>=0) clip->tracking.objectnr= index;
334         else clip->tracking.objectnr= 0;
335 }
336
337 void rna_trackingObject_name_set(PointerRNA *ptr, const char *value)
338 {
339         MovieClip *clip= (MovieClip *)ptr->id.data;
340         MovieTrackingObject *object= (MovieTrackingObject *)ptr->data;
341
342         BLI_strncpy(object->name, value, sizeof(object->name));
343
344         BKE_tracking_object_unique_name(&clip->tracking, object);
345 }
346
347 static void rna_trackingObject_flushUpdate(Main *UNUSED(bmain), Scene *scene, PointerRNA *ptr)
348 {
349         MovieClip *clip= (MovieClip*)ptr->id.data;
350
351         WM_main_add_notifier(NC_OBJECT|ND_TRANSFORM, NULL);
352         DAG_id_tag_update(&clip->id, 0);
353 }
354
355 /* API */
356
357 static void add_tracks_to_base(MovieClip *clip, MovieTracking *tracking, ListBase *tracksbase, int frame, int number)
358 {
359         int a, width, height;
360         MovieClipUser user= {0};
361
362         user.framenr= 1;
363
364         BKE_movieclip_get_size(clip, &user, &width, &height);
365
366         for(a= 0; a<number; a++)
367                 BKE_tracking_add_track(tracking, tracksbase, 0, 0, frame, width, height);
368 }
369
370 static void rna_tracking_tracks_add(ID *id, MovieTracking *tracking, int frame, int number)
371 {
372         MovieClip *clip= (MovieClip *) id;
373
374         add_tracks_to_base(clip, tracking, &tracking->tracks, frame, number);
375
376         WM_main_add_notifier(NC_MOVIECLIP|NA_EDITED, clip);
377 }
378
379 static void rna_trackingObject_tracks_add(ID *id, MovieTrackingObject *object, int frame, int number)
380 {
381         MovieClip *clip= (MovieClip *) id;
382         ListBase *tracksbase= &object->tracks;
383
384         if(object->flag&TRACKING_OBJECT_CAMERA)
385                 tracksbase= &clip->tracking.tracks;
386
387         add_tracks_to_base(clip, &clip->tracking, tracksbase, frame, number);
388
389         WM_main_add_notifier(NC_MOVIECLIP|NA_EDITED, NULL);
390 }
391
392 static MovieTrackingObject *rna_tracking_object_new(MovieTracking *tracking, const char *name)
393 {
394         MovieTrackingObject *object= BKE_tracking_new_object(tracking, name);
395
396         WM_main_add_notifier(NC_MOVIECLIP|NA_EDITED, NULL);
397
398         return object;
399 }
400
401 void rna_tracking_object_remove(MovieTracking *tracking, MovieTrackingObject *object)
402 {
403         BKE_tracking_remove_object(tracking, object);
404
405         WM_main_add_notifier(NC_MOVIECLIP|NA_EDITED, NULL);
406 }
407
408 static void rna_trackingMarker_frame_set(PointerRNA *ptr, int value)
409 {
410         MovieClip *clip = (MovieClip *) ptr->id.data;
411         MovieTracking *tracking = &clip->tracking;
412         MovieTrackingTrack *track;
413         MovieTrackingMarker *marker = (MovieTrackingMarker *) ptr->data;
414
415         track = tracking->tracks.first;
416         while (track) {
417                 if (marker >= track->markers && marker < track->markers+track->markersnr) {
418                         break;
419                 }
420
421                 track = track->next;
422         }
423
424         if (track) {
425                 MovieTrackingMarker new_marker = *marker;
426                 new_marker.framenr = value;
427
428                 BKE_tracking_delete_marker(track, marker->framenr);
429                 BKE_tracking_insert_marker(track, &new_marker);
430         }
431 }
432
433 static MovieTrackingMarker *rna_trackingMarkers_find_frame(MovieTrackingTrack *track, int framenr)
434 {
435         return BKE_tracking_exact_marker(track, framenr);
436 }
437
438 static MovieTrackingMarker* rna_trackingMarkers_insert_frame(MovieTrackingTrack *track, int framenr, float *co)
439 {
440         MovieTrackingMarker marker, *new_marker;
441
442         memset(&marker, 0, sizeof(marker));
443         marker.framenr = framenr;
444         copy_v2_v2(marker.pos, co);
445
446         new_marker = BKE_tracking_insert_marker(track, &marker);
447
448         WM_main_add_notifier(NC_MOVIECLIP|NA_EDITED, NULL);
449
450         return new_marker;
451 }
452
453 void rna_trackingMarkers_delete_frame(MovieTrackingTrack *track, int framenr)
454 {
455         if(track->markersnr==1)
456                 return;
457
458         BKE_tracking_delete_marker(track, framenr);
459
460         WM_main_add_notifier(NC_MOVIECLIP|NA_EDITED, NULL);
461 }
462
463 #else
464
465 static EnumPropertyItem tracker_items[] = {
466         {TRACKER_KLT, "KLT", 0, "KLT", "Kanade–Lucas–Tomasi tracker which works with most of video clips, a bit slower than SAD"},
467         {TRACKER_SAD, "SAD", 0, "SAD", "Sum of Absolute Differences tracker which can be used when KLT tracker fails"},
468         {TRACKER_HYBRID, "Hybrid", 0, "Hybrid", "A hybrid tracker that uses SAD for rough tracking, KLT for refinement."},
469         {0, NULL, 0, NULL, NULL}};
470
471 static EnumPropertyItem pattern_match_items[] = {
472         {TRACK_MATCH_KEYFRAME, "KEYFRAME", 0, "Keyframe", "Track pattern from keyframe to next frame"},
473         {TRACK_MATCH_PREVFRAME, "PREV_FRAME", 0, "Previous frame", "Track pattern from current frame to next frame"},
474         {0, NULL, 0, NULL, NULL}};
475
476 static int rna_matrix_dimsize_4x4[]= {4, 4};
477
478 static void rna_def_trackingSettings(BlenderRNA *brna)
479 {
480         StructRNA *srna;
481         PropertyRNA *prop;
482
483         static EnumPropertyItem speed_items[] = {
484                 {0, "FASTEST", 0, "Fastest", "Track as fast as it's possible"},
485             {TRACKING_SPEED_DOUBLE, "DOUBLE", 0, "Double", "Track with double speed"},
486                 {TRACKING_SPEED_REALTIME, "REALTIME", 0, "Realtime", "Track with realtime speed"},
487                 {TRACKING_SPEED_HALF, "HALF", 0, "Half", "Track with half of realtime speed"},
488                 {TRACKING_SPEED_QUARTER, "QUARTER", 0, "Quarter", "Track with quarter of realtime speed"},
489                 {0, NULL, 0, NULL, NULL}};
490
491         static EnumPropertyItem cleanup_items[] = {
492                 {TRACKING_CLEAN_SELECT, "SELECT", 0, "Select", "Select unclean tracks"},
493                 {TRACKING_CLEAN_DELETE_TRACK, "DELETE_TRACK", 0, "Delete Track", "Delete unclean tracks"},
494                 {TRACKING_CLEAN_DELETE_SEGMENT, "DELETE_SEGMENTS", 0, "Delete Segments", "Delete unclean segments of tracks"},
495                 {0, NULL, 0, NULL, NULL}
496         };
497
498         static EnumPropertyItem refine_items[] = {
499                 {0, "NONE", 0, "Nothing", "Do not refine camera intrinsics"},
500                 {REFINE_FOCAL_LENGTH, "FOCAL_LENGTH", 0, "Focal Length", "Refine focal length"},
501                 {REFINE_FOCAL_LENGTH|REFINE_RADIAL_DISTORTION_K1, "FOCAL_LENGTH_RADIAL_K1", 0, "Focal length, K1", "Refine focal length and radial distortion K1"},
502                 {REFINE_FOCAL_LENGTH|
503                  REFINE_RADIAL_DISTORTION_K1|
504                  REFINE_RADIAL_DISTORTION_K2, "FOCAL_LENGTH_RADIAL_K1_K2", 0, "Focal length, K1, K2", "Refine focal length and radial distortion K1 and K2"},
505                 {REFINE_FOCAL_LENGTH|
506                  REFINE_PRINCIPAL_POINT|
507                  REFINE_RADIAL_DISTORTION_K1|
508                  REFINE_RADIAL_DISTORTION_K2,
509                  "FOCAL_LENGTH_PRINCIPAL_POINT_RADIAL_K1_K2", 0, "Focal Length, Optical Center, K1, K2", "Refine focal length, optical center and radial distortion K1 and K2"},
510                 {REFINE_FOCAL_LENGTH|
511                  REFINE_PRINCIPAL_POINT, "FOCAL_LENGTH_PRINCIPAL_POINT", 0, "Focal Length, Optical Center", "Refine focal length and optical center"},
512                 {0, NULL, 0, NULL, NULL}
513         };
514
515         srna= RNA_def_struct(brna, "MovieTrackingSettings", NULL);
516         RNA_def_struct_ui_text(srna, "Movie tracking settings", "Match moving settings");
517
518         /* speed */
519         prop= RNA_def_property(srna, "speed", PROP_ENUM, PROP_NONE);
520         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
521         RNA_def_property_enum_items(prop, speed_items);
522         RNA_def_property_ui_text(prop, "Speed", "Limit speed of tracking to make visual feedback easier (this does not affect the tracking quality)");
523
524         /* keyframe_a */
525         prop= RNA_def_property(srna, "keyframe_a", PROP_INT, PROP_NONE);
526         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
527         RNA_def_property_int_sdna(prop, NULL, "keyframe1");
528         RNA_def_property_ui_text(prop, "Keyframe A", "First keyframe used for reconstruction initialization");
529
530         /* keyframe_b */
531         prop= RNA_def_property(srna, "keyframe_b", PROP_INT, PROP_NONE);
532         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
533         RNA_def_property_int_sdna(prop, NULL, "keyframe2");
534         RNA_def_property_ui_text(prop, "Keyframe B", "Second keyframe used for reconstruction initialization");
535
536         /* intrinsics refinement during bundle adjustment */
537         prop= RNA_def_property(srna, "refine_intrinsics", PROP_ENUM, PROP_NONE);
538         RNA_def_property_enum_sdna(prop, NULL, "refine_camera_intrinsics");
539         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
540         RNA_def_property_enum_items(prop, refine_items);
541         RNA_def_property_ui_text(prop, "Refine", "Refine intrinsics during camera solving");
542
543         /* tool settings */
544
545         /* distance */
546         prop= RNA_def_property(srna, "distance", PROP_FLOAT, PROP_NONE);
547         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
548         RNA_def_property_float_sdna(prop, NULL, "dist");
549         RNA_def_property_ui_text(prop, "Distance", "Distance between two bundles used for scene scaling");
550
551         /* frames count */
552         prop= RNA_def_property(srna, "clean_frames", PROP_INT, PROP_NONE);
553         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
554         RNA_def_property_int_sdna(prop, NULL, "clean_frames");
555         RNA_def_property_range(prop, 0, INT_MAX);
556         RNA_def_property_ui_text(prop, "Tracked Frames", "Effect on tracks which are tracked less than the specified amount of frames");
557
558         /* reprojection error */
559         prop= RNA_def_property(srna, "clean_error", PROP_FLOAT, PROP_NONE);
560         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
561         RNA_def_property_float_sdna(prop, NULL, "clean_error");
562         RNA_def_property_range(prop, 0, FLT_MAX);
563         RNA_def_property_ui_text(prop, "Reprojection Error", "Effect on tracks which have a larger reprojection error");
564
565         /* cleanup action */
566         prop= RNA_def_property(srna, "clean_action", PROP_ENUM, PROP_NONE);
567         RNA_def_property_enum_sdna(prop, NULL, "clean_action");
568         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
569         RNA_def_property_enum_items(prop, cleanup_items);
570         RNA_def_property_ui_text(prop, "Action", "Cleanup action to execute");
571
572         /* ** default tracker settings ** */
573         prop= RNA_def_property(srna, "show_default_expanded", PROP_BOOLEAN, PROP_NONE);
574         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
575         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACKING_SETTINGS_SHOW_DEFAULT_EXPANDED);
576         RNA_def_property_ui_text(prop, "Show Expanded", "Show the expanded in the user interface");
577         RNA_def_property_ui_icon(prop, ICON_TRIA_RIGHT, 1);
578
579         /* limit frames */
580         prop= RNA_def_property(srna, "default_frames_limit", PROP_INT, PROP_NONE);
581         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
582         RNA_def_property_int_sdna(prop, NULL, "default_frames_limit");
583         RNA_def_property_range(prop, 0, SHRT_MAX);
584         RNA_def_property_ui_text(prop, "Frames Limit", "Every tracking cycle, this number of frames are tracked");
585
586         /* pattern match */
587         prop= RNA_def_property(srna, "default_pattern_match", PROP_ENUM, PROP_NONE);
588         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
589         RNA_def_property_enum_sdna(prop, NULL, "default_pattern_match");
590         RNA_def_property_enum_items(prop, pattern_match_items);
591         RNA_def_property_ui_text(prop, "Pattern Match", "Track pattern from given frame when tracking marker to next frame");
592
593         /* margin */
594         prop= RNA_def_property(srna, "default_margin", PROP_INT, PROP_NONE);
595         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
596         RNA_def_property_int_sdna(prop, NULL, "default_margin");
597         RNA_def_property_range(prop, 0, 300);
598         RNA_def_property_ui_text(prop, "Margin", "Default distance from image boudary at which marker stops tracking");
599
600         /* tracking algorithm */
601         prop= RNA_def_property(srna, "default_tracker", PROP_ENUM, PROP_NONE);
602         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
603         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
604         RNA_def_property_enum_items(prop, tracker_items);
605         RNA_def_property_update(prop, 0, "rna_tracking_defaultSettings_levelsUpdate");
606         RNA_def_property_ui_text(prop, "Tracker", "Default tracking algorithm to use");
607
608         /* pyramid level for pyramid klt tracking */
609         prop= RNA_def_property(srna, "default_pyramid_levels", PROP_INT, PROP_NONE);
610         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
611         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
612         RNA_def_property_int_sdna(prop, NULL, "default_pyramid_levels");
613         RNA_def_property_range(prop, 1, 16);
614         RNA_def_property_update(prop, 0, "rna_tracking_defaultSettings_levelsUpdate");
615         RNA_def_property_ui_text(prop, "Pyramid levels", "Default number of pyramid levels (increase on blurry footage)");
616
617         /* minmal correlation - only used for SAD tracker */
618         prop= RNA_def_property(srna, "default_correlation_min", PROP_FLOAT, PROP_NONE);
619         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
620         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
621         RNA_def_property_float_sdna(prop, NULL, "default_minimum_correlation");
622         RNA_def_property_range(prop, -1.0f, 1.0f);
623         RNA_def_property_ui_range(prop, -1.0f, 1.0f, 0.1, 3);
624         RNA_def_property_ui_text(prop, "Correlation", "Default minimal value of correlation between matched pattern and reference which is still treated as successful tracking");
625
626         /* default pattern size */
627         prop= RNA_def_property(srna, "default_pattern_size", PROP_INT, PROP_NONE);
628         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
629         RNA_def_property_int_sdna(prop, NULL, "default_pattern_size");
630         RNA_def_property_range(prop, 5, 1000);
631         RNA_def_property_update(prop, 0, "rna_tracking_defaultSettings_patternUpdate");
632         RNA_def_property_ui_text(prop, "Pattern Size", "Size of pattern area for newly created tracks");
633
634         /* default search size */
635         prop= RNA_def_property(srna, "default_search_size", PROP_INT, PROP_NONE);
636         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
637         RNA_def_property_int_sdna(prop, NULL, "default_search_size");
638         RNA_def_property_range(prop, 5, 1000);
639         RNA_def_property_update(prop, 0, "rna_tracking_defaultSettings_searchUpdate");
640         RNA_def_property_ui_text(prop, "Search Size", "Size of search area for newly created tracks");
641
642         /* object distance */
643         prop= RNA_def_property(srna, "object_distance", PROP_FLOAT, PROP_NONE);
644         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
645         RNA_def_property_float_sdna(prop, NULL, "object_distance");
646         RNA_def_property_ui_text(prop, "Distance", "Distance between two bundles used for object scaling");
647         RNA_def_property_range(prop, 0.001, 10000);
648         RNA_def_property_ui_range(prop, 0.001, 10000.0, 1, 3);
649 }
650
651 static void rna_def_trackingCamera(BlenderRNA *brna)
652 {
653         StructRNA *srna;
654         PropertyRNA *prop;
655
656         static EnumPropertyItem camera_units_items[] = {
657                 {CAMERA_UNITS_PX, "PIXELS", 0, "px", "Use pixels for units of focal length"},
658                 {CAMERA_UNITS_MM, "MILLIMETERS", 0, "mm", "Use millimeters for units of focal length"},
659                 {0, NULL, 0, NULL, NULL}};
660
661         srna= RNA_def_struct(brna, "MovieTrackingCamera", NULL);
662         RNA_def_struct_ui_text(srna, "Movie tracking camera data", "Match-moving camera data for tracking");
663
664         /* Sensor */
665         prop= RNA_def_property(srna, "sensor_width", PROP_FLOAT, PROP_NONE);
666         RNA_def_property_float_sdna(prop, NULL, "sensor_width");
667         RNA_def_property_range(prop, 0.0f, 500.0f);
668         RNA_def_property_ui_text(prop, "Sensor", "Width of CCD sensor in millimeters");
669         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, NULL);
670
671         /* Focal Length */
672         prop= RNA_def_property(srna, "focal_length", PROP_FLOAT, PROP_NONE);
673         RNA_def_property_float_sdna(prop, NULL, "focal");
674         RNA_def_property_range(prop, 0.0001f, 5000.0f);
675         RNA_def_property_float_funcs(prop, "rna_trackingCamera_focal_mm_get", "rna_trackingCamera_focal_mm_set", NULL);
676         RNA_def_property_ui_text(prop, "Focal Length", "Camera's focal length");
677         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, NULL);
678
679         /* Focal Length in pixels */
680         prop= RNA_def_property(srna, "focal_length_pixels", PROP_FLOAT, PROP_NONE);
681         RNA_def_property_float_sdna(prop, NULL, "focal");
682         RNA_def_property_range(prop, 0.0f, 5000.0f);
683         RNA_def_property_ui_text(prop, "Focal Length", "Camera's focal length");
684         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, NULL);
685
686         /* Units */
687         prop= RNA_def_property(srna, "units", PROP_ENUM, PROP_NONE);
688         RNA_def_property_enum_sdna(prop, NULL, "units");
689         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
690         RNA_def_property_enum_items(prop, camera_units_items);
691         RNA_def_property_ui_text(prop, "Units", "Units used for camera focal length");
692
693         /* Principal Point */
694         prop= RNA_def_property(srna, "principal", PROP_FLOAT, PROP_NONE);
695         RNA_def_property_array(prop, 2);
696         RNA_def_property_float_sdna(prop, NULL, "principal");
697         RNA_def_property_ui_text(prop, "Principal Point", "Optical center of lens");
698         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, NULL);
699
700         /* Radial distortion parameters */
701         prop= RNA_def_property(srna, "k1", PROP_FLOAT, PROP_NONE);
702         RNA_def_property_float_sdna(prop, NULL, "k1");
703         RNA_def_property_ui_range(prop, -10, 10, .1, 3);
704         RNA_def_property_ui_text(prop, "K1", "First coefficient of third order polynomial radial distortion");
705         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_flushUpdate");
706
707         prop= RNA_def_property(srna, "k2", PROP_FLOAT, PROP_NONE);
708         RNA_def_property_float_sdna(prop, NULL, "k2");
709         RNA_def_property_ui_range(prop, -10, 10, .1, 3);
710         RNA_def_property_ui_text(prop, "K2", "Second coefficient of third order polynomial radial distortion");
711         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_flushUpdate");
712
713         prop= RNA_def_property(srna, "k3", PROP_FLOAT, PROP_NONE);
714         RNA_def_property_float_sdna(prop, NULL, "k3");
715         RNA_def_property_ui_range(prop, -10, 10, .1, 3);
716         RNA_def_property_ui_text(prop, "K3", "Third coefficient of third order polynomial radial distortion");
717         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_flushUpdate");
718
719         /* pixel aspect */
720         prop= RNA_def_property(srna, "pixel_aspect", PROP_FLOAT, PROP_XYZ);
721         RNA_def_property_float_sdna(prop, NULL, "pixel_aspect");
722         RNA_def_property_range(prop, 0.1f, 5000.0f);
723         RNA_def_property_ui_range(prop, 0.1f, 5000.0f, 1, 2);
724         RNA_def_property_ui_text(prop, "Pixel Aspect Ratio", "Pixel aspect ratio");
725         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
726 }
727
728 static void rna_def_trackingMarker(BlenderRNA *brna)
729 {
730         StructRNA *srna;
731         PropertyRNA *prop;
732
733         srna= RNA_def_struct(brna, "MovieTrackingMarker", NULL);
734         RNA_def_struct_ui_text(srna, "Movie tracking marker data", "Match-moving marker data for tracking");
735
736         /* position */
737         prop= RNA_def_property(srna, "co", PROP_FLOAT, PROP_TRANSLATION);
738         RNA_def_property_array(prop, 2);
739         RNA_def_property_ui_range(prop, -FLT_MAX, FLT_MAX, 1, 5);
740         RNA_def_property_float_sdna(prop, NULL, "pos");
741         RNA_def_property_ui_text(prop, "Position", "Marker position at frame in normalized coordinates");
742         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, NULL);
743
744         /* frame */
745         prop= RNA_def_property(srna, "frame", PROP_INT, PROP_NONE);
746         RNA_def_property_int_sdna(prop, NULL, "framenr");
747         RNA_def_property_ui_text(prop, "Frame", "Frame number marker is keyframed on");
748         RNA_def_property_int_funcs(prop, NULL, "rna_trackingMarker_frame_set", NULL);
749         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, 0);
750
751         /* enable */
752         prop= RNA_def_property(srna, "mute", PROP_BOOLEAN, PROP_NONE);
753         RNA_def_property_boolean_sdna(prop, NULL, "flag", MARKER_DISABLED);
754         RNA_def_property_ui_text(prop, "Mode", "Is marker muted for current frame");
755         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, NULL);
756 }
757
758 static void rna_def_trackingMarkers(BlenderRNA *brna, PropertyRNA *cprop)
759 {
760         StructRNA *srna;
761         FunctionRNA *func;
762         PropertyRNA *parm;
763
764         RNA_def_property_srna(cprop, "MovieTrackingMarkers");
765         srna= RNA_def_struct(brna, "MovieTrackingMarkers", NULL);
766         RNA_def_struct_sdna(srna, "MovieTrackingTrack");
767         RNA_def_struct_ui_text(srna, "Movie Tracking Markers", "Collection of markers for movie tracking track");
768
769         func= RNA_def_function(srna, "find_frame", "rna_trackingMarkers_find_frame");
770         RNA_def_function_ui_description(func, "Get marker for specified frame");
771         parm= RNA_def_int(func, "frame", 1, MINFRAME, MAXFRAME, "Frame",
772                          "Frame number to find marker for", MINFRAME, MAXFRAME);
773         RNA_def_property_flag(parm, PROP_REQUIRED);
774         parm= RNA_def_pointer(func, "marker", "MovieTrackingMarker", "", "Marker for specified frame");
775         RNA_def_function_return(func, parm);
776
777         func= RNA_def_function(srna, "insert_frame", "rna_trackingMarkers_insert_frame");
778         RNA_def_function_ui_description(func, "Add a number of tracks to this movie clip");
779         parm= RNA_def_int(func, "frame", 1, MINFRAME, MAXFRAME, "Frame",
780                         "Frame number to insert marker to", MINFRAME, MAXFRAME);
781         RNA_def_property_flag(parm, PROP_REQUIRED);
782         RNA_def_float_vector(func, "co", 2, 0, -1.0, 1.0, "Coordinate",
783                         "Place new marker at the given frame using specified in normalized space coordinates", -1.0, 1.0);
784         RNA_def_property_flag(parm, PROP_REQUIRED);
785         parm= RNA_def_pointer(func, "marker", "MovieTrackingMarker", "", "Newly created marker");
786         RNA_def_function_return(func, parm);
787
788         func= RNA_def_function(srna, "delete_frame", "rna_trackingMarkers_delete_frame");
789         RNA_def_function_ui_description(func, "Delete marker at specified frame");
790         parm= RNA_def_int(func, "frame", 1, MINFRAME, MAXFRAME, "Frame",
791                         "Frame number to delete marker from", MINFRAME, MAXFRAME);
792         RNA_def_property_flag(parm, PROP_REQUIRED);
793 }
794
795 static void rna_def_trackingTrack(BlenderRNA *brna)
796 {
797         StructRNA *srna;
798         PropertyRNA *prop;
799
800         rna_def_trackingMarker(brna);
801
802         srna= RNA_def_struct(brna, "MovieTrackingTrack", NULL);
803         RNA_def_struct_ui_text(srna, "Movie tracking track data", "Match-moving track data for tracking");
804         RNA_def_struct_ui_icon(srna, ICON_ANIM_DATA);
805
806         /* name */
807         prop= RNA_def_property(srna, "name", PROP_STRING, PROP_NONE);
808         RNA_def_property_ui_text(prop, "Name", "Unique name of track");
809         RNA_def_property_string_funcs(prop, NULL, NULL, "rna_trackingTrack_name_set");
810         RNA_def_property_string_maxlength(prop, MAX_ID_NAME);
811         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, NULL);
812         RNA_def_struct_name_property(srna, prop);
813
814         /* Pattern */
815         prop= RNA_def_property(srna, "pattern_min", PROP_FLOAT, PROP_TRANSLATION);
816         RNA_def_property_array(prop, 2);
817         RNA_def_property_ui_range(prop, -FLT_MAX, FLT_MAX, 1, 5);
818         RNA_def_property_float_sdna(prop, NULL, "pat_min");
819         RNA_def_property_ui_text(prop, "Pattern Min", "Left-bottom corner of pattern area in normalized coordinates relative to marker position");
820         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_trackerPattern_update");
821
822         prop= RNA_def_property(srna, "pattern_max", PROP_FLOAT, PROP_TRANSLATION);
823         RNA_def_property_array(prop, 2);
824         RNA_def_property_ui_range(prop, -FLT_MAX, FLT_MAX, 1, 5);
825         RNA_def_property_float_sdna(prop, NULL, "pat_max");
826         RNA_def_property_ui_text(prop, "Pattern Max", "Right-bottom corner of pattern area in normalized coordinates relative to marker position");
827         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_trackerPattern_update");
828
829         /* Search */
830         prop= RNA_def_property(srna, "search_min", PROP_FLOAT, PROP_TRANSLATION);
831         RNA_def_property_array(prop, 2);
832         RNA_def_property_ui_range(prop, -FLT_MAX, FLT_MAX, 1, 5);
833         RNA_def_property_float_sdna(prop, NULL, "search_min");
834         RNA_def_property_ui_text(prop, "Search Min", "Left-bottom corner of search area in normalized coordinates relative to marker position");
835         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_trackerSearch_update");
836
837         prop= RNA_def_property(srna, "search_max", PROP_FLOAT, PROP_TRANSLATION);
838         RNA_def_property_array(prop, 2);
839         RNA_def_property_ui_range(prop, -FLT_MAX, FLT_MAX, 1, 5);
840         RNA_def_property_float_sdna(prop, NULL, "search_max");
841         RNA_def_property_ui_text(prop, "Search Max", "Right-bottom corner of search area in normalized coordinates relative to marker position");
842         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_trackerSearch_update");
843
844         /* limit frames */
845         prop= RNA_def_property(srna, "frames_limit", PROP_INT, PROP_NONE);
846         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
847         RNA_def_property_int_sdna(prop, NULL, "frames_limit");
848         RNA_def_property_range(prop, 0, SHRT_MAX);
849         RNA_def_property_ui_text(prop, "Frames Limit", "Every tracking cycle, this number of frames are tracked");
850
851         /* pattern match */
852         prop= RNA_def_property(srna, "pattern_match", PROP_ENUM, PROP_NONE);
853         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
854         RNA_def_property_enum_sdna(prop, NULL, "pattern_match");
855         RNA_def_property_enum_items(prop, pattern_match_items);
856         RNA_def_property_ui_text(prop, "Pattern Match", "Track pattern from given frame when tracking marker to next frame");
857
858         /* margin */
859         prop= RNA_def_property(srna, "margin", PROP_INT, PROP_NONE);
860         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
861         RNA_def_property_int_sdna(prop, NULL, "margin");
862         RNA_def_property_range(prop, 0, 300);
863         RNA_def_property_ui_text(prop, "Margin", "Distance from image boudary at which marker stops tracking");
864
865         /* tracking algorithm */
866         prop= RNA_def_property(srna, "tracker", PROP_ENUM, PROP_NONE);
867         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
868         RNA_def_property_enum_items(prop, tracker_items);
869         RNA_def_property_ui_text(prop, "Tracker", "Tracking algorithm to use");
870         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_trackerAlgorithm_update");
871
872         /* pyramid level for pyramid klt tracking */
873         prop= RNA_def_property(srna, "pyramid_levels", PROP_INT, PROP_NONE);
874         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
875         RNA_def_property_int_sdna(prop, NULL, "pyramid_levels");
876         RNA_def_property_range(prop, 1, 16);
877         RNA_def_property_ui_text(prop, "Pyramid levels", "Number of pyramid levels (increase on blurry footage)");
878         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_trackerPyramid_update");
879
880         /* minmal correlation - only used for SAD tracker */
881         prop= RNA_def_property(srna, "correlation_min", PROP_FLOAT, PROP_NONE);
882         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
883         RNA_def_property_float_sdna(prop, NULL, "minimum_correlation");
884         RNA_def_property_range(prop, -1.0f, 1.0f);
885         RNA_def_property_ui_range(prop, -1.0f, 1.0f, 0.1, 3);
886         RNA_def_property_ui_text(prop, "Correlation", "Minimal value of correlation between matched pattern and reference which is still treated as successful tracking");
887
888         /* markers */
889         prop= RNA_def_property(srna, "markers", PROP_COLLECTION, PROP_NONE);
890         RNA_def_property_struct_type(prop, "MovieTrackingMarker");
891         RNA_def_property_collection_sdna(prop, NULL, "markers", "markersnr");
892         RNA_def_property_ui_text(prop, "Markers", "Collection of markers in track");
893         rna_def_trackingMarkers(brna, prop);
894
895         /* ** channels ** */
896
897         /* use_red_channel */
898         prop= RNA_def_property(srna, "use_red_channel", PROP_BOOLEAN, PROP_NONE);
899         RNA_def_property_boolean_negative_sdna(prop, NULL, "flag", TRACK_DISABLE_RED);
900         RNA_def_property_ui_text(prop, "Use Red Channel", "Use red channel from footage for tracking");
901         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
902
903         /* use_green_channel */
904         prop= RNA_def_property(srna, "use_green_channel", PROP_BOOLEAN, PROP_NONE);
905         RNA_def_property_boolean_negative_sdna(prop, NULL, "flag", TRACK_DISABLE_GREEN);
906         RNA_def_property_ui_text(prop, "Use Green Channel", "Use green channel from footage for tracking");
907         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
908
909         /* use_blue_channel */
910         prop= RNA_def_property(srna, "use_blue_channel", PROP_BOOLEAN, PROP_NONE);
911         RNA_def_property_boolean_negative_sdna(prop, NULL, "flag", TRACK_DISABLE_BLUE);
912         RNA_def_property_ui_text(prop, "Use Blue Channel", "Use blue channel from footage for tracking");
913         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
914
915         /* preview_grayscale */
916         prop= RNA_def_property(srna, "use_grayscale_preview", PROP_BOOLEAN, PROP_NONE);
917         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACK_PREVIEW_GRAYSCALE);
918         RNA_def_property_ui_text(prop, "Grayscale", "Display what the tracking algorithm sees in the preview");
919         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
920
921         /* has bundle */
922         prop= RNA_def_property(srna, "has_bundle", PROP_BOOLEAN, PROP_NONE);
923         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACK_HAS_BUNDLE);
924         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
925         RNA_def_property_ui_text(prop, "Has Bundle", "True if track has a valid bundle");
926
927         /* bundle position */
928         prop= RNA_def_property(srna, "bundle", PROP_FLOAT, PROP_TRANSLATION);
929         RNA_def_property_array(prop, 3);
930         RNA_def_property_float_sdna(prop, NULL, "bundle_pos");
931         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
932         RNA_def_property_ui_text(prop, "Bundle", "Position of bundle reconstructed from this track");
933
934         /* hide */
935         prop= RNA_def_property(srna, "hide", PROP_BOOLEAN, PROP_NONE);
936         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACK_HIDDEN);
937         RNA_def_property_ui_text(prop, "Hide", "Track is hidden");
938         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
939
940         /* select */
941         prop= RNA_def_property(srna, "select", PROP_BOOLEAN, PROP_NONE);
942         RNA_def_property_boolean_funcs(prop, "rna_trackingTrack_select_get", "rna_trackingTrack_select_set");
943         RNA_def_property_ui_text(prop, "Select", "Track is selected");
944         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
945
946         /* select_anchor */
947         prop= RNA_def_property(srna, "select_anchor", PROP_BOOLEAN, PROP_NONE);
948         RNA_def_property_boolean_sdna(prop, NULL, "flag", SELECT);
949         RNA_def_property_ui_text(prop, "Select Anchor", "Track's anchor point is selected");
950         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
951
952         /* select_pattern */
953         prop= RNA_def_property(srna, "select_pattern", PROP_BOOLEAN, PROP_NONE);
954         RNA_def_property_boolean_sdna(prop, NULL, "pat_flag", SELECT);
955         RNA_def_property_ui_text(prop, "Select Pattern", "Track's pattern area is selected");
956         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
957
958         /* select_search */
959         prop= RNA_def_property(srna, "select_search", PROP_BOOLEAN, PROP_NONE);
960         RNA_def_property_boolean_sdna(prop, NULL, "search_flag", SELECT);
961         RNA_def_property_ui_text(prop, "Select Search", "Track's search area is selected");
962         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
963
964         /* locked */
965         prop= RNA_def_property(srna, "lock", PROP_BOOLEAN, PROP_NONE);
966         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACK_LOCKED);
967         RNA_def_property_ui_text(prop, "Lock", "Track is locked and all changes to it are disabled");
968         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
969
970         /* custom color */
971         prop= RNA_def_property(srna, "use_custom_color", PROP_BOOLEAN, PROP_NONE);
972         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACK_CUSTOMCOLOR);
973         RNA_def_property_ui_text(prop, "Custom Color", "Use custom color instead of theme-defined");
974         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
975
976         /* color */
977         prop= RNA_def_property(srna, "color", PROP_FLOAT, PROP_COLOR);
978         RNA_def_property_array(prop, 3);
979         RNA_def_property_range(prop, 0.0f, 1.0f);
980         RNA_def_property_ui_text(prop, "Color", "Color of the track in the Movie Track Editor and the 3D viewport after a solve");
981         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
982
983         /* average error */
984         prop= RNA_def_property(srna, "average_error", PROP_FLOAT, PROP_NONE);
985         RNA_def_property_float_sdna(prop, NULL, "error");
986         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
987         RNA_def_property_ui_text(prop, "Average Error", "Average error of re-projection");
988 }
989
990 static void rna_def_trackingStabilization(BlenderRNA *brna)
991 {
992         StructRNA *srna;
993         PropertyRNA *prop;
994
995         srna= RNA_def_struct(brna, "MovieTrackingStabilization", NULL);
996         RNA_def_struct_ui_text(srna, "Movie tracking stabilization data", "Match-moving stabilization data for tracking");
997
998         /* 2d stabilization */
999         prop= RNA_def_property(srna, "use_2d_stabilization", PROP_BOOLEAN, PROP_NONE);
1000         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACKING_2D_STABILIZATION);
1001         RNA_def_property_ui_text(prop, "Use 2D stabilization", "Use 2D stabilization for footage");
1002         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
1003
1004         /* tracks */
1005         prop= RNA_def_property(srna, "tracks", PROP_COLLECTION, PROP_NONE);
1006         RNA_def_property_collection_funcs(prop, "rna_tracking_stabTracks_begin", "rna_iterator_listbase_next", "rna_iterator_listbase_end", "rna_iterator_listbase_get", 0, 0, 0, 0);
1007         RNA_def_property_struct_type(prop, "MovieTrackingTrack");
1008         RNA_def_property_ui_text(prop, "Tracks", "Collection of tracks used for stabilization");
1009         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
1010
1011         /* rotation track */
1012         prop= RNA_def_property(srna, "rotation_track", PROP_POINTER, PROP_NONE);
1013         RNA_def_property_pointer_sdna(prop, NULL, "rot_track");
1014         RNA_def_property_flag(prop, PROP_EDITABLE);
1015         RNA_def_property_ui_text(prop, "Rotation Track", "Track used to compensate rotation");
1016         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_tracking_flushUpdate");
1017
1018         /* active track index */
1019         prop= RNA_def_property(srna, "active_track_index", PROP_INT, PROP_NONE);
1020         RNA_def_property_int_sdna(prop, NULL, "act_track");
1021         RNA_def_property_int_funcs(prop, "rna_tracking_stabTracks_active_index_get", "rna_tracking_stabTracks_active_index_set", "rna_tracking_stabTracks_active_index_range");
1022         RNA_def_property_ui_text(prop, "Active Track Index", "Index of active track in stabilization tracks list");
1023
1024         /* autoscale */
1025         prop= RNA_def_property(srna, "use_autoscale", PROP_BOOLEAN, PROP_NONE);
1026         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACKING_AUTOSCALE);
1027         RNA_def_property_ui_text(prop, "Autoscale", "Automatically scale footage to cover unfilled areas when stabilizating");
1028         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
1029
1030         /* max scale */
1031         prop= RNA_def_property(srna, "scale_max", PROP_FLOAT, PROP_FACTOR);
1032         RNA_def_property_float_sdna(prop, NULL, "maxscale");
1033         RNA_def_property_range(prop, 0.0f, 10.0f);
1034         RNA_def_property_ui_text(prop, "Maximal Scale", "Limit the amount of automatic scaling");
1035         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
1036
1037         /* influence_location */
1038         prop= RNA_def_property(srna, "influence_location", PROP_FLOAT, PROP_FACTOR);
1039         RNA_def_property_float_sdna(prop, NULL, "locinf");
1040         RNA_def_property_range(prop, 0.0f, 1.0f);
1041         RNA_def_property_ui_text(prop, "Location Influence", "Influence of stabilization algorithm on footage location");
1042         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
1043
1044         /* influence_scale */
1045         prop= RNA_def_property(srna, "influence_scale", PROP_FLOAT, PROP_FACTOR);
1046         RNA_def_property_float_sdna(prop, NULL, "scaleinf");
1047         RNA_def_property_range(prop, 0.0f, 1.0f);
1048         RNA_def_property_ui_text(prop, "Scale Influence", "Influence of stabilization algorithm on footage scale");
1049         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
1050
1051         /* use_stabilize_rotation */
1052         prop= RNA_def_property(srna, "use_stabilize_rotation", PROP_BOOLEAN, PROP_NONE);
1053         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACKING_STABILIZE_ROTATION);
1054         RNA_def_property_ui_text(prop, "Stabilize Rotation", "Stabilize horizon line on the shot");
1055         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
1056
1057         /* influence_rotation */
1058         prop= RNA_def_property(srna, "influence_rotation", PROP_FLOAT, PROP_FACTOR);
1059         RNA_def_property_float_sdna(prop, NULL, "rotinf");
1060         RNA_def_property_range(prop, 0.0f, 1.0f);
1061         RNA_def_property_ui_text(prop, "Rotation Influence", "Influence of stabilization algorithm on footage rotation");
1062         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, "rna_tracking_flushUpdate");
1063 }
1064
1065 static void rna_def_reconstructedCamera(BlenderRNA *brna)
1066 {
1067         StructRNA *srna;
1068         PropertyRNA *prop;
1069
1070         srna= RNA_def_struct(brna, "MovieReconstructedCamera", NULL);
1071         RNA_def_struct_ui_text(srna, "Movie tracking reconstructed camera data", "Match-moving reconstructed camera data from tracker");
1072
1073         /* frame */
1074         prop= RNA_def_property(srna, "frame", PROP_INT, PROP_NONE);
1075         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
1076         RNA_def_property_int_sdna(prop, NULL, "framenr");
1077         RNA_def_property_ui_text(prop, "Frame", "Frame number marker is keyframed on");
1078
1079         /* matrix */
1080         prop= RNA_def_property(srna, "matrix", PROP_FLOAT, PROP_MATRIX);
1081         RNA_def_property_float_sdna(prop, NULL, "mat");
1082         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
1083         RNA_def_property_multi_array(prop, 2, rna_matrix_dimsize_4x4);
1084         RNA_def_property_ui_text(prop, "Matrix", "Worldspace transformation matrix");
1085
1086         /* average_error */
1087         prop= RNA_def_property(srna, "average_error", PROP_FLOAT, PROP_NONE);
1088         RNA_def_property_float_sdna(prop, NULL, "error");
1089         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
1090         RNA_def_property_ui_text(prop, "Average Error", "Average error of resonctruction");
1091 }
1092
1093 static void rna_def_trackingReconstruction(BlenderRNA *brna)
1094 {
1095         StructRNA *srna;
1096         PropertyRNA *prop;
1097
1098         rna_def_reconstructedCamera(brna);
1099
1100         srna= RNA_def_struct(brna, "MovieTrackingReconstruction", NULL);
1101         RNA_def_struct_ui_text(srna, "Movie tracking reconstruction data", "Match-moving reconstruction data from tracker");
1102
1103         /* is_valid */
1104         prop= RNA_def_property(srna, "is_valid", PROP_BOOLEAN, PROP_NONE);
1105         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
1106         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACKING_RECONSTRUCTED);
1107         RNA_def_property_ui_text(prop, "Reconstructed", "Is tracking data contains valid reconstruction information");
1108
1109         /* average_error */
1110         prop= RNA_def_property(srna, "average_error", PROP_FLOAT, PROP_NONE);
1111         RNA_def_property_float_sdna(prop, NULL, "error");
1112         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
1113         RNA_def_property_ui_text(prop, "Average Error", "Average error of resonctruction");
1114
1115         /* cameras */
1116         prop= RNA_def_property(srna, "cameras", PROP_COLLECTION, PROP_NONE);
1117         RNA_def_property_struct_type(prop, "MovieReconstructedCamera");
1118         RNA_def_property_collection_sdna(prop, NULL, "cameras", "camnr");
1119         RNA_def_property_ui_text(prop, "Cameras", "Collection of solved cameras");
1120 }
1121
1122 static void rna_def_trackingTracks(BlenderRNA *brna)
1123 {
1124         StructRNA *srna;
1125         FunctionRNA *func;
1126         PropertyRNA *prop;
1127
1128         srna= RNA_def_struct(brna, "MovieTrackingTracks", NULL);
1129         RNA_def_struct_sdna(srna, "MovieTracking");
1130         RNA_def_struct_ui_text(srna, "Movie Tracks", "Collection of movie tracking tracks");
1131
1132         func= RNA_def_function(srna, "add", "rna_tracking_tracks_add");
1133         RNA_def_function_flag(func, FUNC_USE_SELF_ID);
1134         RNA_def_function_ui_description(func, "Add a number of tracks to this movie clip");
1135         RNA_def_int(func, "frame", 1, MINFRAME, MAXFRAME, "Frame", "Frame number to add tracks on", MINFRAME, MAXFRAME);
1136         RNA_def_int(func, "count", 1, 0, INT_MAX, "Number", "Number of tracks to add to the movie clip", 0, INT_MAX);
1137
1138         /* active track */
1139         prop= RNA_def_property(srna, "active", PROP_POINTER, PROP_NONE);
1140         RNA_def_property_struct_type(prop, "MovieTrackingTrack");
1141         RNA_def_property_pointer_funcs(prop, "rna_tracking_active_track_get", "rna_tracking_active_track_set", NULL, NULL);
1142         RNA_def_property_flag(prop, PROP_EDITABLE|PROP_NEVER_UNLINK);
1143         RNA_def_property_ui_text(prop, "Active Track", "Active track in this tracking data object");
1144 }
1145
1146 static void rna_def_trackingObjectTracks(BlenderRNA *brna)
1147 {
1148         StructRNA *srna;
1149         FunctionRNA *func;
1150         PropertyRNA *prop;
1151
1152         srna= RNA_def_struct(brna, "MovieTrackingObjectTracks", NULL);
1153         RNA_def_struct_sdna(srna, "MovieTrackingObject");
1154         RNA_def_struct_ui_text(srna, "Movie Tracks", "Collection of movie tracking tracks");
1155
1156         func= RNA_def_function(srna, "add", "rna_trackingObject_tracks_add");
1157         RNA_def_function_flag(func, FUNC_USE_SELF_ID);
1158         RNA_def_function_ui_description(func, "Add a number of tracks to this movie clip");
1159         RNA_def_int(func, "frame", 1, MINFRAME, MAXFRAME, "Frame", "Frame number to add tracks on", MINFRAME, MAXFRAME);
1160         RNA_def_int(func, "count", 1, 0, INT_MAX, "Number", "Number of tracks to add to the movie clip", 0, INT_MAX);
1161
1162         /* active track */
1163         prop= RNA_def_property(srna, "active", PROP_POINTER, PROP_NONE);
1164         RNA_def_property_struct_type(prop, "MovieTrackingTrack");
1165         RNA_def_property_pointer_funcs(prop, "rna_tracking_active_track_get", "rna_tracking_active_track_set", NULL, NULL);
1166         RNA_def_property_flag(prop, PROP_EDITABLE|PROP_NEVER_UNLINK);
1167         RNA_def_property_ui_text(prop, "Active Track", "Active track in this tracking data object");
1168 }
1169
1170 static void rna_def_trackingObject(BlenderRNA *brna)
1171 {
1172         StructRNA *srna;
1173         PropertyRNA *prop;
1174
1175         srna= RNA_def_struct(brna, "MovieTrackingObject", NULL);
1176         RNA_def_struct_ui_text(srna, "Movie tracking object data", "Match-moving object tracking and reconstruction data");
1177
1178         /* name */
1179         prop= RNA_def_property(srna, "name", PROP_STRING, PROP_NONE);
1180         RNA_def_property_ui_text(prop, "Name", "Unique name of object");
1181         RNA_def_property_string_funcs(prop, NULL, NULL, "rna_trackingObject_name_set");
1182         RNA_def_property_string_maxlength(prop, MAX_ID_NAME);
1183         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, NULL);
1184         RNA_def_struct_name_property(srna, prop);
1185
1186         /* is_camera */
1187         prop= RNA_def_property(srna, "is_camera", PROP_BOOLEAN, PROP_NONE);
1188         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
1189         RNA_def_property_boolean_sdna(prop, NULL, "flag", TRACKING_OBJECT_CAMERA);
1190         RNA_def_property_ui_text(prop, "Camera", "Object is used for camera tracking");
1191         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
1192
1193         /* tracks */
1194         prop= RNA_def_property(srna, "tracks", PROP_COLLECTION, PROP_NONE);
1195         RNA_def_property_collection_funcs(prop, "rna_trackingObject_tracks_begin", "rna_iterator_listbase_next", "rna_iterator_listbase_end", "rna_iterator_listbase_get", 0, 0, 0, 0);
1196         RNA_def_property_struct_type(prop, "MovieTrackingTrack");
1197         RNA_def_property_ui_text(prop, "Tracks", "Collection of tracks in this tracking data object");
1198         RNA_def_property_srna(prop, "MovieTrackingObjectTracks");
1199
1200         /* reconstruction */
1201         prop= RNA_def_property(srna, "reconstruction", PROP_POINTER, PROP_NONE);
1202         RNA_def_property_struct_type(prop, "MovieTrackingReconstruction");
1203
1204         /* scale */
1205         prop= RNA_def_property(srna, "scale", PROP_FLOAT, PROP_NONE);
1206         RNA_def_property_clear_flag(prop, PROP_ANIMATABLE);
1207         RNA_def_property_float_sdna(prop, NULL, "scale");
1208         RNA_def_property_range(prop, 0.0001f, 10000.0f);
1209         RNA_def_property_ui_range(prop, 0.0001f, 10000.0, 1, 4);
1210         RNA_def_property_ui_text(prop, "Scale", "Scale of object solution in camera space");
1211         RNA_def_property_update(prop, NC_MOVIECLIP|NA_EDITED, "rna_trackingObject_flushUpdate");
1212 }
1213
1214 static void rna_def_trackingObjects(BlenderRNA *brna, PropertyRNA *cprop)
1215 {
1216         StructRNA *srna;
1217         PropertyRNA *prop;
1218
1219         FunctionRNA *func;
1220         PropertyRNA *parm;
1221
1222         RNA_def_property_srna(cprop, "MovieTrackingObjects");
1223         srna= RNA_def_struct(brna, "MovieTrackingObjects", NULL);
1224         RNA_def_struct_sdna(srna, "MovieTracking");
1225         RNA_def_struct_ui_text(srna, "Movie Objects", "Collection of movie trackingobjects");
1226
1227         func= RNA_def_function(srna, "new", "rna_tracking_object_new");
1228         RNA_def_function_ui_description(func, "Add tracking object to this movie clip");
1229         RNA_def_string(func, "name", "", 0, "", "Name of new object");
1230         parm= RNA_def_pointer(func, "object", "MovieTrackingObject", "", "New motion tracking object");
1231         RNA_def_function_return(func, parm);
1232
1233         func= RNA_def_function(srna, "remove", "rna_tracking_object_remove");
1234         RNA_def_function_ui_description(func, "Remove tracking object from this movie clip");
1235         parm= RNA_def_pointer(func, "object", "MovieTrackingObject", "", "Motion tracking object to be removed");
1236
1237         /* active object */
1238         prop= RNA_def_property(srna, "active", PROP_POINTER, PROP_NONE);
1239         RNA_def_property_struct_type(prop, "MovieTrackingObject");
1240         RNA_def_property_pointer_funcs(prop, "rna_tracking_active_object_get", "rna_tracking_active_object_set", NULL, NULL);
1241         RNA_def_property_flag(prop, PROP_EDITABLE|PROP_NEVER_UNLINK);
1242         RNA_def_property_ui_text(prop, "Active Object", "Active object in this tracking data object");
1243 }
1244
1245 static void rna_def_tracking(BlenderRNA *brna)
1246 {
1247         StructRNA *srna;
1248         PropertyRNA *prop;
1249
1250         rna_def_trackingSettings(brna);
1251         rna_def_trackingCamera(brna);
1252         rna_def_trackingTrack(brna);
1253         rna_def_trackingTracks(brna);
1254         rna_def_trackingObjectTracks(brna);
1255         rna_def_trackingStabilization(brna);
1256         rna_def_trackingReconstruction(brna);
1257         rna_def_trackingObject(brna);
1258
1259         srna= RNA_def_struct(brna, "MovieTracking", NULL);
1260         RNA_def_struct_ui_text(srna, "Movie tracking data", "Match-moving data for tracking");
1261
1262         /* settings */
1263         prop= RNA_def_property(srna, "settings", PROP_POINTER, PROP_NONE);
1264         RNA_def_property_struct_type(prop, "MovieTrackingSettings");
1265
1266         /* camera properties */
1267         prop= RNA_def_property(srna, "camera", PROP_POINTER, PROP_NONE);
1268         RNA_def_property_struct_type(prop, "MovieTrackingCamera");
1269
1270         /* tracks */
1271         prop= RNA_def_property(srna, "tracks", PROP_COLLECTION, PROP_NONE);
1272         RNA_def_property_collection_funcs(prop, "rna_tracking_tracks_begin", "rna_iterator_listbase_next", "rna_iterator_listbase_end", "rna_iterator_listbase_get", 0, 0, 0, 0);
1273         RNA_def_property_struct_type(prop, "MovieTrackingTrack");
1274         RNA_def_property_ui_text(prop, "Tracks", "Collection of tracks in this tracking data object");
1275         RNA_def_property_srna(prop, "MovieTrackingTracks");
1276
1277         /* stabilization */
1278         prop= RNA_def_property(srna, "stabilization", PROP_POINTER, PROP_NONE);
1279         RNA_def_property_struct_type(prop, "MovieTrackingStabilization");
1280
1281         /* reconstruction */
1282         prop= RNA_def_property(srna, "reconstruction", PROP_POINTER, PROP_NONE);
1283         RNA_def_property_struct_type(prop, "MovieTrackingReconstruction");
1284
1285         /* objects */
1286         prop= RNA_def_property(srna, "objects", PROP_COLLECTION, PROP_NONE);
1287         RNA_def_property_collection_funcs(prop, "rna_tracking_objects_begin", "rna_iterator_listbase_next", "rna_iterator_listbase_end", "rna_iterator_listbase_get", 0, 0, 0, 0);
1288         RNA_def_property_struct_type(prop, "MovieTrackingObject");
1289         RNA_def_property_ui_text(prop, "Objects", "Collection of objects in this tracking data object");
1290         rna_def_trackingObjects(brna, prop);
1291
1292         /* active object index */
1293         prop= RNA_def_property(srna, "active_object_index", PROP_INT, PROP_NONE);
1294         RNA_def_property_int_sdna(prop, NULL, "objectnr");
1295         RNA_def_property_int_funcs(prop, "rna_tracking_active_object_index_get", "rna_tracking_active_object_index_set", "rna_tracking_active_object_index_range");
1296         RNA_def_property_ui_text(prop, "Active Object Index", "Index of active object");
1297         RNA_def_property_update(prop, NC_MOVIECLIP|ND_DISPLAY, NULL);
1298 }
1299
1300 void RNA_def_tracking(BlenderRNA *brna)
1301 {
1302         rna_def_tracking(brna);
1303 }
1304
1305 #endif