9f7d76d251d0b4d0734f3d2d6bbd6df9a95527fb
[blender.git] / source / blender / blenkernel / intern / fluidsim.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) Blender Foundation
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/blenkernel/intern/fluidsim.c
29  *  \ingroup bke
30  */
31
32
33 // headers for fluidsim bobj meshes
34 #include <stdlib.h>
35 #include <zlib.h>
36 #include <string.h>
37 #include <stdio.h>
38
39 #include "MEM_guardedalloc.h"
40
41 #include "DNA_mesh_types.h"
42 #include "DNA_meshdata_types.h"
43 #include "DNA_object_fluidsim.h"
44 #include "DNA_object_force.h" // for pointcache 
45 #include "DNA_object_types.h"
46 #include "DNA_particle_types.h"
47 #include "DNA_scene_types.h" // N_T
48
49 #include "BLI_math.h"
50 #include "BLI_blenlib.h"
51 #include "BLI_utildefines.h"
52
53 #include "BKE_cdderivedmesh.h"
54 #include "BKE_customdata.h"
55 #include "BKE_DerivedMesh.h"
56 #include "BKE_fluidsim.h"
57 #include "BKE_global.h"
58 #include "BKE_modifier.h"
59 #include "BKE_mesh.h"
60
61 /* ************************* fluidsim bobj file handling **************************** */
62
63
64 //-------------------------------------------------------------------------------
65 // file handling
66 //-------------------------------------------------------------------------------
67
68 void initElbeemMesh(struct Scene *scene, struct Object *ob,
69                         int *numVertices, float **vertices,
70           int *numTriangles, int **triangles,
71           int useGlobalCoords, int modifierIndex)
72 {
73         DerivedMesh *dm = NULL;
74         MVert *mvert;
75         MFace *mface;
76         int countTris=0, i, totvert, totface;
77         float *verts;
78         int *tris;
79
80         dm = mesh_create_derived_index_render(scene, ob, CD_MASK_BAREMESH, modifierIndex);
81         //dm = mesh_create_derived_no_deform(ob,NULL);
82
83         mvert = dm->getVertArray(dm);
84         mface = dm->getTessFaceArray(dm);
85         totvert = dm->getNumVerts(dm);
86         totface = dm->getNumTessFaces(dm);
87
88         *numVertices = totvert;
89         verts = MEM_callocN(totvert*3*sizeof(float), "elbeemmesh_vertices");
90         for (i=0; i<totvert; i++) {
91                 copy_v3_v3(&verts[i*3], mvert[i].co);
92                 if (useGlobalCoords) { mul_m4_v3(ob->obmat, &verts[i*3]); }
93         }
94         *vertices = verts;
95
96         for (i=0; i<totface; i++) {
97                 countTris++;
98                 if (mface[i].v4) { countTris++; }
99         }
100         *numTriangles = countTris;
101         tris = MEM_callocN(countTris*3*sizeof(int), "elbeemmesh_triangles");
102         countTris = 0;
103         for (i=0; i<totface; i++) {
104                 int face[4];
105                 face[0] = mface[i].v1;
106                 face[1] = mface[i].v2;
107                 face[2] = mface[i].v3;
108                 face[3] = mface[i].v4;
109
110                 tris[countTris*3+0] = face[0];
111                 tris[countTris*3+1] = face[1];
112                 tris[countTris*3+2] = face[2];
113                 countTris++;
114                 if (face[3]) {
115                         tris[countTris*3+0] = face[0];
116                         tris[countTris*3+1] = face[2];
117                         tris[countTris*3+2] = face[3];
118                         countTris++;
119                 }
120         }
121         *triangles = tris;
122
123         dm->release(dm);
124 }