mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-06 07:51:47 +00:00
Thomas Krijnen
85 lines
4.1 KiB
C++
85 lines
4.1 KiB
C++
/********************************************************************************
|
|
* *
|
|
* This file is part of IfcOpenShell. *
|
|
* *
|
|
* IfcOpenShell is free software: you can redistribute it and/or modify *
|
|
* it under the terms of the Lesser GNU General Public License as published by *
|
|
* the Free Software Foundation, either version 3.0 of the License, or *
|
|
* (at your option) any later version. *
|
|
* *
|
|
* IfcOpenShell 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 *
|
|
* Lesser GNU General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the Lesser GNU General Public License *
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
|
|
* *
|
|
********************************************************************************/
|
|
|
|
#include "../ifcconvert/SurfaceStyle.h"
|
|
|
|
#include "WavefrontOBJSerializer.h"
|
|
|
|
bool WaveFrontOBJSerializer::ready() {
|
|
return obj_stream.is_open() && mtl_stream.is_open();
|
|
}
|
|
|
|
void WaveFrontOBJSerializer::writeHeader() {
|
|
obj_stream << "# File generated by IfcOpenShell " << IFCOPENSHELL_VERSION << std::endl;
|
|
#ifdef WIN32
|
|
const char dir_separator = '\\';
|
|
#else
|
|
const char dir_separator = '/';
|
|
#endif
|
|
std::string mtl_basename = mtl_filename;
|
|
std::string::size_type slash = mtl_basename.find_last_of(dir_separator);
|
|
if (slash != std::string::npos) {
|
|
mtl_basename = mtl_basename.substr(slash+1);
|
|
}
|
|
obj_stream << "mtllib " << mtl_basename << std::endl;
|
|
mtl_stream << "# File generated by IfcOpenShell " << IFCOPENSHELL_VERSION << std::endl;
|
|
}
|
|
|
|
void WaveFrontOBJSerializer::writeMaterial(const SurfaceStyle& style) {
|
|
mtl_stream << "newmtl " << style.Name() << std::endl
|
|
<< "Kd " << style.Diffuse().R() << " " << style.Diffuse().G() << " " << style.Diffuse().B() << std::endl
|
|
<< "Ks " << style.Specular().R() << " " << style.Specular().G() << " " << style.Specular().B() << std::endl
|
|
<< "Ka " << style.Ambient().R() << " " << style.Ambient().G() << " " << style.Ambient().B() << std::endl
|
|
<< "Ns " << style.Specularity() << std::endl
|
|
<< "Tr " << style.Transparency() << std::endl
|
|
<< "d " << style.Transparency() << std::endl
|
|
<< "D " << style.Transparency() << std::endl;
|
|
}
|
|
|
|
void WaveFrontOBJSerializer::writeTesselated(const IfcGeomObjects::IfcGeomObject* o) {
|
|
const std::string name = o->name.empty() ? o->guid : o->name;
|
|
obj_stream << "g " << name << std::endl;
|
|
obj_stream << "s 1" << std::endl;
|
|
obj_stream << "usemtl " << o->type << std::endl;
|
|
if (materials.find(o->type) == materials.end()) {
|
|
writeMaterial(GetDefaultMaterial(o->type));
|
|
materials.insert(o->type);
|
|
}
|
|
const int vcount = o->mesh->verts.size() / 3;
|
|
for ( IfcGeomObjects::FltIt it = o->mesh->verts.begin(); it != o->mesh->verts.end(); ) {
|
|
const double x = *(it++);
|
|
const double y = *(it++);
|
|
const double z = *(it++);
|
|
obj_stream << "v " << x << " " << y << " " << z << std::endl;
|
|
}
|
|
for ( IfcGeomObjects::FltIt it = o->mesh->normals.begin(); it != o->mesh->normals.end(); ) {
|
|
const double x = *(it++);
|
|
const double y = *(it++);
|
|
const double z = *(it++);
|
|
obj_stream << "vn " << x << " " << y << " " << z << std::endl;
|
|
}
|
|
for ( IfcGeomObjects::IntIt it = o->mesh->faces.begin(); it != o->mesh->faces.end(); ) {
|
|
const int v1 = *(it++)+vcount_total;
|
|
const int v2 = *(it++)+vcount_total;
|
|
const int v3 = *(it++)+vcount_total;
|
|
obj_stream << "f " << v1 << "//" << v1 << " " << v2 << "//" << v2 << " " << v3 << "//" << v3 << std::endl;
|
|
}
|
|
vcount_total += vcount;
|
|
}
|