9bc96e593fd000cdff487468490146bb18420c8d
[blender-staging.git] / source / blender / modifiers / intern / MOD_curve.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_curve.c
32  *  \ingroup modifiers
33  */
34
35 #include <string.h>
36
37 #include "DNA_scene_types.h"
38 #include "DNA_object_types.h"
39
40 #include "BLI_utildefines.h"
41
42
43 #include "BKE_cdderivedmesh.h"
44 #include "BKE_lattice.h"
45 #include "BKE_library_query.h"
46 #include "BKE_modifier.h"
47
48 #include "depsgraph_private.h"
49 #include "DEG_depsgraph_build.h"
50
51
52 static void initData(ModifierData *md)
53 {
54         CurveModifierData *cmd = (CurveModifierData *) md;
55
56         cmd->defaxis = MOD_CURVE_POSX;
57 }
58
59 static void copyData(ModifierData *md, ModifierData *target)
60 {
61 #if 0
62         CurveModifierData *cmd = (CurveModifierData *) md;
63         CurveModifierData *tcmd = (CurveModifierData *) target;
64 #endif
65         modifier_copyData_generic(md, target);
66 }
67
68 static CustomDataMask requiredDataMask(Object *UNUSED(ob), ModifierData *md)
69 {
70         CurveModifierData *cmd = (CurveModifierData *)md;
71         CustomDataMask dataMask = 0;
72
73         /* ask for vertexgroups if we need them */
74         if (cmd->name[0]) dataMask |= CD_MASK_MDEFORMVERT;
75
76         return dataMask;
77 }
78
79 static bool isDisabled(ModifierData *md, int UNUSED(userRenderParams))
80 {
81         CurveModifierData *cmd = (CurveModifierData *) md;
82
83         return !cmd->object;
84 }
85
86 static void foreachObjectLink(
87         ModifierData *md, Object *ob,
88         ObjectWalkFunc walk, void *userData)
89 {
90         CurveModifierData *cmd = (CurveModifierData *) md;
91
92         walk(userData, ob, &cmd->object, IDWALK_NOP);
93 }
94
95 static void updateDepgraph(ModifierData *md, DagForest *forest,
96                            struct Main *UNUSED(bmain),
97                            Scene *UNUSED(scene),
98                            Object *UNUSED(ob),
99                            DagNode *obNode)
100 {
101         CurveModifierData *cmd = (CurveModifierData *) md;
102
103         if (cmd->object) {
104                 DagNode *curNode = dag_get_node(forest, cmd->object);
105                 curNode->eval_flags |= DAG_EVAL_NEED_CURVE_PATH;
106
107                 dag_add_relation(forest, curNode, obNode,
108                                  DAG_RL_DATA_DATA | DAG_RL_OB_DATA, "Curve Modifier");
109         }
110 }
111
112 static void updateDepsgraph(ModifierData *md,
113                             struct Main *UNUSED(bmain),
114                             struct Scene *scene,
115                             Object *object,
116                             struct DepsNodeHandle *node)
117 {
118         CurveModifierData *cmd = (CurveModifierData *)md;
119         if (cmd->object != NULL) {
120                 /* TODO(sergey): Need to do the same eval_flags trick for path
121                  * as happening in legacy depsgraph callback.
122                  */
123                 /* TODO(sergey): Currently path is evaluated as a part of modifier stack,
124                  * might be changed in the future.
125                  */
126                 DEG_add_object_relation(node, cmd->object, DEG_OB_COMP_GEOMETRY, "Curve Modifier");
127                 DEG_add_special_eval_flag(scene->depsgraph, &cmd->object->id, DAG_EVAL_NEED_CURVE_PATH);
128         }
129
130         DEG_add_object_relation(node, object, DEG_OB_COMP_TRANSFORM, "Curve Modifier");
131 }
132
133 static void deformVerts(ModifierData *md, Object *ob,
134                         DerivedMesh *derivedData,
135                         float (*vertexCos)[3],
136                         int numVerts,
137                         ModifierApplyFlag UNUSED(flag))
138 {
139         CurveModifierData *cmd = (CurveModifierData *) md;
140
141         /* silly that defaxis and curve_deform_verts are off by 1
142          * but leave for now to save having to call do_versions */
143         curve_deform_verts(md->scene, cmd->object, ob, derivedData, vertexCos, numVerts,
144                            cmd->name, cmd->defaxis - 1);
145 }
146
147 static void deformVertsEM(
148         ModifierData *md, Object *ob, struct BMEditMesh *em,
149         DerivedMesh *derivedData, float (*vertexCos)[3], int numVerts)
150 {
151         DerivedMesh *dm = derivedData;
152
153         if (!derivedData) dm = CDDM_from_editbmesh(em, false, false);
154
155         deformVerts(md, ob, dm, vertexCos, numVerts, 0);
156
157         if (!derivedData) dm->release(dm);
158 }
159
160
161 ModifierTypeInfo modifierType_Curve = {
162         /* name */              "Curve",
163         /* structName */        "CurveModifierData",
164         /* structSize */        sizeof(CurveModifierData),
165         /* type */              eModifierTypeType_OnlyDeform,
166         /* flags */             eModifierTypeFlag_AcceptsCVs |
167                                 eModifierTypeFlag_AcceptsLattice |
168                                 eModifierTypeFlag_SupportsEditmode,
169
170         /* copyData */          copyData,
171         /* deformVerts */       deformVerts,
172         /* deformMatrices */    NULL,
173         /* deformVertsEM */     deformVertsEM,
174         /* deformMatricesEM */  NULL,
175         /* applyModifier */     NULL,
176         /* applyModifierEM */   NULL,
177         /* initData */          initData,
178         /* requiredDataMask */  requiredDataMask,
179         /* freeData */          NULL,
180         /* isDisabled */        isDisabled,
181         /* updateDepgraph */    updateDepgraph,
182         /* updateDepsgraph */   updateDepsgraph,
183         /* dependsOnTime */     NULL,
184         /* dependsOnNormals */  NULL,
185         /* foreachObjectLink */ foreachObjectLink,
186         /* foreachIDLink */     NULL,
187         /* foreachTexLink */    NULL,
188 };