Salome HOME
Fix PAL8562: rpath (rpath-link) option needs parameter - directory to search shared...
[modules/smesh.git] / src / DriverSTL / DriverSTL_R_SMDS_Mesh.cxx
1 //  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
2 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS 
3 // 
4 //  This library is free software; you can redistribute it and/or 
5 //  modify it under the terms of the GNU Lesser General Public 
6 //  License as published by the Free Software Foundation; either 
7 //  version 2.1 of the License. 
8 // 
9 //  This library is distributed in the hope that it will be useful, 
10 //  but WITHOUT ANY WARRANTY; without even the implied warranty of 
11 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU 
12 //  Lesser General Public License for more details. 
13 // 
14 //  You should have received a copy of the GNU Lesser General Public 
15 //  License along with this library; if not, write to the Free Software 
16 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA 
17 // 
18 //  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org 
19
20 #include <stdio.h>
21
22 #include "DriverSTL_R_SMDS_Mesh.h"
23
24 #include "SMDS_Mesh.hxx"
25 #include "SMDS_MeshElement.hxx"
26 #include "SMDS_MeshNode.hxx"
27 #include <gp_Pnt.hxx>
28 #include <OSD_Path.hxx>
29 #include <OSD_File.hxx>
30 #include <OSD_FromWhere.hxx>
31 #include <OSD_Protection.hxx>
32 #include <OSD_SingleProtection.hxx>
33 #include <NCollection_DataMap.hxx>
34 #include <Standard_NoMoreObject.hxx>
35
36 #include "utilities.h"
37
38 static const int HEADER_SIZE           =  84;
39 static const int SIZEOF_STL_FACET      =  50;
40 //static const int STL_MIN_FILE_SIZE     = 284;
41 static const int ASCII_LINES_PER_FACET =   7;
42
43 static Standard_Real tab1[3];
44 static Standard_Real tab2[3];
45
46 typedef NCollection_DataMap<gp_Pnt,SMDS_MeshNode*> DriverSTL_DataMapOfPntNodePtr;
47 //typedef NCollection_BaseCollection<SMDS_MeshNodePtr> DriverSTL_ColOfNodePtr;
48
49 //=======================================================================
50 //function : HashCode
51 //purpose  : 
52 //=======================================================================
53 inline Standard_Integer HashCode
54   (const gp_Pnt& point, const Standard_Integer Upper)
55 {
56   union 
57     {
58     Standard_Real R[3];
59     Standard_Integer I[6];
60     } U;
61
62   point.Coord(U.R[0],U.R[1],U.R[2]);  
63
64   return ::HashCode(U.I[0]/23+U.I[1]/19+U.I[2]/17+U.I[3]/13+U.I[4]/11+U.I[5]/7,Upper);
65 }
66
67 //=======================================================================
68 //function : IsEqual
69 //purpose  : 
70 //=======================================================================
71 inline Standard_Boolean IsEqual
72   (const gp_Pnt& point1, const gp_Pnt& point2)
73 {
74   point1.Coord(tab1[0],tab1[1],tab1[2]);  
75   point2.Coord(tab2[0],tab2[1],tab2[2]);  
76   return (memcmp(tab1,tab2,sizeof(tab1)) == 0);
77 }
78
79 //=======================================================================
80 //function : DriverSTL_R_SMDS_Mesh
81 //purpose  : 
82 //=======================================================================
83
84 DriverSTL_R_SMDS_Mesh::DriverSTL_R_SMDS_Mesh()
85 {
86   myIsCreateFaces = true;
87   myIsAscii = Standard_True;
88 }
89
90 //=======================================================================
91 //function : SetIsCreateFaces
92 //purpose  : 
93 //=======================================================================
94
95 void DriverSTL_R_SMDS_Mesh::SetIsCreateFaces( const bool theIsCreate )
96 { myIsCreateFaces = theIsCreate; }
97
98 //=======================================================================
99 //function : Perform
100 //purpose  : 
101 //=======================================================================
102
103 Driver_Mesh::Status DriverSTL_R_SMDS_Mesh::Perform()
104 {
105   Status aResult = DRS_OK;
106
107   TCollection_AsciiString aFileName( (char *)myFile.c_str() );
108   if ( aFileName.IsEmpty() ) {
109     fprintf(stderr, ">> ERREOR : invalid file name \n");
110     return DRS_FAIL;
111   }
112
113   filebuf fic;
114   Standard_IStream is(&fic);
115   if (!fic.open(aFileName.ToCString(),ios::in)) {
116     fprintf(stderr, ">> ERROR : cannot open file %s \n", aFileName.ToCString());
117     return DRS_FAIL;
118   }
119
120
121   OSD_Path aPath( aFileName );
122   OSD_File file = OSD_File( aPath );
123   file.Open(OSD_ReadOnly,OSD_Protection(OSD_RWD,OSD_RWD,OSD_RWD,OSD_RWD));
124   unsigned char str[128];
125   Standard_Integer lread,i;
126   Standard_Address ach;
127   ach = (Standard_Address)str;
128   // we skip the header which is in Ascii for both modes
129   file.Read(ach,HEADER_SIZE,lread);
130
131   // we read 128 characters to detect if we have a non-ascii char
132   file.Read(ach,sizeof(str),lread);
133   
134   myIsAscii = Standard_True;
135   for (i = 0; i < lread; ++i) {
136     if (str[i] > '~') {
137       myIsAscii = Standard_False;
138       break;
139     }
140   }
141       
142   file.Close();
143
144   if ( !myMesh ) {
145     fprintf(stderr, ">> ERREOR : cannot create mesh \n");
146     return DRS_FAIL;
147   }
148
149   if ( myIsAscii )
150     aResult = readAscii();
151   else
152     aResult = readBinary();
153
154   return aResult;
155 }
156
157 // static methods
158
159 static Standard_Real readFloat(OSD_File& theFile)
160 {
161   union {
162     Standard_Boolean i; 
163     Standard_ShortReal f;
164   }u;
165
166   char c[4];
167   Standard_Address adr;
168   adr = (Standard_Address)c;
169   Standard_Integer lread;
170   theFile.Read(adr,4,lread);
171   u.i  =  c[0] & 0xFF;
172   u.i |= (c[1] & 0xFF) << 0x08;
173   u.i |= (c[2] & 0xFF) << 0x10;
174   u.i |= (c[3] & 0xFF) << 0x18;
175
176   return u.f;
177 }
178
179 static SMDS_MeshNode* addNode(const gp_Pnt& P,
180                               DriverSTL_DataMapOfPntNodePtr& uniqnodes,
181                               SMDS_Mesh* theMesh)
182 {
183   SMDS_MeshNode* node = 0;
184   if ( uniqnodes.IsBound(P) ) {
185     node = uniqnodes.Find(P);
186   } else {
187     node = theMesh->AddNode(P.X(), P.Y(), P.Z() );
188     uniqnodes.Bind(P,node);
189   }
190   
191   return node;
192 }                                
193
194 static SMDS_MeshNode* readNode(FILE* file,
195                                DriverSTL_DataMapOfPntNodePtr& uniqnodes,
196                                SMDS_Mesh* theMesh)
197 {
198   Standard_ShortReal coord[3];
199   // reading vertex
200   fscanf(file,"%*s %f %f %f\n",&coord[0],&coord[1],&coord[2]);
201
202   gp_Pnt P(coord[0],coord[1],coord[2]);
203   return addNode( P, uniqnodes, theMesh );
204 }
205
206 static SMDS_MeshNode* readNode(OSD_File& theFile,
207                                DriverSTL_DataMapOfPntNodePtr& uniqnodes,
208                                SMDS_Mesh* theMesh)
209 {
210   Standard_ShortReal coord[3];
211   coord[0] = readFloat(theFile);
212   coord[1] = readFloat(theFile);
213   coord[2] = readFloat(theFile);
214
215   gp_Pnt P(coord[0],coord[1],coord[2]);
216   return addNode( P, uniqnodes, theMesh );
217 }
218
219 //=======================================================================
220 //function : readAscii
221 //purpose  : 
222 //=======================================================================
223
224 Driver_Mesh::Status DriverSTL_R_SMDS_Mesh::readAscii() const
225 {
226   Status aResult = DRS_OK;
227   long ipos;
228   Standard_Integer nbLines = 0;
229
230   // Open the file 
231   TCollection_AsciiString aFileName( (char *)myFile.c_str() );
232   FILE* file = fopen(aFileName.ToCString(),"r");
233   fseek(file,0L,SEEK_END);
234   // get the file size
235   long filesize = ftell(file);
236   fclose(file);
237   file = fopen(aFileName.ToCString(),"r");
238   
239   // count the number of lines
240   for (ipos = 0; ipos < filesize; ++ipos) {
241     if (getc(file) == '\n')
242       nbLines++;
243   }
244
245   // go back to the beginning of the file
246 //  fclose(file);
247 //  file = fopen(aFileName.ToCString(),"r");
248   rewind(file);
249   
250   Standard_Integer nbTri = (nbLines / ASCII_LINES_PER_FACET);
251
252   DriverSTL_DataMapOfPntNodePtr uniqnodes;
253   // skip header
254   while (getc(file) != '\n');
255
256   // main reading
257   for (Standard_Integer iTri = 0; iTri < nbTri; ++iTri) {
258
259     // skipping the facet normal
260     Standard_ShortReal normal[3];
261     fscanf(file,"%*s %*s %f %f %f\n",&normal[0],&normal[1],&normal[2]);
262
263     // skip the keywords "outer loop"
264     fscanf(file,"%*s %*s");
265
266     // reading nodes
267     SMDS_MeshNode* node1 = readNode( file, uniqnodes, myMesh );
268     SMDS_MeshNode* node2 = readNode( file, uniqnodes, myMesh );
269     SMDS_MeshNode* node3 = readNode( file, uniqnodes, myMesh );
270
271     if (myIsCreateFaces)
272       myMesh->AddFace(node1,node2,node3);
273
274     // skip the keywords "endloop"
275     fscanf(file,"%*s");
276
277     // skip the keywords "endfacet"
278     fscanf(file,"%*s");
279   }
280
281   fclose(file);
282   return aResult;
283 }
284
285 //=======================================================================
286 //function : readBinary
287 //purpose  : 
288 //=======================================================================
289
290 Driver_Mesh::Status DriverSTL_R_SMDS_Mesh::readBinary() const
291 {
292   Status aResult = DRS_OK;
293
294   char buftest[5];
295   Standard_Address adr;
296   adr = (Standard_Address)buftest;
297
298   TCollection_AsciiString aFileName( (char *)myFile.c_str() );
299   OSD_File aFile = OSD_File(OSD_Path( aFileName ));
300   aFile.Open(OSD_ReadOnly,OSD_Protection(OSD_RWD,OSD_RWD,OSD_RWD,OSD_RWD));
301
302   // the size of the file (minus the header size)
303   // must be a multiple of SIZEOF_STL_FACET
304
305   // compute file size
306   Standard_Integer filesize = aFile.Size();
307
308   if ( (filesize - HEADER_SIZE) % SIZEOF_STL_FACET !=0 
309       // Commented to allow reading small files (ex: 1 face)
310       /*|| (filesize < STL_MIN_FILE_SIZE)*/) {
311     Standard_NoMoreObject::Raise("DriverSTL_R_SMDS_MESH::readBinary (wrong file size)");
312   }
313
314   // don't trust the number of triangles which is coded in the file
315   // sometimes it is wrong, and with this technique we don't need to swap endians for integer
316   Standard_Integer nbTri = ((filesize - HEADER_SIZE) / SIZEOF_STL_FACET);
317
318   // skip the header
319   aFile.Seek(HEADER_SIZE,OSD_FromBeginning);
320
321   DriverSTL_DataMapOfPntNodePtr uniqnodes;
322   Standard_Integer lread;
323   
324   for (Standard_Integer iTri = 0; iTri < nbTri; ++iTri) {
325
326     // ignore normals
327     readFloat(aFile);
328     readFloat(aFile);
329     readFloat(aFile);
330
331     // read vertices
332     SMDS_MeshNode* node1 = readNode( aFile, uniqnodes, myMesh );
333     SMDS_MeshNode* node2 = readNode( aFile, uniqnodes, myMesh );
334     SMDS_MeshNode* node3 = readNode( aFile, uniqnodes, myMesh );
335
336     if (myIsCreateFaces)
337       myMesh->AddFace(node1,node2,node3);
338
339     // skip extra bytes
340     aFile.Read(adr,2,lread);
341   }
342   aFile.Close();
343   return aResult;
344 }