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