/** * $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. * * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. * * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV. * All rights reserved. * * Contributor(s): Blender Foundation, Campbell Barton * * ***** END GPL LICENSE BLOCK ***** */ #include #include #include "MEM_guardedalloc.h" #include "BLI_blenlib.h" #include "BLI_math.h" #include "BLI_heap.h" #include "BLI_edgehash.h" #include "BLI_editVert.h" #include "IMB_imbuf_types.h" #include "IMB_imbuf.h" #include "DNA_meshdata_types.h" #include "DNA_object_types.h" #include "DNA_scene_types.h" #include "BKE_DerivedMesh.h" #include "BKE_global.h" #include "BKE_mesh.h" #include "BKE_context.h" #include "BIF_gl.h" #ifndef DISABLE_PYTHON //#include "BPY_extern.h" //#include "BPY_menus.h" #endif #include "ED_mesh.h" #include "ED_screen.h" #include "ED_view3d.h" #include "WM_api.h" #include "WM_types.h" /* own include */ #include "mesh_intern.h" /* ***************** XXX **************** */ static int pupmenu(const char *UNUSED(dummy)) {return 0;} /* ***************** XXX **************** */ /* copy the face flags, most importantly selection from the mesh to the final derived mesh, * use in object mode when selecting faces (while painting) */ void object_facesel_flush_dm(Object *ob) { Mesh *me= get_mesh(ob); DerivedMesh *dm= ob->derivedFinal; MFace *faces, *mf, *mf_orig; int *index_array = NULL; int totface; int i; if(me==NULL || dm==NULL) return; index_array = dm->getFaceDataArray(dm, CD_ORIGINDEX); if(!index_array) return; faces = dm->getFaceArray(dm); totface = dm->getNumFaces(dm); mf= faces; for (i= 0; imface + index_array[i]; mf->flag= mf_orig->flag;; } } /* returns 0 if not found, otherwise 1 */ int facesel_face_pick(struct bContext *C, Mesh *me, short *mval, unsigned int *index, short rect) { ViewContext vc; view3d_set_viewcontext(C, &vc); if (!me || me->totface==0) return 0; // XXX if (v3d->flag & V3D_INVALID_BACKBUF) { // XXX drawview.c! check_backbuf(); // XXX persp(PERSP_VIEW); // XXX } if (rect) { /* sample rect to increase changes of selecting, so that when clicking on an edge in the backbuf, we can still select a face */ int dist; *index = view3d_sample_backbuf_rect(&vc, mval, 3, 1, me->totface+1, &dist,0,NULL, NULL); } else { /* sample only on the exact position */ *index = view3d_sample_backbuf(&vc, mval[0], mval[1]); } if ((*index)<=0 || (*index)>(unsigned int)me->totface) return 0; (*index)--; return 1; } /* last_sel, use em->act_face otherwise get the last selected face in the editselections * at the moment, last_sel is mainly useful for gaking sure the space image dosnt flicker */ MTFace *EM_get_active_mtface(EditMesh *em, EditFace **act_efa, MCol **mcol, int sloppy) { EditFace *efa = NULL; if(!EM_texFaceCheck(em)) return NULL; efa = EM_get_actFace(em, sloppy); if (efa) { if (mcol) { if (CustomData_has_layer(&em->fdata, CD_MCOL)) *mcol = CustomData_em_get(&em->fdata, efa->data, CD_MCOL); else *mcol = NULL; } if (act_efa) *act_efa = efa; return CustomData_em_get(&em->fdata, efa->data, CD_MTFACE); } if (act_efa) *act_efa= NULL; if(mcol) *mcol = NULL; return NULL; } void reveal_tface(Scene *scene) { Mesh *me; MFace *mface; int a; me= get_mesh(OBACT); if(me==0 || me->totface==0) return; mface= me->mface; a= me->totface; while(a--) { if(mface->flag & ME_HIDE) { mface->flag |= ME_FACE_SEL; mface->flag -= ME_HIDE; } mface++; } object_facesel_flush_dm(OBACT); // XXX notifier! object_tface_flags_changed(OBACT, 0); } void hide_tface(Scene *scene) { Mesh *me; MFace *mface; int a; int shift=0, alt= 0; // XXX me= get_mesh(OBACT); if(me==0 || me->totface==0) return; if(alt) { reveal_tface(scene); return; } mface= me->mface; a= me->totface; while(a--) { if(mface->flag & ME_HIDE); else { if(shift) { if( (mface->flag & ME_FACE_SEL)==0) mface->flag |= ME_HIDE; } else { if( (mface->flag & ME_FACE_SEL)) mface->flag |= ME_HIDE; } } if(mface->flag & ME_HIDE) mface->flag &= ~ME_FACE_SEL; mface++; } object_facesel_flush_dm(OBACT); // XXX notifier! object_tface_flags_changed(OBACT, 0); } /* Set tface seams based on edge data, uses hash table to find seam edges. */ static void hash_add_face(EdgeHash *ehash, MFace *mf) { BLI_edgehash_insert(ehash, mf->v1, mf->v2, NULL); BLI_edgehash_insert(ehash, mf->v2, mf->v3, NULL); if(mf->v4) { BLI_edgehash_insert(ehash, mf->v3, mf->v4, NULL); BLI_edgehash_insert(ehash, mf->v4, mf->v1, NULL); } else BLI_edgehash_insert(ehash, mf->v3, mf->v1, NULL); } void select_linked_tfaces_with_seams(int mode, Mesh *me, unsigned int index) { MFace *mf; int a, doit=1, mark=0; char *linkflag; EdgeHash *ehash, *seamhash; MEdge *med; ehash= BLI_edgehash_new(); seamhash = BLI_edgehash_new(); linkflag= MEM_callocN(sizeof(char)*me->totface, "linkflaguv"); for(med=me->medge, a=0; a < me->totedge; a++, med++) if(med->flag & ME_SEAM) BLI_edgehash_insert(seamhash, med->v1, med->v2, NULL); if (mode==0 || mode==1) { /* only put face under cursor in array */ mf= ((MFace*)me->mface) + index; hash_add_face(ehash, mf); linkflag[index]= 1; } else { /* fill array by selection */ mf= me->mface; for(a=0; atotface; a++, mf++) { if(mf->flag & ME_HIDE); else if(mf->flag & ME_FACE_SEL) { hash_add_face(ehash, mf); linkflag[a]= 1; } } } while(doit) { doit= 0; /* expand selection */ mf= me->mface; for(a=0; atotface; a++, mf++) { if(mf->flag & ME_HIDE) continue; if(!linkflag[a]) { mark= 0; if(!BLI_edgehash_haskey(seamhash, mf->v1, mf->v2)) if(BLI_edgehash_haskey(ehash, mf->v1, mf->v2)) mark= 1; if(!BLI_edgehash_haskey(seamhash, mf->v2, mf->v3)) if(BLI_edgehash_haskey(ehash, mf->v2, mf->v3)) mark= 1; if(mf->v4) { if(!BLI_edgehash_haskey(seamhash, mf->v3, mf->v4)) if(BLI_edgehash_haskey(ehash, mf->v3, mf->v4)) mark= 1; if(!BLI_edgehash_haskey(seamhash, mf->v4, mf->v1)) if(BLI_edgehash_haskey(ehash, mf->v4, mf->v1)) mark= 1; } else if(!BLI_edgehash_haskey(seamhash, mf->v3, mf->v1)) if(BLI_edgehash_haskey(ehash, mf->v3, mf->v1)) mark = 1; if(mark) { linkflag[a]= 1; hash_add_face(ehash, mf); doit= 1; } } } } BLI_edgehash_free(ehash, NULL); BLI_edgehash_free(seamhash, NULL); if(mode==0 || mode==2) { for(a=0, mf=me->mface; atotface; a++, mf++) if(linkflag[a]) mf->flag |= ME_FACE_SEL; else mf->flag &= ~ME_FACE_SEL; } else if(mode==1) { for(a=0, mf=me->mface; atotface; a++, mf++) if(linkflag[a] && (mf->flag & ME_FACE_SEL)) break; if (atotface) { for(a=0, mf=me->mface; atotface; a++, mf++) if(linkflag[a]) mf->flag &= ~ME_FACE_SEL; } else { for(a=0, mf=me->mface; atotface; a++, mf++) if(linkflag[a]) mf->flag |= ME_FACE_SEL; } } MEM_freeN(linkflag); // BIF_undo_push("Select linked UV face"); // object_tface_flags_changed(OBACT, 0); } void select_linked_tfaces(bContext *UNUSED(C), Object *ob, short UNUSED(mval[2]), int mode) { Mesh *me; unsigned int index=0; me = get_mesh(ob); if(me==0 || me->totface==0) return; if (mode==0 || mode==1) { // XXX - Causes glitches, not sure why /* if (!facesel_face_pick(C, me, mval, &index, 1)) return; */ } select_linked_tfaces_with_seams(mode, me, index); object_facesel_flush_dm(ob); } void selectall_tface(Object *ob, int action) { Mesh *me; MFace *mface; int a; me= get_mesh(ob); if(me==0) return; if (action == SEL_TOGGLE) { action = SEL_SELECT; mface= me->mface; a= me->totface; while(a--) { if((mface->flag & ME_HIDE) == 0 && mface->flag & ME_FACE_SEL) { action = SEL_DESELECT; break; } mface++; } } mface= me->mface; a= me->totface; while(a--) { if((mface->flag & ME_HIDE) == 0) { switch (action) { case SEL_SELECT: mface->flag |= ME_FACE_SEL; break; case SEL_DESELECT: mface->flag &= ~ME_FACE_SEL; break; case SEL_INVERT: mface->flag ^= ME_FACE_SEL; break; } } mface++; } object_facesel_flush_dm(ob); // XXX notifier! object_tface_flags_changed(OBACT, 0); } void selectswap_tface(Scene *scene) { Mesh *me; MFace *mface; int a; me= get_mesh(OBACT); if(me==0) return; mface= me->mface; a= me->totface; while(a--) { if(mface->flag & ME_HIDE); else { if(mface->flag & ME_FACE_SEL) mface->flag &= ~ME_FACE_SEL; else mface->flag |= ME_FACE_SEL; } mface++; } object_facesel_flush_dm(OBACT); // XXX notifier! object_tface_flags_changed(OBACT, 0); } int minmax_tface(Object *ob, float *min, float *max) { Mesh *me= get_mesh(ob); MFace *mf; MVert *mv; int a, ok=0; float vec[3]; if(me==NULL) return ok; mv= me->mvert; mf= me->mface; for (a=me->totface; a>0; a--, mf++) { if ((mf->flag & ME_HIDE || !(mf->flag & ME_FACE_SEL)) == 0) { int i= mf->v4 ? 3:2; do { mul_v3_m4v3(vec, ob->obmat, (mv + (*(&mf->v1 + i)))->co); DO_MINMAX(vec, min, max); } while (i--); ok= 1; } } return ok; } /* ******************** edge loop shortest path ********************* */ #define ME_SEAM_DONE 2 /* reuse this flag */ static float edgetag_cut_cost(int e1, int e2, int vert) { EditVert *v = EM_get_vert_for_index(vert); EditEdge *eed1 = EM_get_edge_for_index(e1), *eed2 = EM_get_edge_for_index(e2); EditVert *v1 = EM_get_vert_for_index( (eed1->v1->tmp.l == vert)? eed1->v2->tmp.l: eed1->v1->tmp.l ); EditVert *v2 = EM_get_vert_for_index( (eed2->v1->tmp.l == vert)? eed2->v2->tmp.l: eed2->v1->tmp.l ); float cost, d1[3], d2[3]; cost = len_v3v3(v1->co, v->co); cost += len_v3v3(v->co, v2->co); sub_v3_v3v3(d1, v->co, v1->co); sub_v3_v3v3(d2, v2->co, v->co); cost = cost + 0.5f*cost*(2.0f - fabs(d1[0]*d2[0] + d1[1]*d2[1] + d1[2]*d2[2])); return cost; } static void edgetag_add_adjacent(Heap *heap, int mednum, int vertnum, int *nedges, int *edges, int *prevedge, float *cost) { int startadj, endadj = nedges[vertnum+1]; for (startadj = nedges[vertnum]; startadj < endadj; startadj++) { int adjnum = edges[startadj]; EditEdge *eedadj = EM_get_edge_for_index(adjnum); float newcost; if (eedadj->f2 & ME_SEAM_DONE) continue; newcost = cost[mednum] + edgetag_cut_cost(mednum, adjnum, vertnum); if (cost[adjnum] > newcost) { cost[adjnum] = newcost; prevedge[adjnum] = mednum; BLI_heap_insert(heap, newcost, SET_INT_IN_POINTER(adjnum)); } } } void edgetag_context_set(Scene *scene, EditEdge *eed, int val) { switch (scene->toolsettings->edge_mode) { case EDGE_MODE_SELECT: EM_select_edge(eed, val); break; case EDGE_MODE_TAG_SEAM: if (val) {eed->seam = 255;} else {eed->seam = 0;} break; case EDGE_MODE_TAG_SHARP: if (val) {eed->sharp = 1;} else {eed->sharp = 0;} break; case EDGE_MODE_TAG_CREASE: if (val) {eed->crease = 1.0f;} else {eed->crease = 0.0f;} break; case EDGE_MODE_TAG_BEVEL: if (val) {eed->bweight = 1.0f;} else {eed->bweight = 0.0f;} break; } } int edgetag_context_check(Scene *scene, EditEdge *eed) { switch (scene->toolsettings->edge_mode) { case EDGE_MODE_SELECT: return (eed->f & SELECT) ? 1 : 0; case EDGE_MODE_TAG_SEAM: return eed->seam ? 1 : 0; case EDGE_MODE_TAG_SHARP: return eed->sharp ? 1 : 0; case EDGE_MODE_TAG_CREASE: return eed->crease ? 1 : 0; case EDGE_MODE_TAG_BEVEL: return eed->bweight ? 1 : 0; } return 0; } int edgetag_shortest_path(Scene *scene, EditMesh *em, EditEdge *source, EditEdge *target) { EditEdge *eed; EditVert *ev; Heap *heap; float *cost; int a, totvert=0, totedge=0, *nedges, *edges, *prevedge, mednum = -1, nedgeswap = 0; /* we need the vert */ for (ev= em->verts.first, totvert=0; ev; ev= ev->next) { ev->tmp.l = totvert; totvert++; } for (eed= em->edges.first; eed; eed = eed->next) { eed->f2 = 0; if (eed->h) { eed->f2 |= ME_SEAM_DONE; } eed->tmp.l = totedge; totedge++; } /* alloc */ nedges = MEM_callocN(sizeof(*nedges)*totvert+1, "SeamPathNEdges"); edges = MEM_mallocN(sizeof(*edges)*totedge*2, "SeamPathEdges"); prevedge = MEM_mallocN(sizeof(*prevedge)*totedge, "SeamPathPrevious"); cost = MEM_mallocN(sizeof(*cost)*totedge, "SeamPathCost"); /* count edges, compute adjacent edges offsets and fill adjacent edges */ for (eed= em->edges.first; eed; eed = eed->next) { nedges[eed->v1->tmp.l+1]++; nedges[eed->v2->tmp.l+1]++; } for (a=1; aedges.first; eed; a++, eed = eed->next) { edges[nedges[eed->v1->tmp.l+1]++] = a; edges[nedges[eed->v2->tmp.l+1]++] = a; cost[a] = 1e20f; prevedge[a] = -1; } /* regular dijkstra shortest path, but over edges instead of vertices */ heap = BLI_heap_new(); BLI_heap_insert(heap, 0.0f, SET_INT_IN_POINTER(source->tmp.l)); cost[source->tmp.l] = 0.0f; EM_init_index_arrays(em, 1, 1, 0); while (!BLI_heap_empty(heap)) { mednum = GET_INT_FROM_POINTER(BLI_heap_popmin(heap)); eed = EM_get_edge_for_index( mednum ); if (mednum == target->tmp.l) break; if (eed->f2 & ME_SEAM_DONE) continue; eed->f2 |= ME_SEAM_DONE; edgetag_add_adjacent(heap, mednum, eed->v1->tmp.l, nedges, edges, prevedge, cost); edgetag_add_adjacent(heap, mednum, eed->v2->tmp.l, nedges, edges, prevedge, cost); } MEM_freeN(nedges); MEM_freeN(edges); MEM_freeN(cost); BLI_heap_free(heap, NULL); for (eed= em->edges.first; eed; eed = eed->next) { eed->f2 &= ~ME_SEAM_DONE; } if (mednum != target->tmp.l) { MEM_freeN(prevedge); EM_free_index_arrays(); return 0; } /* follow path back to source and mark as seam */ if (mednum == target->tmp.l) { short allseams = 1; mednum = target->tmp.l; do { eed = EM_get_edge_for_index( mednum ); if (!edgetag_context_check(scene, eed)) { allseams = 0; break; } mednum = prevedge[mednum]; } while (mednum != source->tmp.l); mednum = target->tmp.l; do { eed = EM_get_edge_for_index( mednum ); if (allseams) edgetag_context_set(scene, eed, 0); else edgetag_context_set(scene, eed, 1); mednum = prevedge[mednum]; } while (mednum != -1); } MEM_freeN(prevedge); EM_free_index_arrays(); return 1; } /* *************************************** */ static void seam_edgehash_insert_face(EdgeHash *ehash, MFace *mf) { BLI_edgehash_insert(ehash, mf->v1, mf->v2, NULL); BLI_edgehash_insert(ehash, mf->v2, mf->v3, NULL); if (mf->v4) { BLI_edgehash_insert(ehash, mf->v3, mf->v4, NULL); BLI_edgehash_insert(ehash, mf->v4, mf->v1, NULL); } else BLI_edgehash_insert(ehash, mf->v3, mf->v1, NULL); } void seam_mark_clear_tface(Scene *scene, short mode) { Mesh *me; MFace *mf; MEdge *med; int a; me= get_mesh(OBACT); if(me==0 || me->totface==0) return; if (mode == 0) mode = pupmenu("Seams%t|Mark Border Seam %x1|Clear Seam %x2"); if (mode != 1 && mode != 2) return; if (mode == 2) { EdgeHash *ehash = BLI_edgehash_new(); for (a=0, mf=me->mface; atotface; a++, mf++) if (!(mf->flag & ME_HIDE) && (mf->flag & ME_FACE_SEL)) seam_edgehash_insert_face(ehash, mf); for (a=0, med=me->medge; atotedge; a++, med++) if (BLI_edgehash_haskey(ehash, med->v1, med->v2)) med->flag &= ~ME_SEAM; BLI_edgehash_free(ehash, NULL); } else { /* mark edges that are on both selected and deselected faces */ EdgeHash *ehash1 = BLI_edgehash_new(); EdgeHash *ehash2 = BLI_edgehash_new(); for (a=0, mf=me->mface; atotface; a++, mf++) { if ((mf->flag & ME_HIDE) || !(mf->flag & ME_FACE_SEL)) seam_edgehash_insert_face(ehash1, mf); else seam_edgehash_insert_face(ehash2, mf); } for (a=0, med=me->medge; atotedge; a++, med++) if (BLI_edgehash_haskey(ehash1, med->v1, med->v2) && BLI_edgehash_haskey(ehash2, med->v1, med->v2)) med->flag |= ME_SEAM; BLI_edgehash_free(ehash1, NULL); BLI_edgehash_free(ehash2, NULL); } // XXX if (G.rt == 8) // unwrap_lscm(1); me->drawflag |= ME_DRAWSEAMS; // XXX notifier! object_tface_flags_changed(OBACT, 1); } int face_select(struct bContext *C, Object *ob, short mval[2], int extend) { Mesh *me; MFace *mface, *msel; unsigned int a, index; /* Get the face under the cursor */ me = get_mesh(ob); if (!facesel_face_pick(C, me, mval, &index, 1)) return 0; msel= (((MFace*)me->mface)+index); if (msel->flag & ME_HIDE) return 0; /* clear flags */ mface = me->mface; a = me->totface; if (!extend) { while (a--) { mface->flag &= ~ME_FACE_SEL; mface++; } } me->act_face = (int)index; if (extend) { if (msel->flag & ME_FACE_SEL) msel->flag &= ~ME_FACE_SEL; else msel->flag |= ME_FACE_SEL; } else msel->flag |= ME_FACE_SEL; /* image window redraw */ object_facesel_flush_dm(ob); // XXX notifier! object_tface_flags_changed(OBACT, 1); WM_event_add_notifier(C, NC_GEOM|ND_SELECT, ob->data); ED_region_tag_redraw(CTX_wm_region(C)); // XXX - should redraw all 3D views return 1; } void face_borderselect(struct bContext *C, Object *ob, rcti *rect, int select, int extend) { Mesh *me; MFace *mface; struct ImBuf *ibuf; unsigned int *rt; int a, sx, sy, index; char *selar; ViewContext vc; view3d_set_viewcontext(C, &vc); me= get_mesh(ob); if(me==0) return; if(me->totface==0) return; selar= MEM_callocN(me->totface+1, "selar"); sx= (rect->xmax-rect->xmin+1); sy= (rect->ymax-rect->ymin+1); if(sx*sy<=0) return; if (extend == 0 && select) { mface= me->mface; for(a=1; a<=me->totface; a++, mface++) { if((mface->flag & ME_HIDE) == 0) mface->flag &= ~ME_FACE_SEL; } } view3d_validate_backbuf(&vc); ibuf = IMB_allocImBuf(sx,sy,32,IB_rect); rt = ibuf->rect; glReadPixels(rect->xmin+vc.ar->winrct.xmin, rect->ymin+vc.ar->winrct.ymin, sx, sy, GL_RGBA, GL_UNSIGNED_BYTE, ibuf->rect); if(ENDIAN_ORDER==B_ENDIAN) IMB_convert_rgba_to_abgr(ibuf); a= sx*sy; while(a--) { if(*rt) { index= WM_framebuffer_to_index(*rt); if(index<=me->totface) selar[index]= 1; } rt++; } mface= me->mface; for(a=1; a<=me->totface; a++, mface++) { if(selar[a]) { if(mface->flag & ME_HIDE); else { if(select) mface->flag |= ME_FACE_SEL; else mface->flag &= ~ME_FACE_SEL; } } } IMB_freeImBuf(ibuf); MEM_freeN(selar); // XXX notifier! object_tface_flags_changed(OBACT, 0); #ifdef __APPLE__ glReadBuffer(GL_BACK); #endif object_facesel_flush_dm(ob); }