X-Git-Url: http://git.xonotic.org/?a=blobdiff_plain;f=contrib%2Fbrushexport%2Fexport.cpp;h=223545591865633cd0609576be23abd3e95d351d;hb=9dfae1c9b270ee369c6362903a9205b30751b95f;hp=0407432e35d6c4b56589b07d9c1b72b64ac3dc4a;hpb=4741e6535e6bbdf58ab856a3edd7f2b37761592b;p=xonotic%2Fnetradiant.git diff --git a/contrib/brushexport/export.cpp b/contrib/brushexport/export.cpp index 0407432e..22354559 100644 --- a/contrib/brushexport/export.cpp +++ b/contrib/brushexport/export.cpp @@ -1,4 +1,5 @@ #include "export.h" +#include "globaldefs.h" #include "debugging/debugging.h" #include "ibrush.h" #include "iscenegraph.h" @@ -12,365 +13,353 @@ #include "../../radiant/brush.h" // for limNames -#define MAX_MATERIAL_NAME 20 +const int MAX_MATERIAL_NAME = 20; /* - Abstract baseclass for modelexporters - the class collects all the data which then gets - exported through the WriteToFile method. -*/ -class ExportData -{ + Abstract baseclass for modelexporters + the class collects all the data which then gets + exported through the WriteToFile method. + */ +class ExportData { public: - ExportData(const std::set& ignorelist, collapsemode mode, bool limNames, bool objs); - virtual ~ExportData(void); - - virtual void BeginBrush(Brush& b); - virtual void AddBrushFace(Face& f); - virtual void EndBrush(void); - - virtual bool WriteToFile(const std::string& path, collapsemode mode) const = 0; - + ExportData(const std::set &ignorelist, collapsemode mode, bool limNames, bool objs); + + virtual ~ExportData(void); + + virtual void BeginBrush(Brush &b); + + virtual void AddBrushFace(Face &f); + + virtual void EndBrush(void); + + virtual bool WriteToFile(const std::string &path, collapsemode mode) const = 0; + protected: - // a group of faces - class group - { - public: - std::string name; - std::list faces; - }; - - std::list groups; - +// a group of faces + class group { + public: + std::string name; + std::list faces; + }; + + std::list groups; + private: - // "textures/common/caulk" -> "caulk" - void GetShaderNameFromShaderPath(const char* path, std::string& name); +// "textures/common/caulk" -> "caulk" + void GetShaderNameFromShaderPath(const char *path, std::string &name); - group* current; - collapsemode mode; - const std::set& ignorelist; + group *current; + collapsemode mode; + const std::set &ignorelist; }; -ExportData::ExportData(const std::set& _ignorelist, collapsemode _mode, bool _limNames, bool _objs) - : mode(_mode), - ignorelist(_ignorelist) +ExportData::ExportData(const std::set &_ignorelist, collapsemode _mode, bool _limNames, bool _objs) + : mode(_mode), + ignorelist(_ignorelist) { - current = 0; - - // in this mode, we need just one group - if(mode == COLLAPSE_ALL) - { - groups.push_back(group()); - current = &groups.back(); - current->name = "all"; - } + current = 0; + + // in this mode, we need just one group + if (mode == COLLAPSE_ALL) { + groups.push_back(group()); + current = &groups.back(); + current->name = "all"; + } } ExportData::~ExportData(void) { - + } -void ExportData::BeginBrush(Brush& b) +void ExportData::BeginBrush(Brush &b) { - // create a new group for each brush - if(mode == COLLAPSE_NONE) - { - groups.push_back(group()); - current = &groups.back(); - - StringOutputStream str(256); - str << "Brush" << (const unsigned int)groups.size(); - current->name = str.c_str(); - } + // create a new group for each brush + if (mode == COLLAPSE_NONE) { + groups.push_back(group()); + current = &groups.back(); + + StringOutputStream str(256); + str << "Brush" << (const unsigned int) groups.size(); + current->name = str.c_str(); + } } void ExportData::EndBrush(void) { - // all faces of this brush were on the ignorelist, discard the emptygroup - if(mode == COLLAPSE_NONE) - { - ASSERT_NOTNULL(current); - if(current->faces.empty()) - { - groups.pop_back(); - current = 0; - } - } + // all faces of this brush were on the ignorelist, discard the emptygroup + if (mode == COLLAPSE_NONE) { + ASSERT_NOTNULL(current); + if (current->faces.empty()) { + groups.pop_back(); + current = 0; + } + } } -void ExportData::AddBrushFace(Face& f) +void ExportData::AddBrushFace(Face &f) { - std::string shadername; - GetShaderNameFromShaderPath(f.GetShader(), shadername); - - // ignore faces from ignore list - if(ignorelist.find(shadername) != ignorelist.end()) - return; - - if(mode == COLLAPSE_BY_MATERIAL) - { - // find a group for this material - current = 0; - const std::list::iterator end(groups.end()); - for(std::list::iterator it(groups.begin()); it != end; ++it) - { - if(it->name == shadername) - current = &(*it); - } - - // no group found, create one - if(!current) - { - groups.push_back(group()); - current = &groups.back(); - current->name = shadername; - } - } - - ASSERT_NOTNULL(current); - - // add face to current group - current->faces.push_back(&f); - -#ifdef _DEBUG - globalOutputStream() << "Added Face to group " << current->name.c_str() << "\n"; + std::string shadername; + GetShaderNameFromShaderPath(f.GetShader(), shadername); + + // ignore faces from ignore list + if (ignorelist.find(shadername) != ignorelist.end()) { + return; + } + + if (mode == COLLAPSE_BY_MATERIAL) { + // find a group for this material + current = 0; + const std::list::iterator end(groups.end()); + for (std::list::iterator it(groups.begin()); it != end; ++it) { + if (it->name == shadername) { + current = &(*it); + } + } + + // no group found, create one + if (!current) { + groups.push_back(group()); + current = &groups.back(); + current->name = shadername; + } + } + + ASSERT_NOTNULL(current); + + // add face to current group + current->faces.push_back(&f); + +#if GDEF_DEBUG + globalOutputStream() << "Added Face to group " << current->name.c_str() << "\n"; #endif } -void ExportData::GetShaderNameFromShaderPath(const char* path, std::string& name) +void ExportData::GetShaderNameFromShaderPath(const char *path, std::string &name) { - std::string tmp(path); + std::string tmp(path); - size_t last_slash = tmp.find_last_of("/"); - - if(last_slash != std::string::npos && last_slash == (tmp.length() - 1)) - name = path; - else - name = tmp.substr(last_slash + 1, tmp.length() - last_slash); + size_t last_slash = tmp.find_last_of("/"); -#ifdef _DEBUG - globalOutputStream() << "Last: " << (const unsigned int) last_slash << " " << "length: " << (const unsigned int)tmp.length() << "Name: " << name.c_str() << "\n"; + if (last_slash != std::string::npos && last_slash == (tmp.length() - 1)) { + name = path; + } else { + name = tmp.substr(last_slash + 1, tmp.length() - last_slash); + } + +#if GDEF_DEBUG + globalOutputStream() << "Last: " << (const unsigned int) last_slash << " " << "length: " + << (const unsigned int) tmp.length() << "Name: " << name.c_str() << "\n"; #endif } /* - Exporter writing facedata as wavefront object -*/ -class ExportDataAsWavefront : public ExportData -{ + Exporter writing facedata as wavefront object + */ +class ExportDataAsWavefront : public ExportData { private: - bool expmat; - bool limNames; - bool objs; + bool expmat; + bool limNames; + bool objs; public: - ExportDataAsWavefront(const std::set& _ignorelist, collapsemode _mode, bool _expmat, bool _limNames, bool _objs) - : ExportData(_ignorelist, _mode, _limNames, _objs) - { - expmat = _expmat; - limNames = _limNames; - objs = _objs; - } - - bool WriteToFile(const std::string& path, collapsemode mode) const; + ExportDataAsWavefront(const std::set &_ignorelist, collapsemode _mode, bool _expmat, bool _limNames, + bool _objs) + : ExportData(_ignorelist, _mode, _limNames, _objs) + { + expmat = _expmat; + limNames = _limNames; + objs = _objs; + } + + bool WriteToFile(const std::string &path, collapsemode mode) const; }; -bool ExportDataAsWavefront::WriteToFile(const std::string& path, collapsemode mode) const +bool ExportDataAsWavefront::WriteToFile(const std::string &path, collapsemode mode) const { - std::string objFile = path.substr(0, path.length() -4) + ".obj"; - std::string mtlFile = path.substr(0, path.length() -4) + ".mtl"; - - std::set materials; - - TextFileOutputStream out(objFile.c_str()); - - if(out.failed()) - { - globalErrorStream() << "Unable to open file\n"; - return false; - } - - out << "# Wavefront Objectfile exported with radiants brushexport plugin 3.0 by Thomas 'namespace' Nitschke, spam@codecreator.net\n\n"; - - if(expmat) - { - size_t last = mtlFile.find_last_of("//"); - std::string mtllib = mtlFile.substr(last + 1, mtlFile.size() - last).c_str(); - out << "mtllib " << mtllib.c_str() << "\n"; - } - - unsigned int vertex_count = 0; - - const std::list::const_iterator gend(groups.end()); - for(std::list::const_iterator git(groups.begin()); git != gend; ++git) - { - typedef std::multimap bm; - bm brushMaterials; - typedef std::pair String_Pair; - - const std::list::const_iterator end(git->faces.end()); - - // submesh starts here - if(objs) - { - out << "\no "; - } else { - out << "\ng "; - } - out << git->name.c_str() << "\n"; - - // material - if(expmat && mode == COLLAPSE_ALL) - { - out << "usemtl material" << "\n\n"; - materials.insert("material"); - } - - for(std::list::const_iterator it(git->faces.begin()); it != end; ++it) - { - const Winding& w((*it)->getWinding()); - - // vertices - for(size_t i = 0; i < w.numpoints; ++i) - out << "v " << FloatFormat(w[i].vertex.x(), 1, 6) << " " << FloatFormat(w[i].vertex.z(), 1, 6) << " " << FloatFormat(w[i].vertex.y(), 1, 6) << "\n"; - } - out << "\n"; - - for(std::list::const_iterator it(git->faces.begin()); it != end; ++it) - { - const Winding& w((*it)->getWinding()); - - // texcoords - for(size_t i = 0; i < w.numpoints; ++i) - out << "vt " << FloatFormat(w[i].texcoord.x(), 1, 6) << " " << FloatFormat(w[i].texcoord.y(), 1, 6) << "\n"; - } - - for(std::list::const_iterator it(git->faces.begin()); it != end; ++it) - { - const Winding& w((*it)->getWinding()); - - // faces - StringOutputStream faceLine(256); - faceLine << "\nf"; - for(size_t i = 0; i < w.numpoints; ++i, ++vertex_count) - { - faceLine << " " << vertex_count+1 << "/" << vertex_count+1; - } - - if(mode != COLLAPSE_ALL) - { - materials.insert((*it)->getShader().getShader()); - brushMaterials.insert(String_Pair((*it)->getShader().getShader(), faceLine.c_str())); - } else { - out << faceLine.c_str(); - } - } - - if(mode != COLLAPSE_ALL) - { - std::string lastMat; - std::string mat; - std::string faces; - - for(bm::iterator iter = brushMaterials.begin(); iter != brushMaterials.end(); iter++) - { - mat = (*iter).first.c_str(); - faces = (*iter).second.c_str(); - - if(mat != lastMat) - { - if(limNames && mat.size() > MAX_MATERIAL_NAME) - { - out << "\nusemtl " << mat.substr(mat.size() - MAX_MATERIAL_NAME, mat.size()).c_str(); - } else { - out << "\nusemtl " << mat.c_str(); - } - } - - out << faces.c_str(); - lastMat = mat; - } - } - - out << "\n"; - } - - if(expmat) - { - TextFileOutputStream outMtl(mtlFile.c_str()); - if(outMtl.failed()) - { - globalErrorStream() << "Unable to open material file\n"; - return false; - } - - outMtl << "# Wavefront material file exported with NetRadiants brushexport plugin.\n"; - outMtl << "# Material Count: " << (const Unsigned)materials.size() << "\n\n"; - for(std::set::const_iterator it(materials.begin()); it != materials.end(); ++it) - { - if(limNames && it->size() > MAX_MATERIAL_NAME) - { - outMtl << "newmtl " << it->substr(it->size() - MAX_MATERIAL_NAME, it->size()).c_str() << "\n"; - } else { - outMtl << "newmtl " << it->c_str() << "\n"; - } - } - } - - return true; + std::string objFile = path; + + if (path.compare(path.length() - 4, 4, ".obj") != 0) { + objFile += ".obj"; + } + + std::string mtlFile = objFile.substr(0, objFile.length() - 4) + ".mtl"; + + std::set materials; + + TextFileOutputStream out(objFile.c_str()); + + if (out.failed()) { + globalErrorStream() << "Unable to open file\n"; + return false; + } + + out + << "# Wavefront Objectfile exported with radiants brushexport plugin 3.0 by Thomas 'namespace' Nitschke, spam@codecreator.net\n\n"; + + if (expmat) { + size_t last = mtlFile.find_last_of("//"); + std::string mtllib = mtlFile.substr(last + 1, mtlFile.size() - last).c_str(); + out << "mtllib " << mtllib.c_str() << "\n"; + } + + unsigned int vertex_count = 0; + + const std::list::const_iterator gend(groups.end()); + for (std::list::const_iterator git(groups.begin()); git != gend; ++git) { + typedef std::multimap bm; + bm brushMaterials; + typedef std::pair String_Pair; + + const std::list::const_iterator end(git->faces.end()); + + // submesh starts here + if (objs) { + out << "\no "; + } else { + out << "\ng "; + } + out << git->name.c_str() << "\n"; + + // material + if (expmat && mode == COLLAPSE_ALL) { + out << "usemtl material" << "\n\n"; + materials.insert("material"); + } + + for (std::list::const_iterator it(git->faces.begin()); it != end; ++it) { + const Winding &w((*it)->getWinding()); + + // vertices + for (size_t i = 0; i < w.numpoints; ++i) { + out << "v " << FloatFormat(w[i].vertex.x(), 1, 6) << " " << FloatFormat(w[i].vertex.z(), 1, 6) << " " + << FloatFormat(w[i].vertex.y(), 1, 6) << "\n"; + } + } + out << "\n"; + + for (std::list::const_iterator it(git->faces.begin()); it != end; ++it) { + const Winding &w((*it)->getWinding()); + + // texcoords + for (size_t i = 0; i < w.numpoints; ++i) { + out << "vt " << FloatFormat(w[i].texcoord.x(), 1, 6) << " " << FloatFormat(w[i].texcoord.y(), 1, 6) + << "\n"; + } + } + + for (std::list::const_iterator it(git->faces.begin()); it != end; ++it) { + const Winding &w((*it)->getWinding()); + + // faces + StringOutputStream faceLine(256); + faceLine << "\nf"; + for (size_t i = 0; i < w.numpoints; ++i, ++vertex_count) { + faceLine << " " << vertex_count + 1 << "/" << vertex_count + 1; + } + + if (mode != COLLAPSE_ALL) { + materials.insert((*it)->getShader().getShader()); + brushMaterials.insert(String_Pair((*it)->getShader().getShader(), faceLine.c_str())); + } else { + out << faceLine.c_str(); + } + } + + if (mode != COLLAPSE_ALL) { + std::string lastMat; + std::string mat; + std::string faces; + + for (bm::iterator iter = brushMaterials.begin(); iter != brushMaterials.end(); iter++) { + mat = (*iter).first.c_str(); + faces = (*iter).second.c_str(); + + if (mat != lastMat) { + if (limNames && mat.size() > MAX_MATERIAL_NAME) { + out << "\nusemtl " << mat.substr(mat.size() - MAX_MATERIAL_NAME, mat.size()).c_str(); + } else { + out << "\nusemtl " << mat.c_str(); + } + } + + out << faces.c_str(); + lastMat = mat; + } + } + + out << "\n"; + } + + if (expmat) { + TextFileOutputStream outMtl(mtlFile.c_str()); + if (outMtl.failed()) { + globalErrorStream() << "Unable to open material file\n"; + return false; + } + + outMtl << "# Wavefront material file exported with NetRadiants brushexport plugin.\n"; + outMtl << "# Material Count: " << (const Unsigned) materials.size() << "\n\n"; + for (std::set::const_iterator it(materials.begin()); it != materials.end(); ++it) { + if (limNames && it->size() > MAX_MATERIAL_NAME) { + outMtl << "newmtl " << it->substr(it->size() - MAX_MATERIAL_NAME, it->size()).c_str() << "\n"; + } else { + outMtl << "newmtl " << it->c_str() << "\n"; + } + } + } + + return true; } -class ForEachFace : public BrushVisitor -{ +class ForEachFace : public BrushVisitor { public: - ForEachFace(ExportData& _exporter) - : exporter(_exporter) - {} - - void visit(Face& face) const - { - exporter.AddBrushFace(face); - } - + ForEachFace(ExportData &_exporter) + : exporter(_exporter) + {} + + void visit(Face &face) const + { + exporter.AddBrushFace(face); + } + private: - ExportData& exporter; + ExportData &exporter; }; -class ForEachSelected : public SelectionSystem::Visitor -{ +class ForEachSelected : public SelectionSystem::Visitor { public: - ForEachSelected(ExportData& _exporter) - : exporter(_exporter) - {} - - void visit(scene::Instance& instance) const + ForEachSelected(ExportData &_exporter) + : exporter(_exporter) + {} + + void visit(scene::Instance &instance) const { - BrushInstance* bptr = InstanceTypeCast::cast(instance); - if(bptr) - { - Brush& brush(bptr->getBrush()); - - exporter.BeginBrush(brush); - ForEachFace face_vis(exporter); - brush.forEachFace(face_vis); - exporter.EndBrush(); - } + BrushInstance *bptr = InstanceTypeCast::cast(instance); + if (bptr) { + Brush &brush(bptr->getBrush()); + + exporter.BeginBrush(brush); + ForEachFace face_vis(exporter); + brush.forEachFace(face_vis); + exporter.EndBrush(); + } } - + private: - ExportData& exporter; + ExportData &exporter; }; - -bool ExportSelection(const std::set& ignorelist, collapsemode m, bool exmat, const std::string& path, bool limNames, bool objs) + +bool ExportSelection(const std::set &ignorelist, collapsemode m, bool exmat, const std::string &path, + bool limNames, bool objs) { - ExportDataAsWavefront exporter(ignorelist, m, exmat, limNames, objs); - - ForEachSelected vis(exporter); - GlobalSelectionSystem().foreachSelected(vis); - - return exporter.WriteToFile(path, m); + ExportDataAsWavefront exporter(ignorelist, m, exmat, limNames, objs); + + ForEachSelected vis(exporter); + GlobalSelectionSystem().foreachSelected(vis); + + return exporter.WriteToFile(path, m); }