bb56a5b176a74dd0ef9f9ad78c613f4ec44a250e
[blender.git] / source / blender / blenkernel / intern / fluidsim.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.
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * The Original Code is Copyright (C) Blender Foundation
21  * All rights reserved.
22  *
23  * The Original Code is: all of this file.
24  *
25  * Contributor(s): none yet.
26  *
27  * ***** END GPL LICENSE BLOCK *****
28  */
29
30 #include <stddef.h>
31 #include "BLI_storage.h" /* _LARGEFILE_SOURCE */
32
33 #include "MEM_guardedalloc.h"
34
35 #include "DNA_mesh_types.h"
36 #include "DNA_meshdata_types.h"
37 #include "DNA_object_fluidsim.h"
38 #include "DNA_object_force.h" // for pointcache 
39 #include "DNA_object_types.h"
40 #include "DNA_particle_types.h"
41 #include "DNA_scene_types.h" // N_T
42
43 #include "BLI_math.h"
44 #include "BLI_blenlib.h"
45 #include "BLI_utildefines.h"
46
47 #include "BKE_cdderivedmesh.h"
48 #include "BKE_customdata.h"
49 #include "BKE_DerivedMesh.h"
50 #include "BKE_fluidsim.h"
51 #include "BKE_global.h"
52 #include "BKE_modifier.h"
53 #include "BKE_mesh.h"
54
55
56 // headers for fluidsim bobj meshes
57 #include <stdlib.h>
58 #include "LBM_fluidsim.h"
59 #include <zlib.h>
60 #include <string.h>
61 #include <stdio.h>
62
63 /* ************************* fluidsim bobj file handling **************************** */
64
65
66 //-------------------------------------------------------------------------------
67 // file handling
68 //-------------------------------------------------------------------------------
69
70 void initElbeemMesh(struct Scene *scene, struct Object *ob,
71                         int *numVertices, float **vertices,
72           int *numTriangles, int **triangles,
73           int useGlobalCoords, int modifierIndex)
74 {
75         DerivedMesh *dm = NULL;
76         MVert *mvert;
77         MFace *mface;
78         int countTris=0, i, totvert, totface;
79         float *verts;
80         int *tris;
81
82         dm = mesh_create_derived_index_render(scene, ob, CD_MASK_BAREMESH, modifierIndex);
83         //dm = mesh_create_derived_no_deform(ob,NULL);
84
85         mvert = dm->getVertArray(dm);
86         mface = dm->getFaceArray(dm);
87         totvert = dm->getNumVerts(dm);
88         totface = dm->getNumFaces(dm);
89
90         *numVertices = totvert;
91         verts = MEM_callocN( totvert*3*sizeof(float), "elbeemmesh_vertices");
92         for(i=0; i<totvert; i++) {
93                 VECCOPY( &verts[i*3], mvert[i].co);
94                 if(useGlobalCoords) { mul_m4_v3(ob->obmat, &verts[i*3]); }
95         }
96         *vertices = verts;
97
98         for(i=0; i<totface; i++) {
99                 countTris++;
100                 if(mface[i].v4) { countTris++; }
101         }
102         *numTriangles = countTris;
103         tris = MEM_callocN( countTris*3*sizeof(int), "elbeemmesh_triangles");
104         countTris = 0;
105         for(i=0; i<totface; i++) {
106                 int face[4];
107                 face[0] = mface[i].v1;
108                 face[1] = mface[i].v2;
109                 face[2] = mface[i].v3;
110                 face[3] = mface[i].v4;
111
112                 tris[countTris*3+0] = face[0];
113                 tris[countTris*3+1] = face[1];
114                 tris[countTris*3+2] = face[2];
115                 countTris++;
116                 if(face[3]) {
117                         tris[countTris*3+0] = face[0];
118                         tris[countTris*3+1] = face[2];
119                         tris[countTris*3+2] = face[3];
120                         countTris++;
121                 }
122         }
123         *triangles = tris;
124
125         dm->release(dm);
126 }