svn merge ^/trunk/blender -r49945:49947
[blender-staging.git] / source / blender / editors / space_clip / clip_buttons.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  * The Original Code is Copyright (C) 2011 Blender Foundation.
19  * All rights reserved.
20  *
21  *
22  * Contributor(s): Blender Foundation,
23  *                 Sergey Sharybin
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/editors/space_clip/clip_buttons.c
29  *  \ingroup spclip
30  */
31
32 #include <string.h>
33 #include <stdio.h>
34
35 #include "MEM_guardedalloc.h"
36
37 #include "DNA_scene_types.h"
38 #include "DNA_screen_types.h"
39 #include "DNA_space_types.h"
40
41 #include "BLI_math.h"
42 #include "BLI_utildefines.h"
43 #include "BLI_listbase.h"
44
45 #include "BKE_context.h"
46 #include "BKE_depsgraph.h"
47 #include "BKE_screen.h"
48 #include "BKE_movieclip.h"
49 #include "BKE_tracking.h"
50
51 #include "ED_clip.h"
52 #include "ED_gpencil.h"
53
54 #include "UI_interface.h"
55 #include "UI_resources.h"
56
57 #include "RNA_access.h"
58
59 #include "WM_api.h"
60 #include "WM_types.h"
61
62 #include "clip_intern.h"  /* own include */
63
64 /* Panels */
65
66 static int clip_grease_pencil_panel_poll(const bContext *C, PanelType *UNUSED(pt))
67 {
68         SpaceClip *sc = CTX_wm_space_clip(C);
69
70         return sc->view == SC_VIEW_CLIP;
71 }
72
73 void ED_clip_buttons_register(ARegionType *art)
74 {
75         PanelType *pt;
76
77         pt = MEM_callocN(sizeof(PanelType), "spacetype clip panel gpencil");
78         strcpy(pt->idname, "CLIP_PT_gpencil");
79         strcpy(pt->label, "Grease Pencil");
80         pt->draw = gpencil_panel_standard;
81         pt->flag |= PNL_DEFAULT_CLOSED;
82         pt->poll = clip_grease_pencil_panel_poll;
83         BLI_addtail(&art->paneltypes, pt);
84 }
85
86 /********************* MovieClip Template ************************/
87
88 void uiTemplateMovieClip(uiLayout *layout, bContext *C, PointerRNA *ptr, const char *propname, int compact)
89 {
90         PropertyRNA *prop;
91         PointerRNA clipptr;
92         MovieClip *clip;
93         uiLayout *row, *split;
94         uiBlock *block;
95
96         if (!ptr->data)
97                 return;
98
99         prop = RNA_struct_find_property(ptr, propname);
100         if (!prop) {
101                 printf("%s: property not found: %s.%s\n",
102                        __func__, RNA_struct_identifier(ptr->type), propname);
103                 return;
104         }
105
106         if (RNA_property_type(prop) != PROP_POINTER) {
107                 printf("%s: expected pointer property for %s.%s\n",
108                        __func__, RNA_struct_identifier(ptr->type), propname);
109                 return;
110         }
111
112         clipptr = RNA_property_pointer_get(ptr, prop);
113         clip = clipptr.data;
114
115         uiLayoutSetContextPointer(layout, "edit_movieclip", &clipptr);
116
117         if (!compact)
118                 uiTemplateID(layout, C, ptr, propname, NULL, "CLIP_OT_open", NULL);
119
120         if (clip) {
121                 uiLayout *col;
122
123                 row = uiLayoutRow(layout, FALSE);
124                 block = uiLayoutGetBlock(row);
125                 uiDefBut(block, LABEL, 0, "File Path:", 0, 19, 145, 19, NULL, 0, 0, 0, 0, "");
126
127                 row = uiLayoutRow(layout, FALSE);
128                 split = uiLayoutSplit(row, 0.0f, FALSE);
129                 row = uiLayoutRow(split, TRUE);
130
131                 uiItemR(row, &clipptr, "filepath", 0, "", ICON_NONE);
132                 uiItemO(row, "", ICON_FILE_REFRESH, "clip.reload");
133
134                 col = uiLayoutColumn(layout, FALSE);
135                 uiTemplateColorspaceSettings(col, &clipptr, "colorspace_settings");
136         }
137 }
138
139 /********************* Track Template ************************/
140
141 void uiTemplateTrack(uiLayout *layout, PointerRNA *ptr, const char *propname)
142 {
143         PropertyRNA *prop;
144         PointerRNA scopesptr;
145         uiBlock *block;
146         rctf rect;
147         MovieClipScopes *scopes;
148
149         if (!ptr->data)
150                 return;
151
152         prop = RNA_struct_find_property(ptr, propname);
153         if (!prop) {
154                 printf("%s: property not found: %s.%s\n",
155                        __func__, RNA_struct_identifier(ptr->type), propname);
156                 return;
157         }
158
159         if (RNA_property_type(prop) != PROP_POINTER) {
160                 printf("%s: expected pointer property for %s.%s\n",
161                        __func__, RNA_struct_identifier(ptr->type), propname);
162                 return;
163         }
164
165         scopesptr = RNA_property_pointer_get(ptr, prop);
166         scopes = (MovieClipScopes *)scopesptr.data;
167
168         rect.xmin = 0; rect.xmax = 200;
169         rect.ymin = 0; rect.ymax = 120;
170
171         block = uiLayoutAbsoluteBlock(layout);
172
173         scopes->track_preview_height =
174                 (scopes->track_preview_height <= UI_UNIT_Y) ? UI_UNIT_Y : scopes->track_preview_height;
175
176         uiDefBut(block, TRACKPREVIEW, 0, "", rect.xmin, rect.ymin, rect.xmax - rect.xmin,
177                  scopes->track_preview_height, scopes, 0, 0, 0, 0, "");
178 }
179
180 /********************* Marker Template ************************/
181
182 #define B_MARKER_POS            3
183 #define B_MARKER_OFFSET         4
184 #define B_MARKER_PAT_DIM        5
185 #define B_MARKER_SEARCH_POS     6
186 #define B_MARKER_SEARCH_DIM     7
187 #define B_MARKER_FLAG           8
188
189 typedef struct {
190         int compact;                                /* compact mode */
191
192         MovieClip *clip;
193         MovieClipUser *user;                        /* user of clip */
194         MovieTrackingTrack *track;
195         MovieTrackingMarker *marker;
196
197         int framenr;                                    /* current frame number */
198         float marker_pos[2];                            /* position of marker in pixel coords */
199         float marker_pat[2];                            /* position and dimensions of marker pattern in pixel coords */
200         float track_offset[2];                          /* offset of "parenting" point */
201         float marker_search_pos[2], marker_search[2];   /* position and dimensions of marker search in pixel coords */
202         int marker_flag;                                /* marker's flags */
203 } MarkerUpdateCb;
204
205 static void to_pixel_space(float r[2], float a[2], int width, int height)
206 {
207         copy_v2_v2(r, a);
208         r[0] *= width;
209         r[1] *= height;
210 }
211
212 static void marker_update_cb(bContext *C, void *arg_cb, void *UNUSED(arg))
213 {
214         MarkerUpdateCb *cb = (MarkerUpdateCb *) arg_cb;
215         MovieTrackingMarker *marker;
216
217         if (!cb->compact)
218                 return;
219
220         marker = BKE_tracking_marker_ensure(cb->track, cb->framenr);
221
222         marker->flag = cb->marker_flag;
223
224         WM_event_add_notifier(C, NC_MOVIECLIP | NA_EDITED, NULL);
225 }
226
227 static void marker_block_handler(bContext *C, void *arg_cb, int event)
228 {
229         MarkerUpdateCb *cb = (MarkerUpdateCb *) arg_cb;
230         MovieTrackingMarker *marker;
231         int width, height, ok = FALSE;
232
233         BKE_movieclip_get_size(cb->clip, cb->user, &width, &height);
234
235         marker = BKE_tracking_marker_ensure(cb->track, cb->framenr);
236
237         if (event == B_MARKER_POS) {
238                 marker->pos[0] = cb->marker_pos[0] / width;
239                 marker->pos[1] = cb->marker_pos[1] / height;
240
241                 /* to update position of "parented" objects */
242                 DAG_id_tag_update(&cb->clip->id, 0);
243                 WM_event_add_notifier(C, NC_SPACE | ND_SPACE_VIEW3D, NULL);
244
245                 ok = TRUE;
246         }
247         else if (event == B_MARKER_PAT_DIM) {
248                 float dim[2], pat_dim[2], pat_min[2], pat_max[2];
249                 float scale_x, scale_y;
250                 int a;
251
252                 BKE_tracking_marker_pattern_minmax(cb->marker, pat_min, pat_max);
253
254                 sub_v2_v2v2(pat_dim, pat_max, pat_min);
255
256                 dim[0] = cb->marker_pat[0] / width;
257                 dim[1] = cb->marker_pat[1] / height;
258
259                 scale_x = dim[0] / pat_dim[0];
260                 scale_y = dim[1] / pat_dim[1];
261
262                 for (a = 0; a < 4; a++) {
263                         cb->marker->pattern_corners[a][0] *= scale_x;
264                         cb->marker->pattern_corners[a][1] *= scale_y;
265                 }
266
267                 BKE_tracking_marker_clamp(cb->marker, CLAMP_PAT_DIM);
268
269                 ok = TRUE;
270         }
271         else if (event == B_MARKER_SEARCH_POS) {
272                 float delta[2], side[2];
273
274                 sub_v2_v2v2(side, cb->marker->search_max, cb->marker->search_min);
275                 mul_v2_fl(side, 0.5f);
276
277                 delta[0] = cb->marker_search_pos[0] / width;
278                 delta[1] = cb->marker_search_pos[1] / height;
279
280                 sub_v2_v2v2(cb->marker->search_min, delta, side);
281                 add_v2_v2v2(cb->marker->search_max, delta, side);
282
283                 BKE_tracking_marker_clamp(cb->marker, CLAMP_SEARCH_POS);
284
285                 ok = TRUE;
286         }
287         else if (event == B_MARKER_SEARCH_DIM) {
288                 float dim[2], search_dim[2];
289
290                 sub_v2_v2v2(search_dim, cb->marker->search_max, cb->marker->search_min);
291
292                 dim[0] = cb->marker_search[0] / width;
293                 dim[1] = cb->marker_search[1] / height;
294
295                 sub_v2_v2(dim, search_dim);
296                 mul_v2_fl(dim, 0.5f);
297
298                 cb->marker->search_min[0] -= dim[0];
299                 cb->marker->search_min[1] -= dim[1];
300
301                 cb->marker->search_max[0] += dim[0];
302                 cb->marker->search_max[1] += dim[1];
303
304                 BKE_tracking_marker_clamp(cb->marker, CLAMP_SEARCH_DIM);
305
306                 ok = TRUE;
307         }
308         else if (event == B_MARKER_FLAG) {
309                 marker->flag = cb->marker_flag;
310
311                 ok = TRUE;
312         }
313         else if (event == B_MARKER_OFFSET) {
314                 float offset[2], delta[2];
315                 int i;
316
317                 offset[0] = cb->track_offset[0] / width;
318                 offset[1] = cb->track_offset[1] / height;
319
320                 sub_v2_v2v2(delta, offset, cb->track->offset);
321                 copy_v2_v2(cb->track->offset, offset);
322
323                 for (i = 0; i < cb->track->markersnr; i++)
324                         sub_v2_v2(cb->track->markers[i].pos, delta);
325
326                 /* to update position of "parented" objects */
327                 DAG_id_tag_update(&cb->clip->id, 0);
328                 WM_event_add_notifier(C, NC_SPACE | ND_SPACE_VIEW3D, NULL);
329
330                 ok = TRUE;
331         }
332
333         if (ok)
334                 WM_event_add_notifier(C, NC_MOVIECLIP | NA_EDITED, cb->clip);
335 }
336
337 void uiTemplateMarker(uiLayout *layout, PointerRNA *ptr, const char *propname, PointerRNA *userptr,
338                       PointerRNA *trackptr, int compact)
339 {
340         PropertyRNA *prop;
341         uiBlock *block;
342         uiBut *bt;
343         PointerRNA clipptr;
344         MovieClip *clip;
345         MovieClipUser *user;
346         MovieTrackingTrack *track;
347         MovieTrackingMarker *marker;
348         MarkerUpdateCb *cb;
349         const char *tip;
350         float pat_min[2], pat_max[2];
351
352         if (!ptr->data)
353                 return;
354
355         prop = RNA_struct_find_property(ptr, propname);
356         if (!prop) {
357                 printf("%s: property not found: %s.%s\n",
358                        __func__, RNA_struct_identifier(ptr->type), propname);
359                 return;
360         }
361
362         if (RNA_property_type(prop) != PROP_POINTER) {
363                 printf("%s: expected pointer property for %s.%s\n",
364                        __func__, RNA_struct_identifier(ptr->type), propname);
365                 return;
366         }
367
368         clipptr = RNA_property_pointer_get(ptr, prop);
369         clip = (MovieClip *)clipptr.data;
370         user = userptr->data;
371         track = trackptr->data;
372
373         marker = BKE_tracking_marker_get(track, user->framenr);
374
375         cb = MEM_callocN(sizeof(MarkerUpdateCb), "uiTemplateMarker update_cb");
376         cb->compact = compact;
377         cb->clip = clip;
378         cb->user = user;
379         cb->track = track;
380         cb->marker = marker;
381         cb->marker_flag = marker->flag;
382         cb->framenr = user->framenr;
383
384         if (compact) {
385                 block = uiLayoutGetBlock(layout);
386
387                 if (cb->marker_flag & MARKER_DISABLED)
388                         tip = "Marker is disabled at current frame";
389                 else
390                         tip = "Marker is enabled at current frame";
391
392                 bt = uiDefIconButBitI(block, TOGN, MARKER_DISABLED, 0, ICON_RESTRICT_VIEW_OFF, 0, 0, 20, 20,
393                                       &cb->marker_flag, 0, 0, 1, 0, tip);
394                 uiButSetNFunc(bt, marker_update_cb, cb, NULL);
395         }
396         else {
397                 int width, height, step, digits;
398                 float pat_dim[2], search_dim[2], search_pos[2];
399                 uiLayout *col;
400
401                 BKE_movieclip_get_size(clip, user, &width, &height);
402
403                 if (track->flag & TRACK_LOCKED) {
404                         uiLayoutSetActive(layout, FALSE);
405                         block = uiLayoutAbsoluteBlock(layout);
406                         uiDefBut(block, LABEL, 0, "Track is locked", 0, 0, 300, 19, NULL, 0, 0, 0, 0, "");
407
408                         return;
409                 }
410
411                 step = 100;
412                 digits = 2;
413
414                 BKE_tracking_marker_pattern_minmax(marker, pat_min, pat_max);
415
416                 sub_v2_v2v2(pat_dim, pat_max, pat_min);
417                 sub_v2_v2v2(search_dim, marker->search_max, marker->search_min);
418
419                 add_v2_v2v2(search_pos, marker->search_max, marker->search_min);
420                 mul_v2_fl(search_pos, 0.5);
421
422                 to_pixel_space(cb->marker_pos, marker->pos, width, height);
423                 to_pixel_space(cb->marker_pat, pat_dim, width, height);
424                 to_pixel_space(cb->marker_search, search_dim, width, height);
425                 to_pixel_space(cb->marker_search_pos, search_pos, width, height);
426                 to_pixel_space(cb->track_offset, track->offset, width, height);
427
428                 cb->marker_flag = marker->flag;
429
430                 block = uiLayoutAbsoluteBlock(layout);
431                 uiBlockSetHandleFunc(block, marker_block_handler, cb);
432                 uiBlockSetNFunc(block, marker_update_cb, cb, NULL);
433
434                 if (cb->marker_flag & MARKER_DISABLED)
435                         tip = "Marker is disabled at current frame";
436                 else
437                         tip = "Marker is enabled at current frame";
438
439                 uiDefButBitI(block, OPTIONN, MARKER_DISABLED, B_MARKER_FLAG,  "Enabled", 10, 190, 145, 19, &cb->marker_flag,
440                              0, 0, 0, 0, tip);
441
442                 col = uiLayoutColumn(layout, TRUE);
443                 uiLayoutSetActive(col, (cb->marker_flag & MARKER_DISABLED) == 0);
444
445                 block = uiLayoutAbsoluteBlock(col);
446                 uiBlockBeginAlign(block);
447
448                 uiDefBut(block, LABEL, 0, "Position:", 0, 190, 300, 19, NULL, 0, 0, 0, 0, "");
449                 uiDefButF(block, NUM, B_MARKER_POS, "X:", 10, 171, 145, 19, &cb->marker_pos[0],
450                           -10 * width, 10.0 * width, step, digits, "X-position of marker at frame in screen coordinates");
451                 uiDefButF(block, NUM, B_MARKER_POS, "Y:", 165, 171, 145, 19, &cb->marker_pos[1],
452                           -10 * height, 10.0 * height, step, digits, "Y-position of marker at frame in screen coordinates");
453
454                 uiDefBut(block, LABEL, 0, "Offset:", 0, 152, 300, 19, NULL, 0, 0, 0, 0, "");
455                 uiDefButF(block, NUM, B_MARKER_OFFSET, "X:", 10, 133, 145, 19, &cb->track_offset[0],
456                           -10 * width, 10.0 * width, step, digits, "X-offset to parenting point");
457                 uiDefButF(block, NUM, B_MARKER_OFFSET, "Y:", 165, 133, 145, 19, &cb->track_offset[1],
458                           -10 * height, 10.0 * height, step, digits, "Y-offset to parenting point");
459
460                 uiDefBut(block, LABEL, 0, "Pattern Area:", 0, 114, 300, 19, NULL, 0, 0, 0, 0, "");
461                 uiDefButF(block, NUM, B_MARKER_PAT_DIM, "Width:", 10, 95, 300, 19, &cb->marker_pat[0], 3.0f,
462                           10.0 * width, step, digits, "Width of marker's pattern in screen coordinates");
463                 uiDefButF(block, NUM, B_MARKER_PAT_DIM, "Height:", 10, 76, 300, 19, &cb->marker_pat[1], 3.0f,
464                           10.0 * height, step, digits, "Height of marker's pattern in screen coordinates");
465
466                 uiDefBut(block, LABEL, 0, "Search Area:", 0, 57, 300, 19, NULL, 0, 0, 0, 0, "");
467                 uiDefButF(block, NUM, B_MARKER_SEARCH_POS, "X:", 10, 38, 145, 19, &cb->marker_search_pos[0],
468                           -width, width, step, digits, "X-position of search at frame relative to marker's position");
469                 uiDefButF(block, NUM, B_MARKER_SEARCH_POS, "Y:", 165, 38, 145, 19, &cb->marker_search_pos[1],
470                           -height, height, step, digits, "X-position of search at frame relative to marker's position");
471                 uiDefButF(block, NUM, B_MARKER_SEARCH_DIM, "Width:", 10, 19, 300, 19, &cb->marker_search[0], 3.0f,
472                           10.0 * width, step, digits, "Width of marker's search in screen soordinates");
473                 uiDefButF(block, NUM, B_MARKER_SEARCH_DIM, "Height:", 10, 0, 300, 19, &cb->marker_search[1], 3.0f,
474                           10.0 * height, step, digits, "Height of marker's search in screen soordinates");
475
476                 uiBlockEndAlign(block);
477         }
478 }