Salome HOME
Update of CheckDone
[modules/smesh.git] / src / DriverMED / DriverMED_W_SMESHDS_Mesh.h
1 // Copyright (C) 2007-2024  CEA, EDF, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License, or (at your option) any later version.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22
23 //  SMESH DriverMED : driver to read and write 'med' files
24 //  File   : DriverMED_W_SMESHDS_Mesh.h
25 //  Module : SMESH
26 //
27 #ifndef _INCLUDE_DRIVERMED_W_SMESHDS_MESH
28 #define _INCLUDE_DRIVERMED_W_SMESHDS_MESH
29
30 #include "SMESH_DriverMED.hxx"
31
32 #include "Driver_SMESHDS_Mesh.h"
33 //#include "MED_Common.hxx"
34
35 #include <string>
36 #include <list>
37 #include <map>
38 #include <vector>
39
40 class SMESHDS_Mesh;
41 class SMESHDS_GroupBase;
42 class SMESHDS_SubMesh;
43 class SMDS_MeshElement;
44
45 /*!
46  * \brief Write a mesh to a MED file
47  */
48 class MESHDRIVERMED_EXPORT DriverMED_W_SMESHDS_Mesh: public Driver_SMESHDS_Mesh
49 {
50   public:
51
52   DriverMED_W_SMESHDS_Mesh();
53
54   void SetFile(const std::string& theFileName, int theVersion=-1);
55   void SetAutoDimension(bool toFindOutDimension) { myAutoDimension = toFindOutDimension; }
56   void SetZTolerance(double tol) { myZTolerance = tol; }
57   void SetSaveNumbers(bool toSave) { mySaveNumbers = toSave; }
58
59   static std::string GetVersionString(int theMinor, int theNbDigits=2);
60
61   void AddGroupOfNodes();
62   void AddGroupOfEdges();
63   void AddGroupOfFaces();
64   void AddGroupOfVolumes();
65   void AddGroupOf0DElems();
66   void AddGroupOfBalls();
67
68   /*! functions to prepare adding one mesh
69    */
70   void AddGroup(SMESHDS_GroupBase * theGroup);
71   void AddAllSubMeshes();
72   void AddSubMesh(SMESHDS_SubMesh* theSubMesh, int theID);
73   void AddODOnVertices(bool toAdd) { myAddODOnVertices = toAdd; }
74
75   static bool getNodesOfMissing0DOnVert(SMESHDS_Mesh*                         mesh,
76                                         std::vector<const SMDS_MeshElement*>& nodes);
77
78   /*! add one mesh
79    */
80   Status Perform() override;
81
82   template<class LowLevelWriter>
83   Driver_Mesh::Status PerformInternal(LowLevelWriter myMed);
84
85  private:
86
87   std::list<SMESHDS_GroupBase*> myGroups;
88   bool                          myAllSubMeshes;
89   std::vector<SMESHDS_SubMesh*> mySubMeshes;
90   bool                          myDoGroupOfNodes;
91   bool                          myDoGroupOfEdges;
92   bool                          myDoGroupOfFaces;
93   bool                          myDoGroupOfVolumes;
94   bool                          myDoGroupOf0DElems;
95   bool                          myDoGroupOfBalls;
96   bool                          myAutoDimension;
97   bool                          myAddODOnVertices;
98   bool                          myDoAllInGroups;
99   int                           myVersion;
100   double                        myZTolerance;
101   bool                          mySaveNumbers;
102 };
103
104 #include "MEDCouplingMemArray.hxx"
105
106 /*!
107  * \brief Write a mesh to a MEDCoupling DS
108  */
109 class MESHDRIVERMED_EXPORT DriverMED_W_SMESHDS_Mesh_Mem : public DriverMED_W_SMESHDS_Mesh
110 {
111 public:
112   Status Perform() override;
113   MEDCoupling::MCAuto<MEDCoupling::DataArrayByte> getData() { return _data; }
114 private:
115   MEDCoupling::MCAuto<MEDCoupling::DataArrayByte> _data;
116 };
117
118 #endif