07ce9215e26503c2868085cce2897fabb98fea97
[blender-staging.git] / source / blender / editors / mask / mask_shapekey.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_shapekey.c
29  *  \ingroup edmask
30  */
31
32 #include "BLI_utildefines.h"
33
34 #include "BKE_context.h"
35 #include "BKE_depsgraph.h"
36 #include "BKE_mask.h"
37
38 #include "DNA_mask_types.h"
39 #include "DNA_scene_types.h"
40
41 #include "WM_api.h"
42 #include "WM_types.h"
43
44 #include "ED_mask.h"  /* own include */
45
46 #include "mask_intern.h"  /* own include */
47
48 static int mask_shape_key_insert_exec(bContext *C, wmOperator *UNUSED(op))
49 {
50         Scene *scene = CTX_data_scene(C);
51         const int frame = CFRA;
52         Mask *mask = CTX_data_edit_mask(C);
53         MaskLayer *masklay;
54         int change = FALSE;
55
56         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
57                 MaskLayerShape *masklay_shape;
58
59                 if (!ED_mask_layer_select_check(masklay)) {
60                         continue;
61                 }
62
63                 masklay_shape = BKE_mask_layer_shape_varify_frame(masklay, frame);
64                 BKE_mask_layer_shape_from_mask(masklay, masklay_shape);
65                 change = TRUE;
66         }
67
68         if (change) {
69                 WM_event_add_notifier(C, NC_MASK | ND_DATA, mask);
70                 DAG_id_tag_update(&mask->id, 0);
71
72                 return OPERATOR_FINISHED;
73         }
74         else {
75                 return OPERATOR_CANCELLED;
76         }
77 }
78
79 void MASK_OT_shape_key_insert(wmOperatorType *ot)
80 {
81         /* identifiers */
82         ot->name = "Insert Shape Key";
83         ot->description = "";
84         ot->idname = "MASK_OT_shape_key_insert";
85
86         /* api callbacks */
87         ot->exec = mask_shape_key_insert_exec;
88         ot->poll = ED_maskediting_mask_poll;
89
90         /* flags */
91         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
92 }
93
94 static int mask_shape_key_clear_exec(bContext *C, wmOperator *UNUSED(op))
95 {
96         Scene *scene = CTX_data_scene(C);
97         const int frame = CFRA;
98         Mask *mask = CTX_data_edit_mask(C);
99         MaskLayer *masklay;
100         int change = FALSE;
101
102         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
103                 MaskLayerShape *masklay_shape;
104
105                 if (!ED_mask_layer_select_check(masklay)) {
106                         continue;
107                 }
108
109                 masklay_shape = BKE_mask_layer_shape_find_frame(masklay, frame);
110
111                 if (masklay_shape) {
112                         BKE_mask_layer_shape_unlink(masklay, masklay_shape);
113                         change = TRUE;
114                 }
115         }
116
117         if (change) {
118                 WM_event_add_notifier(C, NC_MASK | ND_DATA, mask);
119                 DAG_id_tag_update(&mask->id, 0);
120
121                 return OPERATOR_FINISHED;
122         }
123         else {
124                 return OPERATOR_CANCELLED;
125         }
126 }
127
128 void MASK_OT_shape_key_clear(wmOperatorType *ot)
129 {
130         /* identifiers */
131         ot->name = "Clear Shape Key";
132         ot->description = "";
133         ot->idname = "MASK_OT_shape_key_clear";
134
135         /* api callbacks */
136         ot->exec = mask_shape_key_clear_exec;
137         ot->poll = ED_maskediting_mask_poll;
138
139         /* flags */
140         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
141 }
142
143 int ED_mask_layer_shape_auto_key_all(Mask *mask, const int frame)
144 {
145         MaskLayer *masklay;
146         int change = FALSE;
147
148         for (masklay = mask->masklayers.first; masklay; masklay = masklay->next) {
149                 MaskLayerShape *masklay_shape;
150
151                 masklay_shape = BKE_mask_layer_shape_varify_frame(masklay, frame);
152                 BKE_mask_layer_shape_from_mask(masklay, masklay_shape);
153                 change = TRUE;
154         }
155
156         return change;
157 }