Salome HOME
23514: EDF 16031 - SMESH freezes
[modules/smesh.git] / src / DriverMED / DriverMED_W_SMESHDS_Mesh.h
index a5385d742f43b4c7acb37388a408ddcf3014e6dc..a9539b9e8741fe509dc95803d76eab4d74c5e345 100644 (file)
@@ -1,37 +1,95 @@
+// Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
+//
+// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
+//
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+
+//  SMESH DriverMED : driver to read and write 'med' files
+//  File   : DriverMED_W_SMESHDS_Mesh.h
+//  Module : SMESH
+//
 #ifndef _INCLUDE_DRIVERMED_W_SMESHDS_MESH
 #define _INCLUDE_DRIVERMED_W_SMESHDS_MESH
 
-#include "SMESHDS_Mesh.hxx"
-#include "Mesh_Writer.h"
+#include "SMESH_DriverMED.hxx"
+
+#include "Driver_SMESHDS_Mesh.h"
+#include "MED_Common.hxx"
 
 #include <string>
+#include <list>
+#include <map>
+#include <vector>
+
+class SMESHDS_Mesh;
+class SMESHDS_GroupBase;
+class SMESHDS_SubMesh;
 
-extern "C"
+class MESHDRIVERMED_EXPORT DriverMED_W_SMESHDS_Mesh: public Driver_SMESHDS_Mesh
 {
-#include <med.h>
-}
+  public:
 
-class DriverMED_W_SMESHDS_Mesh : public Mesh_Writer {
+  DriverMED_W_SMESHDS_Mesh();
 
-  public :
-    DriverMED_W_SMESHDS_Mesh();
-    ~DriverMED_W_SMESHDS_Mesh();
+  virtual void SetFile(const std::string& theFileName);
+  void SetFile(const std::string& theFileName, MED::EVersion theId);
+  void SetAutoDimension(bool toFindOutDimension) { myAutoDimension = toFindOutDimension; }
 
-    void Add(); 
-    void Write();
-    void SetMesh(Handle(SMDS_Mesh)& aMesh);
-    void SetFile(string);
+  static std::string GetVersionString(const MED::EVersion theVersion, int theNbDigits=2);
 
-    void SetFileId(med_idt);
-    void SetMeshId(int);
+  void AddGroupOfNodes();
+  void AddGroupOfEdges();
+  void AddGroupOfFaces();
+  void AddGroupOfVolumes();
+  void AddGroupOf0DElems();
+  void AddGroupOfBalls();
+  void AddAllToGroup();
 
-    void CreateFamily(char*, char*, int, med_int);
+  /*! functions to prepare adding one mesh
+   */
+  void AddGroup(SMESHDS_GroupBase * theGroup);
+  void AddAllSubMeshes();
+  void AddSubMesh(SMESHDS_SubMesh* theSubMesh, int theID);
+  void AddODOnVertices(bool toAdd) { myAddODOnVertices = toAdd; }
 
-private :
-    Handle_SMDS_Mesh myMesh;
-    string myFile; 
-    med_idt myFileId;
-    int myMeshId;
+  static bool getNodesOfMissing0DOnVert(SMESHDS_Mesh*                         mesh,
+                                        std::vector<const SMDS_MeshElement*>& nodes);
 
+  /*! add one mesh
+   */
+  virtual Status Perform();
+
+ private:
+
+  MED::EVersion myMedVersion;
+  std::list<SMESHDS_GroupBase*> myGroups;
+  bool myAllSubMeshes;
+  std::vector<SMESHDS_SubMesh*> mySubMeshes;
+  bool myDoGroupOfNodes;
+  bool myDoGroupOfEdges;
+  bool myDoGroupOfFaces;
+  bool myDoGroupOfVolumes;
+  bool myDoGroupOf0DElems;
+  bool myDoGroupOfBalls;
+  bool myAutoDimension;
+  bool myAddODOnVertices;
+  bool myDoAllInGroups;
 };
+
 #endif