#include "tiny_obj_loader.h"

#include <cstdio>
#include <cstdlib>
#include <cassert>
#include <iostream>

static bool
TestLoadObj(
	const char* fileName,
	bool verbose)
{
	const char* prefix[] = {"./data/", "../data/", "../../data/", "../../../data/", "../../../../data/"};
	char fullPath[1024];
	int index = -1;
	{
		int numPrefixes = sizeof(prefix) / sizeof(char*);

		for (int i = 0; i < numPrefixes; i++)
		{
			sprintf(fullPath, "%s%s", prefix[i], fileName);
			FILE* f;
			f = fopen(fullPath, "r");
			if (f)
			{
				index = i;
				fclose(f);
				break;
			}
		}
	}

	if (index < 0)
	{
		printf("file not found %s\n", fileName);
		return false;
	}

	std::cout << "Loading " << fullPath << std::endl;

	std::vector<tinyobj::shape_t> shapes;
	std::string err = tinyobj::LoadObj(shapes, fullPath, prefix[index]);

	if (!err.empty())
	{
		std::cerr << err << std::endl;
		return false;
	}

	std::cout << "# of shapes : " << shapes.size() << std::endl;

	if (verbose)
	{
		for (size_t i = 0; i < shapes.size(); i++)
		{
			printf("shape[%ld].name = %s\n", i, shapes[i].name.c_str());
			printf("shape[%ld].indices: %ld\n", i, shapes[i].mesh.indices.size());
			assert((shapes[i].mesh.indices.size() % 3) == 0);
			for (size_t f = 0; f < shapes[i].mesh.indices.size(); f++)
			{
				printf("  idx[%ld] = %d\n", f, shapes[i].mesh.indices[f]);
			}

			printf("shape[%ld].vertices: %ld\n", i, shapes[i].mesh.positions.size());
			assert((shapes[i].mesh.positions.size() % 3) == 0);
			for (size_t v = 0; v < shapes[i].mesh.positions.size() / 3; v++)
			{
				printf("  v[%ld] = (%f, %f, %f)\n", v,
					   shapes[i].mesh.positions[3 * v + 0],
					   shapes[i].mesh.positions[3 * v + 1],
					   shapes[i].mesh.positions[3 * v + 2]);
			}

			printf("shape[%ld].material.name = %s\n", i, shapes[i].material.name.c_str());
			printf("  material.Ka = (%f, %f ,%f)\n", shapes[i].material.ambient[0], shapes[i].material.ambient[1], shapes[i].material.ambient[2]);
			printf("  material.Kd = (%f, %f ,%f)\n", shapes[i].material.diffuse[0], shapes[i].material.diffuse[1], shapes[i].material.diffuse[2]);
			printf("  material.Ks = (%f, %f ,%f)\n", shapes[i].material.specular[0], shapes[i].material.specular[1], shapes[i].material.specular[2]);
			printf("  material.Tr = (%f, %f ,%f)\n", shapes[i].material.transmittance[0], shapes[i].material.transmittance[1], shapes[i].material.transmittance[2]);
			printf("  material.Ke = (%f, %f ,%f)\n", shapes[i].material.emission[0], shapes[i].material.emission[1], shapes[i].material.emission[2]);
			printf("  material.Ns = %f\n", shapes[i].material.shininess);
			printf("  material.map_Ka = %s\n", shapes[i].material.ambient_texname.c_str());
			printf("  material.map_Kd = %s\n", shapes[i].material.diffuse_texname.c_str());
			printf("  material.map_Ks = %s\n", shapes[i].material.specular_texname.c_str());
			printf("  material.map_Ns = %s\n", shapes[i].material.normal_texname.c_str());
			std::map<std::string, std::string>::iterator it(shapes[i].material.unknown_parameter.begin());
			std::map<std::string, std::string>::iterator itEnd(shapes[i].material.unknown_parameter.end());
			for (; it != itEnd; it++)
			{
				printf("  material.%s = %s\n", it->first.c_str(), it->second.c_str());
			}
			printf("\n");
		}
	}

	return true;
}

int main(int argc, char** argv)
{
	//	assert(true == TestLoadObj("cornell_box.obj",true));
	//	assert(true == TestLoadObj("cube.obj",true));
	assert(true == TestLoadObj("samurai_monastry.obj", false));
	assert(true == TestLoadObj("teddy2_VHACD_CHs.obj", true));
	return 0;
}