Merge branch 'master' into blender2.8
[blender.git] / source / blender / gpencil_modifiers / intern / MOD_gpencilarmature.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) 2018, Blender Foundation
19  * This is a new part of Blender
20  *
21  * Contributor(s): Antonio Vazquez
22  *
23  * ***** END GPL LICENSE BLOCK *****
24  *
25  */
26
27 /** \file blender/gpencil_modifiers/intern/MOD_gpencilarmature.c
28  *  \ingroup modifiers
29  */
30
31 #include <stdio.h>
32
33 #include "DNA_armature_types.h"
34 #include "DNA_meshdata_types.h"
35 #include "DNA_scene_types.h"
36 #include "DNA_object_types.h"
37 #include "DNA_gpencil_types.h"
38 #include "DNA_gpencil_modifier_types.h"
39 #include "DNA_modifier_types.h"
40 #include "BLI_math.h"
41
42 #include "BLI_listbase.h"
43 #include "BLI_task.h"
44 #include "BLI_utildefines.h"
45
46 #include "BKE_lattice.h"
47 #include "BKE_context.h"
48 #include "BKE_gpencil.h"
49 #include "BKE_gpencil_modifier.h"
50 #include "BKE_modifier.h"
51 #include "BKE_library_query.h"
52 #include "BKE_scene.h"
53 #include "BKE_main.h"
54
55 #include "MEM_guardedalloc.h"
56
57 #include "MOD_gpencil_util.h"
58 #include "MOD_gpencil_modifiertypes.h"
59
60 #include "DEG_depsgraph.h"
61 #include "DEG_depsgraph_build.h"
62 #include "DEG_depsgraph_query.h"
63
64 static void initData(GpencilModifierData *md)
65 {
66         ArmatureGpencilModifierData *gpmd = (ArmatureGpencilModifierData *)md;
67         gpmd->object = NULL;
68         gpmd->deformflag = ARM_DEF_VGROUP;
69 }
70
71 static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
72 {
73         BKE_gpencil_modifier_copyData_generic(md, target);
74 }
75
76 static void gpencil_deform_verts(
77         ArmatureGpencilModifierData *mmd, Object *target,
78         bGPDstroke *gps)
79 {
80         bGPDspoint *pt = gps->points;
81         float *all_vert_coords = MEM_callocN(sizeof(float) * 3 * gps->totpoints, __func__);
82         int i;
83
84         BKE_gpencil_dvert_ensure(gps);
85
86         /* prepare array of points */
87         for (i = 0; i < gps->totpoints; i++, pt++) {
88                 float *pt_coords = &all_vert_coords[3 * i];
89                 float co[3];
90                 copy_v3_v3(co, &pt->x);
91                 copy_v3_v3(pt_coords, co);
92         }
93
94         /* deform verts */
95         armature_deform_verts(
96                 mmd->object, target, NULL,
97                 (float(*)[3])all_vert_coords,
98                 NULL, gps->totpoints,
99                 mmd->deformflag,
100                 (float(*)[3])mmd->prevCos,
101                 mmd->vgname, gps);
102
103         /* Apply deformed coordinates */
104         pt = gps->points;
105         for (i = 0; i < gps->totpoints; i++, pt++) {
106                 float *pt_coords = &all_vert_coords[3 * i];
107                 copy_v3_v3(&pt->x, pt_coords);
108         }
109
110         MEM_SAFE_FREE(all_vert_coords);
111
112 }
113
114 /* deform stroke */
115 static void deformStroke(
116         GpencilModifierData *md, Depsgraph *UNUSED(depsgraph),
117         Object *ob, bGPDlayer *UNUSED(gpl), bGPDstroke *gps)
118 {
119         ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
120         if (!mmd->object) {
121                 return;
122         }
123
124         gpencil_deform_verts(mmd, ob, gps);
125 }
126
127 static void bakeModifier(
128         Main *bmain, Depsgraph *depsgraph,
129         GpencilModifierData *md, Object *ob)
130 {
131         ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
132         Scene *scene = DEG_get_evaluated_scene(depsgraph);
133         bGPdata *gpd = ob->data;
134         int oldframe = (int)DEG_get_ctime(depsgraph);
135
136         if (mmd->object == NULL)
137                 return;
138
139         for (bGPDlayer *gpl = gpd->layers.first; gpl; gpl = gpl->next) {
140                 for (bGPDframe *gpf = gpl->frames.first; gpf; gpf = gpf->next) {
141                         /* apply armature effects on this frame
142                          * NOTE: this assumes that we don't want armature animation on non-keyframed frames
143                          */
144                         CFRA = gpf->framenum;
145                         BKE_scene_graph_update_for_newframe(depsgraph, bmain);
146
147                         /* compute armature effects on this frame */
148                         for (bGPDstroke *gps = gpf->strokes.first; gps; gps = gps->next) {
149                                 deformStroke(md, depsgraph, ob, gpl, gps);
150                         }
151                 }
152         }
153
154         /* return frame state and DB to original state */
155         CFRA = oldframe;
156         BKE_scene_graph_update_for_newframe(depsgraph, bmain);
157 }
158
159 static bool isDisabled(GpencilModifierData *md, int UNUSED(userRenderParams))
160 {
161         ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
162
163         return !mmd->object;
164 }
165
166 static void updateDepsgraph(GpencilModifierData *md, const ModifierUpdateDepsgraphContext *ctx)
167 {
168         ArmatureGpencilModifierData *lmd = (ArmatureGpencilModifierData *)md;
169         if (lmd->object != NULL) {
170                 DEG_add_object_relation(ctx->node, lmd->object, DEG_OB_COMP_EVAL_POSE, "Armature Modifier");
171                 DEG_add_object_relation(ctx->node, lmd->object, DEG_OB_COMP_TRANSFORM, "Armature Modifier");
172         }
173         DEG_add_object_relation(ctx->node, ctx->object, DEG_OB_COMP_TRANSFORM, "Armature Modifier");
174 }
175
176 static void foreachObjectLink(
177         GpencilModifierData *md, Object *ob,
178         ObjectWalkFunc walk, void *userData)
179 {
180         ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
181
182         walk(userData, ob, &mmd->object, IDWALK_CB_NOP);
183 }
184
185 GpencilModifierTypeInfo modifierType_Gpencil_Armature = {
186         /* name */              "Armature",
187         /* structName */        "ArmatureGpencilModifierData",
188         /* structSize */        sizeof(ArmatureGpencilModifierData),
189         /* type */              eGpencilModifierTypeType_Gpencil,
190         /* flags */             0,
191
192         /* copyData */          copyData,
193
194         /* deformStroke */      deformStroke,
195         /* generateStrokes */   NULL,
196         /* bakeModifier */      bakeModifier,
197         /* remapTime */         NULL,
198         /* initData */          initData,
199         /* freeData */          NULL,
200         /* isDisabled */        isDisabled,
201         /* updateDepsgraph */   updateDepsgraph,
202         /* dependsOnTime */     NULL,
203         /* foreachObjectLink */ foreachObjectLink,
204         /* foreachIDLink */     NULL,
205         /* foreachTexLink */    NULL,
206         /* getDuplicationFactor */ NULL,
207 };