363 lines
14 KiB
C++
363 lines
14 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2006 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_
|
|
# include <algorithm>
|
|
# include <QFileInfo>
|
|
# include <QInputDialog>
|
|
# include <Inventor/events/SoMouseButtonEvent.h>
|
|
#endif
|
|
|
|
#include <Base/Exception.h>
|
|
#include <Base/Matrix.h>
|
|
#include <App/Application.h>
|
|
#include <App/Document.h>
|
|
#include <Gui/Application.h>
|
|
#include <Gui/Document.h>
|
|
#include <Gui/MainWindow.h>
|
|
#include <Gui/Command.h>
|
|
#include <Gui/FileDialog.h>
|
|
#include <Gui/Selection.h>
|
|
#include <Gui/ViewProvider.h>
|
|
#include <Gui/View3DInventor.h>
|
|
#include <Gui/View3DInventorViewer.h>
|
|
#include <Gui/WaitCursor.h>
|
|
|
|
#include "../App/PointsFeature.h"
|
|
#include "../App/Properties.h"
|
|
#include "DlgPointsReadImp.h"
|
|
#include "ViewProvider.h"
|
|
|
|
|
|
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
|
|
//===========================================================================
|
|
// CmdPointsImport
|
|
//===========================================================================
|
|
DEF_STD_CMD_A(CmdPointsImport);
|
|
|
|
CmdPointsImport::CmdPointsImport()
|
|
: Command("Points_Import")
|
|
{
|
|
sAppModule = "Points";
|
|
sGroup = QT_TR_NOOP("Points");
|
|
sMenuText = QT_TR_NOOP("Import points...");
|
|
sToolTipText = QT_TR_NOOP("Imports a point cloud");
|
|
sWhatsThis = QT_TR_NOOP("Imports a point cloud");
|
|
sStatusTip = QT_TR_NOOP("Imports a point cloud");
|
|
sPixmap = "Points_Import_Point_cloud";
|
|
}
|
|
|
|
void CmdPointsImport::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
|
|
QString fn = Gui::FileDialog::getOpenFileName(Gui::getMainWindow(),
|
|
QString::null, QString(), QString::fromLatin1("%1 (*.asc);;%2 (*.*)")
|
|
.arg(QObject::tr("Ascii Points")).arg(QObject::tr("All Files")));
|
|
if (fn.isEmpty())
|
|
return;
|
|
|
|
if (!fn.isEmpty()) {
|
|
QFileInfo fi;
|
|
fi.setFile(fn);
|
|
|
|
openCommand("Import points");
|
|
QByteArray name = fi.baseName().toLatin1();
|
|
Points::Feature* pts = static_cast<Points::Feature*>(getActiveGuiDocument()->getDocument()->
|
|
addObject("Points::Feature", static_cast<const char*>(name)));
|
|
Points::PointKernel* kernel = pts->Points.startEditing();
|
|
kernel->load(static_cast<const char*>(fn.toLatin1()));
|
|
pts->Points.finishEditing();
|
|
commitCommand();
|
|
|
|
updateActive();
|
|
}
|
|
}
|
|
|
|
bool CmdPointsImport::isActive(void)
|
|
{
|
|
if (getActiveGuiDocument())
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
DEF_STD_CMD_A(CmdPointsExport);
|
|
|
|
CmdPointsExport::CmdPointsExport()
|
|
: Command("Points_Export")
|
|
{
|
|
sAppModule = "Points";
|
|
sGroup = QT_TR_NOOP("Points");
|
|
sMenuText = QT_TR_NOOP("Export points...");
|
|
sToolTipText = QT_TR_NOOP("Exports a point cloud");
|
|
sWhatsThis = QT_TR_NOOP("Exports a point cloud");
|
|
sStatusTip = QT_TR_NOOP("Exports a point cloud");
|
|
sPixmap = "Points_Export_Point_cloud";
|
|
}
|
|
|
|
void CmdPointsExport::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
|
|
std::vector<App::DocumentObject*> points = getSelection().getObjectsOfType(Points::Feature::getClassTypeId());
|
|
for (std::vector<App::DocumentObject*>::const_iterator it = points.begin(); it != points.end(); ++it) {
|
|
QString fn = Gui::FileDialog::getSaveFileName(Gui::getMainWindow(),
|
|
QString::null, QString(), QString::fromLatin1("%1 (*.asc);;%2 (*.*)")
|
|
.arg(QObject::tr("Ascii Points")).arg(QObject::tr("All Files")));
|
|
if (fn.isEmpty())
|
|
break;
|
|
|
|
if (!fn.isEmpty()) {
|
|
QFileInfo fi;
|
|
fi.setFile(fn);
|
|
|
|
Points::Feature* pts = static_cast<Points::Feature*>(*it);
|
|
pts->Points.getValue().save(static_cast<const char*>(fn.toLatin1()));
|
|
}
|
|
}
|
|
}
|
|
|
|
bool CmdPointsExport::isActive(void)
|
|
{
|
|
return getSelection().countObjectsOfType(Points::Feature::getClassTypeId()) > 0;
|
|
}
|
|
|
|
DEF_STD_CMD_A(CmdPointsTransform);
|
|
|
|
CmdPointsTransform::CmdPointsTransform()
|
|
:Command("Points_Transform")
|
|
{
|
|
sAppModule = "Points";
|
|
sGroup = QT_TR_NOOP("Points");
|
|
sMenuText = QT_TR_NOOP("Transform Points");
|
|
sToolTipText = QT_TR_NOOP("Test to transform a point cloud");
|
|
sWhatsThis = QT_TR_NOOP("Test to transform a point cloud");
|
|
sStatusTip = QT_TR_NOOP("Test to transform a point cloud");
|
|
sPixmap = "Test1";
|
|
}
|
|
|
|
void CmdPointsTransform::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
|
|
// This is a test command to transform a point cloud directly written in C++ (not Python)
|
|
Base::Placement trans;
|
|
trans.setRotation(Base::Rotation(Base::Vector3d(0.0, 0.0, 1.0), 1.570796));
|
|
|
|
openCommand("Transform points");
|
|
//std::vector<App::DocumentObject*> points = getSelection().getObjectsOfType(Points::Feature::getClassTypeId());
|
|
//for (std::vector<App::DocumentObject*>::const_iterator it = points.begin(); it != points.end(); ++it) {
|
|
// Base::Placement p = static_cast<Points::Feature*>(*it)->Placement.getValue();
|
|
// p._rot *= Base::Rotation(Base::Vector3d(0.0, 0.0, 1.0), 1.570796);
|
|
// static_cast<Points::Feature*>(*it)->Placement.setValue(p);
|
|
//}
|
|
commitCommand();
|
|
}
|
|
|
|
bool CmdPointsTransform::isActive(void)
|
|
{
|
|
return getSelection().countObjectsOfType(Points::Feature::getClassTypeId()) > 0;
|
|
}
|
|
|
|
DEF_STD_CMD_A(CmdPointsConvert);
|
|
|
|
CmdPointsConvert::CmdPointsConvert()
|
|
:Command("Points_Convert")
|
|
{
|
|
sAppModule = "Points";
|
|
sGroup = QT_TR_NOOP("Points");
|
|
sMenuText = QT_TR_NOOP("Convert to points...");
|
|
sToolTipText = QT_TR_NOOP("Convert to points");
|
|
sWhatsThis = QT_TR_NOOP("Convert to points");
|
|
sStatusTip = QT_TR_NOOP("Convert to points");
|
|
}
|
|
|
|
void CmdPointsConvert::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
|
|
bool ok;
|
|
double tol = QInputDialog::getDouble(Gui::getMainWindow(), QObject::tr("Distance"),
|
|
QObject::tr("Enter maximum distance:"), 0.1, 0.05, 10.0, 2, &ok);
|
|
if (!ok)
|
|
return;
|
|
|
|
Gui::WaitCursor wc;
|
|
openCommand("Convert to points");
|
|
std::vector<App::DocumentObject*> geoObject = getSelection().getObjectsOfType(Base::Type::fromName("App::GeoFeature"));
|
|
|
|
bool addedPoints = false;
|
|
for (std::vector<App::DocumentObject*>::iterator it = geoObject.begin(); it != geoObject.end(); ++it) {
|
|
const App::PropertyComplexGeoData* prop = static_cast<App::GeoFeature*>(*it)->getPropertyOfGeometry();
|
|
if (prop) {
|
|
const Data::ComplexGeoData* data = prop->getComplexData();
|
|
std::vector<Base::Vector3d> vertexes;
|
|
std::vector<Base::Vector3d> normals;
|
|
data->getPoints(vertexes, normals, static_cast<float>(tol));
|
|
if (!vertexes.empty()) {
|
|
Points::Feature* fea = 0;
|
|
if (vertexes.size() == normals.size()) {
|
|
fea = static_cast<Points::Feature*>(Base::Type::fromName("Points::FeatureCustom").createInstance());
|
|
if (!fea) {
|
|
Base::Console().Error("Failed to create instance of 'Points::FeatureCustom'\n");
|
|
continue;
|
|
}
|
|
Points::PropertyNormalList* prop = static_cast<Points::PropertyNormalList*>
|
|
(fea->addDynamicProperty("Points::PropertyNormalList", "Normal"));
|
|
if (prop) {
|
|
std::vector<Base::Vector3f> normf;
|
|
normf.resize(normals.size());
|
|
std::transform(normals.begin(), normals.end(), normf.begin(), Base::toVector<float, double>);
|
|
prop->setValues(normf);
|
|
}
|
|
}
|
|
else {
|
|
fea = new Points::Feature;
|
|
}
|
|
|
|
Points::PointKernel kernel;
|
|
kernel.reserve(vertexes.size());
|
|
for (std::vector<Base::Vector3d>::iterator pt = vertexes.begin(); pt != vertexes.end(); ++pt)
|
|
kernel.push_back(*pt);
|
|
fea->Points.setValue(kernel);
|
|
|
|
App::Document* doc = (*it)->getDocument();
|
|
doc->addObject(fea, "Points");
|
|
fea->purgeTouched();
|
|
addedPoints = true;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (addedPoints)
|
|
commitCommand();
|
|
else
|
|
abortCommand();
|
|
}
|
|
|
|
bool CmdPointsConvert::isActive(void)
|
|
{
|
|
return getSelection().countObjectsOfType(Base::Type::fromName("App::GeoFeature")) > 0;
|
|
}
|
|
|
|
DEF_STD_CMD_A(CmdPointsPolyCut);
|
|
|
|
CmdPointsPolyCut::CmdPointsPolyCut()
|
|
:Command("Points_PolyCut")
|
|
{
|
|
sAppModule = "Points";
|
|
sGroup = QT_TR_NOOP("Points");
|
|
sMenuText = QT_TR_NOOP("Cut point cloud");
|
|
sToolTipText = QT_TR_NOOP("Cuts a point cloud with a picked polygon");
|
|
sWhatsThis = QT_TR_NOOP("Cuts a point cloud with a picked polygon");
|
|
sStatusTip = QT_TR_NOOP("Cuts a point cloud with a picked polygon");
|
|
sPixmap = "PolygonPick";
|
|
}
|
|
|
|
void CmdPointsPolyCut::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
|
|
std::vector<App::DocumentObject*> docObj = Gui::Selection().getObjectsOfType(Points::Feature::getClassTypeId());
|
|
for (std::vector<App::DocumentObject*>::iterator it = docObj.begin(); it != docObj.end(); ++it) {
|
|
if (it == docObj.begin()) {
|
|
Gui::Document* doc = getActiveGuiDocument();
|
|
Gui::MDIView* view = doc->getActiveView();
|
|
if (view->getTypeId().isDerivedFrom(Gui::View3DInventor::getClassTypeId())) {
|
|
Gui::View3DInventorViewer* viewer = ((Gui::View3DInventor*)view)->getViewer();
|
|
viewer->setEditing(true);
|
|
viewer->startSelection(Gui::View3DInventorViewer::Lasso);
|
|
viewer->addEventCallback(SoMouseButtonEvent::getClassTypeId(), PointsGui::ViewProviderPoints::clipPointsCallback);
|
|
}
|
|
else {
|
|
return;
|
|
}
|
|
}
|
|
|
|
Gui::ViewProvider* pVP = getActiveGuiDocument()->getViewProvider( *it );
|
|
pVP->startEditing();
|
|
}
|
|
}
|
|
|
|
bool CmdPointsPolyCut::isActive(void)
|
|
{
|
|
// Check for the selected mesh feature (all Mesh types)
|
|
return getSelection().countObjectsOfType(Points::Feature::getClassTypeId()) > 0;
|
|
}
|
|
|
|
DEF_STD_CMD_A(CmdPointsMerge)
|
|
|
|
CmdPointsMerge::CmdPointsMerge()
|
|
:Command("Points_Merge")
|
|
{
|
|
sAppModule = "Points";
|
|
sGroup = QT_TR_NOOP("Points");
|
|
sMenuText = QT_TR_NOOP("Merge point clouds");
|
|
sToolTipText = QT_TR_NOOP("Merge several point clouds into one");
|
|
sWhatsThis = QT_TR_NOOP("Merge several point clouds into one");
|
|
sStatusTip = QT_TR_NOOP("Merge several point clouds into one");
|
|
}
|
|
|
|
void CmdPointsMerge::activated(int iMsg)
|
|
{
|
|
Q_UNUSED(iMsg);
|
|
|
|
App::Document* doc = App::GetApplication().getActiveDocument();
|
|
doc->openTransaction("Merge point clouds");
|
|
Points::Feature* pts = static_cast<Points::Feature*>(doc->addObject("Points::Feature", "Merged Points"));
|
|
Points::PointKernel* kernel = pts->Points.startEditing();
|
|
|
|
std::vector<App::DocumentObject*> docObj = Gui::Selection().getObjectsOfType(Points::Feature::getClassTypeId());
|
|
for (std::vector<App::DocumentObject*>::iterator it = docObj.begin(); it != docObj.end(); ++it) {
|
|
const Points::PointKernel& k = static_cast<Points::Feature*>(*it)->Points.getValue();
|
|
std::size_t numPts = kernel->size();
|
|
kernel->resize(numPts + k.size());
|
|
for (std::size_t i=0; i<k.size(); ++i) {
|
|
kernel->setPoint(i+numPts, k.getPoint(i));
|
|
}
|
|
}
|
|
|
|
pts->Points.finishEditing();
|
|
doc->commitTransaction();
|
|
updateActive();
|
|
}
|
|
|
|
bool CmdPointsMerge::isActive(void)
|
|
{
|
|
return getSelection().countObjectsOfType(Points::Feature::getClassTypeId()) > 1;
|
|
}
|
|
|
|
void CreatePointsCommands(void)
|
|
{
|
|
Gui::CommandManager &rcCmdMgr = Gui::Application::Instance->commandManager();
|
|
rcCmdMgr.addCommand(new CmdPointsImport());
|
|
rcCmdMgr.addCommand(new CmdPointsExport());
|
|
rcCmdMgr.addCommand(new CmdPointsTransform());
|
|
rcCmdMgr.addCommand(new CmdPointsConvert());
|
|
rcCmdMgr.addCommand(new CmdPointsPolyCut());
|
|
rcCmdMgr.addCommand(new CmdPointsMerge());
|
|
}
|