+ add Thomas Anderson's remove splitter algorithm

git-svn-id: https://free-cad.svn.sourceforge.net/svnroot/free-cad/trunk@5228 e8eeb9e2-ec13-0410-a4a9-efa5cf37419d
This commit is contained in:
wmayer 2011-12-07 15:16:57 +00:00
parent c618af29dc
commit 271291a2d0
9 changed files with 729 additions and 11 deletions

View File

@ -1237,17 +1237,6 @@ static std::list<TopoDS_Edge> sort_Edges(double tol3d, const std::vector<TopoDS_
if ((pEI == edge_points.end()) || (last.SquareDistance(first) <= tol3d)) {
// no adjacent edge found or polyline is closed
return sorted;
/*
rclBorders.push_back(std::vector<TopoDS_Edge>(sorted.begin(), sorted.end()));
sorted.clear();
if (!edge_points.empty()) {
// new wire
first = edge_points.front()->v1;
last = edge_points.front()->v2;
sorted.push_back(edge_points.front().edge);
edge_points.erase(edge_points.begin());
}*/
}
}

View File

@ -227,6 +227,8 @@ SET(Part_SRCS
TopoShape.h
edgecluster.cpp
edgecluster.h
modelRefine.cpp
modelRefine.h
)
SET(Part_Scripts

View File

@ -127,6 +127,7 @@ libPart_la_SOURCES=\
Geometry.cpp \
ImportIges.cpp \
ImportStep.cpp \
modelRefine.cpp \
CustomFeature.cpp \
PartFeature.cpp \
PartFeatureReference.cpp \
@ -179,6 +180,7 @@ include_HEADERS=\
Geometry.h \
ImportIges.h \
ImportStep.h \
modelRefine.h \
PartFeature.h \
PartFeatureReference.h \
CustomFeature.h \

View File

@ -44,6 +44,7 @@
# include <BRepBuilderAPI_MakeEdge.hxx>
# include <BRepBuilderAPI_MakeFace.hxx>
# include <BRepBuilderAPI_MakePolygon.hxx>
# include <BRepBuilderAPI_MakeSolid.hxx>
# include <BRepBuilderAPI_MakeVertex.hxx>
# include <BRepBuilderAPI_MakeWire.hxx>
# include <BRepBuilderAPI_NurbsConvert.hxx>
@ -97,6 +98,7 @@
# include <TopoDS.hxx>
# include <TopoDS_Compound.hxx>
# include <TopoDS_Iterator.hxx>
# include <TopoDS_Solid.hxx>
# include <TopoDS_Vertex.hxx>
# include <TopExp.hxx>
# include <TopExp_Explorer.hxx>
@ -152,6 +154,7 @@
#include "TopoShapeEdgePy.h"
#include "TopoShapeVertexPy.h"
#include "ProgressIndicator.h"
#include "modelRefine.h"
using namespace Part;
@ -1717,6 +1720,36 @@ bool TopoShape::removeInternalWires(double minArea)
return ok;
}
void TopoShape::removeSplitter()
{
if (_Shape.IsNull())
Standard_Failure::Raise("Cannot remove splitter from empty shape");
if (_Shape.ShapeType() == TopAbs_SOLID) {
TopoDS_Solid& solid = TopoDS::Solid(_Shape);
ModelRefine::FaceUniter uniter(solid);
if (uniter.process()) {
TopoDS_Solid solidMod;
if (!uniter.getSolid(solidMod))
Standard_Failure::Raise("Getting solid failed");
_Shape = solidMod;
}
else {
Standard_Failure::Raise("Removing splitter failed");
}
}
else if (_Shape.ShapeType() == TopAbs_SHELL) {
TopoDS_Shell& shell = TopoDS::Shell(_Shape);
ModelRefine::FaceUniter uniter(shell);
if (uniter.process()) {
_Shape = uniter.getShell();
}
else {
Standard_Failure::Raise("Removing splitter failed");
}
}
}
namespace Part {
struct MeshVertex
{

View File

@ -219,6 +219,7 @@ public:
void sewShape();
bool fix(double, double, double);
bool removeInternalWires(double);
void removeSplitter();
//@}
/** @name Getting basic geometric entities */

View File

@ -283,6 +283,11 @@ The parameter is a list of shapes.</UserDocu>
<UserDocu>Checks whether a point is inside or outside a given shape</UserDocu>
</Documentation>
</Methode>
<Methode Name="removeSplitter" Const="true">
<Documentation>
<UserDocu>Removes redundant edges from the B-REP model</UserDocu>
</Documentation>
</Methode>
<!--
<Attribute Name="Location" ReadOnly="false">
<Documentation>

View File

@ -1244,6 +1244,23 @@ PyObject* TopoShapePy::isInside(PyObject *args)
}
}
PyObject* TopoShapePy::removeSplitter(PyObject *args)
{
if (!PyArg_ParseTuple(args, ""))
return NULL;
try {
// Remove redundant splitter
this->getTopoShapePtr()->removeSplitter();
Py_Return;
}
catch (Standard_Failure) {
Handle_Standard_Failure e = Standard_Failure::Caught();
PyErr_SetString(PyExc_Exception, e->GetMessageString());
return NULL;
}
}
#if 0 // see ComplexGeoDataPy::Matrix which does the same
Py::Object TopoShapePy::getLocation(void) const
{

View File

@ -0,0 +1,505 @@
/***************************************************************************
* Copyright (c) 2011 Thomas Anderson <blobfish[at]gmx.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"
#include <algorithm>
#include <Geom_Surface.hxx>
#include <GeomAdaptor_Surface.hxx>
#include <Geom_Plane.hxx>
#include <Geom_CylindricalSurface.hxx>
#include <gp_Pln.hxx>
#include <gp_Cylinder.hxx>
#include <TopoDS_Shape.hxx>
#include <TopoDS.hxx>
#include <TopExp_Explorer.hxx>
#include <BRep_Tool.hxx>
#include <BRepLib_MakeWire.hxx>
#include <BRepLib_FuseEdges.hxx>
#include <BRepBuilderAPI_MakeFace.hxx>
#include <BRepBuilderAPI_Sewing.hxx>
#include <BRepBuilderAPI_MakeSolid.hxx>
#include <ShapeBuild_ReShape.hxx>
#include "modelRefine.h"
using namespace ModelRefine;
//this following struct and function was stole from freecad AppPartPy.cpp
namespace ModelRefine {
struct EdgePoints {
gp_Pnt v1, v2;
TopoDS_Edge edge;
};
static std::list<TopoDS_Edge> sort_Edges(double tol3d, const std::vector<TopoDS_Edge>& edges)
{
tol3d = tol3d * tol3d;
std::list<EdgePoints> edge_points;
TopExp_Explorer xp;
for (std::vector<TopoDS_Edge>::const_iterator it = edges.begin(); it != edges.end(); ++it) {
EdgePoints ep;
xp.Init(*it,TopAbs_VERTEX);
ep.v1 = BRep_Tool::Pnt(TopoDS::Vertex(xp.Current()));
xp.Next();
ep.v2 = BRep_Tool::Pnt(TopoDS::Vertex(xp.Current()));
ep.edge = *it;
edge_points.push_back(ep);
}
if (edge_points.empty())
return std::list<TopoDS_Edge>();
std::list<TopoDS_Edge> sorted;
gp_Pnt first, last;
first = edge_points.front().v1;
last = edge_points.front().v2;
sorted.push_back(edge_points.front().edge);
edge_points.erase(edge_points.begin());
while (!edge_points.empty()) {
// search for adjacent edge
std::list<EdgePoints>::iterator pEI;
for (pEI = edge_points.begin(); pEI != edge_points.end(); ++pEI) {
if (pEI->v1.SquareDistance(last) <= tol3d) {
last = pEI->v2;
sorted.push_back(pEI->edge);
edge_points.erase(pEI);
break;
}
else if (pEI->v2.SquareDistance(first) <= tol3d) {
first = pEI->v1;
sorted.push_front(pEI->edge);
edge_points.erase(pEI);
break;
}
else if (pEI->v2.SquareDistance(last) <= tol3d) {
last = pEI->v1;
sorted.push_back(pEI->edge);
edge_points.erase(pEI);
break;
}
else if (pEI->v1.SquareDistance(first) <= tol3d) {
first = pEI->v2;
sorted.push_front(pEI->edge);
edge_points.erase(pEI);
break;
}
}
if ((pEI == edge_points.end()) || (last.SquareDistance(first) <= tol3d)) {
// no adjacent edge found or polyline is closed
return sorted;
}
}
return sorted;
}
}
//end stolen freecad.
void ModelRefine::getFaceEdges(TopoDS_Shape face, std::vector<TopoDS_Edge> &edges)
{
TopExp_Explorer it;
for (it.Init(face, TopAbs_EDGE); it.More(); it.Next())
edges.push_back(TopoDS::Edge(it.Current()));
}
bool ModelRefine::hasSharedEdges(const TopoDS_Shape &faceOne, const TopoDS_Shape &faceTwo)
{
std::vector<TopoDS_Edge> faceOneEdges, faceTwoEdges;
getFaceEdges(faceOne, faceOneEdges);
getFaceEdges(faceTwo, faceTwoEdges);
std::vector<TopoDS_Edge>::iterator itOne, itTwo;
for (itOne = faceOneEdges.begin(); itOne != faceOneEdges.end(); ++itOne)
{
for (itTwo = faceTwoEdges.begin(); itTwo != faceTwoEdges.end(); ++itTwo)
{
if ((*itOne).IsSame(*itTwo))
return true;
}
}
return false;
}
TopoDS_Wire ModelRefine::facesBoundary(const FaceVectorType &faces)
{
//this finds the perimeter of equal, adjacent faces.
std::list<TopoDS_Edge> edges;
FaceVectorType::const_iterator faceIt;
for (faceIt = faces.begin(); faceIt != faces.end(); ++faceIt)
{
std::vector<TopoDS_Edge> faceEdges;
std::vector<TopoDS_Edge>::iterator faceEdgesIt;
getFaceEdges(*faceIt, faceEdges);
for (faceEdgesIt = faceEdges.begin(); faceEdgesIt != faceEdges.end(); ++faceEdgesIt)
{
bool foundSignal(false);
std::list<TopoDS_Edge>::iterator edgesIt;
for (edgesIt = edges.begin(); edgesIt != edges.end(); ++edgesIt)
{
if ((*edgesIt).IsSame(*faceEdgesIt))
{
edgesIt = edges.erase(edgesIt);
foundSignal = true;
break;
}
}
if (!foundSignal)
edges.push_back(*faceEdgesIt);
}
}
//put into vector for stolen function.
std::vector<TopoDS_Edge> edgeVector;
edgeVector.reserve(edges.size());
std::copy(edges.begin(), edges.end(), back_inserter(edgeVector));
std::list<TopoDS_Edge> edgeSorted;
edgeSorted = sort_Edges(Precision::Confusion(), edgeVector);
BRepLib_MakeWire wireMaker;
std::list<TopoDS_Edge>::iterator sortedIt;
for (sortedIt = edgeSorted.begin(); sortedIt != edgeSorted.end(); ++sortedIt)
wireMaker.Add(*sortedIt);
return wireMaker.Wire();
}
TopoDS_Shell ModelRefine::removeFaces(const TopoDS_Shell &shell, const FaceVectorType &faces)
{
ShapeBuild_ReShape rebuilder;
FaceVectorType::const_iterator it;
for (it = faces.begin(); it != faces.end(); ++it)
rebuilder.Remove(*it);
return TopoDS::Shell(rebuilder.Apply(shell));
}
////////////////////////////////////////////////////////////////////////////////////////////
void FaceTypeSplitter::addShell(const TopoDS_Shell &shellIn)
{
shell = shellIn;
}
void FaceTypeSplitter::registerType(const GeomAbs_SurfaceType &type)
{
typeMap.insert(SplitMapType::value_type(type, FaceVectorType()));
}
bool FaceTypeSplitter::hasType(const GeomAbs_SurfaceType &type) const
{
return typeMap.find(type) != typeMap.end();
}
void FaceTypeSplitter::split()
{
TopExp_Explorer shellIt;
for (shellIt.Init(shell, TopAbs_FACE); shellIt.More(); shellIt.Next())
{
GeomAbs_SurfaceType currentType = FaceTypedBase::getFaceType(shellIt.Current());
SplitMapType::iterator mapIt = typeMap.find(currentType);
if (mapIt == typeMap.end())
continue;
(*mapIt).second.push_back(TopoDS::Face(shellIt.Current()));
}
}
FaceVectorType FaceTypeSplitter::getTypedFaceVector(const GeomAbs_SurfaceType &type) const
{
if (this->hasType(type))
return (*(typeMap.find(type))).second;
//error here.
return FaceVectorType();
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////
void FaceAdjacencySplitter::split(const FaceVectorType &facesIn)
{
//the reserve call guarantees the vector will never get "pushed back" in the
//recursiveFind calls, thus invalidating the iterators. We can be sure of this as any one
//matched set can't be bigger than the set passed in. if we have seg faults, we will
//want to turn this tempFaces vector back into a std::list ensuring valid iterators
//at the expense of std::find speed.
FaceVectorType tempFaces;
tempFaces.reserve(facesIn.size() + 1);
FaceVectorType::const_iterator it;
for (it = facesIn.begin(); it != facesIn.end(); ++it)
{
//skip already processed shapes.
if (hasBeenMapped(*it))
continue;
tempFaces.clear();
tempFaces.push_back(*it);
recursiveFind(tempFaces, facesIn);
if (tempFaces.size() > 1)
{
FaceVectorType temp;
temp.reserve(tempFaces.size()+10);
std::copy(tempFaces.begin(), tempFaces.end(), back_inserter(temp));
adjacencyArray.push_back(temp);
}
}
}
void FaceAdjacencySplitter::recursiveFind(FaceVectorType &tempFaces, const FaceVectorType &facesIn)
{
FaceVectorType::iterator tempIt;
FaceVectorType::const_iterator faceIt;
for (tempIt = tempFaces.begin(); tempIt != tempFaces.end(); ++tempIt)
{
for(faceIt = facesIn.begin(); faceIt != facesIn.end(); ++faceIt)
{
if ((*tempIt).IsSame(*faceIt))
continue;
if (std::find(tempFaces.begin(), tempFaces.end(), *faceIt) != tempFaces.end())
continue;
if (hasSharedEdges(*tempIt, *faceIt))
{
tempFaces.push_back(*faceIt);
recursiveFind(tempFaces, facesIn);
}
}
}
}
bool FaceAdjacencySplitter::hasBeenMapped(const TopoDS_Face &shape)
{
for (std::vector<FaceVectorType>::iterator it(adjacencyArray.begin()); it != adjacencyArray.end(); ++it)
{
if (std::find((*it).begin(), (*it).end(), shape) != (*it).end())
return true;
}
return false;
}
//////////////////////////////////////////////////////////////////////////////////////////////////////
void FaceEqualitySplitter::split(const FaceVectorType &faces, FaceTypedBase *object)
{
std::vector<FaceVectorType> tempVector;
tempVector.reserve(faces.size());
FaceVectorType::const_iterator faceIt;
for (faceIt = faces.begin(); faceIt != faces.end(); ++faceIt)
{
bool foundMatch(false);
std::vector<FaceVectorType>::iterator tempIt;
for (tempIt = tempVector.begin(); tempIt != tempVector.end(); ++tempIt)
{
if (object->isEqual((*tempIt).front(), *faceIt))
{
(*tempIt).push_back(*faceIt);
foundMatch = true;
break;
}
}
if (!foundMatch)
{
FaceVectorType another;
another.reserve(faces.size());
another.push_back(*faceIt);
tempVector.push_back(another);
}
}
std::vector<FaceVectorType>::iterator it;
for (it = tempVector.begin(); it != tempVector.end(); ++it)
{
if ((*it).size() < 2)
continue;
equalityVector.push_back(*it);
}
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////
GeomAbs_SurfaceType FaceTypedBase::getFaceType(TopoDS_Shape faceIn)
{
Handle(Geom_Surface) surface = BRep_Tool::Surface(TopoDS::Face(faceIn));
GeomAdaptor_Surface surfaceTest(surface);
return surfaceTest.GetType();
}
///////////////////////////////////////////////////////////////////////////////////////////////////////
FaceTypedPlane::FaceTypedPlane() : FaceTypedBase(GeomAbs_Plane)
{
}
bool FaceTypedPlane::isEqual(const TopoDS_Shape &faceOne, const TopoDS_Shape &faceTwo) const
{
Handle(Geom_Plane) planeSurfaceOne = Handle(Geom_Plane)::DownCast(BRep_Tool::Surface(TopoDS::Face(faceOne)));
Handle(Geom_Plane) planeSurfaceTwo = Handle(Geom_Plane)::DownCast(BRep_Tool::Surface(TopoDS::Face(faceTwo)));
if (planeSurfaceOne.IsNull() || planeSurfaceTwo.IsNull())
return false;//error?
gp_Pln planeOne(planeSurfaceOne->Pln());
gp_Pln planeTwo(planeSurfaceTwo->Pln());
return (planeOne.Position().Direction().IsParallel(planeTwo.Position().Direction(), Precision::Confusion()) &&
planeOne.Distance(planeTwo) < Precision::Confusion());
}
GeomAbs_SurfaceType FaceTypedPlane::getType() const
{
return GeomAbs_Plane;
}
TopoDS_Face FaceTypedPlane::buildFace(const FaceVectorType &faces) const
{
TopoDS_Wire newWire = ModelRefine::facesBoundary(faces);
return BRepBuilderAPI_MakeFace(newWire);
}
FaceTypedPlane& ModelRefine::getPlaneObject()
{
static FaceTypedPlane object;
return object;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////
FaceTypedCylinder::FaceTypedCylinder() : FaceTypedBase(GeomAbs_Cylinder)
{
}
bool FaceTypedCylinder::isEqual(const TopoDS_Shape &faceOne, const TopoDS_Shape &faceTwo) const
{
//check if these handles are valid?
Handle(Geom_CylindricalSurface) surfaceOne = Handle(Geom_CylindricalSurface)::DownCast(BRep_Tool::Surface(TopoDS::Face(faceOne)));
Handle(Geom_CylindricalSurface) surfaceTwo = Handle(Geom_CylindricalSurface)::DownCast(BRep_Tool::Surface(TopoDS::Face(faceTwo)));
if (surfaceOne.IsNull() || surfaceTwo.IsNull())
return false;//probably need an error
gp_Cylinder cylinderOne = surfaceOne->Cylinder();
gp_Cylinder cylinderTwo = surfaceTwo->Cylinder();
if (cylinderOne.Radius() != cylinderTwo.Radius())
return false;
if (!cylinderOne.Axis().IsCoaxial(cylinderTwo.Axis(), Precision::Confusion(), Precision::Confusion()))
return false;
return true;
}
GeomAbs_SurfaceType FaceTypedCylinder::getType() const
{
return GeomAbs_Cylinder;
}
TopoDS_Face FaceTypedCylinder::buildFace(const FaceVectorType &faces) const
{
//to do.
return TopoDS_Face();
}
FaceTypedCylinder& ModelRefine::getCylinderObject()
{
static FaceTypedCylinder object;
return object;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////
FaceUniter::FaceUniter(const TopoDS_Shell &shellIn)
{
workShell = shellIn;
}
FaceUniter::FaceUniter(const TopoDS_Solid &solidIn)
{
//get first shell
TopExp_Explorer it;
it.Init(solidIn, TopAbs_SHELL);
workShell = TopoDS::Shell(it.Current());
}
bool FaceUniter::process()
{
if (workShell.IsNull())
return false;
typeObjects.push_back(&getPlaneObject());
// typeObjects.push_back(&getCylinderObject());
//add more face types.
ModelRefine::FaceTypeSplitter splitter;
splitter.addShell(workShell);
std::vector<FaceTypedBase *>::iterator typeIt;
for(typeIt = typeObjects.begin(); typeIt != typeObjects.end(); ++typeIt)
splitter.registerType((*typeIt)->getType());
splitter.split();
ModelRefine::FaceVectorType facesToRemove;
ModelRefine::FaceVectorType facesToSew;
for(typeIt = typeObjects.begin(); typeIt != typeObjects.end(); ++typeIt)
{
ModelRefine::FaceVectorType typedFaces = splitter.getTypedFaceVector((*typeIt)->getType());
ModelRefine::FaceEqualitySplitter equalitySplitter;
equalitySplitter.split(typedFaces, *typeIt);
for (std::size_t indexEquality(0); indexEquality < equalitySplitter.getGroupCount(); ++indexEquality)
{
ModelRefine::FaceAdjacencySplitter adjacencySplitter;
adjacencySplitter.split(equalitySplitter.getGroup(indexEquality));
// std::cout << " adjacency group count: " << adjacencySplitter.getGroupCount() << std::endl;
for (std::size_t adjacentIndex(0); adjacentIndex < adjacencySplitter.getGroupCount(); ++adjacentIndex)
{
// std::cout << " face count is: " << adjacencySplitter.getGroup(adjacentIndex).size() << std::endl;
TopoDS_Face newFace = (*typeIt)->buildFace(adjacencySplitter.getGroup(adjacentIndex));
if (!newFace.IsNull())
{
facesToSew.push_back(newFace);
if (facesToRemove.capacity() <= facesToRemove.size() + adjacencySplitter.getGroup(adjacentIndex).size())
facesToRemove.reserve(facesToRemove.size() + adjacencySplitter.getGroup(adjacentIndex).size());
FaceVectorType temp = adjacencySplitter.getGroup(adjacentIndex);
facesToRemove.insert(facesToRemove.end(), temp.begin(), temp.end());
}
}
}
}
if (facesToSew.size() > 0)
{
workShell = ModelRefine::removeFaces(workShell, facesToRemove);
BRepBuilderAPI_Sewing sew;
sew.Add(workShell);
FaceVectorType::iterator sewIt;
for(sewIt = facesToSew.begin(); sewIt != facesToSew.end(); ++sewIt)
sew.Add(*sewIt);
sew.Perform();
workShell = TopoDS::Shell(sew.SewedShape());
BRepLib_FuseEdges edgeFuse(workShell, true);
workShell = TopoDS::Shell(edgeFuse.Shape());
}
return true;
}
bool FaceUniter::getSolid(TopoDS_Solid &outSolid) const
{
BRepBuilderAPI_MakeSolid solidMaker;
solidMaker.Add(workShell);
outSolid = solidMaker.Solid();
return solidMaker.IsDone() ? true : false;
}

View File

@ -0,0 +1,164 @@
/***************************************************************************
* Copyright (c) 2011 Thomas Anderson <blobfish[at]gmx.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 *
* *
***************************************************************************/
#ifndef MODELREFINE_H
#define MODELREFINE_H
#include <vector>
#include <map>
#include <list>
#include <set>
#include <GeomAbs_SurfaceType.hxx>
#include <TopoDS_Shell.hxx>
#include <TopoDS_Face.hxx>
#include <TopoDS_Solid.hxx>
#include <TopoDS_Wire.hxx>
#include <TopoDS_Edge.hxx>
#include <TopTools_ListOfShape.hxx>
#include <TopTools_MapOfShape.hxx>
namespace ModelRefine
{
typedef std::vector<TopoDS_Face> FaceVectorType;
void getFaceEdges(TopoDS_Shape face, std::vector<TopoDS_Edge> &edges);
bool hasSharedEdges(const TopoDS_Shape &faceOne, const TopoDS_Shape &faceTwo);
TopoDS_Wire facesBoundary(const FaceVectorType &faces);
TopoDS_Shell removeFaces(const TopoDS_Shell &shell, const FaceVectorType &faces);
class FaceTypedBase
{
private:
FaceTypedBase(){}
protected:
FaceTypedBase(GeomAbs_SurfaceType typeIn){surfaceType = typeIn;}
public:
virtual bool isEqual(const TopoDS_Shape &faceOne, const TopoDS_Shape &faceTwo) const = 0;
virtual GeomAbs_SurfaceType getType() const = 0;
virtual TopoDS_Face buildFace(const FaceVectorType &faces) const = 0;
static GeomAbs_SurfaceType getFaceType(TopoDS_Shape faceIn);
protected:
GeomAbs_SurfaceType surfaceType;
};
class FaceTypedPlane : public FaceTypedBase
{
private:
FaceTypedPlane();
public:
virtual bool isEqual(const TopoDS_Shape &faceOne, const TopoDS_Shape &faceTwo) const;
virtual GeomAbs_SurfaceType getType() const;
virtual TopoDS_Face buildFace(const FaceVectorType &faces) const;
friend FaceTypedPlane& getPlaneObject();
};
FaceTypedPlane& getPlaneObject();
class FaceTypedCylinder : public FaceTypedBase
{
private:
FaceTypedCylinder();
public:
virtual bool isEqual(const TopoDS_Shape &faceOne, const TopoDS_Shape &faceTwo) const;
virtual GeomAbs_SurfaceType getType() const;
virtual TopoDS_Face buildFace(const FaceVectorType &faces) const;
friend FaceTypedCylinder& getCylinderObject();
};
FaceTypedCylinder& getCylinderObject();
class FaceTypeSplitter
{
typedef std::map<GeomAbs_SurfaceType, FaceVectorType> SplitMapType;
public:
FaceTypeSplitter(){}
void addShell(const TopoDS_Shell &shellIn);
void registerType(const GeomAbs_SurfaceType &type);
bool hasType(const GeomAbs_SurfaceType &type) const;
void split();
FaceVectorType getTypedFaceVector(const GeomAbs_SurfaceType &type) const;
private:
SplitMapType typeMap;
TopoDS_Shell shell;
};
class FaceAdjacencySplitter
{
public:
FaceAdjacencySplitter(){}
void split(const FaceVectorType &facesIn);
int getGroupCount() const {return adjacencyArray.size();}
FaceVectorType getGroup(const std::size_t &index) const {return adjacencyArray[index];}
private:
bool hasBeenMapped(const TopoDS_Face &shape);
void recursiveFind(FaceVectorType &tempSet, const FaceVectorType &facesIn);
std::vector<FaceVectorType> adjacencyArray;
};
class FaceEqualitySplitter
{
public:
FaceEqualitySplitter(){}
void split(const FaceVectorType &faces, FaceTypedBase *object);
int getGroupCount() const {return equalityVector.size();}
FaceVectorType getGroup(const std::size_t &index) const {return equalityVector[index];}
private:
std::vector<FaceVectorType> equalityVector;
};
class FaceUniter
{
private:
FaceUniter(){}
public:
FaceUniter(const TopoDS_Shell &shellIn);
FaceUniter(const TopoDS_Solid &solidIn);//get first shell
bool process();
const TopoDS_Shell& getShell() const {return workShell;}
bool getSolid(TopoDS_Solid &outSolid) const;//tries to make solid from shell.
private:
TopoDS_Shell workShell;
std::vector<FaceTypedBase *> typeObjects;
};
}
/* excerpt from GeomAbs_SurfaceType.hxx
enum GeomAbs_SurfaceType {
GeomAbs_Plane,
GeomAbs_Cylinder,
GeomAbs_Cone,
GeomAbs_Sphere,
GeomAbs_Torus,
GeomAbs_BezierSurface,
GeomAbs_BSplineSurface,
GeomAbs_SurfaceOfRevolution,
GeomAbs_SurfaceOfExtrusion,
GeomAbs_OffsetSurface,
GeomAbs_OtherSurface
};
*/
#endif // MODELREFINE_H