Salome HOME
DCQ : Merge with Ecole_Ete_a6.
[modules/smesh.git] / src / DriverMED / DriverMED_W_SMESHDS_Mesh.h
index ec2ad8d862f4df6c44c1f5db5562542e31c0ab7f..df91d0d4955b046fe08e29962b46d6cba4be7bca 100644 (file)
 #include "Mesh_Writer.h"
 
 #include <string>
+#include <list>
+#include <map>
 
 extern "C"
 {
 #include <med.h>
 }
 
-class DriverMED_W_SMESHDS_Mesh : public Mesh_Writer {
+using namespace std;
 
-  public :
-    DriverMED_W_SMESHDS_Mesh();
-    ~DriverMED_W_SMESHDS_Mesh();
+class SMESHDS_Group;
+class SMESHDS_SubMesh;
 
-    void Add(); 
-    void Write();
-    void SetMesh(Handle(SMDS_Mesh)& aMesh);
-    void SetFile(string);
+class DriverMED_W_SMESHDS_Mesh:public Mesh_Writer
+{
+  public:
+
+  DriverMED_W_SMESHDS_Mesh();
+  ~DriverMED_W_SMESHDS_Mesh();
+
+  /*! sets file name; only for usage with Add(), not Write()
+   */
+  void SetFile(string);
+  void AddGroupOfNodes();
+  void AddGroupOfEdges();
+  void AddGroupOfFaces();
+  void AddGroupOfVolumes();
 
-    void SetFileId(med_idt);
-    void SetMeshId(int);
+  /*! functions to prepare adding one mesh
+   */
+  void SetMesh(SMDS_Mesh * aMesh);
+  void SetMeshId(int);
+  void SetMeshName(string theMeshName);
+  void AddGroup(SMESHDS_Group* theGroup);
+  void AddAllSubMeshes();
+  void AddSubMesh(SMESHDS_SubMesh* theSubMesh, int theID);
 
-    void CreateFamily(char*, char*, int, med_int);
+  /*! add one mesh
+   */
+  void Add();
 
-private :
-    Handle_SMDS_Mesh myMesh;
-    string myFile; 
-    med_idt myFileId;
-    int myMeshId;
+  /*! functions to write via DriverMED_W_SMDS_Mesh (no groups)
+   */
+  void SetFileId(med_idt);
+  void Write();
 
+ private:
+
+  SMDS_Mesh * myMesh;
+  string myFile;
+  med_idt myFileId;
+  int myMeshId;
+  string myMeshName;
+  list<SMESHDS_Group*> myGroups;
+  bool myAllSubMeshes;
+  map<int,SMESHDS_SubMesh*> mySubMeshes;
+  bool myDoGroupOfNodes;
+  bool myDoGroupOfEdges;
+  bool myDoGroupOfFaces;
+  bool myDoGroupOfVolumes;
 };
 #endif
+