Internal refactoring of tracking module, should be no functional changes
[blender.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_mask.h"  /* own include */
48
49 #include "mask_intern.h"  /* own include */
50
51 static int mask_parent_clear_exec(bContext *C, wmOperator *UNUSED(op))
52 {
53         Mask *mask = CTX_data_edit_mask(C);
54         MaskLayer *masklay;
55
56         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
57                 MaskSpline *spline;
58                 int i;
59
60                 if (masklay->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
61                         continue;
62                 }
63
64                 for (spline = masklay->splines.first; spline; spline = spline->next) {
65                         for (i = 0; i < spline->tot_point; i++) {
66                                 MaskSplinePoint *point = &spline->points[i];
67
68                                 if (MASKPOINT_ISSEL_ANY(point)) {
69                                         point->parent.id = NULL;
70                                 }
71                         }
72                 }
73         }
74
75         WM_event_add_notifier(C, NC_MASK | ND_DATA, mask);
76         DAG_id_tag_update(&mask->id, 0);
77
78         return OPERATOR_FINISHED;
79 }
80
81 void MASK_OT_parent_clear(wmOperatorType *ot)
82 {
83         /* identifiers */
84         ot->name = "Clear Parent";
85         ot->description = "Clear the mask's parenting";
86         ot->idname = "MASK_OT_parent_clear";
87
88         /* api callbacks */
89         ot->exec = mask_parent_clear_exec;
90
91         ot->poll = ED_maskedit_mask_poll;
92
93         /* flags */
94         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
95 }
96
97 static int mask_parent_set_exec(bContext *C, wmOperator *UNUSED(op))
98 {
99         Mask *mask = CTX_data_edit_mask(C);
100         MaskLayer *masklay;
101
102         /* parent info */
103         SpaceClip *sc;
104         MovieClip *clip;
105         MovieTrackingTrack *track;
106         MovieTrackingMarker *marker;
107         MovieTrackingObject *tracking;
108         /* done */
109
110         float marker_pos_ofs[2];
111         float parmask_pos[2];
112
113         if ((NULL == (sc = CTX_wm_space_clip(C))) ||
114             (NULL == (clip = sc->clip)) ||
115             (NULL == (track = clip->tracking.act_track)) ||
116             (NULL == (marker = BKE_tracking_marker_get(track, sc->user.framenr))) ||
117             (NULL == (tracking = BKE_tracking_object_get_active(&clip->tracking))))
118         {
119                 return OPERATOR_CANCELLED;
120         }
121
122         add_v2_v2v2(marker_pos_ofs, marker->pos, track->offset);
123
124         BKE_mask_coord_from_movieclip(clip, &sc->user, parmask_pos, marker_pos_ofs);
125
126         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
127                 MaskSpline *spline;
128                 int i;
129
130                 if (masklay->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
131                         continue;
132                 }
133
134                 for (spline = masklay->splines.first; spline; spline = spline->next) {
135                         for (i = 0; i < spline->tot_point; i++) {
136                                 MaskSplinePoint *point = &spline->points[i];
137
138                                 if (MASKPOINT_ISSEL_ANY(point)) {
139                                         point->parent.id_type = ID_MC;
140                                         point->parent.id = &clip->id;
141                                         strcpy(point->parent.parent, tracking->name);
142                                         strcpy(point->parent.sub_parent, track->name);
143
144                                         copy_v2_v2(point->parent.parent_orig, parmask_pos);
145                                 }
146                         }
147                 }
148         }
149
150         WM_event_add_notifier(C, NC_MASK | ND_DATA, mask);
151         DAG_id_tag_update(&mask->id, 0);
152
153         return OPERATOR_FINISHED;
154 }
155
156 /** based on #OBJECT_OT_parent_set */
157 void MASK_OT_parent_set(wmOperatorType *ot)
158 {
159         /* identifiers */
160         ot->name = "Make Parent";
161         ot->description = "Set the mask's parenting";
162         ot->idname = "MASK_OT_parent_set";
163
164         /* api callbacks */
165         //ot->invoke = mask_parent_set_invoke;
166         ot->exec = mask_parent_set_exec;
167
168         ot->poll = ED_maskedit_mask_poll;
169
170         /* flags */
171         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
172 }