281 lines
9.6 KiB
C++
281 lines
9.6 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2015 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 <gp_Dir.hxx>
|
|
# include <gp_Lin.hxx>
|
|
# include <gp_Pln.hxx>
|
|
# include <gp_Vec.hxx>
|
|
# include <ElCLib.hxx>
|
|
# include <ElSLib.hxx>
|
|
# include <Geom_BezierCurve.hxx>
|
|
# include <Geom_BezierSurface.hxx>
|
|
# include <Geom_BSplineCurve.hxx>
|
|
# include <Geom_BSplineSurface.hxx>
|
|
# include <Precision.hxx>
|
|
# include <TColgp_Array2OfPnt.hxx>
|
|
# include <TopoDS.hxx>
|
|
# include <TopoDS_Shape.hxx>
|
|
# include <TopoDS_Edge.hxx>
|
|
# include <TopoDS_Face.hxx>
|
|
#endif
|
|
|
|
#include "FemTools.h"
|
|
|
|
Base::Vector3d Fem::Tools::getDirectionFromShape(const TopoDS_Shape& shape)
|
|
{
|
|
gp_XYZ dir(0, 0, 0);
|
|
|
|
// "Direction must be a planar face or linear edge"
|
|
//
|
|
if (shape.ShapeType() == TopAbs_FACE) {
|
|
if (isPlanar(TopoDS::Face(shape))) {
|
|
dir = getDirection(TopoDS::Face(shape));
|
|
}
|
|
}
|
|
else if (shape.ShapeType() == TopAbs_EDGE) {
|
|
if (isLinear(TopoDS::Edge(shape))) {
|
|
dir = getDirection(TopoDS::Edge(shape));
|
|
}
|
|
}
|
|
|
|
Base::Vector3d the_direction(dir.X(), dir.Y(), dir.Z());
|
|
return the_direction;
|
|
}
|
|
|
|
bool Fem::Tools::isPlanar(const TopoDS_Face& face)
|
|
{
|
|
BRepAdaptor_Surface surface(face);
|
|
if (surface.GetType() == GeomAbs_Plane) {
|
|
return true;
|
|
}
|
|
else if (surface.GetType() == GeomAbs_BSplineSurface) {
|
|
Handle_Geom_BSplineSurface spline = surface.BSpline();
|
|
try {
|
|
TColgp_Array2OfPnt poles(1,spline->NbUPoles(),1,spline->NbVPoles());
|
|
spline->Poles(poles);
|
|
|
|
// get the plane from three control points
|
|
gp_Pnt p1 = poles(poles.LowerRow(),poles.LowerCol());
|
|
gp_Pnt p2 = poles(poles.UpperRow(),poles.LowerCol());
|
|
gp_Pnt p3 = poles(poles.LowerRow(),poles.UpperCol());
|
|
gp_Vec vec1(p1, p2);
|
|
gp_Vec vec2(p1, p3);
|
|
gp_Vec vec3 = vec1.Crossed(vec2);
|
|
gp_Pln plane(p1, gp_Dir(vec3));
|
|
|
|
for (int i=poles.LowerRow(); i<=poles.UpperRow(); i++) {
|
|
for (int j=poles.LowerCol(); j<poles.UpperCol(); j++) {
|
|
// are control points coplanar?
|
|
const gp_Pnt& pole = poles(i,j);
|
|
Standard_Real dist = plane.Distance(pole);
|
|
if (dist > Precision::Confusion()) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
catch (Standard_Failure) {
|
|
return false;
|
|
}
|
|
}
|
|
else if (surface.GetType() == GeomAbs_BezierSurface) {
|
|
Handle_Geom_BezierSurface bezier = surface.Bezier();
|
|
try {
|
|
TColgp_Array2OfPnt poles(1,bezier->NbUPoles(),1,bezier->NbVPoles());
|
|
bezier->Poles(poles);
|
|
|
|
// get the plane from three control points
|
|
gp_Pnt p1 = poles(poles.LowerRow(),poles.LowerCol());
|
|
gp_Pnt p2 = poles(poles.UpperRow(),poles.LowerCol());
|
|
gp_Pnt p3 = poles(poles.LowerRow(),poles.UpperCol());
|
|
gp_Vec vec1(p1, p2);
|
|
gp_Vec vec2(p1, p3);
|
|
gp_Vec vec3 = vec1.Crossed(vec2);
|
|
gp_Pln plane(p1, gp_Dir(vec3));
|
|
|
|
for (int i=poles.LowerRow(); i<=poles.UpperRow(); i++) {
|
|
for (int j=poles.LowerCol(); j<poles.UpperCol(); j++) {
|
|
// are control points coplanar?
|
|
const gp_Pnt& pole = poles(i,j);
|
|
Standard_Real dist = plane.Distance(pole);
|
|
if (dist > Precision::Confusion()) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
catch (Standard_Failure) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
gp_XYZ Fem::Tools::getDirection(const TopoDS_Face& face)
|
|
{
|
|
gp_XYZ dir(0, 0, 0);
|
|
|
|
BRepAdaptor_Surface surface(face);
|
|
if (surface.GetType() == GeomAbs_Plane) {
|
|
dir = surface.Plane().Axis().Direction().XYZ();
|
|
}
|
|
else if (surface.GetType() == GeomAbs_BSplineSurface) {
|
|
Handle_Geom_BSplineSurface spline = surface.BSpline();
|
|
try {
|
|
TColgp_Array2OfPnt poles(1,spline->NbUPoles(),1,spline->NbVPoles());
|
|
spline->Poles(poles);
|
|
|
|
// get the plane from three control points
|
|
gp_Pnt p1 = poles(poles.LowerRow(),poles.LowerCol());
|
|
gp_Pnt p2 = poles(poles.UpperRow(),poles.LowerCol());
|
|
gp_Pnt p3 = poles(poles.LowerRow(),poles.UpperCol());
|
|
gp_Vec vec1(p1, p2);
|
|
gp_Vec vec2(p1, p3);
|
|
gp_Vec vec3 = vec1.Crossed(vec2);
|
|
gp_Pln plane(p1, gp_Dir(vec3));
|
|
dir = plane.Axis().Direction().XYZ();
|
|
}
|
|
catch (Standard_Failure) {
|
|
}
|
|
}
|
|
else if (surface.GetType() == GeomAbs_BezierSurface) {
|
|
Handle_Geom_BezierSurface bezier = surface.Bezier();
|
|
try {
|
|
TColgp_Array2OfPnt poles(1,bezier->NbUPoles(),1,bezier->NbVPoles());
|
|
bezier->Poles(poles);
|
|
|
|
// get the plane from three control points
|
|
gp_Pnt p1 = poles(poles.LowerRow(),poles.LowerCol());
|
|
gp_Pnt p2 = poles(poles.UpperRow(),poles.LowerCol());
|
|
gp_Pnt p3 = poles(poles.LowerRow(),poles.UpperCol());
|
|
gp_Vec vec1(p1, p2);
|
|
gp_Vec vec2(p1, p3);
|
|
gp_Vec vec3 = vec1.Crossed(vec2);
|
|
gp_Pln plane(p1, gp_Dir(vec3));
|
|
dir = plane.Axis().Direction().XYZ();
|
|
}
|
|
catch (Standard_Failure) {
|
|
}
|
|
}
|
|
|
|
return dir;
|
|
}
|
|
|
|
bool Fem::Tools::isLinear(const TopoDS_Edge& edge)
|
|
{
|
|
BRepAdaptor_Curve curve(edge);
|
|
if (curve.GetType() == GeomAbs_Line) {
|
|
return true;
|
|
}
|
|
else if (curve.GetType() == GeomAbs_BSplineCurve) {
|
|
Handle_Geom_BSplineCurve spline = curve.BSpline();
|
|
try {
|
|
gp_Pnt s1 = spline->Pole(1);
|
|
gp_Pnt sn = spline->Pole(spline->NbPoles());
|
|
gp_Vec vec(s1, sn);
|
|
gp_Lin line(s1, gp_Dir(vec));
|
|
|
|
for (int i=2; i<spline->NbPoles(); i++) {
|
|
// are control points collinear?
|
|
Standard_Real dist = line.Distance(spline->Pole(i));
|
|
if (dist > Precision::Confusion()) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
catch (Standard_Failure) {
|
|
return false;
|
|
}
|
|
}
|
|
else if (curve.GetType() == GeomAbs_BezierCurve) {
|
|
Handle_Geom_BezierCurve bezier = curve.Bezier();
|
|
try {
|
|
gp_Pnt s1 = bezier->Pole(1);
|
|
gp_Pnt sn = bezier->Pole(bezier->NbPoles());
|
|
gp_Vec vec(s1, sn);
|
|
gp_Lin line(s1, gp_Dir(vec));
|
|
|
|
for (int i=2; i<bezier->NbPoles(); i++) {
|
|
// are control points collinear?
|
|
Standard_Real dist = line.Distance(bezier->Pole(i));
|
|
if (dist > Precision::Confusion()) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
catch (Standard_Failure) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
gp_XYZ Fem::Tools::getDirection(const TopoDS_Edge& edge)
|
|
{
|
|
gp_XYZ dir(0, 0, 0);
|
|
|
|
BRepAdaptor_Curve curve(edge);
|
|
if (curve.GetType() == GeomAbs_Line) {
|
|
dir = curve.Line().Direction().XYZ();
|
|
}
|
|
else if (curve.GetType() == GeomAbs_BSplineCurve) {
|
|
Handle_Geom_BSplineCurve spline = curve.BSpline();
|
|
try {
|
|
gp_Pnt s1 = spline->Pole(1);
|
|
gp_Pnt sn = spline->Pole(spline->NbPoles());
|
|
gp_Vec vec(s1, sn);
|
|
gp_Lin line(s1, gp_Dir(vec));
|
|
dir = line.Direction().XYZ();
|
|
}
|
|
catch (Standard_Failure) {
|
|
}
|
|
}
|
|
else if (curve.GetType() == GeomAbs_BezierCurve) {
|
|
Handle_Geom_BezierCurve bezier = curve.Bezier();
|
|
try {
|
|
gp_Pnt s1 = bezier->Pole(1);
|
|
gp_Pnt sn = bezier->Pole(bezier->NbPoles());
|
|
gp_Vec vec(s1, sn);
|
|
gp_Lin line(s1, gp_Dir(vec));
|
|
dir = line.Direction().XYZ();
|
|
}
|
|
catch (Standard_Failure) {
|
|
}
|
|
}
|
|
|
|
return dir;
|
|
}
|