svn merge ^/trunk/blender -r41226:41227 .
[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_utildefines.h"
40 #include "BLI_kdtree.h"
41 #include "BLI_rand.h"
42 #include "BLI_math.h"
43 #include "BLI_edgehash.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->getTessFaceArray(dm);
112         totface= dm->getNumTessFaces(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_tessface(split, cur);
202         DM_copy_tessface_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->getTessFaceArray(dm);
549         MVert *dupve, *mv;
550         EdgeHash *edgehash;
551         EdgeHashIterator *ehi;
552         int totvert=dm->getNumVerts(dm);
553         int totface=dm->getNumTessFaces(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,j,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
564         edgehash= BLI_edgehash_new();
565
566         /* recreate vertpa from facepa calculation */
567         for (i=0,mf=mface; i<totface; i++,mf++) {
568                 vertpa[mf->v1]=facepa[i];
569                 vertpa[mf->v2]=facepa[i];
570                 vertpa[mf->v3]=facepa[i];
571                 if(mf->v4)
572                         vertpa[mf->v4]=facepa[i];
573         }
574
575         /* mark edges for splitting and how to split faces */
576         for (i=0,mf=mface,fs=facesplit; i<totface; i++,mf++,fs++) {
577                 v1=vertpa[mf->v1];
578                 v2=vertpa[mf->v2];
579                 v3=vertpa[mf->v3];
580
581                 if(v1!=v2){
582                         BLI_edgehash_insert(edgehash, mf->v1, mf->v2, NULL);
583                         (*fs) |= 1;
584                 }
585
586                 if(v2!=v3){
587                         BLI_edgehash_insert(edgehash, mf->v2, mf->v3, NULL);
588                         (*fs) |= 2;
589                 }
590
591                 if(mf->v4){
592                         v4=vertpa[mf->v4];
593
594                         if(v3!=v4){
595                                 BLI_edgehash_insert(edgehash, mf->v3, mf->v4, NULL);
596                                 (*fs) |= 4;
597                         }
598
599                         if(v1!=v4){
600                                 BLI_edgehash_insert(edgehash, mf->v1, mf->v4, NULL);
601                                 (*fs) |= 8;
602                         }
603
604                         /* mark center vertex as a fake edge split */
605                         if(*fs == 15)
606                                 BLI_edgehash_insert(edgehash, mf->v1, mf->v3, NULL);
607                 }
608                 else {
609                         (*fs) |= 16; /* mark face as tri */
610
611                         if(v1!=v3){
612                                 BLI_edgehash_insert(edgehash, mf->v1, mf->v3, NULL);
613                                 (*fs) |= 4;
614                         }
615                 }
616         }
617
618         /* count splits & create indexes for new verts */
619         ehi= BLI_edgehashIterator_new(edgehash);
620         totesplit=totvert;
621         for(; !BLI_edgehashIterator_isDone(ehi); BLI_edgehashIterator_step(ehi)) {
622                 BLI_edgehashIterator_setValue(ehi, SET_INT_IN_POINTER(totesplit));
623                 totesplit++;
624         }
625         BLI_edgehashIterator_free(ehi);
626
627         /* count new faces due to splitting */
628         for(i=0,fs=facesplit; i<totface; i++,fs++)
629                 totfsplit += add_faces[*fs];
630         
631         splitdm= CDDM_from_template(dm, totesplit, 0, totface+totfsplit, 0, 0);
632         numlayer = CustomData_number_of_layers(&splitdm->faceData, CD_MTFACE);
633
634         /* copy new faces & verts (is it really this painful with custom data??) */
635         for(i=0; i<totvert; i++){
636                 MVert source;
637                 MVert *dest;
638                 dm->getVert(dm, i, &source);
639                 dest = CDDM_get_vert(splitdm, i);
640
641                 DM_copy_vert_data(dm, splitdm, i, i, 1);
642                 *dest = source;
643         }
644
645         /* override original facepa (original pointer is saved in caller function) */
646         facepa= MEM_callocN(sizeof(int)*(totface+totfsplit),"explode_facepa");
647         //memcpy(facepa,emd->facepa,totface*sizeof(int));
648         emd->facepa=facepa;
649
650         /* create new verts */
651         ehi= BLI_edgehashIterator_new(edgehash);
652         for(; !BLI_edgehashIterator_isDone(ehi); BLI_edgehashIterator_step(ehi)) {
653                 BLI_edgehashIterator_getKey(ehi, &i, &j);
654                 esplit= GET_INT_FROM_POINTER(BLI_edgehashIterator_getValue(ehi));
655                 mv=CDDM_get_vert(splitdm,j);
656                 dupve=CDDM_get_vert(splitdm,esplit);
657
658                 DM_copy_vert_data(splitdm,splitdm,j,esplit,1);
659
660                 *dupve=*mv;
661
662                 mv=CDDM_get_vert(splitdm,i);
663
664                 add_v3_v3(dupve->co, mv->co);
665                 mul_v3_fl(dupve->co, 0.5f);
666         }
667         BLI_edgehashIterator_free(ehi);
668
669         /* create new faces */
670         curdupface=0;//=totface;
671         //curdupin=totesplit;
672         for(i=0,fs=facesplit; i<totface; i++,fs++){
673                 mf = dm->getTessFaceData(dm, i, CD_MFACE);
674
675                 switch(*fs) {
676                 case 3:
677                 case 10:
678                 case 11:
679                 case 15:
680                         SET_VERTS(1, 2, 3, 4)
681                         break;
682                 case 5:
683                 case 6:
684                 case 7:
685                         SET_VERTS(2, 3, 4, 1)
686                         break;
687                 case 9:
688                 case 13:
689                         SET_VERTS(4, 1, 2, 3)
690                         break;
691                 case 12:
692                 case 14:
693                         SET_VERTS(3, 4, 1, 2)
694                         break;
695                 case 21:
696                 case 23:
697                         SET_VERTS(1, 2, 3, 4)
698                         break;
699                 case 19:
700                         SET_VERTS(2, 3, 1, 4)
701                         break;
702                 case 22:
703                         SET_VERTS(3, 1, 2, 4)
704                         break;
705                 }
706
707                 switch(*fs) {
708                 case 3:
709                 case 6:
710                 case 9:
711                 case 12:
712                         remap_faces_3_6_9_12(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2], v[3]);
713                         if(numlayer)
714                                 remap_uvs_3_6_9_12(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2], uv[3]);
715                         break;
716                 case 5:
717                 case 10:
718                         remap_faces_5_10(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2], v[3]);
719                         if(numlayer)
720                                 remap_uvs_5_10(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2], uv[3]);
721                         break;
722                 case 15:
723                         remap_faces_15(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2], v[3]);
724                         if(numlayer)
725                                 remap_uvs_15(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2], uv[3]);
726                         break;
727                 case 7:
728                 case 11:
729                 case 13:
730                 case 14:
731                         remap_faces_7_11_13_14(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2], v[3]);
732                         if(numlayer)
733                                 remap_uvs_7_11_13_14(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2], uv[3]);
734                         break;
735                 case 19:
736                 case 21:
737                 case 22:
738                         remap_faces_19_21_22(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2]);
739                         if(numlayer)
740                                 remap_uvs_19_21_22(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2]);
741                         break;
742                 case 23:
743                         remap_faces_23(dm, splitdm, mf, facepa, vertpa, i, edgehash, curdupface, v[0], v[1], v[2]);
744                         if(numlayer)
745                                 remap_uvs_23(dm, splitdm, numlayer, i, curdupface, uv[0], uv[1], uv[2]);
746                         break;
747                 case 0:
748                 case 16:
749                         df1 = get_dface(dm, splitdm, curdupface, i, mf);
750                         facepa[curdupface] = vertpa[mf->v1];
751
752                         if(df1->v4)
753                                 df1->flag |= ME_FACE_SEL;
754                         else
755                                 df1->flag &= ~ME_FACE_SEL;
756                         break;
757                 }
758
759                 curdupface += add_faces[*fs]+1;
760         }
761
762         for(i=0; i<curdupface; i++) {
763                 mf = CDDM_get_tessface(splitdm, i);
764                 test_index_face(mf, &splitdm->faceData, i, (mf->flag & ME_FACE_SEL ? 4 : 3));
765         }
766
767         BLI_edgehash_free(edgehash, NULL);
768         MEM_freeN(facesplit);
769         MEM_freeN(vertpa);
770
771         dm = CDDM_copy(splitdm, 1); /*builds ngon faces from tess (mface) faces*/
772         splitdm->needsFree = 1;
773         splitdm->release(splitdm);
774
775         return dm;
776 }
777 static DerivedMesh * explodeMesh(ExplodeModifierData *emd, 
778                 ParticleSystemModifierData *psmd, Scene *scene, Object *ob, 
779   DerivedMesh *to_explode)
780 {
781         DerivedMesh *explode, *dm=to_explode;
782         MFace *mf= NULL, *mface;
783         /* ParticleSettings *part=psmd->psys->part; */ /* UNUSED */
784         ParticleSimulationData sim= {NULL};
785         ParticleData *pa=NULL, *pars=psmd->psys->particles;
786         ParticleKey state, birth;
787         EdgeHash *vertpahash;
788         EdgeHashIterator *ehi;
789         float *vertco= NULL, imat[4][4];
790         float rot[4];
791         float cfra;
792         /* float timestep; */
793         int *facepa=emd->facepa;
794         int totdup=0,totvert=0,totface=0,totpart=0;
795         int i, j, v, mindex=0;
796         MTFace *mtface = NULL, *mtf;
797
798         totface= dm->getNumTessFaces(dm);
799         totvert= dm->getNumVerts(dm);
800         mface= dm->getTessFaceArray(dm);
801         totpart= psmd->psys->totpart;
802
803         sim.scene= scene;
804         sim.ob= ob;
805         sim.psys= psmd->psys;
806         sim.psmd= psmd;
807
808         /* timestep= psys_get_timestep(&sim); */
809
810         //if(part->flag & PART_GLOB_TIME)
811                 cfra= BKE_curframe(scene);
812         //else
813         //      cfra=bsystem_time(scene, ob,(float)scene->r.cfra,0.0);
814
815         /* hash table for vertice <-> particle relations */
816         vertpahash= BLI_edgehash_new();
817
818         for (i=0; i<totface; i++) {
819                 /* do mindex + totvert to ensure the vertex index to be the first
820                  * with BLI_edgehashIterator_getKey */
821                 if(facepa[i]==totpart || cfra < (pars+facepa[i])->time)
822                         mindex = totvert+totpart;
823                 else 
824                         mindex = totvert+facepa[i];
825
826                 mf= &mface[i];
827
828                 /* set face vertices to exist in particle group */
829                 BLI_edgehash_insert(vertpahash, mf->v1, mindex, NULL);
830                 BLI_edgehash_insert(vertpahash, mf->v2, mindex, NULL);
831                 BLI_edgehash_insert(vertpahash, mf->v3, mindex, NULL);
832                 if(mf->v4)
833                         BLI_edgehash_insert(vertpahash, mf->v4, mindex, NULL);
834         }
835
836         /* make new vertice indexes & count total vertices after duplication */
837         ehi= BLI_edgehashIterator_new(vertpahash);
838         for(; !BLI_edgehashIterator_isDone(ehi); BLI_edgehashIterator_step(ehi)) {
839                 BLI_edgehashIterator_setValue(ehi, SET_INT_IN_POINTER(totdup));
840                 totdup++;
841         }
842         BLI_edgehashIterator_free(ehi);
843
844         /* the final duplicated vertices */
845         explode= CDDM_from_template(dm, totdup, 0,totface, 0, 0);
846         mtface = CustomData_get_layer_named(&explode->faceData, CD_MTFACE, emd->uvname);
847         /*dupvert= CDDM_get_verts(explode);*/
848
849         /* getting back to object space */
850         invert_m4_m4(imat,ob->obmat);
851
852         psmd->psys->lattice = psys_get_lattice(&sim);
853
854         /* duplicate & displace vertices */
855         ehi= BLI_edgehashIterator_new(vertpahash);
856         for(; !BLI_edgehashIterator_isDone(ehi); BLI_edgehashIterator_step(ehi)) {
857                 MVert source;
858                 MVert *dest;
859
860                 /* get particle + vertex from hash */
861                 BLI_edgehashIterator_getKey(ehi, &j, &i);
862                 i -= totvert;
863                 v= GET_INT_FROM_POINTER(BLI_edgehashIterator_getValue(ehi));
864
865                 dm->getVert(dm, j, &source);
866                 dest = CDDM_get_vert(explode,v);
867
868                 DM_copy_vert_data(dm,explode,j,v,1);
869                 *dest = source;
870
871                 if(i!=totpart) {
872                         /* get particle */
873                         pa= pars+i;
874
875                         psys_get_birth_coordinates(&sim, pa, &birth, 0, 0);
876
877                         state.time=cfra;
878                         psys_get_particle_state(&sim, i, &state, 1);
879
880                         vertco=CDDM_get_vert(explode,v)->co;
881                         mul_m4_v3(ob->obmat,vertco);
882
883                         sub_v3_v3(vertco, birth.co);
884
885                         /* apply rotation, size & location */
886                         sub_qt_qtqt(rot, state.rot, birth.rot);
887                         mul_qt_v3(rot, vertco);
888
889                         if(emd->flag & eExplodeFlag_PaSize)
890                                 mul_v3_fl(vertco,pa->size);
891
892                         add_v3_v3(vertco, state.co);
893
894                         mul_m4_v3(imat, vertco);
895                 }
896         }
897         BLI_edgehashIterator_free(ehi);
898
899         /*map new vertices to faces*/
900         for (i=0; i<totface; i++) {
901                 MFace source;
902                 int orig_v4;
903
904                 if(facepa[i]!=totpart)
905                 {
906                         pa=pars+facepa[i];
907
908                         if(pa->alive==PARS_UNBORN && (emd->flag&eExplodeFlag_Unborn)==0) continue;
909                         if(pa->alive==PARS_ALIVE && (emd->flag&eExplodeFlag_Alive)==0) continue;
910                         if(pa->alive==PARS_DEAD && (emd->flag&eExplodeFlag_Dead)==0) continue;
911                 }
912
913                 dm->getTessFace(dm,i,&source);
914                 mf=CDDM_get_tessface(explode,i);
915                 
916                 orig_v4 = source.v4;
917
918                 if(facepa[i]!=totpart && cfra < pa->time)
919                         mindex = totvert+totpart;
920                 else 
921                         mindex = totvert+facepa[i];
922
923                 source.v1 = edgecut_get(vertpahash, source.v1, mindex);
924                 source.v2 = edgecut_get(vertpahash, source.v2, mindex);
925                 source.v3 = edgecut_get(vertpahash, source.v3, mindex);
926                 if(source.v4)
927                         source.v4 = edgecut_get(vertpahash, source.v4, mindex);
928
929                 DM_copy_tessface_data(dm,explode,i,i,1);
930
931                 *mf = source;
932
933                 /* override uv channel for particle age */
934                 if(mtface) {
935                         float age = (cfra - pa->time)/pa->lifetime;
936                         /* Clamp to this range to avoid flipping to the other side of the coordinates. */
937                         CLAMP(age, 0.001f, 0.999f);
938
939                         mtf = mtface + i;
940
941                         mtf->uv[0][0] = mtf->uv[1][0] = mtf->uv[2][0] = mtf->uv[3][0] = age;
942                         mtf->uv[0][1] = mtf->uv[1][1] = mtf->uv[2][1] = mtf->uv[3][1] = 0.5f;
943                 }
944
945                 test_index_face(mf, &explode->faceData, i, (orig_v4 ? 4 : 3));
946         }
947
948         /* cleanup */
949         BLI_edgehash_free(vertpahash, NULL);
950
951         /* finalization */
952         CDDM_calc_edges(explode);
953         CDDM_tessfaces_to_faces(explode);
954         CDDM_calc_normals(explode);
955
956         if(psmd->psys->lattice){
957                 end_latt_deform(psmd->psys->lattice);
958                 psmd->psys->lattice= NULL;
959         }
960
961         return explode;
962 }
963
964 static ParticleSystemModifierData * findPrecedingParticlesystem(Object *ob, ModifierData *emd)
965 {
966         ModifierData *md;
967         ParticleSystemModifierData *psmd= NULL;
968
969         for (md=ob->modifiers.first; emd!=md; md=md->next){
970                 if(md->type==eModifierType_ParticleSystem)
971                         psmd= (ParticleSystemModifierData*) md;
972         }
973         return psmd;
974 }
975 static DerivedMesh * applyModifier(ModifierData *md, Object *ob,
976                                                 DerivedMesh *derivedData,
977                                                 int UNUSED(useRenderParams),
978                                                 int UNUSED(isFinalCalc))
979 {
980         DerivedMesh *dm = derivedData;
981         ExplodeModifierData *emd= (ExplodeModifierData*) md;
982         ParticleSystemModifierData *psmd=findPrecedingParticlesystem(ob,md);
983
984         if(psmd){
985                 ParticleSystem * psys=psmd->psys;
986
987                 if(psys==NULL || psys->totpart==0) return derivedData;
988                 if(psys->part==NULL || psys->particles==NULL) return derivedData;
989                 if(psmd->dm==NULL) return derivedData;
990
991                 /* 1. find faces to be exploded if needed */
992                 if(emd->facepa == NULL
993                                  || psmd->flag&eParticleSystemFlag_Pars
994                                  || emd->flag&eExplodeFlag_CalcFaces
995                                  || MEM_allocN_len(emd->facepa)/sizeof(int) != dm->getNumTessFaces(dm))
996                 {
997                         if(psmd->flag & eParticleSystemFlag_Pars)
998                                 psmd->flag &= ~eParticleSystemFlag_Pars;
999                         
1000                         if(emd->flag & eExplodeFlag_CalcFaces)
1001                                 emd->flag &= ~eExplodeFlag_CalcFaces;
1002
1003                         createFacepa(emd,psmd,derivedData);
1004                 }
1005                 /* 2. create new mesh */
1006                 if(emd->flag & eExplodeFlag_EdgeCut){
1007                         int *facepa = emd->facepa;
1008                         DerivedMesh *splitdm=cutEdges(emd,dm);
1009                         DerivedMesh *explode=explodeMesh(emd, psmd, md->scene, ob, splitdm);
1010
1011                         MEM_freeN(emd->facepa);
1012                         emd->facepa=facepa;
1013                         splitdm->release(splitdm);
1014                         return explode;
1015                 }
1016                 else
1017                         return explodeMesh(emd, psmd, md->scene, ob, derivedData);
1018         }
1019         return derivedData;
1020 }
1021
1022
1023 ModifierTypeInfo modifierType_Explode = {
1024         /* name */              "Explode",
1025         /* structName */        "ExplodeModifierData",
1026         /* structSize */        sizeof(ExplodeModifierData),
1027         /* type */              eModifierTypeType_Constructive,
1028         /* flags */             eModifierTypeFlag_AcceptsMesh,
1029         /* copyData */          copyData,
1030         /* deformVerts */       NULL,
1031         /* deformMatrices */    NULL,
1032         /* deformVertsEM */     NULL,
1033         /* deformMatricesEM */  NULL,
1034         /* applyModifier */     applyModifier,
1035         /* applyModifierEM */   NULL,
1036         /* initData */          initData,
1037         /* requiredDataMask */  requiredDataMask,
1038         /* freeData */          freeData,
1039         /* isDisabled */        NULL,
1040         /* updateDepgraph */    NULL,
1041         /* dependsOnTime */     dependsOnTime,
1042         /* dependsOnNormals */  NULL,
1043         /* foreachObjectLink */ NULL,
1044         /* foreachIDLink */     NULL,
1045         /* foreachTexLink */    NULL,
1046 };