netradiant-custom/contrib/brushexport/export.cpp

#include "export.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
#define 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 );

#ifdef _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 );
	}

#ifdef _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";

	typedef std::pair<std::string, Colour3> Material;
	auto materials_comparator = []( const Material& ma1, const Material& ma2 ) {
		return ma1.first < ma2.first;
	};
	auto materials  = std::set<Material, decltype( materials_comparator )>( materials_comparator );

	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( std::make_pair( std::string( "material" ), Colour3( 0.5, 0.5, 0.5 ) ) );
		}

		for ( std::list<const Face*>::const_iterator it( git->faces.begin() ); it != end; ++it )
		{
			const Winding& w( ( *it )->getWinding() );

			// vertices
			size_t i = w.numpoints;
			do{
				--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";
			}
			while( i != 0 );
		}
		out << "\n";

		for ( std::list<const Face*>::const_iterator it( git->faces.begin() ); it != end; ++it )
		{
			const Winding& w( ( *it )->getWinding() );

			// texcoords
			size_t i = w.numpoints;
			do{
				--i;
				out << "vt " << FloatFormat( w[i].texcoord.x(), 1, 6 ) << " " << FloatFormat( -w[i].texcoord.y(), 1, 6 ) << "\n";
			}
			while( i != 0 );
		}

		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";

			size_t i = w.numpoints;
			do{
				--i;
				++vertex_count;
				faceLine << " " << vertex_count << "/" << vertex_count;
			}
			while( i != 0 );

			if ( mode != COLLAPSE_ALL ) {
				materials.insert( std::make_pair( std::string( ( *it )->getShader().getShader() ), ( *it )->getShader().state()->getTexture().color ) );
				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::pair<std::string, Colour3>>::const_iterator it( materials.begin() ); it != materials.end(); ++it )
		{
			const std::string& str = it->first;
			const Colour3& clr = it->second;
			if ( limNames && str.size() > MAX_MATERIAL_NAME ) {
				outMtl << "newmtl " << str.substr( str.size() - MAX_MATERIAL_NAME, str.size() ).c_str() << "\n";
			}
			else {
				outMtl << "newmtl " << str.c_str() << "\n";
			}
			outMtl << "Kd " << clr.x() << " " << clr.y() << " " << clr.z() << "\n";
			outMtl << "map_Kd " << str.c_str() << "\n";

		}
	}

	return true;
}


class ForEachFace : public BrushVisitor
{
public:
ForEachFace( ExportData& _exporter )
	: exporter( _exporter )
{}

void visit( Face& face ) const {
	if( face.contributes() )
		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 );
}