b18be28ac7f2c701ee73fea90c7d2fd059741f5f
[blender.git] / source / blender / modifiers / intern / MOD_explode.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_explode.c
32  *  \ingroup modifiers
33  */
34
35
36 #include "DNA_meshdata_types.h"
37 #include "DNA_scene_types.h"
38
39 #include "BLI_kdtree.h"
40 #include "BLI_rand.h"
41 #include "BLI_math.h"
42 #include "BLI_edgehash.h"
43 #include "BLI_utildefines.h"
44
45 #include "BKE_cdderivedmesh.h"
46 #include "BKE_deform.h"
47 #include "BKE_lattice.h"
48 #include "BKE_mesh.h"
49 #include "BKE_modifier.h"
50 #include "BKE_object.h"
51 #include "BKE_particle.h"
52 #include "BKE_scene.h"
53
54
55 #include "MEM_guardedalloc.h"
56
57 #include "MOD_util.h"
58
59 static void initData(ModifierData *md)
60 {
61         ExplodeModifierData *emd= (ExplodeModifierData*) md;
62
63         emd->facepa= NULL;
64         emd->flag |= eExplodeFlag_Unborn+eExplodeFlag_Alive+eExplodeFlag_Dead;
65 }
66 static void freeData(ModifierData *md)
67 {
68         ExplodeModifierData *emd= (ExplodeModifierData*) md;
69         
70         if(emd->facepa) MEM_freeN(emd->facepa);
71 }
72 static void copyData(ModifierData *md, ModifierData *target)
73 {
74         ExplodeModifierData *emd= (ExplodeModifierData*) md;
75         ExplodeModifierData *temd= (ExplodeModifierData*) target;
76
77         temd->facepa = NULL;
78         temd->flag = emd->flag;
79         temd->protect = emd->protect;
80         temd->vgroup = emd->vgroup;
81 }
82 static int dependsOnTime(ModifierData *UNUSED(md)) 
83 {
84         return 1;
85 }
86 static CustomDataMask requiredDataMask(Object *UNUSED(ob), ModifierData *md)
87 {
88         ExplodeModifierData *emd= (ExplodeModifierData*) md;
89         CustomDataMask dataMask = 0;
90
91         if(emd->vgroup)
92                 dataMask |= CD_MASK_MDEFORMVERT;
93
94         return dataMask;
95 }
96
97 static void createFacepa(ExplodeModifierData *emd,
98                                                 ParticleSystemModifierData *psmd,
99                                                 DerivedMesh *dm)
100 {
101         ParticleSystem *psys=psmd->psys;
102         MFace *fa=NULL, *mface=NULL;
103         MVert *mvert = NULL;
104         ParticleData *pa;
105         KDTree *tree;
106         float center[3], co[3];
107         int *facepa=NULL,*vertpa=NULL,totvert=0,totface=0,totpart=0;
108         int i,p,v1,v2,v3,v4=0;
109
110         mvert = dm->getVertArray(dm);
111         mface = dm->getFaceArray(dm);
112         totface= dm->getNumFaces(dm);
113         totvert= dm->getNumVerts(dm);
114         totpart= psmd->psys->totpart;
115
116         BLI_srandom(psys->seed);
117
118         if(emd->facepa)
119                 MEM_freeN(emd->facepa);
120
121         facepa = emd->facepa = MEM_callocN(sizeof(int)*totface, "explode_facepa");
122
123         vertpa = MEM_callocN(sizeof(int)*totvert, "explode_vertpa");
124
125         /* initialize all faces & verts to no particle */
126         for(i=0; i<totface; i++)
127                 facepa[i]=totpart;
128
129         for (i=0; i<totvert; i++)
130                 vertpa[i]=totpart;
131
132         /* set protected verts */
133         if(emd->vgroup){
134                 MDeformVert *dvert = dm->getVertDataArray(dm, CD_MDEFORMVERT);
135                 if(dvert){
136                         const int defgrp_index= emd->vgroup-1;
137                         for(i=0; i<totvert; i++, dvert++){
138                                 float val = BLI_frand();
139                                 val = (1.0f-emd->protect)*val + emd->protect*0.5f;
140                                 if(val < defvert_find_weight(dvert, defgrp_index))
141                                         vertpa[i] = -1;
142                         }
143                 }
144         }
145
146         /* make tree of emitter locations */
147         tree=BLI_kdtree_new(totpart);
148         for(p=0,pa=psys->particles; p<totpart; p++,pa++){
149                 psys_particle_on_emitter(psmd,psys->part->from,pa->num,pa->num_dmcache,pa->fuv,pa->foffset,co,NULL,NULL,NULL,NULL,NULL);
150                 BLI_kdtree_insert(tree, p, co, NULL);
151         }
152         BLI_kdtree_balance(tree);
153
154         /* set face-particle-indexes to nearest particle to face center */
155         for(i=0,fa=mface; i<totface; i++,fa++){
156                 add_v3_v3v3(center,mvert[fa->v1].co,mvert[fa->v2].co);
157                 add_v3_v3(center, mvert[fa->v3].co);
158                 if(fa->v4){
159                         add_v3_v3(center, mvert[fa->v4].co);
160                         mul_v3_fl(center,0.25);
161                 }
162                 else
163                         mul_v3_fl(center,0.3333f);
164
165                 p= BLI_kdtree_find_nearest(tree,center,NULL,NULL);
166
167                 v1=vertpa[fa->v1];
168                 v2=vertpa[fa->v2];
169                 v3=vertpa[fa->v3];
170                 if(fa->v4)
171                         v4=vertpa[fa->v4];
172
173                 if(v1>=0 && v2>=0 && v3>=0 && (fa->v4==0 || v4>=0))
174                         facepa[i]=p;
175
176                 if(v1>=0) vertpa[fa->v1]=p;
177                 if(v2>=0) vertpa[fa->v2]=p;
178                 if(v3>=0) vertpa[fa->v3]=p;
179                 if(fa->v4 && v4>=0) vertpa[fa->v4]=p;
180         }
181
182         if(vertpa) MEM_freeN(vertpa);
183         BLI_kdtree_free(tree);
184 }
185
186 static int edgecut_get(EdgeHash *edgehash, int v1, int v2)
187 {
188         return GET_INT_FROM_POINTER(BLI_edgehash_lookup(edgehash, v1, v2));
189 }
190
191  
192 static const short add_faces[24] = {
193         0,
194         0, 0, 2, 0, 1, 2, 2, 0, 2, 1,
195         2, 2, 2, 2, 3, 0, 0, 0, 1, 0,
196         1, 1, 2
197  };
198
199 static MFace *get_dface(DerivedMesh *dm, DerivedMesh *split, int cur, int i, MFace *mf)
200 {
201         MFace *df = CDDM_get_face(split, cur);
202         DM_copy_face_data(dm, split, i, cur, 1);
203         *df = *mf;
204         return df;
205 }
206
207 #define SET_VERTS(a, b, c, d) \
208                         v[0]=mf->v##a; uv[0]=a-1; \
209                         v[1]=mf->v##b; uv[1]=b-1; \
210                         v[2]=mf->v##c; uv[2]=c-1; \
211                         v[3]=mf->v##d; uv[3]=d-1;
212
213 #define GET_ES(v1, v2) edgecut_get(eh, v1, v2);
214 #define INT_UV(uvf, c0, c1) interp_v2_v2v2(uvf, mf->uv[c0], mf->uv[c1], 0.5f);
215
216 static void remap_faces_3_6_9_12(DerivedMesh *dm, DerivedMesh *split, MFace *mf, int *facepa, int *vertpa, int i, EdgeHash *eh, int cur, int v1, int v2, int v3, int v4)
217 {
218         MFace *df1 = get_dface(dm, split, cur, i, mf);
219         MFace *df2 = get_dface(dm, split, cur+1, i, mf);
220         MFace *df3 = get_dface(dm, split, cur+2, i, mf);
221
222         facepa[cur] = vertpa[v1];
223         df1->v1 = v1;
224         df1->v2 = GET_ES(v1, v2)
225         df1->v3 = GET_ES(v2, v3)
226         df1->v4 = v3;
227         df1->flag |= ME_FACE_SEL;
228
229         facepa[cur+1] = vertpa[v2];
230         df2->v1 = GET_ES(v1, v2)
231         df2->v2 = v2;
232         df2->v3 = GET_ES(v2, v3)
233         df2->v4 = 0;
234         df2->flag &= ~ME_FACE_SEL;
235
236         facepa[cur+2] = vertpa[v1];
237         df3->v1 = v1;
238         df3->v2 = v3;
239         df3->v3 = v4;
240         df3->v4 = 0;
241         df3->flag &= ~ME_FACE_SEL;
242 }
243
244 static void remap_uvs_3_6_9_12(DerivedMesh *dm, DerivedMesh *split, int numlayer, int i, int cur, int c0, int c1, int c2, int c3)
245 {
246         MTFace *mf, *df1, *df2, *df3;
247         int l;
248
249         for(l=0; l<numlayer; l++) {
250                 mf = CustomData_get_layer_n(&split->faceData, CD_MTFACE, l);
251                 df1 = mf+cur;
252                 df2 = df1 + 1;
253                 df3 = df1 + 2;
254                 mf = CustomData_get_layer_n(&dm->faceData, CD_MTFACE, l);
255                 mf += i;
256
257                 copy_v2_v2(df1->uv[0], mf->uv[c0]);
258                 INT_UV(df1->uv[1], c0, c1)
259                 INT_UV(df1->uv[2], c1, c2)
260                 copy_v2_v2(df1->uv[3], mf->uv[c2]);
261
262                 INT_UV(df2->uv[0], c0, c1)
263                 copy_v2_v2(df2->uv[1], mf->uv[c1]);
264                 INT_UV(df2->uv[2], c1, c2)
265
266                 copy_v2_v2(df3->uv[0], mf->uv[c0]);
267                 copy_v2_v2(df3->uv[1], mf->uv[c2]);
268                 copy_v2_v2(df3->uv[2], mf->uv[c3]);
269         }
270 }
271
272 static void remap_faces_5_10(DerivedMesh *dm, DerivedMesh *split, MFace *mf, int *facepa, int *vertpa, int i, EdgeHash *eh, int cur, int v1, int v2, int v3, int v4)
273 {
274         MFace *df1 = get_dface(dm, split, cur, i, mf);
275         MFace *df2 = get_dface(dm, split, cur+1, i, mf);
276
277         facepa[cur] = vertpa[v1];
278         df1->v1 = v1;
279         df1->v2 = v2;
280         df1->v3 = GET_ES(v2, v3)
281         df1->v4 = GET_ES(v1, v4)
282         df1->flag |= ME_FACE_SEL;
283
284         facepa[cur+1] = vertpa[v3];
285         df2->v1 = GET_ES(v1, v4)
286         df2->v2 = GET_ES(v2, v3)
287         df2->v3 = v3;
288         df2->v4 = v4;
289         df2->flag |= ME_FACE_SEL;
290 }
291
292 static void remap_uvs_5_10(DerivedMesh *dm, DerivedMesh *split, int numlayer, int i, int cur, int c0, int c1, int c2, int c3)
293 {
294         MTFace *mf, *df1, *df2;
295         int l;
296
297         for(l=0; l<numlayer; l++) {
298                 mf = CustomData_get_layer_n(&split->faceData, CD_MTFACE, l);
299                 df1 = mf+cur;
300                 df2 = df1 + 1;
301                 mf = CustomData_get_layer_n(&dm->faceData, CD_MTFACE, l);
302                 mf += i;
303
304                 copy_v2_v2(df1->uv[0], mf->uv[c0]);
305                 copy_v2_v2(df1->uv[1], mf->uv[c1]);
306                 INT_UV(df1->uv[2], c1, c2)
307                 INT_UV(df1->uv[3], c0, c3)
308
309                 INT_UV(df2->uv[0], c0, c3)
310                 INT_UV(df2->uv[1], c1, c2)
311                 copy_v2_v2(df2->uv[2], mf->uv[c2]);
312                 copy_v2_v2(df2->uv[3], mf->uv[c3]);
313
314         }
315 }
316
317 static void remap_faces_15(DerivedMesh *dm, DerivedMesh *split, MFace *mf, int *facepa, int *vertpa, int i, EdgeHash *eh, int cur, int v1, int v2, int v3, int v4)
318 {
319         MFace *df1 = get_dface(dm, split, cur, i, mf);
320         MFace *df2 = get_dface(dm, split, cur+1, i, mf);
321         MFace *df3 = get_dface(dm, split, cur+2, i, mf);
322         MFace *df4 = get_dface(dm, split, cur+3, i, mf);
323
324         facepa[cur] = vertpa[v1];
325         df1->v1 = v1;
326         df1->v2 = GET_ES(v1, v2)
327         df1->v3 = GET_ES(v1, v3)
328         df1->v4 = GET_ES(v1, v4)
329         df1->flag |= ME_FACE_SEL;
330
331         facepa[cur+1] = vertpa[v2];
332         df2->v1 = GET_ES(v1, v2)
333         df2->v2 = v2;
334         df2->v3 = GET_ES(v2, v3)
335         df2->v4 = GET_ES(v1, v3)
336         df2->flag |= ME_FACE_SEL;
337
338         facepa[cur+2] = vertpa[v3];
339         df3->v1 = GET_ES(v1, v3)
340         df3->v2 = GET_ES(v2, v3)
341         df3->v3 = v3;
342         df3->v4 = GET_ES(v3, v4)
343         df3->flag |= ME_FACE_SEL;
344
345         facepa[cur+3] = vertpa[v4];
346         df4->v1 = GET_ES(v1, v4)
347         df4->v2 = GET_ES(v1, v3)
348         df4->v3 = GET_ES(v3, v4)
349         df4->v4 = v4;
350         df4->flag |= ME_FACE_SEL;
351 }
352
353 static void remap_uvs_15(DerivedMesh *dm, DerivedMesh *split, int numlayer, int i, int cur, int c0, int c1, int c2, int c3)
354 {
355         MTFace *mf, *df1, *df2, *df3, *df4;
356         int l;
357
358         for(l=0; l<numlayer; l++) {
359                 mf = CustomData_get_layer_n(&split->faceData, CD_MTFACE, l);
360                 df1 = mf+cur;
361                 df2 = df1 + 1;
362                 df3 = df1 + 2;
363                 df4 = df1 + 3;
364                 mf = CustomData_get_layer_n(&dm->faceData, CD_MTFACE, l);
365                 mf += i;
366
367                 copy_v2_v2(df1->uv[0], mf->uv[c0]);
368                 INT_UV(df1->uv[1], c0, c1)
369                 INT_UV(df1->uv[2], c0, c2)
370                 INT_UV(df1->uv[3], c0, c3)
371
372                 INT_UV(df2->uv[0], c0, c1)
373                 copy_v2_v2(df2->uv[1], mf->uv[c1]);
374                 INT_UV(df2->uv[2], c1, c2)
375                 INT_UV(df2->uv[3], c0, c2)
376
377                 INT_UV(df3->uv[0], c0, c2)
378                 INT_UV(df3->uv[1], c1, c2)
379                 copy_v2_v2(df3->uv[2], mf->uv[c2]);
380                 INT_UV(df3->uv[3], c2, c3)
381
382                 INT_UV(df4->uv[0], c0, c3)
383                 INT_UV(df4->uv[1], c0, c2)
384                 INT_UV(df4->uv[2], c2, c3)
385                 copy_v2_v2(df4->uv[3], mf->uv[c3]);
386         }
387 }
388
389 static void remap_faces_7_11_13_14(DerivedMesh *dm, DerivedMesh *split, MFace *mf, int *facepa, int *vertpa, int i, EdgeHash *eh, int cur, int v1, int v2, int v3, int v4)
390 {
391         MFace *df1 = get_dface(dm, split, cur, i, mf);
392         MFace *df2 = get_dface(dm, split, cur+1, i, mf);
393         MFace *df3 = get_dface(dm, split, cur+2, i, mf);
394
395         facepa[cur] = vertpa[v1];
396         df1->v1 = v1;
397         df1->v2 = GET_ES(v1, v2)
398         df1->v3 = GET_ES(v2, v3)
399         df1->v4 = GET_ES(v1, v4)
400         df1->flag |= ME_FACE_SEL;
401
402         facepa[cur+1] = vertpa[v2];
403         df2->v1 = GET_ES(v1, v2)
404         df2->v2 = v2;
405         df2->v3 = GET_ES(v2, v3)
406         df2->v4 = 0;
407         df2->flag &= ~ME_FACE_SEL;
408
409         facepa[cur+2] = vertpa[v4];
410         df3->v1 = GET_ES(v1, v4)
411         df3->v2 = GET_ES(v2, v3)
412         df3->v3 = v3;
413         df3->v4 = v4;
414         df3->flag |= ME_FACE_SEL;
415 }
416
417 static void remap_uvs_7_11_13_14(DerivedMesh *dm, DerivedMesh *split, int numlayer, int i, int cur, int c0, int c1, int c2, int c3)
418 {
419         MTFace *mf, *df1, *df2, *df3;
420         int l;
421
422         for(l=0; l<numlayer; l++) {
423                 mf = CustomData_get_layer_n(&split->faceData, CD_MTFACE, l);
424                 df1 = mf+cur;
425                 df2 = df1 + 1;
426                 df3 = df1 + 2;
427                 mf = CustomData_get_layer_n(&dm->faceData, CD_MTFACE, l);
428                 mf += i;
429
430                 copy_v2_v2(df1->uv[0], mf->uv[c0]);
431                 INT_UV(df1->uv[1], c0, c1)
432                 INT_UV(df1->uv[2], c1, c2)
433                 INT_UV(df1->uv[3], c0, c3)
434
435                 INT_UV(df2->uv[0], c0, c1)
436                 copy_v2_v2(df2->uv[1], mf->uv[c1]);
437                 INT_UV(df2->uv[2], c1, c2)
438
439                 INT_UV(df3->uv[0], c0, c3)
440                 INT_UV(df3->uv[1], c1, c2)
441                 copy_v2_v2(df3->uv[2], mf->uv[c2]);
442                 copy_v2_v2(df3->uv[3], mf->uv[c3]);
443         }
444 }
445
446 static void remap_faces_19_21_22(DerivedMesh *dm, DerivedMesh *split, MFace *mf, int *facepa, int *vertpa, int i, EdgeHash *eh, int cur, int v1, int v2, int v3)
447 {
448         MFace *df1 = get_dface(dm, split, cur, i, mf);
449         MFace *df2 = get_dface(dm, split, cur+1, i, mf);
450
451         facepa[cur] = vertpa[v1];
452         df1->v1 = v1;
453         df1->v2 = GET_ES(v1, v2)
454         df1->v3 = GET_ES(v1, v3)
455         df1->v4 = 0;
456         df1->flag &= ~ME_FACE_SEL;
457
458         facepa[cur+1] = vertpa[v2];
459         df2->v1 = GET_ES(v1, v2)
460         df2->v2 = v2;
461         df2->v3 = v3;
462         df2->v4 = GET_ES(v1, v3)
463         df2->flag |= ME_FACE_SEL;
464 }
465
466 static void remap_uvs_19_21_22(DerivedMesh *dm, DerivedMesh *split, int numlayer, int i, int cur, int c0, int c1, int c2)
467 {
468         MTFace *mf, *df1, *df2;
469         int l;
470
471         for(l=0; l<numlayer; l++) {
472                 mf = CustomData_get_layer_n(&split->faceData, CD_MTFACE, l);
473                 df1 = mf+cur;
474                 df2 = df1 + 1;
475                 mf = CustomData_get_layer_n(&dm->faceData, CD_MTFACE, l);
476                 mf += i;
477
478                 copy_v2_v2(df1->uv[0], mf->uv[c0]);
479                 INT_UV(df1->uv[1], c0, c1)
480                 INT_UV(df1->uv[2], c0, c2)
481
482                 INT_UV(df2->uv[0], c0, c1)
483                 copy_v2_v2(df2->uv[1], mf->uv[c1]);
484                 copy_v2_v2(df2->uv[2], mf->uv[c2]);
485                 INT_UV(df2->uv[3], c0, c2)
486         }
487 }
488
489 static void remap_faces_23(DerivedMesh *dm, DerivedMesh *split, MFace *mf, int *facepa, int *vertpa, int i, EdgeHash *eh, int cur, int v1, int v2, int v3)
490 {
491         MFace *df1 = get_dface(dm, split, cur, i, mf);
492         MFace *df2 = get_dface(dm, split, cur+1, i, mf);
493         MFace *df3 = get_dface(dm, split, cur+2, i, mf);
494
495         facepa[cur] = vertpa[v1];
496         df1->v1 = v1;
497         df1->v2 = GET_ES(v1, v2)
498         df1->v3 = GET_ES(v2, v3)
499         df1->v4 = GET_ES(v1, v3)
500         df1->flag |= ME_FACE_SEL;
501
502         facepa[cur+1] = vertpa[v2];
503         df2->v1 = GET_ES(v1, v2)
504         df2->v2 = v2;
505         df2->v3 = GET_ES(v2, v3)
506         df2->v4 = 0;
507         df2->flag &= ~ME_FACE_SEL;
508
509         facepa[cur+2] = vertpa[v3];
510         df3->v1 = GET_ES(v1, v3)
511         df3->v2 = GET_ES(v2, v3)
512         df3->v3 = v3;
513         df3->v4 = 0;
514         df3->flag &= ~ME_FACE_SEL;
515 }
516
517 static void remap_uvs_23(DerivedMesh *dm, DerivedMesh *split, int numlayer, int i, int cur, int c0, int c1, int c2)
518 {
519         MTFace *mf, *df1, *df2;
520         int l;
521
522         for(l=0; l<numlayer; l++) {
523                 mf = CustomData_get_layer_n(&split->faceData, CD_MTFACE, l);
524                 df1 = mf+cur;
525                 df2 = df1 + 1;
526                 mf = CustomData_get_layer_n(&dm->faceData, CD_MTFACE, l);
527                 mf += i;
528
529                 copy_v2_v2(df1->uv[0], mf->uv[c0]);
530                 INT_UV(df1->uv[1], c0, c1)
531                 INT_UV(df1->uv[2], c1, c2)
532                 INT_UV(df1->uv[3], c0, c2)
533
534                 INT_UV(df2->uv[0], c0, c1)
535                 copy_v2_v2(df2->uv[1], mf->uv[c1]);
536                 INT_UV(df2->uv[2], c1, c2)
537
538                 INT_UV(df2->uv[0], c0, c2)
539                 INT_UV(df2->uv[1], c1, c2)
540                 copy_v2_v2(df2->uv[2], mf->uv[c2]);
541         }
542 }
543
544 static DerivedMesh * cutEdges(ExplodeModifierData *emd, DerivedMesh *dm)
545 {
546         DerivedMesh *splitdm;
547         MFace *mf=NULL,*df1=NULL;
548         MFace *mface=dm->getFaceArray(dm);
549         MVert *dupve, *mv;
550         EdgeHash *edgehash;
551         EdgeHashIterator *ehi;
552         int totvert=dm->getNumVerts(dm);
553         int totface=dm->getNumFaces(dm);
554
555         int *facesplit = MEM_callocN(sizeof(int)*totface,"explode_facesplit");
556         int *vertpa = MEM_callocN(sizeof(int)*totvert,"explode_vertpa2");
557         int *facepa = emd->facepa;
558         int *fs, totesplit=0,totfsplit=0,curdupface=0;
559         int i, v1, v2, v3, v4, esplit,
560             v[4]  = {0, 0, 0, 0}, /* To quite gcc barking... */
561             uv[4] = {0, 0, 0, 0}; /* To quite gcc barking... */
562         int numlayer;
563         unsigned int ed_v1, ed_v2;
564
565         edgehash= BLI_edgehash_new();
566
567         /* recreate vertpa from facepa calculation */
568         for (i=0,mf=mface; i<totface; i++,mf++) {
569                 vertpa[mf->v1]=facepa[i];
570                 vertpa[mf->v2]=facepa[i];
571                 vertpa[mf->v3]=facepa[i];
572                 if(mf->v4)
573                         vertpa[mf->v4]=facepa[i];
574         }
575
576         /* mark edges for splitting and how to split faces */
577         for (i=0,mf=mface,fs=facesplit; i<totface; i++,mf++,fs++) {
578                 v1=vertpa[mf->v1];
579                 v2=vertpa[mf->v2];
580                 v3=vertpa[mf->v3];
581
582                 if(v1!=v2){
583                         BLI_edgehash_insert(edgehash, mf->v1, mf->v2, NULL);
584                         (*fs) |= 1;
585                 }
586
587                 if(v2!=v3){
588                         BLI_edgehash_insert(edgehash, mf->v2, mf->v3, NULL);
589                         (*fs) |= 2;
590                 }
591
592                 if(mf->v4){
593                         v4=vertpa[mf->v4];
594
595                         if(v3!=v4){
596                                 BLI_edgehash_insert(edgehash, mf->v3, mf->v4, NULL);
597                                 (*fs) |= 4;
598                         }
599
600                         if(v1!=v4){
601                                 BLI_edgehash_insert(edgehash, mf->v1, mf->v4, NULL);
602                                 (*fs) |= 8;
603                         }
604
605                         /* mark center vertex as a fake edge split */
606                         if(*fs == 15)
607                                 BLI_edgehash_insert(edgehash, mf->v1, mf->v3, NULL);
608                 }
609                 else {
610                         (*fs) |= 16; /* mark face as tri */
611
612                         if(v1!=v3){
613                                 BLI_edgehash_insert(edgehash, mf->v1, mf->v3, NULL);
614                                 (*fs) |= 4;
615                         }
616                 }
617         }
618
619         /* count splits & create indexes for new verts */
620         ehi= BLI_edgehashIterator_new(edgehash);
621         totesplit=totvert;
622         for(; !BLI_edgehashIterator_isDone(ehi); BLI_edgehashIterator_step(ehi)) {
623                 BLI_edgehashIterator_setValue(ehi, SET_INT_IN_POINTER(totesplit));
624                 totesplit++;
625         }
626         BLI_edgehashIterator_free(ehi);
627
628         /* count new faces due to splitting */
629         for(i=0,fs=facesplit; i<totface; i++,fs++)
630                 totfsplit += add_faces[*fs];
631         
632         splitdm= CDDM_from_template(dm, totesplit, 0, totface+totfsplit);
633         numlayer = CustomData_number_of_layers(&splitdm->faceData, CD_MTFACE);
634
635         /* copy new faces & verts (is it really this painful with custom data??) */
636         for(i=0; i<totvert; i++){
637                 MVert source;
638                 MVert *dest;
639                 dm->getVert(dm, i, &source);
640                 dest = CDDM_get_vert(splitdm, i);
641
642                 DM_copy_vert_data(dm, splitdm, i, i, 1);
643                 *dest = source;
644         }
645
646         /* override original facepa (original pointer is saved in caller function) */
647         facepa= MEM_callocN(sizeof(int)*(totface+totfsplit),"explode_facepa");
648         //memcpy(facepa,emd->facepa,totface*sizeof(int));
649         emd->facepa=facepa;
650
651         /* create new verts */
652         ehi= BLI_edgehashIterator_new(edgehash);
653         for(; !BLI_edgehashIterator_isDone(ehi); BLI_edgehashIterator_step(ehi)) {
654                 BLI_edgehashIterator_getKey(ehi, &ed_v1, &ed_v2);
655                 esplit= GET_INT_FROM_POINTER(BLI_edgehashIterator_getValue(ehi));
656                 mv=CDDM_get_vert(splitdm, ed_v2);
657                 dupve=CDDM_get_vert(splitdm,esplit);
658
659                 DM_copy_vert_data(splitdm,splitdm, ed_v2, esplit,1);
660
661                 *dupve=*mv;
662
663                 mv=CDDM_get_vert(splitdm, ed_v1);
664
665                 add_v3_v3(dupve->co, mv->co);
666                 mul_v3_fl(dupve->co, 0.5f);
667         }
668         BLI_edgehashIterator_free(ehi);
669
670         /* create new faces */
671         curdupface=0;//=totface;
672         //curdupin=totesplit;
673         for(i=0,fs=facesplit; i<totface; i++,fs++){
674                 mf = dm->getFaceData(dm, i, CD_MFACE);
675
676                 switch(*fs) {
677                 case 3:
678                 case 10:
679                 case 11:
680                 case 15:
681                         SET_VERTS(1, 2, 3, 4)
682                         break;
683                 case 5:
684                 case 6:
685                 case 7:
686                         SET_VERTS(2, 3, 4, 1)
687                         break;
688                 case 9:
689                 case 13:
690                         SET_VERTS(4, 1, 2, 3)
691                         break;
692                 case 12:
693                 case 14:
694                         SET_VERTS(3, 4, 1, 2)
695                         break;
696                 case 21:
697                 case 23:
698                         SET_VERTS(1, 2, 3, 4)
699                         break;
700                 case 19:
701                         SET_VERTS(2, 3, 1, 4)
702                         break;
703                 case 22:
704                         SET_VERTS(3, 1, 2, 4)
705                         break;
706                 }
707
708                 switch(*fs) {
709                 case 3:
710                 case 6:
711                 case 9:
712                 case 12:
713                         remap_faces_3_6_9_12(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2], v[3]);
714                         if(numlayer)
715                                 remap_uvs_3_6_9_12(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2], uv[3]);
716                         break;
717                 case 5:
718                 case 10:
719                         remap_faces_5_10(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2], v[3]);
720                         if(numlayer)
721                                 remap_uvs_5_10(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2], uv[3]);
722                         break;
723                 case 15:
724                         remap_faces_15(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2], v[3]);
725                         if(numlayer)
726                                 remap_uvs_15(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2], uv[3]);
727                         break;
728                 case 7:
729                 case 11:
730                 case 13:
731                 case 14:
732                         remap_faces_7_11_13_14(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2], v[3]);
733                         if(numlayer)
734                                 remap_uvs_7_11_13_14(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2], uv[3]);
735                         break;
736                 case 19:
737                 case 21:
738                 case 22:
739                         remap_faces_19_21_22(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2]);
740                         if(numlayer)
741                                 remap_uvs_19_21_22(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2]);
742                         break;
743                 case 23:
744                         remap_faces_23(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2]);
745                         if(numlayer)
746                                 remap_uvs_23(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2]);
747                         break;
748                 case 0:
749                 case 16:
750                         df1 = get_dface(dm, splitdm, curdupface, i, mf);
751                         facepa[curdupface] = vertpa[mf->v1];
752
753                         if(df1->v4)
754                                 df1->flag |= ME_FACE_SEL;
755                         else
756                                 df1->flag &= ~ME_FACE_SEL;
757                         break;
758                 }
759
760                 curdupface += add_faces[*fs]+1;
761         }
762
763         for(i=0; i<curdupface; i++) {
764                 mf = CDDM_get_face(splitdm, i);
765                 test_index_face(mf, &splitdm->faceData, i, (mf->flag & ME_FACE_SEL ? 4 : 3));
766         }
767
768         BLI_edgehash_free(edgehash, NULL);
769         MEM_freeN(facesplit);
770         MEM_freeN(vertpa);
771
772         return splitdm;
773
774 }
775 static DerivedMesh * explodeMesh(ExplodeModifierData *emd, 
776                 ParticleSystemModifierData *psmd, Scene *scene, Object *ob, 
777   DerivedMesh *to_explode)
778 {
779         DerivedMesh *explode, *dm=to_explode;
780         MFace *mf= NULL, *mface;
781         /* ParticleSettings *part=psmd->psys->part; */ /* UNUSED */
782         ParticleSimulationData sim= {NULL};
783         ParticleData *pa=NULL, *pars=psmd->psys->particles;
784         ParticleKey state, birth;
785         EdgeHash *vertpahash;
786         EdgeHashIterator *ehi;
787         float *vertco= NULL, imat[4][4];
788         float rot[4];
789         float cfra;
790         /* float timestep; */
791         int *facepa=emd->facepa;
792         int totdup=0,totvert=0,totface=0,totpart=0;
793         int i, v;
794         unsigned int ed_v1, ed_v2, mindex=0;
795         MTFace *mtface = NULL, *mtf;
796
797         totface= dm->getNumFaces(dm);
798         totvert= dm->getNumVerts(dm);
799         mface= dm->getFaceArray(dm);
800         totpart= psmd->psys->totpart;
801
802         sim.scene= scene;
803         sim.ob= ob;
804         sim.psys= psmd->psys;
805         sim.psmd= psmd;
806
807         /* timestep= psys_get_timestep(&sim); */
808
809         cfra= BKE_curframe(scene);
810
811         /* hash table for vertice <-> particle relations */
812         vertpahash= BLI_edgehash_new();
813
814         for (i=0; i<totface; i++) {
815                 /* do mindex + totvert to ensure the vertex index to be the first
816                  * with BLI_edgehashIterator_getKey */
817                 if(facepa[i]==totpart || cfra < (pars+facepa[i])->time)
818                         mindex = totvert+totpart;
819                 else 
820                         mindex = totvert+facepa[i];
821
822                 mf= &mface[i];
823
824                 /* set face vertices to exist in particle group */
825                 BLI_edgehash_insert(vertpahash, mf->v1, mindex, NULL);
826                 BLI_edgehash_insert(vertpahash, mf->v2, mindex, NULL);
827                 BLI_edgehash_insert(vertpahash, mf->v3, mindex, NULL);
828                 if(mf->v4)
829                         BLI_edgehash_insert(vertpahash, mf->v4, mindex, NULL);
830         }
831
832         /* make new vertice indexes & count total vertices after duplication */
833         ehi= BLI_edgehashIterator_new(vertpahash);
834         for(; !BLI_edgehashIterator_isDone(ehi); BLI_edgehashIterator_step(ehi)) {
835                 BLI_edgehashIterator_setValue(ehi, SET_INT_IN_POINTER(totdup));
836                 totdup++;
837         }
838         BLI_edgehashIterator_free(ehi);
839
840         /* the final duplicated vertices */
841         explode= CDDM_from_template(dm, totdup, 0,totface);
842         mtface = CustomData_get_layer_named(&explode->faceData, CD_MTFACE, emd->uvname);
843         /*dupvert= CDDM_get_verts(explode);*/
844
845         /* getting back to object space */
846         invert_m4_m4(imat,ob->obmat);
847
848         psmd->psys->lattice = psys_get_lattice(&sim);
849
850         /* duplicate & displace vertices */
851         ehi= BLI_edgehashIterator_new(vertpahash);
852         for(; !BLI_edgehashIterator_isDone(ehi); BLI_edgehashIterator_step(ehi)) {
853                 MVert source;
854                 MVert *dest;
855
856                 /* get particle + vertex from hash */
857                 BLI_edgehashIterator_getKey(ehi, &ed_v1, &ed_v2);
858                 ed_v2 -= totvert;
859                 v= GET_INT_FROM_POINTER(BLI_edgehashIterator_getValue(ehi));
860
861                 dm->getVert(dm, ed_v1, &source);
862                 dest = CDDM_get_vert(explode,v);
863
864                 DM_copy_vert_data(dm, explode, ed_v1, v, 1);
865                 *dest = source;
866
867                 if(ed_v2 != totpart) {
868                         /* get particle */
869                         pa= pars + ed_v2;
870
871                         psys_get_birth_coordinates(&sim, pa, &birth, 0, 0);
872
873                         state.time=cfra;
874                         psys_get_particle_state(&sim, ed_v2, &state, 1);
875
876                         vertco=CDDM_get_vert(explode,v)->co;
877                         mul_m4_v3(ob->obmat,vertco);
878
879                         sub_v3_v3(vertco, birth.co);
880
881                         /* apply rotation, size & location */
882                         sub_qt_qtqt(rot, state.rot, birth.rot);
883                         mul_qt_v3(rot, vertco);
884
885                         if(emd->flag & eExplodeFlag_PaSize)
886                                 mul_v3_fl(vertco,pa->size);
887
888                         add_v3_v3(vertco, state.co);
889
890                         mul_m4_v3(imat, vertco);
891                 }
892         }
893         BLI_edgehashIterator_free(ehi);
894
895         /*map new vertices to faces*/
896         for (i=0; i<totface; i++) {
897                 MFace source;
898                 int orig_v4;
899
900                 if(facepa[i]!=totpart)
901                 {
902                         pa=pars+facepa[i];
903
904                         if(pa->alive==PARS_UNBORN && (emd->flag&eExplodeFlag_Unborn)==0) continue;
905                         if(pa->alive==PARS_ALIVE && (emd->flag&eExplodeFlag_Alive)==0) continue;
906                         if(pa->alive==PARS_DEAD && (emd->flag&eExplodeFlag_Dead)==0) continue;
907                 }
908
909                 dm->getFace(dm,i,&source);
910                 mf=CDDM_get_face(explode,i);
911                 
912                 orig_v4 = source.v4;
913
914                 if(facepa[i]!=totpart && cfra < pa->time)
915                         mindex = totvert+totpart;
916                 else 
917                         mindex = totvert+facepa[i];
918
919                 source.v1 = edgecut_get(vertpahash, source.v1, mindex);
920                 source.v2 = edgecut_get(vertpahash, source.v2, mindex);
921                 source.v3 = edgecut_get(vertpahash, source.v3, mindex);
922                 if(source.v4)
923                         source.v4 = edgecut_get(vertpahash, source.v4, mindex);
924
925                 DM_copy_face_data(dm,explode,i,i,1);
926
927                 *mf = source;
928
929                 /* override uv channel for particle age */
930                 if(mtface) {
931                         float age = (cfra - pa->time)/pa->lifetime;
932                         /* Clamp to this range to avoid flipping to the other side of the coordinates. */
933                         CLAMP(age, 0.001f, 0.999f);
934
935                         mtf = mtface + i;
936
937                         mtf->uv[0][0] = mtf->uv[1][0] = mtf->uv[2][0] = mtf->uv[3][0] = age;
938                         mtf->uv[0][1] = mtf->uv[1][1] = mtf->uv[2][1] = mtf->uv[3][1] = 0.5f;
939                 }
940
941                 test_index_face(mf, &explode->faceData, i, (orig_v4 ? 4 : 3));
942         }
943
944         /* cleanup */
945         BLI_edgehash_free(vertpahash, NULL);
946
947         /* finalization */
948         CDDM_calc_edges(explode);
949         CDDM_calc_normals(explode);
950
951         if(psmd->psys->lattice){
952                 end_latt_deform(psmd->psys->lattice);
953                 psmd->psys->lattice= NULL;
954         }
955
956         return explode;
957 }
958
959 static ParticleSystemModifierData * findPrecedingParticlesystem(Object *ob, ModifierData *emd)
960 {
961         ModifierData *md;
962         ParticleSystemModifierData *psmd= NULL;
963
964         for (md=ob->modifiers.first; emd!=md; md=md->next){
965                 if(md->type==eModifierType_ParticleSystem)
966                         psmd= (ParticleSystemModifierData*) md;
967         }
968         return psmd;
969 }
970 static DerivedMesh * applyModifier(ModifierData *md, Object *ob,
971                                                 DerivedMesh *derivedData,
972                                                 int UNUSED(useRenderParams),
973                                                 int UNUSED(isFinalCalc))
974 {
975         DerivedMesh *dm = derivedData;
976         ExplodeModifierData *emd= (ExplodeModifierData*) md;
977         ParticleSystemModifierData *psmd=findPrecedingParticlesystem(ob,md);
978
979         if(psmd){
980                 ParticleSystem * psys=psmd->psys;
981
982                 if(psys==NULL || psys->totpart==0) return derivedData;
983                 if(psys->part==NULL || psys->particles==NULL) return derivedData;
984                 if(psmd->dm==NULL) return derivedData;
985
986                 /* 1. find faces to be exploded if needed */
987                 if(emd->facepa == NULL
988                                  || psmd->flag&eParticleSystemFlag_Pars
989                                  || emd->flag&eExplodeFlag_CalcFaces
990                                  || MEM_allocN_len(emd->facepa)/sizeof(int) != dm->getNumFaces(dm))
991                 {
992                         if(psmd->flag & eParticleSystemFlag_Pars)
993                                 psmd->flag &= ~eParticleSystemFlag_Pars;
994                         
995                         if(emd->flag & eExplodeFlag_CalcFaces)
996                                 emd->flag &= ~eExplodeFlag_CalcFaces;
997
998                         createFacepa(emd,psmd,derivedData);
999                 }
1000                 /* 2. create new mesh */
1001                 if(emd->flag & eExplodeFlag_EdgeCut){
1002                         int *facepa = emd->facepa;
1003                         DerivedMesh *splitdm=cutEdges(emd,dm);
1004                         DerivedMesh *explode=explodeMesh(emd, psmd, md->scene, ob, splitdm);
1005
1006                         MEM_freeN(emd->facepa);
1007                         emd->facepa=facepa;
1008                         splitdm->release(splitdm);
1009                         return explode;
1010                 }
1011                 else
1012                         return explodeMesh(emd, psmd, md->scene, ob, derivedData);
1013         }
1014         return derivedData;
1015 }
1016
1017
1018 ModifierTypeInfo modifierType_Explode = {
1019         /* name */              "Explode",
1020         /* structName */        "ExplodeModifierData",
1021         /* structSize */        sizeof(ExplodeModifierData),
1022         /* type */              eModifierTypeType_Constructive,
1023         /* flags */             eModifierTypeFlag_AcceptsMesh,
1024         /* copyData */          copyData,
1025         /* deformVerts */       NULL,
1026         /* deformMatrices */    NULL,
1027         /* deformVertsEM */     NULL,
1028         /* deformMatricesEM */  NULL,
1029         /* applyModifier */     applyModifier,
1030         /* applyModifierEM */   NULL,
1031         /* initData */          initData,
1032         /* requiredDataMask */  requiredDataMask,
1033         /* freeData */          freeData,
1034         /* isDisabled */        NULL,
1035         /* updateDepgraph */    NULL,
1036         /* dependsOnTime */     dependsOnTime,
1037         /* dependsOnNormals */  NULL,
1038         /* foreachObjectLink */ NULL,
1039         /* foreachIDLink */     NULL,
1040         /* foreachTexLink */    NULL,
1041 };