X-Git-Url: http://git.xonotic.org/?a=blobdiff_plain;f=contrib%2Fbrushexport%2Fexport.cpp;h=86a7767ba4f542298444394a0b68a1e57b3b6348;hb=1c26c420f1b5dabd39b492ab017447172ef5a6c8;hp=0407432e35d6c4b56589b07d9c1b72b64ac3dc4a;hpb=4741e6535e6bbdf58ab856a3edd7f2b37761592b;p=xonotic%2Fnetradiant.git diff --git a/contrib/brushexport/export.cpp b/contrib/brushexport/export.cpp index 0407432e..86a7767b 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,280 +13,270 @@ #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. -*/ + 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()); + if ( mode == COLLAPSE_ALL ) { + groups.push_back( group() ); current = &groups.back(); current->name = "all"; } } -ExportData::~ExportData(void) -{ - +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()); + if ( mode == COLLAPSE_NONE ) { + groups.push_back( group() ); current = &groups.back(); - - StringOutputStream str(256); + + StringOutputStream str( 256 ); str << "Brush" << (const unsigned int)groups.size(); current->name = str.c_str(); } } -void ExportData::EndBrush(void) -{ +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()) - { + 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); - + GetShaderNameFromShaderPath( f.GetShader(), shadername ); + // ignore faces from ignore list - if(ignorelist.find(shadername) != ignorelist.end()) + if ( ignorelist.find( shadername ) != ignorelist.end() ) { return; - - if(mode == COLLAPSE_BY_MATERIAL) - { + } + + 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) + const std::list::iterator end( groups.end() ); + for ( std::list::iterator it( groups.begin() ); it != end; ++it ) { - if(it->name == shadername) - current = &(*it); + if ( it->name == shadername ) { + current = &( *it ); + } } - + // no group found, create one - if(!current) - { - groups.push_back(group()); + if ( !current ) { + groups.push_back( group() ); current = &groups.back(); current->name = shadername; } } - - ASSERT_NOTNULL(current); - + + ASSERT_NOTNULL( current ); + // add face to current group - current->faces.push_back(&f); - -#ifdef _DEBUG + 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) -{ - std::string tmp(path); +void ExportData::GetShaderNameFromShaderPath( const char* path, std::string& name ){ + std::string tmp( path ); + + size_t last_slash = tmp.find_last_of( "/" ); - size_t last_slash = tmp.find_last_of("/"); - - if(last_slash != std::string::npos && last_slash == (tmp.length() - 1)) + if ( last_slash != std::string::npos && last_slash == ( tmp.length() - 1 ) ) { name = path; - else - name = tmp.substr(last_slash + 1, tmp.length() - last_slash); + } + else{ + name = tmp.substr( last_slash + 1, tmp.length() - last_slash ); + } -#ifdef _DEBUG +#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 -*/ + 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 -{ - std::string objFile = path.substr(0, path.length() -4) + ".obj"; - std::string mtlFile = path.substr(0, path.length() -4) + ".mtl"; +bool ExportDataAsWavefront::WriteToFile( const std::string& path, collapsemode mode ) const { + 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()) - { + 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(); + 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) + 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()); - + const std::list::const_iterator end( git->faces.end() ); + // submesh starts here - if(objs) - { + if ( objs ) { out << "\no "; - } else { + } + else { out << "\ng "; } out << git->name.c_str() << "\n"; // material - if(expmat && mode == COLLAPSE_ALL) - { + if ( expmat && mode == COLLAPSE_ALL ) { out << "usemtl material" << "\n\n"; - materials.insert("material"); + materials.insert( "material" ); } - for(std::list::const_iterator it(git->faces.begin()); it != end; ++it) + for ( std::list::const_iterator it( git->faces.begin() ); it != end; ++it ) { - const Winding& w((*it)->getWinding()); - + 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"; + 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) + out << "\n"; + + for ( std::list::const_iterator it( git->faces.begin() ); it != end; ++it ) { - const Winding& w((*it)->getWinding()); - + 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 ( 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) + + for ( std::list::const_iterator it( git->faces.begin() ); it != end; ++it ) { - const Winding& w((*it)->getWinding()); - + const Winding& w( ( *it )->getWinding() ); + // faces - StringOutputStream faceLine(256); + StringOutputStream faceLine( 256 ); faceLine << "\nf"; - for(size_t i = 0; i < w.numpoints; ++i, ++vertex_count) + for ( size_t i = 0; i < w.numpoints; ++i, ++vertex_count ) { - faceLine << " " << vertex_count+1 << "/" << vertex_count+1; + 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 { + 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) - { + if ( mode != COLLAPSE_ALL ) { std::string lastMat; std::string mat; std::string faces; - for(bm::iterator iter = brushMaterials.begin(); iter != brushMaterials.end(); iter++) + 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 { + 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(); } } @@ -298,28 +289,26 @@ bool ExportDataAsWavefront::WriteToFile(const std::string& path, collapsemode mo out << "\n"; } - if(expmat) - { - TextFileOutputStream outMtl(mtlFile.c_str()); - if(outMtl.failed()) - { + 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 << "# Wavefront material file exported with " RADIANT_NAME " brushexport plugin.\n"; outMtl << "# Material Count: " << (const Unsigned)materials.size() << "\n\n"; - for(std::set::const_iterator it(materials.begin()); it != materials.end(); ++it) + 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 { + 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; } @@ -327,50 +316,46 @@ bool ExportDataAsWavefront::WriteToFile(const std::string& path, collapsemode mo 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 { public: - 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(); - } - } - +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(); + } +} + private: - ExportData& exporter; +ExportData& exporter; }; - -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); + +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 ); }