Danbias/Code/OysterGraphics/FileLoader/ObjReader.cpp

#include "OBJReader.h"
#include "..\Definitions\GraphicalDefinition.h"
#include <sstream>
#include <fstream>
#include "GeneralLoader.h"

using namespace std;
OBJReader::OBJReader() 
{
	_mPos		= 0;
	_mNormal	= 0;
	_mTexel		= 0;
}

OBJReader::~OBJReader()
{

}

void OBJReader::readOBJFile( std::wstring fileName )
{
	std::fstream inStream;
	std::string typeOfData = " ";
	float vertexData;
	std::string face1, face2, face3;

	inStream.open( fileName + L".obj", std::fstream::in );
	
	if( inStream.is_open() )
	{
		while( !inStream.eof() )
		{
			inStream >> typeOfData;

			if( typeOfData == "v" )
			{
				Oyster::Math::Float3 position;

				inStream >> vertexData;
				position.x = vertexData;
				
				inStream >> vertexData;
				position.y = vertexData;

				inStream >> vertexData;
				position.z = vertexData;

				_mVertexCoord.push_back( position );

			}
			else if( typeOfData == "vt" )
			{
				Oyster::Math::Float2 texel;
				inStream >> vertexData;
				texel.x = vertexData;

				inStream >> vertexData;
				texel.y = 1 - vertexData;

				_mVertexTexture.push_back( texel );
			}
			else if( typeOfData == "vn" )
			{
				Oyster::Math::Float3 normal;
				inStream >> vertexData;
				normal.x = vertexData;

				inStream >> vertexData;
				normal.y = vertexData;

				inStream >> vertexData;
				normal.z = vertexData;

				_mVertexNormal.push_back( normal );
			}
			else if( typeOfData == "f" )
			{
				inStream >> face1;
				stringSplit( face1 );
				
				addToOBJarray();

				inStream >> face2;
				stringSplit( face2 );

				addToOBJarray();

				inStream >> face3;
				stringSplit( face3 );

				addToOBJarray();
			}
		}
	}

	inStream.close();

	Mat = Oyster::Resource::OysterResource::LoadResource((fileName + L".png").c_str(),Oyster::Graphics::Loading::LoadTexture);
}

Oyster::Graphics::Model::ModelInfo* OBJReader::toModel()
{
	Oyster::Graphics::Core::Buffer* b = new Oyster::Graphics::Core::Buffer();
	Oyster::Graphics::Core::Buffer::BUFFER_INIT_DESC desc;
	Oyster::Graphics::Model::ModelInfo* modelInfo = new Oyster::Graphics::Model::ModelInfo();
	
	desc.ElementSize = 32;
	desc.InitData = &this->_myOBJ[0];
	desc.NumElements = this->_myOBJ.size();
	desc.Type = Oyster::Graphics::Core::Buffer::BUFFER_TYPE::VERTEX_BUFFER;
	desc.Usage = Oyster::Graphics::Core::Buffer::BUFFER_DEFAULT;
	HRESULT hr = S_OK;
	
	hr = b->Init(desc);
	if(FAILED(hr))
	{
		//Something isn't okay here
	}
	modelInfo->Indexed = false;
	modelInfo->VertexCount = (int)desc.NumElements;
	modelInfo->Vertices = b;
	modelInfo->Material.push_back((ID3D11ShaderResourceView*)Mat);

	return modelInfo;
}

//Private functions
void OBJReader::stringSplit( std::string strToSplit )
{
	char delim = '/';
	std::string vPos, vNormal, vTexel;
	std::stringstream aStream(strToSplit);
	getline( aStream, vPos, delim );
	getline( aStream, vTexel, delim );
	getline( aStream, vNormal );

	_mPos		= atoi( vPos.c_str() );
	_mNormal	= atoi( vNormal.c_str() );
	_mTexel		= atoi( vTexel.c_str() );

}

void OBJReader::addToOBJarray()
{
	OBJFormat temp;

	temp._d3VertexCoord		= _mVertexCoord.at( _mPos - 1 );
	temp._d3VertexNormal	= _mVertexNormal.at( _mNormal - 1 );
	temp._d3VertexTexture	= _mVertexTexture.at( _mTexel - 1 );

	_myOBJ.push_back( temp );
}