| /* |
| 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. |
| |
| @author: Richard Steffen, 2015 |
| ---------------------------------------------------------------------- |
| */ |
| |
| |
| #ifndef ASSIMP_BUILD_NO_EXPORT |
| #ifndef ASSIMP_BUILD_NO_STEP_EXPORTER |
| |
| #include "StepExporter.h" |
| #include "ConvertToLHProcess.h" |
| #include "Bitmap.h" |
| #include "BaseImporter.h" |
| #include "fast_atof.h" |
| #include <assimp/SceneCombiner.h> |
| #include <iostream> |
| #include <ctime> |
| #include <set> |
| #include <map> |
| #include <list> |
| #include <memory> |
| #include "Exceptional.h" |
| #include <assimp/DefaultIOSystem.h> |
| #include <assimp/IOSystem.hpp> |
| #include <assimp/scene.h> |
| #include <assimp/light.h> |
| |
| // |
| #if _MSC_VER > 1500 || (defined __GNUC___) |
| # define ASSIMP_STEP_USE_UNORDERED_MULTIMAP |
| # else |
| # define step_unordered_map map |
| # define step_unordered_multimap multimap |
| #endif |
| |
| #ifdef ASSIMP_STEP_USE_UNORDERED_MULTIMAP |
| # include <unordered_map> |
| # if _MSC_VER > 1600 |
| # define step_unordered_map unordered_map |
| # define step_unordered_multimap unordered_multimap |
| # else |
| # define step_unordered_map tr1::unordered_map |
| # define step_unordered_multimap tr1::unordered_multimap |
| # endif |
| #endif |
| |
| typedef std::step_unordered_map<aiVector3D*, int> VectorIndexUMap; |
| |
| /* Tested with Step viewer v4 from www.ida-step.net */ |
| |
| using namespace Assimp; |
| |
| namespace Assimp |
| { |
| |
| // ------------------------------------------------------------------------------------------------ |
| // Worker function for exporting a scene to Collada. Prototyped and registered in Exporter.cpp |
| void ExportSceneStep(const char* pFile,IOSystem* pIOSystem, const aiScene* pScene, const ExportProperties* pProperties) |
| { |
| std::string path = DefaultIOSystem::absolutePath(std::string(pFile)); |
| std::string file = DefaultIOSystem::completeBaseName(std::string(pFile)); |
| |
| // create/copy Properties |
| ExportProperties props(*pProperties); |
| |
| // invoke the exporter |
| StepExporter iDoTheExportThing( pScene, pIOSystem, path, file, &props); |
| |
| // we're still here - export successfully completed. Write result to the given IOSYstem |
| std::unique_ptr<IOStream> outfile (pIOSystem->Open(pFile,"wt")); |
| if(outfile == NULL) { |
| throw DeadlyExportError("could not open output .stp file: " + std::string(pFile)); |
| } |
| |
| // XXX maybe use a small wrapper around IOStream that behaves like std::stringstream in order to avoid the extra copy. |
| outfile->Write( iDoTheExportThing.mOutput.str().c_str(), static_cast<size_t>(iDoTheExportThing.mOutput.tellp()),1); |
| } |
| |
| } // end of namespace Assimp |
| |
| |
| namespace { |
| // Collect world transformations for each node |
| void CollectTrafos(const aiNode* node, std::map<const aiNode*, aiMatrix4x4>& trafos) { |
| const aiMatrix4x4& parent = node->mParent ? trafos[node->mParent] : aiMatrix4x4(); |
| trafos[node] = parent * node->mTransformation; |
| for (unsigned int i = 0; i < node->mNumChildren; ++i) { |
| CollectTrafos(node->mChildren[i], trafos); |
| } |
| } |
| |
| // Generate a flat list of the meshes (by index) assigned to each node |
| void CollectMeshes(const aiNode* node, std::multimap<const aiNode*, unsigned int>& meshes) { |
| for (unsigned int i = 0; i < node->mNumMeshes; ++i) { |
| meshes.insert(std::make_pair(node, node->mMeshes[i])); |
| } |
| for (unsigned int i = 0; i < node->mNumChildren; ++i) { |
| CollectMeshes(node->mChildren[i], meshes); |
| } |
| } |
| } |
| |
| // ------------------------------------------------------------------------------------------------ |
| // Constructor for a specific scene to export |
| StepExporter::StepExporter(const aiScene* pScene, IOSystem* pIOSystem, const std::string& path, |
| const std::string& file, const ExportProperties* pProperties): |
| mProperties(pProperties),mIOSystem(pIOSystem),mFile(file), mPath(path), |
| mScene(pScene), endstr(";\n") { |
| CollectTrafos(pScene->mRootNode, trafos); |
| CollectMeshes(pScene->mRootNode, meshes); |
| |
| // make sure that all formatting happens using the standard, C locale and not the user's current locale |
| mOutput.imbue( std::locale("C") ); |
| mOutput.precision(16); |
| |
| // start writing |
| WriteFile(); |
| } |
| |
| // ------------------------------------------------------------------------------------------------ |
| // Starts writing the contents |
| void StepExporter::WriteFile() |
| { |
| // see http://shodhganga.inflibnet.ac.in:8080/jspui/bitstream/10603/14116/11/11_chapter%203.pdf |
| // note, that all realnumber values must be comma separated in x files |
| mOutput.setf(std::ios::fixed); |
| // precission for double |
| // see http://stackoverflow.com/questions/554063/how-do-i-print-a-double-value-with-full-precision-using-cout |
| mOutput.precision(16); |
| |
| // standard color |
| aiColor4D fColor; |
| fColor.r = 0.8f; |
| fColor.g = 0.8f; |
| fColor.b = 0.8f; |
| |
| int ind = 100; // the start index to be used |
| int faceEntryLen = 30; // number of entries for a triangle/face |
| // prepare unique (count triangles and vertices) |
| |
| VectorIndexUMap uniqueVerts; // use a map to reduce find complexity to log(n) |
| VectorIndexUMap::iterator it; |
| int countFace = 0; |
| |
| for (unsigned int i=0; i<mScene->mNumMeshes; ++i) |
| { |
| aiMesh* mesh = mScene->mMeshes[i]; |
| for (unsigned int j=0; j<mesh->mNumFaces; ++j) |
| { |
| aiFace* face = &(mesh->mFaces[j]); |
| |
| if (face->mNumIndices == 3) countFace++; |
| } |
| for (unsigned int j=0; j<mesh->mNumVertices; ++j) |
| { |
| aiVector3D* v = &(mesh->mVertices[j]); |
| it =uniqueVerts.find(v); |
| if (it == uniqueVerts.end()) |
| { |
| uniqueVerts[v] = -1; // first mark the vector as not transformed |
| } |
| } |
| } |
| |
| static const unsigned int date_nb_chars = 20; |
| char date_str[date_nb_chars]; |
| std::time_t date = std::time(NULL); |
| std::strftime(date_str, date_nb_chars, "%Y-%m-%dT%H:%M:%S", std::localtime(&date)); |
| |
| // write the header |
| mOutput << "ISO-10303-21" << endstr; |
| mOutput << "HEADER" << endstr; |
| mOutput << "FILE_DESCRIPTION(('STEP AP214'),'1')" << endstr; |
| mOutput << "FILE_NAME('" << mFile << ".stp','" << date_str << "',(' '),(' '),'Spatial InterOp 3D',' ',' ')" << endstr; |
| mOutput << "FILE_SCHEMA(('automotive_design'))" << endstr; |
| mOutput << "ENDSEC" << endstr; |
| |
| // write the top of data |
| mOutput << "DATA" << endstr; |
| mOutput << "#1=MECHANICAL_DESIGN_GEOMETRIC_PRESENTATION_REPRESENTATION(' ',("; |
| for (int i=0; i<countFace; ++i) |
| { |
| mOutput << "#" << i*faceEntryLen + ind + 2*uniqueVerts.size(); |
| if (i!=countFace-1) mOutput << ","; |
| } |
| mOutput << "),#6)" << endstr; |
| |
| mOutput << "#2=PRODUCT_DEFINITION_CONTEXT('',#7,'design')" << endstr; |
| mOutput << "#3=APPLICATION_PROTOCOL_DEFINITION('INTERNATIONAL STANDARD','automotive_design',1994,#7)" << endstr; |
| mOutput << "#4=PRODUCT_CATEGORY_RELATIONSHIP('NONE','NONE',#8,#9)" << endstr; |
| mOutput << "#5=SHAPE_DEFINITION_REPRESENTATION(#10,#11)" << endstr; |
| mOutput << "#6= (GEOMETRIC_REPRESENTATION_CONTEXT(3)GLOBAL_UNCERTAINTY_ASSIGNED_CONTEXT((#12))GLOBAL_UNIT_ASSIGNED_CONTEXT((#13,#14,#15))REPRESENTATION_CONTEXT('NONE','WORKSPACE'))" << endstr; |
| mOutput << "#7=APPLICATION_CONTEXT(' ')" << endstr; |
| mOutput << "#8=PRODUCT_CATEGORY('part','NONE')" << endstr; |
| mOutput << "#9=PRODUCT_RELATED_PRODUCT_CATEGORY('detail',' ',(#17))" << endstr; |
| mOutput << "#10=PRODUCT_DEFINITION_SHAPE('NONE','NONE',#18)" << endstr; |
| mOutput << "#11=MANIFOLD_SURFACE_SHAPE_REPRESENTATION('Root',(#16,#19),#6)" << endstr; |
| mOutput << "#12=UNCERTAINTY_MEASURE_WITH_UNIT(LENGTH_MEASURE(1.0E-006),#13,'','')" << endstr; |
| mOutput << "#13=(CONVERSION_BASED_UNIT('METRE',#20)LENGTH_UNIT()NAMED_UNIT(#21))" << endstr; |
| mOutput << "#14=(NAMED_UNIT(#22)PLANE_ANGLE_UNIT()SI_UNIT($,.RADIAN.))" << endstr; |
| mOutput << "#15=(NAMED_UNIT(#22)SOLID_ANGLE_UNIT()SI_UNIT($,.STERADIAN.))" << endstr; |
| mOutput << "#16=SHELL_BASED_SURFACE_MODEL('Root',(#29))" << endstr; |
| mOutput << "#17=PRODUCT('Root','Root','Root',(#23))" << endstr; |
| mOutput << "#18=PRODUCT_DEFINITION('NONE','NONE',#24,#2)" << endstr; |
| mOutput << "#19=AXIS2_PLACEMENT_3D('',#25,#26,#27)" << endstr; |
| mOutput << "#20=LENGTH_MEASURE_WITH_UNIT(LENGTH_MEASURE(1.0),#28)" << endstr; |
| mOutput << "#21=DIMENSIONAL_EXPONENTS(1.0,0.0,0.0,0.0,0.0,0.0,0.0)" << endstr; |
| mOutput << "#22=DIMENSIONAL_EXPONENTS(0.0,0.0,0.0,0.0,0.0,0.0,0.0)" << endstr; |
| mOutput << "#23=PRODUCT_CONTEXT('',#7,'mechanical')" << endstr; |
| mOutput << "#24=PRODUCT_DEFINITION_FORMATION_WITH_SPECIFIED_SOURCE(' ','NONE',#17,.NOT_KNOWN.)" << endstr; |
| mOutput << "#25=CARTESIAN_POINT('',(0.0,0.0,0.0))" << endstr; |
| mOutput << "#26=DIRECTION('',(0.0,0.0,1.0))" << endstr; |
| mOutput << "#27=DIRECTION('',(1.0,0.0,0.0))" << endstr; |
| mOutput << "#28= (NAMED_UNIT(#21)LENGTH_UNIT()SI_UNIT(.MILLI.,.METRE.))" << endstr; |
| mOutput << "#29=CLOSED_SHELL('',("; |
| for (int i=0; i<countFace; ++i) |
| { |
| mOutput << "#" << i*faceEntryLen + ind + 2*uniqueVerts.size() + 8; |
| if (i!=countFace-1) mOutput << ","; |
| } |
| mOutput << "))" << endstr; |
| |
| // write all the unique transformed CARTESIAN and VERTEX |
| for (MeshesByNodeMap::const_iterator it2 = meshes.begin(); it2 != meshes.end(); ++it2) |
| { |
| const aiNode& node = *(*it2).first; |
| unsigned int mesh_idx = (*it2).second; |
| |
| const aiMesh* mesh = mScene->mMeshes[mesh_idx]; |
| aiMatrix4x4& trafo = trafos[&node]; |
| for (unsigned int i = 0; i < mesh->mNumVertices; ++i) |
| { |
| aiVector3D* v = &(mesh->mVertices[i]); |
| it = uniqueVerts.find(v); |
| if (it->second >=0 ) continue; |
| it->second = ind; // this one is new, so set the index (ind) |
| aiVector3D vt = trafo * (*v); // transform the coordinate |
| mOutput << "#" << it->second << "=CARTESIAN_POINT('',(" << vt.x << "," << vt.y << "," << vt.z << "))" << endstr; |
| mOutput << "#" << it->second+1 << "=VERTEX_POINT('',#" << it->second << ")" << endstr; |
| ind += 2; |
| } |
| } |
| |
| // write the triangles |
| for (unsigned int i=0; i<mScene->mNumMeshes; ++i) |
| { |
| aiMesh* mesh = mScene->mMeshes[i]; |
| for (unsigned int j=0; j<mesh->mNumFaces; ++j) |
| { |
| aiFace* face = &(mesh->mFaces[j]); |
| |
| if (face->mNumIndices != 3) continue; |
| |
| aiVector3D* v1 = &(mesh->mVertices[face->mIndices[0]]); |
| aiVector3D* v2 = &(mesh->mVertices[face->mIndices[1]]); |
| aiVector3D* v3 = &(mesh->mVertices[face->mIndices[2]]); |
| aiVector3D dv12 = *v2 - *v1; |
| aiVector3D dv23 = *v3 - *v2; |
| aiVector3D dv31 = *v1 - *v3; |
| aiVector3D dv13 = *v3 - *v1; |
| dv12.Normalize(); |
| dv23.Normalize(); |
| dv31.Normalize(); |
| dv13.Normalize(); |
| |
| int pid1 = uniqueVerts.find(v1)->second; |
| int pid2 = uniqueVerts.find(v2)->second; |
| int pid3 = uniqueVerts.find(v3)->second; |
| |
| // mean vertex color for the face if available |
| if (mesh->HasVertexColors(0)) |
| { |
| fColor.r = 0.0; |
| fColor.g = 0.0; |
| fColor.b = 0.0; |
| fColor += mesh->mColors[0][face->mIndices[0]]; |
| fColor += mesh->mColors[0][face->mIndices[1]]; |
| fColor += mesh->mColors[0][face->mIndices[2]]; |
| fColor /= 3.0f; |
| } |
| |
| int sid = ind; // the sub index |
| mOutput << "#" << sid << "=STYLED_ITEM('',(#" << sid+1 << "),#" << sid+8 << ")" << endstr; /* the item that must be referenced in #1 */ |
| /* This is the color information of the Triangle */ |
| mOutput << "#" << sid+1 << "=PRESENTATION_STYLE_ASSIGNMENT((#" << sid+2 << "))" << endstr; |
| mOutput << "#" << sid+2 << "=SURFACE_STYLE_USAGE(.BOTH.,#" << sid+3 << ")" << endstr; |
| mOutput << "#" << sid+3 << "=SURFACE_SIDE_STYLE('',(#" << sid+4 << "))" << endstr; |
| mOutput << "#" << sid+4 << "=SURFACE_STYLE_FILL_AREA(#" << sid+5 << ")" << endstr; |
| mOutput << "#" << sid+5 << "=FILL_AREA_STYLE('',(#" << sid+6 << "))" << endstr; |
| mOutput << "#" << sid+6 << "=FILL_AREA_STYLE_COLOUR('',#" << sid+7 << ")" << endstr; |
| mOutput << "#" << sid+7 << "=COLOUR_RGB(''," << fColor.r << "," << fColor.g << "," << fColor.b << ")" << endstr; |
| |
| /* this is the geometry */ |
| mOutput << "#" << sid+8 << "=FACE_SURFACE('',(#" << sid+13 << "),#" << sid+9<< ",.T.)" << endstr; /* the face that must be referenced in 29 */ |
| |
| /* 2 directions of the plane */ |
| mOutput << "#" << sid+9 << "=PLANE('',#" << sid+10 << ")" << endstr; |
| mOutput << "#" << sid+10 << "=AXIS2_PLACEMENT_3D('',#" << pid1 << ", #" << sid+11 << ",#" << sid+12 << ")" << endstr; |
| |
| mOutput << "#" << sid+11 << "=DIRECTION('',(" << dv12.x << "," << dv12.y << "," << dv12.z << "))" << endstr; |
| mOutput << "#" << sid+12 << "=DIRECTION('',(" << dv13.x << "," << dv13.y << "," << dv13.z << "))" << endstr; |
| |
| mOutput << "#" << sid+13 << "=FACE_BOUND('',#" << sid+14 << ",.T.)" << endstr; |
| mOutput << "#" << sid+14 << "=EDGE_LOOP('',(#" << sid+15 << ",#" << sid+16 << ",#" << sid+17 << "))" << endstr; |
| |
| /* edge loop */ |
| mOutput << "#" << sid+15 << "=ORIENTED_EDGE('',*,*,#" << sid+18 << ",.T.)" << endstr; |
| mOutput << "#" << sid+16 << "=ORIENTED_EDGE('',*,*,#" << sid+19 << ",.T.)" << endstr; |
| mOutput << "#" << sid+17 << "=ORIENTED_EDGE('',*,*,#" << sid+20 << ",.T.)" << endstr; |
| |
| /* oriented edges */ |
| mOutput << "#" << sid+18 << "=EDGE_CURVE('',#" << pid1+1 << ",#" << pid2+1 << ",#" << sid+21 << ",.F.)" << endstr; |
| mOutput << "#" << sid+19 << "=EDGE_CURVE('',#" << pid2+1 << ",#" << pid3+1 << ",#" << sid+22 << ",.T.)" << endstr; |
| mOutput << "#" << sid+20 << "=EDGE_CURVE('',#" << pid3+1 << ",#" << pid1+1 << ",#" << sid+23 << ",.T.)" << endstr; |
| |
| /* 3 lines and 3 vectors for the lines for the 3 edge curves */ |
| mOutput << "#" << sid+21 << "=LINE('',#" << pid1 << ",#" << sid+24 << ")" << endstr; |
| mOutput << "#" << sid+22 << "=LINE('',#" << pid2 << ",#" << sid+25 << ")" << endstr; |
| mOutput << "#" << sid+23 << "=LINE('',#" << pid3 << ",#" << sid+26 << ")" << endstr; |
| mOutput << "#" << sid+24 << "=VECTOR('',#" << sid+27 << ",1.0)" << endstr; |
| mOutput << "#" << sid+25 << "=VECTOR('',#" << sid+28 << ",1.0)" << endstr; |
| mOutput << "#" << sid+26 << "=VECTOR('',#" << sid+29 << ",1.0)" << endstr; |
| mOutput << "#" << sid+27 << "=DIRECTION('',(" << dv12.x << "," << dv12.y << "," << dv12.z << "))" << endstr; |
| mOutput << "#" << sid+28 << "=DIRECTION('',(" << dv23.x << "," << dv23.y << "," << dv23.z << "))" << endstr; |
| mOutput << "#" << sid+29 << "=DIRECTION('',(" << dv31.x << "," << dv31.y << "," << dv31.z << "))" << endstr; |
| ind += faceEntryLen; // increase counter |
| } |
| } |
| |
| mOutput << "ENDSEC" << endstr; // end of data section |
| mOutput << "END-ISO-10303-21" << endstr; // end of file |
| } |
| |
| #endif |
| #endif |