Cycles / Sky Texture:
[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 #include "BLI_string.h"
35
36 #include "BKE_context.h"
37 #include "BKE_depsgraph.h"
38 #include "BKE_mask.h"
39 #include "BKE_tracking.h"
40
41 #include "DNA_mask_types.h"
42 #include "DNA_object_types.h"  /* SELECT */
43
44 #include "WM_api.h"
45 #include "WM_types.h"
46
47 #include "ED_screen.h"
48 #include "ED_clip.h"  /* frame remapping functions */
49 #include "ED_mask.h"  /* own include */
50
51 #include "mask_intern.h"  /* own include */
52
53 static int mask_parent_clear_exec(bContext *C, wmOperator *UNUSED(op))
54 {
55         Mask *mask = CTX_data_edit_mask(C);
56         MaskLayer *masklay;
57
58         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
59                 MaskSpline *spline;
60                 int i;
61
62                 if (masklay->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
63                         continue;
64                 }
65
66                 for (spline = masklay->splines.first; spline; spline = spline->next) {
67                         for (i = 0; i < spline->tot_point; i++) {
68                                 MaskSplinePoint *point = &spline->points[i];
69
70                                 if (MASKPOINT_ISSEL_ANY(point)) {
71                                         point->parent.id = NULL;
72                                 }
73                         }
74                 }
75         }
76
77         WM_event_add_notifier(C, NC_MASK | ND_DATA, mask);
78         DAG_id_tag_update(&mask->id, 0);
79
80         return OPERATOR_FINISHED;
81 }
82
83 void MASK_OT_parent_clear(wmOperatorType *ot)
84 {
85         /* identifiers */
86         ot->name = "Clear Parent";
87         ot->description = "Clear the mask's parenting";
88         ot->idname = "MASK_OT_parent_clear";
89
90         /* api callbacks */
91         ot->exec = mask_parent_clear_exec;
92
93         ot->poll = ED_maskedit_mask_poll;
94
95         /* flags */
96         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
97 }
98
99 static int mask_parent_set_exec(bContext *C, wmOperator *UNUSED(op))
100 {
101         Mask *mask = CTX_data_edit_mask(C);
102         MaskLayer *masklay;
103
104         /* parent info */
105         SpaceClip *sc = CTX_wm_space_clip(C);
106         MovieClip *clip = ED_space_clip_get_clip(sc);
107         MovieTracking *tracking;
108         MovieTrackingTrack *track;
109         MovieTrackingPlaneTrack *plane_track;
110         MovieTrackingObject *tracking_object;
111         /* done */
112
113         int framenr, parent_type;
114         float parmask_pos[2], orig_corners[4][2];
115         char *sub_parent_name;
116
117         if (ELEM(NULL, sc, clip)) {
118                 return OPERATOR_CANCELLED;
119         }
120
121         framenr = ED_space_clip_get_clip_frame_number(sc);
122
123         tracking = &clip->tracking;
124         tracking_object = BKE_tracking_object_get_active(&clip->tracking);
125
126         if (tracking_object == NULL) {
127                 return OPERATOR_CANCELLED;
128         }
129
130         if ((track = BKE_tracking_track_get_active(tracking)) != NULL) {
131                 MovieTrackingMarker *marker = BKE_tracking_marker_get(track, framenr);
132                 float marker_pos_ofs[2];
133
134                 add_v2_v2v2(marker_pos_ofs, marker->pos, track->offset);
135
136                 BKE_mask_coord_from_movieclip(clip, &sc->user, parmask_pos, marker_pos_ofs);
137
138                 sub_parent_name = track->name;
139                 parent_type = MASK_PARENT_POINT_TRACK;
140                 memset(orig_corners, 0, sizeof(orig_corners));
141         }
142         else if ((plane_track = BKE_tracking_plane_track_get_active(tracking)) != NULL) {
143                 MovieTrackingPlaneMarker *plane_marker = BKE_tracking_plane_marker_get(plane_track, framenr);
144
145                 zero_v2(parmask_pos);
146                 sub_parent_name = plane_track->name;
147                 parent_type = MASK_PARENT_PLANE_TRACK;
148                 memcpy(orig_corners, plane_marker->corners, sizeof(orig_corners));
149         }
150         else {
151                 return OPERATOR_CANCELLED;
152         }
153
154         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
155                 MaskSpline *spline;
156                 int i;
157
158                 if (masklay->restrictflag & (MASK_RESTRICT_VIEW | MASK_RESTRICT_SELECT)) {
159                         continue;
160                 }
161
162                 for (spline = masklay->splines.first; spline; spline = spline->next) {
163                         for (i = 0; i < spline->tot_point; i++) {
164                                 MaskSplinePoint *point = &spline->points[i];
165
166                                 if (MASKPOINT_ISSEL_ANY(point)) {
167                                         point->parent.id_type = ID_MC;
168                                         point->parent.id = &clip->id;
169                                         point->parent.type = parent_type;
170                                         BLI_strncpy(point->parent.parent, tracking_object->name, sizeof(point->parent.parent));
171                                         BLI_strncpy(point->parent.sub_parent, sub_parent_name, sizeof(point->parent.sub_parent));
172
173                                         copy_v2_v2(point->parent.parent_orig, parmask_pos);
174                                         memcpy(point->parent.parent_corners_orig, orig_corners, sizeof(point->parent.parent_corners_orig));
175                                 }
176                         }
177                 }
178         }
179
180         WM_event_add_notifier(C, NC_MASK | ND_DATA, mask);
181         DAG_id_tag_update(&mask->id, 0);
182
183         return OPERATOR_FINISHED;
184 }
185
186 /** based on #OBJECT_OT_parent_set */
187 void MASK_OT_parent_set(wmOperatorType *ot)
188 {
189         /* identifiers */
190         ot->name = "Make Parent";
191         ot->description = "Set the mask's parenting";
192         ot->idname = "MASK_OT_parent_set";
193
194         /* api callbacks */
195         //ot->invoke = mask_parent_set_invoke;
196         ot->exec = mask_parent_set_exec;
197
198         ot->poll = ED_maskedit_mask_poll;
199
200         /* flags */
201         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
202 }