189 lines
7.2 KiB
C++
189 lines
7.2 KiB
C++
/*
|
|
Open Asset Import Library (assimp)
|
|
----------------------------------------------------------------------
|
|
|
|
Copyright (c) 2006-2017, assimp team
|
|
|
|
All rights reserved.
|
|
|
|
Redistribution and use of this software in source and binary forms,
|
|
with or without modification, are permitted provided that the
|
|
following conditions are met:
|
|
|
|
* Redistributions of source code must retain the above
|
|
copyright notice, this list of conditions and the
|
|
following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above
|
|
copyright notice, this list of conditions and the
|
|
following disclaimer in the documentation and/or other
|
|
materials provided with the distribution.
|
|
|
|
* Neither the name of the assimp team, nor the names of its
|
|
contributors may be used to endorse or promote products
|
|
derived from this software without specific prior
|
|
written permission of the assimp team.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
----------------------------------------------------------------------
|
|
*/
|
|
|
|
|
|
|
|
#if !defined(ASSIMP_BUILD_NO_EXPORT) && !defined(ASSIMP_BUILD_NO_STL_EXPORTER)
|
|
|
|
#include "STLExporter.h"
|
|
#include <assimp/version.h>
|
|
#include <assimp/IOSystem.hpp>
|
|
#include <assimp/scene.h>
|
|
#include <assimp/Exporter.hpp>
|
|
#include <memory>
|
|
#include "Exceptional.h"
|
|
#include "ByteSwapper.h"
|
|
|
|
using namespace Assimp;
|
|
namespace Assimp {
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// Worker function for exporting a scene to Stereolithograpy. Prototyped and registered in Exporter.cpp
|
|
void ExportSceneSTL(const char* pFile,IOSystem* pIOSystem, const aiScene* pScene, const ExportProperties* /*pProperties*/)
|
|
{
|
|
// invoke the exporter
|
|
STLExporter exporter(pFile, pScene);
|
|
|
|
if (exporter.mOutput.fail()) {
|
|
throw DeadlyExportError("output data creation failed. Most likely the file became too large: " + std::string(pFile));
|
|
}
|
|
|
|
// we're still here - export successfully completed. Write the file.
|
|
std::unique_ptr<IOStream> outfile (pIOSystem->Open(pFile,"wt"));
|
|
if(outfile == NULL) {
|
|
throw DeadlyExportError("could not open output .stl file: " + std::string(pFile));
|
|
}
|
|
|
|
outfile->Write( exporter.mOutput.str().c_str(), static_cast<size_t>(exporter.mOutput.tellp()),1);
|
|
}
|
|
void ExportSceneSTLBinary(const char* pFile,IOSystem* pIOSystem, const aiScene* pScene, const ExportProperties* /*pProperties*/)
|
|
{
|
|
// invoke the exporter
|
|
STLExporter exporter(pFile, pScene, true);
|
|
|
|
if (exporter.mOutput.fail()) {
|
|
throw DeadlyExportError("output data creation failed. Most likely the file became too large: " + std::string(pFile));
|
|
}
|
|
|
|
// we're still here - export successfully completed. Write the file.
|
|
std::unique_ptr<IOStream> outfile (pIOSystem->Open(pFile,"wb"));
|
|
if(outfile == NULL) {
|
|
throw DeadlyExportError("could not open output .stl file: " + std::string(pFile));
|
|
}
|
|
|
|
outfile->Write( exporter.mOutput.str().c_str(), static_cast<size_t>(exporter.mOutput.tellp()),1);
|
|
}
|
|
|
|
} // end of namespace Assimp
|
|
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
STLExporter :: STLExporter(const char* _filename, const aiScene* pScene, bool binary)
|
|
: filename(_filename)
|
|
, endl("\n")
|
|
{
|
|
// make sure that all formatting happens using the standard, C locale and not the user's current locale
|
|
const std::locale& l = std::locale("C");
|
|
mOutput.imbue(l);
|
|
mOutput.precision(16);
|
|
if (binary) {
|
|
char buf[80] = {0} ;
|
|
buf[0] = 'A'; buf[1] = 's'; buf[2] = 's'; buf[3] = 'i'; buf[4] = 'm'; buf[5] = 'p';
|
|
buf[6] = 'S'; buf[7] = 'c'; buf[8] = 'e'; buf[9] = 'n'; buf[10] = 'e';
|
|
mOutput.write(buf, 80);
|
|
unsigned int meshnum = 0;
|
|
for(unsigned int i = 0; i < pScene->mNumMeshes; ++i) {
|
|
for (unsigned int j = 0; j < pScene->mMeshes[i]->mNumFaces; ++j) {
|
|
meshnum++;
|
|
}
|
|
}
|
|
AI_SWAP4(meshnum);
|
|
mOutput.write((char *)&meshnum, 4);
|
|
for(unsigned int i = 0; i < pScene->mNumMeshes; ++i) {
|
|
WriteMeshBinary(pScene->mMeshes[i]);
|
|
}
|
|
} else {
|
|
const std::string& name = "AssimpScene";
|
|
|
|
mOutput << "solid " << name << endl;
|
|
for(unsigned int i = 0; i < pScene->mNumMeshes; ++i) {
|
|
WriteMesh(pScene->mMeshes[i]);
|
|
}
|
|
mOutput << "endsolid " << name << endl;
|
|
}
|
|
}
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
void STLExporter :: WriteMesh(const aiMesh* m)
|
|
{
|
|
for (unsigned int i = 0; i < m->mNumFaces; ++i) {
|
|
const aiFace& f = m->mFaces[i];
|
|
|
|
// we need per-face normals. We specified aiProcess_GenNormals as pre-requisite for this exporter,
|
|
// but nonetheless we have to expect per-vertex normals.
|
|
aiVector3D nor;
|
|
if (m->mNormals) {
|
|
for(unsigned int a = 0; a < f.mNumIndices; ++a) {
|
|
nor += m->mNormals[f.mIndices[a]];
|
|
}
|
|
nor.Normalize();
|
|
}
|
|
mOutput << " facet normal " << nor.x << " " << nor.y << " " << nor.z << endl;
|
|
mOutput << " outer loop" << endl;
|
|
for(unsigned int a = 0; a < f.mNumIndices; ++a) {
|
|
const aiVector3D& v = m->mVertices[f.mIndices[a]];
|
|
mOutput << " vertex " << v.x << " " << v.y << " " << v.z << endl;
|
|
}
|
|
|
|
mOutput << " endloop" << endl;
|
|
mOutput << " endfacet" << endl << endl;
|
|
}
|
|
}
|
|
|
|
void STLExporter :: WriteMeshBinary(const aiMesh* m)
|
|
{
|
|
for (unsigned int i = 0; i < m->mNumFaces; ++i) {
|
|
const aiFace& f = m->mFaces[i];
|
|
// we need per-face normals. We specified aiProcess_GenNormals as pre-requisite for this exporter,
|
|
// but nonetheless we have to expect per-vertex normals.
|
|
aiVector3D nor;
|
|
if (m->mNormals) {
|
|
for(unsigned int a = 0; a < f.mNumIndices; ++a) {
|
|
nor += m->mNormals[f.mIndices[a]];
|
|
}
|
|
nor.Normalize();
|
|
}
|
|
ai_real nx = nor.x, ny = nor.y, nz = nor.z;
|
|
AI_SWAP4(nx); AI_SWAP4(ny); AI_SWAP4(nz);
|
|
mOutput.write((char *)&nx, 4); mOutput.write((char *)&ny, 4); mOutput.write((char *)&nz, 4);
|
|
for(unsigned int a = 0; a < f.mNumIndices; ++a) {
|
|
const aiVector3D& v = m->mVertices[f.mIndices[a]];
|
|
ai_real vx = v.x, vy = v.y, vz = v.z;
|
|
AI_SWAP4(vx); AI_SWAP4(vy); AI_SWAP4(vz);
|
|
mOutput.write((char *)&vx, 4); mOutput.write((char *)&vy, 4); mOutput.write((char *)&vz, 4);
|
|
}
|
|
char dummy[2] = {0};
|
|
mOutput.write(dummy, 2);
|
|
}
|
|
}
|
|
|
|
#endif
|