Salome HOME
Merge branch merge_1_2_d
[modules/smesh.git] / src / DriverUNV / DriverUNV_W_SMDS_Mesh.cxx
1 using namespace std;
2 #include "DriverUNV_W_SMDS_Mesh.h"
3
4 #include "SMDS_MeshElement.hxx"
5 #include "SMDS_MeshNode.hxx"
6
7
8
9
10
11 #include <utilities.h>
12
13 #define sNODE_UNV_ID "  2411"
14 #define sELT_UNV_ID  "  2412"
15 #define sUNV_SEPARATOR "    -1"
16 #define sNODE_UNV_DESCR "%10d         1         1        11\n"
17 #define sELT_SURF_DESC  "%10d        %2d         1         1        11         %1d\n"
18 #define sELT_VOLU_DESC  "%10d        %2d         1         1         9         %1d\n"
19 #define sELT_BEAM_DESC1 "%10d        %2d         1         1         7         %1d\n"
20 #define sELT_BEAM_DESC2 "         0         1         1\n"
21
22 DriverUNV_W_SMDS_Mesh::DriverUNV_W_SMDS_Mesh()
23 {
24         ;
25 }
26
27 DriverUNV_W_SMDS_Mesh::~DriverUNV_W_SMDS_Mesh()
28 {
29         ;
30 }
31
32 void DriverUNV_W_SMDS_Mesh::SetMesh(SMDS_Mesh * aMesh)
33 {
34         myMesh = aMesh;
35 }
36
37 void DriverUNV_W_SMDS_Mesh::SetFile(string aFile)
38 {
39         myFile = aFile;
40 }
41
42 void DriverUNV_W_SMDS_Mesh::SetFileId(FILE * aFileId)
43 {
44         myFileId = aFileId;
45 }
46
47 void DriverUNV_W_SMDS_Mesh::SetMeshId(int aMeshId)
48 {
49         myMeshId = aMeshId;
50 }
51
52 void DriverUNV_W_SMDS_Mesh::Add()
53 {
54         ;
55 }
56
57 void DriverUNV_W_SMDS_Mesh::Write()
58 {
59
60         int nbNodes, nbCells;
61         int i;
62
63         char *file2Read = (char *)myFile.c_str();
64         myFileId = fopen(file2Read, "w+");
65         if (myFileId < 0)
66         {
67                 fprintf(stderr, ">> ERREUR : ouverture du fichier %s \n", file2Read);
68                 exit(EXIT_FAILURE);
69         }
70         SCRUTE(myMesh);
71   /****************************************************************************
72   *                       NOMBRES D'OBJETS                                    *
73   ****************************************************************************/
74         fprintf(stdout, "\n(****************************)\n");
75         fprintf(stdout, "(* INFORMATIONS GENERALES : *)\n");
76         fprintf(stdout, "(****************************)\n");
77
78         /* Combien de noeuds ? */
79         nbNodes = myMesh->NbNodes();
80
81         /* Combien de mailles, faces ou aretes ? */
82         int nb_of_nodes, nb_of_edges, nb_of_faces, nb_of_volumes;
83         nb_of_edges = myMesh->NbEdges();
84         nb_of_faces = myMesh->NbFaces();
85         nb_of_volumes = myMesh->NbVolumes();
86         nbCells = nb_of_edges + nb_of_faces + nb_of_volumes;
87         SCRUTE(nb_of_edges);
88         SCRUTE(nb_of_faces);
89         SCRUTE(nb_of_volumes);
90
91         fprintf(stdout, "%d %d\n", nbNodes, nbCells);
92         fprintf(myFileId, "%d %d\n", nbNodes, nbCells);
93
94   /****************************************************************************
95   *                       ECRITURE DES NOEUDS                                 *
96   ****************************************************************************/
97         fprintf(stdout, "\n(************************)\n");
98         fprintf(stdout, "(* NOEUDS DU MAILLAGE : *)\n");
99         fprintf(stdout, "(************************)\n");
100
101         fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
102         fprintf(myFileId, "%s\n", sNODE_UNV_ID);
103
104         SMDS_Iterator<const SMDS_MeshNode *> * itNodes=myMesh->nodesIterator();
105         while(itNodes->more())
106         {
107                 const SMDS_MeshNode * node = itNodes->next();
108
109                 fprintf(myFileId, sNODE_UNV_DESCR, node->GetID());
110                 fprintf(myFileId, "%25.16E%25.16E%25.16E\n", node->X(), node->Y(),
111                         node->Z());
112         }
113         delete itNodes;
114         fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
115
116   /****************************************************************************
117   *                       ECRITURE DES ELEMENTS                                *
118   ****************************************************************************/
119         fprintf(stdout, "\n(**************************)\n");
120         fprintf(stdout, "(* ELEMENTS DU MAILLAGE : *)\n");
121         fprintf(stdout, "(**************************)");
122         /* Ecriture des connectivites, noms, numeros des mailles */
123
124         fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
125         fprintf(myFileId, "%s\n", sELT_UNV_ID);
126
127         SMDS_Iterator<const SMDS_MeshEdge *> * itEdges=myMesh->edgesIterator();
128         while(itEdges->more())
129         {
130                 const SMDS_MeshElement * elem = itEdges->next();
131                 SMDS_Iterator<const SMDS_MeshElement*> *itn=elem->nodesIterator();
132
133                 switch (elem->NbNodes())
134                 {
135                 case 2:
136                         fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 21,
137                                 elem->NbNodes());
138                         fprintf(myFileId, sELT_BEAM_DESC2);
139                         fprintf(myFileId, "%10d%10d\n", itn->next()->GetID(),
140                                 itn->next()->GetID());
141                         break;
142
143                 case 3:
144                         fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 24,
145                                 elem->NbNodes());
146                         fprintf(myFileId, sELT_BEAM_DESC2);
147                         fprintf(myFileId, "%10d%10d%10d\n", itn->next()->GetID(),
148                                 itn->next()->GetID(), itn->next()->GetID());
149
150                         break;
151                 }
152                 delete itn;
153         }
154         delete itEdges;
155
156         SMDS_Iterator<const SMDS_MeshFace*> * itFaces=myMesh->facesIterator();
157         while(itFaces->more())
158         {
159                 const SMDS_MeshElement * elem = itFaces->next();
160
161                 switch (elem->NbNodes())
162                 {
163                 case 3:
164                         // linear triangle
165                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 74,
166                                 elem->NbNodes());
167                         break;
168                 case 4:
169                         // linear quadrilateral
170                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 71,
171                                 elem->NbNodes());
172                         break;
173                 case 6:
174                         // parabolic triangle
175                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 72,
176                                 elem->NbNodes());
177                         break;
178                 case 8:
179                         // parabolic quadrilateral
180                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 75,
181                                 elem->NbNodes());
182                         break;
183                 default:
184                         fprintf(myFileId, "element not registered\n");
185                 }
186
187                 SMDS_Iterator<const SMDS_MeshElement*> *itn=elem->nodesIterator();
188                 while(itn->more()) fprintf(myFileId, "%10d", itn->next()->GetID());
189                 delete itn;
190
191                 fprintf(myFileId, "\n");
192         }
193         delete itFaces;
194
195         SMDS_Iterator<const SMDS_MeshVolume*> * itVolumes=myMesh->volumesIterator();
196         while(itVolumes->more())
197         {
198                 const SMDS_MeshElement * elem = itVolumes->next();
199
200                 switch (elem->NbNodes())
201                 {
202                 case 4:
203                         // linear tetrahedron
204                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 111,
205                                 elem->NbNodes());
206                         break;
207                 case 6:
208                         // linear tetrahedron
209                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 112,
210                                 elem->NbNodes());
211                         break;
212                 case 8:
213                         // linear brick
214                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 115,
215                                 elem->NbNodes());
216                         break;
217                 }
218
219                 SMDS_Iterator<const SMDS_MeshElement*> *itn=elem->nodesIterator();
220                 while(itn->more()) fprintf(myFileId, "%10d", itn->next()->GetID());
221                 delete itn;
222
223                 fprintf(myFileId, "\n");
224         }
225         delete itVolumes;
226
227         fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
228
229         fclose(myFileId);
230 }