FEATURE: First prototype for ObjFile material-importer, the assimp data structure will not be created yet.

git-svn-id: https://assimp.svn.sourceforge.net/svnroot/assimp/trunk@17 67173fc5-114c-0410-ac8e-9d2fd5bffc1f

code/ObjFileMtlImporter.cpp

@@ -1,12 +1,31 @@
 #include "ObjFileMtlImporter.h"
+#include "aiTypes.h"
+#include "ObjTools.h"
+#include "ObjFileData.h"
+#include "aiAssert.h"
+#include "fast_atof.h"
+
 
 namespace Assimp
 {
 
 // -------------------------------------------------------------------
-ObjFileMtlImporter::ObjFileMtlImporter()
+ObjFileMtlImporter::ObjFileMtlImporter( std::vector<char> &buffer, 
+									   const std::string &strAbsPath,
+									   ObjFile::Model *pModel ) :
+	m_DataIt( buffer.begin() ),
+	m_DataItEnd( buffer.end() ),
+	m_uiLine( 0 ),
+	m_pModel( NULL )
 {
-	// TODO: Inplement this
+	ai_assert ( NULL != m_pModel );
+	if ( NULL == m_pModel->m_pDefaultMaterial )
+	{
+		m_pModel->m_pDefaultMaterial = new ObjFile::Material();
+		
+		//m_pModel->m_pDefaultMaterial->
+	}
+	load();
 }
 
 // -------------------------------------------------------------------
@@ -22,33 +41,154 @@ ObjFileMtlImporter::ObjFileMtlImporter(const ObjFileMtlImporter &rOther)
 }
 	
 // -------------------------------------------------------------------
-ObjFileMtlImporter &ObjFileMtlImporter::operator = (const ObjFileMtlImporter &rOther)
+ObjFileMtlImporter &ObjFileMtlImporter::operator = (
+	const ObjFileMtlImporter &rOther)
 {
 	return *this;
 }
+
 // -------------------------------------------------------------------
-void ObjFileMtlImporter::getColorRGBA()
+void ObjFileMtlImporter::load()
 {
+	if ( m_DataIt == m_DataItEnd )
+		return;
+
+	while ( m_DataIt != m_DataItEnd )
+	{
+		switch (*m_DataIt)
+		{
+		case 'K':
+			{
+				++m_DataIt;
+				if (*m_DataIt == 'a') // Ambient color
+				{
+					getColorRGBA( &m_pModel->m_pCurrentMaterial->ambient );
+				}
+				else if (*m_DataIt == 'd')	// Diffuse color
+				{
+					getColorRGBA( &m_pModel->m_pCurrentMaterial->diffuse );
+				}
+				else if (*m_DataIt == 's')
+				{
+					getColorRGBA( &m_pModel->m_pCurrentMaterial->specular );
+				}
+				m_DataIt = skipLine<DataArrayIt>( m_DataIt, m_DataItEnd, m_uiLine );
+			}
+			break;
+
+		case 'd':	// Alpha value
+			{
+				getFloatValue( m_pModel->m_pCurrentMaterial->alpha );
+				m_DataIt = skipLine<DataArrayIt>( m_DataIt, m_DataItEnd, m_uiLine );
+			}
+			break;
+
+		case 'N':	// Shineness
+			{
+				getIlluminationModel( m_pModel->m_pCurrentMaterial->illumination_model );
+				m_DataIt = skipLine<DataArrayIt>( m_DataIt, m_DataItEnd, m_uiLine );
+			}
+			break;
+
+		case 'm':	// Texture
+			{
+				getTexture();
+				m_DataIt = skipLine<DataArrayIt>( m_DataIt, m_DataItEnd, m_uiLine );
+			}
+			break;
+
+		case 'n':	// New material name
+			{
+				createMaterial();
+				m_DataIt = skipLine<DataArrayIt>( m_DataIt, m_DataItEnd, m_uiLine );
+			}
+			break;
+
+		case 'i':	// Illumination model
+			{
+				m_DataIt = skipLine<DataArrayIt>( m_DataIt, m_DataItEnd, m_uiLine );
+			}
+			break;
+
+		default:
+			{
+				m_DataIt = skipLine<DataArrayIt>( m_DataIt, m_DataItEnd, m_uiLine );
+			}
+			break;
+		}
+	}
 }
 
 // -------------------------------------------------------------------
-void ObjFileMtlImporter::getIlluminationModel()
+void ObjFileMtlImporter::getColorRGBA( aiColor3D *pColor )
 {
+	ai_assert( NULL != pColor );
+	
+	float r, g, b;
+	m_DataIt = CopyNextWord<DataArrayIt>( m_DataIt, m_DataItEnd, m_buffer, BUFFERSIZE );
+	r = (float) fast_atof(m_buffer);
+
+	m_DataIt = CopyNextWord<DataArrayIt>( m_DataIt, m_DataItEnd, m_buffer, BUFFERSIZE );
+	g = (float) fast_atof(m_buffer);
+
+	m_DataIt = CopyNextWord<DataArrayIt>( m_DataIt, m_DataItEnd, m_buffer, BUFFERSIZE );
+	b = (float) fast_atof(m_buffer);
+
+	pColor->r = r;
+	pColor->g = g;
+	pColor->b = b;
 }
 
 // -------------------------------------------------------------------
-void ObjFileMtlImporter::getFloatValue()
+void ObjFileMtlImporter::getIlluminationModel( int &illum_model )
 {
+	m_DataIt = CopyNextWord<DataArrayIt>( m_DataIt, m_DataItEnd, m_buffer, BUFFERSIZE );
+	illum_model = atoi(m_buffer);
+}
+
+// -------------------------------------------------------------------
+void ObjFileMtlImporter::getFloatValue( float &value )
+{
+	m_DataIt = CopyNextWord<DataArrayIt>( m_DataIt, m_DataItEnd, m_buffer, BUFFERSIZE );
+	value = (float) fast_atof(m_buffer);
 }
 
 // -------------------------------------------------------------------
 void ObjFileMtlImporter::createMaterial()
 {
+	m_pModel->m_pCurrentMaterial = new ObjFile::Material();
+	
+	/*m_DataIt = getNextToken<DataArrayIt>( m_DataIt, m_DataItEnd );
+	if (m_DataIt == m_DataItEnd)
+		return;
+	
+	char *pStart = &(*m_DataIt);
+	while ( !isSpace(*m_DataIt) && m_DataIt != m_DataItEnd )
+		++m_DataIt;
+
+	// Get name
+	std::string strName(pStart, &(*m_DataIt));
+	if ( strName.empty() )
+		return;*/
+	std::string strName;
+	m_DataIt = getName<DataArrayIt>( m_DataIt, m_DataItEnd, strName );
+	if ( m_DataItEnd == m_DataIt )
+		return;
+
+	m_pModel->m_pCurrentMaterial->MaterialName.Set( strName );
+	m_pModel->m_MaterialLib.push_back( strName );
+	m_pModel->m_MaterialMap[ strName ] = m_pModel->m_pCurrentMaterial;
 }
 
 // -------------------------------------------------------------------
 void ObjFileMtlImporter::getTexture()
 {
+	std::string strTexture;
+	m_DataIt = getName<DataArrayIt>( m_DataIt, m_DataItEnd, strTexture );
+	if ( m_DataItEnd == m_DataIt )
+		return;
+
+	m_pModel->m_pCurrentMaterial->texture.Set( strTexture );
 }
 
 // -------------------------------------------------------------------