]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOMImpl/GEOMImpl_HealingDriver.cxx
Salome HOME
Add new command GetShapesOnBox.
[modules/geom.git] / src / GEOMImpl / GEOMImpl_HealingDriver.cxx
1 // Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
3 // 
4 // This library is free software; you can redistribute it and/or
5 // modify it under the terms of the GNU Lesser General Public
6 // License as published by the Free Software Foundation; either 
7 // version 2.1 of the License.
8 // 
9 // This library is distributed in the hope that it will be useful 
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
12 // Lesser General Public License for more details.
13 //
14 // You should have received a copy of the GNU Lesser General Public  
15 // License along with this library; if not, write to the Free Software 
16 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17 //
18 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 //
20
21 #include <Standard_Stream.hxx>
22
23 #include <GEOMImpl_HealingDriver.hxx>
24 #include <GEOMImpl_Types.hxx>
25 #include <GEOMImpl_IHealing.hxx>
26 #include <GEOM_Function.hxx>
27
28 #include <ShHealOper_ShapeProcess.hxx>
29 #include <ShHealOper_RemoveFace.hxx>
30 #include <ShHealOper_CloseContour.hxx>
31 #include <ShHealOper_RemoveInternalWires.hxx>
32 #include <ShHealOper_FillHoles.hxx>
33 #include <ShHealOper_Sewing.hxx>
34 #include <ShHealOper_EdgeDivide.hxx>
35 #include <ShHealOper_ChangeOrientation.hxx>
36
37 #include <TopoDS.hxx>
38 #include <TopExp.hxx>
39 #include <TopTools_IndexedMapOfShape.hxx>
40
41 #include <StdFail_NotDone.hxx>
42
43 //=======================================================================
44 //function :  raiseNotDoneExeption
45 //purpose  :  global function: forms error message and raises exeption
46 //=======================================================================
47 void raiseNotDoneExeption( const int theErrorStatus )
48 {
49   switch ( theErrorStatus )
50   {
51   case ShHealOper_NotError:           StdFail_NotDone::Raise( "ShHealOper_NotError_msg" );
52   case ShHealOper_InvalidParameters:  StdFail_NotDone::Raise( "ShHealOper_InvalidParameters_msg" );
53   case ShHealOper_ErrorExecution:
54   default:                            StdFail_NotDone::Raise( "ShHealOper_ErrorExecution_msg" );
55   }
56 }
57
58 //=======================================================================
59 //function : GetID
60 //purpose  :
61 //=======================================================================
62 const Standard_GUID& GEOMImpl_HealingDriver::GetID()
63 {
64   static Standard_GUID aHealingDriver("FF1BBB61-5D14-4df2-980B-3A668264EA16");
65   return aHealingDriver;
66 }
67
68
69 //=======================================================================
70 //function : GEOMImpl_HealingDriver
71 //purpose  :
72 //=======================================================================
73
74 GEOMImpl_HealingDriver::GEOMImpl_HealingDriver()
75 {
76 }
77
78 //=======================================================================
79 //function : Execute
80 //purpose  :
81 //=======================================================================
82 Standard_Integer GEOMImpl_HealingDriver::Execute(TFunction_Logbook& log) const
83 {
84   if (Label().IsNull()) return 0;
85   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
86
87   if (aFunction.IsNull()) return 0;
88
89   GEOMImpl_IHealing HI (aFunction);
90   Standard_Integer aType = aFunction->GetType();
91   Handle(GEOM_Function) anOriginalFunction = HI.GetOriginal();
92   if (anOriginalFunction.IsNull()) return 0;
93   TopoDS_Shape aShape, anOriginalShape = anOriginalFunction->GetValue();
94   if (anOriginalShape.IsNull()) return 0;
95
96   switch (aType)
97   {
98   case SHAPE_PROCESS:
99     ShapeProcess(&HI, anOriginalShape, aShape);
100     break;
101   case SUPPRESS_FACES:
102     SuppressFaces(&HI, anOriginalShape, aShape);
103     break;
104   case CLOSE_CONTOUR:
105     CloseContour(&HI, anOriginalShape, aShape);
106     break;
107   case REMOVE_INT_WIRES:
108     RemoveIntWires(&HI, anOriginalShape, aShape);
109     break;
110   case FILL_HOLES:
111     RemoveHoles(&HI, anOriginalShape, aShape);
112     break;
113   case SEWING:
114     Sew(&HI, anOriginalShape, aShape);
115     break;
116   case DIVIDE_EDGE:
117     AddPointOnEdge(&HI, anOriginalShape, aShape);
118     break;
119   case CHANGE_ORIENTATION:
120     ChangeOrientation(&HI, anOriginalShape, aShape);
121     break;
122   default:
123     return 0;
124   }
125
126   if (aShape.IsNull())
127     raiseNotDoneExeption( ShHealOper_ErrorExecution );
128
129   aFunction->SetValue(aShape);
130
131   log.SetTouched(Label());
132   return 1;
133 }
134
135 //=======================================================================
136 //function :  ShapeProcess
137 //purpose  :
138 //=======================================================================
139 Standard_Boolean GEOMImpl_HealingDriver::ShapeProcess (GEOMImpl_IHealing* theHI,
140                                                        const TopoDS_Shape& theOriginalShape,
141                                                        TopoDS_Shape& theOutShape) const
142 {
143   Handle(TColStd_HArray1OfExtendedString) anOperators = theHI->GetOperators();
144   Handle(TColStd_HArray1OfExtendedString) aParams = theHI->GetParameters();
145   Handle(TColStd_HArray1OfExtendedString) aValues = theHI->GetValues();
146
147   if (anOperators.IsNull() || anOperators->Length() <= 0)
148     return Standard_False;
149
150   Standard_Integer nbParams = 0, nbValues = 0;
151   if (!aParams.IsNull()) {
152     nbParams = aParams->Length();
153   }
154   if (!aValues.IsNull()) {
155     nbValues = aValues->Length();
156   }
157   if (nbParams != nbValues)
158     return Standard_False;
159
160   ShHealOper_ShapeProcess aHealer;
161   TColStd_SequenceOfAsciiString anOperatorsAS, aParamsAS, aValuesAS;
162   int i;
163   for (i = 1; i <= anOperators->Length(); i++)
164     anOperatorsAS.Append(TCollection_AsciiString(anOperators->Value(i)));
165
166   aHealer.SetOperators(anOperatorsAS);
167
168   for (i = 1; i <= nbParams; i++) {
169     aHealer.SetParameter(TCollection_AsciiString(aParams->Value(i)),
170                          TCollection_AsciiString(aValues->Value(i)));
171   }
172
173   aHealer.Perform(theOriginalShape, theOutShape);
174
175   if (!aHealer.isDone())
176     raiseNotDoneExeption( ShHealOper_NotError );
177
178   return Standard_True;
179 }
180
181 //=======================================================================
182 //function :  SupressFaces
183 //purpose  :
184 //=======================================================================
185 Standard_Boolean GEOMImpl_HealingDriver::SuppressFaces (GEOMImpl_IHealing* theHI,
186                                                         const TopoDS_Shape& theOriginalShape,
187                                                         TopoDS_Shape& theOutShape) const
188 {
189   Handle(TColStd_HArray1OfInteger) aFaces = theHI->GetFaces();
190
191   ShHealOper_RemoveFace aHealer (theOriginalShape);
192
193   Standard_Boolean aResult = Standard_False;
194   if (aFaces.IsNull()) // remove all faces
195   {
196     aResult = aHealer.Perform();
197   } else {
198     TopTools_SequenceOfShape aShapesFaces;
199     TopTools_IndexedMapOfShape aShapes;
200     TopExp::MapShapes(theOriginalShape, aShapes);
201     for (int i = 1; i <= aFaces->Length(); i++) {
202       int indexOfFace = aFaces->Value(i);
203       TopoDS_Shape aFace = aShapes.FindKey(indexOfFace);
204       aShapesFaces.Append(aFace);
205     }
206
207     aResult = aHealer.Perform(aShapesFaces);
208   }
209
210   if ( aResult )
211     theOutShape = aHealer.GetResultShape();
212   else
213     raiseNotDoneExeption( aHealer.GetErrorStatus() );
214
215   return aResult;
216 }
217
218 //=======================================================================
219 //function :  CloseContour
220 //purpose  :
221 //=======================================================================
222 Standard_Boolean GEOMImpl_HealingDriver::CloseContour (GEOMImpl_IHealing* theHI,
223                                                        const TopoDS_Shape& theOriginalShape,
224                                                        TopoDS_Shape& theOutShape) const
225 {
226   Standard_Boolean isByVertex = theHI->GetIsCommonVertex();
227   Handle(TColStd_HArray1OfInteger) aWires = theHI->GetWires();
228
229   ShHealOper_CloseContour aHealer (theOriginalShape);
230
231   Standard_Boolean aResult = Standard_False;
232   if ( aWires.IsNull() ) {
233     if ( theOriginalShape.ShapeType() == TopAbs_WIRE )
234       aResult = aHealer.Perform(TopoDS::Wire(theOriginalShape), isByVertex, !isByVertex);
235   }
236   else {
237     TopTools_SequenceOfShape aShapesWires;
238     TopTools_IndexedMapOfShape aShapes;
239     TopExp::MapShapes(theOriginalShape, aShapes);
240     for (int i = 1; i <= aWires->Length(); i++) {
241       int indexOfWire = aWires->Value(i);
242       TopoDS_Shape aWire = aShapes.FindKey(indexOfWire);
243       aShapesWires.Append(aWire);
244     }
245
246     aResult = aHealer.Perform( aShapesWires, isByVertex, !isByVertex );
247   }
248
249   if (aResult)
250     theOutShape = aHealer.GetResultShape();
251   else
252     raiseNotDoneExeption( aHealer.GetErrorStatus() );
253
254   return aResult;
255 }
256
257 //=======================================================================
258 //function :  RemoveIntWires
259 //purpose  :
260 //=======================================================================
261 Standard_Boolean GEOMImpl_HealingDriver::RemoveIntWires (GEOMImpl_IHealing* theHI,
262                                                          const TopoDS_Shape& theOriginalShape,
263                                                          TopoDS_Shape& theOutShape) const
264 {
265   Handle(TColStd_HArray1OfInteger) aWires = theHI->GetWires();
266
267   ShHealOper_RemoveInternalWires aHealer(theOriginalShape);
268
269   Standard_Boolean aResult = Standard_False;
270   if (aWires.IsNull()) { // remove all faces
271     aResult = aHealer.Remove();
272   } else {
273     TopTools_SequenceOfShape aShapesWires;
274     TopTools_IndexedMapOfShape aShapes;
275     TopExp::MapShapes(theOriginalShape, aShapes);
276     for (int i = 1; i <= aWires->Length(); i++) {
277       int indexOfWire = aWires->Value(i);
278       TopoDS_Shape aWire = aShapes.FindKey(indexOfWire);
279       aShapesWires.Append(aWire);
280     }
281
282     aResult = aHealer.Remove(aShapesWires);
283   }
284
285   if (aResult)
286     theOutShape = aHealer.GetResultShape();
287   else
288     raiseNotDoneExeption( aHealer.GetErrorStatus() );
289
290   return aResult;
291 }
292
293 //=======================================================================
294 //function :  RemoveHoles
295 //purpose  :
296 //=======================================================================
297 Standard_Boolean GEOMImpl_HealingDriver::RemoveHoles (GEOMImpl_IHealing* theHI,
298                                                       const TopoDS_Shape& theOriginalShape,
299                                                       TopoDS_Shape& theOutShape) const
300 {
301   Handle(TColStd_HArray1OfInteger) aWires = theHI->GetWires();
302
303   ShHealOper_FillHoles aHealer (theOriginalShape);
304
305   Standard_Boolean aResult = Standard_False;
306   if (aWires.IsNull()) { // remove all faces
307     aResult = aHealer.Fill();
308   } else {
309     TopTools_SequenceOfShape aShapesWires;
310     TopTools_IndexedMapOfShape aShapes;
311     TopExp::MapShapes(theOriginalShape, aShapes);
312     for (int i = 1; i <= aWires->Length(); i++) {
313       int indexOfWire = aWires->Value(i);
314       TopoDS_Shape aWire = aShapes.FindKey(indexOfWire);
315       aShapesWires.Append(aWire);
316     }
317
318     aResult = aHealer.Fill(aShapesWires);
319   }
320
321   if (aResult)
322     theOutShape = aHealer.GetResultShape();
323   else
324     raiseNotDoneExeption( aHealer.GetErrorStatus() );
325
326   return aResult;
327 }
328
329 //=======================================================================
330 //function :  Sew
331 //purpose  :
332 //=======================================================================
333 Standard_Boolean GEOMImpl_HealingDriver::Sew (GEOMImpl_IHealing* theHI,
334                                               const TopoDS_Shape& theOriginalShape,
335                                               TopoDS_Shape& theOutShape) const
336 {
337   Standard_Real aTol = theHI->GetTolerance();
338
339   ShHealOper_Sewing aHealer (theOriginalShape, aTol);
340
341   Standard_Boolean aResult = aHealer.Perform();
342
343   if (aResult)
344     theOutShape = aHealer.GetResultShape();
345   else
346     raiseNotDoneExeption( aHealer.GetErrorStatus() );
347
348   return aResult;
349 }
350
351 //=======================================================================
352 //function :  AddPointOnEdge
353 //purpose  :
354 //=======================================================================
355 Standard_Boolean GEOMImpl_HealingDriver::AddPointOnEdge (GEOMImpl_IHealing* theHI,
356                                                          const TopoDS_Shape& theOriginalShape,
357                                                          TopoDS_Shape& theOutShape) const
358 {
359   Standard_Boolean isByParameter = theHI->GetIsByParameter();
360   Standard_Integer anIndex = theHI->GetIndex();
361   Standard_Real aValue = theHI->GetDevideEdgeValue();
362
363   ShHealOper_EdgeDivide aHealer (theOriginalShape);
364
365   Standard_Boolean aResult = Standard_False;
366   if (anIndex == -1) { // apply algorythm for the whole shape which is EDGE
367     if (theOriginalShape.ShapeType() == TopAbs_EDGE)
368       aResult = aHealer.Perform(TopoDS::Edge(theOriginalShape), aValue, isByParameter);
369   } else {
370     TopTools_IndexedMapOfShape aShapes;
371     TopExp::MapShapes(theOriginalShape, aShapes);
372     TopoDS_Shape aEdgeShape = aShapes.FindKey(anIndex);
373     if (aEdgeShape.ShapeType() == TopAbs_EDGE)
374       aResult = aHealer.Perform(TopoDS::Edge(aEdgeShape), aValue, isByParameter);
375   }
376
377   if (aResult)
378     theOutShape = aHealer.GetResultShape();
379   else
380     raiseNotDoneExeption( aHealer.GetErrorStatus() );
381
382   return aResult;
383 }
384
385
386 //=======================================================================
387 //function :  ChangeOrientation
388 //purpose  :
389 //=======================================================================
390 Standard_Boolean GEOMImpl_HealingDriver::ChangeOrientation (GEOMImpl_IHealing* theHI,
391                                                             const TopoDS_Shape& theOriginalShape,
392                                                             TopoDS_Shape& theOutShape) const
393 {
394   ShHealOper_ChangeOrientation aHealer (theOriginalShape);
395
396   Standard_Boolean aResult = aHealer.Perform();
397
398   if (aResult)
399     theOutShape = aHealer.GetResultShape();
400   else
401     raiseNotDoneExeption( aHealer.GetErrorStatus() );
402
403   return aResult;
404 }
405
406
407 //=======================================================================
408 //function :  GEOMImpl_HealingDriver_Type_
409 //purpose  :
410 //=======================================================================
411 Standard_EXPORT Handle_Standard_Type& GEOMImpl_HealingDriver_Type_()
412 {
413
414   static Handle_Standard_Type aType1 = STANDARD_TYPE(TFunction_Driver);
415   if ( aType1.IsNull()) aType1 = STANDARD_TYPE(TFunction_Driver);
416   static Handle_Standard_Type aType2 = STANDARD_TYPE(MMgt_TShared);
417   if ( aType2.IsNull()) aType2 = STANDARD_TYPE(MMgt_TShared);
418   static Handle_Standard_Type aType3 = STANDARD_TYPE(Standard_Transient);
419   if ( aType3.IsNull()) aType3 = STANDARD_TYPE(Standard_Transient);
420
421
422   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
423   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_HealingDriver",
424                                                          sizeof(GEOMImpl_HealingDriver),
425                                                          1,
426                                                          (Standard_Address)_Ancestors,
427                                                          (Standard_Address)NULL);
428
429   return _aType;
430 }
431
432 //=======================================================================
433 //function : DownCast
434 //purpose  :
435 //=======================================================================
436
437 const Handle(GEOMImpl_HealingDriver) Handle(GEOMImpl_HealingDriver)::DownCast(const Handle(Standard_Transient)& AnObject)
438 {
439   Handle(GEOMImpl_HealingDriver) _anOtherObject;
440
441   if (!AnObject.IsNull()) {
442      if (AnObject->IsKind(STANDARD_TYPE(GEOMImpl_HealingDriver))) {
443        _anOtherObject = Handle(GEOMImpl_HealingDriver)((Handle(GEOMImpl_HealingDriver)&)AnObject);
444      }
445   }
446
447   return _anOtherObject ;
448 }
449
450