khanat-opennel-code/code/nel/tools/3d/mesh_utils/assimp_material.cpp

// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2015  Winch Gate Property Limited
// Author: Jan Boon <jan.boon@kaetemi.be>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#include <nel/misc/types_nl.h>
#include "assimp_shape.h"

#include <assimp/postprocess.h>
#include <assimp/scene.h>
#include <assimp/Importer.hpp>

#define NL_NODE_INTERNAL_TYPE aiNode
#define NL_SCENE_INTERNAL_TYPE aiScene
#include "scene_context.h"

#include <nel/misc/debug.h>
#include <nel/misc/path.h>
#include <nel/misc/tool_logger.h>

#include <nel/3d/mesh.h>

using namespace std;
using namespace NLMISC;
using namespace NL3D;

// http://assimp.sourceforge.net/lib_html/materials.html

inline CRGBA convColor(const aiColor3D &ac, uint8 a = 255)
{
	return CRGBA(ac.r * 255.99f, ac.g * 255.99f, ac.b * 255.99f);
}

inline CRGBA convColor(const aiColor4D &ac)
{
	return CRGBA(ac.r * 255.99f, ac.g * 255.99f, ac.b * 255.99f, ac.a * 255.99f);
}

void assimpMaterial(NL3D::CMaterial &mat, CMeshUtilsContext &context, const aiMaterial *am)
{
	mat.initLighted();
	mat.setShader(CMaterial::Normal);

	int i;
	float f;
	aiColor3D c3;
	aiColor4D c4;

	if (am->Get(AI_MATKEY_TWOSIDED, i) == aiReturn_SUCCESS)
		mat.setDoubleSided(i != 0);

	if (am->Get(AI_MATKEY_BLEND_FUNC, i) == aiReturn_SUCCESS) switch ((aiBlendMode)i)
	{
	case aiBlendMode_Default:
		mat.setSrcBlend(CMaterial::srcalpha);
		mat.setDstBlend(CMaterial::invsrcalpha);
		break;
	case aiBlendMode_Additive:
		mat.setSrcBlend(CMaterial::one);
		mat.setDstBlend(CMaterial::one);
		break;
	}

	// Colors follow GL convention
	// "While the ambient, diffuse, specular and emission
	// "material parameters all have alpha components, only the diffuse"
	// "alpha component is used in the lighting computation."
	if (am->Get(AI_MATKEY_COLOR_DIFFUSE, c3) == aiReturn_SUCCESS)
		mat.setDiffuse(convColor(c3));

	if (am->Get(AI_MATKEY_OPACITY, f) == aiReturn_SUCCESS)
		mat.setOpacity(f * 255.99f);

	if (am->Get(AI_MATKEY_COLOR_AMBIENT, c3) == aiReturn_SUCCESS)
		mat.setAmbient(convColor(c3));

	if (am->Get(AI_MATKEY_SHININESS, f) == aiReturn_SUCCESS)
		mat.setShininess(f); // (float)pow(2.0, f * 10.0) * 4.f;

	if (am->Get(AI_MATKEY_COLOR_SPECULAR, c3) == aiReturn_SUCCESS)
		mat.setSpecular(convColor(c3));

	if (am->Get(AI_MATKEY_SHININESS_STRENGTH, f) == aiReturn_SUCCESS)
		mat.setSpecular(CRGBAF(mat.getSpecular()) * f);
	else
		mat.setSpecular(NLMISC::CRGBA::Black);

	if (am->Get(AI_MATKEY_COLOR_EMISSIVE, c3) == aiReturn_SUCCESS)
		mat.setEmissive(convColor(c3));
}

CSmartPtr<CMaterial> assimpMaterial(CMeshUtilsContext &context, const aiMaterial *am)
{
	CSmartPtr<CMaterial> matp = new CMaterial();
	CMaterial &mat = *matp;
	assimpMaterial(mat, context, am);
	return matp;
}

void assimpMaterials(CMeshUtilsContext &context)
{
	set<CSString> materialNames;

	const aiScene *scene = context.InternalScene;
	for (unsigned int mi = 0; mi < scene->mNumMaterials; ++mi)
	{
		const aiMaterial *am = scene->mMaterials[mi];

		for (unsigned int pi = 0; pi < am->mNumProperties; ++pi) // DEBUG
		{ // DEBUG
			const aiMaterialProperty *amp = am->mProperties[pi];
			printf("%s\n", amp->mKey.C_Str());
		} // DEBUG

		aiString amname;
		if (am->Get(AI_MATKEY_NAME, amname) != aiReturn_SUCCESS)
		{
			tlerror(context.ToolLogger, context.Settings.SourceFilePath.c_str(),
				"Material has no name");
			continue;
		}

		if (materialNames.find(amname.C_Str()) != materialNames.end())
		{
			tlerror(context.ToolLogger, context.Settings.SourceFilePath.c_str(),
				"Material name '%s' used more than once", amname.C_Str());
			continue;
		}
		
		if (context.SceneMeta.Materials.find(amname.C_Str())
			== context.SceneMeta.Materials.end())
		{
			materialNames.insert(amname.C_Str());
			context.SceneMeta.Materials[amname.C_Str()] = assimpMaterial(context, am);
		}
	}
}

/* end of file */