NPAL18095: Pb. with dump python and mesh group by filter.

[modules/smesh.git] / src / StdMeshers / StdMeshers_ProjectionSource1D.hxx

diff --git a/src/StdMeshers/StdMeshers_ProjectionSource1D.hxx b/src/StdMeshers/StdMeshers_ProjectionSource1D.hxx
index e1d019a1cb1b708975e6811cda5be00c256e8862..1cd9bf6bb0d2c92e4eea903ccc617a08a5a6acb8 100644 (file)
--- a/src/StdMeshers/StdMeshers_ProjectionSource1D.hxx
+++ b/src/StdMeshers/StdMeshers_ProjectionSource1D.hxx
@@ -32,7 +32,6 @@
 #include "SMESH_Hypothesis.hxx"
 #include "Utils_SALOME_Exception.hxx"
 
-#include <TopoDS_Edge.hxx>
 #include <TopoDS_Vertex.hxx>
 
 class SMESH_Gen;
@@ -60,14 +59,20 @@ public:
     throw ( SALOME_Exception );
 
   /*!
-   * Returns the source edge
+   * Returns the source edge or a group containing edges
    */
-  TopoDS_Edge GetSourceEdge() const { return _sourceEdge; }
+  TopoDS_Shape GetSourceEdge() const { return _sourceEdge; }
+
+  /*!
+   * Returns true the source edge is a group of edges
+   */
+  bool IsCompoundSource() const
+  { return !_sourceEdge.IsNull() && _sourceEdge.ShapeType() == TopAbs_COMPOUND; }
 
   /*!
    * Sets source <mesh> to take a mesh pattern from
    */
-  void SetSourceMesh(SMESH_Mesh* mesh) { _sourceMesh = mesh; }
+  void SetSourceMesh(SMESH_Mesh* mesh);
 
   /*!
    * Return source mesh
@@ -133,7 +138,7 @@ public:
 
 protected:
 
-  TopoDS_Edge   _sourceEdge;
+  TopoDS_Shape  _sourceEdge;
   SMESH_Mesh*   _sourceMesh;
   TopoDS_Vertex _sourceVertex;
   TopoDS_Vertex _targetVertex;