merge with trunk at r31523
[blender.git] / source / blender / modifiers / intern / MOD_util.c
1 /**
2  * $Id$
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version. The Blender
10  * Foundation also sells licenses for use in proprietary software under
11  * the Blender License.  See http://www.blender.org/BL/ for information
12  * about this.
13  *
14  * This program is distributed in the hope that it will be useful;
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software Foundation;
21  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
22  *
23  * The Original Code is Copyright (C) 2005 Blender Foundation.
24  * All rights reserved.
25  *
26  * The Original Code is: all of this file.
27  *
28  * Contributor(s): Ben Batt
29  *
30  * ***** END GPL LICENSE BLOCK *****
31  */
32
33 #include <string.h>
34
35 #include "DNA_modifier_types.h"
36 #include "DNA_object_types.h"
37 #include "DNA_curve_types.h"
38
39 #include "BKE_cdderivedmesh.h"
40 #include "BKE_mesh.h"
41 #include "BKE_displist.h"
42 #include "BKE_utildefines.h"
43 #include "BKE_modifier.h"
44
45 #include "MOD_util.h"
46 #include "MOD_modifiertypes.h"
47
48 #include "MEM_guardedalloc.h"
49
50 #include "RE_shader_ext.h"
51
52 void get_texture_value(Tex *texture, float *tex_co, TexResult *texres)
53 {
54         int result_type;
55
56         result_type = multitex_ext(texture, tex_co, NULL, NULL, 0, texres);
57
58         /* if the texture gave an RGB value, we assume it didn't give a valid
59         * intensity, so calculate one (formula from do_material_tex).
60         * if the texture didn't give an RGB value, copy the intensity across
61         */
62         if(result_type & TEX_RGB)
63                 texres->tin = (0.35f * texres->tr + 0.45f * texres->tg
64                                 + 0.2f * texres->tb);
65         else
66                 texres->tr = texres->tg = texres->tb = texres->tin;
67 }
68
69 void modifier_vgroup_cache(ModifierData *md, float (*vertexCos)[3])
70 {
71         while((md=md->next) && md->type==eModifierType_Armature) {
72                 ArmatureModifierData *amd = (ArmatureModifierData*) md;
73                 if(amd->multi && amd->prevCos==NULL)
74                         amd->prevCos= MEM_dupallocN(vertexCos);
75                 else
76                         break;
77         }
78         /* lattice/mesh modifier too */
79 }
80
81 void validate_layer_name(const CustomData *data, int type, char *name, char *outname)
82 {
83         int index = -1;
84
85         /* if a layer name was given, try to find that layer */
86         if(name[0])
87                 index = CustomData_get_named_layer_index(data, CD_MTFACE, name);
88
89         if(index < 0) {
90                 /* either no layer was specified, or the layer we want has been
91                 * deleted, so assign the active layer to name
92                 */
93                 index = CustomData_get_active_layer_index(data, CD_MTFACE);
94                 strcpy(outname, data->layers[index].name);
95         }
96         else
97                 strcpy(outname, name);
98 }
99
100 /* returns a cdderivedmesh if dm == NULL or is another type of derivedmesh */
101 DerivedMesh *get_cddm(struct Scene *scene, Object *ob, struct EditMesh *em, DerivedMesh *dm, float (*vertexCos)[3])
102 {
103         if(dm && dm->type == DM_TYPE_CDDM)
104                 return dm;
105
106         if(!dm) {
107                 dm= get_dm(scene, ob, em, dm, vertexCos, 0);
108         }
109         else {
110                 dm= CDDM_copy(dm, 0);
111                 CDDM_apply_vert_coords(dm, vertexCos);
112         }
113
114         if(dm)
115                 CDDM_calc_normals(dm);
116         
117         return dm;
118 }
119
120 /* returns a derived mesh if dm == NULL, for deforming modifiers that need it */
121 DerivedMesh *get_dm(struct Scene *scene, Object *ob, struct EditMesh *em, DerivedMesh *dm, float (*vertexCos)[3], int orco)
122 {
123         if(dm)
124                 return dm;
125
126         if(ob->type==OB_MESH) {
127                 if(em) dm= CDDM_from_editmesh(em, ob->data);
128                 else dm = CDDM_from_mesh((struct Mesh *)(ob->data), ob);
129
130                 if(vertexCos) {
131                         CDDM_apply_vert_coords(dm, vertexCos);
132                         //CDDM_calc_normals(dm);
133                 }
134                 
135                 if(orco)
136                         DM_add_vert_layer(dm, CD_ORCO, CD_ASSIGN, get_mesh_orco_verts(ob));
137         }
138         else if(ELEM3(ob->type,OB_FONT,OB_CURVE,OB_SURF)) {
139                 dm= CDDM_from_curve(ob);
140         }
141
142         return dm;
143 }
144
145 /* only called by BKE_modifier.h/modifier.c */
146 void modifier_type_init(ModifierTypeInfo *types[], ModifierType type)
147 {
148         memset(types, 0, sizeof(types));
149 #define INIT_TYPE(typeName) (types[eModifierType_##typeName] = &modifierType_##typeName)
150         INIT_TYPE(None);
151         INIT_TYPE(Curve);
152         INIT_TYPE(Lattice);
153         INIT_TYPE(Subsurf);
154         INIT_TYPE(Build);
155         INIT_TYPE(Array);
156         INIT_TYPE(Mirror);
157         INIT_TYPE(EdgeSplit);
158         INIT_TYPE(Bevel);
159         INIT_TYPE(Displace);
160         INIT_TYPE(UVProject);
161         INIT_TYPE(Decimate);
162         INIT_TYPE(Smooth);
163         INIT_TYPE(Cast);
164         INIT_TYPE(Wave);
165         INIT_TYPE(Armature);
166         INIT_TYPE(Hook);
167         INIT_TYPE(Softbody);
168         INIT_TYPE(Cloth);
169         INIT_TYPE(Collision);
170         INIT_TYPE(Boolean);
171         INIT_TYPE(MeshDeform);
172         INIT_TYPE(ParticleSystem);
173         INIT_TYPE(ParticleInstance);
174         INIT_TYPE(Explode);
175         INIT_TYPE(Shrinkwrap);
176         INIT_TYPE(Fluidsim);
177         INIT_TYPE(Mask);
178         INIT_TYPE(SimpleDeform);
179         INIT_TYPE(Multires);
180         INIT_TYPE(Surface);
181         INIT_TYPE(Smoke);
182         INIT_TYPE(ShapeKey);
183         INIT_TYPE(Solidify);
184         INIT_TYPE(Screw);
185 #undef INIT_TYPE
186 }