meshlab/src/meshlabplugins/edit_align/AlignPairWidget.cpp

368 lines
11 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 <GL/glew.h>
#include <common/GLExtensionsManager.h>
#include "edit_align.h"
#include <QGLWidget>
#include "AlignPairWidget.h"
#include "AlignPairDialog.h"
#include <wrap/gl/pick.h>
#include <wrap/qt/trackball.h>
#include <wrap/gl/picking.h>
#include <wrap/gl/space.h>
#include <wrap/qt/gl_label.h>
#include <meshlab/glarea.h>
AlignPairWidget::AlignPairWidget(GLArea* ar, QWidget * parent)
:QGLWidget(parent, ar->mvc()->sharedDataContext())
{
gla = ar;
shared = ar->mvc()->sharedDataContext();
currentTrack = 0;
freeMesh = 0;
gluedTree = 0;
tt[0] = &trackLeft;
tt[1] = &trackRight;
isUsingVertexColor = false;
usePointRendering = false;
freePickedPointVec.clear();
gluedPickedPointVec.clear();
hasToPick = false;
hasToDelete = false;
pointToPick = vcg::Point2i(-1, -1);
shared->addView(context());
setAutoFillBackground(false);
}
void AlignPairWidget::initMesh(MeshNode *_freeMesh, MeshTree *_gluedTree)
{
freeMesh = _freeMesh;
gluedTree = _gluedTree;
assert(freeMesh->glued == false);
assert(gluedTree->gluedNum() > 0);
update();
}
void AlignPairWidget::initializeGL()
{
if (shared == NULL)
return;
GLExtensionsManager::initializeGLextensions(); //needed to init extensions, used by the aligner GL window while rendering
shared->addView(context());
glClearColor(0, 0, 0, 0);
glEnable(GL_LIGHTING);
glEnable(GL_LIGHT0);
glEnable(GL_NORMALIZE);
glEnable(GL_COLOR_MATERIAL);
glEnable(GL_CULL_FACE);
glEnable(GL_DEPTH_TEST);
}
void AlignPairWidget::paintEvent(QPaintEvent *)
{
if ((shared == NULL) || (gla == NULL))
return;
QPainter painter(this);
painter.beginNativePainting();
makeCurrent();
if (!isValid())return;
glEnable(GL_DEPTH_TEST);
glDepthMask(GL_TRUE);
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
if (freeMesh == 0 || gluedTree == 0) return;
for (int i = 0; i < 2; ++i)
{
if (i == 0)
{
MLRenderingData dt;
createRenderingData(freeMesh->m, dt);
shared->setRenderingDataPerMeshView(freeMesh->Id(), context(), dt);
shared->manageBuffers(freeMesh->Id());
glViewport(0, 0, (GLsizei)QTLogicalToDevice(this, width() / 2), (GLsizei)QTLogicalToDevice(this, height()));
}
else
{
for(auto ni=gluedTree->nodeMap.begin();ni!=gluedTree->nodeMap.end();++ni)
{
MeshNode *mn=ni->second;
if ((mn != NULL) && (mn->m != NULL) && mn->glued && mn != freeMesh && mn->m->visible)
{
MLRenderingData dt;
createRenderingData(mn->m, dt);
shared->setRenderingDataPerMeshView(mn->m->id(), context(), dt);
shared->manageBuffers(mn->m->id());
}
}
glViewport(QTLogicalToDevice(this, width() / 2), 0, (GLsizei)QTLogicalToDevice(this, width() / 2), (GLsizei)QTLogicalToDevice(this, height()));
}
glMatrixMode(GL_PROJECTION);
glLoadIdentity();
gluPerspective(30, (AlignPairWidget::width() / 2) / (float)AlignPairWidget::height(), 0.1, 100);
glMatrixMode(GL_MODELVIEW);
glLoadIdentity();
gluLookAt(0, 0, 6, 0, 0, 0, 0, 1, 0);
tt[i]->center = vcg::Point3f(0, 0, 0);
tt[i]->radius = 1;
tt[i]->GetView();
tt[i]->Apply();
Box3m bb;
if (i == 0)
bb.Add(freeMesh->tr(), freeMesh->bbox()); //bb.Import(freeMesh->bbox());
else
bb.Import(gluedTree->gluedBBox());
glPushMatrix();
if (allowscaling)
vcg::glScale(3.0f / bb.Diag());
else
vcg::glScale(3.0f / gluedTree->gluedBBox().Diag());
vcg::glTranslate(-bb.Center());
if (i == 0)
{
shared->draw(freeMesh->Id(), context());
drawPickedPoints(&painter, freePickedPointVec, vcg::Color4b(vcg::Color4b::Red));
}
else
{
// foreach(MeshNode *mn, gluedTree->nodeList)
for(auto ni=gluedTree->nodeMap.begin();ni!=gluedTree->nodeMap.end();++ni)
{
MeshNode *mn=ni->second;
if ((mn != NULL) && (mn->m != NULL) && mn->glued && mn != freeMesh && mn->m->visible)
{
shared->draw(mn->m->id(), context());
}
}
drawPickedPoints(&painter, gluedPickedPointVec, vcg::Color4b(vcg::Color4b::Blue));
}
int pickSide = (pointToPick[0] < QTLogicalToDevice(this, (width() / 2))) ? 0 : 1;
if (hasToPick && pickSide == i)
{
Point3m pp;
hasToPick = false;
if (vcg::Pick<Point3m>(pointToPick[0], pointToPick[1], pp))
{
std::vector<Point3m> &curVec = pickSide ? gluedPickedPointVec : freePickedPointVec;
qDebug("Picked point %i %i -> %f %f %f", pointToPick[0], pointToPick[1], pp[0], pp[1], pp[2]);
if (hasToDelete)
{
int bestInd = -1;
double bestDist = 10e100;
for (size_t i = 0; i < curVec.size(); ++i)
if (Distance(pp, curVec[i]) < bestDist)
{
bestDist = Distance(pp, curVec[i]);
bestInd = i;
}
hasToDelete = false;
if (bestInd >= 0)
curVec.erase(curVec.begin() + bestInd);
}
else curVec.push_back(pp);
hasToPick = false;
update();
}
}
glPopMatrix();
tt[i]->DrawPostApply();
}
painter.endNativePainting();
}
void AlignPairWidget::drawPickedPoints(QPainter *qp, std::vector<Point3m> &pointVec, vcg::Color4b color)
{
glPushAttrib(GL_ENABLE_BIT | GL_POINT_BIT | GL_CURRENT_BIT | GL_DEPTH_BUFFER_BIT);
glDisable(GL_LIGHTING);
glDisable(GL_TEXTURE);
glDepthFunc(GL_ALWAYS);
//glDisable(GL_DEPTH_TEST);
for (uint i = 0; i < pointVec.size(); ++i)
{
Point3m &pt = pointVec[i];
glPointSize(5.0);
glColor(vcg::Color4b(vcg::Color4b::Black));
glBegin(GL_POINTS);
glVertex(pt);
glEnd();
glPointSize(3.0);
glColor(color);
glBegin(GL_POINTS);
glVertex(pt);
glEnd();
vcg::glLabel::render(qp, pt, QString("%1").arg(i));
// renderText( pt[0],pt[1],pt[2], QString("%1").arg(i) );
}
glPopAttrib();
}
void AlignPairWidget::cleanDataOnClosing(int)
{
if (shared == NULL)
return;
shared->removeView(context());
}
void AlignPairWidget::createRenderingData(MeshModel* mm, MLRenderingData& dt)
{
if (mm == NULL)
return;
MLRenderingData::RendAtts atts;
MLPerViewGLOptions opts;
atts[MLRenderingData::ATT_NAMES::ATT_VERTPOSITION] = true;
atts[MLRenderingData::ATT_NAMES::ATT_VERTNORMAL] = true;
atts[MLRenderingData::ATT_NAMES::ATT_FACENORMAL] = true;
atts[MLRenderingData::ATT_NAMES::ATT_VERTCOLOR] = (mm->hasDataMask(MeshModel::MM_VERTCOLOR)) && (isUsingVertexColor);
dt.get(opts);
if ((mm->cm.fn == 0) || (this->usePointRendering))
{
opts._perpoint_mesh_color_enabled = !isUsingVertexColor;
opts._perpoint_fixed_color_enabled = !atts[MLRenderingData::ATT_NAMES::ATT_VERTCOLOR] && !opts._perpoint_mesh_color_enabled;
dt.set(MLRenderingData::PR_POINTS, atts);
}
else
{
opts._persolid_mesh_color_enabled = !isUsingVertexColor;
opts._persolid_fixed_color_enabled = !atts[MLRenderingData::ATT_NAMES::ATT_VERTCOLOR] && !opts._persolid_mesh_color_enabled;
dt.set(MLRenderingData::PR_SOLID, atts);
}
dt.set(opts);
}
void AlignPairWidget::keyReleaseEvent(QKeyEvent * e)
{
e->ignore();
for (int i = 0; i < 2; ++i)
{
if (e->key() == Qt::Key_Control) tt[i]->ButtonUp(QT2VCG(Qt::NoButton, Qt::ControlModifier));
if (e->key() == Qt::Key_Shift) tt[i]->ButtonUp(QT2VCG(Qt::NoButton, Qt::ShiftModifier));
if (e->key() == Qt::Key_Alt) tt[i]->ButtonUp(QT2VCG(Qt::NoButton, Qt::AltModifier));
}
update();
}
void AlignPairWidget::keyPressEvent(QKeyEvent * e)
{
e->ignore();
for (int i = 0; i < 2; ++i)
{
if (e->key() == Qt::Key_Control) tt[i]->ButtonDown(QT2VCG(Qt::NoButton, Qt::ControlModifier));
if (e->key() == Qt::Key_Shift) tt[i]->ButtonDown(QT2VCG(Qt::NoButton, Qt::ShiftModifier));
if (e->key() == Qt::Key_Alt) tt[i]->ButtonDown(QT2VCG(Qt::NoButton, Qt::AltModifier));
}
update();
}
void AlignPairWidget::mouseDoubleClickEvent(QMouseEvent * e)
{
hasToPick = true;
pointToPick = vcg::Point2i(QT2VCG_X(this, e), QT2VCG_Y(this, e));
if (e->modifiers()&Qt::ControlModifier)
hasToDelete = true;
update();
}
void AlignPairWidget::mousePressEvent(QMouseEvent * e)
{
e->accept();
setFocus();
int index = e->x() < (width() / 2) ? 0 : 1;
currentTrack = tt[index];
currentTrack->MouseDown(QT2VCG_X(this, e), QT2VCG_Y(this, e), QT2VCG(e->button(), e->modifiers()));
}
void AlignPairWidget::mouseMoveEvent(QMouseEvent * e)
{
if (!currentTrack)
{
qDebug("Warning useless mousemove");
return;
}
if (e->buttons()&Qt::RightButton)
{
float lx, ly;
lx = (e->x() / (float(width()) / 2.0f)) - 1.0;
ly = ((height() - e->y()) / (float(height()) / 2.0f)) - 1.0;
float lightPosF[] = { lx,ly,1.0,0.0 };
glPushMatrix();
glLoadIdentity();
glLightfv(GL_LIGHT0, GL_POSITION, lightPosF);
glPopMatrix();
update();
}
if (e->buttons()&Qt::LeftButton)
{
currentTrack->MouseMove(QT2VCG_X(this, e), QT2VCG_Y(this, e));
update();
}
}
void AlignPairWidget::mouseReleaseEvent(QMouseEvent * e)
{
if (!currentTrack) {
qDebug("Warning useless mouse release");
return;
}
currentTrack->MouseUp(QT2VCG_X(this, e), QT2VCG_Y(this, e), QT2VCG(e->button(), e->modifiers()));
currentTrack = 0;
}
void AlignPairWidget::wheelEvent(QWheelEvent * e)
{
const int WHEEL_STEP = 120;
//AlignPairDialog * dd = qobject_cast<AlignPairDialog *>(parent());
if (allowscaling)
{
int index = e->x() < (width() / 2) ? 0 : 1;
tt[index]->MouseWheel(e->delta() / float(WHEEL_STEP), QTWheel2VCG(e->modifiers()));
}
else
{
tt[0]->MouseWheel(e->delta() / float(WHEEL_STEP), QTWheel2VCG(e->modifiers()));
tt[1]->MouseWheel(e->delta() / float(WHEEL_STEP), QTWheel2VCG(e->modifiers()));
}
update();
}