]> SALOME platform Git repositories - modules/med.git/blob - medtool/src/MEDLoader/SauvWriter.hxx
Salome HOME
Move medtool folder to MED base repository
[modules/med.git] / medtool / src / MEDLoader / SauvWriter.hxx
1 // Copyright (C) 2007-2015  CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19 // File      : SauvWriter.hxx
20 // Created   : Wed Aug 24 11:18:49 2011
21 // Author    : Edward AGAPOV (eap)
22
23 #ifndef __SAUVWRITER_HXX__
24 #define __SAUVWRITER_HXX__
25
26 #include "MEDLoaderDefines.hxx"
27 #include "MEDCouplingRefCountObject.hxx"
28 #include "NormalizedUnstructuredMesh.hxx"
29 #include "SauvUtilities.hxx"
30 #include "MEDCouplingAutoRefCountObjectPtr.hxx"
31
32 #include <vector>
33 #include <string>
34 #include <map>
35
36 namespace ParaMEDMEM
37 {
38   class MEDFileData;
39   class MEDFileMesh;
40   class MEDFileFieldMultiTS;
41   class DataArrayInt;
42
43   /*!
44    * \brief Class to write a MEDFileData into a SAUVE format file
45    */
46   class SauvWriter : public ParaMEDMEM::RefCountObject
47   {
48   public:
49     MEDLOADER_EXPORT static SauvWriter *New();
50     MEDLOADER_EXPORT void setMEDFileDS(const MEDFileData* medData, unsigned meshIndex = 0);
51     MEDLOADER_EXPORT void write(const std::string& fileName);
52     MEDLOADER_EXPORT void setCpyGrpIfOnASingleFamilyStatus(bool status);
53     MEDLOADER_EXPORT bool getCpyGrpIfOnASingleFamilyStatus() const;
54   private:
55     SauvWriter();
56     std::size_t getHeapMemorySizeWithoutChildren() const;
57     std::vector<const BigMemoryObject *> getDirectChildrenWithNull() const;
58     /*!
59      * \brief Class representing a GIBI sub-mesh (described in the pile 1 of the SAUVE file).
60      * It stands for a named med sub-mesh (family, etc) and contains either cell IDs or other sub-meshes. 
61      */
62     struct SubMesh
63     {
64       std::vector<int>      _cellIDsByType[ INTERP_KERNEL::NORM_MAXTYPE+1 ];
65       std::vector<SubMesh*> _subs;
66       std::string           _name;
67       int                   _id;
68       int                   _nbSauvObjects;
69       int                   _dimRelExt;
70       int nbTypes() const;
71       static int cellIDsByTypeSize() { return INTERP_KERNEL::NORM_MAXTYPE+1; }
72     };
73     SubMesh* addSubMesh(const std::string& name, int dimRelExt);
74
75     void fillSubMeshes(int& nbSauvObjects, std::map<std::string,int>& nameNbMap);
76     void fillFamilySubMeshes();
77     void fillGroupSubMeshes();
78     void fillProfileSubMeshes();
79     int evaluateNbProfileSubMeshes() const;
80     void makeCompNames(const std::string&                  fieldName,
81                        const std::vector<std::string>&     compInfo,
82                        std::map<std::string, std::string>& compMedToGibi );
83     void makeProfileIDs( SubMesh*                          sm,
84                          INTERP_KERNEL::NormalizedCellType type,
85                          const DataArrayInt*               profile );
86     void writeFileHead();
87     void writeSubMeshes();
88     void writeCompoundSubMesh(int iSub);
89     void writeNodes();
90     void writeLongNames();
91     void writeNodalFields(std::map<std::string,int>& fldNamePrefixMap);
92     void writeElemFields(std::map<std::string,int>& fldNamePrefixMap);
93     void writeFieldNames(const bool isNodal, std::map<std::string,int>& fldNamePrefixMap);
94     void writeElemTimeStamp(int iF, int iter, int order);
95     void writeLastRecord();
96     void writeNames( const std::map<std::string,int>& nameNbMap );
97
98   private:
99
100     MEDCouplingAutoRefCountObjectPtr< MEDFileMesh >                        _fileMesh;
101     std::vector< MEDCouplingAutoRefCountObjectPtr< MEDFileFieldMultiTS > > _nodeFields;
102     std::vector< MEDCouplingAutoRefCountObjectPtr< MEDFileFieldMultiTS > > _cellFields;
103
104     std::vector<SubMesh>                      _subs;
105     std::map< int, SubMesh* >                 _famIDs2Sub;
106     std::map< std::string, SubMesh* >         _profile2Sub;
107     enum
108       {
109         LN_MAIL=0, LN_CHAM, LN_COMP, LN_NB
110       };
111     std::vector<SauvUtilities::nameGIBItoMED> _longNames[ LN_NB ];
112
113     std::fstream*                             _sauvFile;
114     bool _cpy_grp_if_on_single_family;
115   };
116 }
117
118 #endif