Salome HOME
NRI : Add KERNEL includes.
[modules/smesh.git] / src / DriverDAT / DriverDAT_W_SMESHDS_Mesh.cxx
1 using namespace std;
2 #include "DriverDAT_W_SMESHDS_Mesh.h"
3 #include "DriverDAT_W_SMDS_Mesh.h"
4
5 #include "SMDS_MeshElement.hxx"
6 #include "SMDS_MeshNode.hxx"
7 #include "SMDS_MeshEdgesIterator.hxx"
8 #include "SMDS_MeshFacesIterator.hxx"
9 #include "SMDS_MeshNodesIterator.hxx"
10 #include "SMDS_MeshVolumesIterator.hxx"
11
12 #include "utilities.h"
13
14 DriverDAT_W_SMESHDS_Mesh::DriverDAT_W_SMESHDS_Mesh() {
15 ;
16 }
17
18 DriverDAT_W_SMESHDS_Mesh::~DriverDAT_W_SMESHDS_Mesh() {
19 ;
20 }
21
22 void DriverDAT_W_SMESHDS_Mesh::SetMesh(Handle(SMDS_Mesh)& aMesh) {
23   //myMesh = Handle(SMESHDS_Mesh)::DownCast(aMesh);
24   myMesh = aMesh;
25 }
26
27 void DriverDAT_W_SMESHDS_Mesh::SetFile(string aFile) {
28   myFile = aFile;
29 }
30
31 void DriverDAT_W_SMESHDS_Mesh::SetFileId(FILE* aFileId) {
32   myFileId = aFileId;
33 }
34
35 void DriverDAT_W_SMESHDS_Mesh::SetMeshId(int aMeshId) {
36   myMeshId = aMeshId;
37 }
38
39 void DriverDAT_W_SMESHDS_Mesh::Write() {
40
41   string myClass = string("SMDS_Mesh");
42   string myExtension = string("DAT");
43
44   DriverDAT_W_SMDS_Mesh* myWriter = new DriverDAT_W_SMDS_Mesh;
45
46   myWriter->SetMesh(myMesh);
47   myWriter->SetFile(myFile);
48   myWriter->SetMeshId(myMeshId);
49   //myWriter->SetFileId(myFileId);
50
51   myWriter->Write();
52
53
54 }
55 void DriverDAT_W_SMESHDS_Mesh::Add() {
56   int nbNodes,nbCells;
57   int i;
58
59   char* file2Read = (char*)myFile.c_str();
60   myFileId = fopen(file2Read,"w+");
61   if (myFileId < 0)
62     {
63       fprintf(stderr,">> ERREUR : ouverture du fichier %s \n",file2Read);
64       exit(EXIT_FAILURE);
65     }
66   
67   /****************************************************************************
68   *                       NOMBRES D'OBJETS                                    *
69   ****************************************************************************/
70   fprintf(stdout,"\n(****************************)\n");
71   fprintf(stdout,"(* INFORMATIONS GENERALES : *)\n");
72   fprintf(stdout,"(****************************)\n");
73
74   /* Combien de noeuds ? */
75   nbNodes = myMesh->NbNodes();
76
77   /* Combien de mailles, faces ou aretes ? */
78   Standard_Integer nb_of_nodes, nb_of_edges,nb_of_faces, nb_of_volumes;
79   nb_of_edges = myMesh->NbEdges();
80   nb_of_faces = myMesh->NbFaces();
81   nb_of_volumes = myMesh->NbVolumes();
82   nbCells = nb_of_edges + nb_of_faces + nb_of_volumes;
83
84   fprintf(stdout,"%d %d\n",nbNodes,nbCells);
85   fprintf(myFileId,"%d %d\n",nbNodes,nbCells);
86
87   /****************************************************************************
88   *                       ECRITURE DES NOEUDS                                 *
89   ****************************************************************************/
90   fprintf(stdout,"\n(************************)\n");
91   fprintf(stdout,"(* NOEUDS DU MAILLAGE : *)\n");
92   fprintf(stdout,"(************************)\n");
93
94   SMDS_MeshNodesIterator itNodes(myMesh);
95   for (;itNodes.More();itNodes.Next()) {
96     const Handle(SMDS_MeshElement)& elem = itNodes.Value();
97     const Handle(SMDS_MeshNode)& node = myMesh->GetNode(1,elem);
98  
99     fprintf(myFileId,"%d %e %e %e\n",node->GetID(),node->X(),node->Y(),node->Z());
100   }
101
102   /****************************************************************************
103   *                       ECRITURE DES ELEMENTS                                *
104   ****************************************************************************/
105   fprintf(stdout,"\n(**************************)\n");
106   fprintf(stdout,"(* ELEMENTS DU MAILLAGE : *)\n");
107   fprintf(stdout,"(**************************)");
108   /* Ecriture des connectivites, noms, numeros des mailles */
109
110   SMDS_MeshEdgesIterator itEdges(myMesh);
111   for (;itEdges.More();itEdges.Next()) {
112     const Handle(SMDS_MeshElement)& elem = itEdges.Value();
113
114     switch (elem->NbNodes()) {
115     case 2 : {
116       fprintf(myFileId,"%d %d ",elem->GetID(),102);
117       break;
118     } 
119     case 3 : {
120       fprintf(myFileId,"%d %d ",elem->GetID(),103);
121       break;
122     }
123     }
124
125     for (i=0;i<elem->NbNodes();i++)
126       fprintf(myFileId,"%d ",elem->GetConnection(i+1));
127     
128     fprintf(myFileId,"\n");
129   }
130
131   SMDS_MeshFacesIterator itFaces(myMesh);
132   for (;itFaces.More();itFaces.Next()) {
133     const Handle(SMDS_MeshElement)& elem = itFaces.Value();
134
135     switch (elem->NbNodes()) {
136     case 3 : {
137       fprintf(myFileId,"%d %d ",elem->GetID(),203);
138       break;
139     }
140     case 4 : {
141       fprintf(myFileId,"%d %d ",elem->GetID(),204);
142       break;
143     }
144     case 6 : {
145       fprintf(myFileId,"%d %d ",elem->GetID(),206);
146       break;
147     }
148     }
149
150     for (i=0;i<elem->NbNodes();i++)
151       fprintf(myFileId,"%d ",elem->GetConnection(i+1));
152     
153     fprintf(myFileId,"\n");
154   }
155
156   SMDS_MeshVolumesIterator itVolumes(myMesh);
157   for (;itVolumes.More();itVolumes.Next()) {
158     const Handle(SMDS_MeshElement)& elem = itVolumes.Value();
159
160     switch (elem->NbNodes()) {
161     case 8 : {
162       fprintf(myFileId,"%d %d ",elem->GetID(),308);
163       break;
164     }
165     }
166
167     for (i=0;i<elem->NbNodes();i++)
168       fprintf(myFileId,"%d ",elem->GetConnection(i+1));
169     
170     fprintf(myFileId,"\n");
171   }
172
173   fclose (myFileId);
174 }
175