232 lines
8.3 KiB
C++
232 lines
8.3 KiB
C++
/******************************************************************************
|
|
* Copyright (c)2012 Konstantinos Poulios <logari81@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 <TopoDS.hxx>
|
|
# include <TopoDS_Edge.hxx>
|
|
# include <TopoDS_Face.hxx>
|
|
# include <BRepAdaptor_Curve.hxx>
|
|
# include <BRepAdaptor_Surface.hxx>
|
|
#endif
|
|
|
|
#include <App/Plane.h>
|
|
#include <App/Part.h>
|
|
#include <App/Line.h>
|
|
#include <Gui/Application.h>
|
|
#include <Gui/Document.h>
|
|
#include <Mod/Part/App/TopoShape.h>
|
|
#include <Mod/Part/App/PartFeature.h>
|
|
#include <Mod/PartDesign/App/Feature.h>
|
|
#include <Mod/PartDesign/App/Body.h>
|
|
#include <Mod/PartDesign/App/DatumPoint.h>
|
|
#include <Mod/PartDesign/App/DatumLine.h>
|
|
#include <Mod/PartDesign/App/DatumPlane.h>
|
|
#include "ReferenceSelection.h"
|
|
|
|
using namespace PartDesignGui;
|
|
using namespace Gui;
|
|
|
|
/* TRANSLATOR PartDesignGui::ReferenceSelection.cpp */
|
|
|
|
bool ReferenceSelection::allow(App::Document* pDoc, App::DocumentObject* pObj, const char* sSubName)
|
|
{
|
|
PartDesign::Body* ActivePartObject = Gui::Application::Instance->activeView()->getActiveObject<PartDesign::Body*>(PDBODYKEY);
|
|
App::Part* activePart = Gui::Application::Instance->activeView()->getActiveObject<App::Part*>("Part");
|
|
|
|
// Don't allow selection in other document
|
|
if ((support != NULL) && (pDoc != support->getDocument()))
|
|
return false;
|
|
|
|
if (plane && (pObj->getTypeId().isDerivedFrom(App::Plane::getClassTypeId())))
|
|
// Note: It is assumed that a Part has exactly 3 App::Plane objects at the root of the feature tree
|
|
return true;
|
|
|
|
if (edge && (pObj->getTypeId().isDerivedFrom(App::Line::getClassTypeId())))
|
|
return true;
|
|
|
|
if (pObj->getTypeId().isDerivedFrom(Part::Datum::getClassTypeId())) {
|
|
// Allow selecting Part::Datum features from the active Body
|
|
if (ActivePartObject == NULL)
|
|
return false;
|
|
if (!allowOtherBody && !ActivePartObject->hasFeature(pObj))
|
|
return false;
|
|
|
|
if (plane && (pObj->getTypeId().isDerivedFrom(PartDesign::Plane::getClassTypeId())))
|
|
return true;
|
|
if (edge && (pObj->getTypeId().isDerivedFrom(PartDesign::Line::getClassTypeId())))
|
|
return true;
|
|
if (point && (pObj->getTypeId().isDerivedFrom(PartDesign::Point::getClassTypeId())))
|
|
return true;
|
|
|
|
return false;
|
|
}
|
|
|
|
// Handle selection of geometry elements
|
|
if (!sSubName || sSubName[0] == '\0')
|
|
return false;
|
|
if (!allowOtherBody) {
|
|
if (support == NULL)
|
|
return false;
|
|
if (pObj != support)
|
|
return false;
|
|
}
|
|
std::string subName(sSubName);
|
|
if (edge && subName.size() > 4 && subName.substr(0,4) == "Edge") {
|
|
const Part::TopoShape &shape = static_cast<const Part::Feature*>(pObj)->Shape.getValue();
|
|
TopoDS_Shape sh = shape.getSubShape(subName.c_str());
|
|
const TopoDS_Edge& edge = TopoDS::Edge(sh);
|
|
if (!edge.IsNull()) {
|
|
if (planar) {
|
|
BRepAdaptor_Curve adapt(edge);
|
|
if (adapt.GetType() == GeomAbs_Line)
|
|
return true;
|
|
} else {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
if (plane && subName.size() > 4 && subName.substr(0,4) == "Face") {
|
|
const Part::TopoShape &shape = static_cast<const Part::Feature*>(pObj)->Shape.getValue();
|
|
TopoDS_Shape sh = shape.getSubShape(subName.c_str());
|
|
const TopoDS_Face& face = TopoDS::Face(sh);
|
|
if (!face.IsNull()) {
|
|
if (planar) {
|
|
BRepAdaptor_Surface adapt(face);
|
|
if (adapt.GetType() == GeomAbs_Plane)
|
|
return true;
|
|
} else {
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
if (point && subName.size() > 6 && subName.substr(0,6) == "Vertex") {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
namespace PartDesignGui
|
|
{
|
|
|
|
void getReferencedSelection(const App::DocumentObject* thisObj, const Gui::SelectionChanges& msg,
|
|
App::DocumentObject*& selObj, std::vector<std::string>& selSub)
|
|
{
|
|
if (strcmp(thisObj->getDocument()->getName(), msg.pDocName) != 0)
|
|
return;
|
|
|
|
selObj = thisObj->getDocument()->getObject(msg.pObjectName);
|
|
if (selObj == thisObj)
|
|
return;
|
|
std::string subname = msg.pSubName;
|
|
|
|
// Remove subname for planes and datum features
|
|
if (PartDesign::Feature::isDatum(selObj)) {
|
|
subname = "";
|
|
}
|
|
|
|
selSub = std::vector<std::string>(1,subname);
|
|
}
|
|
|
|
QString getRefStr(const App::DocumentObject* obj, const std::vector<std::string>& sub)
|
|
{
|
|
if (obj == NULL)
|
|
return QString::fromAscii("");
|
|
|
|
if (PartDesign::Feature::isDatum(obj))
|
|
return QString::fromAscii(obj->getNameInDocument());
|
|
else if (sub.size()>0)
|
|
return QString::fromAscii(obj->getNameInDocument()) + QString::fromAscii(":") +
|
|
QString::fromAscii(sub.front().c_str());
|
|
else
|
|
return QString();
|
|
}
|
|
|
|
std::string buildLinkSubPythonStr(const App::DocumentObject* obj, const std::vector<std::string>& subs)
|
|
{
|
|
if ( obj == NULL)
|
|
return "None";
|
|
|
|
std::string result("[");
|
|
|
|
for (const auto & sub : subs)
|
|
result += "\"" + sub + "\",";
|
|
result += "]";
|
|
|
|
return result;
|
|
}
|
|
|
|
std::string buildLinkSingleSubPythonStr(const App::DocumentObject* obj,
|
|
const std::vector<std::string>& subs)
|
|
{
|
|
if (obj == NULL)
|
|
return "None";
|
|
|
|
if (PartDesign::Feature::isDatum(obj))
|
|
return std::string("(App.ActiveDocument.") + obj->getNameInDocument() + ", [\"\"])";
|
|
else
|
|
return std::string("(App.ActiveDocument.") + obj->getNameInDocument() + ", [\"" + subs.front() + "\"])";
|
|
}
|
|
|
|
std::string buildLinkListPythonStr(const std::vector<App::DocumentObject*> & objs)
|
|
{
|
|
if ( objs.empty() ) {
|
|
return "None";
|
|
}
|
|
|
|
std::string result("[");
|
|
|
|
for (std::vector<App::DocumentObject*>::const_iterator o = objs.begin(); o != objs.end(); o++)
|
|
result += std::string("App.activeDocument().") + (*o)->getNameInDocument() + ",";
|
|
result += "]";
|
|
|
|
return result;
|
|
}
|
|
|
|
std::string buildLinkSubListPythonStr(const std::vector<App::DocumentObject*> & objs,
|
|
const std::vector<std::string>& subs)
|
|
{
|
|
if ( objs.empty() ) {
|
|
return "None";
|
|
}
|
|
|
|
std::string result("[");
|
|
|
|
assert (objs.size () == subs.size () );
|
|
|
|
for (size_t i=0, objs_sz=objs.size(); i < objs_sz; i++) {
|
|
if (objs[i] ) {
|
|
result += '(';
|
|
result += std::string("App.activeDocument().").append( objs[i]->getNameInDocument () );
|
|
result += ",\"";
|
|
result += subs[i];
|
|
result += "\"),";
|
|
}
|
|
}
|
|
|
|
result += "]";
|
|
|
|
return result;
|
|
}
|
|
}
|