17.12.2013. Added Partition algorithm (draft version).

[modules/hydro.git] / src / HYDROData / HYDROData_ImmersibleZone.cxx

diff --git a/src/HYDROData/HYDROData_ImmersibleZone.cxx b/src/HYDROData/HYDROData_ImmersibleZone.cxx
index 286cbaa39a9df099166b418d0d2221189f2f7acb..eb3ae96669cf873e4ab86daa31ea92a30c380084 100644 (file)
--- a/src/HYDROData/HYDROData_ImmersibleZone.cxx
+++ b/src/HYDROData/HYDROData_ImmersibleZone.cxx
@@ -3,18 +3,24 @@
 
 #include "HYDROData_Bathymetry.h"
 #include "HYDROData_Document.h"
-#include "HYDROData_Polyline.h"
+#include "HYDROData_PolylineXY.h"
 
 #include <BRepBuilderAPI_MakeFace.hxx>
 
 #include <TopoDS.hxx>
 #include <TopoDS_Face.hxx>
 #include <TopoDS_Wire.hxx>
-
+#include <TopoDS_Compound.hxx>
+#include <TopExp_Explorer.hxx>
+#include <TopTools_ListIteratorOfListOfShape.hxx>
+#include <BRep_Builder.hxx>
+#include <BRepAlgo_FaceRestrictor.hxx>
+#include <BRepCheck_Analyzer.hxx>
 #include <QColor>
 #include <QStringList>
 
 #define PYTHON_IMMERSIBLE_ZONE_ID "KIND_IMMERSIBLE_ZONE"
+//#define HYDRODATA_IMZONE_DEB 1
 
 IMPLEMENT_STANDARD_HANDLE(HYDROData_ImmersibleZone,HYDROData_NaturalObject)
 IMPLEMENT_STANDARD_RTTIEXT(HYDROData_ImmersibleZone,HYDROData_NaturalObject)
@@ -33,7 +39,7 @@ QStringList HYDROData_ImmersibleZone::DumpToPython( MapOfTreatedObjects& theTrea
 {
   QStringList aResList;
 
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( this );
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
   if ( aDocument.IsNull() )
     return aResList;
 
@@ -63,48 +69,115 @@ QStringList HYDROData_ImmersibleZone::DumpToPython( MapOfTreatedObjects& theTrea
   Handle(HYDROData_Bathymetry) aRefBathymetry = GetBathymetry();
   setPythonReferenceObject( theTreatedObjects, aResList, aRefBathymetry, "SetBathymetry" );
 
-  Handle(HYDROData_Polyline) aRefPolyline = GetPolyline();
+  Handle(HYDROData_PolylineXY) aRefPolyline = GetPolyline();
   setPythonReferenceObject( theTreatedObjects, aResList, aRefPolyline, "SetPolyline" );
 
   return aResList;
 }
 
-TopoDS_Shape HYDROData_ImmersibleZone::GetTopShape() const
+HYDROData_SequenceOfObjects HYDROData_ImmersibleZone::GetAllReferenceObjects() const
 {
-  Handle(HYDROData_Polyline) aPolyline = GetPolyline();
-  if( !aPolyline.IsNull() )
-  {
-    TopoDS_Wire aPolylineWire = TopoDS::Wire( aPolyline->GetTopShape() );
-    if( !aPolylineWire.IsNull() )
-    {
-      BRepBuilderAPI_MakeFace aMakeFace( aPolylineWire, Standard_True );
-      aMakeFace.Build();
-      if( aMakeFace.IsDone() )
-        return aMakeFace.Face();
-    }
-  }
-  return TopoDS_Face();
+  HYDROData_SequenceOfObjects aResSeq = HYDROData_NaturalObject::GetAllReferenceObjects();
+
+  Handle(HYDROData_PolylineXY) aRefPolyline = GetPolyline();
+  if ( !aRefPolyline.IsNull() )
+    aResSeq.Append( aRefPolyline );
+
+  return aResSeq;
 }
 
-TopoDS_Shape HYDROData_ImmersibleZone::GetShape3D() const
+TopoDS_Shape HYDROData_ImmersibleZone::GetTopShape() const
 {
-  // TODO
   return getTopShape();
 }
 
-QColor HYDROData_ImmersibleZone::DefaultFillingColor()
+void HYDROData_ImmersibleZone::Update()
 {
-  return QColor( Qt::green );
+  HYDROData_NaturalObject::Update();
+  
+  TopoDS_Shape aResShape = generateTopShape();
+  SetTopShape( aResShape );
 }
 
-void HYDROData_ImmersibleZone::SetFillingColor( const QColor& theColor )
+TopoDS_Shape HYDROData_ImmersibleZone::generateTopShape() const
 {
-  return SetColor( theColor, DataTag_FillingColor );
+  TopoDS_Shape aResShape = TopoDS_Face();
+
+  Handle(HYDROData_PolylineXY) aPolyline = GetPolyline();
+  if( !aPolyline.IsNull() )
+  {
+    TopoDS_Shape aPolylineShape = aPolyline->GetShape();
+    TopTools_ListOfShape aWiresList;
+
+    if ( !aPolylineShape.IsNull() && 
+         aPolylineShape.ShapeType() == TopAbs_WIRE ) {
+      const TopoDS_Wire& aPolylineWire = TopoDS::Wire( aPolylineShape );
+      if ( !aPolylineWire.IsNull() ) {
+        BRepBuilderAPI_MakeFace aMakeFace( aPolylineWire, Standard_True );
+        aMakeFace.Build();
+        if( aMakeFace.IsDone() ) {
+          return aMakeFace.Face();
+        }
+      }
+    } else {
+           TopExp_Explorer anExp( aPolylineShape, TopAbs_WIRE );
+               for ( ; anExp.More(); anExp.Next() ) {
+                 if(!anExp.Current().IsNull()) {
+            const TopoDS_Wire& aWire = TopoDS::Wire( anExp.Current() );
+                   aWiresList.Append( aWire );
+                 }
+               }
+               if(aWiresList.IsEmpty())
+           return aResShape;
+           BRepAlgo_FaceRestrictor aFR;
+           TopoDS_Face aRefFace;         
+               TopoDS_Shape aS = aWiresList.First();
+               BRepBuilderAPI_MakeFace aMakeFace( TopoDS::Wire(aWiresList.First()), Standard_True );
+        aMakeFace.Build();
+        if( aMakeFace.IsDone() ) {
+          aRefFace = aMakeFace.Face();
+               }
+               if(aRefFace.IsNull())
+                 return aResShape;
+           aFR.Init(aRefFace,Standard_False, Standard_True);
+               TopTools_ListIteratorOfListOfShape anIt( aWiresList );
+        for ( ; anIt.More(); anIt.Next() ) {
+          TopoDS_Wire& aWire = TopoDS::Wire( anIt.Value() );
+                 if ( aWire.IsNull() ) 
+                         continue;
+          aFR.Add(aWire);
+               }
+        aFR.Perform();
+               if (aFR.IsDone()) {
+                 TopoDS_Compound aResult;
+                 BRep_Builder aBB;
+                 aBB.MakeCompound(aResult);
+                 for (; aFR.More(); aFR.Next()) {
+                       const TopoDS_Face& aFF = aFR.Current();
+                       aBB.Add(aResult,aFF);    
+                 }
+                 BRepCheck_Analyzer aChecker(aResult);
+                 if(aChecker.IsValid())
+                   aResShape = aResult;
+#ifdef HYDRODATA_IMZONE_DEB
+                 else {
+                       aResShape = aResult;             
+                 }
+#endif
+               }
+       }
+  }
+  return aResShape;
 }
 
-QColor HYDROData_ImmersibleZone::GetFillingColor() const
+TopoDS_Shape HYDROData_ImmersibleZone::GetShape3D() const
 {
-  return GetColor( DefaultFillingColor(), DataTag_FillingColor );
+  return getTopShape();
+}
+
+QColor HYDROData_ImmersibleZone::DefaultFillingColor()
+{
+  return QColor( Qt::darkBlue );
 }
 
 QColor HYDROData_ImmersibleZone::DefaultBorderColor()
@@ -112,46 +185,32 @@ QColor HYDROData_ImmersibleZone::DefaultBorderColor()
   return QColor( Qt::transparent );
 }
 
-void HYDROData_ImmersibleZone::SetBorderColor( const QColor& theColor )
+QColor HYDROData_ImmersibleZone::getDefaultFillingColor() const
 {
-  return SetColor( theColor, DataTag_BorderColor );
+  return DefaultFillingColor();
 }
 
-QColor HYDROData_ImmersibleZone::GetBorderColor() const
+QColor HYDROData_ImmersibleZone::getDefaultBorderColor() const
 {
-  return GetColor( DefaultBorderColor(), DataTag_BorderColor );
+  return DefaultBorderColor();
 }
 
-void HYDROData_ImmersibleZone::SetPolyline( const Handle(HYDROData_Polyline)& thePolyline )
+void HYDROData_ImmersibleZone::SetPolyline( const Handle(HYDROData_PolylineXY)& thePolyline )
 {
   SetReferenceObject( thePolyline, DataTag_Polyline );
+  SetToUpdate( true );
 }
 
-Handle(HYDROData_Polyline) HYDROData_ImmersibleZone::GetPolyline() const
+Handle(HYDROData_PolylineXY) HYDROData_ImmersibleZone::GetPolyline() const
 {
-  return Handle(HYDROData_Polyline)::DownCast( 
+  return Handle(HYDROData_PolylineXY)::DownCast( 
            GetReferenceObject( DataTag_Polyline ) );
 }
 
 void HYDROData_ImmersibleZone::RemovePolyline()
 {
   ClearReferenceObjects( DataTag_Polyline );
-}
-
-void HYDROData_ImmersibleZone::SetBathymetry( const Handle(HYDROData_Bathymetry)& theBathymetry )
-{
-  SetReferenceObject( theBathymetry, DataTag_Bathymetry );
-}
-
-Handle(HYDROData_Bathymetry) HYDROData_ImmersibleZone::GetBathymetry() const
-{
-  return Handle(HYDROData_Bathymetry)::DownCast( 
-           GetReferenceObject( DataTag_Bathymetry ) );
-}
-
-void HYDROData_ImmersibleZone::RemoveBathymetry()
-{
-  ClearReferenceObjects( DataTag_Bathymetry );
+  SetToUpdate( true );
 }