]> SALOME platform Git repositories - modules/hydro.git/blobdiff - src/HYDROData/HYDROData_CalculationCase.cxx
Salome HOME
Create goups for stream.
[modules/hydro.git] / src / HYDROData / HYDROData_CalculationCase.cxx
index f8d25eb746999b52c0aaa5e42442cedb827d18a9..d40387eee159af849077abd59f4ff15e3d5ee2ec 100644 (file)
@@ -4,18 +4,20 @@
 #include "HYDROData_ArtificialObject.h"
 #include "HYDROData_Bathymetry.h"
 #include "HYDROData_Document.h"
-#include "HYDROData_EdgesGroup.h"
+#include "HYDROData_ShapesGroup.h"
 #include "HYDROData_Iterator.h"
 #include "HYDROData_NaturalObject.h"
 #include "HYDROData_PolylineXY.h"
 #include "HYDROData_SplitToZonesTool.h"
-#include "HYDROData_SplittedEdgesGroup.h"
+#include "HYDROData_SplittedShapesGroup.h"
 #include "HYDROData_Region.h"
 #include "HYDROData_Tool.h"
 #include "HYDROData_Zone.h"
 
 #include <GEOMBase.h>
 
+#include <QSet>
+
 #include <TopoDS.hxx>
 #include <TopoDS_Shell.hxx>
 #include <TopoDS_Edge.hxx>
@@ -98,8 +100,8 @@ void HYDROData_CalculationCase::SetName( const QString& theName )
     anIter.Init( aGroups );
     for ( ; anIter.More(); anIter.Next() )
     {
-      Handle(HYDROData_SplittedEdgesGroup) aGroup =
-        Handle(HYDROData_SplittedEdgesGroup)::DownCast( anIter.Value() );
+      Handle(HYDROData_SplittedShapesGroup) aGroup =
+        Handle(HYDROData_SplittedShapesGroup)::DownCast( anIter.Value() );
       if ( aGroup.IsNull() )
         continue;
 
@@ -204,7 +206,7 @@ void HYDROData_CalculationCase::Update()
   QString aRegsPref = CALCULATION_REGIONS_PREF;
   QString aZonesPref = CALCULATION_ZONES_PREF;
 
-  QMap<QString,Handle(HYDROData_SplittedEdgesGroup)> aSplittedEdgesGroupsMap;
+  QMap<QString,Handle(HYDROData_SplittedShapesGroup)> aSplittedEdgesGroupsMap;
 
   // Create result regions for case, by default one zone for one region
   HYDROData_SplitToZonesTool::SplitDataListIterator anIter( aSplitedObjects );
@@ -244,15 +246,14 @@ void HYDROData_CalculationCase::Update()
     else if ( aSplitData.Type == HYDROData_SplitToZonesTool::SplitData::Data_Edge )
     {
       // Create new edges group
-      if ( aSplitData.ObjectNames.isEmpty() || 
-           aSplitData.Shape.IsNull() || aSplitData.Shape.ShapeType() != TopAbs_EDGE )
+      if ( aSplitData.ObjectNames.isEmpty() || aSplitData.Shape.IsNull() )
         continue;
 
       QString anObjName = aSplitData.ObjectNames.first();
       if ( anObjName.isEmpty() )
         continue;
 
-      Handle(HYDROData_SplittedEdgesGroup) aSplittedGroup;
+      Handle(HYDROData_SplittedShapesGroup) aSplittedGroup;
       if ( !aSplittedEdgesGroupsMap.contains( anObjName ) )
       {
         aSplittedGroup = addNewSplittedGroup();
@@ -270,8 +271,7 @@ void HYDROData_CalculationCase::Update()
       if ( aSplittedGroup.IsNull() )
         continue;
 
-      TopoDS_Edge anEdge = TopoDS::Edge( aSplitData.Shape );
-      aSplittedGroup->AddEdge( anEdge );
+      aSplittedGroup->AddShape( aSplitData.Shape );
     }
   }
 }
@@ -316,7 +316,7 @@ void HYDROData_CalculationCase::RemoveGeometryObjects()
   SetToUpdate( true );
 }
 
-bool HYDROData_CalculationCase::AddGeometryGroup( const Handle(HYDROData_EdgesGroup)& theGroup )
+bool HYDROData_CalculationCase::AddGeometryGroup( const Handle(HYDROData_ShapesGroup)& theGroup )
 {
   if ( theGroup.IsNull() )
     return false;
@@ -337,7 +337,7 @@ HYDROData_SequenceOfObjects HYDROData_CalculationCase::GetGeometryGroups() const
   return GetReferenceObjects( DataTag_GeometryGroup );
 }
 
-void HYDROData_CalculationCase::RemoveGeometryGroup( const Handle(HYDROData_EdgesGroup)& theGroup )
+void HYDROData_CalculationCase::RemoveGeometryGroup( const Handle(HYDROData_ShapesGroup)& theGroup )
 {
   if ( theGroup.IsNull() )
     return;
@@ -499,137 +499,6 @@ void HYDROData_CalculationCase::RemoveSplittedGroups()
   myLab.FindChild( DataTag_SplittedGroups ).ForgetAllAttributes();
 }
 
-TopoDS_Shell HYDROData_CalculationCase::GetShell()
-{
-  TopoDS_Shell aShell;
-
-  TopTools_ListOfShape aFacesList;
-
-  // Make shell containing all region shapes
-  BRepBuilderAPI_Sewing aSewing( Precision::Confusion()*10.0 );
-
-  HYDROData_SequenceOfObjects aCaseRegions = GetRegions();
-  HYDROData_SequenceOfObjects::Iterator aRegionIter( aCaseRegions );
-  for ( ; aRegionIter.More(); aRegionIter.Next() ) {
-    Handle(HYDROData_Region) aRegion =
-      Handle(HYDROData_Region)::DownCast( aRegionIter.Value() );
-    if( aRegion.IsNull() ) {
-      continue;
-    }
-
-    TopoDS_Shape aRegionShape = aRegion->GetShape();
-    if( !aRegionShape.IsNull() ) {
-      if ( aRegionShape.ShapeType() == TopAbs_FACE ) {
-        TopoDS_Face aFace = TopoDS::Face( aRegionShape );
-        if ( !aFace.IsNull() ) {
-          aFacesList.Append( aFace );
-          aSewing.Add( aFace );
-        }
-      } else {
-        TopExp_Explorer anExp( aRegionShape, TopAbs_FACE );
-        for ( ; anExp.More(); anExp.Next() ) {
-          TopoDS_Face aFace = TopoDS::Face( anExp.Current() );
-          if ( !aFace.IsNull() ) {
-            aFacesList.Append( aFace );
-            aSewing.Add( aFace );
-          }
-        }
-      }
-    }
-  } // regions iterator
-  
-  aSewing.Perform();
-  TopoDS_Shape aSewedShape = aSewing.SewedShape();
-
-  if ( !aSewedShape.IsNull() )
-  {
-    if ( aSewedShape.ShapeType() == TopAbs_FACE && aCaseRegions.Length() ==1 ) {
-      // create shell from one face
-      BRep_Builder aBuilder;
-      aBuilder.MakeShell( aShell );
-      aBuilder.Add( aShell, aSewedShape);
-    } else {
-      TopExp_Explorer anExpShells( aSewedShape, TopAbs_SHELL );
-      Standard_Integer aNbOfShells = 0;
-      for ( ; anExpShells.More(); anExpShells.Next() ) {
-        aShell = TopoDS::Shell( anExpShells.Current() );
-        aNbOfShells++;
-      }
-
-      if ( aNbOfShells != 1 ) {
-        aShell.Nullify();
-        BRep_Builder aBuilder;
-        aBuilder.MakeShell( aShell );
-
-        TopExp_Explorer anExpFaces( aSewedShape, TopAbs_FACE );
-        for ( ; anExpFaces.More(); anExpFaces.Next() ) {
-          TopoDS_Face aFace = TopoDS::Face( anExpFaces.Current() );
-          if ( !aFace.IsNull() ) {
-            aBuilder.Add( aShell, aFace );
-          }
-        }
-      }
-    }
-  }
-
-  if ( !aShell.IsNull() ) {
-    TopTools_IndexedMapOfShape aMapOfFaces;
-    TopExp::MapShapes( aShell, TopAbs_FACE, aMapOfFaces );
-    if ( aMapOfFaces.Extent() != aFacesList.Extent() ) {
-      aShell.Nullify();
-      BRep_Builder aBuilder;
-      aBuilder.MakeShell( aShell );
-
-      TopTools_ListIteratorOfListOfShape anIter( aFacesList );
-      for ( ; anIter.More(); anIter.Next() ) {
-        TopoDS_Face aFace = TopoDS::Face( anIter.Value() );
-        aBuilder.Add( aShell, aFace );
-      }
-    }
-  }
-
-/* TODO: old version
-  // Make shell
-  BRep_Builder aBuilder;
-  aBuilder.MakeShell( aShell );
-
-  // Make shell containing all region shapes
-  HYDROData_SequenceOfObjects aCaseRegions = GetRegions();
-  HYDROData_SequenceOfObjects::Iterator aRegionIter( aCaseRegions );
-  for ( ; aRegionIter.More(); aRegionIter.Next() ) {
-    Handle(HYDROData_Region) aRegion =
-      Handle(HYDROData_Region)::DownCast( aRegionIter.Value() );
-    if( aRegion.IsNull() ) {
-      continue;
-    }
-
-    TopoDS_Shape aRegionShape = aRegion->GetShape();
-
-    // Add shape (face or shell) corresponding to the region into the shell
-    if( !aRegionShape.IsNull() ) {
-      if ( aRegionShape.ShapeType() == TopAbs_FACE ) {
-        aBuilder.Add( aShell, aRegionShape );
-      } else {
-        TopExp_Explorer anExp( aRegionShape, TopAbs_FACE );
-        for( ; anExp.More(); anExp.Next() ) {
-          TopoDS_Face aFace = TopoDS::Face( anExp.Current() );
-          if( !aFace.IsNull() ) {
-            aBuilder.Add( aShell, aFace );
-          }
-        }
-      }
-    }
-  } // regions iterator
-*/
-
-  // Nullify shell if it is empty
-  if ( !aShell.IsNull() && !TopoDS_Iterator(aShell).More() ) {
-    aShell.Nullify();
-  }
-
-  return aShell;
-}
-
 double HYDROData_CalculationCase::GetAltitudeForPoint( const gp_XY& thePoint ) const
 {
   double aResAltitude = HYDROData_Bathymetry::GetInvalidAltitude();
@@ -766,12 +635,12 @@ Handle(HYDROData_Region) HYDROData_CalculationCase::addNewRegion()
   return aNewRegion;
 }
 
-Handle(HYDROData_SplittedEdgesGroup) HYDROData_CalculationCase::addNewSplittedGroup()
+Handle(HYDROData_SplittedShapesGroup) HYDROData_CalculationCase::addNewSplittedGroup()
 {
   TDF_Label aNewLab = myLab.FindChild( DataTag_SplittedGroups ).NewChild();
 
-  Handle(HYDROData_SplittedEdgesGroup) aNewGroup =
-    Handle(HYDROData_SplittedEdgesGroup)::DownCast( 
+  Handle(HYDROData_SplittedShapesGroup) aNewGroup =
+    Handle(HYDROData_SplittedShapesGroup)::DownCast( 
       HYDROData_Iterator::CreateObject( aNewLab, KIND_SPLITTED_GROUP ) );
   AddReferenceObject( aNewGroup, DataTag_SplittedGroups );
 
@@ -782,12 +651,24 @@ bool HYDROData_CalculationCase::Export( GEOM::GEOM_Gen_var theGeomEngine,
                                         SALOMEDS::Study_ptr theStudy )
 {
   // Get faces
-  // TODO
+  // TODO to be revised
   TopTools_ListOfShape aFaces;
+  HYDROData_SequenceOfObjects aCaseRegions = GetRegions();
+  HYDROData_SequenceOfObjects::Iterator aRegionIter( aCaseRegions );
+  for ( ; aRegionIter.More(); aRegionIter.Next() ) {
+    Handle(HYDROData_Region) aRegion =
+      Handle(HYDROData_Region)::DownCast( aRegionIter.Value() );
+    if( aRegion.IsNull() ) {
+      continue;
+    }
+
+    TopoDS_Shape aRegionShape = aRegion->GetShape();
+    aFaces.Append( aRegionShape );
+  }
 
   // Get groups
-  // TODO
-  HYDROData_SequenceOfObjects aSplittedGroups;
+  // TODO to be revised
+  HYDROData_SequenceOfObjects aSplittedGroups = GetSplittedGroups();
 
   return Export( theGeomEngine, theStudy, aFaces, aSplittedGroups );
 }
@@ -797,11 +678,9 @@ bool HYDROData_CalculationCase::Export( GEOM::GEOM_Gen_var theGeomEngine,
                                         const TopTools_ListOfShape& theFaces,
                                         const HYDROData_SequenceOfObjects& theSplittedGroups )
 {
-  // Make shell from the faces
-  TopoDS_Shell aShell;
-  TopTools_ListOfShape aFacesList;
-
+  // Sew faces
   BRepBuilderAPI_Sewing aSewing( Precision::Confusion()*10.0 );
+  aSewing.SetNonManifoldMode( Standard_True );
 
   TopTools_ListIteratorOfListOfShape aFaceIter( theFaces );
   for ( ; aFaceIter.More(); aFaceIter.Next() ) {
@@ -809,7 +688,6 @@ bool HYDROData_CalculationCase::Export( GEOM::GEOM_Gen_var theGeomEngine,
     if ( !aShape.IsNull() && (aShape.ShapeType() == TopAbs_FACE) ) {
       TopoDS_Face aFace = TopoDS::Face( aShape );
       if ( !aFace.IsNull() ) {
-        aFacesList.Append( aFace );
         aSewing.Add( aFace );
       }
     } else {
@@ -817,7 +695,6 @@ bool HYDROData_CalculationCase::Export( GEOM::GEOM_Gen_var theGeomEngine,
       for ( ; anExp.More(); anExp.Next() ) {
         TopoDS_Face aFace = TopoDS::Face( anExp.Current() );
         if ( !aFace.IsNull() ) {
-          aFacesList.Append( aFace );
           aSewing.Add( aFace );
         }
       }
@@ -827,77 +704,75 @@ bool HYDROData_CalculationCase::Export( GEOM::GEOM_Gen_var theGeomEngine,
   aSewing.Perform();
   TopoDS_Shape aSewedShape = aSewing.SewedShape();
 
-  if ( !aSewedShape.IsNull() ) {
-    if ( aSewedShape.ShapeType() == TopAbs_FACE && theFaces.Extent() ==1 ) {
-      // create shell from one face
-      BRep_Builder aBuilder;
-      aBuilder.MakeShell( aShell );
-      aBuilder.Add( aShell, aSewedShape);
-    } else {
-      TopExp_Explorer anExpShells( aSewedShape, TopAbs_SHELL );
-      Standard_Integer aNbOfShells = 0;
-      for ( ; anExpShells.More(); anExpShells.Next() ) {
-        aShell = TopoDS::Shell( anExpShells.Current() );
-        aNbOfShells++;
-      }
-
-      if ( aNbOfShells != 1 ) {
-        aShell.Nullify();
-        BRep_Builder aBuilder;
-        aBuilder.MakeShell( aShell );
-
-        TopExp_Explorer anExpFaces( aSewedShape, TopAbs_FACE );
-        for ( ; anExpFaces.More(); anExpFaces.Next() ) {
-          TopoDS_Face aFace = TopoDS::Face( anExpFaces.Current() );
-          if ( !aFace.IsNull() ) {
-            aBuilder.Add( aShell, aFace );
-          }
-        }
-      }
-    }
-  }
-
-  if ( !aShell.IsNull() ) {
-    TopTools_IndexedMapOfShape aMapOfFaces;
-    TopExp::MapShapes( aShell, TopAbs_FACE, aMapOfFaces );
-    if ( aMapOfFaces.Extent() != aFacesList.Extent() ) {
-      aShell.Nullify();
-      BRep_Builder aBuilder;
-      aBuilder.MakeShell( aShell );
-
-      TopTools_ListIteratorOfListOfShape anIter( aFacesList );
-      for ( ; anIter.More(); anIter.Next() ) {
-        TopoDS_Face aFace = TopoDS::Face( anIter.Value() );
-        aBuilder.Add( aShell, aFace );
-      }
-    }
-  }
-
-  // If the shell is empty - return false
-  if ( aShell.IsNull() || !TopoDS_Iterator(aShell).More() ) {
+  // If the sewed shape is empty - return false
+  if ( aSewedShape.IsNull() || !TopoDS_Iterator(aSewedShape).More() ) {
     return false;
   }
 
-  // Publish the shell
+  // Publish the sewed shape
   QString aName = EXPORT_NAME;
-  GEOM::GEOM_Object_ptr aPublishedObj = 
-    publishShapeInGEOM( theGeomEngine, theStudy, aShell, aName );
+  GEOM::GEOM_Object_ptr aMainShape = 
+    publishShapeInGEOM( theGeomEngine, theStudy, aSewedShape, aName );
 
-  if ( aPublishedObj->_is_nil() ) {
+  if ( aMainShape->_is_nil() ) {
     return false;
   }
 
   // Create groups
+  TopTools_IndexedMapOfShape aMapOfSubShapes;
+  TopExp::MapShapes( aSewedShape, aMapOfSubShapes );
+
+  QHash<QString, QSet<int> > aGroupsData;
+
+  HYDROData_SequenceOfObjects::Iterator anIter( theSplittedGroups );
+  for ( ; anIter.More(); anIter.Next() ) {
+    // Get shapes group
+    Handle(HYDROData_ShapesGroup) aGroup =
+      Handle(HYDROData_ShapesGroup)::DownCast( anIter.Value() );
+    if ( aGroup.IsNull() ) {
+      continue;
+    }
+
+    QSet<int> anIndexes;
+
+    // Get shapes of the group
+    TopTools_SequenceOfShape aShapes;
+    aGroup->GetShapes( aShapes );
+    for( int i = 1, aNbShapes = aShapes.Length(); i <= aNbShapes; i++ ) {
+      const TopoDS_Shape& aShape = aShapes.Value( i );
+      const TopoDS_Shape ModifiedShape = aSewing.Modified( aShape );
+      if ( !ModifiedShape.IsNull() ) {
+        int anIndex = aMapOfSubShapes.FindIndex( ModifiedShape );
+        if ( anIndex > 0 ) {
+          anIndexes << anIndex;
+        }
+      }
+    }
+    
+    if ( anIndexes.count() > 0 ) {
+      aGroupsData.insert( aGroup->GetName(), anIndexes );
+    }
+  }
+  
   GEOM::GEOM_IGroupOperations_var aGroupOp = 
-    theGeomEngine->GetIGroupOperations( theStudy->StudyId() );
-
-  GEOM::GEOM_Object_var aGroup = aGroupOp->CreateGroup( aPublishedObj, TopAbs_EDGE );
-  if ( !CORBA::is_nil(aGroup) && aGroupOp->IsDone() ) {
-    GEOM::ListOfLong_var anIndexes = new GEOM::ListOfLong;
-    aGroupOp->UnionIDs( aGroup, anIndexes );
-    if ( aGroupOp->IsDone() ) {
-      SALOMEDS::SObject_var aGroupSO = 
-        theGeomEngine->AddInStudy( theStudy, aGroup, qPrintable( aName ), aPublishedObj );
+    theGeomEngine->GetIGroupOperations( theStudy->StudyId() );  
+
+  foreach ( const QString& aGroupName, aGroupsData.keys() ) {
+    QSet<int> aGroupIndexes = aGroupsData.value( aGroupName );
+
+    GEOM::GEOM_Object_var aGroup = aGroupOp->CreateGroup( aMainShape, TopAbs_EDGE );
+    if ( !CORBA::is_nil(aGroup) && aGroupOp->IsDone() ) {
+      GEOM::ListOfLong_var anIndexes = new GEOM::ListOfLong;
+      int aListIndex = 0;
+      foreach ( const int anIndex, aGroupIndexes ) {
+        anIndexes[aListIndex++] = anIndex;
+      }
+
+      aGroupOp->UnionIDs( aGroup, anIndexes );
+      if ( aGroupOp->IsDone() ) {
+        SALOMEDS::SObject_var aGroupSO = 
+          theGeomEngine->AddInStudy( theStudy, aGroup, qPrintable( aName ), aMainShape );
+      }
     }
   }
 
@@ -910,8 +785,6 @@ GEOM::GEOM_Object_ptr HYDROData_CalculationCase::publishShapeInGEOM(
 {
   GEOM::GEOM_Object_var aGeomObj;
 
-  bool isNil = aGeomObj->_is_nil(); ///@MZN
-
   if ( theGeomEngine->_is_nil() || theStudy->_is_nil() ||
        theShape.IsNull() ) {
     return aGeomObj._retn();
@@ -946,4 +819,4 @@ GEOM::GEOM_Object_ptr HYDROData_CalculationCase::publishShapeInGEOM(
   }
 
   return aGeomObj._retn();
- }
\ No newline at end of file
+ }