X-Git-Url: http://git.xonotic.org/?a=blobdiff_plain;f=contrib%2Fbrushexport%2Fexport.cpp;h=223545591865633cd0609576be23abd3e95d351d;hb=9dfae1c9b270ee369c6362903a9205b30751b95f;hp=4a1e2d4c2d2566796aeebc2033c5b9eefd511c89;hpb=7fc621fc78d0e040dc2c12f38dc53dd9048215dc;p=xonotic%2Fnetradiant.git

diff --git a/contrib/brushexport/export.cpp b/contrib/brushexport/export.cpp
index 4a1e2d4c..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,349 +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
-{
+class ExportData {
 public:
-ExportData( const std::set<std::string>& ignorelist, collapsemode mode, bool limNames, bool objs );
-virtual ~ExportData( void );
+    ExportData(const std::set<std::string> &ignorelist, collapsemode mode, bool limNames, bool objs);
+
+    virtual ~ExportData(void);
+
+    virtual void BeginBrush(Brush &b);
 
-virtual void BeginBrush( Brush& b );
-virtual void AddBrushFace( Face& f );
-virtual void EndBrush( void );
+    virtual void AddBrushFace(Face &f);
 
-virtual bool WriteToFile( const std::string& path, collapsemode mode ) const = 0;
+    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;
-};
+    class group {
+    public:
+        std::string name;
+        std::list<const Face *> faces;
+    };
 
-std::list<group> groups;
+    std::list<group> groups;
 
 private:
 
 // "textures/common/caulk" -> "caulk"
-void GetShaderNameFromShaderPath( const char* path, std::string& name );
+    void GetShaderNameFromShaderPath(const char *path, std::string &name);
 
-group* current;
-collapsemode mode;
-const std::set<std::string>& ignorelist;
+    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(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 ){
+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::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::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 );
-
-#ifdef _DEBUG
-	globalOutputStream() << "Added Face to group " << current->name.c_str() << "\n";
+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 );
+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 ) ) {
-		name = path;
-	}
-	else{
-		name = tmp.substr( last_slash + 1, tmp.length() - last_slash );
-	}
+    if (last_slash != std::string::npos && last_slash == (tmp.length() - 1)) {
+        name = path;
+    } else {
+        name = tmp.substr(last_slash + 1, tmp.length() - last_slash);
+    }
 
-#ifdef _DEBUG
-	globalOutputStream() << "Last: " << (const unsigned int) last_slash << " " << "length: " << (const unsigned int)tmp.length() << "Name: " << name.c_str() << "\n";
+#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
-{
+class ExportDataAsWavefront : public ExportData {
 private:
-bool expmat;
-bool limNames;
-bool objs;
+    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;
+    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 NetRadiants 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;
+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 NetRadiants 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
-{
+class ForEachFace : public BrushVisitor {
 public:
-ForEachFace( ExportData& _exporter )
-	: exporter( _exporter )
-{}
+    ForEachFace(ExportData &_exporter)
+            : exporter(_exporter)
+    {}
 
-void visit( Face& face ) const {
-	exporter.AddBrushFace( face );
-}
+    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 {
-	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();
-	}
-}
+    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;
+    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 );
+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 );
+    ForEachSelected vis(exporter);
+    GlobalSelectionSystem().foreachSelected(vis);
 
-	return exporter.WriteToFile( path, m );
+    return exporter.WriteToFile(path, m);
 }