Salome HOME
NRI : Add KERNEL includes.
[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 #include "SMDS_MeshEdgesIterator.hxx"
7 #include "SMDS_MeshFacesIterator.hxx"
8 #include "SMDS_MeshNodesIterator.hxx"
9 #include "SMDS_MeshVolumesIterator.hxx"
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 DriverUNV_W_SMDS_Mesh::~DriverUNV_W_SMDS_Mesh() {
27 ;
28 }
29
30 void DriverUNV_W_SMDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) {
31   myMesh = aMesh;
32 }
33
34 void DriverUNV_W_SMDS_Mesh::SetFile(string aFile) {
35   myFile = aFile;
36 }
37
38 void DriverUNV_W_SMDS_Mesh::SetFileId(FILE* aFileId) {
39   myFileId = aFileId;
40 }
41
42 void DriverUNV_W_SMDS_Mesh::SetMeshId(int aMeshId) {
43   myMeshId = aMeshId;
44 }
45
46 void DriverUNV_W_SMDS_Mesh::Add() {
47   ;
48 }
49
50 void DriverUNV_W_SMDS_Mesh::Write() {
51
52   int nbNodes,nbCells;
53   int i;
54
55   char* file2Read = (char*)myFile.c_str();
56   myFileId = fopen(file2Read,"w+");
57   if (myFileId < 0)
58     {
59       fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read);
60       exit(EXIT_FAILURE);
61     }
62   SCRUTE(myMesh);
63   /****************************************************************************
64   *                       NOMBRES D'OBJETS                                    *
65   ****************************************************************************/
66   fprintf(stdout,"\n(****************************)\n");
67   fprintf(stdout,"(* INFORMATIONS GENERALES : *)\n");
68   fprintf(stdout,"(****************************)\n");
69
70   /* Combien de noeuds ? */
71   nbNodes = myMesh->NbNodes();
72
73   /* Combien de mailles, faces ou aretes ? */
74   Standard_Integer nb_of_nodes, nb_of_edges,nb_of_faces, nb_of_volumes;
75   nb_of_edges = myMesh->NbEdges();
76   nb_of_faces = myMesh->NbFaces();
77   nb_of_volumes = myMesh->NbVolumes();
78   nbCells = nb_of_edges + nb_of_faces + nb_of_volumes;
79   SCRUTE(nb_of_edges);
80   SCRUTE(nb_of_faces);
81   SCRUTE(nb_of_volumes);
82
83   fprintf(stdout,"%d %d\n",nbNodes,nbCells);
84   fprintf(myFileId,"%d %d\n",nbNodes,nbCells);
85
86   /****************************************************************************
87   *                       ECRITURE DES NOEUDS                                 *
88   ****************************************************************************/
89   fprintf(stdout,"\n(************************)\n");
90   fprintf(stdout,"(* NOEUDS DU MAILLAGE : *)\n");
91   fprintf(stdout,"(************************)\n");
92
93   SMDS_MeshNodesIterator itNodes(myMesh);
94
95   fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
96   fprintf(myFileId,"%s\n", sNODE_UNV_ID  );
97
98   for (;itNodes.More();itNodes.Next()) {
99     const Handle(SMDS_MeshElement)& elem = itNodes.Value();
100     const Handle(SMDS_MeshNode   )& node = myMesh->GetNode(1, elem);
101
102     fprintf(myFileId, sNODE_UNV_DESCR, node->GetID());
103     fprintf(myFileId, "%25.16E%25.16E%25.16E\n", node->X(), node->Y(), node->Z());
104   }
105   fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
106
107   /****************************************************************************
108   *                       ECRITURE DES ELEMENTS                                *
109   ****************************************************************************/
110   fprintf(stdout,"\n(**************************)\n");
111   fprintf(stdout,"(* ELEMENTS DU MAILLAGE : *)\n");
112   fprintf(stdout,"(**************************)");
113   /* Ecriture des connectivites, noms, numeros des mailles */
114
115   fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
116   fprintf(myFileId,"%s\n", sELT_UNV_ID   );
117
118   SMDS_MeshEdgesIterator itEdges(myMesh);
119   for (;itEdges.More();itEdges.Next()) {
120     const Handle(SMDS_MeshElement)& elem = itEdges.Value();
121
122     switch (elem->NbNodes()) {
123       case 2 : {
124         fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 21, elem->NbNodes());
125         fprintf(myFileId, sELT_BEAM_DESC2);
126         fprintf(myFileId, "%10d%10d\n", elem->GetConnection(1), elem->GetConnection(2));
127         break;
128       } 
129       case 3 : {
130         fprintf(myFileId, sELT_BEAM_DESC1, elem->GetID(), 24, elem->NbNodes());
131         fprintf(myFileId, sELT_BEAM_DESC2);
132         fprintf(myFileId, "%10d%10d%10d\n",elem->GetConnection(1), elem->GetConnection(2), elem->GetConnection(3));
133         break;
134       }
135     }
136   }
137
138   SMDS_MeshFacesIterator itFaces(myMesh);
139   for (;itFaces.More();itFaces.Next()) {
140     const Handle(SMDS_MeshElement)& elem = itFaces.Value();
141
142     switch (elem->NbNodes()) {
143       case 3 :
144         // linear triangle
145         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 74, elem->NbNodes());
146         break;
147       case 4 :
148         // linear quadrilateral
149         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 71, elem->NbNodes());
150         break;
151       case 6 :
152         // parabolic triangle
153         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 72, elem->NbNodes());
154         break;
155       case 8 :
156         // parabolic quadrilateral
157         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 75, elem->NbNodes());
158         break;
159       default:
160         fprintf(myFileId, "element not registered\n");
161     }
162
163     for (i=0;i<elem->NbNodes();i++)
164       fprintf(myFileId,"%10d",elem->GetConnection(i+1));
165     
166     fprintf(myFileId,"\n");
167   }
168
169   SMDS_MeshVolumesIterator itVolumes(myMesh);
170   for (;itVolumes.More();itVolumes.Next()) {
171     const Handle(SMDS_MeshElement)& elem = itVolumes.Value();
172
173     switch (elem->NbNodes()) {
174       case 4 : 
175         // linear tetrahedron
176         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 111, elem->NbNodes());
177         break;
178       case 6 : 
179         // linear tetrahedron
180         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 112, elem->NbNodes());
181         break;
182       case 8 : 
183         // linear brick
184         fprintf(myFileId, sELT_SURF_DESC, elem->GetID(), 115, elem->NbNodes());
185         break;
186     }
187
188     for (i=0;i<elem->NbNodes();i++)
189       fprintf(myFileId,"%10d",elem->GetConnection(i+1));
190     
191     fprintf(myFileId,"\n");
192   }
193   fprintf(myFileId,"%s\n", sUNV_SEPARATOR);
194
195   fclose (myFileId);
196 }