]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOMAlgo/GEOMAlgo_FinderShapeOn.cxx
Salome HOME
775522cab969c71cd94ec16e84ffe69da9f2fe03
[modules/geom.git] / src / GEOMAlgo / GEOMAlgo_FinderShapeOn.cxx
1 // File:        GEOMAlgo_FinderShapeOn.cxx
2 // Created:     Tue Jan 11 14:44:31 2005
3 // Author:      Peter KURNEV
4 //              <pkv@irinox>
5
6
7 #include <GEOMAlgo_FinderShapeOn.ixx>
8
9 #include <TopAbs_ShapeEnum.hxx>
10
11 #include <TopoDS.hxx>
12 #include <TopoDS_Face.hxx>
13 #include <TopoDS_Shape.hxx>
14 #include <TopoDS_Compound.hxx>
15 #include <TopoDS_Shell.hxx>
16 #include <TopoDS_Solid.hxx>
17 #include <TopoDS_Vertex.hxx>
18 #include <TopoDS_Edge.hxx>
19
20 #include <TopoDS_Iterator.hxx>
21
22 #include <TopTools_ListIteratorOfListOfShape.hxx>
23 #include <TopTools_IndexedMapOfShape.hxx>
24 #include <TopTools_DataMapOfShapeShape.hxx>
25
26 #include <BRep_Builder.hxx>
27
28 #include <TopExp.hxx>
29 #include <TopExp_Explorer.hxx>
30
31 #include <BRepLib_MakeFace.hxx>
32 #include <BRepLib_FaceError.hxx>
33
34 #include <BOPTools_DSFiller.hxx>
35
36 #include <GEOMAlgo_WireSolid.hxx>
37 #include <GEOMAlgo_ShellSolid.hxx>
38 #include <GEOMAlgo_VertexSolid.hxx>
39 #include <GEOMAlgo_ShapeSolid.hxx>
40
41
42 //=======================================================================
43 //function : GEOMAlgo_FinderShapeOn
44 //purpose  : 
45 //=======================================================================
46 GEOMAlgo_FinderShapeOn::GEOMAlgo_FinderShapeOn()
47 :
48   GEOMAlgo_ShapeAlgo()
49 {
50   myTolerance=0.0001;
51   myShapeType=TopAbs_VERTEX;
52   myState=GEOMAlgo_ST_UNKNOWN; 
53 }
54 //=======================================================================
55 //function : ~
56 //purpose  : 
57 //=======================================================================
58 GEOMAlgo_FinderShapeOn::~GEOMAlgo_FinderShapeOn()
59 {
60 }
61 //=======================================================================
62 //function : SetSurface
63 //purpose  : 
64 //=======================================================================
65 void GEOMAlgo_FinderShapeOn::SetSurface(const Handle(Geom_Surface)& aS)
66 {
67   mySurface=aS;
68 }
69 //=======================================================================
70 //function : Surface
71 //purpose  : 
72 //=======================================================================
73 const Handle(Geom_Surface)& GEOMAlgo_FinderShapeOn::Surface() const
74 {
75   return mySurface;
76 }
77 //=======================================================================
78 //function : SetShapeType
79 //purpose  : 
80 //=======================================================================
81 void GEOMAlgo_FinderShapeOn::SetShapeType(const TopAbs_ShapeEnum aType)
82 {
83   myShapeType=aType;
84 }
85 //=======================================================================
86 //function : ShapeType
87 //purpose  : 
88 //=======================================================================
89 TopAbs_ShapeEnum GEOMAlgo_FinderShapeOn::ShapeType()const
90 {
91   return myShapeType;
92 }
93 //=======================================================================
94 //function : SetState
95 //purpose  : 
96 //=======================================================================
97 void GEOMAlgo_FinderShapeOn::SetState(const GEOMAlgo_State aState)
98 {
99   myState=aState;
100 }
101 //=======================================================================
102 //function : State
103 //purpose  : 
104 //=======================================================================
105 GEOMAlgo_State GEOMAlgo_FinderShapeOn::State() const
106 {
107   return myState;
108 }
109 //=======================================================================
110 // function: Shapes
111 // purpose: 
112 //=======================================================================
113 const TopTools_ListOfShape& GEOMAlgo_FinderShapeOn::Shapes() const
114 {
115   return myLS;
116 }
117 //=======================================================================
118 //function : Perform
119 //purpose  : 
120 //=======================================================================
121 void GEOMAlgo_FinderShapeOn::Perform()
122 {
123   myErrorStatus=0;
124   myWarningStatus=0;
125   myLS.Clear();
126   //
127   if (!myResult.IsNull()){
128     myResult.Nullify();
129   }
130   //
131   CheckData();
132   if(myErrorStatus) {
133     return;
134   }
135   //
136   MakeArguments();
137   if(myErrorStatus || myWarningStatus) {
138     return;
139   }
140   //
141   Find();
142   if(myErrorStatus) {
143     return;
144   }
145   //
146 }
147 //=======================================================================
148 //function : Find
149 //purpose  : 
150 //=======================================================================
151 void GEOMAlgo_FinderShapeOn::Find()
152 {
153   myErrorStatus=0;
154   //
155   Standard_Boolean bIsDone;
156   Standard_Integer iErr;
157   TopTools_ListIteratorOfListOfShape aIt;
158   BRep_Builder aBB;
159   BOPTools_DSFiller aDF;
160   GEOMAlgo_ShapeSolid* pSS;
161   //
162   // 1. Prepare DSFiller
163   aDF.SetShapes (myArg1, myArg2);
164   bIsDone=aDF.IsDone();
165   if (!bIsDone) {
166     myErrorStatus=30; // wrong args are used for DSFiller
167     return;
168   }
169   aDF.Perform();
170   bIsDone=aDF.IsDone();
171   if (!bIsDone) {
172     myErrorStatus=31; // DSFiller failed
173     return;
174   }
175   // 
176   // 2. Find shapes
177   myLS.Clear();
178   //
179   if (myShapeType==TopAbs_VERTEX) {
180     pSS=new GEOMAlgo_VertexSolid;
181   }
182   else if (myShapeType==TopAbs_EDGE) {
183     pSS=new GEOMAlgo_WireSolid;
184   }
185   else if (myShapeType==TopAbs_FACE) {
186     pSS=new GEOMAlgo_ShellSolid;
187   }
188   //
189   pSS->SetFiller(aDF);
190   pSS->Perform();
191   iErr=pSS->ErrorStatus();
192   if (iErr) {
193     myErrorStatus=32; // builder ShapeSolid failed
194     delete pSS;
195     return;
196   }
197   //
198   const TopTools_ListOfShape& aLS=pSS->Shapes(myState);
199   //
200   aIt.Initialize(aLS);
201   for (; aIt.More(); aIt.Next()) {
202     const TopoDS_Shape& aSImage=aIt.Value(); 
203     if (myImages.IsBound(aSImage)) { 
204       const TopoDS_Shape& aS=myImages.Find(aSImage); 
205       myLS.Append(aS);
206     }
207     else {
208       myErrorStatus=33;// can not find original shape
209       return; 
210     }
211   }
212   //
213   delete pSS;
214 }
215 //=======================================================================
216 //function : MakeArguments
217 //purpose  : 
218 //=======================================================================
219 void GEOMAlgo_FinderShapeOn::MakeArguments()
220 {
221   myErrorStatus=0;
222   //
223   Standard_Integer i, aNb;
224   BRepLib_FaceError aFErr;
225   BRepLib_MakeFace aMF;
226   TopTools_IndexedMapOfShape aM;
227   BRep_Builder aBB;
228   TopoDS_Compound aCmp;
229   TopoDS_Shell aSh;
230   TopoDS_Solid aSd;
231   TopoDS_Shape aSC;
232   TopTools_DataMapOfShapeShape aOriginals;
233   TopExp_Explorer aExp;
234   //
235   // Argument 1
236   aMF.Init(mySurface, Standard_True);
237   aFErr=aMF.Error();
238   if (aFErr!=BRepLib_FaceDone) {
239     myErrorStatus=20; // can not build the face
240     return;
241   }
242   //
243   const TopoDS_Shape& aF=aMF.Shape();
244   //
245   // update tolerance
246   aExp.Init(aF, TopAbs_VERTEX);
247   for (; aExp.More(); aExp.Next()) {
248     const TopoDS_Vertex& aV=TopoDS::Vertex(aExp.Current());
249     aBB.UpdateVertex(aV, myTolerance);
250   }
251   aExp.Init(aF, TopAbs_EDGE);
252   for (; aExp.More(); aExp.Next()) {
253     const TopoDS_Edge& aE=TopoDS::Edge(aExp.Current());
254     aBB.UpdateEdge(aE, myTolerance);
255   }
256   const TopoDS_Face& aFace=TopoDS::Face(aF);
257   aBB.UpdateFace(aFace, myTolerance);
258   //
259   // make solid
260   aBB.MakeShell(aSh);
261   aBB.Add(aSh, aFace);
262   aBB.MakeSolid(aSd);
263   aBB.Add(aSd, aSh);
264   myArg1=aSd;
265   //
266   // Argument 2
267   //
268   myImages.Clear();
269   //
270   GEOMAlgo_FinderShapeOn::CopySource(myShape, myImages, aOriginals, aSC);
271   //
272   TopExp::MapShapes(aSC, myShapeType, aM);
273   aNb=aM.Extent();
274   if (!aNb) {
275     myWarningStatus=10; // No found subshapes of type myShapeType
276     return;
277   }
278   //
279   aBB.MakeCompound(aCmp);
280   for (i=1; i<=aNb; ++i) {
281     const TopoDS_Shape& aS=aM(i);
282     aBB.Add(aCmp, aS);
283   }
284   myArg2=aCmp;
285 }
286 //=======================================================================
287 //function : CheckData
288 //purpose  : 
289 //=======================================================================
290 void GEOMAlgo_FinderShapeOn::CheckData()
291 {
292   myErrorStatus=0;
293   //
294   if(mySurface.IsNull()) {
295     myErrorStatus=10; // mySurface=NULL
296     return;
297   }
298   //
299   if (myShape.IsNull()) {
300     myErrorStatus=11; // myShape=NULL
301     return;
302   }
303   //
304   if (!(myShapeType==TopAbs_VERTEX ||
305         myShapeType==TopAbs_EDGE ||
306         myShapeType==TopAbs_FACE)) {
307     myErrorStatus=12; // unallowed subshape type
308     return;
309   }
310   //
311   if (myState==GEOMAlgo_ST_UNKNOWN || 
312       myState==GEOMAlgo_ST_INOUT) {
313     myErrorStatus=13; // unallowed state type
314     return;
315   }
316 }
317 //
318 //=======================================================================
319 //function : CopySource
320 //purpose  : 
321 //=======================================================================
322 void GEOMAlgo_FinderShapeOn::CopySource(const TopoDS_Shape& aE,
323                                         TopTools_DataMapOfShapeShape& aImages,
324                                         TopTools_DataMapOfShapeShape& aOriginals,
325                                         TopoDS_Shape& aEx)
326 {
327   Standard_Boolean bFree;
328   TopAbs_ShapeEnum aType;
329   Standard_Integer aR;
330   BRep_Builder BB;
331   TopoDS_Iterator aIt;
332   //
333   aType=aE.ShapeType();
334   //
335   if (aOriginals.IsBound(aE)) {
336     aEx=aOriginals.ChangeFind(aE);
337     if (aType==TopAbs_EDGE) {
338       return;
339     }
340   }
341   else {
342     aEx=aE.EmptyCopied();
343     aOriginals.Bind(aE, aEx);
344     aImages.Bind(aEx, aE);
345   }
346   //
347   aR=(Standard_Integer)aType+1;
348   if (aR>TopAbs_VERTEX) {
349     return;
350   }
351   //
352   bFree=aEx.Free();
353   aEx.Free(Standard_True);
354   //
355   aType=(TopAbs_ShapeEnum) aR;
356   //
357   aIt.Initialize(aE);//, Standard_False);
358   for (; aIt.More();  aIt.Next()) {
359     const TopoDS_Shape& aV=aIt.Value();
360     TopoDS_Shape aVx;
361     //
362     CopySource (aV, aImages, aOriginals, aVx);  
363     //
364     aVx.Orientation(aV.Orientation());
365     BB.Add(aEx, aVx);
366   }
367   //
368   aEx.Free(bFree);
369 }
370
371 //
372 // myErrorStatus :
373 //
374 // 10 -mySurface=NULL
375 // 11 -myShape=NULL
376 // 12 -unallowed type of subshapes 
377 // 13 -unallowed state  
378 // 20 -can not build the face
379 // 30 -wrong args are used for DSFiller
380 // 31 -DSFiller failed
381 // 32 -builder ShapeSolid failed
382 // 33 -can not find original shape
383 //
384 // myWarningStatus
385 //
386 // 10 -subshapes of type myShapeType can not be fond in myShape
387