Salome HOME
Bug 0020413: Dump file has many GetMainShape instructions.
[modules/geom.git] / src / GEOMImpl / GEOMImpl_FillingDriver.cxx
1 //  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 //  This library is free software; you can redistribute it and/or
7 //  modify it under the terms of the GNU Lesser General Public
8 //  License as published by the Free Software Foundation; either
9 //  version 2.1 of the License.
10 //
11 //  This library is distributed in the hope that it will be useful,
12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 //  Lesser General Public License for more details.
15 //
16 //  You should have received a copy of the GNU Lesser General Public
17 //  License along with this library; if not, write to the Free Software
18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 //  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 #include <Standard_Stream.hxx>
23
24 #include <GEOMImpl_FillingDriver.hxx>
25 #include <GEOM_Function.hxx>
26 #include <GEOMImpl_IFilling.hxx>
27 #include <GEOMImpl_Types.hxx>
28
29 #include <BRep_Tool.hxx>
30 #include <BRepAlgo.hxx>
31 #include <BRepBuilderAPI_MakeFace.hxx>
32
33 #include <TopAbs.hxx>
34 #include <TopoDS.hxx>
35 #include <TopoDS_Shape.hxx>
36 #include <TopExp_Explorer.hxx>
37
38 #include <Geom_Curve.hxx>
39 #include <Geom_Surface.hxx>
40 #include <Geom_TrimmedCurve.hxx>
41 #include <Geom_BSplineSurface.hxx>
42 #include <GeomFill_Line.hxx>
43 #include <GeomFill_AppSurf.hxx>
44 #include <GeomFill_SectionGenerator.hxx>
45
46 #include <Precision.hxx>
47 #include <Standard_ConstructionError.hxx>
48
49 #include <TColGeom_SequenceOfCurve.hxx>
50 #include <ShapeFix_Face.hxx>
51 #include <GeomAPI_PointsToBSplineSurface.hxx>
52 #include <Geom_BSplineCurve.hxx>
53
54 //=======================================================================
55 //function : GetID
56 //purpose  :
57 //=======================================================================
58 const Standard_GUID& GEOMImpl_FillingDriver::GetID()
59 {
60   static Standard_GUID aFillingDriver ("FF1BBB62-5D14-4df2-980B-3A668264EA16");
61   return aFillingDriver;
62 }
63
64
65 //=======================================================================
66 //function : GEOMImpl_FillingDriver
67 //purpose  :
68 //=======================================================================
69
70 GEOMImpl_FillingDriver::GEOMImpl_FillingDriver()
71 {
72 }
73
74 //=======================================================================
75 //function : Execute
76 //purpose  :
77 //=======================================================================
78 Standard_Integer GEOMImpl_FillingDriver::Execute(TFunction_Logbook& log) const
79 {
80   if (Label().IsNull()) return 0;
81   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
82   if (aFunction.IsNull()) return 0;
83
84   if (aFunction->GetType() != BASIC_FILLING) return 0;
85
86   GEOMImpl_IFilling IF (aFunction);
87   Handle(GEOM_Function) aShapeFunction = IF.GetShape();
88   if (aShapeFunction.IsNull()) return 0;
89   TopoDS_Shape aShape = aShapeFunction->GetValue();
90   if (aShape.IsNull() || aShape.ShapeType() != TopAbs_COMPOUND) return 0;
91
92   Standard_Integer mindeg = IF.GetMinDeg();
93   Standard_Integer maxdeg = IF.GetMaxDeg();
94   Standard_Real tol3d = IF.GetTol2D();
95   Standard_Real tol2d = IF.GetTol3D();
96   Standard_Integer nbiter = IF.GetNbIter();
97   Standard_Boolean isApprox = IF.GetApprox();
98
99   if (mindeg > maxdeg) {
100     Standard_RangeError::Raise("Minimal degree can not be more than maximal degree");
101   }
102
103   /* we verify the contents of the shape */
104   TopExp_Explorer Ex;
105   TopoDS_Shape Scurrent;
106   Standard_Real First, Last;
107   Handle(Geom_Curve) C;
108
109   TopoDS_Iterator It (aShape);
110   for (; It.More(); It.Next()) {
111     Scurrent = It.Value();
112     if (Scurrent.ShapeType() != TopAbs_EDGE)
113       Standard_ConstructionError::Raise("The argument compound must contain only edges");
114   }
115
116   if (!isApprox) {
117     // make filling as in old version of SALOME (before 4.1.1)
118     GeomFill_SectionGenerator Section;
119     Standard_Integer i = 0;
120     for (Ex.Init(aShape, TopAbs_EDGE); Ex.More(); Ex.Next()) {
121       Scurrent = Ex.Current();
122       if (Scurrent.IsNull() || Scurrent.ShapeType() != TopAbs_EDGE) return 0;
123       if (BRep_Tool::Degenerated(TopoDS::Edge(Scurrent))) continue;
124       C = BRep_Tool::Curve(TopoDS::Edge(Scurrent), First, Last);
125       C = new Geom_TrimmedCurve(C, First, Last);
126       Section.AddCurve(C);
127       i++;
128     }
129
130     /* a 'tolerance' is used to compare 2 knots : see GeomFill_Generator.cdl */
131     Section.Perform(Precision::Confusion());
132     Handle(GeomFill_Line) Line = new GeomFill_Line(i);
133
134     GeomFill_AppSurf App (mindeg, maxdeg, tol3d, tol2d, nbiter); /* user parameters */
135     App.Perform(Line, Section);
136
137     if (!App.IsDone()) return 0;
138     Standard_Integer UDegree, VDegree, NbUPoles, NbVPoles, NbUKnots, NbVKnots;
139     App.SurfShape(UDegree, VDegree, NbUPoles, NbVPoles, NbUKnots, NbVKnots);
140     Handle(Geom_BSplineSurface) GBS = new Geom_BSplineSurface
141       (App.SurfPoles(), App.SurfWeights(), App.SurfUKnots(), App.SurfVKnots(),
142        App.SurfUMults(), App.SurfVMults(), App.UDegree(), App.VDegree());
143     
144     if (GBS.IsNull()) return 0;
145     aShape = BRepBuilderAPI_MakeFace(GBS);
146   }    
147   else {
148     // implemented by skl 20.03.2008 for bug 16568
149     // make approximation - try to create bspline surface
150     // using GeomAPI_PointsToBSplineSurface
151     TColGeom_SequenceOfCurve aSeq;
152     int MaxNbPoles = 0;
153     // add curves from edges to sequence and find maximal
154     // number of poles if some of them are bsplines
155     for (Ex.Init(aShape, TopAbs_EDGE); Ex.More(); Ex.Next()) {
156       Scurrent = Ex.Current();
157       if (Scurrent.IsNull() || Scurrent.ShapeType() != TopAbs_EDGE) return 0;
158       if (BRep_Tool::Degenerated(TopoDS::Edge(Scurrent))) continue;
159       C = BRep_Tool::Curve(TopoDS::Edge(Scurrent), First, Last);
160       Handle(Geom_TrimmedCurve) TC = Handle(Geom_TrimmedCurve)::DownCast(C);
161       if(TC.IsNull()) {
162         Handle(Geom_BSplineCurve) BC = Handle(Geom_BSplineCurve)::DownCast(C);
163         if(!BC.IsNull()) {
164           MaxNbPoles = Max(MaxNbPoles,BC->NbPoles());
165         }
166       }
167       else {
168         Handle(Geom_BSplineCurve) BC = Handle(Geom_BSplineCurve)::DownCast(TC->BasisCurve());
169         if(BC.IsNull()) {
170           Handle(Geom_TrimmedCurve) TC1 = Handle(Geom_TrimmedCurve)::DownCast(TC->BasisCurve());
171           if(!TC1.IsNull()) {
172             BC = Handle(Geom_BSplineCurve)::DownCast(TC1->BasisCurve());
173           }
174         }
175         if(!BC.IsNull()) {
176           MaxNbPoles = Max(MaxNbPoles,BC->NbPoles());
177         }
178       }
179       aSeq.Append(C);
180     }
181     // prepare array of points for creation bspline surface
182     // size of this array: by U parameter - number of curves,
183     // by V parameter - determ using MaxNbPoles but it's
184     // value must be between 21(min) and 101(max)
185     int nbc = aSeq.Length();
186     int nbp = Max(21,2*MaxNbPoles-1);
187     if(nbp>101) nbp = 101;
188     TColgp_Array2OfPnt Points(1,nbc,1,nbp);
189     int ic = 1;
190     for(; ic<=nbc; ic++) {
191       Handle(Geom_Curve) C = aSeq.Value(ic);
192       double fp = C->FirstParameter();
193       double lp = C->LastParameter();
194       double dp = (lp-fp)/(nbp-1);
195       int j = 0;
196       gp_Pnt P;
197       for(; j<nbp; j++) {
198         C->D0(fp+dp*j,P);
199         Points.SetValue(ic,j+1,P);
200       }
201     }
202     GeomAPI_PointsToBSplineSurface PTB(Points,mindeg,maxdeg,GeomAbs_C2,tol3d);
203     Handle(Geom_BSplineSurface) BS = PTB.Surface();
204     BRepBuilderAPI_MakeFace BB(BS);
205     TopoDS_Face NewF = BB.Face();
206     Handle(ShapeFix_Face) sff = new ShapeFix_Face(NewF);
207     sff->Perform();
208     sff->FixOrientation();
209     aShape = sff->Face();
210   }
211
212   /* We test the validity of resulting shape */
213   if (!BRepAlgo::IsValid((aShape))) {
214     Standard_ConstructionError::Raise("Algorithm have produced an invalid shape result");
215     return 0;
216   }
217
218   aFunction->SetValue(aShape);
219
220   log.SetTouched(Label());
221   return 1;
222 }
223
224
225 //=======================================================================
226 //function :  GEOMImpl_FillingDriver_Type_
227 //purpose  :
228 //=======================================================================
229 Standard_EXPORT Handle_Standard_Type& GEOMImpl_FillingDriver_Type_()
230 {
231
232   static Handle_Standard_Type aType1 = STANDARD_TYPE(TFunction_Driver);
233   if ( aType1.IsNull()) aType1 = STANDARD_TYPE(TFunction_Driver);
234   static Handle_Standard_Type aType2 = STANDARD_TYPE(MMgt_TShared);
235   if ( aType2.IsNull()) aType2 = STANDARD_TYPE(MMgt_TShared);
236   static Handle_Standard_Type aType3 = STANDARD_TYPE(Standard_Transient);
237   if ( aType3.IsNull()) aType3 = STANDARD_TYPE(Standard_Transient);
238
239
240   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
241   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_FillingDriver",
242                                                          sizeof(GEOMImpl_FillingDriver),
243                                                          1,
244                                                          (Standard_Address)_Ancestors,
245                                                          (Standard_Address)NULL);
246
247   return _aType;
248 }
249
250 //=======================================================================
251 //function : DownCast
252 //purpose  :
253 //=======================================================================
254
255 const Handle(GEOMImpl_FillingDriver) Handle(GEOMImpl_FillingDriver)::DownCast(const Handle(Standard_Transient)& AnObject)
256 {
257   Handle(GEOMImpl_FillingDriver) _anOtherObject;
258
259   if (!AnObject.IsNull()) {
260      if (AnObject->IsKind(STANDARD_TYPE(GEOMImpl_FillingDriver))) {
261        _anOtherObject = Handle(GEOMImpl_FillingDriver)((Handle(GEOMImpl_FillingDriver)&)AnObject);
262      }
263   }
264
265   return _anOtherObject ;
266 }
267
268