2020-10-21 17:24:01 +02:00

569 lines
18 KiB
C++

/****************************************************************************
* MeshLab o o *
* An extendible mesh processor o o *
* _ O _ *
* Copyright(C) 2005, 2009 \/)\/ *
* Visual Computing Lab /\/| *
* ISTI - Italian National Research Council | *
* \ *
* All rights reserved. *
* *
* 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 (http://www.gnu.org/licenses/gpl.txt) *
* for more details. *
* *
****************************************************************************/
#include <editsegment.h>
using namespace std;
using namespace vcg;
inline bool isFront(const QPointF &a, const QPointF &b, const QPointF &c) {
return (b.x()-a.x())*(c.y()-a.y())-(b.y()-a.y())*(c.x()-a.x())>0;
}
inline bool isIn(const QPointF &pointer, const QPointF &pointer_prev, float X,
float Y, float radius, float *dist) {
if (pointer != pointer_prev) {
float mid_x = (pointer_prev.x() - pointer.x());
float mid_y = (pointer_prev.y() - pointer.y());
float l_square = mid_x * mid_x + mid_y * mid_y;
float r = (X - pointer.x()) * (pointer_prev.x() - pointer.x()) +
(Y - pointer.y()) * (pointer_prev.y() - pointer.y());
r /= l_square;
float P_x = pointer.x() + r * (pointer_prev.x() - pointer.x()) - X;
float P_y = pointer.y() + r * (pointer_prev.y() - pointer.y()) - Y;
if (r >= 0 && r <= 1 && (P_x * P_x + P_y * P_y < radius * radius)) {
*dist=sqrt(P_x*P_x + P_y*P_y)/radius;
return true;
}
}
// there could be some problem when point is nearer p0 or p1 and viceversa
// so i have to check both. is only needed with smooth_borders
float x1 = (X - pointer_prev.x());
float y1 = (Y - pointer_prev.y());
float tmp = x1*x1 + y1*y1;
if (tmp < radius*radius) {
*dist = sqrt(tmp)/radius;
return true;
}
if (pointer == pointer_prev)
return false;
else {
float x0 = (X - pointer.x());
float y0 = (Y - pointer.y());
tmp = x0*x0 + y0*y0;
if (tmp < radius*radius) {
*dist = sqrt(tmp)/radius;
return true;
} else
return false;
}
}
inline bool lineHitsCircle(QPointF& LineStart, QPointF& LineEnd,
QPointF& CircleCenter, float Radius, QPointF* const pOut = 0) {
const float RadiusSq = Radius * Radius;
QPointF PMinusM =LineStart - CircleCenter;
float pm_squ=PMinusM.x()*PMinusM.x()+PMinusM.y()*PMinusM.y();
if (pm_squ <= RadiusSq) { /// startpoint in circle
if (pOut)
*pOut = LineStart;
return true;
}
QPointF LineDir=LineEnd - LineStart; /// line direction
// u * (p - m)
const float UDotPMinusM = LineDir.x()*PMinusM.x()+LineDir.y()*PMinusM.y();//Vector2D_Dot(LineDir, PMinusM);
// u*u
const float LineDirSq = LineDir.x()*LineDir.x()+LineDir.y()*LineDir.y();
// (u * (p - m))^2
// - (u*u * ((p - m)^2 - r^2))
const float d = UDotPMinusM * UDotPMinusM - LineDirSq * (pm_squ - RadiusSq);
if (d < 0.0f)
return false;
else if (d < 0.0001f) {
const float s = -UDotPMinusM / LineDirSq;
if (s< 0.0f || s> 1.0f)return false;
else {
if(pOut) *pOut = LineStart + s * LineDir;
return true;
}
}
else {
const float s = (-UDotPMinusM - sqrtf(d)) / LineDirSq;
if(s < 0.0f || s > 1.0f) return false;
else {
if(pOut) *pOut = LineStart + s * LineDir;
return true;
}
}
}
inline bool pointInTriangle(const QPointF &p, const QPointF &a,
const QPointF &b, const QPointF &c) {
float fab=(p.y()-a.y())*(b.x()-a.x()) - (p.x()-a.x())*(b.y()-a.y());
float fbc=(p.y()-c.y())*(a.x()-c.x()) - (p.x()-c.x())*(a.y()-c.y());
float fca=(p.y()-b.y())*(c.x()-b.x()) - (p.x()-b.x())*(c.y()-b.y());
if (fab*fbc>0 && fbc*fca>0)
return true;
return false;
}
inline bool pointInTriangle(const float p_x, const float p_y, const float a_x,
const float a_y, const float b_x, const float b_y, const float c_x,
const float c_y) {
float fab=(p_y-a_y)*(b_x-a_x) - (p_x-a_x)*(b_y-a_y);
float fbc=(p_y-c_y)*(a_x-c_x) - (p_x-c_x)*(a_y-c_y);
float fca=(p_y-b_y)*(c_x-b_x) - (p_x-b_x)*(c_y-b_y);
if (fab*fbc>0 && fbc*fca>0)
return true;
return false;
}
inline void getSurroundingFacesVF(CFaceO * fac, int vert_pos,
vector<CFaceO *> *surround) {
CVertexO * vert=fac->V(vert_pos);
int pos=vert->VFi();
CFaceO * first_fac=vert->VFp();
CFaceO * curr_f=first_fac;
do {
CFaceO * temp=curr_f->VFp(pos);
if (curr_f!=0 && !curr_f->IsD()) {
surround->push_back(curr_f);
pos=curr_f->VFi(pos);
}
curr_f=temp;
} while (curr_f!=first_fac && curr_f!=0);
}
void getInternFaces(MeshModel & m, vector<CMeshO::FacePointer> *actual, vector<CMeshO::VertexPointer> * risult,
vector<CMeshO::FacePointer> * face_risult, GLArea * gla, Penn &pen, QPoint &current_point, QPoint &previous_point, GLfloat * pixels,
double mvmatrix[16], double projmatrix[16], GLint viewport[4]) {
QHash <CFaceO *,CFaceO *> selected;
QHash <CVertexO *,CVertexO *> sel_vert;
vector<CMeshO::FacePointer>::iterator fpi;
vector<CMeshO::FacePointer> temp_po;
if (actual->size()==0) {
CMeshO::FaceIterator fi;
for (fi=m.cm.face.begin(); fi!=m.cm.face.end(); ++fi)
if (!(*fi).IsD()) {
temp_po.push_back((&*fi));
}
} else
for (fpi=actual->begin(); fpi!=actual->end(); ++fpi) {
temp_po.push_back(*fpi);
}
actual->clear();
QPointF mid = QPointF(current_point.x(), gla->curSiz.height()
- current_point.y());
QPointF mid_prev = QPointF(previous_point.x(), gla->curSiz.height()
- previous_point.y());
QPointF p[3], z[3];
double tx, ty, tz;
bool backface=pen.backface;
bool invisible=pen.invisible;
for (int j=0; j<temp_po.size(); j++) {
CFaceO * fac=temp_po.at(j);
int intern=0;
int checkable=0; /// to avoid problems when the pen is smaller then the polys
for (int i=0; i<3; i++) {
if (gluProject((fac)->V(i)->P()[0], (fac)->V(i)->P()[1], (fac)->V(i)->P()[2], mvmatrix, projmatrix, viewport, &tx, &ty, &tz)
==GL_TRUE)
checkable++;
if (tz<0 || tz>1)checkable --;
p[i].setX(tx); p[i].setY(ty);
if (tx>=0 && tx<viewport[2] && ty>=0 && ty<viewport[3]) {
z[i].setX(tz);
z[i].setY((float)pixels[(int)(((int)ty)*viewport[2]+(int)tx)]);
}
else {
z[i].setX(1); z[i].setY(0);
}
}
if (backface || isFront(p[0],p[1],p[2])) {
//checkable++;
/// colud be useful to calc the medium of z in the matrix
for (int i=0; i<3; i++) {
if (invisible || (z[i].x()<=z[i].y()+0.003)) {
float dist;
if (isIn(mid,mid_prev,p[i].x(),p[i].y(),pen.radius,&dist)==1) {
intern=1;
if (!sel_vert.contains(fac->V(i))) {
risult->push_back(fac->V(i));
sel_vert.insert(fac->V(i),fac->V(i));
}
}
QPointF pos_res;
if (intern==0 && lineHitsCircle(p[i],p[(i+1)%3],mid,pen.radius,&pos_res)) {
intern=1; continue;
}
}
}
if (checkable==3 && intern==0 && (pointInTriangle(mid,p[0],p[1],p[2]) || pointInTriangle(mid_prev,p[0],p[1],p[2]))) {
intern=-1;
}
}
if (checkable==3 && intern!=0 && !selected.contains(fac)) {
selected.insert((fac),(fac));
actual->push_back(fac);
vector <CFaceO *> surround;
for (int i=0; i<3; i++) getSurroundingFacesVF(fac,i,&surround);
for (int i=0; i<surround.size(); i++) {
if (!selected.contains(surround[i])) {
temp_po.push_back(surround[i]);
}
}
if (intern!=0) face_risult->push_back(fac);
}
}
}
Color4b toVcgColor(QColor c) {
return Color4b(c.red(), c.green(), c.blue(), 255);
}
EditSegment::EditSegment() {
pixels = 0;
pen.radius = 5;
pen.backface = false;
pen.invisible = false;
pressed = false;
dragging = false;
first = true;
meshCut = 0;
meshCutDialog = 0;
meshcut_dock = 0;
normalWeight = 5;
curvatureWeight = 5;
selectForeground = true;
doRefine = true;
}
EditSegment::~EditSegment() {
delete meshCut;
}
const QString EditSegment::Info()
{
return tr("Segment the mesh by selecting the foreground and background");
}
bool EditSegment::StartEdit(MeshModel & m, GLArea * parent) {
parent->setCursor(QCursor(
QPixmap(":/images/editsegment_cursor.png", "PNG"), 1, 1));
if (!glarea_map.contains(parent)) {
glarea_map.insert(parent, new MeshCutting<CMeshO>(&m.cm));
}
glarea = parent;
meshCut = glarea_map[glarea];
if (!meshCutDialog) {
meshCutDialog = new MeshCutDialog(parent->window());
meshcut_dock = new QDockWidget(parent->window());
meshcut_dock->setAllowedAreas(Qt::NoDockWidgetArea);
meshcut_dock->setWidget(meshCutDialog);
QPoint pos = parent->window()->mapToGlobal(QPoint(0, 0));
meshcut_dock->setGeometry(-5+pos.x()+parent->window()->width()-meshCutDialog->width(), pos.y(),
meshCutDialog->width(), meshCutDialog->height());
meshcut_dock->setFloating(true);
QObject::connect(meshCutDialog, SIGNAL(meshCutSignal()), this,
SLOT(MeshCutSlot()));
QObject::connect(meshCutDialog, SIGNAL(updateCurvatureSignal()), this,
SLOT(UpdateCurvatureSlot()));
QObject::connect(meshCutDialog, SIGNAL(selectForegroundSignal(bool)),this, SLOT(SelectForegroundSlot(bool)));
QObject::connect(meshCutDialog, SIGNAL(doRefineSignal(bool)),this, SLOT(doRefineSlot(bool)));
QObject::connect(meshCutDialog, SIGNAL(resetSignal()), this,
SLOT(ResetSlot()));
QObject::connect(meshCutDialog, SIGNAL(colorizeGaussianSignal()), this,
SLOT(ColorizeGaussianSlot()));
QObject::connect(meshCutDialog, SIGNAL(colorizeMeanSignal()), this,
SLOT(ColorizeMeanSlot()));
QObject::connect(meshCutDialog, SIGNAL(normalWeightSignal(int)),this, SLOT(changeNormalWeight(int)));
QObject::connect(meshCutDialog, SIGNAL(curvatureWeightSignal(int)),this, SLOT(changeCurvatureWeight(int)));
QObject::connect(meshCutDialog, SIGNAL(penRadiusSignal(int)),this, SLOT(changePenRadius(int)));
}
meshcut_dock->setVisible(true);
meshcut_dock->layout()->update();
m.updateDataMask(MeshModel::MM_FACEFACETOPO | MeshModel::MM_VERTFACETOPO | MeshModel::MM_VERTMARK );
parent->getCurrentRenderMode().colorMode=vcg::GLW::CMPerVert;
parent->mm()->updateDataMask(MeshModel::MM_VERTCOLOR);
//ioMask|=MeshModel::IOM_VERTCOLOR;
first = true;
pressed = false;
dragging = false;
parent->update();
return true;
}
void EditSegment::EndEdit(MeshModel & , GLArea * ) {
//qDebug() << "Mesh Segmentation End Edit" << endl;
delete meshCutDialog;
meshCutDialog = 0;
delete meshcut_dock;
meshcut_dock = 0;
}
void EditSegment::Decorate(MeshModel & m, GLArea * gla) {
if (!glarea_map.contains(gla)) {
glarea_map.insert(gla, new MeshCutting<CMeshO>(&m.cm));
}
glarea = gla;
meshCut = glarea_map[glarea];
updateMatrixes();
QPoint mid=QPoint(current_point.x(), gla->curSiz.height()
- current_point.y());
if (first) {
first=false;
if (pixels) {
free(pixels);
}
pixels=(GLfloat *)malloc(sizeof(GLfloat)*gla->curSiz.width()
*gla->curSiz.height());
glReadPixels(0, 0, gla->curSiz.width(), gla->curSiz.height(),
GL_DEPTH_COMPONENT, GL_FLOAT, pixels);
}
if (dragging) {
dragging = false;
DrawXORCircle(m, glarea, false);
vector<CMeshO::VertexPointer> newSel;
vector<CMeshO::FacePointer> faceSel;
getInternFaces(m, &currentSelection, &newSel, &faceSel, gla, pen,
current_point, previous_point, pixels, mvmatrix, projmatrix,
viewport);
vector<CMeshO::VertexPointer>::iterator vi;
for (vi=newSel.begin(); vi!=newSel.end(); ++vi) {
if (mouse_button_pressed==Qt::LeftButton) {
meshCut->Mark(*vi, iF);
(*vi)->C()
= toVcgColor(meshCutDialog->getForegroundColor());
} else if (mouse_button_pressed==Qt::RightButton ){
meshCut->Mark(*vi, iB);
(*vi)->C()
= toVcgColor(meshCutDialog->getBackgroundColor());
} else if (mouse_button_pressed==Qt::MidButton) {
meshCut->Mark(*vi, U);
(*vi)->C()
= toVcgColor(Qt::white);
}
}
}
}
void EditSegment::mousePressEvent(QMouseEvent *event, MeshModel & m, GLArea * gla) {
//if (event->button() == Qt::MidButton) {
//gla->trackball.MouseDown(event->x(),gla->height()-event->y(), QT2VCG(event->button(), event->modifiers() ) );
//} else {
if (!glarea_map.contains(gla)) {
glarea_map.insert(gla, new MeshCutting<CMeshO>(&m.cm));
}
glarea = gla;
meshCut = glarea_map[glarea];
show_trackball = gla->isTrackBallVisible();
gla->showTrackBall(false);
first = true;
dragging = true;
mouse_button_pressed = event->button();
pressed = true;
current_point = event->pos();
previous_point = current_point;
currentSelection.clear();
//}
}
void EditSegment::mouseReleaseEvent(QMouseEvent *event, MeshModel & m, GLArea * gla) {
//if (event->button() == Qt::MidButton) {
//gla->trackball.MouseUp(event->x(),gla->height()-event->y(), QT2VCG(event->button(), event->modifiers() ) );
//} else {
if (!glarea_map.contains(gla)) {
glarea_map.insert(gla, new MeshCutting<CMeshO>(&m.cm));
}
glarea = gla;
meshCut = glarea_map[glarea];
gla->showTrackBall(show_trackball);
gla->update();
previous_point = current_point;
current_point = event->pos();
dragging = false;
pressed = false;
//}
}
void EditSegment::mouseMoveEvent(QMouseEvent *event, MeshModel & m, GLArea * gla) {
//if (event->button() == Qt::MidButton) {
//gla->trackball.MouseMove(event->x(),gla->height()-event->y());
//gla->setCursorTrack(gla->trackball.current_mode);
//} else {
if (!glarea_map.contains(gla)) {
glarea_map.insert(gla, new MeshCutting<CMeshO>(&m.cm));
}
glarea = gla;
meshCut = glarea_map[glarea];
if (!dragging)
previous_point = current_point;
current_point = event->pos();
dragging = true;
//}
gla->update();
}
void EditSegment::DrawXORCircle(MeshModel &m, GLArea * gla, bool doubleDraw) {
int PEZ=18;
if (!glarea_map.contains(gla)) {
glarea_map.insert(gla, new MeshCutting<CMeshO>(&m.cm));
}
glarea = gla;
meshCut = glarea_map[glarea];
/** paint the normal circle in pixel-mode */
//if (paintbox->paintType()==1) {
glMatrixMode(GL_PROJECTION);
glPushMatrix();
glLoadIdentity();
glOrtho(0, gla->curSiz.width(), gla->curSiz.height(), 0, -1, 1);
glMatrixMode(GL_MODELVIEW);
glPushMatrix();
glLoadIdentity();
glPushAttrib(GL_ENABLE_BIT);
glDisable(GL_DEPTH_TEST);
glDisable(GL_LIGHTING);
glDisable(GL_TEXTURE_2D);
glEnable(GL_COLOR_LOGIC_OP);
glLogicOp(GL_XOR);
glColor3f(1, 1, 1);
QPoint mid= QPoint(current_point.x(),/*gla->curSiz.height()-*/
current_point.y());
if (doubleDraw) {
glBegin(GL_LINE_LOOP);
for (int lauf=0; lauf<PEZ; lauf++)
glVertex2f(mid.x()+sin(M_PI*(float)lauf/9.0)*pen.radius, mid.y()
+cos(M_PI*(float)lauf/9.0)*pen.radius);
glEnd();
}
glBegin(GL_LINE_LOOP);
for (int lauf=0; lauf<PEZ; lauf++)
glVertex2f(mid.x()+sin(M_PI*(float)lauf/9.0)*pen.radius, mid.y()
+cos(M_PI*(float)lauf/9.0)*pen.radius);
glEnd();
glDisable(GL_LOGIC_OP);
// Closing 2D
glPopAttrib();
glPopMatrix(); // restore modelview
glMatrixMode(GL_PROJECTION);
glPopMatrix();
glMatrixMode(GL_MODELVIEW);
}
void EditSegment::MeshCutSlot() {
if (meshCut) {
meshCut->MeshCut(normalWeight,curvatureWeight);
meshCut->Colorize(selectForeground, doRefine);
}
glarea->update();
}
void EditSegment::UpdateCurvatureSlot() {
if (meshCut) {
meshCut->UpdateCurvature();
}
}
void EditSegment::SelectForegroundSlot(bool value) {
selectForeground = value;
}
void EditSegment::doRefineSlot(bool value) {
doRefine = value;
}
void EditSegment::changeCurvatureWeight(int value) {
curvatureWeight = value;
}
void EditSegment::changeNormalWeight(int value) {
normalWeight = value;
}
void EditSegment::changePenRadius(int value) {
pen.radius = value;
}
void EditSegment::ColorizeGaussianSlot() {
if (meshCut) {
meshCut->ColorizeCurvature(true);
glarea->update();
}
}
void EditSegment::ColorizeMeanSlot() {
if (meshCut) {
meshCut->ColorizeCurvature(false);
glarea->update();
}
}
void EditSegment::ResetSlot() {
if (meshCut) {
meshCut->Reset();
glarea->update();
}
}