521 lines
12 KiB
C++
521 lines
12 KiB
C++
/*
|
|
* This program source code file is part of KiCad, a free EDA CAD application.
|
|
*
|
|
* Copyright (C) 2015 Cirilo Bernardo <cirilo.bernardo@gmail.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, you may find one here:
|
|
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
|
|
* or you may search the http://www.gnu.org website for the version 2 license,
|
|
* or you may write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <streambuf>
|
|
#include <iomanip>
|
|
#include <string>
|
|
#include <utility>
|
|
#include <map>
|
|
|
|
#include "3d_cache/sg/sg_helpers.h"
|
|
#include "3d_cache/sg/sg_node.h"
|
|
|
|
|
|
// formats a floating point number for text output to a VRML file
|
|
void S3D::FormatFloat( std::string& result, double value )
|
|
{
|
|
if( value < 1e-8 && value > -1e-8 )
|
|
{
|
|
result = "0";
|
|
return;
|
|
}
|
|
|
|
// note: many VRML implementations use float so we use the max.
|
|
// precision here of 8 digits.
|
|
std::ostringstream out;
|
|
out << std::setprecision( 8 ) << value;
|
|
|
|
result = out.str();
|
|
|
|
size_t p = result.find( "." );
|
|
|
|
// trim trailing 0 if appropriate
|
|
|
|
if( std::string::npos == p )
|
|
return;
|
|
|
|
p = result.find_first_of( "eE" );
|
|
|
|
if( std::string::npos == p )
|
|
{
|
|
while( '0' == *(result.rbegin()) )
|
|
result.erase( result.size() - 1 );
|
|
|
|
return;
|
|
}
|
|
|
|
if( '0' != result.at( p -1 ) )
|
|
return;
|
|
|
|
// trim all 0 to the left of 'p'
|
|
std::string tmp = result.substr( p );
|
|
result = result.substr( 0, p );
|
|
|
|
while( '0' == *(result.rbegin()) )
|
|
result.erase( result.size() - 1 );
|
|
|
|
result.append( tmp );
|
|
|
|
return;
|
|
}
|
|
|
|
// format orientation data for VRML output
|
|
void S3D::FormatOrientation( std::string& result, const SGVECTOR& axis, double rotation )
|
|
{
|
|
double aX;
|
|
double aY;
|
|
double aZ;
|
|
|
|
axis.GetVector( aX, aY, aZ );
|
|
FormatFloat( result, aX );
|
|
std::string tmp;
|
|
FormatFloat( tmp, aY );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
FormatFloat( tmp, aZ );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
FormatFloat( tmp, rotation );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
|
|
return;
|
|
}
|
|
|
|
// format point data for VRML output
|
|
void S3D::FormatPoint( std::string& result, const SGPOINT& point )
|
|
{
|
|
FormatFloat( result, point.x );
|
|
|
|
std::string tmp;
|
|
FormatFloat( tmp, point.y );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
|
|
FormatFloat( tmp, point.z );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
// format vector data for VRML output
|
|
void S3D::FormatVector( std::string& result, const SGVECTOR& aVector )
|
|
{
|
|
double X, Y, Z;
|
|
aVector.GetVector( X, Y, Z );
|
|
FormatFloat( result, X );
|
|
|
|
std::string tmp;
|
|
FormatFloat( tmp, Y );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
|
|
FormatFloat( tmp, Z );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
// format Color data for VRML output
|
|
void S3D::FormatColor( std::string& result, const SGCOLOR& aColor )
|
|
{
|
|
float R, G, B;
|
|
aColor.GetColor( R, G, B );
|
|
FormatFloat( result, R );
|
|
|
|
std::string tmp;
|
|
FormatFloat( tmp, G );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
|
|
FormatFloat( tmp, B );
|
|
result.append( " " );
|
|
result.append( tmp );
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
bool S3D::WritePoint( std::ofstream& aFile, const SGPOINT& aPoint )
|
|
{
|
|
aFile.write( (char*)&aPoint.x, sizeof(aPoint.x) );
|
|
aFile.write( (char*)&aPoint.y, sizeof(aPoint.y) );
|
|
aFile.write( (char*)&aPoint.z, sizeof(aPoint.z) );
|
|
|
|
if( aFile.fail() )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool S3D::WriteVector( std::ofstream& aFile, const SGVECTOR& aVector )
|
|
{
|
|
double x, y, z;
|
|
aVector.GetVector( x, y, z );
|
|
aFile.write( (char*)&x, sizeof(double) );
|
|
aFile.write( (char*)&y, sizeof(double) );
|
|
aFile.write( (char*)&z, sizeof(double) );
|
|
|
|
if( aFile.fail() )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool S3D::WriteColor( std::ofstream& aFile, const SGCOLOR& aColor )
|
|
{
|
|
float r, g, b;
|
|
aColor.GetColor( r, g, b );
|
|
aFile.write( (char*)&r, sizeof(float) );
|
|
aFile.write( (char*)&g, sizeof(float) );
|
|
aFile.write( (char*)&b, sizeof(float) );
|
|
|
|
if( aFile.fail() )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
S3D::SGTYPES S3D::ReadTag( std::ifstream& aFile, std::string& aName )
|
|
{
|
|
char schar;
|
|
aFile.get( schar );
|
|
|
|
if( '[' != schar )
|
|
{
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [INFO] corrupt data; missing left bracket at position ";
|
|
std::cerr << aFile.tellg() << "\n";
|
|
#endif
|
|
|
|
return S3D::SGTYPE_END;
|
|
}
|
|
|
|
std::string name;
|
|
aFile.get( schar );
|
|
|
|
while( ']' != schar && aFile.good() )
|
|
{
|
|
name.push_back( schar );
|
|
aFile.get( schar );
|
|
}
|
|
|
|
if( schar != ']' )
|
|
{
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [INFO] corrupt data; could not find right bracket\n";
|
|
#endif
|
|
|
|
return S3D::SGTYPE_END;
|
|
}
|
|
|
|
aName = name;
|
|
size_t upos = name.find( '_' );
|
|
|
|
if( std::string::npos == upos )
|
|
{
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [INFO] corrupt data; no underscore in name '";
|
|
std::cerr << name << "'\n";
|
|
#endif
|
|
|
|
return S3D::SGTYPE_END;
|
|
}
|
|
|
|
name = name.substr( 0, upos );
|
|
S3D::SGTYPES types[S3D::SGTYPE_END] = {
|
|
SGTYPE_TRANSFORM,
|
|
SGTYPE_APPEARANCE,
|
|
SGTYPE_COLORS,
|
|
SGTYPE_COLORINDEX,
|
|
SGTYPE_FACESET,
|
|
SGTYPE_COORDS,
|
|
SGTYPE_COORDINDEX,
|
|
SGTYPE_NORMALS,
|
|
SGTYPE_SHAPE
|
|
};
|
|
|
|
for( int i = 0; i < S3D::SGTYPE_END; ++i )
|
|
{
|
|
if( !name.compare( S3D::GetNodeTypeName( types[i] ) ) )
|
|
return types[i];
|
|
}
|
|
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [INFO] corrupt data; no node type matching '";
|
|
std::cerr << name << "'\n";
|
|
#endif
|
|
|
|
return S3D::SGTYPE_END;
|
|
}
|
|
|
|
|
|
bool S3D::ReadPoint( std::ifstream& aFile, SGPOINT& aPoint )
|
|
{
|
|
aFile.read( (char*)&aPoint.x, sizeof( aPoint.x ) );
|
|
aFile.read( (char*)&aPoint.y, sizeof( aPoint.y ) );
|
|
aFile.read( (char*)&aPoint.z, sizeof( aPoint.z ) );
|
|
|
|
if( aFile.fail() )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool S3D::ReadVector( std::ifstream& aFile, SGVECTOR& aVector )
|
|
{
|
|
double x, y, z;
|
|
aFile.read( (char*)&x, sizeof(double) );
|
|
aFile.read( (char*)&y, sizeof(double) );
|
|
aFile.read( (char*)&z, sizeof(double) );
|
|
aVector.SetVector( x, y, z );
|
|
|
|
if( aFile.fail() )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool S3D::ReadColor( std::ifstream& aFile, SGCOLOR& aColor )
|
|
{
|
|
float r, g, b;
|
|
aFile.read( (char*)&r, sizeof(float) );
|
|
aFile.read( (char*)&g, sizeof(float) );
|
|
aFile.read( (char*)&b, sizeof(float) );
|
|
aColor.SetColor( r, g, b );
|
|
|
|
if( aFile.fail() )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool S3D::degenerate( glm::dvec3* pts )
|
|
{
|
|
double dx, dy, dz;
|
|
|
|
dx = pts[1].x - pts[0].x;
|
|
dy = pts[1].y - pts[0].y;
|
|
dz = pts[1].z - pts[0].z;
|
|
|
|
if( ( dx*dx + dy*dy + dz*dz ) < 1e-15 )
|
|
return true;
|
|
|
|
dx = pts[2].x - pts[0].x;
|
|
dy = pts[2].y - pts[0].y;
|
|
dz = pts[2].z - pts[0].z;
|
|
|
|
if( ( dx*dx + dy*dy + dz*dz ) < 1e-15 )
|
|
return true;
|
|
|
|
dx = pts[2].x - pts[1].x;
|
|
dy = pts[2].y - pts[1].y;
|
|
dz = pts[2].z - pts[1].z;
|
|
|
|
if( ( dx*dx + dy*dy + dz*dz ) < 1e-15 )
|
|
return true;
|
|
|
|
return false;
|
|
};
|
|
|
|
|
|
static void calcTriad( glm::dvec3* pts, glm::dvec3& tri )
|
|
{
|
|
if( S3D::degenerate( pts ) )
|
|
{
|
|
// degenerate points should contribute nothing to the result
|
|
tri = glm::dvec3( 0.0, 0.0, 0.0 );
|
|
return;
|
|
}
|
|
|
|
// normal * 2 * area
|
|
tri = cross( pts[1] - pts[0], pts[2] - pts[0] );
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
bool S3D::CalcTriangleNormals( std::vector< SGPOINT > coords,
|
|
std::vector< int >& index, std::vector< SGVECTOR >& norms )
|
|
{
|
|
size_t vsize = coords.size();
|
|
|
|
if( vsize < 3 )
|
|
{
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [INFO] invalid vertex set (fewer than 3 vertices)\n";
|
|
#endif
|
|
|
|
return false;
|
|
}
|
|
|
|
size_t isize = index.size();
|
|
|
|
if( 0 != isize % 3 || index.empty() )
|
|
{
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [INFO] invalid index set (not multiple of 3)\n";
|
|
#endif
|
|
|
|
return false;
|
|
}
|
|
|
|
if( !norms.empty() )
|
|
{
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [INFO] normals set is not empty\n";
|
|
#endif
|
|
|
|
return false;
|
|
}
|
|
|
|
std::map< int, std::list< glm::dvec3 > >vmap;
|
|
|
|
int p1, p2, p3;
|
|
|
|
// create the map of indices to facet sets
|
|
for( size_t i = 0; i < isize; )
|
|
{
|
|
p1 = index[i++];
|
|
p2 = index[i++];
|
|
p3 = index[i++];
|
|
|
|
if( p1 < 0 || p1 >= vsize || p2 < 0 || p2 >= vsize ||
|
|
p3 < 0 || p3 >= vsize )
|
|
{
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [INFO] invalid index set; index out of bounds\n";
|
|
#endif
|
|
|
|
return false;
|
|
}
|
|
|
|
glm::dvec3 tri;
|
|
glm::dvec3 trip[3];
|
|
trip[0] = glm::dvec3( coords[p1].x, coords[p1].y, coords[p1].z );
|
|
trip[1] = glm::dvec3( coords[p2].x, coords[p2].y, coords[p2].z );
|
|
trip[2] = glm::dvec3( coords[p3].x, coords[p3].y, coords[p3].z );
|
|
calcTriad( trip, tri );
|
|
|
|
std::map< int, std::list< glm::dvec3 > >::iterator ip = vmap.find( p1 );
|
|
|
|
if( ip != vmap.end() )
|
|
{
|
|
ip->second.push_back( tri );
|
|
}
|
|
else
|
|
{
|
|
vmap.insert( std::pair < int, std::list < glm::dvec3 > >
|
|
( p1, std::list < glm::dvec3 >( 1, tri ) ) );
|
|
}
|
|
|
|
ip = vmap.find( p2 );
|
|
|
|
if( ip != vmap.end() )
|
|
{
|
|
ip->second.push_back( tri );
|
|
}
|
|
else
|
|
{
|
|
vmap.insert( std::pair < int, std::list < glm::dvec3 > >
|
|
( p2, std::list < glm::dvec3 >( 1, tri ) ) );
|
|
}
|
|
|
|
ip = vmap.find( p3 );
|
|
|
|
if( ip != vmap.end() )
|
|
{
|
|
ip->second.push_back( tri );
|
|
}
|
|
else
|
|
{
|
|
vmap.insert( std::pair < int, std::list < glm::dvec3 > >
|
|
( p3, std::list < glm::dvec3 >( 1, tri ) ) );
|
|
}
|
|
}
|
|
|
|
std::map< int, std::list< glm::dvec3 > >::iterator sM = vmap.begin();
|
|
std::map< int, std::list< glm::dvec3 > >::iterator eM = vmap.end();
|
|
size_t idx = 0;
|
|
|
|
while( sM != eM )
|
|
{
|
|
size_t item = sM->first;
|
|
|
|
// assign any skipped coordinates a normal of (0,0,1)
|
|
while( item > idx )
|
|
{
|
|
norms.push_back( SGVECTOR( 0, 0, 1 ) );
|
|
++idx;
|
|
}
|
|
|
|
std::list< glm::dvec3 >::iterator sT = sM->second.begin();
|
|
std::list< glm::dvec3 >::iterator eT = sM->second.end();
|
|
glm::dvec3 norm( 0.0, 0.0, 0.0 );
|
|
|
|
while( sT != eT )
|
|
{
|
|
norm += *sT;
|
|
++sT;
|
|
}
|
|
|
|
norms.push_back( SGVECTOR( norm.x, norm.y, norm.z ) );
|
|
|
|
++idx;
|
|
++sM;
|
|
}
|
|
|
|
if( norms.size() != coords.size() )
|
|
{
|
|
#ifdef DEBUG
|
|
std::cerr << __FILE__ << ": " << __FUNCTION__ << ": " << __LINE__ << "\n";
|
|
std::cerr << " * [BUG] number of normals does not equal number of vertices\n";
|
|
#endif
|
|
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|