command to creation section from plane and mesh

This commit is contained in:
wmayer 2017-02-19 14:05:00 +01:00
parent 845d6766f9
commit 46f5554c43
3 changed files with 205 additions and 57 deletions

View File

@ -943,61 +943,7 @@ CmdMeshTrimByPlane::CmdMeshTrimByPlane()
void CmdMeshTrimByPlane::activated(int)
{
Base::Type partType = Base::Type::fromName("Part::Plane");
std::vector<App::DocumentObject*> plane = getSelection().getObjectsOfType(partType);
if (plane.empty()) {
QMessageBox::warning(Gui::getMainWindow(),
qApp->translate("Mesh_TrimByPlane", "Select plane"),
qApp->translate("Mesh_TrimByPlane", "Please select a plane at which you trim the mesh."));
return;
}
Base::Placement plm = static_cast<App::GeoFeature*>(plane.front())->Placement.getValue();
Base::Vector3d normal(0,0,1);
plm.getRotation().multVec(normal, normal);
Base::Vector3d up(-1,0,0);
plm.getRotation().multVec(up, up);
Base::Vector3d view(0,1,0);
plm.getRotation().multVec(view, view);
Base::Vector3d base = plm.getPosition();
Base::Rotation rot(Base::Vector3d(0,0,1), view);
Base::Matrix4D mat;
rot.getValue(mat);
Base::ViewProjMatrix proj(mat);
openCommand("Trim with plane");
std::vector<App::DocumentObject*> docObj = Gui::Selection().getObjectsOfType(Mesh::Feature::getClassTypeId());
for (std::vector<App::DocumentObject*>::iterator it = docObj.begin(); it != docObj.end(); ++it) {
Mesh::MeshObject* mesh = static_cast<Mesh::Feature*>(*it)->Mesh.startEditing();
Base::BoundBox3d bbox = mesh->getBoundBox();
double len = bbox.CalcDiagonalLength();
// project center of bbox onto plane and use this as base point
Base::Vector3d cnt = bbox.GetCenter();
double dist = (cnt-base)*normal;
base = cnt - normal * dist;
Base::Vector3d p1 = base + up * len;
Base::Vector3d p2 = base - up * len;
Base::Vector3d p3 = p2 + normal * len;
Base::Vector3d p4 = p1 + normal * len;
p1 = mat * p1;
p2 = mat * p2;
p3 = mat * p3;
p4 = mat * p4;
Base::Polygon2d polygon2d;
polygon2d.Add(Base::Vector2d(p1.x, p1.y));
polygon2d.Add(Base::Vector2d(p2.x, p2.y));
polygon2d.Add(Base::Vector2d(p3.x, p3.y));
polygon2d.Add(Base::Vector2d(p4.x, p4.y));
Mesh::MeshObject::CutType type = Mesh::MeshObject::INNER;
mesh->trim(polygon2d, proj, type);
static_cast<Mesh::Feature*>(*it)->Mesh.finishEditing();
(*it)->purgeTouched();
}
commitCommand();
doCommand(Doc,"import MeshPartGui, FreeCADGui\nFreeCADGui.runCommand('MeshPart_TrimByPlane')\n");
}
bool CmdMeshTrimByPlane::isActive(void)
@ -1011,6 +957,34 @@ bool CmdMeshTrimByPlane::isActive(void)
//--------------------------------------------------------------------------------------
DEF_STD_CMD_A(CmdMeshSectionByPlane);
CmdMeshSectionByPlane::CmdMeshSectionByPlane()
: Command("Mesh_SectionByPlane")
{
sAppModule = "Mesh";
sGroup = QT_TR_NOOP("Mesh");
sMenuText = QT_TR_NOOP("Create section from mesh and plane");
sToolTipText = QT_TR_NOOP("Section from mesh and plane");
sStatusTip = QT_TR_NOOP("Section from mesh and plane");
}
void CmdMeshSectionByPlane::activated(int)
{
doCommand(Doc,"import MeshPartGui, FreeCADGui\nFreeCADGui.runCommand('MeshPart_SectionByPlane')\n");
}
bool CmdMeshSectionByPlane::isActive(void)
{
// Check for the selected mesh feature (all Mesh types)
if (getSelection().countObjectsOfType(Mesh::Feature::getClassTypeId()) != 1)
return false;
return true;
}
//--------------------------------------------------------------------------------------
DEF_STD_CMD_A(CmdMeshPolySplit);
CmdMeshPolySplit::CmdMeshPolySplit()
@ -1696,6 +1670,7 @@ void CreateMeshCommands(void)
rcCmdMgr.addCommand(new CmdMeshPolySplit());
rcCmdMgr.addCommand(new CmdMeshPolyTrim());
rcCmdMgr.addCommand(new CmdMeshTrimByPlane());
rcCmdMgr.addCommand(new CmdMeshSectionByPlane());
rcCmdMgr.addCommand(new CmdMeshToolMesh());
rcCmdMgr.addCommand(new CmdMeshTransform());
rcCmdMgr.addCommand(new CmdMeshEvaluation());

View File

@ -190,7 +190,8 @@ Gui::MenuItem* Workbench::setupMenuBar() const
<< "Mesh_FillupHoles" << "Mesh_FillInteractiveHole" << "Mesh_RemoveComponents"
<< "Mesh_RemoveCompByHand" << "Mesh_AddFacet" << "Mesh_Smoothing" << "Separator"
<< "Mesh_BuildRegularSolid" << boolean << "Separator" << "Mesh_Merge" << "Mesh_PolySelect" << "Mesh_PolyCut"
<< "Mesh_PolySplit" << "Mesh_PolySegm" << "Mesh_PolyTrim" << "Mesh_TrimByPlane" << "Mesh_Segmentation"
<< "Mesh_PolySplit" << "Mesh_PolySegm" << "Mesh_PolyTrim" << "Separator"
<< "Mesh_TrimByPlane" << "Mesh_SectionByPlane" << "Mesh_Segmentation"
<< "Mesh_VertexCurvature";
return root;
}

View File

@ -23,8 +23,12 @@
#include "PreCompiled.h"
#ifndef _PreComp_
# include <QApplication>
# include <QMessageBox>
#endif
#include <Mod/Mesh/App/MeshFeature.h>
#include <Gui/Application.h>
#include <Gui/Command.h>
#include <Gui/Control.h>
@ -37,7 +41,7 @@ using namespace std;
//===========================================================================
// MeshPart_Mesher
//===========================================================================
DEF_STD_CMD_A(CmdMeshPartMesher);
DEF_STD_CMD_A(CmdMeshPartMesher)
CmdMeshPartMesher::CmdMeshPartMesher()
: Command("MeshPart_Mesher")
@ -60,9 +64,177 @@ bool CmdMeshPartMesher::isActive(void)
return (hasActiveDocument() && !Gui::Control().activeDialog());
}
//--------------------------------------------------------------------------------------
DEF_STD_CMD_A(CmdMeshPartTrimByPlane)
CmdMeshPartTrimByPlane::CmdMeshPartTrimByPlane()
: Command("MeshPart_TrimByPlane")
{
sAppModule = "Mesh";
sGroup = QT_TR_NOOP("Mesh");
sMenuText = QT_TR_NOOP("Trim mesh with a plane");
sToolTipText = QT_TR_NOOP("Trims a mesh with a plane");
sStatusTip = QT_TR_NOOP("Trims a mesh with a plane");
}
void CmdMeshPartTrimByPlane::activated(int)
{
Base::Type partType = Base::Type::fromName("Part::Plane");
std::vector<App::DocumentObject*> plane = getSelection().getObjectsOfType(partType);
if (plane.empty()) {
QMessageBox::warning(Gui::getMainWindow(),
qApp->translate("MeshPart_TrimByPlane", "Select plane"),
qApp->translate("MeshPart_TrimByPlane", "Please select a plane at which you trim the mesh."));
return;
}
Base::Placement plm = static_cast<App::GeoFeature*>(plane.front())->Placement.getValue();
Base::Vector3d normal(0,0,1);
plm.getRotation().multVec(normal, normal);
Base::Vector3d up(-1,0,0);
plm.getRotation().multVec(up, up);
Base::Vector3d view(0,1,0);
plm.getRotation().multVec(view, view);
Base::Vector3d base = plm.getPosition();
Base::Rotation rot(Base::Vector3d(0,0,1), view);
Base::Matrix4D mat;
rot.getValue(mat);
Base::ViewProjMatrix proj(mat);
openCommand("Trim with plane");
std::vector<App::DocumentObject*> docObj = Gui::Selection().getObjectsOfType(Mesh::Feature::getClassTypeId());
for (std::vector<App::DocumentObject*>::iterator it = docObj.begin(); it != docObj.end(); ++it) {
Mesh::MeshObject* mesh = static_cast<Mesh::Feature*>(*it)->Mesh.startEditing();
Base::BoundBox3d bbox = mesh->getBoundBox();
double len = bbox.CalcDiagonalLength();
// project center of bbox onto plane and use this as base point
Base::Vector3d cnt = bbox.GetCenter();
double dist = (cnt-base)*normal;
base = cnt - normal * dist;
Base::Vector3d p1 = base + up * len;
Base::Vector3d p2 = base - up * len;
Base::Vector3d p3 = p2 + normal * len;
Base::Vector3d p4 = p1 + normal * len;
p1 = mat * p1;
p2 = mat * p2;
p3 = mat * p3;
p4 = mat * p4;
Base::Polygon2d polygon2d;
polygon2d.Add(Base::Vector2d(p1.x, p1.y));
polygon2d.Add(Base::Vector2d(p2.x, p2.y));
polygon2d.Add(Base::Vector2d(p3.x, p3.y));
polygon2d.Add(Base::Vector2d(p4.x, p4.y));
Mesh::MeshObject::CutType type = Mesh::MeshObject::INNER;
mesh->trim(polygon2d, proj, type);
static_cast<Mesh::Feature*>(*it)->Mesh.finishEditing();
(*it)->purgeTouched();
}
commitCommand();
}
bool CmdMeshPartTrimByPlane::isActive(void)
{
// Check for the selected mesh feature (all Mesh types)
if (getSelection().countObjectsOfType(Mesh::Feature::getClassTypeId()) != 1)
return false;
return true;
}
//===========================================================================
// MeshPart_Section
//===========================================================================
DEF_STD_CMD_A(CmdMeshPartSection)
CmdMeshPartSection::CmdMeshPartSection()
: Command("MeshPart_SectionByPlane")
{
sAppModule = "MeshPart";
sGroup = QT_TR_NOOP("Mesh");
sMenuText = QT_TR_NOOP("Create section from mesh and plane");
sToolTipText = QT_TR_NOOP("Section");
sWhatsThis = "MeshPart_Section";
sStatusTip = sToolTipText;
}
void CmdMeshPartSection::activated(int)
{
Base::Type partType = Base::Type::fromName("Part::Plane");
std::vector<App::DocumentObject*> plane = getSelection().getObjectsOfType(partType);
if (plane.empty()) {
QMessageBox::warning(Gui::getMainWindow(),
qApp->translate("MeshPart_Section", "Select plane"),
qApp->translate("MeshPart_Section", "Please select a plane at which you section the mesh."));
return;
}
Base::Placement plm = static_cast<App::GeoFeature*>(plane.front())->Placement.getValue();
Base::Vector3d normal(0,0,1);
plm.getRotation().multVec(normal, normal);
Base::Vector3d base = plm.getPosition();
openCommand("Section with plane");
std::vector<App::DocumentObject*> docObj = Gui::Selection().getObjectsOfType(Mesh::Feature::getClassTypeId());
Mesh::MeshObject::TPlane tplane;
tplane.first = Base::convertTo<Base::Vector3f>(base);
tplane.second = Base::convertTo<Base::Vector3f>(normal);
std::vector<Mesh::MeshObject::TPlane> sections;
sections.push_back(tplane);
Py::Module partModule(PyImport_ImportModule("Part"), true);
Py::Callable makeWire(partModule.getAttr("makePolygon"));
Py::Module appModule(PyImport_ImportModule("FreeCAD"), true);
Py::Callable addObject(appModule.getAttr("ActiveDocument").getAttr("addObject"));
for (std::vector<App::DocumentObject*>::iterator it = docObj.begin(); it != docObj.end(); ++it) {
const Mesh::MeshObject* mesh = static_cast<Mesh::Feature*>(*it)->Mesh.getValuePtr();
std::vector<Mesh::MeshObject::TPolylines> polylines;
mesh->crossSections(sections, polylines);
for (auto it2 : polylines) {
for (auto it3 : it2) {
Py::Tuple arg(1);
Py::List list;
for (auto it4 : it3) {
Py::Tuple pnt(3);
pnt.setItem(0, Py::Float(it4.x));
pnt.setItem(1, Py::Float(it4.y));
pnt.setItem(2, Py::Float(it4.z));
list.append(pnt);
}
arg.setItem(0, list);
Py::Object wire = makeWire.apply(arg);
Py::Tuple arg2(2);
arg2.setItem(0, Py::String("Part::Feature"));
arg2.setItem(1, Py::String("Section"));
Py::Object obj = addObject.apply(arg2);
obj.setAttr("Shape", wire);
}
}
}
updateActive();
commitCommand();
}
bool CmdMeshPartSection::isActive(void)
{
// Check for the selected mesh feature (all Mesh types)
if (getSelection().countObjectsOfType(Mesh::Feature::getClassTypeId()) != 1)
return false;
return true;
}
void CreateMeshPartCommands(void)
{
Gui::CommandManager &rcCmdMgr = Gui::Application::Instance->commandManager();
rcCmdMgr.addCommand(new CmdMeshPartMesher());
rcCmdMgr.addCommand(new CmdMeshPartTrimByPlane());
rcCmdMgr.addCommand(new CmdMeshPartSection());
}