Salome HOME
Mantis issue 0020998: EDF 1572 GEOM: Some limitations to the PipeTShape.
[modules/geom.git] / src / GEOMImpl / GEOMImpl_FillingDriver.cxx
index c9771babec9da95ae7ffaa605cef10c4ebb97094..c70a3b04f4483c3a9f20f24d25f9217c06965aef 100644 (file)
@@ -1,21 +1,23 @@
-// Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
-// 
-// This library is free software; you can redistribute it and/or
-// modify it under the terms of the GNU Lesser General Public
-// License as published by the Free Software Foundation; either 
-// version 2.1 of the License.
-// 
-// This library is distributed in the hope that it will be useful 
-// but WITHOUT ANY WARRANTY; without even the implied warranty of 
-// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
-// Lesser General Public License for more details.
+//  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
 //
-// You should have received a copy of the GNU Lesser General Public  
-// License along with this library; if not, write to the Free Software 
-// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+//  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
 //
-// See http://www.salome-platform.org/
+//  This library is free software; you can redistribute it and/or
+//  modify it under the terms of the GNU Lesser General Public
+//  License as published by the Free Software Foundation; either
+//  version 2.1 of the License.
+//
+//  This library is distributed in the hope that it will be useful,
+//  but WITHOUT ANY WARRANTY; without even the implied warranty of
+//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+//  Lesser General Public License for more details.
+//
+//  You should have received a copy of the GNU Lesser General Public
+//  License along with this library; if not, write to the Free Software
+//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+//  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
 //
 
 #include <Standard_Stream.hxx>
 #include <BRep_Tool.hxx>
 #include <BRepAlgo.hxx>
 #include <BRepBuilderAPI_MakeFace.hxx>
+#include <BRep_Builder.hxx>
+#include <BRepBuilderAPI_Copy.hxx>
 
 #include <TopAbs.hxx>
 #include <TopoDS.hxx>
 #include <TopoDS_Shape.hxx>
+#include <TopoDS_Compound.hxx>
+#include <TopoDS_Edge.hxx>
+#include <TopoDS_Vertex.hxx>
 #include <TopExp_Explorer.hxx>
 
 #include <Geom_Curve.hxx>
 #include <Geom_Surface.hxx>
 #include <Geom_TrimmedCurve.hxx>
+#include <Geom_Line.hxx>
+#include <Geom_Circle.hxx>
+#include <Geom_Ellipse.hxx>
+#include <Geom_BezierCurve.hxx>
 #include <Geom_BSplineSurface.hxx>
 #include <GeomFill_Line.hxx>
 #include <GeomFill_AppSurf.hxx>
 #include <Precision.hxx>
 #include <Standard_ConstructionError.hxx>
 
+#include <TColGeom_SequenceOfCurve.hxx>
+#include <ShapeFix_Face.hxx>
+#include <GeomAPI_PointsToBSplineSurface.hxx>
+#include <Geom_BSplineCurve.hxx>
+#include <GeomAPI_PointsToBSpline.hxx>
+
+#include <TColgp_SequenceOfPnt.hxx>
+#include <TColgp_Array1OfPnt.hxx>
+
+//#include <BRepTools.hxx>
+
+
 //=======================================================================
 //function : GetID
 //purpose  :
@@ -80,7 +103,12 @@ Standard_Integer GEOMImpl_FillingDriver::Execute(TFunction_Logbook& log) const
   GEOMImpl_IFilling IF (aFunction);
   Handle(GEOM_Function) aShapeFunction = IF.GetShape();
   if (aShapeFunction.IsNull()) return 0;
-  TopoDS_Shape aShape = aShapeFunction->GetValue();
+  TopoDS_Shape aShape;
+
+  BRepBuilderAPI_Copy Copy(aShapeFunction->GetValue());
+  if( Copy.IsDone() )
+    aShape = Copy.Shape();
+
   if (aShape.IsNull() || aShape.ShapeType() != TopAbs_COMPOUND) return 0;
 
   Standard_Integer mindeg = IF.GetMinDeg();
@@ -88,6 +116,8 @@ Standard_Integer GEOMImpl_FillingDriver::Execute(TFunction_Logbook& log) const
   Standard_Real tol3d = IF.GetTol2D();
   Standard_Real tol2d = IF.GetTol3D();
   Standard_Integer nbiter = IF.GetNbIter();
+  Standard_Boolean isApprox = IF.GetApprox();
+  Standard_Integer aMethod = IF.GetMethod();
 
   if (mindeg > maxdeg) {
     Standard_RangeError::Raise("Minimal degree can not be more than maximal degree");
@@ -98,34 +128,234 @@ Standard_Integer GEOMImpl_FillingDriver::Execute(TFunction_Logbook& log) const
   TopoDS_Shape Scurrent;
   Standard_Real First, Last;
   Handle(Geom_Curve) C;
-  GeomFill_SectionGenerator Section;
-
-  Standard_Integer i = 0;
-  for (Ex.Init(aShape, TopAbs_EDGE); Ex.More(); Ex.Next()) {
-    Scurrent = Ex.Current() ;
-    if (Scurrent.IsNull() || Scurrent.ShapeType() != TopAbs_EDGE) return 0;
-    C = BRep_Tool::Curve(TopoDS::Edge(Scurrent), First, Last);
-    C = new Geom_TrimmedCurve(C, First, Last);
-    Section.AddCurve(C);
-    i++;
-  }
-
-  /* a 'tolerance' is used to compare 2 knots : see GeomFill_Generator.cdl */
-  Section.Perform(Precision::Confusion());
-  Handle(GeomFill_Line) Line = new GeomFill_Line(i);
-
-  GeomFill_AppSurf App (mindeg, maxdeg, tol3d, tol2d, nbiter); /* user parameters */
-  App.Perform(Line, Section);
 
-  if (!App.IsDone()) return 0;
-  Standard_Integer UDegree, VDegree, NbUPoles, NbVPoles, NbUKnots, NbVKnots;
-  App.SurfShape(UDegree, VDegree, NbUPoles, NbVPoles, NbUKnots, NbVKnots);
-  Handle(Geom_BSplineSurface) GBS = new Geom_BSplineSurface
-    (App.SurfPoles(), App.SurfWeights(), App.SurfUKnots(), App.SurfVKnots(),
-     App.SurfUMults(), App.SurfVMults(), App.UDegree(), App.VDegree());
-
-  if (GBS.IsNull()) return 0;
-  aShape = BRepBuilderAPI_MakeFace(GBS);
+  TopoDS_Compound aComp;
+  BRep_Builder B;
+  B.MakeCompound(aComp);
+
+  TopoDS_Iterator It (aShape);
+  for (; It.More(); It.Next()) {
+    Scurrent = It.Value();
+    if (Scurrent.ShapeType() != TopAbs_EDGE) {
+      Handle(Geom_BSplineCurve) newC;
+      if (Scurrent.ShapeType() == TopAbs_WIRE) {
+        TColgp_SequenceOfPnt PntSeq;
+        // collect points
+        for (Ex.Init(Scurrent, TopAbs_EDGE); Ex.More(); Ex.Next()) {
+          TopoDS_Edge E = TopoDS::Edge(Ex.Current());
+          if (BRep_Tool::Degenerated(E)) continue;
+          C = BRep_Tool::Curve(E, First, Last);
+          if( E.Orientation() == TopAbs_REVERSED ) {
+            C->Reverse();
+          }
+          Handle(Geom_TrimmedCurve) tc = Handle(Geom_TrimmedCurve)::DownCast(C);
+          while( !tc.IsNull() ) {
+            C = tc->BasisCurve();
+            tc = Handle(Geom_TrimmedCurve)::DownCast(C);
+          }
+          int nbp = 10;
+          if( C->IsKind(STANDARD_TYPE(Geom_Line)) ) {
+            nbp = 4;
+          }
+          else if( C->IsKind(STANDARD_TYPE(Geom_Circle)) || 
+                   C->IsKind(STANDARD_TYPE(Geom_Ellipse)) ) {
+            nbp = (int)25*fabs(Last-First)/(2*PI);
+          }
+          else if( C->IsKind(STANDARD_TYPE(Geom_BezierCurve)) ) {
+            Handle(Geom_BezierCurve) C3d = Handle(Geom_BezierCurve)::DownCast(C);
+            nbp = C3d->NbPoles();
+          }
+          else if( C->IsKind(STANDARD_TYPE(Geom_BSplineCurve)) ) {
+            Handle(Geom_BSplineCurve) C3d = Handle(Geom_BSplineCurve)::DownCast(C);
+            nbp = C3d->NbPoles();
+          }
+          else {
+          }
+          if( nbp<4 ) nbp = 4;
+          double dp = (Last-First)/(nbp-1);
+          for(int i=1; i<nbp; i++) {
+            gp_Pnt P;
+            C->D0(First+dp*(i-1),P);
+            PntSeq.Append(P);
+          }
+        }
+        // add last point
+        gp_Pnt P;
+        C->D0(Last,P);
+        PntSeq.Append(P);
+        // create BSpline 
+        if(PntSeq.Length()>1) {
+          TColgp_Array1OfPnt Pnts(1,PntSeq.Length());
+          // check orientation of wire
+          if( Scurrent.Orientation() == TopAbs_REVERSED ) {
+            for(int i=1; i<=PntSeq.Length(); i++) {
+              Pnts.SetValue(PntSeq.Length()-i+1,PntSeq.Value(i));
+            }
+          }
+          else {
+            for(int i=1; i<=PntSeq.Length(); i++) {
+              Pnts.SetValue(i,PntSeq.Value(i));
+            }
+          }
+          GeomAPI_PointsToBSpline PTB(Pnts);
+          newC = Handle(Geom_BSplineCurve)::DownCast(PTB.Curve());
+          // set periodic flag if curve is closed
+          //if( newC->IsClosed() ) {
+          //  newC->SetPeriodic();
+          //}
+          // create edge
+          double fp = newC->FirstParameter();
+          double lp = newC->FirstParameter();
+          gp_Pnt PF,PL;
+          newC->D0(fp,PF);
+          newC->D0(lp,PL);
+          TopoDS_Vertex VF,VL;
+          B.MakeVertex(VF,PF,1.e-7);
+          B.MakeVertex(VL,PL,1.e-7);
+          TopoDS_Edge newE;
+          B.MakeEdge(newE,newC,1.e-7);
+          B.Add(newE,VF);
+          B.Add(newE,VL.Reversed());
+          Scurrent = newE;
+        }
+      }
+      if(newC.IsNull()) {
+        Standard_ConstructionError::Raise("The argument compound must contain only edges");
+      }
+    }
+    B.Add(aComp,Scurrent);
+  }
+  aShape = aComp;
+
+  if (!isApprox) {
+    // make filling as in old version of SALOME (before 4.1.1)
+    GeomFill_SectionGenerator Section;
+    Standard_Integer i = 0;
+    Handle(Geom_Curve) aLastC;
+    gp_Pnt PL1,PL2;
+    for (Ex.Init(aShape, TopAbs_EDGE); Ex.More(); Ex.Next()) {
+      Scurrent = Ex.Current();
+      if (Scurrent.IsNull() || Scurrent.ShapeType() != TopAbs_EDGE) return 0;
+      if (BRep_Tool::Degenerated(TopoDS::Edge(Scurrent))) continue;
+      C = BRep_Tool::Curve(TopoDS::Edge(Scurrent), First, Last);
+      //if (Scurrent.Orientation() == TopAbs_REVERSED)
+      //  // Mantis isuue 0020659: consider the orientation of the edges
+      //  C = new Geom_TrimmedCurve(C, Last, First);
+      //else
+      //  C = new Geom_TrimmedCurve(C, First, Last);
+      C = new Geom_TrimmedCurve(C, First, Last);
+      gp_Pnt P1,P2;
+      C->D0(First,P1);
+      C->D0(Last,P2);
+
+      if( aMethod==1 && Scurrent.Orientation() == TopAbs_REVERSED ) {
+        C->Reverse();
+      }
+      else if( aMethod==2 ) {
+        if( i==0 ) {
+          PL1 = P1;
+          PL2 = P2;
+        }
+        else {
+          double d1 = PL1.Distance(P1) + PL2.Distance(P2);
+          double d2 = PL1.Distance(P2) + PL2.Distance(P1);
+          if(d2<d1) {
+            C->Reverse();
+            PL1 = P2;
+            PL2 = P1;
+          }
+          else {
+            PL1 = P1;
+            PL2 = P2;
+          }
+        }
+      }
+
+      Section.AddCurve(C);
+      i++;
+    }
+
+    /* a 'tolerance' is used to compare 2 knots : see GeomFill_Generator.cdl */
+    Section.Perform(Precision::Confusion());
+    Handle(GeomFill_Line) Line = new GeomFill_Line(i);
+
+    GeomFill_AppSurf App (mindeg, maxdeg, tol3d, tol2d, nbiter); /* user parameters */
+    App.Perform(Line, Section);
+
+    if (!App.IsDone()) return 0;
+    Standard_Integer UDegree, VDegree, NbUPoles, NbVPoles, NbUKnots, NbVKnots;
+    App.SurfShape(UDegree, VDegree, NbUPoles, NbVPoles, NbUKnots, NbVKnots);
+    Handle(Geom_BSplineSurface) GBS = new Geom_BSplineSurface
+      (App.SurfPoles(), App.SurfWeights(), App.SurfUKnots(), App.SurfVKnots(),
+       App.SurfUMults(), App.SurfVMults(), App.UDegree(), App.VDegree());
+
+    if (GBS.IsNull()) return 0;
+    aShape = BRepBuilderAPI_MakeFace(GBS);
+  }
+  else {
+    // implemented by skl 20.03.2008 for bug 16568
+    // make approximation - try to create bspline surface
+    // using GeomAPI_PointsToBSplineSurface
+    TColGeom_SequenceOfCurve aSeq;
+    int MaxNbPoles = 0;
+    // add curves from edges to sequence and find maximal
+    // number of poles if some of them are bsplines
+    for (Ex.Init(aShape, TopAbs_EDGE); Ex.More(); Ex.Next()) {
+      Scurrent = Ex.Current();
+      if (Scurrent.IsNull() || Scurrent.ShapeType() != TopAbs_EDGE) return 0;
+      if (BRep_Tool::Degenerated(TopoDS::Edge(Scurrent))) continue;
+      C = BRep_Tool::Curve(TopoDS::Edge(Scurrent), First, Last);
+      Handle(Geom_TrimmedCurve) TC = Handle(Geom_TrimmedCurve)::DownCast(C);
+      if(TC.IsNull()) {
+        Handle(Geom_BSplineCurve) BC = Handle(Geom_BSplineCurve)::DownCast(C);
+        if(!BC.IsNull()) {
+          MaxNbPoles = Max(MaxNbPoles,BC->NbPoles());
+        }
+      }
+      else {
+        Handle(Geom_BSplineCurve) BC = Handle(Geom_BSplineCurve)::DownCast(TC->BasisCurve());
+        if(BC.IsNull()) {
+          Handle(Geom_TrimmedCurve) TC1 = Handle(Geom_TrimmedCurve)::DownCast(TC->BasisCurve());
+          if(!TC1.IsNull()) {
+            BC = Handle(Geom_BSplineCurve)::DownCast(TC1->BasisCurve());
+          }
+        }
+        if(!BC.IsNull()) {
+          MaxNbPoles = Max(MaxNbPoles,BC->NbPoles());
+        }
+      }
+      aSeq.Append(C);
+    }
+    // prepare array of points for creation bspline surface
+    // size of this array: by U parameter - number of curves,
+    // by V parameter - determ using MaxNbPoles but it's
+    // value must be between 21(min) and 101(max)
+    int nbc = aSeq.Length();
+    int nbp = Max(21,2*MaxNbPoles-1);
+    if(nbp>101) nbp = 101;
+    TColgp_Array2OfPnt Points(1,nbc,1,nbp);
+    int ic = 1;
+    for(; ic<=nbc; ic++) {
+      Handle(Geom_Curve) C = aSeq.Value(ic);
+      double fp = C->FirstParameter();
+      double lp = C->LastParameter();
+      double dp = (lp-fp)/(nbp-1);
+      int j = 0;
+      gp_Pnt P;
+      for(; j<nbp; j++) {
+        C->D0(fp+dp*j,P);
+        Points.SetValue(ic,j+1,P);
+      }
+    }
+    GeomAPI_PointsToBSplineSurface PTB(Points,mindeg,maxdeg,GeomAbs_C2,tol3d);
+    Handle(Geom_BSplineSurface) BS = PTB.Surface();
+    BRepBuilderAPI_MakeFace BB(BS);
+    TopoDS_Face NewF = BB.Face();
+    Handle(ShapeFix_Face) sff = new ShapeFix_Face(NewF);
+    sff->Perform();
+    sff->FixOrientation();
+    aShape = sff->Face();
+  }
 
   /* We test the validity of resulting shape */
   if (!BRepAlgo::IsValid((aShape))) {
@@ -157,10 +387,10 @@ Standard_EXPORT Handle_Standard_Type& GEOMImpl_FillingDriver_Type_()
 
   static Handle_Standard_Transient _Ancestors[]= {aType1,aType2,aType3,NULL};
   static Handle_Standard_Type _aType = new Standard_Type("GEOMImpl_FillingDriver",
-                                                        sizeof(GEOMImpl_FillingDriver),
-                                                        1,
-                                                        (Standard_Address)_Ancestors,
-                                                        (Standard_Address)NULL);
+                                                         sizeof(GEOMImpl_FillingDriver),
+                                                         1,
+                                                         (Standard_Address)_Ancestors,
+                                                         (Standard_Address)NULL);
 
   return _aType;
 }
@@ -180,7 +410,5 @@ const Handle(GEOMImpl_FillingDriver) Handle(GEOMImpl_FillingDriver)::DownCast(co
      }
   }
 
-  return _anOtherObject ;
+  return _anOtherObject;
 }
-
-