X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_ShapesTool.cxx;h=d11b369168ddfac2bb90a472f71886eef06a5c13;hb=2f217128f8d34b2082de78d2af90bf243bf9e8b1;hp=283d64c0c891095364fc927d8daefd66f067ff28;hpb=9b2bfc45ba9bf3430d7fb6ed9bfe8352a91885a3;p=modules%2Fhydro.git

diff --git a/src/HYDROData/HYDROData_ShapesTool.cxx b/src/HYDROData/HYDROData_ShapesTool.cxx
index 283d64c0..d11b3691 100644
--- a/src/HYDROData/HYDROData_ShapesTool.cxx
+++ b/src/HYDROData/HYDROData_ShapesTool.cxx
@@ -9,10 +9,15 @@
 
 #include <TopExp.hxx>
 
+#include <TopoDS.hxx>
+#include <TopoDS_Edge.hxx>
 #include <TopoDS_Shape.hxx>
 #include <TopoDS_Vertex.hxx>
+#include <TopoDS_Wire.hxx>
 
 #include <TopTools_SequenceOfShape.hxx>
+#include <TopTools_ListOfShape.hxx>
+#include <TopTools_ListIteratorOfListOfShape.hxx>
 
 #include <TopExp_Explorer.hxx>
 
@@ -33,22 +38,143 @@ void HYDROData_ShapesTool::ExploreShapeToShapes( const TopoDS_Shape&       theIn
   }
 }
 
+bool HYDROData_ShapesTool::Vertices( const TopoDS_Shape& theInShape,
+                                     TopoDS_Vertex&      theFirstVertex,
+                                     TopoDS_Vertex&      theLastVertex,
+                                     const bool          theIsCumOri  )
+{
+  if ( theInShape.IsNull() )
+    return false;
+
+  if ( theInShape.ShapeType() == TopAbs_EDGE )
+  {
+    TopoDS_Edge anEdge = TopoDS::Edge( theInShape );
+    TopExp::Vertices( anEdge, theFirstVertex, theLastVertex, theIsCumOri );
+  }
+  else if ( theInShape.ShapeType() == TopAbs_WIRE )
+  {
+    TopoDS_Wire aWire = TopoDS::Wire( theInShape );
+    TopExp::Vertices( aWire, theFirstVertex, theLastVertex );
+  }
+  else
+  {
+    return false;
+  }
+
+  return true;
+}
+
 bool HYDROData_ShapesTool::IsVerticesEquals( const TopoDS_Vertex& theFirstVert,
-                                             const TopoDS_Vertex& theSecondVert )
+                                             const TopoDS_Vertex& theSecondVert,
+                                             const bool           theIs2D )
 {
   gp_Pnt aFirstPoint = BRep_Tool::Pnt( theFirstVert );
   gp_Pnt aSecondPoint = BRep_Tool::Pnt( theSecondVert );
+
+  // Set the Z coord to zero for 2D equations
+  if ( theIs2D )
+  {
+    aFirstPoint.SetZ( 0.0 );
+    aSecondPoint.SetZ( 0.0 );
+  }
+
   return aFirstPoint.IsEqual( aSecondPoint, Precision::Confusion() );
 }
 
+TopoDS_Shape HYDROData_ShapesTool::Translated( const TopoDS_Shape& theShape,
+                                               const double        theDx,
+                                               const double        theDy,
+                                               const double        theDz )
+{
+  TopoDS_Shape aTranslatedShape;
+  if ( theShape.IsNull() )
+    return aTranslatedShape;
+
+  gp_Vec aVec( theDx, theDy, theDz );
+
+  gp_Trsf aTrsf;
+  aTrsf.SetTranslation( aVec );
+
+  TopLoc_Location aLocOrig = theShape.Location();
+  gp_Trsf aTrsfOrig = aLocOrig.Transformation();
+
+  TopLoc_Location aLocRes( aTrsf * aTrsfOrig );
+  aTranslatedShape = theShape.Located( aLocRes );
+
+  return aTranslatedShape;
+}
+
 bool HYDROData_ShapesTool::IsEdgesEquals( const TopoDS_Edge& theFirstEdge,
-                                          const TopoDS_Edge& theSecondEdge )
+                                          const TopoDS_Edge& theSecondEdge,
+                                          const bool         theIs2D )
 {
   TopoDS_Vertex aFFirstVert, aFLastVert, aSFirstVert, aSLastVert;
   TopExp::Vertices( theFirstEdge, aFFirstVert, aFLastVert );
   TopExp::Vertices( theSecondEdge, aSFirstVert, aSLastVert );
-  return IsVerticesEquals( aFFirstVert, aSFirstVert ) &&
-         IsVerticesEquals( aFLastVert, aSLastVert );
+  return IsVerticesEquals( aFFirstVert, aSFirstVert, theIs2D ) &&
+         IsVerticesEquals( aFLastVert, aSLastVert, theIs2D );
+}
+
+void HYDROData_ShapesTool::AddShapes( TopTools_SequenceOfShape&       theShapes,
+                                      const TopTools_SequenceOfShape& theShapesToAdd )
+{
+  for ( int i = 1, n = theShapesToAdd.Length(); i <= n; ++i )
+  {
+    const TopoDS_Shape& aShape = theShapesToAdd.Value( i );
+    theShapes.Append( aShape );
+  }
 }
 
+void HYDROData_ShapesTool::AddShapes( TopTools_SequenceOfShape&   theShapes,
+                                      const TopTools_ListOfShape& theShapesToAdd )
+{
+  TopTools_ListIteratorOfListOfShape anIter( theShapesToAdd );
+  for ( ; anIter.More(); anIter.Next() )
+  {
+    const TopoDS_Shape& aShape = anIter.Value();
+    theShapes.Append( aShape );
+  }
+}
+
+void HYDROData_ShapesTool::AddShapes( TopTools_ListOfShape&           theShapes,
+                                      const TopTools_SequenceOfShape& theShapesToAdd )
+{
+  for ( int i = 1, n = theShapesToAdd.Length(); i <= n; ++i )
+  {
+    const TopoDS_Shape& aShape = theShapesToAdd.Value( i );
+    theShapes.Append( aShape );
+  }
+}
 
+void HYDROData_ShapesTool::AddShapes( TopTools_ListOfShape&       theShapes,
+                                      const TopTools_ListOfShape& theShapesToAdd )
+{
+  TopTools_ListIteratorOfListOfShape anIter( theShapesToAdd );
+  for ( ; anIter.More(); anIter.Next() )
+  {
+    const TopoDS_Shape& aShape = anIter.Value();
+    theShapes.Append( aShape );
+  }
+}
+
+void HYDROData_ShapesTool::DumpShapeSubShapes( std::ostream&           theStream,
+                                               const char*             theTitle,
+                                               const TopoDS_Shape&     theShape,
+                                               const TopAbs_ShapeEnum& theExpType )
+{
+  TopTools_SequenceOfShape aShapeSubShapes;
+  ExploreShapeToShapes( theShape, theExpType, aShapeSubShapes );
+
+  theStream << theTitle << "\n";
+  DumpSequenceOfShapes( theStream, aShapeSubShapes );
+}
+
+void HYDROData_ShapesTool::DumpSequenceOfShapes( std::ostream&                   theStream,
+                                                 const TopTools_SequenceOfShape& theShapes )
+{
+  for ( int i = 1, n = theShapes.Length(); i <= n; ++i )
+  {
+    const TopoDS_Shape& aShape = theShapes.Value( i );
+    theStream << "Shape " << i << " : " << aShape.TShape() << "\n";
+  }
+}
\ No newline at end of file