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