Salome HOME
DCQ : Merge with Ecole_Ete_a6.
[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_NodeIteratorPtr 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         fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
114
115   /****************************************************************************
116   *                       ECRITURE DES ELEMENTS                                *
117   ****************************************************************************/
118         fprintf(stdout, "\n(**************************)\n");
119         fprintf(stdout, "(* ELEMENTS DU MAILLAGE : *)\n");
120         fprintf(stdout, "(**************************)");
121         /* Ecriture des connectivites, noms, numeros des mailles */
122
123         fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
124         fprintf(myFileId, "%s\n", sELT_UNV_ID);
125
126         SMDS_EdgeIteratorPtr itEdges=myMesh->edgesIterator();
127         while(itEdges->more())
128         {
129                 const SMDS_MeshElement * elem = itEdges->next();
130                 SMDS_ElemIteratorPtr itn=elem->nodesIterator();
131
132                 switch (elem->NbNodes())
133                 {
134                 case 2:
135                         fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 21,
136                                 elem->NbNodes());
137                         fprintf(myFileId, sELT_BEAM_DESC2);
138                         fprintf(myFileId, "%10d%10d\n", itn->next()->GetID(),
139                                 itn->next()->GetID());
140                         break;
141
142                 case 3:
143                         fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 24,
144                                 elem->NbNodes());
145                         fprintf(myFileId, sELT_BEAM_DESC2);
146                         fprintf(myFileId, "%10d%10d%10d\n", itn->next()->GetID(),
147                                 itn->next()->GetID(), itn->next()->GetID());
148
149                         break;
150                 }
151         }
152
153         SMDS_FaceIteratorPtr itFaces=myMesh->facesIterator();
154         while(itFaces->more())
155         {
156                 const SMDS_MeshElement * elem = itFaces->next();
157
158                 switch (elem->NbNodes())
159                 {
160                 case 3:
161                         // linear triangle
162                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 74,
163                                 elem->NbNodes());
164                         break;
165                 case 4:
166                         // linear quadrilateral
167                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 71,
168                                 elem->NbNodes());
169                         break;
170                 case 6:
171                         // parabolic triangle
172                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 72,
173                                 elem->NbNodes());
174                         break;
175                 case 8:
176                         // parabolic quadrilateral
177                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 75,
178                                 elem->NbNodes());
179                         break;
180                 default:
181                         fprintf(myFileId, "element not registered\n");
182                 }
183
184                 SMDS_ElemIteratorPtr itn=elem->nodesIterator();
185                 while(itn->more()) fprintf(myFileId, "%10d", itn->next()->GetID());
186
187                 fprintf(myFileId, "\n");
188         }
189
190         SMDS_VolumeIteratorPtr itVolumes=myMesh->volumesIterator();
191         while(itVolumes->more())
192         {
193                 const SMDS_MeshElement * elem = itVolumes->next();
194
195                 switch (elem->NbNodes())
196                 {
197                 case 4:
198                         // linear tetrahedron
199                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 111,
200                                 elem->NbNodes());
201                         break;
202                 case 6:
203                         // linear tetrahedron
204                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 112,
205                                 elem->NbNodes());
206                         break;
207                 case 8:
208                         // linear brick
209                         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 115,
210                                 elem->NbNodes());
211                         break;
212                 }
213
214                 SMDS_ElemIteratorPtr itn=elem->nodesIterator();
215                 while(itn->more()) fprintf(myFileId, "%10d", itn->next()->GetID());
216
217                 fprintf(myFileId, "\n");
218         }
219
220         fprintf(myFileId, "%s\n", sUNV_SEPARATOR);
221
222         fclose(myFileId);
223 }