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