src/GEOMImpl/GEOMImpl_FillingDriver.cxx

   1 //  Copyright (C) 2007-2010  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
  23 #include <Standard_Stream.hxx>
  24
  25 #include <GEOMImpl_FillingDriver.hxx>
  26 #include <GEOM_Function.hxx>
  27 #include <GEOMImpl_IFilling.hxx>
  28 #include <GEOMImpl_Types.hxx>
  29
  30 #include <BRep_Tool.hxx>
  31 #include <BRepAlgo.hxx>
  32 #include <BRepBuilderAPI_MakeFace.hxx>
  33 #include <BRep_Builder.hxx>
  34 #include <BRepBuilderAPI_Copy.hxx>
  35
  36 #include <TopAbs.hxx>
  37 #include <TopoDS.hxx>
  38 #include <TopoDS_Shape.hxx>
  39 #include <TopoDS_Compound.hxx>
  40 #include <TopoDS_Edge.hxx>
  41 #include <TopoDS_Vertex.hxx>
  42 #include <TopExp_Explorer.hxx>
  43
  44 #include <Geom_Curve.hxx>
  45 #include <Geom_Surface.hxx>
  46 #include <Geom_TrimmedCurve.hxx>
  47 #include <Geom_Line.hxx>
  48 #include <Geom_Circle.hxx>
  49 #include <Geom_Ellipse.hxx>
  50 #include <Geom_BezierCurve.hxx>
  51 #include <Geom_BSplineSurface.hxx>
  52 #include <GeomFill_Line.hxx>
  53 #include <GeomFill_AppSurf.hxx>
  54 #include <GeomFill_SectionGenerator.hxx>
  55
  56 #include <Precision.hxx>
  57 #include <Standard_ConstructionError.hxx>
  58
  59 #include <TColGeom_SequenceOfCurve.hxx>
  60 #include <ShapeFix_Face.hxx>
  61 #include <GeomAPI_PointsToBSplineSurface.hxx>
  62 #include <Geom_BSplineCurve.hxx>
  63 #include <GeomAPI_PointsToBSpline.hxx>
  64
  65 #include <TColgp_SequenceOfPnt.hxx>
  66 #include <TColgp_Array1OfPnt.hxx>
  67
  68 //#include <BRepTools.hxx>
  69
  70
  71 //=======================================================================
  72 //function : GetID
  73 //purpose  :
  74 //=======================================================================
  75 const Standard_GUID& GEOMImpl_FillingDriver::GetID()
  76 {
  77   static Standard_GUID aFillingDriver ("FF1BBB62-5D14-4df2-980B-3A668264EA16");
  78   return aFillingDriver;
  79 }
  80
  81
  82 //=======================================================================
  83 //function : GEOMImpl_FillingDriver
  84 //purpose  :
  85 //=======================================================================
  86
  87 GEOMImpl_FillingDriver::GEOMImpl_FillingDriver()
  88 {
  89 }
  90
  91 //=======================================================================
  92 //function : Execute
  93 //purpose  :
  94 //=======================================================================
  95 Standard_Integer GEOMImpl_FillingDriver::Execute(TFunction_Logbook& log) const
  96 {
  97   if (Label().IsNull()) return 0;
  98   Handle(GEOM_Function) aFunction = GEOM_Function::GetFunction(Label());
  99   if (aFunction.IsNull()) return 0;
 100
 101   if (aFunction->GetType() != BASIC_FILLING) return 0;
 102
 103   GEOMImpl_IFilling IF (aFunction);
 104   Handle(GEOM_Function) aShapeFunction = IF.GetShape();
 105   if (aShapeFunction.IsNull()) return 0;
 106   TopoDS_Shape aShape;
 107
 108   BRepBuilderAPI_Copy Copy(aShapeFunction->GetValue());
 109   if( Copy.IsDone() )
 110     aShape = Copy.Shape();
 111
 112   if (aShape.IsNull() || aShape.ShapeType() != TopAbs_COMPOUND) return 0;
 113
 114   Standard_Integer mindeg = IF.GetMinDeg();
 115   Standard_Integer maxdeg = IF.GetMaxDeg();
 116   Standard_Real tol3d = IF.GetTol2D();
 117   Standard_Real tol2d = IF.GetTol3D();
 118   Standard_Integer nbiter = IF.GetNbIter();
 119   Standard_Boolean isApprox = IF.GetApprox();
 120   Standard_Integer aMethod = IF.GetMethod();
 121
 122   if (mindeg > maxdeg) {
 123     Standard_RangeError::Raise("Minimal degree can not be more than maximal degree");
 124   }
 125
 126   /* we verify the contents of the shape */
 127   TopExp_Explorer Ex;
 128   TopoDS_Shape Scurrent;
 129   Standard_Real First, Last;
 130   Handle(Geom_Curve) C;
 131
 132   TopoDS_Compound aComp;
 133   BRep_Builder B;
 134   B.MakeCompound(aComp);
 135
 136   TopoDS_Iterator It (aShape);
 137   for (; It.More(); It.Next()) {
 138     Scurrent = It.Value();
 139     if (Scurrent.ShapeType() != TopAbs_EDGE) {
 140       Handle(Geom_BSplineCurve) newC;
 141       if (Scurrent.ShapeType() == TopAbs_WIRE) {
 142         TColgp_SequenceOfPnt PntSeq;
 143         // collect points
 144         for (Ex.Init(Scurrent, TopAbs_EDGE); Ex.More(); Ex.Next()) {
 145           TopoDS_Edge E = TopoDS::Edge(Ex.Current());
 146           if (BRep_Tool::Degenerated(E)) continue;
 147           C = BRep_Tool::Curve(E, First, Last);
 148           if( E.Orientation() == TopAbs_REVERSED ) {
 149             C->Reverse();
 150           }
 151           Handle(Geom_TrimmedCurve) tc = Handle(Geom_TrimmedCurve)::DownCast(C);
 152           while( !tc.IsNull() ) {
 153             C = tc->BasisCurve();
 154             tc = Handle(Geom_TrimmedCurve)::DownCast(C);
 155           }
 156           int nbp = 10;
 157           if( C->IsKind(STANDARD_TYPE(Geom_Line)) ) {
 158             nbp = 4;
 159           }
 160           else if( C->IsKind(STANDARD_TYPE(Geom_Circle)) ||
 161                    C->IsKind(STANDARD_TYPE(Geom_Ellipse)) ) {
 162             nbp = (int)25*fabs(Last-First)/(2*PI);
 163           }
 164           else if( C->IsKind(STANDARD_TYPE(Geom_BezierCurve)) ) {
 165             Handle(Geom_BezierCurve) C3d = Handle(Geom_BezierCurve)::DownCast(C);
 166             nbp = C3d->NbPoles();
 167           }
 168           else if( C->IsKind(STANDARD_TYPE(Geom_BSplineCurve)) ) {
 169             Handle(Geom_BSplineCurve) C3d = Handle(Geom_BSplineCurve)::DownCast(C);
 170             nbp = C3d->NbPoles();
 171           }
 172           else {
 173           }
 174           if( nbp<4 ) nbp = 4;
 175           double dp = (Last-First)/(nbp-1);
 176           for(int i=1; i<nbp; i++) {
 177             gp_Pnt P;
 178             C->D0(First+dp*(i-1),P);
 179             PntSeq.Append(P);
 180           }
 181         }
 182         // add last point
 183         gp_Pnt P;
 184         C->D0(Last,P);
 185         PntSeq.Append(P);
 186         // create BSpline
 187         if(PntSeq.Length()>1) {
 188           TColgp_Array1OfPnt Pnts(1,PntSeq.Length());
 189           // check orientation of wire
 190           if( Scurrent.Orientation() == TopAbs_REVERSED ) {
 191             for(int i=1; i<=PntSeq.Length(); i++) {
 192               Pnts.SetValue(PntSeq.Length()-i+1,PntSeq.Value(i));
 193             }
 194           }
 195           else {
 196             for(int i=1; i<=PntSeq.Length(); i++) {
 197               Pnts.SetValue(i,PntSeq.Value(i));
 198             }
 199           }
 200           GeomAPI_PointsToBSpline PTB(Pnts);
 201           newC = Handle(Geom_BSplineCurve)::DownCast(PTB.Curve());
 202           // set periodic flag if curve is closed
 203           //if( newC->IsClosed() ) {
 204           //  newC->SetPeriodic();
 205           //}
 206           // create edge
 207           double fp = newC->FirstParameter();
 208           double lp = newC->FirstParameter();
 209           gp_Pnt PF,PL;
 210           newC->D0(fp,PF);
 211           newC->D0(lp,PL);
 212           TopoDS_Vertex VF,VL;
 213           B.MakeVertex(VF,PF,1.e-7);
 214           B.MakeVertex(VL,PL,1.e-7);
 215           TopoDS_Edge newE;
 216           B.MakeEdge(newE,newC,1.e-7);
 217           B.Add(newE,VF);
 218           B.Add(newE,VL.Reversed());
 219           Scurrent = newE;
 220         }
 221       }
 222       if(newC.IsNull()) {
 223         Standard_ConstructionError::Raise("The argument compound must contain only edges");
 224       }
 225     }
 226     B.Add(aComp,Scurrent);
 227   }
 228   aShape = aComp;
 229
 230   if (!isApprox) {
 231     // make filling as in old version of SALOME (before 4.1.1)
 232     GeomFill_SectionGenerator Section;
 233     Standard_Integer i = 0;
 234     Handle(Geom_Curve) aLastC;
 235     gp_Pnt PL1,PL2;
 236     for (Ex.Init(aShape, TopAbs_EDGE); Ex.More(); Ex.Next()) {
 237       Scurrent = Ex.Current();
 238       if (Scurrent.IsNull() || Scurrent.ShapeType() != TopAbs_EDGE) return 0;
 239       if (BRep_Tool::Degenerated(TopoDS::Edge(Scurrent))) continue;
 240       C = BRep_Tool::Curve(TopoDS::Edge(Scurrent), First, Last);
 241       //if (Scurrent.Orientation() == TopAbs_REVERSED)
 242       //  // Mantis isuue 0020659: consider the orientation of the edges
 243       //  C = new Geom_TrimmedCurve(C, Last, First);
 244       //else
 245       //  C = new Geom_TrimmedCurve(C, First, Last);
 246       C = new Geom_TrimmedCurve(C, First, Last);
 247       gp_Pnt P1,P2;
 248       C->D0(First,P1);
 249       C->D0(Last,P2);
 250
 251       if( aMethod==1 && Scurrent.Orientation() == TopAbs_REVERSED ) {
 252         C->Reverse();
 253       }
 254       else if( aMethod==2 ) {
 255         if( i==0 ) {
 256           PL1 = P1;
 257           PL2 = P2;
 258         }
 259         else {
 260           double d1 = PL1.Distance(P1) + PL2.Distance(P2);
 261           double d2 = PL1.Distance(P2) + PL2.Distance(P1);
 262           if(d2<d1) {
 263             C->Reverse();
 264             PL1 = P2;
 265             PL2 = P1;
 266           }
 267           else {
 268             PL1 = P1;
 269             PL2 = P2;
 270           }
 271         }
 272       }
 273
 274       Section.AddCurve(C);
 275       i++;
 276     }
 277
 278     /* a 'tolerance' is used to compare 2 knots : see GeomFill_Generator.cdl */
 279     Section.Perform(Precision::Confusion());
 280     Handle(GeomFill_Line) Line = new GeomFill_Line(i);
 281
 282     GeomFill_AppSurf App (mindeg, maxdeg, tol3d, tol2d, nbiter); /* user parameters */
 283     App.Perform(Line, Section);
 284
 285     if (!App.IsDone()) return 0;
 286     Standard_Integer UDegree, VDegree, NbUPoles, NbVPoles, NbUKnots, NbVKnots;
 287     App.SurfShape(UDegree, VDegree, NbUPoles, NbVPoles, NbUKnots, NbVKnots);
 288     Handle(Geom_BSplineSurface) GBS = new Geom_BSplineSurface
 289       (App.SurfPoles(), App.SurfWeights(), App.SurfUKnots(), App.SurfVKnots(),
 290        App.SurfUMults(), App.SurfVMults(), App.UDegree(), App.VDegree());
 291
 292     if (GBS.IsNull()) return 0;
 293     aShape = BRepBuilderAPI_MakeFace(GBS);
 294   }
 295   else {
 296     // implemented by skl 20.03.2008 for bug 16568
 297     // make approximation - try to create bspline surface
 298     // using GeomAPI_PointsToBSplineSurface
 299     TColGeom_SequenceOfCurve aSeq;
 300     int MaxNbPoles = 0;
 301     // add curves from edges to sequence and find maximal
 302     // number of poles if some of them are bsplines
 303     for (Ex.Init(aShape, TopAbs_EDGE); Ex.More(); Ex.Next()) {
 304       Scurrent = Ex.Current();
 305       if (Scurrent.IsNull() || Scurrent.ShapeType() != TopAbs_EDGE) return 0;
 306       if (BRep_Tool::Degenerated(TopoDS::Edge(Scurrent))) continue;
 307       C = BRep_Tool::Curve(TopoDS::Edge(Scurrent), First, Last);
 308       Handle(Geom_TrimmedCurve) TC = Handle(Geom_TrimmedCurve)::DownCast(C);
 309       if(TC.IsNull()) {
 310         Handle(Geom_BSplineCurve) BC = Handle(Geom_BSplineCurve)::DownCast(C);
 311         if(!BC.IsNull()) {
 312           MaxNbPoles = Max(MaxNbPoles,BC->NbPoles());
 313         }
 314       }
 315       else {
 316         Handle(Geom_BSplineCurve) BC = Handle(Geom_BSplineCurve)::DownCast(TC->BasisCurve());
 317         if(BC.IsNull()) {
 318           Handle(Geom_TrimmedCurve) TC1 = Handle(Geom_TrimmedCurve)::DownCast(TC->BasisCurve());
 319           if(!TC1.IsNull()) {
 320             BC = Handle(Geom_BSplineCurve)::DownCast(TC1->BasisCurve());
 321           }
 322         }
 323         if(!BC.IsNull()) {
 324           MaxNbPoles = Max(MaxNbPoles,BC->NbPoles());
 325         }
 326       }
 327       aSeq.Append(C);
 328     }
 329     // prepare array of points for creation bspline surface
 330     // size of this array: by U parameter - number of curves,
 331     // by V parameter - determ using MaxNbPoles but it's
 332     // value must be between 21(min) and 101(max)
 333     int nbc = aSeq.Length();
 334     int nbp = Max(21,2*MaxNbPoles-1);
 335     if(nbp>101) nbp = 101;
 336     TColgp_Array2OfPnt Points(1,nbc,1,nbp);
 337     int ic = 1;
 338     for(; ic<=nbc; ic++) {
 339       Handle(Geom_Curve) C = aSeq.Value(ic);
 340       double fp = C->FirstParameter();
 341       double lp = C->LastParameter();
 342       double dp = (lp-fp)/(nbp-1);
 343       int j = 0;
 344       gp_Pnt P;
 345       for(; j<nbp; j++) {
 346         C->D0(fp+dp*j,P);
 347         Points.SetValue(ic,j+1,P);
 348       }
 349     }
 350     GeomAPI_PointsToBSplineSurface PTB(Points,mindeg,maxdeg,GeomAbs_C2,tol3d);
 351     Handle(Geom_BSplineSurface) BS = PTB.Surface();
 352     BRepBuilderAPI_MakeFace BB(BS);
 353     TopoDS_Face NewF = BB.Face();
 354     Handle(ShapeFix_Face) sff = new ShapeFix_Face(NewF);
 355     sff->Perform();
 356     sff->FixOrientation();
 357     aShape = sff->Face();
 358   }
 359
 360   /* We test the validity of resulting shape */
 361   if (!BRepAlgo::IsValid((aShape))) {
 362     Standard_ConstructionError::Raise("Algorithm have produced an invalid shape result");
 363     return 0;
 364   }
 365
 366   aFunction->SetValue(aShape);
 367
 368   log.SetTouched(Label());
 369   return 1;
 370 }
 371
 372
 373 //=======================================================================
 374 //function :  GEOMImpl_FillingDriver_Type_
 375 //purpose  :
 376 //=======================================================================
 377 Standard_EXPORT Handle_Standard_Type& GEOMImpl_FillingDriver_Type_()
 378 {
 379
 380   static Handle_Standard_Type aType1 = STANDARD_TYPE(TFunction_Driver);
 381   if ( aType1.IsNull()) aType1 = STANDARD_TYPE(TFunction_Driver);
 382   static Handle_Standard_Type aType2 = STANDARD_TYPE(MMgt_TShared);
 383   if ( aType2.IsNull()) aType2 = STANDARD_TYPE(MMgt_TShared);
 384   static Handle_Standard_Type aType3 = STANDARD_TYPE(Standard_Transient);
 385   if ( aType3.IsNull()) aType3 = STANDARD_TYPE(Standard_Transient);
 386
 387
 388   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
 389   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_FillingDriver",
 390                                                          sizeof(GEOMImpl_FillingDriver),
 391                                                          1,
 392                                                          (Standard_Address)_Ancestors,
 393                                                          (Standard_Address)NULL);
 394
 395   return _aType;
 396 }
 397
 398 //=======================================================================
 399 //function : DownCast
 400 //purpose  :
 401 //=======================================================================
 402
 403 const Handle(GEOMImpl_FillingDriver) Handle(GEOMImpl_FillingDriver)::DownCast(const Handle(Standard_Transient)& AnObject)
 404 {
 405   Handle(GEOMImpl_FillingDriver) _anOtherObject;
 406
 407   if (!AnObject.IsNull()) {
 408      if (AnObject->IsKind(STANDARD_TYPE(GEOMImpl_FillingDriver))) {
 409        _anOtherObject = Handle(GEOMImpl_FillingDriver)((Handle(GEOMImpl_FillingDriver)&)AnObject);
 410      }
 411   }
 412
 413   return _anOtherObject;
 414 }