cgal/Polyhedron_IO/include/CGAL/IO/STL_reader.h

111 lines
3.0 KiB
C++

// Copyright (c) 2015 GeometryFactory
//
// This file is part of CGAL (www.cgal.org); you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; either version 3 of the License,
// or (at your option) any later version.
//
// Licensees holding a valid commercial license may use this file in
// accordance with the commercial license agreement provided with the software.
//
// This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
// WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
//
// $URL$
// $Id$
//
// Author(s) : Andreas Fabri
#ifndef CGAL_IO_STL_READER_H
#define CGAL_IO_STL_READER_H
#include <CGAL/array.h>
#include <vector>
#include <map>
#include <iostream>
namespace CGAL{
bool
read_STL( std::istream& input,
std::vector< cpp11::array<double,3> >& points,
std::vector< cpp11::array<int,3> >& facets,
bool verbose = false)
{
std::string s,
solid("solid"),
facet("facet"),
outer("outer"),
loop("loop"),
vertex("vertex"),
endloop("endloop"),
endsolid("endsolid");
std::map<cpp11::array<double,3>, int> vertex_index;
int index = 0;
cpp11::array<int,3> ijk;
cpp11::array<double,3> p;
input >> s;
if(s == solid){
std::getline(input, s);
} else {
if (verbose)
std::cerr << "We expect keyword 'solid'" << std::endl;
return false;
}
while(input >> s){
if(s == endsolid){
//std::cerr << "found endsolid" << std::endl;
} else if(s == facet){
//std::cerr << "found facet" << std::endl;
std::getline(input, s); // ignore the normal
input >> s;
if(s != outer){
if (verbose)
std::cerr << "Expect 'outer' and got " << s << std::endl;
return false;
}
input >> s;
if(s != loop){
if (verbose)
std::cerr << "Expect 'loop' and got " << s << std::endl;
return false;
}
int count = 0;
do {
input >> s;
if(s == vertex){
// std::cerr << "found vertex" << std::endl;
if(count < 3){
input >> p[0] >> p[1] >> p[2];
std::map<cpp11::array<double,3>, int>::iterator iti=
vertex_index.insert(std::make_pair(p,-1)).first;
if(iti->second==-1){
ijk[count] = index;
iti->second = index++;
points.push_back(p);
} else {
ijk[count] = iti->second;
}
++count;
} else {
if (verbose)
std::cerr << "We can only read triangulated surfaces" << std::endl;
return false;
}
}
}while(s != endloop);
facets.push_back(ijk);
}
}
return true;
}
} // namespace CGAL
#endif // CGAL_IO_STL_READER_H