1 // File: BlockFix_UnionEdges.cxx
2 // Created: 07.12.04 15:27:30
6 #include <BlockFix_UnionEdges.ixx>
8 #include <Approx_Curve3d.hxx>
9 #include <BRepAdaptor_HCompCurve.hxx>
10 #include <BRep_Builder.hxx>
11 #include <BRep_Tool.hxx>
12 #include <GC_MakeCircle.hxx>
13 #include <Geom_BSplineCurve.hxx>
14 #include <Geom_Circle.hxx>
15 #include <Geom_Curve.hxx>
16 #include <Geom_Line.hxx>
17 #include <Geom_TrimmedCurve.hxx>
18 #include <ShapeAnalysis_Edge.hxx>
19 #include <ShapeFix_Edge.hxx>
20 #include <ShapeFix_Face.hxx>
21 #include <ShapeFix_Shell.hxx>
22 #include <TColgp_SequenceOfPnt.hxx>
23 #include <TColStd_MapOfInteger.hxx>
25 #include <TopExp_Explorer.hxx>
26 #include <TopTools_IndexedDataMapOfShapeListOfShape.hxx>
27 #include <TopTools_IndexedMapOfShape.hxx>
28 #include <TopTools_ListOfShape.hxx>
29 #include <TopTools_MapOfShape.hxx>
30 #include <TopTools_ListIteratorOfListOfShape.hxx>
31 #include <TopTools_SequenceOfShape.hxx>
33 #include <TopoDS_Edge.hxx>
34 #include <TopoDS_Face.hxx>
35 #include <TopoDS_Shell.hxx>
36 #include <TopoDS_Solid.hxx>
37 #include <TopoDS_Vertex.hxx>
38 #include <TopoDS_Iterator.hxx>
41 //=======================================================================
42 //function : BlockFix_UnionEdges()
43 //purpose : Constructor
44 //=======================================================================
46 BlockFix_UnionEdges::BlockFix_UnionEdges ( )
51 //=======================================================================
52 //function : MergeEdges
54 //=======================================================================
55 static Standard_Boolean MergeEdges(const TopTools_SequenceOfShape& SeqEdges,
56 const TopoDS_Face& aFace,
57 const Standard_Real Tol,
60 // make chain for union
62 ShapeAnalysis_Edge sae;
63 TopoDS_Edge FirstE = TopoDS::Edge(SeqEdges.Value(1));
64 TopoDS_Edge LastE = FirstE;
65 TopoDS_Vertex VF = sae.FirstVertex(FirstE);
66 TopoDS_Vertex VL = sae.LastVertex(LastE);
67 TopTools_SequenceOfShape aChain;
68 aChain.Append(FirstE);
69 TColStd_MapOfInteger IndUsedEdges;
72 for(j=2; j<=SeqEdges.Length(); j++) {
73 for(Standard_Integer k=2; k<=SeqEdges.Length(); k++) {
74 if(IndUsedEdges.Contains(k)) continue;
75 TopoDS_Edge edge = TopoDS::Edge(SeqEdges.Value(k));
76 TopoDS_Vertex VF2 = sae.FirstVertex(edge);
77 TopoDS_Vertex VL2 = sae.LastVertex(edge);
78 if(sae.FirstVertex(edge).IsSame(VL)) {
81 VL = sae.LastVertex(LastE);
84 else if(sae.LastVertex(edge).IsSame(VF)) {
87 VF = sae.FirstVertex(FirstE);
92 if(aChain.Length()<SeqEdges.Length()) {
93 cout<<"can not create correct chain..."<<endl;
94 return Standard_False;
96 // union edges in chain
97 // first step: union lines and circles
99 Standard_Real fp1,lp1,fp2,lp2;
100 for(j=1; j<aChain.Length(); j++) {
101 TopoDS_Edge edge1 = TopoDS::Edge(aChain.Value(j));
102 Handle(Geom_Curve) c3d1 = BRep_Tool::Curve(edge1,Loc,fp1,lp1);
103 if(c3d1.IsNull()) break;
104 while(c3d1->IsKind(STANDARD_TYPE(Geom_TrimmedCurve))) {
105 Handle(Geom_TrimmedCurve) tc =
106 Handle(Geom_TrimmedCurve)::DownCast(c3d1);
107 c3d1 = tc->BasisCurve();
109 TopoDS_Edge edge2 = TopoDS::Edge(aChain.Value(j+1));
110 Handle(Geom_Curve) c3d2 = BRep_Tool::Curve(edge2,Loc,fp2,lp2);
111 if(c3d2.IsNull()) break;
112 while(c3d2->IsKind(STANDARD_TYPE(Geom_TrimmedCurve))) {
113 Handle(Geom_TrimmedCurve) tc =
114 Handle(Geom_TrimmedCurve)::DownCast(c3d2);
115 c3d2 = tc->BasisCurve();
117 if( c3d1->IsKind(STANDARD_TYPE(Geom_Line)) && c3d2->IsKind(STANDARD_TYPE(Geom_Line)) ) {
119 Handle(Geom_Line) L1 = Handle(Geom_Line)::DownCast(c3d1);
120 Handle(Geom_Line) L2 = Handle(Geom_Line)::DownCast(c3d2);
121 gp_Dir Dir1 = L1->Position().Direction();
122 gp_Dir Dir2 = L2->Position().Direction();
123 if(!Dir1.IsEqual(Dir2,Precision::Angular())) continue;
124 // can union lines => create new edge
125 TopoDS_Vertex V1 = sae.FirstVertex(edge1);
126 gp_Pnt PV1 = BRep_Tool::Pnt(V1);
127 TopoDS_Vertex V2 = sae.LastVertex(edge2);
128 gp_Pnt PV2 = BRep_Tool::Pnt(V2);
130 Handle(Geom_Line) L = new Geom_Line(gp_Ax1(PV1,Vec));
131 Standard_Real dist = PV1.Distance(PV2);
132 Handle(Geom_TrimmedCurve) tc = new Geom_TrimmedCurve(L,0.0,dist);
134 B.MakeEdge (E,tc,Precision::Confusion());
135 B.Add (E,V1); B.Add (E,V2);
136 B.UpdateVertex(V1, 0., E, 0.);
137 B.UpdateVertex(V2, dist, E, 0.);
139 sfe.FixAddPCurve(E,aFace,Standard_False);
140 sfe.FixSameParameter(E);
142 aChain.SetValue(j,E);
145 if( c3d1->IsKind(STANDARD_TYPE(Geom_Circle)) && c3d2->IsKind(STANDARD_TYPE(Geom_Circle)) ) {
147 Handle(Geom_Circle) C1 = Handle(Geom_Circle)::DownCast(c3d1);
148 Handle(Geom_Circle) C2 = Handle(Geom_Circle)::DownCast(c3d2);
149 gp_Pnt P01 = C1->Location();
150 gp_Pnt P02 = C2->Location();
151 if(P01.Distance(P02)>Precision::Confusion()) continue;
152 // can union circles => create new edge
153 TopoDS_Vertex V1 = sae.FirstVertex(edge1);
154 gp_Pnt PV1 = BRep_Tool::Pnt(V1);
155 TopoDS_Vertex V2 = sae.LastVertex(edge2);
156 gp_Pnt PV2 = BRep_Tool::Pnt(V2);
157 TopoDS_Vertex VM = sae.LastVertex(edge1);
158 gp_Pnt PVM = BRep_Tool::Pnt(VM);
159 GC_MakeCircle MC(PV1,PVM,PV2);
160 Handle(Geom_Circle) C = MC.Value();
161 gp_Pnt P0 = C->Location();
162 gp_Dir D1(gp_Vec(P0,PV1));
163 gp_Dir D2(gp_Vec(P0,PV2));
164 Standard_Real fpar = C->XAxis().Direction().Angle(D1);
165 Standard_Real lpar = C->XAxis().Direction().Angle(D2);
166 Handle(Geom_TrimmedCurve) tc = new Geom_TrimmedCurve(C,fpar,lpar);
168 B.MakeEdge (E,tc,Precision::Confusion());
169 B.Add (E,V1); B.Add (E,V2);
170 B.UpdateVertex(V1, fpar, E, 0.);
171 B.UpdateVertex(V2, lpar, E, 0.);
173 sfe.FixAddPCurve(E,aFace,Standard_False);
174 sfe.FixSameParameter(E);
176 aChain.SetValue(j,E);
180 if(j<aChain.Length()) {
181 cout<<"null curve3d in edge..."<<endl;
182 return Standard_False;
184 if(aChain.Length()>1) {
185 // second step: union edges with various curves
186 cout<<"can not make analitical union => make approximation"<<endl;
189 for(j=1; j<=aChain.Length(); j++) {
190 TopoDS_Edge edge = TopoDS::Edge(aChain.Value(j));
193 Handle(BRepAdaptor_HCompCurve) Adapt = new BRepAdaptor_HCompCurve(W);
194 Approx_Curve3d Conv(Adapt,Tol,GeomAbs_C1,9,1000);
195 Handle(Geom_BSplineCurve) bc = Conv.Curve();
197 B.MakeEdge (E,bc,Precision::Confusion());
200 aChain.SetValue(1,E);
203 anEdge = TopoDS::Edge(aChain.Value(1));
204 return Standard_True;
208 //=======================================================================
211 //=======================================================================
213 TopoDS_Shape BlockFix_UnionEdges::Perform(const TopoDS_Shape& Shape,
214 const Standard_Real Tol)
216 myContext = new ShapeBuild_ReShape;
218 TopoDS_Shape aResult = myContext->Apply(Shape);
220 // processing each solid
221 TopExp_Explorer exps;
222 for(exps.Init(Shape, TopAbs_SOLID); exps.More(); exps.Next()) {
223 TopoDS_Solid aSolid = TopoDS::Solid(exps.Current());
225 TopTools_IndexedMapOfShape ChangedFaces;
227 // creating map of edge faces
228 TopTools_IndexedDataMapOfShapeListOfShape aMapEdgeFaces;
229 TopExp::MapShapesAndAncestors(aSolid, TopAbs_EDGE, TopAbs_FACE, aMapEdgeFaces);
231 Handle(ShapeBuild_ReShape) aContext = new ShapeBuild_ReShape;
232 TopoDS_Shape aRes = aSolid;
233 aRes = aContext->Apply(aSolid);
235 // processing each face
237 for(exp.Init(aRes, TopAbs_FACE); exp.More(); exp.Next()) {
239 TopoDS::Face(aContext->Apply(exp.Current().Oriented(TopAbs_FORWARD)));
240 TopTools_IndexedDataMapOfShapeListOfShape aMapFacesEdges;
242 for(TopExp_Explorer expe(aFace,TopAbs_EDGE); expe.More(); expe.Next()) {
243 TopoDS_Edge edge = TopoDS::Edge(expe.Current());
244 if(!aMapEdgeFaces.Contains(edge)) continue;
245 const TopTools_ListOfShape& aList = aMapEdgeFaces.FindFromKey(edge);
246 TopTools_ListIteratorOfListOfShape anIter(aList);
247 for( ; anIter.More(); anIter.Next()) {
248 TopoDS_Face face = TopoDS::Face(anIter.Value());
249 TopoDS_Face face1 = TopoDS::Face(aContext->Apply(anIter.Value()));
250 if(face1.IsSame(aFace)) continue;
251 if(aMapFacesEdges.Contains(face)) {
252 aMapFacesEdges.ChangeFromKey(face).Append(edge);
255 TopTools_ListOfShape ListEdges;
256 ListEdges.Append(edge);
257 aMapFacesEdges.Add(face,ListEdges);
262 for(Standard_Integer i=1; i<=aMapFacesEdges.Extent(); i++) {
263 const TopTools_ListOfShape& ListEdges = aMapFacesEdges.FindFromIndex(i);
264 TopTools_SequenceOfShape SeqEdges;
265 TopTools_ListIteratorOfListOfShape anIter(ListEdges);
266 for( ; anIter.More(); anIter.Next()) {
267 SeqEdges.Append(anIter.Value());
269 if(SeqEdges.Length()==1) continue;
271 if( MergeEdges(SeqEdges,aFace,Tol,E) ) {
272 // now we have only one edge - aChain.Value(1)
273 // we have to replace old ListEdges with this new edge
274 aContext->Replace(SeqEdges(1),E);
275 for(Standard_Integer j=2; j<=SeqEdges.Length(); j++) {
276 aContext->Remove(SeqEdges(j));
278 TopoDS_Face tmpF = TopoDS::Face(exp.Current());
279 if( !ChangedFaces.Contains(tmpF) )
280 ChangedFaces.Add(tmpF);
281 tmpF = TopoDS::Face(aMapFacesEdges.FindKey(i));
282 if( !ChangedFaces.Contains(tmpF) )
283 ChangedFaces.Add(tmpF);
287 } // end processing each face
289 // fix changed faces and replace them in the local context
290 for(Standard_Integer i=1; i<=ChangedFaces.Extent(); i++) {
291 TopoDS_Face aFace = TopoDS::Face(aContext->Apply(ChangedFaces.FindKey(i)));
292 Handle(ShapeFix_Face) sff = new ShapeFix_Face(aFace);
293 sff->SetContext(myContext);
294 sff->SetPrecision(myTolerance);
295 sff->SetMinTolerance(myTolerance);
296 sff->SetMaxTolerance(Max(1.,myTolerance*1000.));
298 aContext->Replace(aFace,sff->Face());
301 if(ChangedFaces.Extent()>0) {
302 // fix changed shell and replace it in the local context
303 TopoDS_Shape aRes1 = aContext->Apply(aRes);
304 TopExp_Explorer expsh;
305 for(expsh.Init(aRes1, TopAbs_SHELL); expsh.More(); expsh.Next()) {
306 TopoDS_Shell aShell = TopoDS::Shell(expsh.Current());
307 Handle(ShapeFix_Shell) sfsh = new ShapeFix_Shell;
308 sfsh->FixFaceOrientation(aShell);
309 aContext->Replace(aShell,sfsh->Shell());
311 TopoDS_Shape aRes2 = aContext->Apply(aRes1);
312 // put new solid into global context
313 myContext->Replace(aSolid,aRes2);
316 } // end processing each solid
318 aResult = myContext->Apply(Shape);