diff options
Diffstat (limited to 'modelFile/modelFile.cpp')
| -rw-r--r-- | modelFile/modelFile.cpp | 177 |
1 files changed, 177 insertions, 0 deletions
diff --git a/modelFile/modelFile.cpp b/modelFile/modelFile.cpp new file mode 100644 index 0000000..ad1a269 --- /dev/null +++ b/modelFile/modelFile.cpp @@ -0,0 +1,177 @@ +/** Copyright (C) 2013 David Braam - Released under terms of the AGPLv3 License */
+#include <string.h>
+#include <stdio.h>
+
+#include "modelFile/modelFile.h"
+
+FILE* binaryMeshBlob = NULL;
+
+/* Custom fgets function to support Mac line-ends in Ascii STL files. OpenSCAD produces this when used on Mac */
+void* fgets_(char* ptr, size_t len, FILE* f)
+{
+ while(len && fread(ptr, 1, 1, f) > 0)
+ {
+ if (*ptr == '\n' || *ptr == '\r')
+ {
+ *ptr = '\0';
+ return ptr;
+ }
+ ptr++;
+ len--;
+ }
+ return NULL;
+}
+
+SimpleModel* loadModelSTL_ascii(const char* filename, FMatrix3x3& matrix)
+{
+ SimpleModel* m = new SimpleModel();
+ m->volumes.push_back(SimpleVolume());
+ SimpleVolume* vol = &m->volumes[0];
+ FILE* f = fopen(filename, "rt");
+ char buffer[1024];
+ FPoint3 vertex;
+ int n = 0;
+ Point3 v0(0,0,0), v1(0,0,0), v2(0,0,0);
+ while(fgets_(buffer, sizeof(buffer), f))
+ {
+ if (sscanf(buffer, " vertex %lf %lf %lf", &vertex.x, &vertex.y, &vertex.z) == 3)
+ {
+ n++;
+ switch(n)
+ {
+ case 1:
+ v0 = matrix.apply(vertex);
+ break;
+ case 2:
+ v1 = matrix.apply(vertex);
+ break;
+ case 3:
+ v2 = matrix.apply(vertex);
+ vol->addFace(v0, v1, v2);
+ n = 0;
+ break;
+ }
+ }
+ }
+ fclose(f);
+ return m;
+}
+
+SimpleModel* loadModelSTL_binary(const char* filename, FMatrix3x3& matrix)
+{
+ FILE* f = fopen(filename, "rb");
+ char buffer[80];
+ uint32_t faceCount;
+ //Skip the header
+ if (fread(buffer, 80, 1, f) != 1)
+ {
+ fclose(f);
+ return NULL;
+ }
+ //Read the face count
+ if (fread(&faceCount, sizeof(uint32_t), 1, f) != 1)
+ {
+ fclose(f);
+ return NULL;
+ }
+ //For each face read:
+ //float(x,y,z) = normal, float(X,Y,Z)*3 = vertexes, uint16_t = flags
+ SimpleModel* m = new SimpleModel();
+ m->volumes.push_back(SimpleVolume());
+ SimpleVolume* vol = &m->volumes[0];
+ if(vol == NULL)
+ {
+ fclose(f);
+ return NULL;
+ }
+
+ for(unsigned int i=0;i<faceCount;i++)
+ {
+ if (fread(buffer, sizeof(float) * 3, 1, f) != 1)
+ {
+ fclose(f);
+ return NULL;
+ }
+ float v[9];
+ if (fread(v, sizeof(float) * 9, 1, f) != 1)
+ {
+ fclose(f);
+ return NULL;
+ }
+ Point3 v0 = matrix.apply(FPoint3(v[0], v[1], v[2]));
+ Point3 v1 = matrix.apply(FPoint3(v[3], v[4], v[5]));
+ Point3 v2 = matrix.apply(FPoint3(v[6], v[7], v[8]));
+ vol->addFace(v0, v1, v2);
+ if (fread(buffer, sizeof(uint16_t), 1, f) != 1)
+ {
+ fclose(f);
+ return NULL;
+ }
+ }
+ fclose(f);
+ return m;
+}
+
+SimpleModel* loadModelSTL(const char* filename, FMatrix3x3& matrix)
+{
+ FILE* f = fopen(filename, "r");
+ char buffer[6];
+ if (f == NULL)
+ return NULL;
+
+ if (fread(buffer, 5, 1, f) != 1)
+ {
+ fclose(f);
+ return NULL;
+ }
+ fclose(f);
+
+ buffer[5] = '\0';
+ if (strcasecmp(buffer, "SOLID") == 0)
+ {
+ return loadModelSTL_ascii(filename, matrix);
+ }
+ return loadModelSTL_binary(filename, matrix);
+}
+
+SimpleModel* loadModel(const char* filename, FMatrix3x3& matrix)
+{
+ const char* ext = strrchr(filename, '.');
+ if (ext && strcasecmp(ext, ".stl") == 0)
+ {
+ return loadModelSTL(filename, matrix);
+ }
+ if (filename[0] == '#' && binaryMeshBlob != NULL)
+ {
+ SimpleModel* m = new SimpleModel();
+
+ while(*filename == '#')
+ {
+ filename++;
+
+ m->volumes.push_back(SimpleVolume());
+ SimpleVolume* vol = &m->volumes[m->volumes.size()-1];
+ int32_t n, pNr = 0;
+ if (fread(&n, 1, sizeof(int32_t), binaryMeshBlob) < 1)
+ return NULL;
+ printf("Reading mesh from binary blob with %i vertexes\n", n);
+ Point3 v[3];
+ while(n)
+ {
+ float f[3];
+ if (fread(f, 3, sizeof(float), binaryMeshBlob) < 1)
+ return NULL;
+ FPoint3 fp(f[0], f[1], f[2]);
+ v[pNr++] = matrix.apply(fp);
+ if (pNr == 3)
+ {
+ vol->addFace(v[0], v[1], v[2]);
+ pNr = 0;
+ }
+ n--;
+ }
+ }
+ return m;
+ }
+ return NULL;
+}
|