Merging r50625 through r51896 from trunk into soc-2011-tomato
[blender-staging.git] / source / blender / editors / mask / mask_relationships.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) 2012 Blender Foundation.
19  * All rights reserved.
20  *
21  *
22  * Contributor(s): Blender Foundation,
23  *                 Campbell Barton
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/editors/mask/mask_relationships.c
29  *  \ingroup edmask
30  */
31
32
33 #include "BLI_math.h"
34
35 #include "BKE_context.h"
36 #include "BKE_depsgraph.h"
37 #include "BKE_mask.h"
38 #include "BKE_tracking.h"
39
40 #include "DNA_mask_types.h"
41 #include "DNA_object_types.h"  /* SELECT */
42
43 #include "WM_api.h"
44 #include "WM_types.h"
45
46 #include "ED_screen.h"
47 #include "ED_clip.h"  /* frame remapping functions */
48 #include "ED_mask.h"  /* own include */
49
50 #include "mask_intern.h"  /* own include */
51
52 static int mask_parent_clear_exec(bContext *C, wmOperator *UNUSED(op))
53 {
54         Mask *mask = CTX_data_edit_mask(C);
55         MaskLayer *masklay;
56
57         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
58                 MaskSpline *spline;
59                 int i;
60
61                 if (masklay->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
62                         continue;
63                 }
64
65                 for (spline = masklay->splines.first; spline; spline = spline->next) {
66                         for (i = 0; i < spline->tot_point; i++) {
67                                 MaskSplinePoint *point = &spline->points[i];
68
69                                 if (MASKPOINT_ISSEL_ANY(point)) {
70                                         point->parent.id = NULL;
71                                 }
72                         }
73                 }
74         }
75
76         WM_event_add_notifier(C, NC_MASK | ND_DATA, mask);
77         DAG_id_tag_update(&mask->id, 0);
78
79         return OPERATOR_FINISHED;
80 }
81
82 void MASK_OT_parent_clear(wmOperatorType *ot)
83 {
84         /* identifiers */
85         ot->name = "Clear Parent";
86         ot->description = "Clear the mask's parenting";
87         ot->idname = "MASK_OT_parent_clear";
88
89         /* api callbacks */
90         ot->exec = mask_parent_clear_exec;
91
92         ot->poll = ED_maskedit_mask_poll;
93
94         /* flags */
95         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
96 }
97
98 static int mask_parent_set_exec(bContext *C, wmOperator *UNUSED(op))
99 {
100         Mask *mask = CTX_data_edit_mask(C);
101         MaskLayer *masklay;
102
103         /* parent info */
104         SpaceClip *sc;
105         MovieClip *clip;
106         MovieTrackingTrack *track;
107         MovieTrackingMarker *marker;
108         MovieTrackingObject *tracking_object;
109         /* done */
110
111         int framenr;
112
113         float marker_pos_ofs[2];
114         float parmask_pos[2];
115
116         if ((NULL == (sc = CTX_wm_space_clip(C))) ||
117             (NULL == (clip = sc->clip)) ||
118             (NULL == (track = clip->tracking.act_track)) ||
119             (NULL == (tracking_object = BKE_tracking_object_get_active(&clip->tracking))))
120         {
121                 return OPERATOR_CANCELLED;
122         }
123
124         framenr = ED_space_clip_get_clip_frame_number(sc);
125         marker = BKE_tracking_marker_get(track, framenr);
126
127         add_v2_v2v2(marker_pos_ofs, marker->pos, track->offset);
128
129         BKE_mask_coord_from_movieclip(clip, &sc->user, parmask_pos, marker_pos_ofs);
130
131         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
132                 MaskSpline *spline;
133                 int i;
134
135                 if (masklay->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
136                         continue;
137                 }
138
139                 for (spline = masklay->splines.first; spline; spline = spline->next) {
140                         for (i = 0; i < spline->tot_point; i++) {
141                                 MaskSplinePoint *point = &spline->points[i];
142
143                                 if (MASKPOINT_ISSEL_ANY(point)) {
144                                         point->parent.id_type = ID_MC;
145                                         point->parent.id = &clip->id;
146                                         strcpy(point->parent.parent, tracking_object->name);
147                                         strcpy(point->parent.sub_parent, track->name);
148
149                                         copy_v2_v2(point->parent.parent_orig, parmask_pos);
150                                 }
151                         }
152                 }
153         }
154
155         WM_event_add_notifier(C, NC_MASK | ND_DATA, mask);
156         DAG_id_tag_update(&mask->id, 0);
157
158         return OPERATOR_FINISHED;
159 }
160
161 /** based on #OBJECT_OT_parent_set */
162 void MASK_OT_parent_set(wmOperatorType *ot)
163 {
164         /* identifiers */
165         ot->name = "Make Parent";
166         ot->description = "Set the mask's parenting";
167         ot->idname = "MASK_OT_parent_set";
168
169         /* api callbacks */
170         //ot->invoke = mask_parent_set_invoke;
171         ot->exec = mask_parent_set_exec;
172
173         ot->poll = ED_maskedit_mask_poll;
174
175         /* flags */
176         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
177 }