Salome HOME
Initialisation module SMESH_SRC de la base SMESH
[modules/smesh.git] / src / DriverMED / DriverMED_R_SMDS_Mesh.cxx
1 #include "DriverMED_R_SMDS_Mesh.h"
2 #include "utilities.h"
3
4 DriverMED_R_SMDS_Mesh::DriverMED_R_SMDS_Mesh() {
5 }
6
7 DriverMED_R_SMDS_Mesh::~DriverMED_R_SMDS_Mesh() {
8 ;
9 }
10
11 void DriverMED_R_SMDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) {
12   myMesh = aMesh;
13 }
14
15 void DriverMED_R_SMDS_Mesh::SetFile(string aFile) {
16   myFileId = -1;
17   myFile = aFile;
18 }
19
20 void DriverMED_R_SMDS_Mesh::SetFileId(med_idt aFileId) {
21   myFileId = aFileId;
22 }
23
24 void DriverMED_R_SMDS_Mesh::SetMeshId(int aMeshId) {
25   myMeshId = aMeshId;
26 }
27
28 void DriverMED_R_SMDS_Mesh::Add() {
29   ;
30 }
31
32 void DriverMED_R_SMDS_Mesh::Read() {
33
34   med_err ret = 0;
35   int i,j,k,l;
36   int numero;
37   char message[200];
38   Standard_Boolean ok;
39   /* nombre d'objets MED */
40   char nom_universel[MED_TAILLE_LNOM+1];
41   med_int long_fichier_en_tete; 
42   char *fichier_en_tete;
43   char version_hdf[10];
44   char version_med[10];
45   med_int nmaa,mdim,nnoe;
46   med_int nmai[MED_NBR_GEOMETRIE_MAILLE],nfac[MED_NBR_GEOMETRIE_FACE];
47   med_int nare[MED_NBR_GEOMETRIE_ARETE];
48   /* nom du maillage */
49   char nommaa[MED_TAILLE_NOM+1];
50   /* noeuds */
51   med_float *coo;
52   char nomcoo[3*MED_TAILLE_PNOM+1];
53   char unicoo[3*MED_TAILLE_PNOM+1];
54   char *nomnoe;
55   med_int *numnoe;
56   med_int *nufano; 
57   med_repere rep;
58   med_booleen inonoe,inunoe;
59   med_mode_switch mode_coo;
60   char str[MED_TAILLE_PNOM+1];
61   /* elements */
62   med_int nsup;
63   med_int edim;
64   med_int taille;
65   med_int elem_id;
66   med_int cmpt = 0;
67   med_int *connectivite;
68   char *nomele;
69   med_int *numele;
70   med_int *nufael;
71   med_booleen inoele, inuele;
72   med_connectivite typ_con;
73   med_geometrie_element typgeo;
74   med_geometrie_element typmai[MED_NBR_GEOMETRIE_MAILLE] = {MED_POINT1,MED_SEG2, 
75                                                    MED_SEG3,MED_TRIA3,
76                                                    MED_TRIA6,MED_QUAD4,
77                                                    MED_QUAD8,MED_TETRA4,
78                                                    MED_TETRA10,MED_HEXA8,
79                                                    MED_HEXA20,MED_PENTA6,
80                                                    MED_PENTA15,MED_PYRA5,
81                                                    MED_PYRA13};
82   med_int desmai[MED_NBR_GEOMETRIE_MAILLE] = {0,2,3,3,3,4,4,4,4,6,6,5,5,5,5};
83   med_int nmailles[MED_NBR_GEOMETRIE_MAILLE];
84   char nommai[MED_NBR_GEOMETRIE_MAILLE] [MED_TAILLE_NOM+1] = {"MED_POINT1",
85                                                           "MED_SEG2", 
86                                                           "MED_SEG3",
87                                                           "MED_TRIA3",
88                                                           "MED_TRIA6",
89                                                           "MED_QUAD4",
90                                                           "MED_QUAD8",
91                                                           "MED_TETRA4",
92                                                           "MED_TETRA10",
93                                                           "MED_HEXA8",
94                                                           "MED_HEXA20",
95                                                           "MED_PENTA6",
96                                                           "MED_PENTA15",
97                                                           "MED_PYRA5",
98                                                           "MED_PYRA13"};
99   med_geometrie_element typfac[MED_NBR_GEOMETRIE_FACE] = {MED_TRIA3,MED_TRIA6,
100                                                  MED_QUAD4,MED_QUAD8};
101   med_int desfac[MED_NBR_GEOMETRIE_FACE] = {3,3,4,4};
102   med_int nfaces[MED_NBR_GEOMETRIE_FACE];
103   char nomfac[MED_NBR_GEOMETRIE_FACE][MED_TAILLE_NOM+1] = {"MED_TRIA3","MED_TRIA6",
104                                                        "MED_QUAD4","MED_QUAD8"};
105   med_geometrie_element typare[MED_NBR_GEOMETRIE_ARETE] = {MED_SEG2,MED_SEG3};
106   med_int desare[MED_NBR_GEOMETRIE_ARETE] = {2,3};
107   med_int naretes[MED_NBR_GEOMETRIE_ARETE];
108   char nomare[MED_NBR_GEOMETRIE_ARETE] [MED_TAILLE_NOM+1] = {"MED_SEG2","MED_SEG3"};
109   /* familles */
110   med_int nfam;
111   med_int natt,ngro;
112   char *attdes,*gro;
113   med_int *attval,*attide;
114   char nomfam[MED_TAILLE_NOM+1];
115   med_int numfam;
116   char str1[MED_TAILLE_DESC+1];
117   char str2[MED_TAILLE_LNOM+1];
118   
119   char* file2Read;
120   bool locally_managed;
121   SCRUTE(myFileId);
122   if (myFileId==-1) 
123     locally_managed = true;
124   else
125     locally_managed = false;
126
127   if (locally_managed)
128     { 
129     file2Read = (char*)myFile.c_str();
130     myFileId = MEDouvrir(file2Read,MED_LECT);
131     if (myFileId < 0)
132       {
133         fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read);
134         exit(EXIT_FAILURE);
135       }
136     numero = 1;
137     }
138   else
139     numero = myMeshId;
140     
141   typ_con = MED_NOD;
142   mode_coo = MED_FULL_INTERLACE;
143   
144
145   /****************************************************************************
146   *                       NOMBRES D'OBJETS MED                                *
147   ****************************************************************************/
148   fprintf(stdout,"\n(****************************)\n");
149   fprintf(stdout,"(* INFORMATIONS GENERALES : *)\n");
150   fprintf(stdout,"(****************************)\n");
151
152   /* lecture du nom et de la dimension du maillage */
153   fprintf(stdout,"%d %d\n",myFileId,numero);
154   ret = MEDmaaInfo(myFileId,numero,nommaa,&mdim);
155   fprintf(stdout,"%d\n",ret);
156   if (ret < 0)
157     {
158       fprintf(stderr,">> ERREUR : lecture du nom du maillage \n");
159       exit(EXIT_FAILURE);
160     }
161   fprintf(stdout,"- Nom du maillage : <<%s>>\n",nommaa);
162   fprintf(stdout,"- Dimension du maillage : %d\n",mdim);
163
164   /* Combien de noeuds ? */
165   nnoe = MEDnEntMaa(myFileId,nommaa,MED_COOR,MED_NOEUD,MED_POINT1,typ_con);
166   if (nnoe < 0)
167     {
168       fprintf(stderr,">> ERREUR : lecture du nombre de noeuds \n");
169       exit(EXIT_FAILURE);
170     }
171   fprintf(stdout,"- Nombre de noeuds : %d \n",nnoe);
172
173   /* Combien de mailles, faces ou aretes ? */
174   for (i=0;i<MED_NBR_GEOMETRIE_MAILLE;i++)
175     {
176       nmailles[i] = MEDnEntMaa(myFileId,nommaa,MED_CONN,MED_MAILLE,typmai[i],
177                                typ_con);
178       if (nmailles[i] < 0)
179         {
180           fprintf(stderr,">> ERREUR : lecture du nombre de mailles \n");
181           exit(EXIT_FAILURE);
182         }
183       fprintf (stdout,"- Nombre de mailles de type %s : %d \n",nommai[i],nmailles[i]);
184     }
185
186   for (i=0;i<MED_NBR_GEOMETRIE_FACE;i++)
187     {
188       nfaces[i] = MEDnEntMaa(myFileId,nommaa,MED_CONN,MED_FACE,typfac[i],
189                              typ_con);
190       if (nfaces[i] < 0)
191         {
192           fprintf(stderr,">> ERREUR : lecture du nombre de faces \n");
193           exit(EXIT_FAILURE);
194         }
195       fprintf (stdout,"- Nombre de faces de type %s : %d \n",nomfac[i],nfaces[i]);
196     }    
197
198   for (i=0;i<MED_NBR_GEOMETRIE_ARETE;i++)
199     {
200       naretes[i] = MEDnEntMaa(myFileId,nommaa,MED_CONN,MED_ARETE,typare[i],
201                               typ_con); 
202       if (naretes[i] < 0)
203         {
204           fprintf(stderr,">> ERREUR : lecture du nombre d'aretes \n");
205           exit(EXIT_FAILURE);
206         }
207       fprintf (stdout,"- Nombre d'aretes de type %s : %d \n",nomare[i],naretes[i]);
208     }
209
210   /* nombre de familles */
211   nfam = MEDnFam(myFileId,nommaa,0,MED_FAMILLE);
212   if (nfam < 0)
213     {
214       fprintf(stderr,">> ERREUR : lecture du nombre de familles \n");
215       exit(EXIT_FAILURE);
216     }   
217   fprintf(stdout,"- Nombre de familles : %d \n",nfam);
218
219   /****************************************************************************
220   *                       LECTURE DES NOEUDS                                  *
221   ****************************************************************************/
222   fprintf(stdout,"\n(************************)\n");
223   fprintf(stdout,"(* NOEUDS DU MAILLAGE : *)\n");
224   fprintf(stdout,"(************************)\n");
225
226   /* Allocations memoires */
227   /* table des coordonnees 
228      profil : (dimension * nombre de noeuds ) */
229   coo = (med_float*) malloc(sizeof(med_float)*nnoe*mdim);
230   /* table  des numeros, des numeros de familles des noeuds
231      profil : (nombre de noeuds) */
232   numnoe = (med_int*) malloc(sizeof(med_int)*nnoe);
233   nufano = (med_int*) malloc(sizeof(med_int)*nnoe);
234   /* table des noms des noeuds 
235      profil : (nnoe*MED_TAILLE_PNOM+1) */
236   nomnoe = (char*) malloc(MED_TAILLE_PNOM*nnoe+1);
237
238   /* lecture des noeuds : 
239      - coordonnees
240      - noms (optionnel dans un fichier MED) 
241      - numeros (optionnel dans un fichier MED) 
242      - numeros des familles */
243   ret = MEDnoeudsLire(myFileId,nommaa,mdim,coo,mode_coo,&rep,
244                       nomcoo,unicoo,nomnoe,&inonoe,numnoe,&inunoe,
245                       nufano,nnoe);
246   if (ret < 0)
247     strcpy(message,">> ERREUR : lecture des noeuds \n");
248
249   if (inunoe) {
250     for (int i=0;i<nnoe;i++) {
251       ok = myMesh->AddNodeWithID(coo[i*3],coo[i*3+1],coo[i*3+2],numnoe[i]);
252       //fprintf(Out,"%d %f %f %f\n",numnoe[i],coo[i*3],coo[i*3+1],coo[i*3+2]);
253     }
254   }
255   else {
256     for (int i=0;i<nnoe;i++) {
257       ok = myMesh->AddNodeWithID(coo[i*3],coo[i*3+1],coo[i*3+2],i+1);
258       //fprintf(Out,"%d %f %f %f\n",numnoe[i],coo[i*3],coo[i*3+1],i);
259     }
260   }
261
262   //fprintf(stdout,"\n- Numeros des familles des noeuds : \n");
263   //for (i=0;i<nnoe;i++)
264   //fprintf(stdout," %d ",*(nufano+i));
265   //fprintf(stdout,"\n");
266
267   SCRUTE(myMesh->NbNodes());
268
269   /* liberation memoire */
270   free(coo);
271   free(nomnoe);
272   free(numnoe);
273   free(nufano);
274
275   /****************************************************************************
276   *                       LECTURE DES ELEMENTS                                *
277   ****************************************************************************/
278   fprintf(stdout,"\n(**************************)\n");
279   fprintf(stdout,"(* ELEMENTS DU MAILLAGE : *)\n");
280   fprintf(stdout,"(**************************)");
281   //fprintf(Out,"CELLS\n");
282   /* Lecture des connectivites, noms, numeros des mailles */
283   //printf("%d %d %d %d\n",nmailles[3],nmailles[4],nmailles[5],nmailles[9]);
284
285   if (ret == 0)
286     for (i=0;i<MED_NBR_GEOMETRIE_MAILLE;i++)
287       {
288         if (nmailles[i] > 0 && ret == 0)
289           {
290             /* dimension de la maille */
291             edim = typmai[i] / 100;
292             nsup = 0;
293             if (mdim  == 2 || mdim == 3)
294               if (edim == 1)
295                 nsup = 1;
296             if (mdim == 3)
297               if (edim == 2)
298                 nsup = 1;
299
300             taille = nsup+typmai[i]%100;
301             //taille = typmai[i]%100;
302             
303             /* allocation memoire */
304             connectivite = (med_int*)malloc(sizeof(med_int)*
305                                             taille*nmailles[i]);
306             nomele = (char*)malloc(sizeof(char)*MED_TAILLE_PNOM*
307                                    nmailles[i]+1);
308             numele = (med_int*)malloc(sizeof(med_int)*
309                                       nmailles[i]);
310             nufael = (med_int*)malloc(sizeof(med_int)*
311                                       nmailles[i]);
312             
313             /* lecture des données */
314             ret = MEDelementsLire(myFileId,nommaa,mdim,connectivite,mode_coo,
315                                   nomele,&inoele,numele,&inuele,nufael,
316                                   nmailles[i],MED_MAILLE,typmai[i],
317                                   typ_con);
318             SCRUTE(typmai[i]);
319             switch (typmai[i])
320               {
321               case MED_SEG2  : {
322                 if (inuele) {
323                   for (j=0;j<nmailles[i];j++) {
324                     elem_id=*(numele+j);
325                     ok = myMesh->AddEdgeWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),elem_id);
326                   }
327                 }
328                 else {
329                   for (j=0;j<nmailles[i];j++) {
330                     cmpt++;
331                     ok = myMesh->AddEdgeWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),cmpt);
332                   }
333                 }
334
335                 break;
336               }
337               case MED_TRIA3  : {
338                 if (inuele) {
339                   for (j=0;j<nmailles[i];j++) {
340                     elem_id=*(numele+j);
341                     ok = myMesh->AddFaceWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),elem_id);
342                     //fprintf(Out,"%d %d %d %d\n",elem_id,*(connectivite+j*(taille-nsup)),*(connectivite+j*(taille-nsup)+1),*(connectivite+j*(taille-nsup)+2));
343                   }
344                 }
345                 else {
346                   for (j=0;j<nmailles[i];j++) {
347                     cmpt++;
348                     ok = myMesh->AddFaceWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),cmpt);
349                     //fprintf(Out,"%d %d %d %d\n",j,*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2));
350                   }
351                 }
352
353                 break;
354               }
355               case MED_QUAD4  : {
356                 if (inuele) {
357                   for (j=0;j<nmailles[i];j++) {
358                     elem_id=*(numele+j);
359                     ok = myMesh->AddFaceWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3),elem_id);
360                     //fprintf(Out,"%d %d %d %d\n",elem_id,*(connectivite+j*(taille-nsup)),*(connectivite+j*(taille-nsup)+1),*(connectivite+j*(taille-nsup)+2),*(connectivite+j*(taille-nsup)+3));
361                   }
362                 }
363                 else {
364                   for (j=0;j<nmailles[i];j++) {
365                     cmpt++;
366                     ok = myMesh->AddFaceWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3),cmpt);
367                     //fprintf(Out,"%d %d %d %d\n",j,*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3));
368                   }
369                 }
370                 break;
371               }
372               case MED_HEXA8  : {
373                 if (inuele) {
374                   for (j=0;j<nmailles[i];j++) {
375                     elem_id=*(numele+j);
376                     ok = myMesh->AddVolumeWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3),*(connectivite+j*(taille)+4),*(connectivite+j*(taille)+5),*(connectivite+j*(taille)+6),*(connectivite+j*(taille)+7),elem_id);
377                     //fprintf(Out,"%d %d %d %d\n",elem_id,*(connectivite+j*(taille-nsup)),*(connectivite+j*(taille-nsup)+1),*(connectivite+j*(taille-nsup)+2),*(connectivite+j*(taille-nsup)+3),*(connectivite+j*(taille-nsup)+4),*(connectivite+j*(taille-nsup)+5),*(connectivite+j*(taille-nsup)+6),*(connectivite+j*(taille-nsup)+7));
378                   }
379                 }
380                 else {
381                   for (j=0;j<nmailles[i];j++) {
382                     cmpt++;
383                     ok = myMesh->AddVolumeWithID(*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3),*(connectivite+j*(taille)+4),*(connectivite+j*(taille)+5),*(connectivite+j*(taille)+6),*(connectivite+j*(taille)+7),cmpt);
384                     //fprintf(Out,"%d %d %d %d\n",j,*(connectivite+j*(taille)),*(connectivite+j*(taille)+1),*(connectivite+j*(taille)+2),*(connectivite+j*(taille)+3),*(connectivite+j*(taille)+4),*(connectivite+j*(taille)+5),*(connectivite+j*(taille)+6),*(connectivite+j*(taille)+7));
385                   }
386                 }
387                 break;
388               }
389               default : {
390                 break ;
391               }
392               }
393
394             //fprintf(stdout,"\n  - Numéros de familles : \n");
395             //for (j=0;j<nmailles[i];j++)
396             //fprintf(stdout," %d ",*(nufael+j));
397             
398             /* liberation memoire */
399             free(connectivite);
400             free(nomele);
401             free(numele);
402             free(nufael);
403           }
404       }
405
406   SCRUTE(myMesh->NbEdges());
407   SCRUTE(myMesh->NbFaces());
408   /****************************************************************************
409    *                       LECTURE DES FAMILLES                                *
410    ****************************************************************************/
411   printf("\n(*************************)\n");
412   printf("(* FAMILLES DU MAILLAGE : *)\n");
413   printf("(*************************)\n");
414   if (ret == 0)
415     for (i=0;i<nfam;i++)
416       {
417         
418         /* nombre de groupes */
419         ngro = MEDnFam(myFileId,nommaa,i+1,MED_GROUPE);
420         if (ngro < 0)  
421           {
422             ret = -1;
423             strcpy(message,
424                    ">> ERREUR : lecture du nombre de groupes d'une famille \n");
425           }
426         
427         /* nombre d'attributs */
428         if (ret == 0)
429           {
430             natt = MEDnFam(myFileId,nommaa,i+1,MED_ATTR);
431             if (natt < 0)
432               {
433                 ret = -1;
434                 strcpy(message,
435                    ">> ERREUR : lecture du nombre d'attributs d'une famille\n");
436               }
437           }
438
439         if (ret == 0)
440           fprintf(stdout,"- Famille %d a %d attributs et %d groupes \n",i+1,natt,ngro); 
441
442         /* nom,numero,attributs,groupes */
443         if (ret == 0)
444           {
445             attide = (med_int*) malloc(sizeof(med_int)*natt);
446             attval = (med_int*) malloc(sizeof(med_int)*natt);       
447             attdes = (char *) malloc(MED_TAILLE_DESC*natt+1);
448             gro = (char*) malloc(MED_TAILLE_LNOM*ngro+1);
449             ret = MEDfamInfo(myFileId,nommaa,i+1,nomfam,&numfam,attide,attval,
450                              attdes,&natt,gro,&ngro);
451             fprintf(stdout,"  - Famille de nom %s et de numero %d : \n",nomfam,numfam);
452             fprintf(stdout,"  - Attributs : \n");
453             for (j=0;j<natt;j++)
454               {
455                 strncpy(str1,attdes+j*MED_TAILLE_DESC,MED_TAILLE_DESC);
456                 str1[MED_TAILLE_DESC] = '\0';
457                 fprintf(stdout,"   ide = %d - val = %d - des = %s\n",*(attide+j),
458                        *(attval+j),str1);
459               }
460             free(attide);
461             free(attval);
462             free(attdes);
463             fprintf(stdout,"  - Groupes :\n");
464             for (j=0;j<ngro;j++)
465               {
466                 strncpy(str2,gro+j*MED_TAILLE_LNOM,MED_TAILLE_LNOM);
467                 str2[MED_TAILLE_LNOM] = '\0';
468                 fprintf(stdout,"   gro = %s\n",str2);
469               }
470             free(gro);
471           }
472       }
473
474   if (locally_managed)
475     ret = MEDfermer(myFileId);
476
477 }