Salome HOME
PAL10332 - references are interpreted as original objects in filters, dialogs, etc.
[modules/smesh.git] / src / SMESHDS / SMESHDS_SubMesh.hxx
index bda8f04bb48c4a53e78ec0d0c31a52d8d7472228..ac3cd1c3f81b79ea618a963aa2ccb5c2983dda37 100644 (file)
 #include "SMDS_Mesh.hxx"
 #include <set>
 
-using namespace std;
+#if defined WNT && defined WIN32 && defined SMESHDS_EXPORTS
+#define SMESHDS_WNT_EXPORT __declspec( dllexport )
+#else
+#define SMESHDS_WNT_EXPORT
+#endif
 
-class SMESHDS_SubMesh
+class SMESHDS_WNT_EXPORT SMESHDS_SubMesh
 {
-  public:
-       void AddElement(const SMDS_MeshElement * ME);
-       void RemoveElement(const SMDS_MeshElement * ME);
-       void AddNode(const SMDS_MeshNode * ME);
-       void RemoveNode(const SMDS_MeshNode * ME);
-       int NbElements() const;
-       SMDS_Iterator<const SMDS_MeshElement*> * GetElements() const;
-       int NbNodes() const;
-       SMDS_Iterator<const SMDS_MeshNode*> * GetNodes() const;
-
-  private:
-       const SMDS_Mesh * myMesh;
-       set<const SMDS_MeshElement*> myElements;
-       set<const SMDS_MeshNode*> myNodes;
+ public:
+
+  bool IsComplexSubmesh() const { return !mySubMeshes.empty(); }
+
+  // if !IsComplexSubmesh()
+  void AddElement(const SMDS_MeshElement * ME);
+  bool RemoveElement(const SMDS_MeshElement * ME); // ret true if ME was in
+  void AddNode(const SMDS_MeshNode * ME);
+  bool RemoveNode(const SMDS_MeshNode * ME);       // ret true if ME was in
+
+  // if IsComplexSubmesh()
+  void AddSubMesh( const SMESHDS_SubMesh* theSubMesh );
+  bool RemoveSubMesh( const SMESHDS_SubMesh* theSubMesh );
+  bool ContainsSubMesh( const SMESHDS_SubMesh* theSubMesh ) const;
+
+  // for both types
+  int NbElements() const;
+  SMDS_ElemIteratorPtr GetElements() const;
+  int NbNodes() const;
+  SMDS_NodeIteratorPtr GetNodes() const;
+  bool Contains(const SMDS_MeshElement * ME) const;      // check if elem or node is in
+
+ private:
+  //const SMDS_Mesh * myMesh;
+  std::set<const SMDS_MeshElement*> myElements;
+  std::set<const SMDS_MeshNode*>    myNodes;
+  std::set<const SMESHDS_SubMesh*>  mySubMeshes;
 };
 #endif