Extract common modifier parameters into ModifierEvalContext struct
[blender.git] / source / blender / modifiers / intern / MOD_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) 2005 by the Blender Foundation.
19  * All rights reserved.
20  *
21  * Contributor(s): Daniel Dunbar
22  *                 Ton Roosendaal,
23  *                 Ben Batt,
24  *                 Brecht Van Lommel,
25  *                 Campbell Barton
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  *
29  */
30
31 /** \file blender/modifiers/intern/MOD_shapekey.c
32  *  \ingroup modifiers
33  */
34
35 #include "BLI_math.h"
36
37 #include "DNA_key_types.h"
38
39 #include "BLI_utildefines.h"
40
41 #include "BKE_cdderivedmesh.h"
42 #include "BKE_key.h"
43 #include "BKE_particle.h"
44
45 #include "MOD_modifiertypes.h"
46
47 static void deformVerts(ModifierData *UNUSED(md), const ModifierEvalContext *ctx,
48                         DerivedMesh *UNUSED(derivedData),
49                         float (*vertexCos)[3],
50                         int numVerts)
51 {
52         Key *key = BKE_key_from_object(ctx->object);
53
54         if (key && key->block.first) {
55                 int deformedVerts_tot;
56                 BKE_key_evaluate_object_ex(
57                             ctx->object, &deformedVerts_tot,
58                             (float *)vertexCos, sizeof(*vertexCos) * numVerts);
59
60         }
61 }
62
63 static void deformMatrices(ModifierData *md, const ModifierEvalContext *ctx, DerivedMesh *derivedData,
64                            float (*vertexCos)[3], float (*defMats)[3][3], int numVerts)
65 {
66         Key *key = BKE_key_from_object(ctx->object);
67         KeyBlock *kb = BKE_keyblock_from_object(ctx->object);
68         float scale[3][3];
69
70         (void)vertexCos; /* unused */
71
72         if (kb && kb->totelem == numVerts && kb != key->refkey) {
73                 int a;
74
75                 if (ctx->object->shapeflag & OB_SHAPE_LOCK) scale_m3_fl(scale, 1);
76                 else scale_m3_fl(scale, kb->curval);
77
78                 for (a = 0; a < numVerts; a++)
79                         copy_m3_m3(defMats[a], scale);
80         }
81
82         deformVerts(md, ctx, derivedData, vertexCos, numVerts);
83 }
84
85 static void deformVertsEM(ModifierData *md, const ModifierEvalContext *ctx,
86                           struct BMEditMesh *UNUSED(editData),
87                           DerivedMesh *derivedData,
88                           float (*vertexCos)[3],
89                           int numVerts)
90 {
91         Key *key = BKE_key_from_object(ctx->object);
92
93         if (key && key->type == KEY_RELATIVE)
94                 deformVerts(md, ctx, derivedData, vertexCos, numVerts);
95 }
96
97 static void deformMatricesEM(ModifierData *UNUSED(md), const ModifierEvalContext *ctx,
98                              struct BMEditMesh *UNUSED(editData),
99                              DerivedMesh *UNUSED(derivedData),
100                              float (*vertexCos)[3],
101                              float (*defMats)[3][3],
102                              int numVerts)
103 {
104         Key *key = BKE_key_from_object(ctx->object);
105         KeyBlock *kb = BKE_keyblock_from_object(ctx->object);
106         float scale[3][3];
107
108         (void)vertexCos; /* unused */
109
110         if (kb && kb->totelem == numVerts && kb != key->refkey) {
111                 int a;
112                 scale_m3_fl(scale, kb->curval);
113
114                 for (a = 0; a < numVerts; a++)
115                         copy_m3_m3(defMats[a], scale);
116         }
117 }
118
119
120 ModifierTypeInfo modifierType_ShapeKey = {
121         /* name */              "ShapeKey",
122         /* structName */        "ShapeKeyModifierData",
123         /* structSize */        sizeof(ShapeKeyModifierData),
124         /* type */              eModifierTypeType_OnlyDeform,
125         /* flags */             eModifierTypeFlag_AcceptsCVs |
126                                 eModifierTypeFlag_AcceptsLattice |
127                                 eModifierTypeFlag_SupportsEditmode,
128
129         /* copyData */          NULL,
130
131         /* deformVerts_DM */    deformVerts,
132         /* deformMatrices_DM */ deformMatrices,
133         /* deformVertsEM_DM */  deformVertsEM,
134         /* deformMatricesEM_DM*/deformMatricesEM,
135         /* applyModifier_DM */  NULL,
136         /* applyModifierEM_DM */NULL,
137
138         /* deformVerts */       NULL,
139         /* deformMatrices */    NULL,
140         /* deformVertsEM */     NULL,
141         /* deformMatricesEM */  NULL,
142         /* applyModifier */     NULL,
143         /* applyModifierEM */   NULL,
144
145         /* initData */          NULL,
146         /* requiredDataMask */  NULL,
147         /* freeData */          NULL,
148         /* isDisabled */        NULL,
149         /* updateDepsgraph */   NULL,
150         /* dependsOnTime */     NULL,
151         /* dependsOnNormals */  NULL,
152         /* foreachObjectLink */ NULL,
153         /* foreachIDLink */     NULL,
154         /* foreachTexLink */    NULL,
155 };