Cleanup: style, use braces for gpencil modifiers
[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(ArmatureGpencilModifierData *mmd, Object *target, bGPDstroke *gps)
68 {
69   bGPDspoint *pt = gps->points;
70   float *all_vert_coords = MEM_callocN(sizeof(float) * 3 * gps->totpoints, __func__);
71   int i;
72
73   BKE_gpencil_dvert_ensure(gps);
74
75   /* prepare array of points */
76   for (i = 0; i < gps->totpoints; i++, pt++) {
77     float *pt_coords = &all_vert_coords[3 * i];
78     float co[3];
79     copy_v3_v3(co, &pt->x);
80     copy_v3_v3(pt_coords, co);
81   }
82
83   /* deform verts */
84   armature_deform_verts(mmd->object,
85                         target,
86                         NULL,
87                         (float(*)[3])all_vert_coords,
88                         NULL,
89                         gps->totpoints,
90                         mmd->deformflag,
91                         (float(*)[3])mmd->prevCos,
92                         mmd->vgname,
93                         gps);
94
95   /* Apply deformed coordinates */
96   pt = gps->points;
97   for (i = 0; i < gps->totpoints; i++, pt++) {
98     float *pt_coords = &all_vert_coords[3 * i];
99     copy_v3_v3(&pt->x, pt_coords);
100   }
101
102   MEM_SAFE_FREE(all_vert_coords);
103 }
104
105 /* deform stroke */
106 static void deformStroke(GpencilModifierData *md,
107                          Depsgraph *UNUSED(depsgraph),
108                          Object *ob,
109                          bGPDlayer *UNUSED(gpl),
110                          bGPDstroke *gps)
111 {
112   ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
113   if (!mmd->object) {
114     return;
115   }
116
117   gpencil_deform_verts(mmd, ob, gps);
118 }
119
120 static void bakeModifier(Main *bmain, Depsgraph *depsgraph, GpencilModifierData *md, Object *ob)
121 {
122   Scene *scene = DEG_get_evaluated_scene(depsgraph);
123   Object *object_eval = DEG_get_evaluated_object(depsgraph, ob);
124   ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
125   GpencilModifierData *md_eval = BKE_gpencil_modifiers_findByName(object_eval, md->name);
126   bGPdata *gpd = (bGPdata *)ob->data;
127   int oldframe = (int)DEG_get_ctime(depsgraph);
128
129   if (mmd->object == NULL) {
130     return;
131   }
132
133   for (bGPDlayer *gpl = gpd->layers.first; gpl; gpl = gpl->next) {
134     for (bGPDframe *gpf = gpl->frames.first; gpf; gpf = gpf->next) {
135       /* apply armature effects on this frame
136        * NOTE: this assumes that we don't want armature animation on non-keyframed frames
137        */
138       CFRA = gpf->framenum;
139       BKE_scene_graph_update_for_newframe(depsgraph, bmain);
140
141       /* compute armature effects on this frame */
142       for (bGPDstroke *gps = gpf->strokes.first; gps; gps = gps->next) {
143         deformStroke(md_eval, depsgraph, object_eval, gpl, gps);
144       }
145     }
146   }
147
148   /* return frame state and DB to original state */
149   CFRA = oldframe;
150   BKE_scene_graph_update_for_newframe(depsgraph, bmain);
151 }
152
153 static bool isDisabled(GpencilModifierData *md, int UNUSED(userRenderParams))
154 {
155   ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
156
157   return !mmd->object;
158 }
159
160 static void updateDepsgraph(GpencilModifierData *md, const ModifierUpdateDepsgraphContext *ctx)
161 {
162   ArmatureGpencilModifierData *lmd = (ArmatureGpencilModifierData *)md;
163   if (lmd->object != NULL) {
164     DEG_add_object_relation(ctx->node, lmd->object, DEG_OB_COMP_EVAL_POSE, "Armature Modifier");
165     DEG_add_object_relation(ctx->node, lmd->object, DEG_OB_COMP_TRANSFORM, "Armature Modifier");
166   }
167   DEG_add_object_relation(ctx->node, ctx->object, DEG_OB_COMP_TRANSFORM, "Armature Modifier");
168 }
169
170 static void foreachObjectLink(GpencilModifierData *md,
171                               Object *ob,
172                               ObjectWalkFunc walk,
173                               void *userData)
174 {
175   ArmatureGpencilModifierData *mmd = (ArmatureGpencilModifierData *)md;
176
177   walk(userData, ob, &mmd->object, IDWALK_CB_NOP);
178 }
179
180 GpencilModifierTypeInfo modifierType_Gpencil_Armature = {
181     /* name */ "Armature",
182     /* structName */ "ArmatureGpencilModifierData",
183     /* structSize */ sizeof(ArmatureGpencilModifierData),
184     /* type */ eGpencilModifierTypeType_Gpencil,
185     /* flags */ eGpencilModifierTypeFlag_SupportsEditmode,
186
187     /* copyData */ copyData,
188
189     /* deformStroke */ deformStroke,
190     /* generateStrokes */ NULL,
191     /* bakeModifier */ bakeModifier,
192     /* remapTime */ NULL,
193     /* initData */ initData,
194     /* freeData */ NULL,
195     /* isDisabled */ isDisabled,
196     /* updateDepsgraph */ updateDepsgraph,
197     /* dependsOnTime */ NULL,
198     /* foreachObjectLink */ foreachObjectLink,
199     /* foreachIDLink */ NULL,
200     /* foreachTexLink */ NULL,
201     /* getDuplicationFactor */ NULL,
202 };