Merge branch 'blender2.7'
[blender.git] / source / blender / gpencil_modifiers / intern / MOD_gpencilarmature.c
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software  Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  *
16  * The Original Code is Copyright (C) 2018, Blender Foundation
17  * This is a new part of Blender
18  */
19
20 /** \file
21  * \ingroup modifiers
22  */
23
24 #include <stdio.h>
25
26 #include "BLI_utildefines.h"
27
28 #include "BLI_math.h"
29
30 #include "DNA_armature_types.h"
31 #include "DNA_meshdata_types.h"
32 #include "DNA_scene_types.h"
33 #include "DNA_object_types.h"
34 #include "DNA_gpencil_types.h"
35 #include "DNA_gpencil_modifier_types.h"
36 #include "DNA_modifier_types.h"
37
38 #include "BKE_lattice.h"
39 #include "BKE_gpencil.h"
40 #include "BKE_gpencil_modifier.h"
41 #include "BKE_modifier.h"
42 #include "BKE_library_query.h"
43 #include "BKE_scene.h"
44 #include "BKE_main.h"
45
46 #include "MEM_guardedalloc.h"
47
48 #include "MOD_gpencil_util.h"
49 #include "MOD_gpencil_modifiertypes.h"
50
51 #include "DEG_depsgraph.h"
52 #include "DEG_depsgraph_build.h"
53 #include "DEG_depsgraph_query.h"
54
55 static void initData(GpencilModifierData *md)
56 {
57         ArmatureGpencilModifierData *gpmd = (ArmatureGpencilModifierData *)md;
58         gpmd->object = NULL;
59         gpmd->deformflag = ARM_DEF_VGROUP;
60 }
61
62 static void copyData(const GpencilModifierData *md, GpencilModifierData *target)
63 {
64         BKE_gpencil_modifier_copyData_generic(md, target);
65 }
66
67 static void gpencil_deform_verts(
68         ArmatureGpencilModifierData *mmd, Object *target,
69         bGPDstroke *gps)
70 {
71         bGPDspoint *pt = gps->points;
72         float *all_vert_coords = MEM_callocN(sizeof(float) * 3 * gps->totpoints, __func__);
73         int i;
74
75         BKE_gpencil_dvert_ensure(gps);
76
77         /* prepare array of points */
78         for (i = 0; i < gps->totpoints; i++, pt++) {
79                 float *pt_coords = &all_vert_coords[3 * i];
80                 float co[3];
81                 copy_v3_v3(co, &pt->x);
82                 copy_v3_v3(pt_coords, co);
83         }
84
85         /* deform verts */
86         armature_deform_verts(
87                 mmd->object, target, NULL,
88                 (float(*)[3])all_vert_coords,
89                 NULL, gps->totpoints,
90                 mmd->deformflag,
91                 (float(*)[3])mmd->prevCos,
92                 mmd->vgname, gps);
93
94         /* Apply deformed coordinates */
95         pt = gps->points;
96         for (i = 0; i < gps->totpoints; i++, pt++) {
97                 float *pt_coords = &all_vert_coords[3 * i];
98                 copy_v3_v3(&pt->x, pt_coords);
99         }
100
101         MEM_SAFE_FREE(all_vert_coords);
102
103 }
104
105 /* deform stroke */
106 static void deformStroke(
107         GpencilModifierData *md, Depsgraph *UNUSED(depsgraph),
108         Object *ob, bGPDlayer *UNUSED(gpl), bGPDstroke *gps)
109 {
110         ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
111         if (!mmd->object) {
112                 return;
113         }
114
115         gpencil_deform_verts(mmd, ob, gps);
116 }
117
118 static void bakeModifier(
119         Main *bmain, Depsgraph *depsgraph,
120         GpencilModifierData *md, Object *ob)
121 {
122         ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
123         Scene *scene = DEG_get_evaluated_scene(depsgraph);
124         bGPdata *gpd = ob->data;
125         int oldframe = (int)DEG_get_ctime(depsgraph);
126
127         if (mmd->object == NULL)
128                 return;
129
130         for (bGPDlayer *gpl = gpd->layers.first; gpl; gpl = gpl->next) {
131                 for (bGPDframe *gpf = gpl->frames.first; gpf; gpf = gpf->next) {
132                         /* apply armature effects on this frame
133                          * NOTE: this assumes that we don't want armature animation on non-keyframed frames
134                          */
135                         CFRA = gpf->framenum;
136                         BKE_scene_graph_update_for_newframe(depsgraph, bmain);
137
138                         /* compute armature effects on this frame */
139                         for (bGPDstroke *gps = gpf->strokes.first; gps; gps = gps->next) {
140                                 deformStroke(md, depsgraph, ob, gpl, gps);
141                         }
142                 }
143         }
144
145         /* return frame state and DB to original state */
146         CFRA = oldframe;
147         BKE_scene_graph_update_for_newframe(depsgraph, bmain);
148 }
149
150 static bool isDisabled(GpencilModifierData *md, int UNUSED(userRenderParams))
151 {
152         ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
153
154         return !mmd->object;
155 }
156
157 static void updateDepsgraph(GpencilModifierData *md, const ModifierUpdateDepsgraphContext *ctx)
158 {
159         ArmatureGpencilModifierData *lmd = (ArmatureGpencilModifierData *)md;
160         if (lmd->object != NULL) {
161                 DEG_add_object_relation(ctx->node, lmd->object, DEG_OB_COMP_EVAL_POSE, "Armature Modifier");
162                 DEG_add_object_relation(ctx->node, lmd->object, DEG_OB_COMP_TRANSFORM, "Armature Modifier");
163         }
164         DEG_add_object_relation(ctx->node, ctx->object, DEG_OB_COMP_TRANSFORM, "Armature Modifier");
165 }
166
167 static void foreachObjectLink(
168         GpencilModifierData *md, Object *ob,
169         ObjectWalkFunc walk, void *userData)
170 {
171         ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
172
173         walk(userData, ob, &mmd->object, IDWALK_CB_NOP);
174 }
175
176 GpencilModifierTypeInfo modifierType_Gpencil_Armature = {
177         /* name */              "Armature",
178         /* structName */        "ArmatureGpencilModifierData",
179         /* structSize */        sizeof(ArmatureGpencilModifierData),
180         /* type */              eGpencilModifierTypeType_Gpencil,
181         /* flags */             eGpencilModifierTypeFlag_SupportsEditmode,
182
183         /* copyData */          copyData,
184
185         /* deformStroke */      deformStroke,
186         /* generateStrokes */   NULL,
187         /* bakeModifier */      bakeModifier,
188         /* remapTime */         NULL,
189         /* initData */          initData,
190         /* freeData */          NULL,
191         /* isDisabled */        isDisabled,
192         /* updateDepsgraph */   updateDepsgraph,
193         /* dependsOnTime */     NULL,
194         /* foreachObjectLink */ foreachObjectLink,
195         /* foreachIDLink */     NULL,
196         /* foreachTexLink */    NULL,
197         /* getDuplicationFactor */ NULL,
198 };