-#include "export.h"\r
-#include "debugging/debugging.h"\r
-#include "ibrush.h"\r
-#include "iscenegraph.h"\r
-#include "iselection.h"\r
-#include "stream/stringstream.h"\r
-#include "stream/textfilestream.h"\r
-\r
-// this is very evil, but right now there is no better way\r
-#include "../../radiant/brush.h"\r
-\r
-/*\r
- Abstract baseclass for modelexporters\r
- the class collects all the data which then gets\r
- exported through the WriteToFile method.\r
-*/\r
-class ExportData\r
-{\r
-public:\r
- ExportData(const std::set<std::string>& ignorelist, collapsemode mode);\r
- virtual ~ExportData(void);\r
- \r
- virtual void BeginBrush(Brush& b);\r
- virtual void AddBrushFace(Face& f);\r
- virtual void EndBrush(void);\r
- \r
- virtual bool WriteToFile(const std::string& path) const = 0;\r
- \r
-protected:\r
-\r
- // a group of faces\r
- class group\r
- {\r
- public:\r
- std::string name;\r
- std::list<const Face*> faces;\r
- };\r
- \r
- std::list<group> groups;\r
- \r
-private:\r
-\r
- // "textures/common/caulk" -> "caulk"\r
- void GetShaderNameFromShaderPath(const char* path, std::string& name);\r
-\r
- group* current; \r
- collapsemode mode;\r
- const std::set<std::string>& ignorelist;\r
-};\r
-\r
-ExportData::ExportData(const std::set<std::string>& _ignorelist, collapsemode _mode)\r
- : mode(_mode),\r
- ignorelist(_ignorelist)\r
-{\r
- current = 0;\r
- \r
- // in this mode, we need just one group\r
- if(mode == COLLAPSE_ALL)\r
- {\r
- groups.push_back(group());\r
- current = &groups.back();\r
- current->name = "all";\r
- }\r
-}\r
-\r
-ExportData::~ExportData(void)\r
-{\r
- \r
-}\r
-\r
-void ExportData::BeginBrush(Brush& b)\r
-{\r
- // create a new group for each brush\r
- if(mode == COLLAPSE_NONE)\r
- {\r
- groups.push_back(group());\r
- current = &groups.back();\r
- \r
- StringOutputStream str(256);\r
- str << "Brush" << (const unsigned int)groups.size();\r
- current->name = str.c_str();\r
- }\r
-}\r
-\r
-void ExportData::EndBrush(void)\r
-{\r
- // all faces of this brush were on the ignorelist, discard the emptygroup\r
- if(mode == COLLAPSE_NONE)\r
- {\r
- ASSERT_NOTNULL(current);\r
- if(current->faces.empty())\r
- {\r
- groups.pop_back();\r
- current = 0;\r
- }\r
- }\r
-}\r
-\r
-void ExportData::AddBrushFace(Face& f)\r
-{\r
- std::string shadername;\r
- GetShaderNameFromShaderPath(f.GetShader(), shadername);\r
- \r
- // faces mit materials auf der ignoreliste ignorieren\r
- if(ignorelist.find(shadername) != ignorelist.end())\r
- return;\r
- \r
- if(mode == COLLAPSE_BY_MATERIAL)\r
- {\r
- // find a group for this material\r
- current = 0;\r
- const std::list<group>::iterator end(groups.end());\r
- for(std::list<group>::iterator it(groups.begin()); it != end; ++it)\r
- {\r
- if(it->name == shadername)\r
- current = &(*it);\r
- }\r
- \r
- // no group found, create one\r
- if(!current)\r
- {\r
- groups.push_back(group());\r
- current = &groups.back();\r
- current->name = shadername;\r
- }\r
- }\r
- \r
- ASSERT_NOTNULL(current);\r
- \r
- // add face to current group\r
- current->faces.push_back(&f);\r
- \r
-#ifdef _DEBUG\r
- globalOutputStream() << "Added Face to group " << current->name.c_str() << "\n";\r
-#endif\r
-}\r
-\r
-void ExportData::GetShaderNameFromShaderPath(const char* path, std::string& name)\r
-{\r
- std::string tmp(path);\r
-\r
- size_t last_slash = tmp.find_last_of("/");\r
- \r
- if(last_slash != std::string::npos && last_slash == (tmp.length() - 1))\r
- name = path;\r
- else\r
- name = tmp.substr(last_slash + 1, tmp.length() - last_slash);\r
- \r
- globalOutputStream() << "Last: " << last_slash << " " << "lenght: " << (const unsigned int)tmp.length() << "Name: " << name.c_str() << "\n";\r
-}\r
-\r
-/*\r
- Exporter writing facedata as wavefront object\r
-*/\r
-class ExportDataAsWavefront : public ExportData\r
-{\r
-public:\r
- ExportDataAsWavefront(const std::set<std::string>& _ignorelist, collapsemode _mode)\r
- : ExportData(_ignorelist, _mode)\r
- {\r
- }\r
- \r
- bool WriteToFile(const std::string& path) const;\r
-};\r
-\r
-bool ExportDataAsWavefront::WriteToFile(const std::string& path) const\r
-{\r
- TextFileOutputStream out(path.c_str());\r
- \r
- if(out.failed())\r
- {\r
- globalErrorStream() << "Unable to open file\n";\r
- return false;\r
- }\r
- \r
- out << "# Wavefront Objectfile exported with radiants brushexport plugin 2.0 by Thomas 'namespace' Nitschke, spam@codecreator.net\n\n";\r
- \r
- unsigned int vertex_count = 0;\r
-\r
- const std::list<ExportData::group>::const_iterator gend(groups.end());\r
- for(std::list<ExportData::group>::const_iterator git(groups.begin()); git != gend; ++git)\r
- {\r
- const std::list<const Face*>::const_iterator end(git->faces.end());\r
- \r
- // submesh starts here\r
- out << "\ng " << git->name.c_str() << "\n";\r
- \r
- for(std::list<const Face*>::const_iterator it(git->faces.begin()); it != end; ++it)\r
- {\r
- const Winding& w((*it)->getWinding());\r
- \r
- // vertices\r
- for(size_t i = 0; i < w.numpoints; ++i)\r
- out << "v " << w[i].vertex.x() << " " << w[i].vertex.y() << " " << w[i].vertex.z() << "\n";\r
- }\r
- out << "\n"; \r
- \r
- for(std::list<const Face*>::const_iterator it(git->faces.begin()); it != end; ++it)\r
- {\r
- const Winding& w((*it)->getWinding());\r
- \r
- // texcoords\r
- for(size_t i = 0; i < w.numpoints; ++i)\r
- out << "vt " << w[i].texcoord.x() << " " << w[i].texcoord.y() << "\n";\r
- }\r
- \r
- for(std::list<const Face*>::const_iterator it(git->faces.begin()); it != end; ++it)\r
- {\r
- const Winding& w((*it)->getWinding());\r
- \r
- // faces\r
- out << "\nf";\r
- for(size_t i = 0; i < w.numpoints; ++i, ++vertex_count)\r
- {\r
- out << " " << vertex_count+1 << "/" << vertex_count+1;\r
- }\r
- }\r
- out << "\n";\r
- }\r
- \r
- return true;\r
-}\r
-\r
-\r
-class ForEachFace : public BrushVisitor\r
-{\r
-public:\r
- ForEachFace(ExportData& _exporter)\r
- : exporter(_exporter)\r
- {}\r
- \r
- void visit(Face& face) const\r
- {\r
- exporter.AddBrushFace(face);\r
- }\r
- \r
-private:\r
- ExportData& exporter;\r
-};\r
-\r
-class ForEachSelected : public SelectionSystem::Visitor\r
-{\r
-public:\r
- ForEachSelected(ExportData& _exporter)\r
- : exporter(_exporter)\r
- {}\r
- \r
- void visit(scene::Instance& instance) const\r
- {\r
- BrushInstance* bptr = InstanceTypeCast<BrushInstance>::cast(instance);\r
- if(bptr)\r
- {\r
- Brush& brush(bptr->getBrush());\r
- \r
- exporter.BeginBrush(brush);\r
- ForEachFace face_vis(exporter);\r
- brush.forEachFace(face_vis);\r
- exporter.EndBrush();\r
- }\r
- }\r
- \r
-private:\r
- ExportData& exporter;\r
-};\r
- \r
-bool ExportSelection(const std::set<std::string>& ignorelist, collapsemode m, const std::string& path)\r
-{\r
- ExportDataAsWavefront exporter(ignorelist, m);\r
- \r
- ForEachSelected vis(exporter);\r
- GlobalSelectionSystem().foreachSelected(vis);\r
- \r
- return exporter.WriteToFile(path);\r
-}\r
+#include "export.h"
+#include "globaldefs.h"
+#include "debugging/debugging.h"
+#include "ibrush.h"
+#include "iscenegraph.h"
+#include "iselection.h"
+#include "stream/stringstream.h"
+#include "stream/textfilestream.h"
+
+#include <map>
+
+// this is very evil, but right now there is no better way
+#include "../../radiant/brush.h"
+
+// for limNames
+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
+{
+public:
+ExportData( const std::set<std::string>& 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<const Face*> faces;
+};
+
+std::list<group> groups;
+
+private:
+
+// "textures/common/caulk" -> "caulk"
+void GetShaderNameFromShaderPath( const char* path, std::string& name );
+
+group* current;
+collapsemode mode;
+const std::set<std::string>& ignorelist;
+};
+
+ExportData::ExportData( const std::set<std::string>& _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";
+ }
+}
+
+ExportData::~ExportData( void ){
+
+}
+
+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();
+ }
+}
+
+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;
+ }
+ }
+}
+
+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<group>::iterator end( groups.end() );
+ for ( std::list<group>::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 ){
+ 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 );
+ }
+
+#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
+{
+private:
+bool expmat;
+bool limNames;
+bool objs;
+
+public:
+ExportDataAsWavefront( const std::set<std::string>& _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;
+
+ if ( path.compare( path.length() - 4, 4, ".obj" ) != 0 ) {
+ objFile += ".obj";
+ }
+
+ std::string mtlFile = objFile.substr( 0, objFile.length() - 4 ) + ".mtl";
+
+ std::set<std::string> 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<ExportData::group>::const_iterator gend( groups.end() );
+ for ( std::list<ExportData::group>::const_iterator git( groups.begin() ); git != gend; ++git )
+ {
+ typedef std::multimap<std::string, std::string> bm;
+ bm brushMaterials;
+ typedef std::pair<std::string, std::string> String_Pair;
+
+ const std::list<const Face*>::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 Face*>::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 Face*>::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 Face*>::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 " RADIANT_NAME " brushexport plugin.\n";
+ outMtl << "# Material Count: " << (const Unsigned)materials.size() << "\n\n";
+ for ( std::set<std::string>::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
+{
+public:
+ForEachFace( ExportData& _exporter )
+ : exporter( _exporter )
+{}
+
+void visit( Face& face ) const {
+ exporter.AddBrushFace( face );
+}
+
+private:
+ExportData& exporter;
+};
+
+class ForEachSelected : public SelectionSystem::Visitor
+{
+public:
+ForEachSelected( ExportData& _exporter )
+ : exporter( _exporter )
+{}
+
+void visit( scene::Instance& instance ) const {
+ BrushInstance* bptr = InstanceTypeCast<BrushInstance>::cast( instance );
+ if ( bptr ) {
+ Brush& brush( bptr->getBrush() );
+
+ exporter.BeginBrush( brush );
+ ForEachFace face_vis( exporter );
+ brush.forEachFace( face_vis );
+ exporter.EndBrush();
+ }
+}
+
+private:
+ExportData& exporter;
+};
+
+bool ExportSelection( const std::set<std::string>& 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 );
+}
projects / xonotic / netradiant.git / blobdiff