allow shapes as coordinate system references

This commit is contained in:
Stefan Tröger
2015-05-19 07:08:27 +02:00
parent 3b89ce1c6a
commit bcf168748e

View File

@@ -32,6 +32,11 @@
#include <App/Part.h>
#include <gp_Pln.hxx>
#include <BRepBuilderAPI_MakeFace.hxx>
#include <TopoDS.hxx>
#include <GeomAbs_SurfaceType.hxx>
#include <BRepAdaptor_Surface.hxx>
#include <Geom_Plane.hxx>
#include <GeomAPI_ProjectPointOnSurf.hxx>
#include <QObject>
#ifndef M_PI
@@ -47,22 +52,32 @@ using namespace PartDesign;
#define POINT QObject::tr("DPOINT")
#define ANGLE QObject::tr("Angle")
#define CS QObject::tr("DCOORDINATESYSTEM")
#define DONE QObject::tr("Done")
std::map<std::multiset<QString>, std::set<QString> > CoordinateSystem::hints = std::map<std::multiset<QString>, std::set<QString> >();
void CoordinateSystem::initHints()
{
std::set<QString> DONE;
DONE.insert(QObject::tr("Done"));
std::set<QString> Done;
Done.insert(QObject::tr("Done"));
std::multiset<QString> key;
std::set<QString> value;
key.insert(POINT);
hints[key] = DONE;
hints[key] = Done;
key.clear(); value.clear();
key.insert(PLANE);
hints[key] = DONE;
value.insert(LINE);
value.insert(DONE);
hints[key] = value;
key.clear(); value.clear();
key.insert(PLANE);
key.insert(LINE);
value.insert(LINE);
value.insert(DONE);
hints[key] = value;
}
// ============================================================================
@@ -106,20 +121,20 @@ void CoordinateSystem::onChanged(const App::Property *prop)
//build the placement from the references
bool plane = false;
Base::Vector3d pl_base, pl_normal;
Base::Vector3d pl_base(0,0,0), pl_normal(0,0,1);
int count = 0;
for(App::DocumentObject* obj : refs) {
for(int i = 0; i < refs.size(); i++) {
if (obj->getTypeId().isDerivedFrom(PartDesign::Plane::getClassTypeId())) {
PartDesign::Plane* p = static_cast<PartDesign::Plane*>(obj);
if (refs[i]->getTypeId().isDerivedFrom(PartDesign::Plane::getClassTypeId())) {
PartDesign::Plane* p = static_cast<PartDesign::Plane*>(refs[i]);
if(!plane) {
pl_base = p->getBasePoint();
pl_normal = p->getNormal();
plane=true;
}
} else if (obj->getTypeId().isDerivedFrom(App::Plane::getClassTypeId())) {
PartDesign::Plane* p = static_cast<PartDesign::Plane*>(obj);
} else if (refs[i]->getTypeId().isDerivedFrom(App::Plane::getClassTypeId())) {
PartDesign::Plane* p = static_cast<PartDesign::Plane*>(refs[i]);
if(!plane) {
pl_base = Base::Vector3d(0,0,0);
if (strcmp(p->getNameInDocument(), App::Part::BaseplaneTypes[0]) == 0)
@@ -132,6 +147,50 @@ void CoordinateSystem::onChanged(const App::Property *prop)
plane=true;
}
}
else if (refs[i]->getTypeId().isDerivedFrom(Part::Feature::getClassTypeId())) {
Part::Feature* feature = static_cast<Part::Feature*>(refs[i]);
const TopoDS_Shape& sh = feature->Shape.getValue();
if (sh.IsNull())
return; // "PartDesign::Plane: Reference has NULL shape"
// Get subshape
TopoDS_Shape subshape = feature->Shape.getShape().getSubShape(subrefs[i].c_str());
if (subshape.IsNull())
return; // "PartDesign::Plane: Reference has NULL subshape";
if (subshape.ShapeType() == TopAbs_VERTEX) {/*
TopoDS_Vertex v = TopoDS::Vertex(subshape);
gp_Pnt p = BRep_Tool::Pnt(v);
if (p1 == NULL)
p1 = new Base::Vector3d(p.X(), p.Y(), p.Z());
else if (p2 == NULL)
p2 = new Base::Vector3d(p.X(), p.Y(), p.Z());
else
p3 = new Base::Vector3d(p.X(), p.Y(), p.Z());*/
} else if (subshape.ShapeType() == TopAbs_EDGE) { /*
TopoDS_Edge e = TopoDS::Edge(subshape);
BRepAdaptor_Curve adapt(e);
if (adapt.GetType() != GeomAbs_Line)
return; // Non-linear edge
line = new gp_Lin(adapt.Line());*/
} else if (subshape.ShapeType() == TopAbs_FACE) {
TopoDS_Face f = TopoDS::Face(subshape);
BRepAdaptor_Surface adapt(f);
if (adapt.GetType() == GeomAbs_Plane) {
// Ensure that the front and back of the plane corresponds with the face's idea of front and back
bool reverse = (f.Orientation() == TopAbs_REVERSED);
gp_Pln pl = adapt.Plane();
gp_Dir d = pl.Axis().Direction();
const gp_Pnt& p = pl.Location();
if (reverse) d.Reverse();
pl_base = Base::Vector3d(p.X(), p.Y(), p.Z());
pl_normal = Base::Vector3d(d.X(), d.Y(), d.Z());
plane = true;
} else {
return; // invalid surface type
}
}
}
count++;
};
@@ -142,12 +201,12 @@ void CoordinateSystem::onChanged(const App::Property *prop)
//add the offsets
Base::Vector3d o1;
plm.multVec(Offset.getValue()*Base::Vector3d(1,0,0), o1);
plm.getRotation().multVec(Base::Vector3d(1,0,0), o1);
Base::Vector3d o2;
plm.multVec(Offset2.getValue()*Base::Vector3d(0,1,0), o2);
plm.getRotation().multVec(Offset2.getValue()*Base::Vector3d(0,1,0), o2);
Base::Vector3d o3;
plm.multVec(Offset3.getValue()*Base::Vector3d(0,0,1), o3);
plm.move(o1+o2+o3);
plm.getRotation().multVec(Offset3.getValue()*Base::Vector3d(0,0,1), o3);
plm.move(Offset.getValue()*o1+Offset2.getValue()*o2+Offset3.getValue()*o3);
Placement.setValue(plm);
}