Threaded object update and EvaluationContext
[blender.git] / source / blender / modifiers / intern / MOD_util.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 Blender Foundation.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): Ben Batt
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/modifiers/intern/MOD_util.c
29  *  \ingroup modifiers
30  */
31
32
33 #include <string.h>
34
35 #include "DNA_curve_types.h"
36 #include "DNA_image_types.h"
37 #include "DNA_lattice_types.h"
38 #include "DNA_meshdata_types.h"
39 #include "DNA_modifier_types.h"
40 #include "DNA_object_types.h"
41 #include "DNA_scene_types.h"
42
43 #include "BLI_utildefines.h"
44 #include "BLI_math_vector.h"
45 #include "BLI_math_matrix.h"
46
47 #include "BKE_cdderivedmesh.h"
48 #include "BKE_deform.h"
49 #include "BKE_image.h"
50 #include "BKE_lattice.h"
51 #include "BKE_mesh.h"
52 #include "BKE_displist.h"
53 #include "BKE_scene.h"
54
55 #include "BKE_modifier.h"
56
57 #include "MOD_util.h"
58 #include "MOD_modifiertypes.h"
59
60 #include "MEM_guardedalloc.h"
61
62 #include "RE_shader_ext.h"
63
64 void modifier_init_texture(Scene *scene, Tex *tex)
65 {
66         if (!tex)
67                 return;
68
69         if (tex->ima && ELEM(tex->ima->source, IMA_SRC_MOVIE, IMA_SRC_SEQUENCE))
70                 BKE_image_user_frame_calc(&tex->iuser, scene->r.cfra, 0);
71 }
72
73 void get_texture_coords(MappingInfoModifierData *dmd, Object *ob,
74                         DerivedMesh *dm,
75                         float (*co)[3], float (*texco)[3],
76                         int numVerts)
77 {
78         int i;
79         int texmapping = dmd->texmapping;
80         float mapob_imat[4][4];
81
82         if (texmapping == MOD_DISP_MAP_OBJECT) {
83                 if (dmd->map_object)
84                         invert_m4_m4(mapob_imat, dmd->map_object->obmat);
85                 else /* if there is no map object, default to local */
86                         texmapping = MOD_DISP_MAP_LOCAL;
87         }
88
89         /* UVs need special handling, since they come from faces */
90         if (texmapping == MOD_DISP_MAP_UV) {
91                 if (CustomData_has_layer(&dm->loopData, CD_MLOOPUV)) {
92                         MPoly *mpoly = dm->getPolyArray(dm);
93                         MPoly *mp;
94                         MLoop *mloop = dm->getLoopArray(dm);
95                         char *done = MEM_callocN(sizeof(*done) * numVerts,
96                                                  "get_texture_coords done");
97                         int numPolys = dm->getNumPolys(dm);
98                         char uvname[MAX_CUSTOMDATA_LAYER_NAME];
99                         MLoopUV *mloop_uv;
100
101                         CustomData_validate_layer_name(&dm->loopData, CD_MLOOPUV, dmd->uvlayer_name, uvname);
102                         mloop_uv = CustomData_get_layer_named(&dm->loopData, CD_MLOOPUV, uvname);
103
104                         /* verts are given the UV from the first face that uses them */
105                         for (i = 0, mp = mpoly; i < numPolys; ++i, ++mp) {
106                                 unsigned int fidx = mp->totloop - 1;
107
108                                 do {
109                                         unsigned int lidx = mp->loopstart + fidx;
110                                         unsigned int vidx = mloop[lidx].v;
111
112                                         if (done[vidx] == 0) {
113                                                 /* remap UVs from [0, 1] to [-1, 1] */
114                                                 texco[vidx][0] = (mloop_uv[lidx].uv[0] * 2.0f) - 1.0f;
115                                                 texco[vidx][1] = (mloop_uv[lidx].uv[1] * 2.0f) - 1.0f;
116                                                 done[vidx] = 1;
117                                         }
118
119                                 } while (fidx--);
120                         }
121
122                         MEM_freeN(done);
123                         return;
124                 }
125                 else /* if there are no UVs, default to local */
126                         texmapping = MOD_DISP_MAP_LOCAL;
127         }
128
129         for (i = 0; i < numVerts; ++i, ++co, ++texco) {
130                 switch (texmapping) {
131                         case MOD_DISP_MAP_LOCAL:
132                                 copy_v3_v3(*texco, *co);
133                                 break;
134                         case MOD_DISP_MAP_GLOBAL:
135                                 mul_v3_m4v3(*texco, ob->obmat, *co);
136                                 break;
137                         case MOD_DISP_MAP_OBJECT:
138                                 mul_v3_m4v3(*texco, ob->obmat, *co);
139                                 mul_m4_v3(mapob_imat, *texco);
140                                 break;
141                 }
142         }
143 }
144
145 void modifier_vgroup_cache(ModifierData *md, float (*vertexCos)[3])
146 {
147         while ((md = md->next) && md->type == eModifierType_Armature) {
148                 ArmatureModifierData *amd = (ArmatureModifierData *) md;
149                 if (amd->multi && amd->prevCos == NULL)
150                         amd->prevCos = MEM_dupallocN(vertexCos);
151                 else
152                         break;
153         }
154         /* lattice/mesh modifier too */
155 }
156
157 /* returns a cdderivedmesh if dm == NULL or is another type of derivedmesh */
158 DerivedMesh *get_cddm(Object *ob, struct BMEditMesh *em, DerivedMesh *dm, float (*vertexCos)[3], bool use_normals)
159 {
160         if (dm) {
161                 if (dm->type != DM_TYPE_CDDM) {
162                         dm = CDDM_copy(dm);
163                         CDDM_apply_vert_coords(dm, vertexCos);
164                 }
165
166                 if (use_normals) {
167                         DM_ensure_normals(dm);
168                 }
169         }
170         else {
171                 dm = get_dm(ob, em, dm, vertexCos, use_normals, false);
172         }
173
174         return dm;
175 }
176
177 /* returns a derived mesh if dm == NULL, for deforming modifiers that need it */
178 DerivedMesh *get_dm(Object *ob, struct BMEditMesh *em, DerivedMesh *dm,
179                     float (*vertexCos)[3], bool use_normals, bool use_orco)
180 {
181         if (dm) {
182                 /* pass */
183         }
184         else if (ob->type == OB_MESH) {
185                 if (em) dm = CDDM_from_editbmesh(em, FALSE, FALSE);
186                 else dm = CDDM_from_mesh((struct Mesh *)(ob->data));
187
188                 if (vertexCos) {
189                         CDDM_apply_vert_coords(dm, vertexCos);
190                         dm->dirty |= DM_DIRTY_NORMALS;
191                 }
192                 
193                 if (use_orco) {
194                         DM_add_vert_layer(dm, CD_ORCO, CD_ASSIGN, BKE_mesh_orco_verts_get(ob));
195                 }
196         }
197         else if (ELEM3(ob->type, OB_FONT, OB_CURVE, OB_SURF)) {
198                 dm = CDDM_from_curve(ob);
199         }
200
201         if (use_normals) {
202                 if (LIKELY(dm)) {
203                         DM_ensure_normals(dm);
204                 }
205         }
206
207         return dm;
208 }
209
210 /* Get derived mesh for other object, which is used as an operand for the modifier,
211  * i.e. second operand for boolean modifier.
212  */
213 DerivedMesh *get_dm_for_modifier(Object *ob, ModifierApplyFlag flag)
214 {
215         if (flag & MOD_APPLY_RENDER) {
216                 /* TODO(sergey): Use proper derived render in the future. */
217                 return ob->derivedFinal;
218         }
219         else {
220                 return ob->derivedFinal;
221         }
222 }
223
224 void modifier_get_vgroup(Object *ob, DerivedMesh *dm, const char *name, MDeformVert **dvert, int *defgrp_index)
225 {
226         *defgrp_index = defgroup_name_index(ob, name);
227         *dvert = NULL;
228
229         if (*defgrp_index != -1) {
230                 if (ob->type == OB_LATTICE)
231                         *dvert = BKE_lattice_deform_verts_get(ob);
232                 else if (dm)
233                         *dvert = dm->getVertDataArray(dm, CD_MDEFORMVERT);
234         }
235 }
236
237 /* only called by BKE_modifier.h/modifier.c */
238 void modifier_type_init(ModifierTypeInfo *types[])
239 {
240 #define INIT_TYPE(typeName) (types[eModifierType_##typeName] = &modifierType_##typeName)
241         INIT_TYPE(None);
242         INIT_TYPE(Curve);
243         INIT_TYPE(Lattice);
244         INIT_TYPE(Subsurf);
245         INIT_TYPE(Build);
246         INIT_TYPE(Array);
247         INIT_TYPE(Mirror);
248         INIT_TYPE(EdgeSplit);
249         INIT_TYPE(Bevel);
250         INIT_TYPE(Displace);
251         INIT_TYPE(UVProject);
252         INIT_TYPE(Decimate);
253         INIT_TYPE(Smooth);
254         INIT_TYPE(Cast);
255         INIT_TYPE(Wave);
256         INIT_TYPE(Armature);
257         INIT_TYPE(Hook);
258         INIT_TYPE(Softbody);
259         INIT_TYPE(Cloth);
260         INIT_TYPE(Collision);
261         INIT_TYPE(Boolean);
262         INIT_TYPE(MeshDeform);
263         INIT_TYPE(Ocean);
264         INIT_TYPE(ParticleSystem);
265         INIT_TYPE(ParticleInstance);
266         INIT_TYPE(Explode);
267         INIT_TYPE(Shrinkwrap);
268         INIT_TYPE(Fluidsim);
269         INIT_TYPE(Mask);
270         INIT_TYPE(SimpleDeform);
271         INIT_TYPE(Multires);
272         INIT_TYPE(Surface);
273         INIT_TYPE(Smoke);
274         INIT_TYPE(ShapeKey);
275         INIT_TYPE(Solidify);
276         INIT_TYPE(Screw);
277         INIT_TYPE(Warp);
278         INIT_TYPE(WeightVGEdit);
279         INIT_TYPE(WeightVGMix);
280         INIT_TYPE(WeightVGProximity);
281         INIT_TYPE(DynamicPaint);
282         INIT_TYPE(Remesh);
283         INIT_TYPE(Skin);
284         INIT_TYPE(LaplacianSmooth);
285         INIT_TYPE(Triangulate);
286         INIT_TYPE(UVWarp);
287         INIT_TYPE(MeshCache);
288         INIT_TYPE(LaplacianDeform);
289         INIT_TYPE(Wireframe);
290 #undef INIT_TYPE
291 }