279 lines
10 KiB
C++
279 lines
10 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2013 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 <BRepAdaptor_Curve.hxx>
|
|
# include <BRepAdaptor_Surface.hxx>
|
|
# include <GeomAbs_CurveType.hxx>
|
|
# include <GeomAbs_SurfaceType.hxx>
|
|
# include <Geom_BezierCurve.hxx>
|
|
# include <Geom_BSplineCurve.hxx>
|
|
# include <Geom_BezierSurface.hxx>
|
|
# include <Geom_BSplineSurface.hxx>
|
|
# include <gp_Pnt.hxx>
|
|
# include <TopoDS.hxx>
|
|
# include <TopoDS_Edge.hxx>
|
|
# include <TopoDS_Wire.hxx>
|
|
# include <TopoDS_Face.hxx>
|
|
# include <TopoDS_Shape.hxx>
|
|
# include <TopExp_Explorer.hxx>
|
|
# include <Inventor/nodes/SoCoordinate3.h>
|
|
# include <Inventor/nodes/SoSeparator.h>
|
|
# include <Inventor/nodes/SoSwitch.h>
|
|
#endif
|
|
|
|
|
|
#include <App/PropertyStandard.h>
|
|
#include <Mod/Part/App/PartFeature.h>
|
|
#include "SoFCShapeObject.h"
|
|
#include "ViewProviderSpline.h"
|
|
|
|
|
|
using namespace PartGui;
|
|
|
|
|
|
PROPERTY_SOURCE(PartGui::ViewProviderSpline, PartGui::ViewProviderPartExt)
|
|
|
|
ViewProviderSpline::ViewProviderSpline()
|
|
: pcControlPoints(0)
|
|
{
|
|
ADD_PROPERTY(ControlPoints,(false));
|
|
}
|
|
|
|
ViewProviderSpline::~ViewProviderSpline()
|
|
{
|
|
}
|
|
|
|
void ViewProviderSpline::updateData(const App::Property* prop)
|
|
{
|
|
ViewProviderPartExt::updateData(prop);
|
|
if (prop->getTypeId() == Part::PropertyPartShape::getClassTypeId() && strcmp(prop->getName(), "Shape") == 0) {
|
|
// update control points if there
|
|
if (pcControlPoints) {
|
|
pcControlPoints->removeAllChildren();
|
|
showControlPoints(this->ControlPoints.getValue(), prop);
|
|
}
|
|
}
|
|
}
|
|
|
|
void ViewProviderSpline::onChanged(const App::Property* prop)
|
|
{
|
|
if (prop == &ControlPoints) {
|
|
App::DocumentObject* obj = this->pcObject;
|
|
App::Property* shape = obj->getPropertyByName("Shape");
|
|
showControlPoints(ControlPoints.getValue(), shape);
|
|
}
|
|
else {
|
|
ViewProviderPartExt::onChanged(prop);
|
|
}
|
|
}
|
|
|
|
void ViewProviderSpline::showControlPoints(bool show, const App::Property* prop)
|
|
{
|
|
if (!pcControlPoints && show) {
|
|
pcControlPoints = new SoSwitch();
|
|
pcRoot->addChild(pcControlPoints);
|
|
}
|
|
|
|
if (pcControlPoints) {
|
|
pcControlPoints->whichChild = (show ? SO_SWITCH_ALL : SO_SWITCH_NONE);
|
|
}
|
|
|
|
if (!show || !pcControlPoints || pcControlPoints->getNumChildren() > 0)
|
|
return;
|
|
|
|
// ask for the property we are interested in
|
|
if (prop && prop->getTypeId() == Part::PropertyPartShape::getClassTypeId()) {
|
|
const TopoDS_Shape& shape = static_cast<const Part::PropertyPartShape*>(prop)->getValue();
|
|
if (shape.IsNull())
|
|
return; // empty shape
|
|
|
|
for (TopExp_Explorer xp(shape, TopAbs_SHELL); xp.More(); xp.Next()) {
|
|
const TopoDS_Shell& shell = TopoDS::Shell(xp.Current());
|
|
for (TopExp_Explorer xp2(xp.Current(), TopAbs_FACE); xp2.More(); xp2.Next()) {
|
|
const TopoDS_Face& face = TopoDS::Face(xp2.Current());
|
|
showControlPointsOfFace(face);
|
|
}
|
|
}
|
|
for (TopExp_Explorer xp(shape, TopAbs_FACE, TopAbs_SHELL); xp.More(); xp.Next()) {
|
|
const TopoDS_Face& face = TopoDS::Face(xp.Current());
|
|
showControlPointsOfFace(face);
|
|
}
|
|
for (TopExp_Explorer xp(shape, TopAbs_WIRE, TopAbs_FACE); xp.More(); xp.Next()) {
|
|
const TopoDS_Wire& wire = TopoDS::Wire(xp.Current());
|
|
for (TopExp_Explorer xp2(xp.Current(), TopAbs_EDGE); xp2.More(); xp2.Next()) {
|
|
const TopoDS_Edge& edge = TopoDS::Edge(xp2.Current());
|
|
showControlPointsOfEdge(edge);
|
|
}
|
|
}
|
|
for (TopExp_Explorer xp(shape, TopAbs_EDGE, TopAbs_WIRE); xp.More(); xp.Next()) {
|
|
const TopoDS_Edge& edge = TopoDS::Edge(xp.Current());
|
|
showControlPointsOfEdge(edge);
|
|
}
|
|
}
|
|
}
|
|
|
|
void ViewProviderSpline::showControlPointsOfEdge(const TopoDS_Edge& edge)
|
|
{
|
|
std::list<gp_Pnt> poles, knots;
|
|
Standard_Integer nCt=0;
|
|
BRepAdaptor_Curve curve(edge);
|
|
switch (curve.GetType())
|
|
{
|
|
case GeomAbs_BezierCurve:
|
|
{
|
|
Handle(Geom_BezierCurve) hBezier = curve.Bezier();
|
|
nCt = hBezier->NbPoles();
|
|
for (Standard_Integer i = 1; i <= nCt; i++)
|
|
poles.push_back(hBezier->Pole(i));
|
|
if (hBezier->IsClosed()) {
|
|
nCt++;
|
|
poles.push_back(hBezier->Pole(1));
|
|
}
|
|
} break;
|
|
case GeomAbs_BSplineCurve:
|
|
{
|
|
Handle(Geom_BSplineCurve) hBSpline = curve.BSpline();
|
|
nCt = hBSpline->NbPoles();
|
|
for (Standard_Integer i = 1; i <= nCt; i++)
|
|
poles.push_back(hBSpline->Pole(i));
|
|
if (hBSpline->IsClosed()) {
|
|
nCt++;
|
|
poles.push_back(hBSpline->Pole(1));
|
|
}
|
|
for (Standard_Integer i = hBSpline->FirstUKnotIndex()+1; i <= hBSpline->LastUKnotIndex()-1; i++)
|
|
knots.push_back(hBSpline->Value(hBSpline->Knot(i)));
|
|
} break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
if (poles.empty())
|
|
return; // nothing to do
|
|
|
|
SoCoordinate3 * coords = new SoCoordinate3;
|
|
coords->point.setNum(nCt + knots.size());
|
|
|
|
int index=0;
|
|
SbVec3f* verts = coords->point.startEditing();
|
|
for (std::list<gp_Pnt>::iterator p = poles.begin(); p != poles.end(); ++p) {
|
|
verts[index++].setValue((float)p->X(), (float)p->Y(), (float)p->Z());
|
|
}
|
|
for (std::list<gp_Pnt>::iterator k = knots.begin(); k != knots.end(); ++k) {
|
|
verts[index++].setValue((float)k->X(), (float)k->Y(), (float)k->Z());
|
|
}
|
|
coords->point.finishEditing();
|
|
|
|
|
|
SoFCControlPoints* control = new SoFCControlPoints();
|
|
control->numPolesU = nCt;
|
|
control->numPolesV = 1;
|
|
|
|
SoSeparator* nodes = new SoSeparator();
|
|
nodes->addChild(coords);
|
|
nodes->addChild(control);
|
|
|
|
pcControlPoints->addChild(nodes);
|
|
}
|
|
|
|
void ViewProviderSpline::showControlPointsOfFace(const TopoDS_Face& face)
|
|
{
|
|
std::list<gp_Pnt> knots;
|
|
std::vector<std::vector<gp_Pnt> > poles;
|
|
Standard_Integer nCtU=0, nCtV=0;
|
|
BRepAdaptor_Surface surface(face);
|
|
|
|
BRepAdaptor_Surface clSurface(face);
|
|
switch (clSurface.GetType())
|
|
{
|
|
case GeomAbs_BezierSurface:
|
|
{
|
|
Handle(Geom_BezierSurface) hBezier = surface.Bezier();
|
|
nCtU = hBezier->NbUPoles();
|
|
nCtV = hBezier->NbVPoles();
|
|
poles.resize(nCtU);
|
|
for (Standard_Integer u = 1; u <= nCtU; u++) {
|
|
poles[u-1].resize(nCtV);
|
|
for (Standard_Integer v = 1; v <= nCtV; v++)
|
|
poles[u-1][v-1] = hBezier->Pole(u, v);
|
|
}
|
|
} break;
|
|
case GeomAbs_BSplineSurface:
|
|
{
|
|
Handle(Geom_BSplineSurface) hBSpline = surface.BSpline();
|
|
nCtU = hBSpline->NbUPoles();
|
|
nCtV = hBSpline->NbVPoles();
|
|
poles.resize(nCtU);
|
|
for (Standard_Integer u = 1; u <= nCtU; u++) {
|
|
poles[u-1].resize(nCtV);
|
|
for (Standard_Integer v = 1; v <= nCtV; v++)
|
|
poles[u-1][v-1] = hBSpline->Pole(u, v);
|
|
}
|
|
|
|
//Standard_Integer nKnU = hBSpline->NbUKnots();
|
|
//Standard_Integer nKnV = hBSpline->NbVKnots();
|
|
for (Standard_Integer u = 1; u <= hBSpline->NbUKnots(); u++) {
|
|
for (Standard_Integer v = 1; v <= hBSpline->NbVKnots(); v++)
|
|
knots.push_back(hBSpline->Value(hBSpline->UKnot(u), hBSpline->VKnot(v)));
|
|
}
|
|
} break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
if (poles.empty())
|
|
return; // nothing to do
|
|
|
|
SoCoordinate3 * coords = new SoCoordinate3;
|
|
coords->point.setNum(nCtU * nCtV + knots.size());
|
|
|
|
int index=0;
|
|
SbVec3f* verts = coords->point.startEditing();
|
|
for (std::vector<std::vector<gp_Pnt> >::iterator u = poles.begin(); u != poles.end(); ++u) {
|
|
for (std::vector<gp_Pnt>::iterator v = u->begin(); v != u->end(); ++v) {
|
|
verts[index++].setValue((float)v->X(), (float)v->Y(), (float)v->Z());
|
|
}
|
|
}
|
|
for (std::list<gp_Pnt>::iterator k = knots.begin(); k != knots.end(); ++k) {
|
|
verts[index++].setValue((float)k->X(), (float)k->Y(), (float)k->Z());
|
|
}
|
|
coords->point.finishEditing();
|
|
|
|
|
|
SoFCControlPoints* control = new SoFCControlPoints();
|
|
control->numPolesU = nCtU;
|
|
control->numPolesV = nCtV;
|
|
|
|
//if (knots.size() > 0) {
|
|
// control->numKnotsU = nKnU;
|
|
// control->numKnotsV = nKnV;
|
|
//}
|
|
|
|
SoSeparator* nodes = new SoSeparator();
|
|
nodes->addChild(coords);
|
|
nodes->addChild(control);
|
|
|
|
pcControlPoints->addChild(nodes);
|
|
}
|