Mesh segmentation

2012-05-18 15:57:32 +02:00
parent 9bbc10bdd9
commit 47d0418e7c
11 changed files with 488 additions and 10 deletions
--- a/src/Mod/Mesh/Gui/Segmentation.cpp
+++ b/src/Mod/Mesh/Gui/Segmentation.cpp
@@ -0,0 +1,136 @@
+/***************************************************************************
+ *   Copyright (c) 2012 Werner Mayer <wmayer[at]users.sourceforge.net>     *
+ *                                                                         *
+ *   This file is part of the FreeCAD CAx development system.              *
+ *                                                                         *
+ *   This library is free software; you can redistribute it and/or         *
+ *   modify it under the terms of the GNU Library General Public           *
+ *   License as published by the Free Software Foundation; either          *
+ *   version 2 of the License, or (at your option) any later version.      *
+ *                                                                         *
+ *   This library  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 Library General Public License for more details.                  *
+ *                                                                         *
+ *   You should have received a copy of the GNU Library General Public     *
+ *   License along with this library; see the file COPYING.LIB. If not,    *
+ *   write to the Free Software Foundation, Inc., 59 Temple Place,         *
+ *   Suite 330, Boston, MA  02111-1307, USA                                *
+ *                                                                         *
+ ***************************************************************************/
+
+
+#include "PreCompiled.h"
+
+#ifndef _PreComp_
+#endif
+
+#include "Segmentation.h"
+#include "ui_Segmentation.h"
+#include <App/Application.h>
+#include <App/Document.h>
+
+#include <Mod/Mesh/App/Core/Segmentation.h>
+#include <Mod/Mesh/App/Core/Curvature.h>
+#include <Mod/Mesh/App/Core/Smoothing.h>
+#include <Mod/Mesh/App/Mesh.h>
+#include <Mod/Mesh/App/MeshFeature.h>
+
+using namespace MeshGui;
+
+Segmentation::Segmentation(Mesh::Feature* mesh, QWidget* parent, Qt::WFlags fl)
+  : QWidget(parent, fl), myMesh(mesh)
+{
+    ui = new Ui_Segmentation;
+    ui->setupUi(this);
+    ui->numPln->setRange(1, INT_MAX);
+    ui->numPln->setValue(100);
+    ui->numCyl->setRange(1, INT_MAX);
+    ui->numCyl->setValue(100);
+    ui->numSph->setRange(1, INT_MAX);
+    ui->numSph->setValue(100);
+}
+
+Segmentation::~Segmentation()
+{
+    // no need to delete child widgets, Qt does it all for us
+    delete ui;
+}
+
+void Segmentation::accept()
+{
+    const Mesh::MeshObject* mesh = myMesh->Mesh.getValuePtr();
+    // make a copy because we might smooth the mesh before
+    MeshCore::MeshKernel kernel = mesh->getKernel();
+
+    if (ui->checkBoxSmooth->isChecked()) {
+        MeshCore::LaplaceSmoothing smoother(kernel);
+        smoother.Smooth(ui->smoothSteps->value());
+    }
+
+    MeshCore::MeshSegmentAlgorithm finder(kernel);
+    MeshCore::MeshCurvature meshCurv(kernel);
+    meshCurv.ComputePerVertex();
+
+    std::vector<MeshCore::MeshSurfaceSegment*> segm;
+    if (ui->groupBoxCyl->isChecked()) {
+        segm.push_back(new MeshCore::MeshCurvatureCylindricalSegment
+            (meshCurv.GetCurvature(), ui->numCyl->value(), ui->tolCyl->value(), ui->radCyl->value()));
+    }
+    if (ui->groupBoxSph->isChecked()) {
+        segm.push_back(new MeshCore::MeshCurvatureSphericalSegment
+            (meshCurv.GetCurvature(), ui->numSph->value(), ui->tolSph->value(), ui->radSph->value()));
+    }
+    if (ui->groupBoxPln->isChecked()) {
+        segm.push_back(new MeshCore::MeshCurvaturePlanarSegment
+            (meshCurv.GetCurvature(), ui->numPln->value(), ui->tolPln->value()));
+    }
+    finder.FindSegments(segm);
+
+    App::Document* document = App::GetApplication().getActiveDocument();
+    for (std::vector<MeshCore::MeshSurfaceSegment*>::iterator it = segm.begin(); it != segm.end(); ++it) {
+        const std::vector<MeshCore::MeshSegment>& data = (*it)->GetSegments();
+        for (std::vector<MeshCore::MeshSegment>::const_iterator jt = data.begin(); jt != data.end(); ++jt) {
+            Mesh::MeshObject* segment = mesh->meshFromSegment(*jt);
+            Mesh::Feature* feaSegm = static_cast<Mesh::Feature*>(document->addObject("Mesh::Feature", "Segment"));
+            Mesh::MeshObject* feaMesh = feaSegm->Mesh.startEditing();
+            feaMesh->swap(*segment);
+            feaSegm->Mesh.finishEditing();
+            delete segment;
+        }
+        delete (*it);
+    }
+}
+
+void Segmentation::changeEvent(QEvent *e)
+{
+    if (e->type() == QEvent::LanguageChange) {
+        ui->retranslateUi(this);
+    }
+    QWidget::changeEvent(e);
+}
+
+// ---------------------------------------
+
+/* TRANSLATOR MeshGui::TaskRemoveComponents */
+
+TaskSegmentation::TaskSegmentation(Mesh::Feature* mesh)
+{
+    widget = new Segmentation(mesh);
+    taskbox = new Gui::TaskView::TaskBox(
+        QPixmap(), widget->windowTitle(), false, 0);
+    taskbox->groupLayout()->addWidget(widget);
+    Content.push_back(taskbox);
+}
+
+TaskSegmentation::~TaskSegmentation()
+{
+    // automatically deleted in the sub-class
+}
+
+bool TaskSegmentation::accept()
+{
+    widget->accept();
+    return true;
+}