/** * $Id: * * ***** BEGIN GPL LICENSE BLOCK ***** * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * as published by the Free Software Foundation; either version 2 * of the License, or (at your option) any later version. The Blender * Foundation also sells licenses for use in proprietary software under * the Blender License. See http://www.blender.org/BL/ for information * about this. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software Foundation, * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. * * Contributor(s): Joseph Eagar. * * ***** END GPL LICENSE BLOCK ***** */ #include "MEM_guardedalloc.h" #include "BKE_utildefines.h" #include "BKE_tessmesh.h" #include "BLI_math.h" #include "BLI_rand.h" #include "BLI_ghash.h" #include "BLI_array.h" #include "BLI_utildefines.h" #include "DNA_object_types.h" #include "ED_mesh.h" #include "bmesh.h" #include "bmesh_private.h" #include "mesh_intern.h" #include "subdivideop.h" #include #include #include #include /*flags for all elements share a common bitfield space*/ #define SUBD_SPLIT 1 #define EDGE_PERCENT 2 /*I don't think new faces are flagged, currently, but better safe than sorry.*/ #define FACE_NEW 4 #define FACE_CUSTOMFILL 8 #define ELE_INNER 16 #define ELE_SPLIT 32 #define ELE_CONNECT 64 /*stuff for the flag paramter. note that what used to live in "beauty" and in "seltype" live here. still have to convert the beauty flags over, which is why it starts at 128 (to avoid collision).*/ #define SELTYPE_INNER 128 /* NOTE: beauty has been renamed to flag! */ /*generic subdivision rules: * two selected edges in a face should make a link between them. * one edge should do, what? make pretty topology, or just split the edge only? */ #if 0 //misc. code, maps a parametric coordinate to a fractal line float lastrnd[3], vec2[3] = {0.0f, 0.0f, 0.0f}; int seed = BLI_rand(); int d, i, j, dp, lvl, wid; float df; BLI_srandom(seed); wid = (params->numcuts+2); dp = perc*wid; wid /= 2; d = lvl = 0; while (1) { if (d > dp) { d -= wid; } else if (d < dp) { d += wid; } else { break; } wid = MAX2((wid/2), 1); lvl++; } zero_v3(vec1); df = 1.0f; for (i=0; inumcuts/(float)tot); b = (float)(j+1)*(float)((float)params->numcuts/(float)tot); if (d >= a && d <= b) { interp_v3_v3v3(rnd2, lastrnd, rnd, (((float)d)-a)/(b-a)); mul_v3_fl(rnd2, df); add_v3_v3(vec1, rnd2); } copy_v3_v3(lastrnd, rnd); } } #endif /*connects face with smallest len, which I think should always be correct for edge subdivision*/ static BMEdge *connect_smallest_face(BMesh *bm, BMVert *v1, BMVert *v2, BMFace **nf) { BMIter iter, iter2; BMVert *v; BMLoop *nl; BMFace *face, *curf = NULL; /*this isn't the best thing in the world. it doesn't handle cases where there's multiple faces yet. that might require a convexity test to figure out which face is "best," and who knows what for non-manifold conditions.*/ for (face = BMIter_New(&iter, bm, BM_FACES_OF_VERT, v1); face; face=BMIter_Step(&iter)) { for (v=BMIter_New(&iter2, bm, BM_VERTS_OF_FACE, face); v; v=BMIter_Step(&iter2)) { if (v == v2) { if (!curf || face->len < curf->len) curf = face; } } } if (curf) { face = BM_Split_Face(bm, curf, v1, v2, &nl, NULL); if (nf) *nf = face; return nl ? nl->e : NULL; } return NULL; } /* calculates offset for co, based on fractal, sphere or smooth settings */ static void alter_co(BMesh *bm, BMVert *v, BMEdge *UNUSED(origed), subdparams *params, float perc, BMVert *vsta, BMVert *vend) { float vec1[3], fac; float *co=NULL, *origco=NULL; int i, totlayer = CustomData_number_of_layers(&bm->vdata, CD_SHAPEKEY); BM_Vert_UpdateAllNormals(bm, v); origco = CustomData_bmesh_get_n(&bm->vdata, v->head.data, CD_SHAPEKEY, params->origkey); sub_v3_v3v3(vec1, origco, v->co); for (i=0; ivdata, v->head.data, CD_SHAPEKEY, i); sub_v3_v3(co, vec1); } for (i=0; ivdata, v->head.data, CD_SHAPEKEY, i); if(params->beauty & B_SMOOTH) { /* we calculate an offset vector vec1[], to be added to *co */ float len, fac, nor[3], nor1[3], nor2[3], smooth=params->smooth; sub_v3_v3v3(nor, vsta->co, vend->co); len= 0.5f*normalize_v3(nor); copy_v3_v3(nor1, vsta->no); copy_v3_v3(nor2, vend->no); /* cosine angle */ fac= nor[0]*nor1[0] + nor[1]*nor1[1] + nor[2]*nor1[2] ; vec1[0]= fac*nor1[0]; vec1[1]= fac*nor1[1]; vec1[2]= fac*nor1[2]; /* cosine angle */ fac= -nor[0]*nor2[0] - nor[1]*nor2[1] - nor[2]*nor2[2] ; vec1[0]+= fac*nor2[0]; vec1[1]+= fac*nor2[1]; vec1[2]+= fac*nor2[2]; /* falloff for multi subdivide */ smooth *= sqrt(fabs(1.0f - 2.0f*fabs(perc))); vec1[0]*= smooth*len; vec1[1]*= smooth*len; vec1[2]*= smooth*len; co[0] += vec1[0]; co[1] += vec1[1]; co[2] += vec1[2]; } else if(params->beauty & B_SPHERE) { /* subdivide sphere */ normalize_v3(co); co[0]*= params->smooth; co[1]*= params->smooth; co[2]*= params->smooth; } if(params->beauty & B_FRACTAL) { float len = len_v3v3(vsta->co, vend->co); float vec2[3] = {0.0f, 0.0f, 0.0f}, co2[3]; fac= params->fractal*len; add_v3_v3(vec2, vsta->no); add_v3_v3(vec2, vend->no); mul_v3_fl(vec2, 0.5f); add_v3_v3v3(co2, v->co, params->off); vec1[0] = fac*(BLI_gTurbulence(1.0, co2[0], co2[1], co2[2], 15, 0, 1)-0.5f); vec1[1] = fac*(BLI_gTurbulence(1.0, co2[0], co2[1], co2[2], 15, 0, 1)-0.5f); vec1[2] = fac*(BLI_gTurbulence(1.0, co2[0], co2[1], co2[2], 15, 0, 1)-0.5f); mul_v3_v3(vec2, vec1); /*add displacement*/ add_v3_v3v3(co, co, vec2); } } } /* assumes in the edge is the correct interpolated vertices already */ /* percent defines the interpolation, rad and flag are for special options */ /* results in new vertex with correct coordinate, vertex normal and weight group info */ static BMVert *bm_subdivide_edge_addvert(BMesh *bm, BMEdge *edge,BMEdge *oedge, subdparams *params, float percent, float percent2, BMEdge **out,BMVert *vsta,BMVert *vend) { BMVert *ev; ev = BM_Split_Edge(bm, edge->v1, edge, out, percent); BMO_SetFlag(bm, ev, ELE_INNER); /* offset for smooth or sphere or fractal */ alter_co(bm, ev, oedge, params, percent2, vsta, vend); #if 0 //TODO /* clip if needed by mirror modifier */ if (edge->v1->f2) { if ( edge->v1->f2 & edge->v2->f2 & 1) { co[0]= 0.0f; } if ( edge->v1->f2 & edge->v2->f2 & 2) { co[1]= 0.0f; } if ( edge->v1->f2 & edge->v2->f2 & 4) { co[2]= 0.0f; } } #endif return ev; } static BMVert *subdivideedgenum(BMesh *bm, BMEdge *edge, BMEdge *oedge, int curpoint, int totpoint, subdparams *params, BMEdge **newe, BMVert *vsta, BMVert *vend) { BMVert *ev; float percent, percent2 = 0.0f; if (BMO_TestFlag(bm, edge, EDGE_PERCENT) && totpoint == 1) percent = BMO_Get_MapFloat(bm, params->op, "edgepercents", edge); else { percent= 1.0f/(float)(totpoint+1-curpoint); percent2 = (float)(curpoint+1) / (float)(totpoint+1); } ev= bm_subdivide_edge_addvert(bm, edge, oedge, params, percent, percent2, newe, vsta, vend); return ev; } static void bm_subdivide_multicut(BMesh *bm, BMEdge *edge, subdparams *params, BMVert *vsta, BMVert *vend) { BMEdge *eed = edge, *newe, temp = *edge; BMVert *v, ov1=*edge->v1, ov2=*edge->v2, *v1=edge->v1, *v2=edge->v2; int i, numcuts = params->numcuts; temp.v1 = &ov1; temp.v2 = &ov2; for(i=0;inumcuts, params, &newe, vsta, vend); BMO_SetFlag(bm, v, SUBD_SPLIT); BMO_SetFlag(bm, eed, SUBD_SPLIT); BMO_SetFlag(bm, newe, SUBD_SPLIT); BMO_SetFlag(bm, v, ELE_SPLIT); BMO_SetFlag(bm, eed, ELE_SPLIT); BMO_SetFlag(bm, newe, SUBD_SPLIT); CHECK_ELEMENT(bm, v); if (v->e) CHECK_ELEMENT(bm, v->e); if (v->e && v->e->l) CHECK_ELEMENT(bm, v->e->l->f); } alter_co(bm, v1, &temp, params, 0, &ov1, &ov2); alter_co(bm, v2, &temp, params, 1.0, &ov1, &ov2); } /*note: the patterns are rotated as necassary to match the input geometry. they're based on the pre-split state of the face*/ /* v3---------v2 | | | | | | | | v4---v0---v1 */ static void quad_1edge_split(BMesh *bm, BMFace *UNUSED(face), BMVert **verts, subdparams *params) { BMFace *nf; int i, add, numcuts = params->numcuts; /*if it's odd, the middle face is a quad, otherwise it's a triangle*/ if (numcuts % 2==0) { add = 2; for (i=0; inumcuts; for (i=0; inumcuts; lastv = verts[numcuts]; for (i=numcuts-1; i>=0; i--) { e = connect_smallest_face(bm, verts[i], verts[numcuts+(numcuts-i)], &nf); olde = *e; v = bm_subdivide_edge_addvert(bm, e, &olde, params, 0.5f, 0.5f, &ne, e->v1, e->v2); if (i != numcuts-1) connect_smallest_face(bm, lastv, v, &nf); lastv = v; } connect_smallest_face(bm, lastv, verts[numcuts*2+2], &nf); } static subdpattern quad_2edge_innervert = { {1, 1, 0, 0}, quad_2edge_split_innervert, 4, }; /* v6--------v5 | | | |v4s | |v3s | s s | v7-v0--v1-v2 */ static void quad_2edge_split_fan(BMesh *bm, BMFace *UNUSED(face), BMVert **verts, subdparams *params) { BMFace *nf; // BMVert *v; BMVert *lastv; // BMEdge *e, *ne; int i, numcuts = params->numcuts; lastv = verts[2]; for (i=0; inumcuts; for (i=0; inumcuts; int i, j, a, b, s=numcuts+2 /* , totv=numcuts*4+4 */; lines = MEM_callocN(sizeof(BMVert*)*(numcuts+2)*(numcuts+2), "q_4edge_split"); /*build a 2-dimensional array of verts, containing every vert (and all new ones) in the face.*/ /*first line*/ for (i=0; inumcuts; for (i=0; inumcuts; /*number of verts in each line*/ lines = MEM_callocN(sizeof(void*)*(numcuts+2), "triangle vert table"); lines[0] = (BMVert**) stackarr; lines[0][0] = verts[numcuts*2+1]; lines[1+numcuts] = MEM_callocN(sizeof(void*)*(numcuts+2), "triangle vert table 2"); for (i=0; ivdata, CD_SHAPEKEY); skey = CustomData_number_of_layers(&bmesh->vdata, CD_SHAPEKEY)-1; BM_ITER(v, &viter, bmesh, BM_VERTS_OF_MESH, NULL) { float *co = CustomData_bmesh_get_n(&bmesh->vdata, v->head.data, CD_SHAPEKEY, skey); copy_v3_v3(co, v->co); } /*first go through and tag edges*/ BMO_Flag_To_Slot(bmesh, op, "edges", SUBD_SPLIT, BM_EDGE); params.numcuts = numcuts; params.op = op; params.smooth = smooth; params.seed = seed; params.fractal = fractal; params.beauty = beauty; params.origkey = skey; params.off[0] = BLI_drand()*200.0f; params.off[1] = BLI_drand()*200.0f; params.off[2] = BLI_drand()*200.0f; BMO_Mapping_To_Flag(bmesh, op, "custompatterns", FACE_CUSTOMFILL); BMO_Mapping_To_Flag(bmesh, op, "edgepercents", EDGE_PERCENT); for (face=BMIter_New(&fiter, bmesh, BM_FACES_OF_MESH, NULL); face; face=BMIter_Step(&fiter)) { BMEdge *e1 = NULL, *e2 = NULL; float vec1[3], vec2[3]; /*figure out which pattern to use*/ BLI_array_empty(edges); BLI_array_empty(verts); matched = 0; i = 0; totesel = 0; for (nl=BMIter_New(&liter, bmesh, BM_LOOPS_OF_FACE, face); nl; nl=BMIter_Step(&liter)) { BLI_array_growone(edges); BLI_array_growone(verts); edges[i] = nl->e; verts[i] = nl->v; if (BMO_TestFlag(bmesh, edges[i], SUBD_SPLIT)) { if (!e1) e1 = edges[i]; else e2 = edges[i]; totesel++; } i++; } /*make sure the two edges have a valid angle to each other*/ if (totesel == 2 && (e1->v1 == e2->v1 || e1->v1 == e2->v2 || e1->v2 == e2->v1 || e1->v2 == e2->v1)) { float angle; sub_v3_v3v3(vec1, e1->v2->co, e1->v1->co); sub_v3_v3v3(vec2, e2->v2->co, e2->v1->co); normalize_v3(vec1); normalize_v3(vec2); angle = INPR(vec1, vec2); angle = ABS(angle); if (ABS(angle-1.0) < 0.01) totesel = 0; } if (BMO_TestFlag(bmesh, face, FACE_CUSTOMFILL)) { pat = BMO_Get_MapData(bmesh, op, "custompatterns", face); for (i=0; ilen; i++) { matched = 1; for (j=0; jlen; j++) { a = (j + i) % pat->len; if ((!!BMO_TestFlag(bmesh, edges[a], SUBD_SPLIT)) != (!!pat->seledges[j])) { matched = 0; break; } } if (matched) { BLI_array_growone(facedata); b = BLI_array_count(facedata)-1; facedata[b].pat = pat; facedata[b].start = verts[i]; facedata[b].face = face; facedata[b].totedgesel = totesel; BMO_SetFlag(bmesh, face, SUBD_SPLIT); break; } } /*obvously don't test for other patterns matching*/ continue; } for (i=0; ilen == face->len) { for (a=0; alen; a++) { matched = 1; for (b=0; blen; b++) { j = (b + a) % pat->len; if ((!!BMO_TestFlag(bmesh, edges[j], SUBD_SPLIT)) != (!!pat->seledges[b])) { matched = 0; break; } } if (matched) break; } if (matched) { BLI_array_growone(facedata); j = BLI_array_count(facedata) - 1; BMO_SetFlag(bmesh, face, SUBD_SPLIT); facedata[j].pat = pat; facedata[j].start = verts[a]; facedata[j].face = face; facedata[j].totedgesel = totesel; break; } } } if (!matched && totesel) { BLI_array_growone(facedata); j = BLI_array_count(facedata) - 1; BMO_SetFlag(bmesh, face, SUBD_SPLIT); facedata[j].totedgesel = totesel; facedata[j].face = face; } } einput = BMO_GetSlot(op, "edges"); /*go through and split edges*/ for (i=0; ilen; i++) { edge = ((BMEdge**)einput->data.p)[i]; bm_subdivide_multicut(bmesh, edge, ¶ms, edge->v1, edge->v2); } i = 0; for (i=0; iv, ELE_INNER) && BMO_TestFlag(bmesh, loops[a]->v, ELE_INNER)) break; } if (BMO_TestFlag(bmesh, loops[(a+numcuts+1)%vlen]->v, ELE_INNER)) { b = (a+numcuts+1)%vlen; } else { /*find the boundary of the other edge.*/ for (j=0; jv, ELE_INNER) && BMO_TestFlag(bmesh, loops[b]->v, ELE_INNER)) break; } } b += numcuts - 1; for (j=0; jv, splits[j*2+1]->v, &nl, NULL); } } continue; } else if (!pat) { continue; } j = a = 0; for (nl=BMIter_New(&liter, bmesh, BM_LOOPS_OF_FACE, face); nl; nl=BMIter_Step(&liter)) { if (nl->v == facedata[i].start) { a = j+1; break; } j++; } for (j=0; jlen; j++) { BLI_array_growone(verts); } j = 0; for (nl=BMIter_New(&liter, bmesh, BM_LOOPS_OF_FACE, face); nl; nl=BMIter_Step(&liter)) { b = (j-a+face->len) % face->len; verts[b] = nl->v; j += 1; } CHECK_ELEMENT(bmesh, face); pat->connectexec(bmesh, face, verts, ¶ms); } /*copy original-geometry displacements to current coordinates*/ BM_ITER(v, &viter, bmesh, BM_VERTS_OF_MESH, NULL) { float *co = CustomData_bmesh_get_n(&bmesh->vdata, v->head.data, CD_SHAPEKEY, skey); copy_v3_v3(v->co, co); } BM_free_data_layer_n(bmesh, &bmesh->vdata, CD_SHAPEKEY, skey); if (facedata) BLI_array_free(facedata); if (edges) BLI_array_free(edges); if (verts) BLI_array_free(verts); BLI_array_free(splits); BLI_array_free(loops); BMO_Flag_To_Slot(bmesh, op, "outinner", ELE_INNER, BM_ALL); BMO_Flag_To_Slot(bmesh, op, "outsplit", ELE_SPLIT, BM_ALL); BMO_Flag_To_Slot(bmesh, op, "geomout", ELE_INNER|ELE_SPLIT|SUBD_SPLIT, BM_ALL); } /*editmesh-emulating function*/ void BM_esubdivideflag(Object *UNUSED(obedit), BMesh *bm, int flag, float smooth, float fractal, int beauty, int numcuts, int seltype, int cornertype, int singleedge, int gridfill, int seed) { BMOperator op; BMO_InitOpf(bm, &op, "esubd edges=%he smooth=%f fractal=%f " "beauty=%d numcuts=%d quadcornertype=%d singleedge=%d " "gridfill=%d seed=%d", flag, smooth, fractal, beauty, numcuts, cornertype, singleedge, gridfill, seed); BMO_Exec_Op(bm, &op); if (seltype == SUBDIV_SELECT_INNER) { BMOIter iter; BMHeader *ele; // int i; ele = BMO_IterNew(&iter, bm, &op, "outinner", BM_EDGE|BM_VERT); for (; ele; ele=BMO_IterStep(&iter)) { BM_Select(bm, ele, 1); } } else if (seltype == SUBDIV_SELECT_LOOPCUT) { BMOIter iter; BMHeader *ele; // int i; /*deselect input*/ BM_clear_flag_all(bm, BM_SELECT); ele = BMO_IterNew(&iter, bm, &op, "outinner", BM_EDGE|BM_VERT); for (; ele; ele=BMO_IterStep(&iter)) { BM_Select(bm, ele, 1); if (ele->type == BM_VERT) { BMEdge *e; BMIter eiter; BM_ITER(e, &eiter, bm, BM_EDGES_OF_VERT, ele) { if (!BM_TestHFlag(e, BM_SELECT) && BM_TestHFlag(e->v1, BM_SELECT) && BM_TestHFlag(e->v2, BM_SELECT)) { BM_SetHFlag(e, BM_SELECT); bm->totedgesel += 1; } else if (BM_TestHFlag(e, BM_SELECT) && (!BM_TestHFlag(e->v1, BM_SELECT) || !BM_TestHFlag(e->v2, BM_SELECT))) { BM_ClearHFlag(e, BM_SELECT); bm->totedgesel -= 1; } } } } } BMO_Finish_Op(bm, &op); } void esplit_exec(BMesh *bm, BMOperator *op) { BMOIter siter; BMEdge *e; subdparams params; int skey; params.numcuts = BMO_GetSlot(op, "numcuts")->data.i; params.op = op; BM_add_data_layer(bm, &bm->vdata, CD_SHAPEKEY); skey = CustomData_number_of_layers(&bm->vdata, CD_SHAPEKEY)-1; params.origkey = skey; /*go through and split edges*/ BMO_ITER(e, &siter, bm, op, "edges", BM_EDGE) { bm_subdivide_multicut(bm, e, ¶ms, e->v1, e->v2); } BMO_Flag_To_Slot(bm, op, "outsplit", ELE_SPLIT, BM_ALL); BM_free_data_layer_n(bm, &bm->vdata, CD_SHAPEKEY, skey); }