archive/blender-archive
Archived
1
1
Fork 0
Code Pull Requests Packages Activity
This repository has been archived on 2023-10-09. You can view files and clone it, but cannot push or open issues or pull requests.
Files
1be4eda552aa97aa5f5fe43fd34196c57425a7c2
blender-archive/source/blender/editors/mesh/editmesh_lib.c

2768 lines
67 KiB
C
Raw Blame History

/**
* $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) 2004 by Blender Foundation
* All rights reserved.
*
* The Original Code is: all of this file.
*
* Contributor(s): none yet.
*
* ***** END GPL LICENSE BLOCK *****
*/
/*
editmesh_lib: generic (no UI, no menus) operations/evaluators for editmesh data
*/
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include "MEM_guardedalloc.h"
#include "DNA_meshdata_types.h"
#include "DNA_modifier_types.h"
#include "DNA_object_types.h"
#include "DNA_scene_types.h"
#include "BLI_blenlib.h"
#include "BLI_math.h"
#include "BLI_editVert.h"
#include "BLI_edgehash.h"
#include "BKE_customdata.h"
#include "BKE_context.h"
#include "BKE_global.h"
#include "BKE_mesh.h"
#include "BKE_utildefines.h"
#include "ED_mesh.h"
#include "ED_screen.h"
#include "ED_view3d.h"
#include "mesh_intern.h"
/* Helpers for EM_set_flag_all_selectmode */
#define SET_EVE_FLAG(eve, flag) \
if (eve->h==0) { \
if (flag & SELECT && !(eve->f & SELECT)) { \
++selvert; \
} \
eve->f |= flag; \
}
#define SET_EED_FLAG(eed, flag) \
if (eed->h==0) { \
if (flag & SELECT && !(eed->f & SELECT)) { \
++seledge; \
} \
eed->f |= flag; \
SET_EVE_FLAG(eed->v1, flag); \
SET_EVE_FLAG(eed->v2, flag); \
}
/* ****************** stats *************** */
int EM_nfaces_selected(EditMesh *em)
{
EditFace *efa;
int count= 0;
for (efa= em->faces.first; efa; efa= efa->next)
if (efa->f & SELECT)
count++;
em->totfacesel= count;
return count;
}
int EM_nedges_selected(EditMesh *em)
{
EditEdge *eed;
int count= 0;
for (eed= em->edges.first; eed; eed= eed->next)
if(eed->f & SELECT)
count++;
em->totedgesel= count;
return count;
}
int EM_nvertices_selected(EditMesh *em)
{
EditVert *eve;
int count= 0;
for (eve= em->verts.first; eve; eve= eve->next)
if (eve->f & SELECT)
count++;
em->totvertsel= count;
return count;
}
void EM_stats_update(EditMesh *em)
{
em->totvert= BLI_countlist(&em->verts);
em->totedge= BLI_countlist(&em->edges);
em->totface= BLI_countlist(&em->faces);
EM_nvertices_selected(em);
EM_nedges_selected(em);
EM_nfaces_selected(em);
}
/* ************************************** */
/* this replaces the active flag used in uv/face mode */
void EM_set_actFace(EditMesh *em, EditFace *efa)
{
em->act_face = efa;
}
EditFace *EM_get_actFace(EditMesh *em, int sloppy)
{
if (em->act_face) {
return em->act_face;
} else if (sloppy) {
EditFace *efa= NULL;
EditSelection *ese;
ese = em->selected.last;
for (; ese; ese=ese->prev){
if(ese->type == EDITFACE) {
efa = (EditFace *)ese->data;
if (efa->h) efa= NULL;
else break;
}
}
if (efa==NULL) {
for (efa= em->faces.first; efa; efa= efa->next) {
if (efa->f & SELECT)
break;
}
}
return efa; /* can still be null */
}
return NULL;
}
int EM_get_actSelection(EditMesh *em, EditSelection *ese)
{
EditSelection *ese_last = em->selected.last;
EditFace *efa = EM_get_actFace(em, 0);
ese->next = ese->prev = NULL;
if (ese_last) {
if (ese_last->type == EDITFACE) { /* if there is an active face, use it over the last selected face */
if (efa) {
ese->data = (void *)efa;
} else {
ese->data = ese_last->data;
}
ese->type = EDITFACE;
} else {
ese->data = ese_last->data;
ese->type = ese_last->type;
}
} else if (efa) { /* no */
ese->data = (void *)efa;
ese->type = EDITFACE;
} else {
ese->data = NULL;
return 0;
}
return 1;
}
/* ********* Selection History ************ */
static int EM_check_selection(EditMesh *em, void *data)
{
EditSelection *ese;
for(ese = em->selected.first; ese; ese = ese->next){
if(ese->data == data) return 1;
}
return 0;
}
void EM_remove_selection(EditMesh *em, void *data, int type)
{
EditSelection *ese;
for(ese=em->selected.first; ese; ese = ese->next){
if(ese->data == data){
BLI_freelinkN(&(em->selected),ese);
break;
}
}
}
void EM_store_selection(EditMesh *em, void *data, int type)
{
EditSelection *ese;
if(!EM_check_selection(em, data)){
ese = (EditSelection*) MEM_callocN( sizeof(EditSelection), "Edit Selection");
ese->type = type;
ese->data = data;
BLI_addtail(&(em->selected),ese);
}
}
void EM_validate_selections(EditMesh *em)
{
EditSelection *ese, *nextese;
ese = em->selected.first;
while(ese){
nextese = ese->next;
if(ese->type == EDITVERT && !(((EditVert*)ese->data)->f & SELECT)) BLI_freelinkN(&(em->selected), ese);
else if(ese->type == EDITEDGE && !(((EditEdge*)ese->data)->f & SELECT)) BLI_freelinkN(&(em->selected), ese);
else if(ese->type == EDITFACE && !(((EditFace*)ese->data)->f & SELECT)) BLI_freelinkN(&(em->selected), ese);
ese = nextese;
}
}
static void EM_strip_selections(EditMesh *em)
{
EditSelection *ese, *nextese;
if(!(em->selectmode & SCE_SELECT_VERTEX)){
ese = em->selected.first;
while(ese){
nextese = ese->next;
if(ese->type == EDITVERT) BLI_freelinkN(&(em->selected),ese);
ese = nextese;
}
}
if(!(em->selectmode & SCE_SELECT_EDGE)){
ese=em->selected.first;
while(ese){
nextese = ese->next;
if(ese->type == EDITEDGE) BLI_freelinkN(&(em->selected), ese);
ese = nextese;
}
}
if(!(em->selectmode & SCE_SELECT_FACE)){
ese=em->selected.first;
while(ese){
nextese = ese->next;
if(ese->type == EDITFACE) BLI_freelinkN(&(em->selected), ese);
ese = nextese;
}
}
}
/* generic way to get data from an EditSelection type
These functions were written to be used by the Modifier widget when in Rotate about active mode,
but can be used anywhere.
EM_editselection_center
EM_editselection_normal
EM_editselection_plane
*/
void EM_editselection_center(float *center, EditSelection *ese)
{
if (ese->type==EDITVERT) {
EditVert *eve= ese->data;
copy_v3_v3(center, eve->co);
} else if (ese->type==EDITEDGE) {
EditEdge *eed= ese->data;
add_v3_v3v3(center, eed->v1->co, eed->v2->co);
mul_v3_fl(center, 0.5);
} else if (ese->type==EDITFACE) {
EditFace *efa= ese->data;
copy_v3_v3(center, efa->cent);
}
}
void EM_editselection_normal(float *normal, EditSelection *ese)
{
if (ese->type==EDITVERT) {
EditVert *eve= ese->data;
copy_v3_v3(normal, eve->no);
} else if (ese->type==EDITEDGE) {
EditEdge *eed= ese->data;
float plane[3]; /* need a plane to correct the normal */
float vec[3]; /* temp vec storage */
add_v3_v3v3(normal, eed->v1->no, eed->v2->no);
sub_v3_v3v3(plane, eed->v2->co, eed->v1->co);
/* the 2 vertex normals will be close but not at rightangles to the edge
for rotate about edge we want them to be at right angles, so we need to
do some extra colculation to correct the vert normals,
we need the plane for this */
cross_v3_v3v3(vec, normal, plane);
cross_v3_v3v3(normal, plane, vec);
normalize_v3(normal);
} else if (ese->type==EDITFACE) {
EditFace *efa= ese->data;
copy_v3_v3(normal, efa->n);
}
}
/* Calculate a plane that is rightangles to the edge/vert/faces normal
also make the plane run allong an axis that is related to the geometry,
because this is used for the manipulators Y axis.*/
void EM_editselection_plane(float *plane, EditSelection *ese)
{
if (ese->type==EDITVERT) {
EditVert *eve= ese->data;
float vec[3]={0,0,0};
if (ese->prev) { /*use previously selected data to make a usefull vertex plane */
EM_editselection_center(vec, ese->prev);
sub_v3_v3v3(plane, vec, eve->co);
} else {
/* make a fake plane thats at rightangles to the normal
we cant make a crossvec from a vec thats the same as the vec
unlikely but possible, so make sure if the normal is (0,0,1)
that vec isnt the same or in the same direction even.*/
if (eve->no[0]<0.5) vec[0]=1;
else if (eve->no[1]<0.5) vec[1]=1;
else vec[2]=1;
cross_v3_v3v3(plane, eve->no, vec);
}
} else if (ese->type==EDITEDGE) {
EditEdge *eed= ese->data;
/*the plane is simple, it runs allong the edge
however selecting different edges can swap the direction of the y axis.
this makes it less likely for the y axis of the manipulator
(running along the edge).. to flip less often.
at least its more pradictable */
if (eed->v2->co[1] > eed->v1->co[1]) /*check which to do first */
sub_v3_v3v3(plane, eed->v2->co, eed->v1->co);
else
sub_v3_v3v3(plane, eed->v1->co, eed->v2->co);
} else if (ese->type==EDITFACE) {
EditFace *efa= ese->data;
float vec[3];
if (efa->v4) { /*if its a quad- set the plane along the 2 longest edges.*/
float vecA[3], vecB[3];
sub_v3_v3v3(vecA, efa->v4->co, efa->v3->co);
sub_v3_v3v3(vecB, efa->v1->co, efa->v2->co);
add_v3_v3v3(plane, vecA, vecB);
sub_v3_v3v3(vecA, efa->v1->co, efa->v4->co);
sub_v3_v3v3(vecB, efa->v2->co, efa->v3->co);
add_v3_v3v3(vec, vecA, vecB);
/*use the biggest edge length*/
if (plane[0]*plane[0]+plane[1]*plane[1]+plane[2]*plane[2] < vec[0]*vec[0]+vec[1]*vec[1]+vec[2]*vec[2])
copy_v3_v3(plane, vec);
} else {
/*start with v1-2 */
sub_v3_v3v3(plane, efa->v1->co, efa->v2->co);
/*test the edge between v2-3, use if longer */
sub_v3_v3v3(vec, efa->v2->co, efa->v3->co);
if (plane[0]*plane[0]+plane[1]*plane[1]+plane[2]*plane[2] < vec[0]*vec[0]+vec[1]*vec[1]+vec[2]*vec[2])
copy_v3_v3(plane, vec);
/*test the edge between v1-3, use if longer */
sub_v3_v3v3(vec, efa->v3->co, efa->v1->co);
if (plane[0]*plane[0]+plane[1]*plane[1]+plane[2]*plane[2] < vec[0]*vec[0]+vec[1]*vec[1]+vec[2]*vec[2])
copy_v3_v3(plane, vec);
}
}
normalize_v3(plane);
}
void EM_select_face(EditFace *efa, int sel)
{
if(sel) {
efa->f |= SELECT;
efa->e1->f |= SELECT;
efa->e2->f |= SELECT;
efa->e3->f |= SELECT;
if(efa->e4) efa->e4->f |= SELECT;
efa->v1->f |= SELECT;
efa->v2->f |= SELECT;
efa->v3->f |= SELECT;
if(efa->v4) efa->v4->f |= SELECT;
}
else {
efa->f &= ~SELECT;
efa->e1->f &= ~SELECT;
efa->e2->f &= ~SELECT;
efa->e3->f &= ~SELECT;
if(efa->e4) efa->e4->f &= ~SELECT;
efa->v1->f &= ~SELECT;
efa->v2->f &= ~SELECT;
efa->v3->f &= ~SELECT;
if(efa->v4) efa->v4->f &= ~SELECT;
}
}
void EM_select_edge(EditEdge *eed, int sel)
{
if(sel) {
eed->f |= SELECT;
eed->v1->f |= SELECT;
eed->v2->f |= SELECT;
}
else {
eed->f &= ~SELECT;
eed->v1->f &= ~SELECT;
eed->v2->f &= ~SELECT;
}
}
void EM_select_face_fgon(EditMesh *em, EditFace *efa, int val)
{
short index=0;
if(efa->fgonf==0) EM_select_face(efa, val);
else {
if(efa->e1->fgoni) index= efa->e1->fgoni;
if(efa->e2->fgoni) index= efa->e2->fgoni;
if(efa->e3->fgoni) index= efa->e3->fgoni;
if(efa->v4 && efa->e4->fgoni) index= efa->e4->fgoni;
if((index==0) && (G.f & G_DEBUG))printf("wrong fgon select\n");
// select all ngon faces with index
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->fgonf) {
if(efa->e1->fgoni==index || efa->e2->fgoni==index ||
efa->e3->fgoni==index || (efa->e4 && efa->e4->fgoni==index) ) {
EM_select_face(efa, val);
}
}
}
}
}
/* only vertices */
int faceselectedOR(EditFace *efa, int flag)
{
if ((efa->v1->f | efa->v2->f | efa->v3->f | (efa->v4?efa->v4->f:0))&flag) {
return 1;
} else {
return 0;
}
}
// replace with (efa->f & SELECT)
int faceselectedAND(EditFace *efa, int flag)
{
if ((efa->v1->f & efa->v2->f & efa->v3->f & (efa->v4?efa->v4->f:flag))&flag) {
return 1;
} else {
return 0;
}
}
void EM_clear_flag_all(EditMesh *em, int flag)
{
EditVert *eve;
EditEdge *eed;
EditFace *efa;
for (eve= em->verts.first; eve; eve= eve->next) eve->f &= ~flag;
for (eed= em->edges.first; eed; eed= eed->next) eed->f &= ~flag;
for (efa= em->faces.first; efa; efa= efa->next) efa->f &= ~flag;
if(flag & SELECT) {
BLI_freelistN(&(em->selected));
em->totvertsel= em->totedgesel= em->totfacesel= 0;
}
}
void EM_set_flag_all(EditMesh *em, int flag)
{
EditVert *eve;
EditEdge *eed;
EditFace *efa;
for (eve= em->verts.first; eve; eve= eve->next) if(eve->h==0) eve->f |= flag;
for (eed= em->edges.first; eed; eed= eed->next) if(eed->h==0) eed->f |= flag;
for (efa= em->faces.first; efa; efa= efa->next) if(efa->h==0) efa->f |= flag;
if(flag & SELECT) {
em->totvertsel= em->totvert;
em->totedgesel= em->totedge;
em->totfacesel= em->totface;
}
}
void EM_set_flag_all_selectmode(EditMesh *em, int flag)
{
EditVert *eve;
EditEdge *eed;
EditFace *efa;
int selvert= 0, seledge= 0, selface= 0;
if (em->selectmode & SCE_SELECT_VERTEX) {
/* If vertex select mode enabled all the data could be affected */
for (eve= em->verts.first; eve; eve= eve->next) if(eve->h==0) eve->f |= flag;
for (eed= em->edges.first; eed; eed= eed->next) if(eed->h==0) eed->f |= flag;
for (efa= em->faces.first; efa; efa= efa->next) if(efa->h==0) efa->f |= flag;
if (flag & SELECT) {
selvert= em->totvert;
seledge= em->totedge;
selface= em->totface;
}
} else if (em->selectmode & SCE_SELECT_EDGE) {
/* If edge select mode is enabled we should affect on all edges, faces and */
/* vertices, connected to them */
for (eed= em->edges.first; eed; eed= eed->next) {
SET_EED_FLAG(eed, flag)
}
for (efa= em->faces.first; efa; efa= efa->next) {
if(efa->h==0) {
efa->f |= flag;
if (flag & SELECT) {
++selface;
}
}
}
} else if (em->selectmode & SCE_SELECT_FACE) {
/* No vertex and edge select mode, only face selection */
/* In face select mode only edges and vertices belongs to faces should be affected */
for (efa= em->faces.first; efa; efa= efa->next) {
if(efa->h==0) {
efa->f |= flag;
SET_EED_FLAG(efa->e1, flag);
SET_EED_FLAG(efa->e2, flag);
SET_EED_FLAG(efa->e3, flag);
if (efa->e4) {
SET_EED_FLAG(efa->e4, flag);
}
if (flag & SELECT) {
++selface;
}
}
}
}
if(flag & SELECT) {
em->totvertsel= selvert;
em->totedgesel= seledge;
em->totfacesel= selface;
}
}
/* flush for changes in vertices only */
void EM_deselect_flush(EditMesh *em)
{
EditEdge *eed;
EditFace *efa;
for(eed= em->edges.first; eed; eed= eed->next) {
if(eed->v1->f & eed->v2->f & SELECT);
else eed->f &= ~SELECT;
}
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->v4) {
if(efa->v1->f & efa->v2->f & efa->v3->f & efa->v4->f & SELECT );
else efa->f &= ~SELECT;
}
else {
if(efa->v1->f & efa->v2->f & efa->v3->f & SELECT );
else efa->f &= ~SELECT;
}
}
EM_nedges_selected(em);
EM_nfaces_selected(em);
}
/* flush selection to edges & faces */
/* this only based on coherent selected vertices, for example when adding new
objects. call clear_flag_all() before you select vertices to be sure it ends OK!
*/
void EM_select_flush(EditMesh *em)
{
EditEdge *eed;
EditFace *efa;
for(eed= em->edges.first; eed; eed= eed->next) {
if(eed->v1->f & eed->v2->f & SELECT) eed->f |= SELECT;
}
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->v4) {
if(efa->v1->f & efa->v2->f & efa->v3->f & efa->v4->f & SELECT ) efa->f |= SELECT;
}
else {
if(efa->v1->f & efa->v2->f & efa->v3->f & SELECT ) efa->f |= SELECT;
}
}
EM_nedges_selected(em);
EM_nfaces_selected(em);
}
/* when vertices or edges can be selected, also make fgon consistent */
static void check_fgons_selection(EditMesh *em)
{
EditFace *efa, *efan;
EditEdge *eed;
ListBase *lbar;
int sel, desel, index, totfgon= 0;
/* count amount of fgons */
for(eed= em->edges.first; eed; eed= eed->next)
if(eed->fgoni>totfgon) totfgon= eed->fgoni;
if(totfgon==0) return;
lbar= MEM_callocN((totfgon+1)*sizeof(ListBase), "listbase array");
/* put all fgons in lbar */
for(efa= em->faces.first; efa; efa= efan) {
efan= efa->next;
index= efa->e1->fgoni;
if(index==0) index= efa->e2->fgoni;
if(index==0) index= efa->e3->fgoni;
if(index==0 && efa->e4) index= efa->e4->fgoni;
if(index) {
BLI_remlink(&em->faces, efa);
BLI_addtail(&lbar[index], efa);
}
}
/* now check the fgons */
for(index=1; index<=totfgon; index++) {
/* we count on vertices/faces/edges being set OK, so we only have to set ngon itself */
sel= desel= 0;
for(efa= lbar[index].first; efa; efa= efa->next) {
if(efa->e1->fgoni==0) {
if(efa->e1->f & SELECT) sel++;
else desel++;
}
if(efa->e2->fgoni==0) {
if(efa->e2->f & SELECT) sel++;
else desel++;
}
if(efa->e3->fgoni==0) {
if(efa->e3->f & SELECT) sel++;
else desel++;
}
if(efa->e4 && efa->e4->fgoni==0) {
if(efa->e4->f & SELECT) sel++;
else desel++;
}
if(sel && desel) break;
}
if(sel && desel) sel= 0;
else if(sel) sel= 1;
else sel= 0;
/* select/deselect and put back */
for(efa= lbar[index].first; efa; efa= efa->next) {
if(sel) efa->f |= SELECT;
else efa->f &= ~SELECT;
}
addlisttolist(&em->faces, &lbar[index]);
}
MEM_freeN(lbar);
}
/* flush to edges & faces */
/* based on select mode it selects edges/faces
assumed is that verts/edges/faces were properly selected themselves
with the calls above
*/
void EM_selectmode_flush(EditMesh *em)
{
EditEdge *eed;
EditFace *efa;
// flush to edges & faces
if(em->selectmode & SCE_SELECT_VERTEX) {
for(eed= em->edges.first; eed; eed= eed->next) {
if(eed->v1->f & eed->v2->f & SELECT) eed->f |= SELECT;
else eed->f &= ~SELECT;
}
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->v4) {
if(efa->v1->f & efa->v2->f & efa->v3->f & efa->v4->f & SELECT) efa->f |= SELECT;
else efa->f &= ~SELECT;
}
else {
if(efa->v1->f & efa->v2->f & efa->v3->f & SELECT) efa->f |= SELECT;
else efa->f &= ~SELECT;
}
}
}
// flush to faces
else if(em->selectmode & SCE_SELECT_EDGE) {
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->e4) {
if(efa->e1->f & efa->e2->f & efa->e3->f & efa->e4->f & SELECT) efa->f |= SELECT;
else efa->f &= ~SELECT;
}
else {
if(efa->e1->f & efa->e2->f & efa->e3->f & SELECT) efa->f |= SELECT;
else efa->f &= ~SELECT;
}
}
}
// make sure selected faces have selected edges too, for extrude (hack?)
else if(em->selectmode & SCE_SELECT_FACE) {
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->f & SELECT) EM_select_face(efa, 1);
}
}
if(!(em->selectmode & SCE_SELECT_FACE))
check_fgons_selection(em);
EM_nvertices_selected(em);
EM_nedges_selected(em);
EM_nfaces_selected(em);
}
void EM_convertsel(EditMesh *em, short oldmode, short selectmode)
{
EditVert *eve;
EditEdge *eed;
EditFace *efa;
/*clear flags*/
for(eve= em->verts.first; eve; eve= eve->next) eve->f1 = 0;
for(eed= em->edges.first; eed; eed= eed->next) eed->f1 = 0;
for(efa= em->faces.first; efa; efa= efa->next) efa->f1 = 0;
/*have to find out what the selectionmode was previously*/
if(oldmode == SCE_SELECT_VERTEX) {
if(selectmode == SCE_SELECT_EDGE){
/*select all edges associated with every selected vertex*/
for(eed= em->edges.first; eed; eed= eed->next){
if(eed->v1->f&SELECT) eed->f1 = 1;
else if(eed->v2->f&SELECT) eed->f1 = 1;
}
for(eed= em->edges.first; eed; eed= eed->next){
if(eed->f1 == 1) EM_select_edge(eed,1);
}
}
else if(selectmode == SCE_SELECT_FACE){
/*select all faces associated with every selected vertex*/
for(efa= em->faces.first; efa; efa= efa->next){
if(efa->v1->f&SELECT) efa->f1 = 1;
else if(efa->v2->f&SELECT) efa->f1 = 1;
else if(efa->v3->f&SELECT) efa->f1 = 1;
else{
if(efa->v4){
if(efa->v4->f&SELECT) efa->f1 =1;
}
}
}
for(efa= em->faces.first; efa; efa= efa->next){
if(efa->f1 == 1) EM_select_face(efa,1);
}
}
}
if(oldmode == SCE_SELECT_EDGE){
if(selectmode == SCE_SELECT_FACE){
for(efa= em->faces.first; efa; efa= efa->next){
if(efa->e1->f&SELECT) efa->f1 = 1;
else if(efa->e2->f&SELECT) efa->f1 = 1;
else if(efa->e3->f&SELECT) efa->f1 = 1;
else if(efa->e4){
if(efa->e4->f&SELECT) efa->f1 = 1;
}
}
for(efa= em->faces.first; efa; efa= efa->next){
if(efa->f1 == 1) EM_select_face(efa,1);
}
}
}
check_fgons_selection(em);
EM_nvertices_selected(em);
EM_nedges_selected(em);
EM_nfaces_selected(em);
}
void EM_selectmode_to_scene(struct Scene *scene, struct Object *obedit)
{
scene->toolsettings->selectmode= get_mesh(obedit)->edit_mesh->selectmode;
}
/* when switching select mode, makes sure selection is consistent for editing */
/* also for paranoia checks to make sure edge or face mode works */
void EM_selectmode_set(EditMesh *em)
{
EditVert *eve;
EditEdge *eed;
EditFace *efa;
EM_strip_selections(em); /*strip EditSelections from em->selected that are not relevant to new mode*/
if(em->selectmode & SCE_SELECT_VERTEX) {
/* vertices -> edges -> faces */
for (eed= em->edges.first; eed; eed= eed->next) eed->f &= ~SELECT;
for (efa= em->faces.first; efa; efa= efa->next) efa->f &= ~SELECT;
EM_select_flush(em);
}
else if(em->selectmode & SCE_SELECT_EDGE) {
/* deselect vertices, and select again based on edge select */
for(eve= em->verts.first; eve; eve= eve->next) eve->f &= ~SELECT;
for(eed= em->edges.first; eed; eed= eed->next)
if(eed->f & SELECT) EM_select_edge(eed, 1);
/* selects faces based on edge status */
EM_selectmode_flush(em);
}
else if(em->selectmode & SCE_SELECT_FACE) {
/* deselect eges, and select again based on face select */
for(eed= em->edges.first; eed; eed= eed->next) EM_select_edge(eed, 0);
for(efa= em->faces.first; efa; efa= efa->next)
if(efa->f & SELECT) EM_select_face(efa, 1);
}
EM_nvertices_selected(em);
EM_nedges_selected(em);
EM_nfaces_selected(em);
}
/* paranoia check, actually only for entering editmode. rule:
- vertex hidden, always means edge is hidden too
- edge hidden, always means face is hidden too
- face hidden, dont change anything
*/
void EM_hide_reset(EditMesh *em)
{
EditEdge *eed;
EditFace *efa;
for(eed= em->edges.first; eed; eed= eed->next)
if(eed->v1->h || eed->v2->h) eed->h |= 1;
for(efa= em->faces.first; efa; efa= efa->next)
if((efa->e1->h & 1) || (efa->e2->h & 1) || (efa->e3->h & 1) || (efa->e4 && (efa->e4->h & 1)))
efa->h= 1;
}
void EM_data_interp_from_verts(EditMesh *em, EditVert *v1, EditVert *v2, EditVert *eve, float fac)
{
void *src[2];
float w[2];
if (v1->data && v2->data) {
src[0]= v1->data;
src[1]= v2->data;
w[0] = 1.0f-fac;
w[1] = fac;
CustomData_em_interp(&em->vdata, src, w, NULL, 2, eve->data);
}
}
void EM_data_interp_from_faces(EditMesh *em, EditFace *efa1, EditFace *efa2, EditFace *efan, int i1, int i2, int i3, int i4)
{
float w[2][4][4];
void *src[2];
int count = (efa2)? 2: 1;
if (efa1->data) {
/* set weights for copying from corners directly to other corners */
memset(w, 0, sizeof(w));
w[i1/4][0][i1%4]= 1.0f;
w[i2/4][1][i2%4]= 1.0f;
w[i3/4][2][i3%4]= 1.0f;
if (i4 != -1)
w[i4/4][3][i4%4]= 1.0f;
src[0]= efa1->data;
src[1]= (efa2)? efa2->data: NULL;
CustomData_em_interp(&em->fdata, src, NULL, (float*)w, count, efan->data);
}
}
EditFace *EM_face_from_faces(EditMesh *em, EditFace *efa1, EditFace *efa2, int i1, int i2, int i3, int i4)
{
EditFace *efan;
EditVert **v[2];
v[0]= &efa1->v1;
v[1]= (efa2)? &efa2->v1: NULL;
efan= addfacelist(em, v[i1/4][i1%4], v[i2/4][i2%4], v[i3/4][i3%4],
(i4 == -1)? 0: v[i4/4][i4%4], efa1, NULL);
EM_data_interp_from_faces(em, efa1, efa2, efan, i1, i2, i3, i4);
return efan;
}
static void update_data_blocks(EditMesh *em, CustomData *olddata, CustomData *data)
{
EditFace *efa;
EditVert *eve;
void *block;
if (data == &em->vdata) {
for(eve= em->verts.first; eve; eve= eve->next) {
block = NULL;
CustomData_em_set_default(data, &block);
CustomData_em_copy_data(olddata, data, eve->data, &block);
CustomData_em_free_block(olddata, &eve->data);
eve->data= block;
}
}
else if (data == &em->fdata) {
for(efa= em->faces.first; efa; efa= efa->next) {
block = NULL;
CustomData_em_set_default(data, &block);
CustomData_em_copy_data(olddata, data, efa->data, &block);
CustomData_em_free_block(olddata, &efa->data);
efa->data= block;
}
}
}
void EM_add_data_layer(EditMesh *em, CustomData *data, int type)
{
CustomData olddata;
olddata= *data;
olddata.layers= (olddata.layers)? MEM_dupallocN(olddata.layers): NULL;
CustomData_add_layer(data, type, CD_CALLOC, NULL, 0);
update_data_blocks(em, &olddata, data);
if (olddata.layers) MEM_freeN(olddata.layers);
}
void EM_free_data_layer(EditMesh *em, CustomData *data, int type)
{
CustomData olddata;
olddata= *data;
olddata.layers= (olddata.layers)? MEM_dupallocN(olddata.layers): NULL;
CustomData_free_layer_active(data, type, 0);
update_data_blocks(em, &olddata, data);
if (olddata.layers) MEM_freeN(olddata.layers);
}
/* ******** EXTRUDE ********* */
static void add_normal_aligned(float *nor, float *add)
{
if( INPR(nor, add) < -0.9999f)
sub_v3_v3(nor, add);
else
add_v3_v3(nor, add);
}
static void set_edge_directions_f2(EditMesh *em, int val)
{
EditFace *efa;
int do_all= 1;
/* edge directions are used for extrude, to detect direction of edges that make new faces */
/* we have set 'f2' flags in edges that need to get a direction set (e.g. get new face) */
/* the val argument differs... so we need it as arg */
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->f & SELECT) {
do_all= 0;
if(efa->e1->f2<val) {
if(efa->e1->v1 == efa->v1) efa->e1->dir= 0;
else efa->e1->dir= 1;
}
if(efa->e2->f2<val) {
if(efa->e2->v1 == efa->v2) efa->e2->dir= 0;
else efa->e2->dir= 1;
}
if(efa->e3->f2<val) {
if(efa->e3->v1 == efa->v3) efa->e3->dir= 0;
else efa->e3->dir= 1;
}
if(efa->e4 && efa->e4->f2<val) {
if(efa->e4->v1 == efa->v4) efa->e4->dir= 0;
else efa->e4->dir= 1;
}
}
}
/* ok, no faces done... then we at least set it for exterior edges */
if(do_all) {
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->e1->v1 == efa->v1) efa->e1->dir= 0;
else efa->e1->dir= 1;
if(efa->e2->v1 == efa->v2) efa->e2->dir= 0;
else efa->e2->dir= 1;
if(efa->e3->v1 == efa->v3) efa->e3->dir= 0;
else efa->e3->dir= 1;
if(efa->e4) {
if(efa->e4->v1 == efa->v4) efa->e4->dir= 0;
else efa->e4->dir= 1;
}
}
}
}
/* individual face extrude */
/* will use vertex normals for extrusion directions, so *nor is unaffected */
short extrudeflag_face_indiv(EditMesh *em, short flag, float *nor)
{
EditVert *eve, *v1, *v2, *v3, *v4;
EditEdge *eed;
EditFace *efa, *nextfa;
if(em==NULL) return 0;
/* selected edges with 1 or more selected face become faces */
/* selected faces each makes new faces */
/* always remove old faces, keeps volumes manifold */
/* select the new extrusion, deselect old */
/* step 1; init, count faces in edges */
recalc_editnormals(em);
for(eve= em->verts.first; eve; eve= eve->next) eve->f1= 0; // new select flag
for(eed= em->edges.first; eed; eed= eed->next) {
eed->f2= 0; // amount of unselected faces
}
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->f & SELECT);
else {
efa->e1->f2++;
efa->e2->f2++;
efa->e3->f2++;
if(efa->e4) efa->e4->f2++;
}
}
/* step 2: make new faces from faces */
for(efa= em->faces.last; efa; efa= efa->prev) {
if(efa->f & SELECT) {
v1= addvertlist(em, efa->v1->co, efa->v1);
v2= addvertlist(em, efa->v2->co, efa->v2);
v3= addvertlist(em, efa->v3->co, efa->v3);
v1->f1= v2->f1= v3->f1= 1;
VECCOPY(v1->no, efa->n);
VECCOPY(v2->no, efa->n);
VECCOPY(v3->no, efa->n);
if(efa->v4) {
v4= addvertlist(em, efa->v4->co, efa->v4);
v4->f1= 1;
VECCOPY(v4->no, efa->n);
}
else v4= NULL;
/* side faces, clockwise */
addfacelist(em, efa->v2, v2, v1, efa->v1, efa, NULL);
addfacelist(em, efa->v3, v3, v2, efa->v2, efa, NULL);
if(efa->v4) {
addfacelist(em, efa->v4, v4, v3, efa->v3, efa, NULL);
addfacelist(em, efa->v1, v1, v4, efa->v4, efa, NULL);
}
else {
addfacelist(em, efa->v1, v1, v3, efa->v3, efa, NULL);
}
/* top face */
addfacelist(em, v1, v2, v3, v4, efa, NULL);
}
}
/* step 3: remove old faces */
efa= em->faces.first;
while(efa) {
nextfa= efa->next;
if(efa->f & SELECT) {
BLI_remlink(&em->faces, efa);
free_editface(em, efa);
}
efa= nextfa;
}
/* step 4: redo selection */
EM_clear_flag_all(em, SELECT);
for(eve= em->verts.first; eve; eve= eve->next) {
if(eve->f1) eve->f |= SELECT;
}
EM_select_flush(em);
return 'n';
}
/* extrudes individual edges */
/* nor is filled with constraint vector */
short extrudeflag_edges_indiv(EditMesh *em, short flag, float *nor)
{
EditVert *eve;
EditEdge *eed;
EditFace *efa;
for(eve= em->verts.first; eve; eve= eve->next) eve->tmp.v = NULL;
for(eed= em->edges.first; eed; eed= eed->next) {
eed->tmp.f = NULL;
eed->f2= ((eed->f & flag)!=0);
}
set_edge_directions_f2(em, 2);
/* sample for next loop */
for(efa= em->faces.first; efa; efa= efa->next) {
efa->e1->tmp.f = efa;
efa->e2->tmp.f = efa;
efa->e3->tmp.f = efa;
if(efa->e4) efa->e4->tmp.f = efa;
}
/* make the faces */
for(eed= em->edges.first; eed; eed= eed->next) {
if(eed->f & flag) {
if(eed->v1->tmp.v == NULL)
eed->v1->tmp.v = addvertlist(em, eed->v1->co, eed->v1);
if(eed->v2->tmp.v == NULL)
eed->v2->tmp.v = addvertlist(em, eed->v2->co, eed->v2);
if(eed->dir==1)
addfacelist(em, eed->v1, eed->v2,
eed->v2->tmp.v, eed->v1->tmp.v,
eed->tmp.f, NULL);
else
addfacelist(em, eed->v2, eed->v1,
eed->v1->tmp.v, eed->v2->tmp.v,
eed->tmp.f, NULL);
/* for transform */
if(eed->tmp.f) {
efa = eed->tmp.f;
if (efa->f & SELECT) add_normal_aligned(nor, efa->n);
}
}
}
normalize_v3(nor);
/* set correct selection */
EM_clear_flag_all(em, SELECT);
for(eve= em->verts.last; eve; eve= eve->prev) {
if(eve->tmp.v) {
eve->tmp.v->f |= flag;
}
}
for(eed= em->edges.first; eed; eed= eed->next) {
if(eed->v1->f & eed->v2->f & flag) eed->f |= flag;
}
if(nor[0]==0.0 && nor[1]==0.0 && nor[2]==0.0) return 'g'; // g is grab
return 'n'; // n is for normal constraint
}
/* extrudes individual vertices */
short extrudeflag_verts_indiv(EditMesh *em, short flag, float *nor)
{
EditVert *eve;
/* make the edges */
for(eve= em->verts.first; eve; eve= eve->next) {
if(eve->f & flag) {
eve->tmp.v = addvertlist(em, eve->co, eve);
addedgelist(em, eve, eve->tmp.v, NULL);
}
else eve->tmp.v = NULL;
}
/* set correct selection */
EM_clear_flag_all(em, SELECT);
for(eve= em->verts.last; eve; eve= eve->prev)
if (eve->tmp.v)
eve->tmp.v->f |= flag;
return 'g'; // g is grab
}
/* this is actually a recode of extrudeflag(), using proper edge/face select */
/* hurms, doesnt use 'flag' yet, but its not called by primitive making stuff anyway */
static short extrudeflag_edge(Object *obedit, EditMesh *em, short flag, float *nor, int all)
{
/* all select edges/faces: extrude */
/* old select is cleared, in new ones it is set */
EditVert *eve, *nextve;
EditEdge *eed, *nexted;
EditFace *efa, *nextfa, *efan;
short del_old= 0;
ModifierData *md;
if(em==NULL) return 0;
md = obedit->modifiers.first;
/* selected edges with 0 or 1 selected face become faces */
/* selected faces generate new faces */
/* if *one* selected face has edge with unselected face; remove old selected faces */
/* if selected edge is not used anymore; remove */
/* if selected vertex is not used anymore: remove */
/* select the new extrusion, deselect old */
/* step 1; init, count faces in edges */
recalc_editnormals(em);
for(eve= em->verts.first; eve; eve= eve->next) {
eve->tmp.v = NULL;
eve->f1= 0;
}
for(eed= em->edges.first; eed; eed= eed->next) {
eed->f1= 0; // amount of unselected faces
eed->f2= 0; // amount of selected faces
if(eed->f & SELECT) {
eed->v1->f1= 1; // we call this 'selected vertex' now
eed->v2->f1= 1;
}
eed->tmp.f = NULL; // here we tuck face pointer, as sample
}
for(efa= em->faces.first; efa; efa= efa->next) {
if(efa->f & SELECT) {
efa->e1->f2++;
efa->e2->f2++;
efa->e3->f2++;
if(efa->e4) efa->e4->f2++;
// sample for next loop
efa->e1->tmp.f = efa;
efa->e2->tmp.f = efa;
efa->e3->tmp.f = efa;
if(efa->e4) efa->e4->tmp.f = efa;
}
else {
efa->e1->f1++;
efa->e2->f1++;
efa->e3->f1++;
if(efa->e4) efa->e4->f1++;
}
}
/* If a mirror modifier with clipping is on, we need to adjust some
* of the cases above to handle edges on the line of symmetry.
*/
for (; md; md=md->next) {
if (md->type==eModifierType_Mirror) {
MirrorModifierData *mmd = (MirrorModifierData*) md;
if(mmd->flag & MOD_MIR_CLIPPING) {
float mtx[4][4];
if (mmd->mirror_ob) {
float imtx[4][4];
invert_m4_m4(imtx, mmd->mirror_ob->obmat);
mul_m4_m4m4(mtx, obedit->obmat, imtx);
}
for (eed= em->edges.first; eed; eed= eed->next) {
if(eed->f2 == 1) {
float co1[3], co2[3];
copy_v3_v3(co1, eed->v1->co);
copy_v3_v3(co2, eed->v2->co);
if (mmd->mirror_ob) {
mul_m4_v3(mtx, co1);
mul_m4_v3(mtx, co2);
}
if (mmd->flag & MOD_MIR_AXIS_X)
if ( (fabs(co1[0]) < mmd->tolerance) &&
(fabs(co2[0]) < mmd->tolerance) )
++eed->f2;
if (mmd->flag & MOD_MIR_AXIS_Y)
if ( (fabs(co1[1]) < mmd->tolerance) &&
(fabs(co2[1]) < mmd->tolerance) )
++eed->f2;
if (mmd->flag & MOD_MIR_AXIS_Z)
if ( (fabs(co1[2]) < mmd->tolerance) &&
(fabs(co2[2]) < mmd->tolerance) )
++eed->f2;
}
}
}
}
}
set_edge_directions_f2(em, 2);
/* step 1.5: if *one* selected face has edge with unselected face; remove old selected faces */
if(all == 0) {
for(efa= em->faces.last; efa; efa= efa->prev) {
if(efa->f & SELECT) {
if(efa->e1->f1 || efa->e2->f1 || efa->e3->f1 || (efa->e4 && efa->e4->f1)) {
del_old= 1;
break;
}
}
}
}
/* step 2: make new faces from edges */
for(eed= em->edges.last; eed; eed= eed->prev) {
if(eed->f & SELECT) {
if(eed->f2<2) {
if(eed->v1->tmp.v == NULL)
eed->v1->tmp.v = addvertlist(em, eed->v1->co, eed->v1);
if(eed->v2->tmp.v == NULL)
eed->v2->tmp.v = addvertlist(em, eed->v2->co, eed->v2);
/* if del_old, the preferred normal direction is exact
* opposite as for keep old faces
*/
if(eed->dir!=del_old)
addfacelist(em, eed->v1, eed->v2,
eed->v2->tmp.v, eed->v1->tmp.v,
eed->tmp.f, NULL);
else
addfacelist(em, eed->v2, eed->v1,
eed->v1->tmp.v, eed->v2->tmp.v,
eed->tmp.f, NULL);
}
}
}
/* step 3: make new faces from faces */
for(efa= em->faces.last; efa; efa= efa->prev) {
if(efa->f & SELECT) {
if (efa->v1->tmp.v == NULL)
efa->v1->tmp.v = addvertlist(em, efa->v1->co, efa->v1);
if (efa->v2->tmp.v ==NULL)
efa->v2->tmp.v = addvertlist(em, efa->v2->co, efa->v2);
if (efa->v3->tmp.v ==NULL)
efa->v3->tmp.v = addvertlist(em, efa->v3->co, efa->v3);
if (efa->v4 && (efa->v4->tmp.v == NULL))
efa->v4->tmp.v = addvertlist(em, efa->v4->co, efa->v4);
if(del_old==0) { // keep old faces means flipping normal
if(efa->v4)
efan = addfacelist(em, efa->v4->tmp.v, efa->v3->tmp.v,
efa->v2->tmp.v, efa->v1->tmp.v, efa, efa);
else
efan = addfacelist(em, efa->v3->tmp.v, efa->v2->tmp.v,
efa->v1->tmp.v, NULL, efa, efa);
}
else {
if(efa->v4)
efan = addfacelist(em, efa->v1->tmp.v, efa->v2->tmp.v,
efa->v3->tmp.v, efa->v4->tmp.v, efa, efa);
else
efan = addfacelist(em, efa->v1->tmp.v, efa->v2->tmp.v,
efa->v3->tmp.v, NULL, efa, efa);
}
if (em->act_face == efa) {
em->act_face = efan;
}
/* for transform */
add_normal_aligned(nor, efa->n);
}
}
if(del_old) {
/* step 4: remove old faces, if del_old */
efa= em->faces.first;
while(efa) {
nextfa= efa->next;
if(efa->f & SELECT) {
BLI_remlink(&em->faces, efa);
free_editface(em, efa);
}
efa= nextfa;
}
/* step 5: remove selected unused edges */
/* start tagging again */
for(eed= em->edges.first; eed; eed= eed->next) eed->f1=0;
for(efa= em->faces.first; efa; efa= efa->next) {
efa->e1->f1= 1;
efa->e2->f1= 1;
efa->e3->f1= 1;
if(efa->e4) efa->e4->f1= 1;
}
/* remove */
eed= em->edges.first;
while(eed) {
nexted= eed->next;
if(eed->f & SELECT) {
if(eed->f1==0) {
remedge(em, eed);
free_editedge(em, eed);
}
}
eed= nexted;
}
/* step 6: remove selected unused vertices */
for(eed= em->edges.first; eed; eed= eed->next)
eed->v1->f1= eed->v2->f1= 0;
eve= em->verts.first;
while(eve) {
nextve= eve->next;
if(eve->f1) {
// hack... but we need it for step 7, redoing selection
if(eve->tmp.v) eve->tmp.v->tmp.v= eve->tmp.v;
BLI_remlink(&em->verts, eve);
free_editvert(em, eve);
}
eve= nextve;
}
}
normalize_v3(nor); // translation normal grab
/* step 7: redo selection */
EM_clear_flag_all(em, SELECT);
for(eve= em->verts.first; eve; eve= eve->next) {
if(eve->tmp.v) {
eve->tmp.v->f |= SELECT;
}
}
EM_select_flush(em);
if(nor[0]==0.0 && nor[1]==0.0 && nor[2]==0.0) return 'g'; // grab
return 'n'; // normal constraint
}
short extrudeflag_vert(Object *obedit, EditMesh *em, short flag, float *nor, int all)
{
/* all verts/edges/faces with (f & 'flag'): extrude */
/* from old verts, 'flag' is cleared, in new ones it is set */
EditVert *eve, *v1, *v2, *v3, *v4, *nextve;
EditEdge *eed, *e1, *e2, *e3, *e4, *nexted;
EditFace *efa, *efa2, *nextvl;
short sel=0, del_old= 0, is_face_sel=0;
ModifierData *md;
if(em==NULL) return 0;
md = obedit->modifiers.first;
/* clear vert flag f1, we use this to detect a loose selected vertice */
eve= em->verts.first;
while(eve) {
if(eve->f & flag) eve->f1= 1;
else eve->f1= 0;
eve= eve->next;
}
/* clear edges counter flag, if selected we set it at 1 */
eed= em->edges.first;
while(eed) {
if( (eed->v1->f & flag) && (eed->v2->f & flag) ) {
eed->f2= 1;
eed->v1->f1= 0;
eed->v2->f1= 0;
}
else eed->f2= 0;
eed->f1= 1; /* this indicates it is an 'old' edge (in this routine we make new ones) */
eed->tmp.f = NULL; /* used as sample */
eed= eed->next;
}
/* we set a flag in all selected faces, and increase the associated edge counters */
efa= em->faces.first;
while(efa) {
efa->f1= 0;
if(faceselectedAND(efa, flag)) {
e1= efa->e1;
e2= efa->e2;
e3= efa->e3;
e4= efa->e4;
if(e1->f2 < 3) e1->f2++;
if(e2->f2 < 3) e2->f2++;
if(e3->f2 < 3) e3->f2++;
if(e4 && e4->f2 < 3) e4->f2++;
efa->f1= 1;
is_face_sel= 1; // for del_old
}
else if(faceselectedOR(efa, flag)) {
e1= efa->e1;
e2= efa->e2;
e3= efa->e3;
e4= efa->e4;
if( (e1->v1->f & flag) && (e1->v2->f & flag) ) e1->f1= 2;
if( (e2->v1->f & flag) && (e2->v2->f & flag) ) e2->f1= 2;
if( (e3->v1->f & flag) && (e3->v2->f & flag) ) e3->f1= 2;
if( e4 && (e4->v1->f & flag) && (e4->v2->f & flag) ) e4->f1= 2;
}
// sample for next loop
efa->e1->tmp.f = efa;
efa->e2->tmp.f = efa;
efa->e3->tmp.f = efa;
if(efa->e4) efa->e4->tmp.f = efa;
efa= efa->next;
}
set_edge_directions_f2(em, 3);
/* the current state now is:
eve->f1==1: loose selected vertex
eed->f2==0 : edge is not selected, no extrude
eed->f2==1 : edge selected, is not part of a face, extrude
eed->f2==2 : edge selected, is part of 1 face, extrude
eed->f2==3 : edge selected, is part of more faces, no extrude
eed->f1==0: new edge
eed->f1==1: edge selected, is part of selected face, when eed->f==3: remove
eed->f1==2: edge selected, part of a partially selected face
efa->f1==1 : duplicate this face
*/
/* If a mirror modifier with clipping is on, we need to adjust some
* of the cases above to handle edges on the line of symmetry.
*/
for (; md; md=md->next) {
if (md->type==eModifierType_Mirror) {
MirrorModifierData *mmd = (MirrorModifierData*) md;
if(mmd->flag & MOD_MIR_CLIPPING) {
float mtx[4][4];
if (mmd->mirror_ob) {
float imtx[4][4];
invert_m4_m4(imtx, mmd->mirror_ob->obmat);
mul_m4_m4m4(mtx, obedit->obmat, imtx);
}
for (eed= em->edges.first; eed; eed= eed->next) {
if(eed->f2 == 2) {
float co1[3], co2[3];
copy_v3_v3(co1, eed->v1->co);
copy_v3_v3(co2, eed->v2->co);
if (mmd->mirror_ob) {
mul_m4_v3(mtx, co1);
mul_m4_v3(mtx, co2);
}
if (mmd->flag & MOD_MIR_AXIS_X)
if ( (fabs(co1[0]) < mmd->tolerance) &&
(fabs(co2[0]) < mmd->tolerance) )
++eed->f2;
if (mmd->flag & MOD_MIR_AXIS_Y)
if ( (fabs(co1[1]) < mmd->tolerance) &&
(fabs(co2[1]) < mmd->tolerance) )
++eed->f2;
if (mmd->flag & MOD_MIR_AXIS_Z)
if ( (fabs(co1[2]) < mmd->tolerance) &&
(fabs(co2[2]) < mmd->tolerance) )
++eed->f2;
}
}
}
}
}
/* copy all selected vertices, */
/* write pointer to new vert in old struct at eve->tmp.v */
eve= em->verts.last;
while(eve) {
eve->f &= ~128; /* clear, for later test for loose verts */
if(eve->f & flag) {
sel= 1;
v1= addvertlist(em, 0, NULL);
VECCOPY(v1->co, eve->co);
VECCOPY(v1->no, eve->no);
v1->f= eve->f;
eve->f-= flag;
eve->tmp.v = v1;
}
else eve->tmp.v = 0;
eve= eve->prev;
}
if(sel==0) return 0;
/* all edges with eed->f2==1 or eed->f2==2 become faces */
/* if del_old==1 then extrude is in partial geometry, to keep it manifold.
verts with f1==0 and (eve->f & 128)==0) are removed
edges with eed->f2>2 are removed
faces with efa->f1 are removed
if del_old==0 the extrude creates a volume.
*/
/* find if we delete old faces */
if(is_face_sel && all==0) {
for(eed= em->edges.first; eed; eed= eed->next) {
if( (eed->f2==1 || eed->f2==2) ) {
if(eed->f1==2) {
del_old= 1;
break;
}
}
}
}
eed= em->edges.last;
while(eed) {
nexted= eed->prev;
if( eed->f2<3) {
eed->v1->f |= 128; /* = no loose vert! */
eed->v2->f |= 128;
}
if( (eed->f2==1 || eed->f2==2) ) {
/* if del_old, the preferred normal direction is exact opposite as for keep old faces */
if(eed->dir != del_old)
efa2 = addfacelist(em, eed->v1, eed->v2,
eed->v2->tmp.v, eed->v1->tmp.v,
eed->tmp.f, NULL);
else
efa2 = addfacelist(em, eed->v2, eed->v1,
eed->v1->tmp.v, eed->v2->tmp.v,
eed->tmp.f, NULL);
/* Needs smarter adaption of existing creases.
* If addedgelist is used, make sure seams are set to 0 on these
* new edges, since we do not want to add any seams on extrusion.
*/
efa2->e1->crease= eed->crease;
efa2->e2->crease= eed->crease;
efa2->e3->crease= eed->crease;
if(efa2->e4) efa2->e4->crease= eed->crease;
}
eed= nexted;
}
if(del_old) {
eed= em->edges.first;
while(eed) {
nexted= eed->next;
if(eed->f2==3 && eed->f1==1) {
remedge(em, eed);
free_editedge(em, eed);
}
eed= nexted;
}
}
/* duplicate faces, if necessary remove old ones */
efa= em->faces.first;
while(efa) {
nextvl= efa->next;
if(efa->f1 & 1) {
v1 = efa->v1->tmp.v;
v2 = efa->v2->tmp.v;
v3 = efa->v3->tmp.v;
if(efa->v4)
v4 = efa->v4->tmp.v;
else
v4= 0;
/* hmm .. not sure about edges here */
if(del_old==0) // if we keep old, we flip normal
efa2= addfacelist(em, v3, v2, v1, v4, efa, efa);
else
efa2= addfacelist(em, v1, v2, v3, v4, efa, efa);
/* for transform */
add_normal_aligned(nor, efa->n);
if(del_old) {
BLI_remlink(&em->faces, efa);
free_editface(em, efa);
}
}
efa= nextvl;
}
normalize_v3(nor); // for grab
/* for all vertices with eve->tmp.v!=0
if eve->f1==1: make edge
if flag!=128 : if del_old==1: remove
*/
eve= em->verts.last;
while(eve) {
nextve= eve->prev;
if(eve->tmp.v) {
if(eve->f1==1) addedgelist(em, eve, eve->tmp.v, NULL);
else if( (eve->f & 128)==0) {
if(del_old) {
BLI_remlink(&em->verts,eve);
free_editvert(em, eve);
eve= NULL;
}
}
}
if(eve) {
eve->f &= ~128;
}
eve= nextve;
}
// since its vertex select mode now, it also deselects higher order
EM_selectmode_flush(em);
if(nor[0]==0.0 && nor[1]==0.0 && nor[2]==0.0) return 'g'; // g is grab, for correct undo print
return 'n';
}
/* generic extrude */
short extrudeflag(Object *obedit, EditMesh *em, short flag, float *nor, int all)
{
if(em->selectmode & SCE_SELECT_VERTEX)
return extrudeflag_vert(obedit, em, flag, nor, all);
else
return extrudeflag_edge(obedit, em, flag, nor, all);
}
void rotateflag(EditMesh *em, short flag, float *cent, float rotmat[][3])
{
/* all verts with (flag & 'flag') rotate */
EditVert *eve;
eve= em->verts.first;
while(eve) {
if(eve->f & flag) {
eve->co[0]-=cent[0];
eve->co[1]-=cent[1];
eve->co[2]-=cent[2];
mul_m3_v3(rotmat,eve->co);
eve->co[0]+=cent[0];
eve->co[1]+=cent[1];
eve->co[2]+=cent[2];
}
eve= eve->next;
}
}
void translateflag(EditMesh *em, short flag, float *vec)
{
/* all verts with (flag & 'flag') translate */
EditVert *eve;
eve= em->verts.first;
while(eve) {
if(eve->f & flag) {
eve->co[0]+=vec[0];
eve->co[1]+=vec[1];
eve->co[2]+=vec[2];
}
eve= eve->next;
}
}
/* helper call for below */
static EditVert *adduplicate_vertex(EditMesh *em, EditVert *eve, int flag)
{
/* FIXME: copy deformation weight from eve ok here? */
EditVert *v1= addvertlist(em, eve->co, eve);
v1->f= eve->f;
eve->f-= flag;
eve->f|= 128;
eve->tmp.v = v1;
return v1;
}
/* old selection has flag 128 set, and flag 'flag' cleared
new selection has flag 'flag' set */
void adduplicateflag(EditMesh *em, int flag)
{
EditVert *eve, *v1, *v2, *v3, *v4;
EditEdge *eed, *newed;
EditFace *efa, *newfa, *act_efa = EM_get_actFace(em, 0);
EM_clear_flag_all(em, 128);
EM_selectmode_set(em); // paranoia check, selection now is consistent
/* vertices first */
for(eve= em->verts.last; eve; eve= eve->prev) {
if(eve->f & flag)
adduplicate_vertex(em, eve, flag);
else
eve->tmp.v = NULL;
}
/* copy edges, note that vertex selection can be independent of edge */
for(eed= em->edges.last; eed; eed= eed->prev) {
if( eed->f & flag ) {
v1 = eed->v1->tmp.v;
if(v1==NULL) v1= adduplicate_vertex(em, eed->v1, flag);
v2 = eed->v2->tmp.v;
if(v2==NULL) v2= adduplicate_vertex(em, eed->v2, flag);
newed= addedgelist(em, v1, v2, eed);
newed->f= eed->f;
eed->f -= flag;
eed->f |= 128;
}
}
/* then duplicate faces, again create new vertices if needed */
for(efa= em->faces.last; efa; efa= efa->prev) {
if(efa->f & flag) {
v1 = efa->v1->tmp.v;
if(v1==NULL) v1= adduplicate_vertex(em, efa->v1, flag);
v2 = efa->v2->tmp.v;
if(v2==NULL) v2= adduplicate_vertex(em, efa->v2, flag);
v3 = efa->v3->tmp.v;
if(v3==NULL) v3= adduplicate_vertex(em, efa->v3, flag);
if(efa->v4) {
v4 = efa->v4->tmp.v;
if(v4==NULL) v4= adduplicate_vertex(em, efa->v4, flag);
}
else v4= NULL;
newfa= addfacelist(em, v1, v2, v3, v4, efa, efa);
if (efa==act_efa) {
EM_set_actFace(em, newfa);
}
newfa->f= efa->f;
efa->f -= flag;
efa->f |= 128;
}
}
EM_fgon_flags(em); // redo flags and indices for fgons
}
void delfaceflag(EditMesh *em, int flag)
{
/* delete all faces with 'flag', including loose edges and loose vertices */
/* this is maybe a bit weird, but this function is used for 'split' and 'separate' */
/* in remaining vertices/edges 'flag' is cleared */
EditVert *eve,*nextve;
EditEdge *eed, *nexted;
EditFace *efa,*nextvl;
/* to detect loose edges, we put f2 flag on 1 */
for(eed= em->edges.first; eed; eed= eed->next) {
if(eed->f & flag) eed->f2= 1;
else eed->f2= 0;
}
/* delete faces */
efa= em->faces.first;
while(efa) {
nextvl= efa->next;
if(efa->f & flag) {
efa->e1->f2= 1;
efa->e2->f2= 1;
efa->e3->f2= 1;
if(efa->e4) {
efa->e4->f2= 1;
}
BLI_remlink(&em->faces, efa);
free_editface(em, efa);
}
efa= nextvl;
}
/* all remaining faces: make sure we keep the edges */
for(efa= em->faces.first; efa; efa= efa->next) {
efa->e1->f2= 0;
efa->e2->f2= 0;
efa->e3->f2= 0;
if(efa->e4) {
efa->e4->f2= 0;
}
}
/* remove tagged edges, and clear remaining ones */
eed= em->edges.first;
while(eed) {
nexted= eed->next;
if(eed->f2==1) {
remedge(em, eed);
free_editedge(em, eed);
}
else {
eed->f &= ~flag;
eed->v1->f &= ~flag;
eed->v2->f &= ~flag;
}
eed= nexted;
}
/* vertices with 'flag' now are the loose ones, and will be removed */
eve= em->verts.first;
while(eve) {
nextve= eve->next;
if(eve->f & flag) {
BLI_remlink(&em->verts, eve);
free_editvert(em, eve);
}
eve= nextve;
}
}
/* ********************* */
#if 0
static int check_vnormal_flip(float *n, float *vnorm)
{
float inp;
inp= n[0]*vnorm[0]+n[1]*vnorm[1]+n[2]*vnorm[2];
/* angles 90 degrees: dont flip */
if(inp> -0.000001) return 0;
return 1;
}
#endif
/* does face centers too */
void recalc_editnormals(EditMesh *em)
{
EditFace *efa;
EditVert *eve;
for(eve= em->verts.first; eve; eve=eve->next) {
eve->no[0] = eve->no[1] = eve->no[2] = 0.0;
}
for(efa= em->faces.first; efa; efa=efa->next) {
if(efa->v4) {
normal_quad_v3( efa->n,efa->v1->co, efa->v2->co, efa->v3->co, efa->v4->co);
cent_quad_v3(efa->cent, efa->v1->co, efa->v2->co, efa->v3->co, efa->v4->co);
add_v3_v3(efa->v4->no, efa->n);
}
else {
normal_tri_v3( efa->n,efa->v1->co, efa->v2->co, efa->v3->co);
cent_tri_v3(efa->cent, efa->v1->co, efa->v2->co, efa->v3->co);
}
add_v3_v3(efa->v1->no, efa->n);
add_v3_v3(efa->v2->no, efa->n);
add_v3_v3(efa->v3->no, efa->n);
}
/* following Mesh convention; we use vertex coordinate itself for normal in this case */
for(eve= em->verts.first; eve; eve=eve->next) {
if (normalize_v3(eve->no)==0.0) {
VECCOPY(eve->no, eve->co);
normalize_v3(eve->no);
}
}
}
int compareface(EditFace *vl1, EditFace *vl2)
{
EditVert *v1, *v2, *v3, *v4;
if(vl1->v4 && vl2->v4) {
v1= vl2->v1;
v2= vl2->v2;
v3= vl2->v3;
v4= vl2->v4;
if(vl1->v1==v1 || vl1->v2==v1 || vl1->v3==v1 || vl1->v4==v1) {
if(vl1->v1==v2 || vl1->v2==v2 || vl1->v3==v2 || vl1->v4==v2) {
if(vl1->v1==v3 || vl1->v2==v3 || vl1->v3==v3 || vl1->v4==v3) {
if(vl1->v1==v4 || vl1->v2==v4 || vl1->v3==v4 || vl1->v4==v4) {
return 1;
}
}
}
}
}
else if(vl1->v4==0 && vl2->v4==0) {
v1= vl2->v1;
v2= vl2->v2;
v3= vl2->v3;
if(vl1->v1==v1 || vl1->v2==v1 || vl1->v3==v1) {
if(vl1->v1==v2 || vl1->v2==v2 || vl1->v3==v2) {
if(vl1->v1==v3 || vl1->v2==v3 || vl1->v3==v3) {
return 1;
}
}
}
}
return 0;
}
/* checks for existence, not tria overlapping inside quad */
EditFace *exist_face(EditMesh *em, EditVert *v1, EditVert *v2, EditVert *v3, EditVert *v4)
{
EditFace *efa, efatest;
efatest.v1= v1;
efatest.v2= v2;
efatest.v3= v3;
efatest.v4= v4;
efa= em->faces.first;
while(efa) {
if(compareface(&efatest, efa)) return efa;
efa= efa->next;
}
return NULL;
}
/* evaluate if entire quad is a proper convex quad */
int convex(float *v1, float *v2, float *v3, float *v4)
{
float nor[3], nor1[3], nor2[3], vec[4][2];
/* define projection, do both trias apart, quad is undefined! */
normal_tri_v3( nor1,v1, v2, v3);
normal_tri_v3( nor2,v1, v3, v4);
nor[0]= ABS(nor1[0]) + ABS(nor2[0]);
nor[1]= ABS(nor1[1]) + ABS(nor2[1]);
nor[2]= ABS(nor1[2]) + ABS(nor2[2]);
if(nor[2] >= nor[0] && nor[2] >= nor[1]) {
vec[0][0]= v1[0]; vec[0][1]= v1[1];
vec[1][0]= v2[0]; vec[1][1]= v2[1];
vec[2][0]= v3[0]; vec[2][1]= v3[1];
vec[3][0]= v4[0]; vec[3][1]= v4[1];
}
else if(nor[1] >= nor[0] && nor[1]>= nor[2]) {
vec[0][0]= v1[0]; vec[0][1]= v1[2];
vec[1][0]= v2[0]; vec[1][1]= v2[2];
vec[2][0]= v3[0]; vec[2][1]= v3[2];
vec[3][0]= v4[0]; vec[3][1]= v4[2];
}
else {
vec[0][0]= v1[1]; vec[0][1]= v1[2];
vec[1][0]= v2[1]; vec[1][1]= v2[2];
vec[2][0]= v3[1]; vec[2][1]= v3[2];
vec[3][0]= v4[1]; vec[3][1]= v4[2];
}
/* linetests, the 2 diagonals have to instersect to be convex */
if( isect_line_line_v2(vec[0], vec[2], vec[1], vec[3]) > 0 ) return 1;
return 0;
}
/* ********************* Fake Polgon support (FGon) ***************** */
/* results in:
- faces having ->fgonf flag set (also for draw)
- edges having ->fgoni index set (for select)
*/
float EM_face_area(EditFace *efa)
{
if(efa->v4) return area_quad_v3(efa->v1->co, efa->v2->co, efa->v3->co, efa->v4->co);
else return area_tri_v3(efa->v1->co, efa->v2->co, efa->v3->co);
}
float EM_face_perimeter(EditFace *efa)
{
if(efa->v4) return
len_v3v3(efa->v1->co, efa->v2->co)+
len_v3v3(efa->v2->co, efa->v3->co)+
len_v3v3(efa->v3->co, efa->v4->co)+
len_v3v3(efa->v4->co, efa->v1->co);
else return
len_v3v3(efa->v1->co, efa->v2->co)+
len_v3v3(efa->v2->co, efa->v3->co)+
len_v3v3(efa->v3->co, efa->v1->co);
}
void EM_fgon_flags(EditMesh *em)
{
EditFace *efa, *efan, *efamax;
EditEdge *eed;
ListBase listb={NULL, NULL};
float size, maxsize;
short done, curindex= 1;
// for each face with fgon edge AND not fgon flag set
for(eed= em->edges.first; eed; eed= eed->next) eed->fgoni= 0; // index
for(efa= em->faces.first; efa; efa= efa->next) efa->fgonf= 0; // flag
// for speed & simplicity, put fgon face candidates in new listbase
efa= em->faces.first;
while(efa) {
efan= efa->next;
if( (efa->e1->h & EM_FGON) || (efa->e2->h & EM_FGON) ||
(efa->e3->h & EM_FGON) || (efa->e4 && (efa->e4->h & EM_FGON)) ) {
BLI_remlink(&em->faces, efa);
BLI_addtail(&listb, efa);
}
efa= efan;
}
// find an undone face with fgon edge
for(efa= listb.first; efa; efa= efa->next) {
if(efa->fgonf==0) {
// init this face
efa->fgonf= EM_FGON;
if(efa->e1->h & EM_FGON) efa->e1->fgoni= curindex;
if(efa->e2->h & EM_FGON) efa->e2->fgoni= curindex;
if(efa->e3->h & EM_FGON) efa->e3->fgoni= curindex;
if(efa->e4 && (efa->e4->h & EM_FGON)) efa->e4->fgoni= curindex;
// we search for largest face, to give facedot drawing rights
maxsize= EM_face_area(efa);
efamax= efa;
// now flush curendex over edges and set faceflags
done= 1;
while(done==1) {
done= 0;
for(efan= listb.first; efan; efan= efan->next) {
if(efan->fgonf==0) {
// if one if its edges has index set, do other too
if( (efan->e1->fgoni==curindex) || (efan->e2->fgoni==curindex) ||
(efan->e3->fgoni==curindex) || (efan->e4 && (efan->e4->fgoni==curindex)) ) {
efan->fgonf= EM_FGON;
if(efan->e1->h & EM_FGON) efan->e1->fgoni= curindex;
if(efan->e2->h & EM_FGON) efan->e2->fgoni= curindex;
if(efan->e3->h & EM_FGON) efan->e3->fgoni= curindex;
if(efan->e4 && (efan->e4->h & EM_FGON)) efan->e4->fgoni= curindex;
size= EM_face_area(efan);
if(size>maxsize) {
efamax= efan;
maxsize= size;
}
done= 1;
}
}
}
}
efamax->fgonf |= EM_FGON_DRAW;
curindex++;
}
}
// put fgon face candidates back in listbase
efa= listb.first;
while(efa) {
efan= efa->next;
BLI_remlink(&listb, efa);
BLI_addtail(&em->faces, efa);
efa= efan;
}
// remove fgon flags when edge not in fgon (anymore)
for(eed= em->edges.first; eed; eed= eed->next) {
if(eed->fgoni==0) eed->h &= ~EM_FGON;
}
}
/* editmesh vertmap, copied from intern.mesh.c
* if do_face_idx_array is 0 it means we need to run it as well as freeing
* */
UvVertMap *EM_make_uv_vert_map(EditMesh *em, int selected, int do_face_idx_array, float *limit)
{
EditVert *ev;
EditFace *efa;
int totverts;
/* vars from original func */
UvVertMap *vmap;
UvMapVert *buf;
MTFace *tf;
unsigned int a;
int i, totuv, nverts;
if (do_face_idx_array)
EM_init_index_arrays(em, 0, 0, 1);
/* we need the vert */
for (ev= em->verts.first, totverts=0; ev; ev= ev->next, totverts++) {
ev->tmp.l = totverts;
}
totuv = 0;
/* generate UvMapVert array */
for (efa= em->faces.first; efa; efa= efa->next)
if(!selected || ((!efa->h) && (efa->f & SELECT)))
totuv += (efa->v4)? 4: 3;
if(totuv==0) {
if (do_face_idx_array)
EM_free_index_arrays();
return NULL;
}
vmap= (UvVertMap*)MEM_callocN(sizeof(*vmap), "UvVertMap");
if (!vmap) {
if (do_face_idx_array)
EM_free_index_arrays();
return NULL;
}
vmap->vert= (UvMapVert**)MEM_callocN(sizeof(*vmap->vert)*totverts, "UvMapVert*");
buf= vmap->buf= (UvMapVert*)MEM_callocN(sizeof(*vmap->buf)*totuv, "UvMapVert");
if (!vmap->vert || !vmap->buf) {
free_uv_vert_map(vmap);
if (do_face_idx_array)
EM_free_index_arrays();
return NULL;
}
for (a=0, efa= em->faces.first; efa; a++, efa= efa->next) {
if(!selected || ((!efa->h) && (efa->f & SELECT))) {
nverts= (efa->v4)? 4: 3;
for(i=0; i<nverts; i++) {
buf->tfindex= i;
buf->f= a;
buf->separate = 0;
buf->next= vmap->vert[(*(&efa->v1 + i))->tmp.l];
vmap->vert[(*(&efa->v1 + i))->tmp.l]= buf;
buf++;
}
}
}
/* sort individual uvs for each vert */
for(a=0, ev=em->verts.first; ev; a++, ev= ev->next) {
UvMapVert *newvlist= NULL, *vlist=vmap->vert[a];
UvMapVert *iterv, *v, *lastv, *next;
float *uv, *uv2, uvdiff[2];
while(vlist) {
v= vlist;
vlist= vlist->next;
v->next= newvlist;
newvlist= v;
efa = EM_get_face_for_index(v->f);
tf = CustomData_em_get(&em->fdata, efa->data, CD_MTFACE);
uv = tf->uv[v->tfindex];
lastv= NULL;
iterv= vlist;
while(iterv) {
next= iterv->next;
efa = EM_get_face_for_index(iterv->f);
tf = CustomData_em_get(&em->fdata, efa->data, CD_MTFACE);
uv2 = tf->uv[iterv->tfindex];
sub_v2_v2v2(uvdiff, uv2, uv);
if(fabs(uv[0]-uv2[0]) < limit[0] && fabs(uv[1]-uv2[1]) < limit[1]) {
if(lastv) lastv->next= next;
else vlist= next;
iterv->next= newvlist;
newvlist= iterv;
}
else
lastv=iterv;
iterv= next;
}
newvlist->separate = 1;
}
vmap->vert[a]= newvlist;
}
if (do_face_idx_array)
EM_free_index_arrays();
return vmap;
}
UvMapVert *EM_get_uv_map_vert(UvVertMap *vmap, unsigned int v)
{
return vmap->vert[v];
}
void EM_free_uv_vert_map(UvVertMap *vmap)
{
if (vmap) {
if (vmap->vert) MEM_freeN(vmap->vert);
if (vmap->buf) MEM_freeN(vmap->buf);
MEM_freeN(vmap);
}
}
/* poll call for mesh operators requiring a view3d context */
int EM_view3d_poll(bContext *C)
{
if(ED_operator_editmesh(C) && ED_operator_view3d_active(C))
return 1;
return 0;
}
/* higher quality normals */
/* NormalCalc */
/* NormalCalc modifier: calculates higher quality normals
*/
/* each edge uses this to */
typedef struct EdgeFaceRef {
int f1; /* init as -1 */
int f2;
} EdgeFaceRef;
void EM_make_hq_normals(EditMesh *em)
{
EditFace *efa;
EditVert *eve;
int i;
EdgeHash *edge_hash = BLI_edgehash_new();
EdgeHashIterator *edge_iter;
int edge_ref_count = 0;
int ed_v1, ed_v2; /* use when getting the key */
EdgeFaceRef *edge_ref_array = MEM_callocN(em->totedge * sizeof(EdgeFaceRef), "Edge Connectivity");
EdgeFaceRef *edge_ref;
float edge_normal[3];
EM_init_index_arrays(em, 1, 1, 1);
for(eve= em->verts.first, i=0; eve; eve= eve->next, i++) {
zero_v3(eve->no);
eve->tmp.l= i;
}
/* This function adds an edge hash if its not there, and adds the face index */
#define NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE(EDV1, EDV2); \
edge_ref = (EdgeFaceRef *)BLI_edgehash_lookup(edge_hash, EDV1, EDV2); \
if (!edge_ref) { \
edge_ref = &edge_ref_array[edge_ref_count]; edge_ref_count++; \
edge_ref->f1=i; \
edge_ref->f2=-1; \
BLI_edgehash_insert(edge_hash, EDV1, EDV2, edge_ref); \
} else { \
edge_ref->f2=i; \
}
efa= em->faces.first;
for(i = 0; i < em->totface; i++, efa= efa->next) {
if(efa->v4) {
NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE(efa->v1->tmp.l, efa->v2->tmp.l);
NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE(efa->v2->tmp.l, efa->v3->tmp.l);
NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE(efa->v3->tmp.l, efa->v4->tmp.l);
NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE(efa->v4->tmp.l, efa->v1->tmp.l);
} else {
NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE(efa->v1->tmp.l, efa->v2->tmp.l);
NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE(efa->v2->tmp.l, efa->v3->tmp.l);
NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE(efa->v3->tmp.l, efa->v1->tmp.l);
}
}
#undef NOCALC_EDGEWEIGHT_ADD_EDGEREF_FACE
for(edge_iter = BLI_edgehashIterator_new(edge_hash); !BLI_edgehashIterator_isDone(edge_iter); BLI_edgehashIterator_step(edge_iter)) {
/* Get the edge vert indicies, and edge value (the face indicies that use it)*/
BLI_edgehashIterator_getKey(edge_iter, (int*)&ed_v1, (int*)&ed_v2);
edge_ref = BLI_edgehashIterator_getValue(edge_iter);
if (edge_ref->f2 != -1) {
EditFace *ef1= EM_get_face_for_index(edge_ref->f1), *ef2= EM_get_face_for_index(edge_ref->f2);
float angle= angle_normalized_v3v3(ef1->n, ef2->n);
if(angle > 0.0f) {
/* We have 2 faces using this edge, calculate the edges normal
* using the angle between the 2 faces as a weighting */
add_v3_v3v3(edge_normal, ef1->n, ef2->n);
normalize_v3(edge_normal);
mul_v3_fl(edge_normal, angle);
}
else {
/* cant do anything useful here!
Set the face index for a vert incase it gets a zero normal */
EM_get_vert_for_index(ed_v1)->tmp.l=
EM_get_vert_for_index(ed_v2)->tmp.l= -(edge_ref->f1 + 1);
continue;
}
} else {
/* only one face attached to that edge */
/* an edge without another attached- the weight on this is
* undefined, M_PI/2 is 90d in radians and that seems good enough */
VECCOPY(edge_normal, EM_get_face_for_index(edge_ref->f1)->n)
mul_v3_fl(edge_normal, M_PI/2);
}
add_v3_v3(EM_get_vert_for_index(ed_v1)->no, edge_normal );
add_v3_v3(EM_get_vert_for_index(ed_v2)->no, edge_normal );
}
BLI_edgehashIterator_free(edge_iter);
BLI_edgehash_free(edge_hash, NULL);
MEM_freeN(edge_ref_array);
/* normalize vertex normals and assign */
for(eve= em->verts.first; eve; eve= eve->next) {
if(normalize_v3(eve->no) == 0.0f && eve->tmp.l < 0) {
/* exceptional case, totally flat */
efa= EM_get_face_for_index(-(eve->tmp.l) - 1);
VECCOPY(eve->no, efa->n);
}
}
EM_free_index_arrays();
}
void EM_solidify(EditMesh *em, float dist)
{
EditFace *efa;
EditVert *eve;
float *vert_angles= MEM_callocN(sizeof(float) * em->totvert * 2, "EM_solidify"); /* 2 in 1 */
float *vert_accum= vert_angles + em->totvert;
float face_angles[4];
int i, j;
for(eve= em->verts.first, i=0; eve; eve= eve->next, i++) {
eve->tmp.l= i;
}
efa= em->faces.first;
for(i = 0; i < em->totface; i++, efa= efa->next) {
if(!(efa->f & SELECT))
continue;
if(efa->v4) {
angle_quad_v3(face_angles, efa->v1->co, efa->v2->co, efa->v3->co, efa->v4->co);
j= 3;
}
else {
angle_tri_v3(face_angles, efa->v1->co, efa->v2->co, efa->v3->co);
j= 2;
}
for(; j>=0; j--) {
eve= *(&efa->v1 + j);
vert_accum[eve->tmp.l] += face_angles[j];
vert_angles[eve->tmp.l]+= shell_angle_to_dist(angle_normalized_v3v3(eve->no, efa->n)) * face_angles[j];
}
}
for(eve= em->verts.first, i=0; eve; eve= eve->next, i++) {
if(vert_accum[i]) { /* zero if unselected */
madd_v3_v3fl(eve->co, eve->no, dist * vert_angles[i] / vert_accum[i]);
}
}
MEM_freeN(vert_angles);
}
/* not that optimal!, should be nicer with bmesh */
static void tag_face_edges(EditFace *efa)
{
if(efa->v4)
efa->e1->tmp.l= efa->e2->tmp.l= efa->e3->tmp.l= efa->e4->tmp.l= 1;
else
efa->e1->tmp.l= efa->e2->tmp.l= efa->e3->tmp.l= 1;
}
static int tag_face_edges_test(EditFace *efa)
{
if(efa->v4)
return (efa->e1->tmp.l || efa->e2->tmp.l || efa->e3->tmp.l || efa->e4->tmp.l) ? 1:0;
else
return (efa->e1->tmp.l || efa->e2->tmp.l || efa->e3->tmp.l) ? 1:0;
}
void em_deselect_nth_face(EditMesh *em, int nth, EditFace *efa_act)
{
EditFace *efa;
EditEdge *eed;
int ok= 1;
if(efa_act==NULL) {
return;
}
/* to detect loose edges, we put f2 flag on 1 */
for(eed= em->edges.first; eed; eed= eed->next) {
eed->tmp.l= 0;
}
for (efa= em->faces.first; efa; efa= efa->next) {
efa->tmp.l = 0;
}
efa_act->tmp.l = 1;
while(ok) {
ok = 0;
for (efa= em->faces.first; efa; efa= efa->next) {
if(efa->tmp.l==1) { /* initialize */
tag_face_edges(efa);
}
if(efa->tmp.l)
efa->tmp.l++;
}
for (efa= em->faces.first; efa; efa= efa->next) {
if(efa->tmp.l==0 && tag_face_edges_test(efa)) {
efa->tmp.l= 1;
ok = 1; /* keep looping */
}
}
}
for (efa= em->faces.first; efa; efa= efa->next) {
if(efa->tmp.l > 0 && efa->tmp.l % nth) {
EM_select_face(efa, 0);
}
}
for (efa= em->faces.first; efa; efa= efa->next) {
if(efa->f & SELECT) {
EM_select_face(efa, 1);
}
}
EM_nvertices_selected(em);
EM_nedges_selected(em);
EM_nfaces_selected(em);
}
/* not that optimal!, should be nicer with bmesh */
static void tag_edge_verts(EditEdge *eed)
{
eed->v1->tmp.l= eed->v2->tmp.l= 1;
}
static int tag_edge_verts_test(EditEdge *eed)
{
return (eed->v1->tmp.l || eed->v2->tmp.l) ? 1:0;
}
void em_deselect_nth_edge(EditMesh *em, int nth, EditEdge *eed_act)
{
EditEdge *eed;
EditVert *eve;
int ok= 1;
if(eed_act==NULL) {
return;
}
for(eve= em->verts.first; eve; eve= eve->next) {
eve->tmp.l= 0;
}
for (eed= em->edges.first; eed; eed= eed->next) {
eed->tmp.l = 0;
}
eed_act->tmp.l = 1;
while(ok) {
ok = 0;
for (eed= em->edges.first; eed; eed= eed->next) {
if(eed->tmp.l==1) { /* initialize */
tag_edge_verts(eed);
}
if(eed->tmp.l)
eed->tmp.l++;
}
for (eed= em->edges.first; eed; eed= eed->next) {
if(eed->tmp.l==0 && tag_edge_verts_test(eed)) {
eed->tmp.l= 1;
ok = 1; /* keep looping */
}
}
}
for (eed= em->edges.first; eed; eed= eed->next) {
if(eed->tmp.l > 0 && eed->tmp.l % nth) {
EM_select_edge(eed, 0);
}
}
for (eed= em->edges.first; eed; eed= eed->next) {
if(eed->f & SELECT) {
EM_select_edge(eed, 1);
}
}
{
/* grr, should be a function */
EditFace *efa;
for (efa= em->faces.first; efa; efa= efa->next) {
if(efa->v4) {
if(efa->e1->f & efa->e2->f & efa->e3->f & efa->e4->f & SELECT );
else efa->f &= ~SELECT;
}
else {
if(efa->e1->f & efa->e2->f & efa->e3->f & SELECT );
else efa->f &= ~SELECT;
}
}
}
EM_nvertices_selected(em);
EM_nedges_selected(em);
EM_nfaces_selected(em);
}
void em_deselect_nth_vert(EditMesh *em, int nth, EditVert *eve_act)
{
EditVert *eve;
EditEdge *eed;
int ok= 1;
if(eve_act==NULL) {
return;
}
for (eve= em->verts.first; eve; eve= eve->next) {
eve->tmp.l = 0;
}
eve_act->tmp.l = 1;
while(ok) {
ok = 0;
for (eve= em->verts.first; eve; eve= eve->next) {
if(eve->tmp.l)
eve->tmp.l++;
}
for (eed= em->edges.first; eed; eed= eed->next) {
if(eed->v1->tmp.l==2 && eed->v2->tmp.l==0) { /* initialize */
eed->v2->tmp.l= 1;
ok = 1; /* keep looping */
}
else if(eed->v2->tmp.l==2 && eed->v1->tmp.l==0) { /* initialize */
eed->v1->tmp.l= 1;
ok = 1; /* keep looping */
}
}
}
for (eve= em->verts.first; eve; eve= eve->next) {
if(eve->tmp.l > 0 && eve->tmp.l % nth) {
eve->f &= ~SELECT;
}
}
EM_deselect_flush(em);
EM_nvertices_selected(em);
// EM_nedges_selected(em); // flush does these
// EM_nfaces_selected(em); // flush does these
}
int EM_deselect_nth(EditMesh *em, int nth)
{
EditSelection *ese;
ese = ((EditSelection*)em->selected.last);
if(ese) {
if(ese->type == EDITVERT) {
em_deselect_nth_vert(em, nth, (EditVert*)ese->data);
return 1;
}
if(ese->type == EDITEDGE) {
em_deselect_nth_edge(em, nth, (EditEdge*)ese->data);
return 1;
}
}
else {
EditFace *efa_act = EM_get_actFace(em, 0);
if(efa_act) {
em_deselect_nth_face(em, nth, efa_act);
return 1;
}
}
return 0;
}
View Git Blame Copy Permalink