dust3d/third_party/libigl/include/igl/readSTL.cpp

291 lines
9.6 KiB
C++
Raw Normal View History

// This file is part of libigl, a simple c++ geometry processing library.
//
// Copyright (C) 2014 Alec Jacobson <alecjacobson@gmail.com>
//
// This Source Code Form is subject to the terms of the Mozilla Public License
// v. 2.0. If a copy of the MPL was not distributed with this file, You can
// obtain one at http://mozilla.org/MPL/2.0/.
#include "readSTL.h"
#include "list_to_matrix.h"
#include <iostream>
template <typename DerivedV, typename DerivedF, typename DerivedN>
IGL_INLINE bool igl::readSTL(
const std::string & filename,
Eigen::PlainObjectBase<DerivedV> & V,
Eigen::PlainObjectBase<DerivedF> & F,
Eigen::PlainObjectBase<DerivedN> & N)
{
using namespace std;
vector<vector<typename DerivedV::Scalar> > vV;
vector<vector<typename DerivedN::Scalar> > vN;
vector<vector<typename DerivedF::Scalar> > vF;
if(!readSTL(filename,vV,vF,vN))
{
return false;
}
if(!list_to_matrix(vV,V))
{
return false;
}
if(!list_to_matrix(vF,F))
{
return false;
}
if(!list_to_matrix(vN,N))
{
return false;
}
return true;
}
template <typename TypeV, typename TypeF, typename TypeN>
IGL_INLINE bool igl::readSTL(
const std::string & filename,
std::vector<std::vector<TypeV> > & V,
std::vector<std::vector<TypeF> > & F,
std::vector<std::vector<TypeN> > & N)
{
using namespace std;
// Should test for ascii
// Open file, and check for error
FILE * stl_file = fopen(filename.c_str(),"rb");
if(NULL==stl_file)
{
fprintf(stderr,"IOError: %s could not be opened...\n",
filename.c_str());
return false;
}
return readSTL(stl_file,V,F,N);
}
template <typename TypeV, typename TypeF, typename TypeN>
IGL_INLINE bool igl::readSTL(
FILE * stl_file,
std::vector<std::vector<TypeV> > & V,
std::vector<std::vector<TypeF> > & F,
std::vector<std::vector<TypeN> > & N)
{
using namespace std;
//stl_file = freopen(NULL,"rb",stl_file);
if(NULL==stl_file)
{
fprintf(stderr,"IOError: stl file could not be reopened as binary (1) ...\n");
return false;
}
V.clear();
F.clear();
N.clear();
// Specifically 80 character header
char header[80];
char solid[80];
bool is_ascii = true;
if(fread(header,1,80,stl_file) != 80)
{
cerr<<"IOError: too short (1)."<<endl;
goto close_false;
}
sscanf(header,"%s",solid);
if(string("solid") != solid)
{
// definitely **not** ascii
is_ascii = false;
}else
{
// might still be binary
char buf[4];
if(fread(buf,1,4,stl_file) != 4)
{
cerr<<"IOError: too short (3)."<<endl;
goto close_false;
}
size_t num_faces = *reinterpret_cast<unsigned int*>(buf);
fseek(stl_file,0,SEEK_END);
int file_size = ftell(stl_file);
if(file_size == 80 + 4 + (4*12 + 2) * num_faces)
{
is_ascii = false;
}else
{
is_ascii = true;
}
}
if(is_ascii)
{
// Rewind to end of header
//stl_file = fopen(filename.c_str(),"r");
//stl_file = freopen(NULL,"r",stl_file);
fseek(stl_file, 0, SEEK_SET);
if(NULL==stl_file)
{
fprintf(stderr,"IOError: stl file could not be reopened as ascii ...\n");
return false;
}
// Read 80 header
// Eat file name
#ifndef IGL_LINE_MAX
# define IGL_LINE_MAX 2048
#endif
char name[IGL_LINE_MAX];
if(NULL==fgets(name,IGL_LINE_MAX,stl_file))
{
cerr<<"IOError: ascii too short (2)."<<endl;
goto close_false;
}
// ascii
while(true)
{
int ret;
char facet[IGL_LINE_MAX],normal[IGL_LINE_MAX];
vector<TypeN > n(3);
double nd[3];
ret = fscanf(stl_file,"%s %s %lg %lg %lg",facet,normal,nd,nd+1,nd+2);
if(string("endsolid") == facet)
{
break;
}
if(ret != 5 ||
!(string("facet") == facet ||
string("faced") == facet) ||
string("normal") != normal)
{
cout<<"facet: "<<facet<<endl;
cout<<"normal: "<<normal<<endl;
cerr<<"IOError: bad format (1)."<<endl;
goto close_false;
}
// copy casts to Type
n[0] = nd[0]; n[1] = nd[1]; n[2] = nd[2];
N.push_back(n);
char outer[IGL_LINE_MAX], loop[IGL_LINE_MAX];
ret = fscanf(stl_file,"%s %s",outer,loop);
if(ret != 2 || string("outer") != outer || string("loop") != loop)
{
cerr<<"IOError: bad format (2)."<<endl;
goto close_false;
}
vector<TypeF> f;
while(true)
{
char word[IGL_LINE_MAX];
int ret = fscanf(stl_file,"%s",word);
if(ret == 1 && string("endloop") == word)
{
break;
}else if(ret == 1 && string("vertex") == word)
{
vector<TypeV> v(3);
double vd[3];
int ret = fscanf(stl_file,"%lg %lg %lg",vd,vd+1,vd+2);
if(ret != 3)
{
cerr<<"IOError: bad format (3)."<<endl;
goto close_false;
}
f.push_back(V.size());
// copy casts to Type
v[0] = vd[0]; v[1] = vd[1]; v[2] = vd[2];
V.push_back(v);
}else
{
cerr<<"IOError: bad format (4)."<<endl;
goto close_false;
}
}
F.push_back(f);
char endfacet[IGL_LINE_MAX];
ret = fscanf(stl_file,"%s",endfacet);
if(ret != 1 || string("endfacet") != endfacet)
{
cerr<<"IOError: bad format (5)."<<endl;
goto close_false;
}
}
// read endfacet
goto close_true;
}else
{
// Binary
//stl_file = freopen(NULL,"rb",stl_file);
fseek(stl_file, 0, SEEK_SET);
// Read 80 header
char header[80];
if(fread(header,sizeof(char),80,stl_file)!=80)
{
cerr<<"IOError: bad format (6)."<<endl;
goto close_false;
}
// Read number of triangles
unsigned int num_tri;
if(fread(&num_tri,sizeof(unsigned int),1,stl_file)!=1)
{
cerr<<"IOError: bad format (7)."<<endl;
goto close_false;
}
V.resize(num_tri*3,vector<TypeV >(3,0));
N.resize(num_tri,vector<TypeN >(3,0));
F.resize(num_tri,vector<TypeF >(3,0));
for(int t = 0;t<(int)num_tri;t++)
{
// Read normal
float n[3];
if(fread(n,sizeof(float),3,stl_file)!=3)
{
cerr<<"IOError: bad format (8)."<<endl;
goto close_false;
}
// Read each vertex
for(int c = 0;c<3;c++)
{
F[t][c] = 3*t+c;
N[t][c] = n[c];
float v[3];
if(fread(v,sizeof(float),3,stl_file)!=3)
{
cerr<<"IOError: bad format (9)."<<endl;
goto close_false;
}
V[3*t+c][0] = v[0];
V[3*t+c][1] = v[1];
V[3*t+c][2] = v[2];
}
// Read attribute size
unsigned short att_count;
if(fread(&att_count,sizeof(unsigned short),1,stl_file)!=1)
{
cerr<<"IOError: bad format (10)."<<endl;
goto close_false;
}
}
goto close_true;
}
close_false:
fclose(stl_file);
return false;
close_true:
fclose(stl_file);
return true;
}
#ifdef IGL_STATIC_LIBRARY
// Explicit template instantiation
// generated by autoexplicit.sh
template bool igl::readSTL<Eigen::Matrix<double, -1, -1, 1, -1, -1>, Eigen::Matrix<int, -1, -1, 0, -1, -1>, Eigen::Matrix<double, -1, -1, 0, -1, -1> >(std::basic_string<char, std::char_traits<char>, std::allocator<char> > const&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 1, -1, -1> >&, Eigen::PlainObjectBase<Eigen::Matrix<int, -1, -1, 0, -1, -1> >&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> >&);
// generated by autoexplicit.sh
template bool igl::readSTL<Eigen::Matrix<float, -1, 3, 1, -1, 3>, Eigen::Matrix<unsigned int, -1, 3, 1, -1, 3>, Eigen::Matrix<double, -1, -1, 0, -1, -1> >(std::basic_string<char, std::char_traits<char>, std::allocator<char> > const&, Eigen::PlainObjectBase<Eigen::Matrix<float, -1, 3, 1, -1, 3> >&, Eigen::PlainObjectBase<Eigen::Matrix<unsigned int, -1, 3, 1, -1, 3> >&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> >&);
// generated by autoexplicit.sh
template bool igl::readSTL<Eigen::Matrix<double, -1, 3, 0, -1, 3>, Eigen::Matrix<int, -1, 3, 0, -1, 3>, Eigen::Matrix<double, -1, -1, 0, -1, -1> >(std::basic_string<char, std::char_traits<char>, std::allocator<char> > const&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, 3, 0, -1, 3> >&, Eigen::PlainObjectBase<Eigen::Matrix<int, -1, 3, 0, -1, 3> >&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> >&);
// generated by autoexplicit.sh
template bool igl::readSTL<Eigen::Matrix<double, -1, -1, 0, -1, -1>, Eigen::Matrix<int, -1, -1, 0, -1, -1>, Eigen::Matrix<double, -1, -1, 0, -1, -1> >(std::basic_string<char, std::char_traits<char>, std::allocator<char> > const&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> >&, Eigen::PlainObjectBase<Eigen::Matrix<int, -1, -1, 0, -1, -1> >&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> >&);
template bool igl::readSTL<Eigen::Matrix<float, -1, -1, 0, -1, -1>, Eigen::Matrix<int, -1, -1, 0, -1, -1>, Eigen::Matrix<float, -1, -1, 0, -1, -1> >(std::basic_string<char, std::char_traits<char>, std::allocator<char> > const&, Eigen::PlainObjectBase<Eigen::Matrix<float, -1, -1, 0, -1, -1> >&, Eigen::PlainObjectBase<Eigen::Matrix<int, -1, -1, 0, -1, -1> >&, Eigen::PlainObjectBase<Eigen::Matrix<float, -1, -1, 0, -1, -1> >&);
template bool igl::readSTL<Eigen::Matrix<double, -1, 3, 1, -1, 3>, Eigen::Matrix<int, -1, 3, 1, -1, 3>, Eigen::Matrix<double, -1, -1, 0, -1, -1> >(std::basic_string<char, std::char_traits<char>, std::allocator<char> > const&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, 3, 1, -1, 3> >&, Eigen::PlainObjectBase<Eigen::Matrix<int, -1, 3, 1, -1, 3> >&, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> >&);
#endif