386 lines
14 KiB
C++
386 lines
14 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2016 WandererFan <wandererfan@gmail.com> *
|
|
* *
|
|
* 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 <sstream>
|
|
# include <cstring>
|
|
# include <cstdlib>
|
|
# include <exception>
|
|
# include <boost/regex.hpp>
|
|
# include <QString>
|
|
# include <QStringList>
|
|
# include <QRegExp>
|
|
# include <QMessageBox>
|
|
#include <QRectF>
|
|
|
|
#endif
|
|
|
|
#include <gp_Vec.hxx>
|
|
#include <gp_Pnt.hxx>
|
|
#include <gp_Trsf.hxx>
|
|
#include <gp_Dir.hxx>
|
|
#include <gp_Ax2.hxx>
|
|
#include <BRepAdaptor_Surface.hxx>
|
|
#include <BRepLProp_SLProps.hxx>
|
|
|
|
#include <App/Application.h>
|
|
#include <App/Document.h>
|
|
#include <App/DocumentObject.h>
|
|
#include <App/FeaturePython.h>
|
|
#include <App/Property.h>
|
|
#include <App/PropertyPythonObject.h>
|
|
#include <Base/Console.h>
|
|
#include <Base/Exception.h>
|
|
#include <Base/Parameter.h>
|
|
#include <Base/Type.h>
|
|
#include <Gui/Application.h>
|
|
#include <Gui/BitmapFactory.h>
|
|
#include <Gui/Command.h>
|
|
#include <Gui/Control.h>
|
|
#include <Gui/Document.h>
|
|
#include <Gui/Selection.h>
|
|
#include <Gui/MainWindow.h>
|
|
#include <Gui/MDIView.h>
|
|
#include <Gui/View3DInventor.h>
|
|
#include <Gui/View3DInventorViewer.h>
|
|
|
|
#include <Inventor/SbVec3f.h>
|
|
|
|
#include <Mod/Part/App/PartFeature.h>
|
|
#include <Mod/Part/App/Part2DObject.h>
|
|
#include <Mod/Spreadsheet/App/Sheet.h>
|
|
|
|
#include <Mod/TechDraw/App/DrawPage.h>
|
|
#include <Mod/TechDraw/App/DrawViewPart.h>
|
|
#include <Mod/TechDraw/App/DrawUtil.h>
|
|
#include <Mod/TechDraw/App/Geometry.h>
|
|
#include <Mod/TechDraw/App/ArrowPropEnum.h>
|
|
|
|
#include "QGVPage.h"
|
|
#include "MDIViewPage.h"
|
|
#include "ViewProviderPage.h"
|
|
#include "DrawGuiUtil.h"
|
|
|
|
using namespace TechDrawGui;
|
|
using namespace TechDraw;
|
|
|
|
void DrawGuiUtil::loadArrowBox(QComboBox* qcb)
|
|
{
|
|
qcb->clear();
|
|
int i = 0;
|
|
for (; i < ArrowPropEnum::ArrowCount; i++) {
|
|
qcb->addItem(tr(ArrowPropEnum::ArrowTypeEnums[i]));
|
|
QIcon itemIcon(QString::fromUtf8(ArrowPropEnum::ArrowTypeIcons[i].c_str()));
|
|
qcb->setItemIcon(i, itemIcon);
|
|
}
|
|
}
|
|
|
|
//===========================================================================
|
|
// validate helper routines
|
|
//===========================================================================
|
|
|
|
//find a page in Selection, Document or CurrentWindow.
|
|
TechDraw::DrawPage* DrawGuiUtil::findPage(Gui::Command* cmd)
|
|
{
|
|
TechDraw::DrawPage* page = nullptr;
|
|
|
|
//check Selection for a page
|
|
std::vector<App::DocumentObject*> selPages = cmd->getSelection().
|
|
getObjectsOfType(TechDraw::DrawPage::getClassTypeId());
|
|
if (selPages.empty()) {
|
|
//no page in selection, try document
|
|
selPages = cmd->getDocument()->getObjectsOfType(TechDraw::DrawPage::getClassTypeId());
|
|
if (selPages.empty()) {
|
|
//no page in document
|
|
QMessageBox::warning(Gui::getMainWindow(), QObject::tr("No page found"),
|
|
QObject::tr("No Drawing Pages in document."));
|
|
} else if (selPages.size() > 1) {
|
|
//multiple pages in document, but none selected
|
|
//use active page if there is one
|
|
Gui::MainWindow* w = Gui::getMainWindow();
|
|
Gui::MDIView* mv = w->activeWindow();
|
|
MDIViewPage* mvp = dynamic_cast<MDIViewPage*>(mv);
|
|
if (mvp) {
|
|
QString windowTitle = mvp->windowTitle();
|
|
QGVPage* qp = mvp->getQGVPage();
|
|
page = qp->getDrawPage();
|
|
} else {
|
|
// no active page
|
|
QMessageBox::warning(Gui::getMainWindow(), QObject::tr("Which page?"),
|
|
QObject::tr("Can not determine correct page."));
|
|
}
|
|
} else {
|
|
//only 1 page in document - use it
|
|
page = static_cast<TechDraw::DrawPage*>(selPages.front());
|
|
}
|
|
} else if (selPages.size() > 1) {
|
|
//multiple pages in selection
|
|
QMessageBox::warning(Gui::getMainWindow(), QObject::tr("Too many pages"),
|
|
QObject::tr("Select only 1 page."));
|
|
} else {
|
|
//exactly 1 page in selection, use it
|
|
page = static_cast<TechDraw::DrawPage*>(selPages.front());
|
|
}
|
|
|
|
return page;
|
|
}
|
|
|
|
bool DrawGuiUtil::isDraftObject(App::DocumentObject* obj)
|
|
{
|
|
bool result = false;
|
|
App::Property* proxy = obj->getPropertyByName("Proxy");
|
|
|
|
if (proxy != nullptr) {
|
|
//if no proxy, can not be Draft obj
|
|
//if has proxy, might be Draft obj
|
|
App::PropertyPythonObject* proxyPy = dynamic_cast<App::PropertyPythonObject*>(proxy);
|
|
std::stringstream ss;
|
|
if (proxyPy != nullptr) {
|
|
Py::Object proxyObj = proxyPy->getValue();
|
|
std::stringstream ss;
|
|
if (proxyPy != nullptr) {
|
|
Base::PyGILStateLocker lock;
|
|
try {
|
|
if (proxyObj.hasAttr("__module__")) {
|
|
Py::String mod(proxyObj.getAttr("__module__"));
|
|
ss << (std::string)mod;
|
|
if (ss.str().find("Draft") != std::string::npos) {
|
|
result = true;
|
|
} else if (ss.str().find("draft") != std::string::npos) {
|
|
result = true;
|
|
}
|
|
}
|
|
}
|
|
catch (Py::Exception&) {
|
|
Base::PyException e; // extract the Python error text
|
|
e.ReportException();
|
|
result = false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool DrawGuiUtil::isArchObject(App::DocumentObject* obj)
|
|
{
|
|
bool result = false;
|
|
App::Property* proxy = obj->getPropertyByName("Proxy");
|
|
|
|
if (proxy != nullptr) {
|
|
//if no proxy, can not be Arch obj
|
|
//if has proxy, might be Arch obj
|
|
App::PropertyPythonObject* proxyPy = dynamic_cast<App::PropertyPythonObject*>(proxy);
|
|
Py::Object proxyObj = proxyPy->getValue();
|
|
std::stringstream ss;
|
|
if (proxyPy != nullptr) {
|
|
Base::PyGILStateLocker lock;
|
|
try {
|
|
if (proxyObj.hasAttr("__module__")) {
|
|
Py::String mod(proxyObj.getAttr("__module__"));
|
|
ss << (std::string)mod;
|
|
//does this have to be an ArchSection, or can it be any Arch object?
|
|
if (ss.str().find("Arch") != std::string::npos) {
|
|
result = true;
|
|
}
|
|
}
|
|
}
|
|
catch (Py::Exception&) {
|
|
Base::PyException e; // extract the Python error text
|
|
e.ReportException();
|
|
result = false;
|
|
}
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool DrawGuiUtil::isArchSection(App::DocumentObject* obj)
|
|
{
|
|
bool result = false;
|
|
App::Property* proxy = obj->getPropertyByName("Proxy");
|
|
|
|
if (proxy != nullptr) {
|
|
//if no proxy, can not be Arch obj
|
|
//if has proxy, might be Arch obj
|
|
App::PropertyPythonObject* proxyPy = dynamic_cast<App::PropertyPythonObject*>(proxy);
|
|
Py::Object proxyObj = proxyPy->getValue();
|
|
std::stringstream ss;
|
|
if (proxyPy != nullptr) {
|
|
Base::PyGILStateLocker lock;
|
|
try {
|
|
if (proxyObj.hasAttr("__module__")) {
|
|
Py::String mod(proxyObj.getAttr("__module__"));
|
|
ss << (std::string)mod;
|
|
//does this have to be an ArchSection, or can it be other Arch objects?
|
|
if (ss.str().find("ArchSectionPlane") != std::string::npos) {
|
|
result = true;
|
|
}
|
|
}
|
|
}
|
|
catch (Py::Exception&) {
|
|
Base::PyException e; // extract the Python error text
|
|
e.ReportException();
|
|
result = false;
|
|
}
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool DrawGuiUtil::needPage(Gui::Command* cmd)
|
|
{
|
|
//need a Document and a Page
|
|
bool active = false;
|
|
if (cmd->hasActiveDocument()) {
|
|
auto drawPageType( TechDraw::DrawPage::getClassTypeId() );
|
|
auto selPages = cmd->getDocument()->getObjectsOfType(drawPageType);
|
|
if (!selPages.empty()) {
|
|
active = true;
|
|
}
|
|
}
|
|
return active;
|
|
}
|
|
|
|
bool DrawGuiUtil::needView(Gui::Command* cmd, bool partOnly)
|
|
{
|
|
bool haveView = false;
|
|
if (cmd->hasActiveDocument()) {
|
|
if (partOnly) {
|
|
auto drawPartType (TechDraw::DrawViewPart::getClassTypeId());
|
|
auto selParts = cmd->getDocument()->getObjectsOfType(drawPartType);
|
|
if (!selParts.empty()) {
|
|
haveView = true;
|
|
}
|
|
} else {
|
|
auto drawViewType (TechDraw::DrawView::getClassTypeId());
|
|
auto selParts = cmd->getDocument()->getObjectsOfType(drawViewType);
|
|
if (!selParts.empty()) {
|
|
haveView = true;
|
|
}
|
|
}
|
|
}
|
|
return haveView;
|
|
}
|
|
|
|
void DrawGuiUtil::dumpRectF(const char* text, const QRectF& r)
|
|
{
|
|
Base::Console().Message("DUMP - dumpRectF - %s\n",text);
|
|
double left = r.left();
|
|
double right = r.right();
|
|
double top = r.top();
|
|
double bottom = r.bottom();
|
|
Base::Console().Message("Extents: L: %.3f, R: %.3f, T: %.3f, B: %.3f\n",left,right,top,bottom);
|
|
Base::Console().Message("Size: W: %.3f H: %.3f\n",r.width(),r.height());
|
|
Base::Console().Message("Centre: (%.3f, %.3f)\n",r.center().x(),r.center().y());
|
|
}
|
|
|
|
void DrawGuiUtil::dumpPointF(const char* text, const QPointF& p)
|
|
{
|
|
Base::Console().Message("DUMP - dumpPointF - %s\n",text);
|
|
Base::Console().Message("Point: (%.3f, %.3f)\n",p.x(),p.y());
|
|
}
|
|
|
|
std::pair<Base::Vector3d,Base::Vector3d> DrawGuiUtil::get3DDirAndRot()
|
|
{
|
|
std::pair<Base::Vector3d,Base::Vector3d> result;
|
|
Base::Vector3d viewDir(0.0,-1.0,0.0); //default to front
|
|
Base::Vector3d viewUp(0.0,0.0,1.0); //default to top
|
|
Base::Vector3d viewRight(1.0,0.0,0.0); //default to right
|
|
std::list<Gui::MDIView*> mdis = Gui::Application::Instance->activeDocument()->getMDIViews();
|
|
Gui::View3DInventor *view;
|
|
Gui::View3DInventorViewer *viewer = nullptr;
|
|
for (auto& m: mdis) { //find the 3D viewer
|
|
view = dynamic_cast<Gui::View3DInventor*>(m);
|
|
if (view) {
|
|
viewer = view->getViewer();
|
|
break;
|
|
}
|
|
}
|
|
if (!viewer) {
|
|
Base::Console().Log("LOG - DrawGuiUtil could not find a 3D viewer\n");
|
|
return std::make_pair( viewDir, viewRight);
|
|
}
|
|
|
|
SbVec3f dvec = viewer->getViewDirection();
|
|
SbVec3f upvec = viewer->getUpDirection();
|
|
|
|
viewDir = Base::Vector3d(dvec[0], dvec[1], dvec[2]);
|
|
viewDir = viewDir * (-1.0); // Inventor dir is opposite TD projection dir
|
|
viewUp = Base::Vector3d(upvec[0],upvec[1],upvec[2]);
|
|
|
|
// Base::Vector3d dirXup = viewDir.Cross(viewUp);
|
|
Base::Vector3d right = viewUp.Cross(viewDir);
|
|
|
|
result = std::make_pair(viewDir,right);
|
|
return result;
|
|
}
|
|
|
|
std::pair<Base::Vector3d,Base::Vector3d> DrawGuiUtil::getProjDirFromFace(App::DocumentObject* obj, std::string faceName)
|
|
{
|
|
std::pair<Base::Vector3d,Base::Vector3d> d3Dirs = get3DDirAndRot();
|
|
Base::Vector3d d3Up = (d3Dirs.first).Cross(d3Dirs.second);
|
|
std::pair<Base::Vector3d,Base::Vector3d> dirs;
|
|
dirs.first = Base::Vector3d(0.0,0.0,1.0); //set a default
|
|
dirs.second = Base::Vector3d(1.0,0.0,0.0);
|
|
Base::Vector3d projDir, rotVec;
|
|
projDir = d3Dirs.first;
|
|
rotVec = d3Dirs.second;
|
|
|
|
auto ts = Part::Feature::getShape(obj,faceName.c_str(),true);
|
|
if(ts.IsNull() || ts.ShapeType()!=TopAbs_FACE) {
|
|
Base::Console().Warning("getProjDirFromFace(%s) is not a Face\n",faceName.c_str());
|
|
return dirs;
|
|
}
|
|
|
|
const TopoDS_Face& face = TopoDS::Face(ts);
|
|
TopAbs_Orientation orient = face.Orientation();
|
|
BRepAdaptor_Surface adapt(face);
|
|
|
|
double u1 = adapt.FirstUParameter();
|
|
double u2 = adapt.LastUParameter();
|
|
double v1 = adapt.FirstVParameter();
|
|
double v2 = adapt.LastVParameter();
|
|
double uMid = (u1+u2)/2.0;
|
|
double vMid = (v1+v2)/2.0;
|
|
|
|
BRepLProp_SLProps props(adapt,uMid,vMid,2,Precision::Confusion());
|
|
if (props.IsNormalDefined()) {
|
|
gp_Dir vec = props.Normal();
|
|
projDir = Base::Vector3d(vec.X(),vec.Y(),vec.Z());
|
|
rotVec = projDir.Cross(d3Up);
|
|
if (orient != TopAbs_FORWARD) {
|
|
projDir = projDir * (-1.0);
|
|
}
|
|
}
|
|
else {
|
|
Base::Console().Log("Selected Face has no normal at midpoint\n");
|
|
}
|
|
|
|
dirs = std::make_pair(projDir,rotVec);
|
|
return dirs;
|
|
}
|
|
|
|
|