2012-10-24 10:39:11 +00:00
|
|
|
/*
|
|
|
|
* ***** 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) 2005 by the Blender Foundation.
|
|
|
|
* All rights reserved.
|
|
|
|
*
|
|
|
|
* Contributor(s): Alexander Pinzon
|
|
|
|
*
|
|
|
|
* ***** END GPL LICENSE BLOCK *****
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
|
|
|
/** \file blender/modifiers/intern/MOD_laplaciansmooth.c
|
|
|
|
* \ingroup modifiers
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
|
|
#include "DNA_meshdata_types.h"
|
|
|
|
#include "DNA_object_types.h"
|
|
|
|
|
|
|
|
#include "BLI_math.h"
|
|
|
|
#include "BLI_utildefines.h"
|
|
|
|
#include "BLI_string.h"
|
|
|
|
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
|
|
|
|
#include "BKE_cdderivedmesh.h"
|
|
|
|
#include "BKE_deform.h"
|
|
|
|
#include "BKE_displist.h"
|
|
|
|
#include "BKE_mesh.h"
|
|
|
|
#include "BKE_modifier.h"
|
|
|
|
#include "BKE_object.h"
|
|
|
|
#include "BKE_particle.h"
|
|
|
|
#include "BKE_tessmesh.h"
|
|
|
|
|
|
|
|
#include "MOD_modifiertypes.h"
|
|
|
|
#include "MOD_util.h"
|
|
|
|
|
|
|
|
#include "ONL_opennl.h"
|
|
|
|
|
|
|
|
#define MOD_LAPLACIANSMOOTH_MAX_EDGE_PERCENTAGE 1.8
|
|
|
|
#define MOD_LAPLACIANSMOOTH_MIN_EDGE_PERCENTAGE 0.02
|
|
|
|
|
|
|
|
struct BLaplacianSystem {
|
2012-10-24 11:31:57 +00:00
|
|
|
float *eweights; /* Length weights per Edge */
|
2012-10-24 10:39:11 +00:00
|
|
|
float (*fweights)[3]; /* Cotangent weights per face */
|
2012-10-24 11:31:57 +00:00
|
|
|
float *ring_areas; /* Total area per ring*/
|
|
|
|
float *vlengths; /* Total sum of lengths(edges) per vertice*/
|
|
|
|
float *vweights; /* Total sum of weights per vertice*/
|
|
|
|
int numEdges; /* Number of edges*/
|
|
|
|
int numFaces; /* Number of faces*/
|
|
|
|
int numVerts; /* Number of verts*/
|
|
|
|
short *numNeFa; /* Number of neighboors faces around vertice*/
|
|
|
|
short *numNeEd; /* Number of neighboors Edges around vertice*/
|
|
|
|
short *zerola; /* Is zero area or length*/
|
2012-10-24 10:39:11 +00:00
|
|
|
|
|
|
|
/* Pointers to data*/
|
|
|
|
float (*vertexCos)[3];
|
2012-10-24 11:31:57 +00:00
|
|
|
MFace *mfaces;
|
2012-10-24 10:39:11 +00:00
|
|
|
MEdge *medges;
|
|
|
|
NLContext *context;
|
|
|
|
|
|
|
|
/*Data*/
|
|
|
|
float min_area;
|
|
|
|
float vert_centroid[3];
|
|
|
|
};
|
|
|
|
typedef struct BLaplacianSystem LaplacianSystem;
|
|
|
|
|
|
|
|
static CustomDataMask required_data_mask(Object *UNUSED(ob), ModifierData *md);
|
|
|
|
static int is_disabled(ModifierData *md, int UNUSED(useRenderParams));
|
|
|
|
static float compute_volume(float (*vertexCos)[3], MFace *mfaces, int numFaces);
|
|
|
|
static float cotan_weight(float *v1, float *v2, float *v3);
|
2012-10-27 10:42:28 +00:00
|
|
|
static LaplacianSystem *init_laplacian_system(int a_numEdges, int a_numFaces, int a_numVerts);
|
2012-10-24 10:39:11 +00:00
|
|
|
static void copy_data(ModifierData *md, ModifierData *target);
|
2012-10-24 11:31:57 +00:00
|
|
|
static void delete_laplacian_system(LaplacianSystem *sys);
|
2012-10-27 10:42:28 +00:00
|
|
|
static void delete_void_pointer(void *data);
|
2012-10-24 11:31:57 +00:00
|
|
|
static void fill_laplacian_matrix(LaplacianSystem *sys);
|
2012-10-24 10:39:11 +00:00
|
|
|
static void init_data(ModifierData *md);
|
2012-10-24 11:31:57 +00:00
|
|
|
static void init_laplacian_matrix(LaplacianSystem *sys);
|
2012-10-24 10:39:11 +00:00
|
|
|
static void memset_laplacian_system(LaplacianSystem *sys, int val);
|
|
|
|
static void volume_preservation(LaplacianSystem *sys, float vini, float vend, short flag);
|
2012-10-24 11:31:57 +00:00
|
|
|
static void validate_solution(LaplacianSystem *sys, short flag);
|
2012-10-24 10:39:11 +00:00
|
|
|
|
2012-10-27 10:42:28 +00:00
|
|
|
static void delete_void_pointer(void *data)
|
2012-10-24 10:39:11 +00:00
|
|
|
{
|
|
|
|
if (data) {
|
|
|
|
MEM_freeN(data);
|
|
|
|
data = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-10-24 11:31:57 +00:00
|
|
|
static void delete_laplacian_system(LaplacianSystem *sys)
|
2012-10-24 10:39:11 +00:00
|
|
|
{
|
|
|
|
delete_void_pointer(sys->eweights);
|
|
|
|
delete_void_pointer(sys->fweights);
|
|
|
|
delete_void_pointer(sys->numNeEd);
|
|
|
|
delete_void_pointer(sys->numNeFa);
|
|
|
|
delete_void_pointer(sys->ring_areas);
|
|
|
|
delete_void_pointer(sys->vlengths);
|
|
|
|
delete_void_pointer(sys->vweights);
|
|
|
|
delete_void_pointer(sys->zerola);
|
|
|
|
if (sys->context) {
|
|
|
|
nlDeleteContext(sys->context);
|
|
|
|
}
|
|
|
|
sys->vertexCos = NULL;
|
2012-10-24 11:31:57 +00:00
|
|
|
sys->mfaces = NULL;
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->medges = NULL;
|
|
|
|
MEM_freeN(sys);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void memset_laplacian_system(LaplacianSystem *sys, int val)
|
|
|
|
{
|
2012-10-24 11:31:57 +00:00
|
|
|
memset(sys->eweights, val, sizeof(float) * sys->numEdges);
|
|
|
|
memset(sys->fweights, val, sizeof(float) * sys->numFaces * 3);
|
|
|
|
memset(sys->numNeEd, val, sizeof(short) * sys->numVerts);
|
|
|
|
memset(sys->numNeFa, val, sizeof(short) * sys->numVerts);
|
|
|
|
memset(sys->ring_areas, val, sizeof(float) * sys->numVerts);
|
|
|
|
memset(sys->vlengths, val, sizeof(float) * sys->numVerts);
|
|
|
|
memset(sys->vweights, val, sizeof(float) * sys->numVerts);
|
|
|
|
memset(sys->zerola, val, sizeof(short) * sys->numVerts);
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
|
2012-10-27 10:42:28 +00:00
|
|
|
static LaplacianSystem *init_laplacian_system(int a_numEdges, int a_numFaces, int a_numVerts)
|
2012-10-24 10:39:11 +00:00
|
|
|
{
|
2012-10-24 11:31:57 +00:00
|
|
|
LaplacianSystem *sys;
|
2012-10-24 10:39:11 +00:00
|
|
|
sys = MEM_callocN(sizeof(LaplacianSystem), "ModLaplSmoothSystem");
|
|
|
|
sys->numEdges = a_numEdges;
|
|
|
|
sys->numFaces = a_numFaces;
|
|
|
|
sys->numVerts = a_numVerts;
|
|
|
|
|
|
|
|
sys->eweights = MEM_callocN(sizeof(float) * sys->numEdges, "ModLaplSmoothEWeight");
|
|
|
|
if (!sys->eweights) {
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
return NULL;
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->fweights = MEM_callocN(sizeof(float) * 3 * sys->numFaces, "ModLaplSmoothFWeight");
|
|
|
|
if (!sys->fweights) {
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
sys->numNeEd = MEM_callocN(sizeof(short) * sys->numVerts, "ModLaplSmoothNumNeEd");
|
|
|
|
if (!sys->numNeEd) {
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
return NULL;
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->numNeFa = MEM_callocN(sizeof(short) * sys->numVerts, "ModLaplSmoothNumNeFa");
|
|
|
|
if (!sys->numNeFa) {
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
return NULL;
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->ring_areas = MEM_callocN(sizeof(float) * sys->numVerts, "ModLaplSmoothRingAreas");
|
|
|
|
if (!sys->ring_areas) {
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
return NULL;
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->vlengths = MEM_callocN(sizeof(float) * sys->numVerts, "ModLaplSmoothVlengths");
|
|
|
|
if (!sys->vlengths) {
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
sys->vweights = MEM_callocN(sizeof(float) * sys->numVerts, "ModLaplSmoothVweights");
|
|
|
|
if (!sys->vweights) {
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
sys->zerola = MEM_callocN(sizeof(short) * sys->numVerts, "ModLaplSmoothZeloa");
|
|
|
|
if (!sys->zerola) {
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return sys;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void init_data(ModifierData *md)
|
|
|
|
{
|
|
|
|
LaplacianSmoothModifierData *smd = (LaplacianSmoothModifierData *) md;
|
|
|
|
smd->lambda = 0.00001f;
|
|
|
|
smd->lambda_border = 0.00005f;
|
|
|
|
smd->repeat = 1;
|
|
|
|
smd->flag = MOD_LAPLACIANSMOOTH_X | MOD_LAPLACIANSMOOTH_Y | MOD_LAPLACIANSMOOTH_Z | MOD_LAPLACIANSMOOTH_VOLUME_PRESERVATION;
|
|
|
|
smd->defgrp_name[0] = '\0';
|
|
|
|
}
|
|
|
|
|
|
|
|
static void copy_data(ModifierData *md, ModifierData *target)
|
|
|
|
{
|
|
|
|
LaplacianSmoothModifierData *smd = (LaplacianSmoothModifierData *) md;
|
|
|
|
LaplacianSmoothModifierData *tsmd = (LaplacianSmoothModifierData *) target;
|
|
|
|
|
|
|
|
tsmd->lambda = smd->lambda;
|
|
|
|
tsmd->lambda_border = smd->lambda_border;
|
|
|
|
tsmd->repeat = smd->repeat;
|
|
|
|
tsmd->flag = smd->flag;
|
|
|
|
BLI_strncpy(tsmd->defgrp_name, smd->defgrp_name, sizeof(tsmd->defgrp_name));
|
|
|
|
}
|
|
|
|
|
|
|
|
static int is_disabled(ModifierData *md, int UNUSED(useRenderParams))
|
|
|
|
{
|
|
|
|
LaplacianSmoothModifierData *smd = (LaplacianSmoothModifierData *) md;
|
|
|
|
short flag;
|
|
|
|
|
|
|
|
flag = smd->flag & (MOD_LAPLACIANSMOOTH_X | MOD_LAPLACIANSMOOTH_Y | MOD_LAPLACIANSMOOTH_Z);
|
|
|
|
|
|
|
|
/* disable if modifier is off for X, Y and Z or if factor is 0 */
|
2012-10-27 10:42:28 +00:00
|
|
|
if (flag == 0) return 1;
|
2012-10-24 10:39:11 +00:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static CustomDataMask required_data_mask(Object *UNUSED(ob), ModifierData *md)
|
|
|
|
{
|
|
|
|
LaplacianSmoothModifierData *smd = (LaplacianSmoothModifierData *)md;
|
|
|
|
CustomDataMask dataMask = 0;
|
|
|
|
|
|
|
|
/* ask for vertexgroups if we need them */
|
|
|
|
if (smd->defgrp_name[0]) dataMask |= CD_MASK_MDEFORMVERT;
|
|
|
|
|
|
|
|
return dataMask;
|
|
|
|
}
|
|
|
|
|
|
|
|
static float cotan_weight(float *v1, float *v2, float *v3)
|
|
|
|
{
|
|
|
|
float a[3], b[3], c[3], clen;
|
|
|
|
|
|
|
|
sub_v3_v3v3(a, v2, v1);
|
|
|
|
sub_v3_v3v3(b, v3, v1);
|
|
|
|
cross_v3_v3v3(c, a, b);
|
|
|
|
|
|
|
|
clen = len_v3(c);
|
|
|
|
|
|
|
|
if (clen == 0.0f)
|
|
|
|
return 0.0f;
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
return dot_v3v3(a, b) / clen;
|
|
|
|
}
|
|
|
|
|
|
|
|
static float compute_volume(float (*vertexCos)[3], MFace *mfaces, int numFaces)
|
|
|
|
{
|
|
|
|
float vol = 0.0f;
|
|
|
|
float x1, y1, z1, x2, y2, z2, x3, y3, z3, x4, y4, z4;
|
|
|
|
int i;
|
|
|
|
float *vf[4];
|
2012-10-27 10:42:28 +00:00
|
|
|
for (i = 0; i < numFaces; i++) {
|
2012-10-24 10:39:11 +00:00
|
|
|
vf[0] = vertexCos[mfaces[i].v1];
|
|
|
|
vf[1] = vertexCos[mfaces[i].v2];
|
|
|
|
vf[2] = vertexCos[mfaces[i].v3];
|
|
|
|
|
|
|
|
x1 = vf[0][0];
|
|
|
|
y1 = vf[0][1];
|
|
|
|
z1 = vf[0][2];
|
|
|
|
|
|
|
|
x2 = vf[1][0];
|
|
|
|
y2 = vf[1][1];
|
|
|
|
z2 = vf[1][2];
|
|
|
|
|
|
|
|
x3 = vf[2][0];
|
|
|
|
y3 = vf[2][1];
|
|
|
|
z3 = vf[2][2];
|
|
|
|
|
2012-10-27 10:42:28 +00:00
|
|
|
|
|
|
|
vol += (1.0 / 6.0) * (x2 * y3 * z1 + x3 * y1 * z2 - x1 * y3 * z2 - x2 * y1 * z3 + x1 * y2 * z3 - x3 * y2 * z1);
|
2012-10-24 10:39:11 +00:00
|
|
|
if ((&mfaces[i])->v4) {
|
|
|
|
vf[3] = vertexCos[mfaces[i].v4];
|
|
|
|
x4 = vf[3][0];
|
|
|
|
y4 = vf[3][1];
|
|
|
|
z4 = vf[3][2];
|
2012-10-27 10:42:28 +00:00
|
|
|
vol += (1.0 / 6.0) * (x1 * y3 * z4 - x1 * y4 * z3 - x3 * y1 * z4 + x3 * z1 * y4 + y1 * x4 * z3 - x4 * y3 * z1);
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
return fabsfvol);
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static void volume_preservation(LaplacianSystem *sys, float vini, float vend, short flag)
|
|
|
|
{
|
|
|
|
float beta;
|
|
|
|
int i;
|
|
|
|
|
2012-10-27 10:42:28 +00:00
|
|
|
if (vend != 0.0f) {
|
|
|
|
beta = pow(vini / vend, 1.0f / 3.0f);
|
2012-10-24 10:39:11 +00:00
|
|
|
for (i = 0; i < sys->numVerts; i++) {
|
|
|
|
if (flag & MOD_LAPLACIANSMOOTH_X) {
|
2012-10-24 11:31:57 +00:00
|
|
|
sys->vertexCos[i][0] = (sys->vertexCos[i][0] - sys->vert_centroid[0]) * beta + sys->vert_centroid[0];
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
if (flag & MOD_LAPLACIANSMOOTH_Y) {
|
2012-10-24 11:31:57 +00:00
|
|
|
sys->vertexCos[i][1] = (sys->vertexCos[i][1] - sys->vert_centroid[1]) * beta + sys->vert_centroid[1];
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
if (flag & MOD_LAPLACIANSMOOTH_Z) {
|
2012-10-24 11:31:57 +00:00
|
|
|
sys->vertexCos[i][2] = (sys->vertexCos[i][2] - sys->vert_centroid[2]) * beta + sys->vert_centroid[2];
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-10-24 11:31:57 +00:00
|
|
|
static void init_laplacian_matrix(LaplacianSystem *sys)
|
2012-10-24 10:39:11 +00:00
|
|
|
{
|
|
|
|
float *v1, *v2, *v3, *v4;
|
|
|
|
float w1, w2, w3, w4;
|
|
|
|
float areaf;
|
|
|
|
int i, j;
|
|
|
|
unsigned int idv1, idv2, idv3, idv4, idv[4];
|
2012-10-27 10:42:28 +00:00
|
|
|
int has_4_vert;
|
|
|
|
for (i = 0; i < sys->numEdges; i++) {
|
2012-10-24 10:39:11 +00:00
|
|
|
idv1 = sys->medges[i].v1;
|
|
|
|
idv2 = sys->medges[i].v2;
|
|
|
|
|
|
|
|
v1 = sys->vertexCos[idv1];
|
|
|
|
v2 = sys->vertexCos[idv2];
|
|
|
|
|
|
|
|
sys->numNeEd[idv1] = sys->numNeEd[idv1] + 1;
|
|
|
|
sys->numNeEd[idv2] = sys->numNeEd[idv2] + 1;
|
|
|
|
w1 = len_v3v3(v1, v2);
|
|
|
|
if (w1 < sys->min_area) {
|
|
|
|
sys->zerola[idv1] = 1;
|
|
|
|
sys->zerola[idv2] = 1;
|
2012-10-24 11:31:57 +00:00
|
|
|
}
|
|
|
|
else {
|
2012-10-24 10:39:11 +00:00
|
|
|
w1 = 1.0f / w1;
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->eweights[i] = w1;
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
for (i = 0; i < sys->numFaces; i++) {
|
2012-10-24 10:39:11 +00:00
|
|
|
has_4_vert = ((&sys->mfaces[i])->v4) ? 1 : 0;
|
|
|
|
|
|
|
|
idv1 = sys->mfaces[i].v1;
|
|
|
|
idv2 = sys->mfaces[i].v2;
|
|
|
|
idv3 = sys->mfaces[i].v3;
|
|
|
|
idv4 = has_4_vert ? sys->mfaces[i].v4 : 0;
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->numNeFa[idv1] += 1;
|
|
|
|
sys->numNeFa[idv2] += 1;
|
|
|
|
sys->numNeFa[idv3] += 1;
|
|
|
|
if (has_4_vert) sys->numNeFa[idv4] += 1;
|
|
|
|
|
|
|
|
v1 = sys->vertexCos[idv1];
|
|
|
|
v2 = sys->vertexCos[idv2];
|
|
|
|
v3 = sys->vertexCos[idv3];
|
|
|
|
v4 = has_4_vert ? sys->vertexCos[idv4] : 0;
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
if (has_4_vert) {
|
|
|
|
areaf = area_quad_v3(v1, v2, v3, sys->vertexCos[sys->mfaces[i].v4]);
|
2012-10-24 11:31:57 +00:00
|
|
|
}
|
|
|
|
else {
|
2012-10-24 10:39:11 +00:00
|
|
|
areaf = area_tri_v3(v1, v2, v3);
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
if (fabs(areaf) < sys->min_area) {
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->zerola[idv1] = 1;
|
|
|
|
sys->zerola[idv2] = 1;
|
|
|
|
sys->zerola[idv3] = 1;
|
|
|
|
if (has_4_vert) sys->zerola[idv4] = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
sys->ring_areas[idv1] += areaf;
|
|
|
|
sys->ring_areas[idv2] += areaf;
|
|
|
|
sys->ring_areas[idv3] += areaf;
|
|
|
|
if (has_4_vert) sys->ring_areas[idv4] += areaf;
|
|
|
|
|
|
|
|
if (has_4_vert) {
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
idv[0] = idv1;
|
|
|
|
idv[1] = idv2;
|
|
|
|
idv[2] = idv3;
|
|
|
|
idv[3] = idv4;
|
|
|
|
|
|
|
|
for (j = 0; j < 4; j++) {
|
|
|
|
idv1 = idv[j];
|
|
|
|
idv2 = idv[(j + 1) % 4];
|
|
|
|
idv3 = idv[(j + 2) % 4];
|
|
|
|
idv4 = idv[(j + 3) % 4];
|
|
|
|
|
|
|
|
v1 = sys->vertexCos[idv1];
|
|
|
|
v2 = sys->vertexCos[idv2];
|
|
|
|
v3 = sys->vertexCos[idv3];
|
|
|
|
v4 = sys->vertexCos[idv4];
|
|
|
|
|
|
|
|
w2 = cotan_weight(v4, v1, v2) + cotan_weight(v3, v1, v2);
|
|
|
|
w3 = cotan_weight(v2, v3, v1) + cotan_weight(v4, v1, v3);
|
|
|
|
w4 = cotan_weight(v2, v4, v1) + cotan_weight(v3, v4, v1);
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->vweights[idv1] += (w2 + w3 + w4) / 4.0f;
|
|
|
|
}
|
2012-10-24 11:31:57 +00:00
|
|
|
}
|
|
|
|
else {
|
2012-10-24 10:39:11 +00:00
|
|
|
w1 = cotan_weight(v1, v2, v3);
|
|
|
|
w2 = cotan_weight(v2, v3, v1);
|
|
|
|
w3 = cotan_weight(v3, v1, v2);
|
|
|
|
|
|
|
|
sys->fweights[i][0] = sys->fweights[i][0] + w1;
|
|
|
|
sys->fweights[i][1] = sys->fweights[i][1] + w2;
|
|
|
|
sys->fweights[i][2] = sys->fweights[i][2] + w3;
|
2012-10-27 10:42:28 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->vweights[idv1] = sys->vweights[idv1] + w2 + w3;
|
|
|
|
sys->vweights[idv2] = sys->vweights[idv2] + w1 + w3;
|
|
|
|
sys->vweights[idv3] = sys->vweights[idv3] + w1 + w2;
|
|
|
|
}
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
for (i = 0; i < sys->numEdges; i++) {
|
2012-10-24 10:39:11 +00:00
|
|
|
idv1 = sys->medges[i].v1;
|
|
|
|
idv2 = sys->medges[i].v2;
|
|
|
|
/* if is boundary, apply scale-dependent umbrella operator only with neighboors in boundary */
|
2012-10-27 10:42:28 +00:00
|
|
|
if (sys->numNeEd[idv1] != sys->numNeFa[idv1] && sys->numNeEd[idv2] != sys->numNeFa[idv2]) {
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->vlengths[idv1] += sys->eweights[i];
|
|
|
|
sys->vlengths[idv2] += sys->eweights[i];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2012-10-24 11:31:57 +00:00
|
|
|
static void fill_laplacian_matrix(LaplacianSystem *sys)
|
2012-10-24 10:39:11 +00:00
|
|
|
{
|
|
|
|
float *v1, *v2, *v3, *v4;
|
|
|
|
float w2, w3, w4;
|
|
|
|
int i, j;
|
2012-10-27 10:42:28 +00:00
|
|
|
int has_4_vert;
|
2012-10-24 10:39:11 +00:00
|
|
|
unsigned int idv1, idv2, idv3, idv4, idv[4];
|
2012-10-27 10:42:28 +00:00
|
|
|
|
|
|
|
for (i = 0; i < sys->numFaces; i++) {
|
2012-10-24 10:39:11 +00:00
|
|
|
idv1 = sys->mfaces[i].v1;
|
|
|
|
idv2 = sys->mfaces[i].v2;
|
|
|
|
idv3 = sys->mfaces[i].v3;
|
|
|
|
has_4_vert = ((&sys->mfaces[i])->v4) ? 1 : 0;
|
|
|
|
|
|
|
|
if (has_4_vert) {
|
|
|
|
idv[0] = sys->mfaces[i].v1;
|
|
|
|
idv[1] = sys->mfaces[i].v2;
|
|
|
|
idv[2] = sys->mfaces[i].v3;
|
|
|
|
idv[3] = sys->mfaces[i].v4;
|
|
|
|
for (j = 0; j < 4; j++) {
|
|
|
|
idv1 = idv[j];
|
|
|
|
idv2 = idv[(j + 1) % 4];
|
|
|
|
idv3 = idv[(j + 2) % 4];
|
|
|
|
idv4 = idv[(j + 3) % 4];
|
|
|
|
|
|
|
|
v1 = sys->vertexCos[idv1];
|
|
|
|
v2 = sys->vertexCos[idv2];
|
|
|
|
v3 = sys->vertexCos[idv3];
|
|
|
|
v4 = sys->vertexCos[idv4];
|
|
|
|
|
|
|
|
w2 = cotan_weight(v4, v1, v2) + cotan_weight(v3, v1, v2);
|
|
|
|
w3 = cotan_weight(v2, v3, v1) + cotan_weight(v4, v1, v3);
|
|
|
|
w4 = cotan_weight(v2, v4, v1) + cotan_weight(v3, v4, v1);
|
|
|
|
|
|
|
|
w2 = w2 / 4.0f;
|
|
|
|
w3 = w3 / 4.0f;
|
|
|
|
w4 = w4 / 4.0f;
|
2012-10-27 10:42:28 +00:00
|
|
|
|
|
|
|
if (sys->numNeEd[idv1] == sys->numNeFa[idv1] && sys->zerola[idv1] == 0) {
|
2012-10-24 10:39:11 +00:00
|
|
|
nlMatrixAdd(idv1, idv2, w2 * sys->vweights[idv1]);
|
|
|
|
nlMatrixAdd(idv1, idv3, w3 * sys->vweights[idv1]);
|
|
|
|
nlMatrixAdd(idv1, idv4, w4 * sys->vweights[idv1]);
|
|
|
|
}
|
|
|
|
}
|
2012-10-24 11:31:57 +00:00
|
|
|
}
|
|
|
|
else {
|
2012-10-24 10:39:11 +00:00
|
|
|
/* Is ring if number of faces == number of edges around vertice*/
|
2012-10-27 10:42:28 +00:00
|
|
|
if (sys->numNeEd[idv1] == sys->numNeFa[idv1] && sys->zerola[idv1] == 0) {
|
2012-10-24 10:39:11 +00:00
|
|
|
nlMatrixAdd(idv1, idv2, sys->fweights[i][2] * sys->vweights[idv1]);
|
|
|
|
nlMatrixAdd(idv1, idv3, sys->fweights[i][1] * sys->vweights[idv1]);
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
if (sys->numNeEd[idv2] == sys->numNeFa[idv2] && sys->zerola[idv2] == 0) {
|
2012-10-24 10:39:11 +00:00
|
|
|
nlMatrixAdd(idv2, idv1, sys->fweights[i][2] * sys->vweights[idv2]);
|
|
|
|
nlMatrixAdd(idv2, idv3, sys->fweights[i][0] * sys->vweights[idv2]);
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
if (sys->numNeEd[idv3] == sys->numNeFa[idv3] && sys->zerola[idv3] == 0) {
|
2012-10-24 10:39:11 +00:00
|
|
|
nlMatrixAdd(idv3, idv1, sys->fweights[i][1] * sys->vweights[idv3]);
|
|
|
|
nlMatrixAdd(idv3, idv2, sys->fweights[i][0] * sys->vweights[idv3]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-10-27 10:42:28 +00:00
|
|
|
for (i = 0; i < sys->numEdges; i++) {
|
2012-10-24 10:39:11 +00:00
|
|
|
idv1 = sys->medges[i].v1;
|
|
|
|
idv2 = sys->medges[i].v2;
|
|
|
|
/* Is boundary */
|
2012-10-27 10:42:28 +00:00
|
|
|
if (sys->numNeEd[idv1] != sys->numNeFa[idv1] &&
|
|
|
|
sys->numNeEd[idv2] != sys->numNeFa[idv2] &&
|
|
|
|
sys->zerola[idv1] == 0 &&
|
|
|
|
sys->zerola[idv2] == 0)
|
2012-10-24 10:39:11 +00:00
|
|
|
{
|
|
|
|
nlMatrixAdd(idv1, idv2, sys->eweights[i] * sys->vlengths[idv1]);
|
|
|
|
nlMatrixAdd(idv2, idv1, sys->eweights[i] * sys->vlengths[idv2]);
|
2012-10-27 10:42:28 +00:00
|
|
|
}
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-10-24 11:31:57 +00:00
|
|
|
static void validate_solution(LaplacianSystem *sys, short flag)
|
2012-10-24 10:39:11 +00:00
|
|
|
{
|
|
|
|
int i, idv1, idv2;
|
|
|
|
float leni, lene;
|
|
|
|
float vini, vend;
|
|
|
|
float *vi1, *vi2, ve1[3], ve2[3];
|
|
|
|
if (flag & MOD_LAPLACIANSMOOTH_VOLUME_PRESERVATION) {
|
|
|
|
vini = compute_volume(sys->vertexCos, sys->mfaces, sys->numFaces);
|
|
|
|
}
|
2012-10-27 10:42:28 +00:00
|
|
|
for (i = 0; i < sys->numEdges; i++) {
|
2012-10-24 10:39:11 +00:00
|
|
|
idv1 = sys->medges[i].v1;
|
|
|
|
idv2 = sys->medges[i].v2;
|
|
|
|
vi1 = sys->vertexCos[idv1];
|
|
|
|
vi2 = sys->vertexCos[idv2];
|
|
|
|
ve1[0] = nlGetVariable(0, idv1);
|
|
|
|
ve1[1] = nlGetVariable(1, idv1);
|
|
|
|
ve1[2] = nlGetVariable(2, idv1);
|
|
|
|
ve2[0] = nlGetVariable(0, idv2);
|
|
|
|
ve2[1] = nlGetVariable(1, idv2);
|
|
|
|
ve2[2] = nlGetVariable(2, idv2);
|
|
|
|
leni = len_v3v3(vi1, vi2);
|
|
|
|
lene = len_v3v3(ve1, ve2);
|
2012-10-27 10:42:28 +00:00
|
|
|
if (lene > leni * MOD_LAPLACIANSMOOTH_MAX_EDGE_PERCENTAGE || lene < leni * MOD_LAPLACIANSMOOTH_MIN_EDGE_PERCENTAGE) {
|
2012-10-24 10:39:11 +00:00
|
|
|
sys->zerola[idv1] = 1;
|
|
|
|
sys->zerola[idv2] = 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for (i = 0; i < sys->numVerts; i++) {
|
|
|
|
if (sys->zerola[i] == 0) {
|
|
|
|
if (flag & MOD_LAPLACIANSMOOTH_X) {
|
|
|
|
sys->vertexCos[i][0] = nlGetVariable(0, i);
|
|
|
|
}
|
|
|
|
if (flag & MOD_LAPLACIANSMOOTH_Y) {
|
|
|
|
sys->vertexCos[i][1] = nlGetVariable(1, i);
|
|
|
|
}
|
|
|
|
if (flag & MOD_LAPLACIANSMOOTH_Z) {
|
|
|
|
sys->vertexCos[i][2] = nlGetVariable(2, i);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (flag & MOD_LAPLACIANSMOOTH_VOLUME_PRESERVATION) {
|
|
|
|
vend = compute_volume(sys->vertexCos, sys->mfaces, sys->numFaces);
|
|
|
|
volume_preservation(sys, vini, vend, flag);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void laplaciansmoothModifier_do(
|
2012-10-27 10:42:28 +00:00
|
|
|
LaplacianSmoothModifierData *smd, Object *ob, DerivedMesh *dm,
|
|
|
|
float (*vertexCos)[3], int numVerts)
|
2012-10-24 10:39:11 +00:00
|
|
|
{
|
|
|
|
LaplacianSystem *sys;
|
|
|
|
MDeformVert *dvert = NULL;
|
|
|
|
MDeformVert *dv = NULL;
|
|
|
|
float w, wpaint;
|
|
|
|
int i, iter;
|
|
|
|
int defgrp_index;
|
|
|
|
|
|
|
|
DM_ensure_tessface(dm);
|
|
|
|
|
|
|
|
sys = init_laplacian_system(dm->getNumEdges(dm), dm->getNumTessFaces(dm), numVerts);
|
2012-10-24 11:31:57 +00:00
|
|
|
if (!sys) {
|
|
|
|
return;
|
|
|
|
}
|
2012-10-24 10:39:11 +00:00
|
|
|
|
|
|
|
sys->mfaces = dm->getTessFaceArray(dm);
|
|
|
|
sys->medges = dm->getEdgeArray(dm);
|
|
|
|
sys->vertexCos = vertexCos;
|
|
|
|
sys->min_area = 0.00001f;
|
|
|
|
modifier_get_vgroup(ob, dm, smd->defgrp_name, &dvert, &defgrp_index);
|
|
|
|
|
|
|
|
sys->vert_centroid[0] = 0.0f;
|
|
|
|
sys->vert_centroid[1] = 0.0f;
|
|
|
|
sys->vert_centroid[2] = 0.0f;
|
|
|
|
for (iter = 0; iter < smd->repeat; iter++) {
|
|
|
|
memset_laplacian_system(sys, 0);
|
|
|
|
nlNewContext();
|
|
|
|
sys->context = nlGetCurrent();
|
|
|
|
nlSolverParameteri(NL_NB_VARIABLES, numVerts);
|
|
|
|
nlSolverParameteri(NL_LEAST_SQUARES, NL_TRUE);
|
|
|
|
nlSolverParameteri(NL_NB_ROWS, numVerts);
|
|
|
|
nlSolverParameteri(NL_NB_RIGHT_HAND_SIDES, 3);
|
|
|
|
|
|
|
|
init_laplacian_matrix(sys);
|
|
|
|
|
2012-10-27 10:42:28 +00:00
|
|
|
nlBegin(NL_SYSTEM);
|
2012-10-24 10:39:11 +00:00
|
|
|
for (i = 0; i < numVerts; i++) {
|
|
|
|
nlSetVariable(0, i, vertexCos[i][0]);
|
|
|
|
nlSetVariable(1, i, vertexCos[i][1]);
|
|
|
|
nlSetVariable(2, i, vertexCos[i][2]);
|
|
|
|
if (iter == 0) {
|
2012-10-24 11:31:57 +00:00
|
|
|
add_v3_v3(sys->vert_centroid, vertexCos[i]);
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
}
|
2012-10-24 11:31:57 +00:00
|
|
|
if (iter == 0 && numVerts > 0) {
|
|
|
|
mul_v3_fl(sys->vert_centroid, 1.0f / (float)numVerts);
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
nlBegin(NL_MATRIX);
|
|
|
|
dv = dvert;
|
|
|
|
for (i = 0; i < numVerts; i++) {
|
|
|
|
nlRightHandSideAdd(0, i, vertexCos[i][0]);
|
|
|
|
nlRightHandSideAdd(1, i, vertexCos[i][1]);
|
|
|
|
nlRightHandSideAdd(2, i, vertexCos[i][2]);
|
|
|
|
if (dv) {
|
|
|
|
wpaint = defvert_find_weight(dv, defgrp_index);
|
|
|
|
dv++;
|
2012-10-24 11:31:57 +00:00
|
|
|
}
|
|
|
|
else {
|
2012-10-24 10:39:11 +00:00
|
|
|
wpaint = 1.0f;
|
|
|
|
}
|
2012-10-24 11:31:57 +00:00
|
|
|
|
2012-10-24 10:39:11 +00:00
|
|
|
if (sys->zerola[i] == 0) {
|
|
|
|
w = sys->vweights[i] * sys->ring_areas[i];
|
2012-10-24 11:31:57 +00:00
|
|
|
sys->vweights[i] = (w == 0.0f) ? 0.0f : -smd->lambda * wpaint / (4.0f * w);
|
2012-10-24 10:39:11 +00:00
|
|
|
w = sys->vlengths[i];
|
2012-10-24 11:31:57 +00:00
|
|
|
sys->vlengths[i] = (w == 0.0f) ? 0.0f : -smd->lambda_border * wpaint * 2.0f / w;
|
2012-10-24 10:39:11 +00:00
|
|
|
|
2012-10-27 10:42:28 +00:00
|
|
|
if (sys->numNeEd[i] == sys->numNeFa[i]) {
|
2012-10-24 10:39:11 +00:00
|
|
|
nlMatrixAdd(i, i, 1.0f + smd->lambda * wpaint / (4.0f * sys->ring_areas[i]));
|
2012-10-24 11:31:57 +00:00
|
|
|
}
|
|
|
|
else {
|
2012-10-24 10:39:11 +00:00
|
|
|
nlMatrixAdd(i, i, 1.0f + smd->lambda_border * wpaint * 2.0f);
|
|
|
|
}
|
2012-10-24 11:31:57 +00:00
|
|
|
}
|
|
|
|
else {
|
2012-10-24 10:39:11 +00:00
|
|
|
nlMatrixAdd(i, i, 1.0f);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
fill_laplacian_matrix(sys);
|
|
|
|
|
|
|
|
nlEnd(NL_MATRIX);
|
|
|
|
nlEnd(NL_SYSTEM);
|
|
|
|
|
|
|
|
if (nlSolveAdvanced(NULL, NL_TRUE)) {
|
2012-10-24 11:31:57 +00:00
|
|
|
validate_solution(sys, smd->flag);
|
2012-10-24 10:39:11 +00:00
|
|
|
}
|
|
|
|
nlDeleteContext(sys->context);
|
|
|
|
sys->context = NULL;
|
|
|
|
}
|
|
|
|
delete_laplacian_system(sys);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
static void deformVerts(ModifierData *md, Object *ob, DerivedMesh *derivedData,
|
|
|
|
float (*vertexCos)[3], int numVerts, ModifierApplyFlag UNUSED(flag))
|
|
|
|
{
|
|
|
|
DerivedMesh *dm = get_dm(ob, NULL, derivedData, NULL, 0);
|
|
|
|
|
|
|
|
laplaciansmoothModifier_do((LaplacianSmoothModifierData *)md, ob, dm,
|
2012-10-27 10:42:28 +00:00
|
|
|
vertexCos, numVerts);
|
2012-10-24 10:39:11 +00:00
|
|
|
|
|
|
|
if (dm != derivedData)
|
|
|
|
dm->release(dm);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void deformVertsEM(
|
|
|
|
ModifierData *md, Object *ob, struct BMEditMesh *editData,
|
|
|
|
DerivedMesh *derivedData, float (*vertexCos)[3], int numVerts)
|
|
|
|
{
|
|
|
|
DerivedMesh *dm = get_dm(ob, editData, derivedData, NULL, 0);
|
|
|
|
|
|
|
|
laplaciansmoothModifier_do((LaplacianSmoothModifierData *)md, ob, dm,
|
2012-10-27 10:42:28 +00:00
|
|
|
vertexCos, numVerts);
|
2012-10-24 10:39:11 +00:00
|
|
|
|
|
|
|
if (dm != derivedData)
|
|
|
|
dm->release(dm);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
ModifierTypeInfo modifierType_LaplacianSmooth = {
|
|
|
|
/* name */ "Laplacian Smooth",
|
|
|
|
/* structName */ "LaplacianSmoothModifierData",
|
|
|
|
/* structSize */ sizeof(LaplacianSmoothModifierData),
|
|
|
|
/* type */ eModifierTypeType_OnlyDeform,
|
|
|
|
/* flags */ eModifierTypeFlag_AcceptsMesh |
|
|
|
|
eModifierTypeFlag_SupportsEditmode,
|
|
|
|
|
|
|
|
/* copy_data */ copy_data,
|
|
|
|
/* deformVerts */ deformVerts,
|
|
|
|
/* deformMatrices */ NULL,
|
|
|
|
/* deformVertsEM */ deformVertsEM,
|
|
|
|
/* deformMatricesEM */ NULL,
|
|
|
|
/* applyModifier */ NULL,
|
|
|
|
/* applyModifierEM */ NULL,
|
2012-10-24 11:31:57 +00:00
|
|
|
/* initData */ init_data,
|
|
|
|
/* requiredDataMask */ required_data_mask,
|
2012-10-24 10:39:11 +00:00
|
|
|
/* freeData */ NULL,
|
2012-10-24 11:31:57 +00:00
|
|
|
/* isDisabled */ is_disabled,
|
2012-10-24 10:39:11 +00:00
|
|
|
/* updateDepgraph */ NULL,
|
|
|
|
/* dependsOnTime */ NULL,
|
|
|
|
/* dependsOnNormals */ NULL,
|
|
|
|
/* foreachObjectLink */ NULL,
|
|
|
|
/* foreachIDLink */ NULL,
|
|
|
|
/* foreachTexLink */ NULL,
|
|
|
|
};
|