]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOMImpl/GEOMImpl_PipeDriver.cxx
Salome HOME
Add new command GetShapesOnBox.
[modules/geom.git] / src / GEOMImpl / GEOMImpl_PipeDriver.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_PipeDriver.hxx>
24 #include <GEOMImpl_IPipe.hxx>
25 #include <GEOMImpl_Types.hxx>
26 #include <GEOM_Function.hxx>
27
28 #include <BRep_Tool.hxx>
29 #include <BRepCheck_Analyzer.hxx>
30 #include <BRepOffsetAPI_MakePipe.hxx>
31 #include <BRepBuilderAPI_MakeWire.hxx>
32
33 #include <TopAbs.hxx>
34 #include <TopoDS.hxx>
35 #include <TopoDS_Wire.hxx>
36 #include <TopoDS_Edge.hxx>
37 #include <TopoDS_Shape.hxx>
38 #include <BRepOffsetAPI_MakePipeShell.hxx>
39 #include <TColStd_HSequenceOfTransient.hxx>
40 #include <GEOMImpl_IPipeDiffSect.hxx>
41
42 #include <Standard_NullObject.hxx>
43 #include <Standard_TypeMismatch.hxx>
44 #include <Standard_ConstructionError.hxx>
45 #include "utilities.h"
46 #include <TopExp_Explorer.hxx>
47 #include <TopTools_SequenceOfShape.hxx>
48 #include <BRep_Builder.hxx>
49 #include <TopoDS_Compound.hxx>
50
51 //=======================================================================
52 //function : GetID
53 //purpose  :
54 //=======================================================================
55 const Standard_GUID& GEOMImpl_PipeDriver::GetID()
56 {
57   static Standard_GUID aPipeDriver("FF1BBB19-5D14-4df2-980B-3A668264EA16");
58   return aPipeDriver;
59 }
60
61
62 //=======================================================================
63 //function : GEOMImpl_PipeDriver
64 //purpose  :
65 //=======================================================================
66 GEOMImpl_PipeDriver::GEOMImpl_PipeDriver()
67 {
68 }
69
70 //=======================================================================
71 //function : Execute
72 //purpose  :
73 //=======================================================================
74 Standard_Integer GEOMImpl_PipeDriver::Execute(TFunction_Logbook& log) const
75 {
76   if (Label().IsNull()) return 0;
77   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
78   GEOMImpl_IPipe* aCI= 0;
79   Standard_Integer aType = aFunction->GetType();
80   if(aType == PIPE_BASE_PATH)
81     aCI = new GEOMImpl_IPipe(aFunction);
82   else if(aType == PIPE_DIFFERENT_SECTIONS)
83     aCI = new GEOMImpl_IPipeDiffSect(aFunction);
84   else
85     return 0;
86
87   Handle(GEOM_Function) aRefPath = aCI->GetPath();
88   TopoDS_Shape aShapePath = aRefPath->GetValue();
89
90
91   if (aShapePath.IsNull())
92   {
93     cout<<"Driver : path is null"<<endl;
94     if(aCI) delete aCI;
95     Standard_NullObject::Raise("MakePipe aborted : null path argument");
96   }
97   
98   // Get path contour
99   TopoDS_Wire aWirePath;
100   if (aShapePath.ShapeType() == TopAbs_WIRE) {
101     aWirePath = TopoDS::Wire(aShapePath);
102   } 
103   else {
104     if (aShapePath.ShapeType() == TopAbs_EDGE) {
105          
106         TopoDS_Edge anEdge = TopoDS::Edge(aShapePath);
107         aWirePath = BRepBuilderAPI_MakeWire(anEdge);
108     } 
109     else {
110       if(aCI) delete aCI;
111       Standard_TypeMismatch::Raise("MakePipe aborted : path shape is neither a wire nor an edge");
112     } 
113   }
114   
115   TopoDS_Shape aShape;
116
117   if (aType == PIPE_BASE_PATH) 
118   {
119      
120     Handle(GEOM_Function) aRefBase = aCI->GetBase();
121    
122     TopoDS_Shape aShapeBase = aRefBase->GetValue();
123     
124     if (aShapeBase.IsNull()) {
125       if(aCI) delete aCI;
126       Standard_NullObject::Raise("MakePipe aborted : null base argument");
127     }
128
129     // Make pipe
130     aShape = BRepOffsetAPI_MakePipe(aWirePath, aShapeBase);
131   }
132
133   //building pipe with different sections
134   else if (aType == PIPE_DIFFERENT_SECTIONS) 
135   {
136     GEOMImpl_IPipeDiffSect* aCIDS = (GEOMImpl_IPipeDiffSect*)aCI;
137     //GEOMImpl_IPipeDiffSect* aCIDS = static_cast<GEOMImpl_IPipeDiffSect*>(aCI);
138     BRepOffsetAPI_MakePipeShell aBuilder(aWirePath);
139     Handle(TColStd_HSequenceOfTransient) aBasesObjs = aCIDS->GetBases ();
140     Handle(TColStd_HSequenceOfTransient) aLocObjs = aCIDS->GetLocations ();
141     Standard_Boolean aWithContact = (aCIDS->GetWithContactMode());
142     Standard_Boolean aWithCorrect = (aCIDS->GetWithCorrectionMode());
143
144     Standard_Integer i =1, nbBases = aBasesObjs->Length(), 
145       nbLocs = (aLocObjs.IsNull() ? 0 :aLocObjs->Length());
146     
147     if(nbLocs && nbLocs != nbBases)
148     {
149       if(aCI) delete aCI;
150       Standard_ConstructionError::Raise("Number of sections is not equal to number of locations ");
151     }
152     TopTools_SequenceOfShape aSeqBases;
153     TopTools_SequenceOfShape aSeqLocs;
154     TopTools_SequenceOfShape aSeqFaces;
155     for( ; i <= nbBases; i++) 
156     {
157       Handle(Standard_Transient) anItem = aBasesObjs->Value(i);
158       if(anItem.IsNull())
159         continue;
160       Handle(GEOM_Function) aRefBase = Handle(GEOM_Function)::DownCast(anItem);
161       if(aRefBase.IsNull())
162         continue;
163       TopoDS_Shape aShapeBase = aRefBase->GetValue();
164       if(aShapeBase.IsNull())
165         continue;
166       TopAbs_ShapeEnum aTypeBase = aShapeBase.ShapeType();
167
168       //if for section was specified face with a few wires then a few
169       //    pipes were build and make solid 
170       if(aTypeBase == TopAbs_FACE)
171       {
172         //for case one path should be used other type function
173         aSeqFaces.Append(aShapeBase);
174         TopExp_Explorer aExpW(aShapeBase,TopAbs_WIRE);
175         for( ; aExpW.More(); aExpW.Next())
176         {
177           TopoDS_Shape aWireProf = aExpW.Current();
178           aSeqBases.Append(aWireProf);
179         }
180       }
181       else if(aTypeBase == TopAbs_WIRE || aTypeBase == TopAbs_VERTEX) 
182         aSeqBases.Append(aShapeBase);
183       else if(aTypeBase == TopAbs_EDGE)
184       {
185         TopoDS_Edge anEdge = TopoDS::Edge(aShapeBase);
186         TopoDS_Shape aWireProf = BRepBuilderAPI_MakeWire(anEdge);
187         aSeqBases.Append(aWireProf);
188       }
189       if(nbLocs) 
190       {
191         Handle(Standard_Transient) anItemLoc = aLocObjs->Value(i);
192         if(anItemLoc.IsNull())
193           continue;
194         Handle(GEOM_Function) aRefLoc = Handle(GEOM_Function)::DownCast(anItemLoc);
195         TopoDS_Shape aShapeLoc = aRefLoc->GetValue();
196         if(aShapeLoc.IsNull() || aShapeLoc.ShapeType() != TopAbs_VERTEX)
197           continue;
198         aSeqLocs.Append(aShapeLoc);
199       }
200     }
201     
202     nbLocs = aSeqLocs.Length();
203     Standard_Integer nbShapes = aSeqBases.Length();
204     Standard_Integer step = nbShapes/nbBases;
205     
206     if(nbShapes < nbBases || fmod((double)nbShapes, (double)nbBases))
207     {
208       if(aCI) delete aCI;
209       Standard_ConstructionError::Raise("Invalid sections were specified for building pipe");
210     }
211      
212     Standard_Integer ind =0;     
213     for( i=1; i  <= nbShapes && ind < nbShapes; i++) //i+nbBases <= nbShapes
214     {
215       TopTools_SequenceOfShape usedBases;
216       Standard_Integer j = 1;
217       for( ; j <= nbBases ; j++) 
218       {
219         ind = i + (j-1)*step;
220         
221          TopoDS_Shape aWireProf = aSeqBases.Value(ind);
222          usedBases.Append(aWireProf);
223          if(nbLocs) 
224          {
225            TopoDS_Shape aShapeLoc = aSeqLocs.Value(j);
226            TopoDS_Vertex aVert = TopoDS::Vertex(aShapeLoc);
227            aBuilder.Add(aWireProf,aVert,aWithContact,aWithCorrect);
228          }
229          else
230            aBuilder.Add(aWireProf,aWithContact,aWithCorrect);
231       }
232       if(!aBuilder.IsReady())
233       {
234         if(aCI) delete aCI;
235         Standard_ConstructionError::Raise("Invalid input data for building PIPE: bases are invalid");
236       }
237       aBuilder.Build();
238       aShape = aBuilder.Shape();
239       aSeqFaces.Append(aShape); 
240       for( j = 1; j <=usedBases.Length(); j++)
241         aBuilder.Delete(usedBases.Value(j));
242     }
243         
244     //for case if section is face 
245     if(aSeqFaces.Length() >1)
246     {
247       BRep_Builder aB;
248       TopoDS_Compound aComp;
249       aB.MakeCompound(aComp);
250       for( i = 1; i <= aSeqFaces.Length(); i++)
251         aB.Add(aComp,aSeqFaces.Value(i));
252       aShape = aComp;
253     }
254   }
255   if (aShape.IsNull()) return 0;
256
257   BRepCheck_Analyzer ana (aShape, Standard_False);
258   if (!ana.IsValid()) {
259     if(aCI) delete aCI;
260     Standard_ConstructionError::Raise("Algorithm have produced an invalid shape result");
261   }
262
263   aFunction->SetValue(aShape);
264
265   log.SetTouched(Label());
266   if(aCI) delete aCI;
267   return 1;
268 }
269
270
271 //=======================================================================
272 //function :  GEOMImpl_PipeDriver_Type_
273 //purpose  :
274 //=======================================================================
275 Standard_EXPORT Handle_Standard_Type& GEOMImpl_PipeDriver_Type_()
276 {
277
278   static Handle_Standard_Type aType1 = STANDARD_TYPE(TFunction_Driver);
279   if ( aType1.IsNull()) aType1 = STANDARD_TYPE(TFunction_Driver);
280   static Handle_Standard_Type aType2 = STANDARD_TYPE(MMgt_TShared);
281   if ( aType2.IsNull()) aType2 = STANDARD_TYPE(MMgt_TShared);
282   static Handle_Standard_Type aType3 = STANDARD_TYPE(Standard_Transient);
283   if ( aType3.IsNull()) aType3 = STANDARD_TYPE(Standard_Transient);
284
285
286   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
287   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_PipeDriver",
288                                                          sizeof(GEOMImpl_PipeDriver),
289                                                          1,
290                                                          (Standard_Address)_Ancestors,
291                                                          (Standard_Address)NULL);
292
293   return _aType;
294 }
295
296 //=======================================================================
297 //function : DownCast
298 //purpose  :
299 //=======================================================================
300 const Handle(GEOMImpl_PipeDriver) Handle(GEOMImpl_PipeDriver)::DownCast(const Handle(Standard_Transient)& AnObject)
301 {
302   Handle(GEOMImpl_PipeDriver) _anOtherObject;
303
304   if (!AnObject.IsNull()) {
305      if (AnObject->IsKind(STANDARD_TYPE(GEOMImpl_PipeDriver))) {
306        _anOtherObject = Handle(GEOMImpl_PipeDriver)((Handle(GEOMImpl_PipeDriver)&)AnObject);
307      }
308   }
309
310   return _anOtherObject ;
311 }