Salome HOME
TShell cannot be disconnected; use TCompound of shells/faces instead
[modules/hydro.git] / src / HYDROData / HYDROData_ShapeFile.cxx
index 969bba9c286e53e64f5465888772ca3384cd7de5..b6a4e737c336a6bff237188c8653a794f6831e70 100644 (file)
@@ -33,6 +33,7 @@
 #include <TopoDS_Edge.hxx>
 #include <TopoDS_Face.hxx>
 #include <BRep_Tool.hxx>
+#include <BRepTools.hxx>
 #include <Precision.hxx>
 #include <Handle_Geom_Curve.hxx>
 #include <Handle_Geom_Line.hxx>
@@ -47,6 +48,7 @@
 #include <TopTools_SequenceOfShape.hxx>
 #include <QColor>
 #include <BRepTopAdaptor_FClass2d.hxx>
+#include <TopExp.hxx>
 
 #ifdef WIN32
   #pragma warning( disable: 4996 )
@@ -64,7 +66,6 @@ HYDROData_ShapeFile::~HYDROData_ShapeFile()
 void HYDROData_ShapeFile::Export(const QString& aFileName, 
   NCollection_Sequence<Handle_HYDROData_PolylineXY> aPolyXYSeq,
   NCollection_Sequence<Handle_HYDROData_Polyline3D> aPoly3DSeq,
-  const Handle_HYDROData_LandCoverMap& aLCM,
   QStringList& aNonExpList)
 {
   SHPHandle hSHPHandle;
@@ -74,7 +75,6 @@ void HYDROData_ShapeFile::Export(const QString& aFileName,
     for (int i = 1; i <= aPolyXYSeq.Size(); i++)
       if (WriteObjectPolyXY(hSHPHandle, aPolyXYSeq(i)) != 1)
         aNonExpList.append(aPolyXYSeq(i)->GetName());
-
   }
   else if (aPolyXYSeq.IsEmpty() && !aPoly3DSeq.IsEmpty())
   {
@@ -83,19 +83,31 @@ void HYDROData_ShapeFile::Export(const QString& aFileName,
       if (WriteObjectPoly3D(hSHPHandle, aPoly3DSeq(i)) != 1)
         aNonExpList.append(aPoly3DSeq(i)->GetName());
   }
-  else if (aPolyXYSeq.IsEmpty() && aPoly3DSeq.IsEmpty() && !aLCM->IsEmpty())
+  if (hSHPHandle->nRecords > 0)
+    SHPClose( hSHPHandle );
+  else
+  {
+    SHPClose( hSHPHandle );
+    QString aFN = aFileName.simplified();
+    remove (aFN.toStdString().c_str());
+    remove (aFN.replace( ".shp", ".shx", Qt::CaseInsensitive).toStdString().c_str());
+  }
+}
+
+void HYDROData_ShapeFile::Export(const QString& aFileName,
+  const Handle_HYDROData_LandCoverMap& aLCM, QStringList& aNonExpList)
+{
+  SHPHandle hSHPHandle;
+  if ( !aLCM.IsNull() && !aLCM->IsEmpty())
   {
     hSHPHandle = SHPCreate( aFileName.toAscii().data(), SHPT_POLYGON );
-    HYDROData_LandCoverMap::Iterator It( aLCM );
+    HYDROData_LandCoverMap::Explorer It( aLCM );
     for( ; It.More(); It.Next() )
     {
       TopoDS_Face aFace = It.Face();
       if (WriteObjectPolygon(hSHPHandle, aFace) != 1)
         aNonExpList.append(aLCM->GetName() + "_" +  QString::number(It.Index()));
     }
-    /*for (int i = 1; i <= aLCSeq.Size(); i++)
-      if (WriteObjectLC(hSHPHandle, aLCSeq(i)) != 1)
-        aNonExpList.append(aLCSeq(i)->GetName());*/
   }
   if (hSHPHandle->nRecords > 0)
     SHPClose( hSHPHandle );
@@ -108,6 +120,7 @@ void HYDROData_ShapeFile::Export(const QString& aFileName,
   }
 }
 
+
 int HYDROData_ShapeFile::WriteObjectPolyXY(SHPHandle theShpHandle, Handle_HYDROData_PolylineXY thePoly )
 {
   SHPObject    *aSHPObj;
@@ -224,53 +237,88 @@ int HYDROData_ShapeFile::WriteObjectPolygon(SHPHandle theShpHandle, const TopoDS
 
 void HYDROData_ShapeFile::ProcessFace(TopoDS_Face theFace, SHPHandle theShpHandle)
 {
+  if (theFace.ShapeType() != TopAbs_FACE)
+     return;
   SHPObject    *aSHPObj;
   std::vector<double> x, y;
   std::vector<int> anPartStart;
-  if (theFace.ShapeType() == TopAbs_FACE)
+  TopoDS_Wire OuterW = BRepTools::OuterWire(theFace);
+  NCollection_Sequence<TopoDS_Wire> aWires;
+
+  //write an outer wire first
+  aWires.Append(OuterW); 
+  TopExp_Explorer Ex(theFace, TopAbs_WIRE);  
+  for (; Ex.More(); Ex.Next()) 
   {
-    TopExp_Explorer Ex(theFace, TopAbs_WIRE);
-    int NbWires = 0;
-    for (; Ex.More(); Ex.Next()) 
+    TopoDS_Wire aW = TopoDS::Wire(Ex.Current());
+    if (aW.IsEqual(OuterW))
+      continue;
+    aWires.Append(aW); 
+  }
+
+  //TopExp_Explorer Ex(theFace, TopAbs_WIRE);
+  int NbWires = 0;
+  for (int k = 1; k <= aWires.Length(); k++) 
+  {
+    TopoDS_Wire aW = aWires(k);   
+    if (aW.IsNull())
+      continue;
+    NbWires++;
+    // Try to reorder edges
+    Handle(ShapeFix_Wire) aSFW = new ShapeFix_Wire( aW, theFace, Precision::Confusion() );
+    aSFW->FixReorder();
+    Handle(ShapeExtend_WireData) aSEWD = aSFW->WireData();
+    Standard_Integer nbE = aSEWD->NbEdges();
+    //
+    anPartStart.push_back(x.size());
+    NCollection_Sequence<gp_Pnt2d> aPnts;
+    for (Standard_Integer i = 1; i <= nbE; i++)
     {
-      TopoDS_Wire aW = TopoDS::Wire(Ex.Current());   
-      if (aW.IsNull())
+      TopoDS_Edge E = aSEWD->Edge(i);
+      TopoDS_Vertex aV = TopExp::LastVertex(E, 1);
+      if (aV.IsNull())
         continue;
-      NbWires++;
-      anPartStart.push_back(x.size());
-      TopExp_Explorer aVEx(aW, TopAbs_VERTEX);
-      NCollection_Sequence<gp_Pnt2d> aPnts;
-      for (; aVEx.More(); aVEx.Next())
-      {
-        TopoDS_Vertex aV = TopoDS::Vertex(aVEx.Current()); 
-        if (aV.IsNull())
-          continue;
-        gp_Pnt P = BRep_Tool::Pnt(aV);
-        aPnts.Append(gp_Pnt2d(P.X(), P.Y()));
-      }
-      NCollection_Sequence<gp_Pnt2d> aNPnts;
-      aNPnts.Append(aPnts.First());
-      for (int j = 1; j <= aPnts.Size() - 1; j++)
-      {
-        if (!aPnts(j).IsEqual(aPnts(j + 1), Precision::Confusion())) 
-          aNPnts.Append(aPnts(j + 1));
-      }
+      gp_Pnt P = BRep_Tool::Pnt(aV);
+      aPnts.Append(gp_Pnt2d(P.X(), P.Y()));
+    }
+    NCollection_Sequence<gp_Pnt2d> aNPnts;
+    aNPnts.Append(aPnts.First());
+    for (int j = 1; j <= aPnts.Size() - 1; j++)
+    {
+      if (!aPnts(j).IsEqual(aPnts(j + 1), Precision::Confusion())) 
+        aNPnts.Append(aPnts(j + 1));
+    }
 
+    if (k == 1)
+    {
+      //an outer wire
+      //clockwise direction
       for (int j = 1; j <= aNPnts.Size(); j++)
       { 
         x.push_back( aNPnts(j).X());
         y.push_back( aNPnts(j).Y()); 
       }
-      //x.push_back( aNPnts(1).X());
-      //y.push_back( aNPnts(1).Y()); 
+      x.push_back( aNPnts(1).X());
+      y.push_back( aNPnts(1).Y()); 
+    }
+    else
+    {
+      //holes
+      //counter-clockwise direction
+      for (int j = aNPnts.Size(); j > 0; j--)
+      { 
+        x.push_back( aNPnts(j).X());
+        y.push_back( aNPnts(j).Y()); 
+      }
+      x.push_back( aNPnts(aNPnts.Size()).X());
+      y.push_back( aNPnts(aNPnts.Size()).Y()); 
     }
-    
-    aSHPObj = SHPCreateObject( SHPT_POLYGON, -1, NbWires, &anPartStart[0], NULL, x.size(), &x[0], &y[0], NULL, NULL );
-    SHPWriteObject( theShpHandle, -1, aSHPObj );
-    SHPDestroyObject( aSHPObj );
   }
-  else
-    return;
+  
+  aSHPObj = SHPCreateObject( SHPT_POLYGON, -1, NbWires, &anPartStart[0], NULL, x.size(), &x[0], &y[0], NULL, NULL );
+  SHPWriteObject( theShpHandle, -1, aSHPObj );
+  SHPDestroyObject( aSHPObj );
+
 }
 
 bool HYDROData_ShapeFile::Parse(SHPHandle theHandle, ShapeType theType, int& theShapeTypeOfFile)