/*
   model/asefile.cc
   This file is part of the Osirion project and is distributed under
   the terms of the GNU General Public License version 2
*/

#include "sys/sys.h"
#include "model/asefile.h"

#include <sstream>
#include <cstring>

namespace model
{

// max geometry bounds
const float MAX_BOUNDS = 16384.0f;

ASEFile::ASEFile(std::string const &name)
{
	asefile_name.append(name);
	asefile_name.append(".ase");
	asefile_ifs.open(asefile_name);

	for (int i = 0; i < 3; i++) {
		ase_minbbox[i] = 0;
		ase_maxbbox[i] = 0;
	}

	// a single fragmentgroup wil contain all the model triangles
	ase_fragmentgroup = new FragmentGroup();
	ase_fragmentgroup->set_type(FragmentGroup::None);
	
	// reset counters
	ase_vertexcount = 0;
	ase_facecount = 0 ;
}

ASEFile::~ASEFile()
{
	for (VertexList::iterator it = ase_tvertexlist.begin(); it != ase_tvertexlist.end(); it++) {
		delete(*it).second;
	}
	ase_tvertexlist.clear();

	for (VertexList::iterator it = ase_vertexlist.begin(); it != ase_vertexlist.end(); it++) {
		delete(*it).second;
	}
	ase_vertexlist.clear();

	for (FaceList::iterator it = ase_facelist.begin(); it != ase_facelist.end(); it++) {
		delete(*it).second;
	}
	ase_facelist.clear();

	for (MaterialList::iterator it = ase_materials.begin(); it != ase_materials.end(); it++) {
		SubMaterialList *submaterials = (*it).second;
		
		for (SubMaterialList::iterator smit = submaterials->begin(); smit != submaterials->end(); smit++) {
			(*smit).second = 0;
		}
		submaterials->clear();
		delete submaterials;
		(*it).second = 0;		
	}
	ase_materials.clear();

	if (asefile_ifs.is_open())
		asefile_ifs.close();
}

bool ASEFile::read_header(std::istream &is)
{
	if (!is.good()) {
		return false;
	}

	char data[1024];
	memset(data, 0, sizeof(data));

	if (!is.getline(data, sizeof(data) - 1)) {
		return false;
	}

	std::istringstream line(data);
	std::string word;
	line >> word;

	if (!word.compare("*3DSMAX_ASCIIEXPORT") == 0) {
		return false;
	}

	return true;
}

Material *ASEFile::read_submaterial(std::istream &is)
{
	Material *material = 0;
	char data[1024];
	memset(data, 0, sizeof(data));
	int level = 1;

	while (is.getline(data, sizeof(data) - 1)) {

		std::istringstream line(data);
		std::string word;
		line >> word;

		if ((level == 1) && (word.compare("*MATERIAL_NAME") == 0)) {
			// read material name
			std::string n;
			char c;
			while ((line.get(c)) && (c != '"'));
			while ((line.get(c)) && (c != '"'))
				n += c;
			
			// find material
			material = Material::find(n);
			if (!material) {
				material = new Material(n);
				Material::add(material);
				material->set_texture(material->name());
			}
				
		} else {
			do {
				if (word.compare("{") == 0) {
					level++;
				} else if (word.compare("}") == 0) {
					level--;
				}
			} while (line >> word);
		}

		if (level == 0) {
			return material;
		}
	}

	return material;
}

ASEFile::SubMaterialList *ASEFile::read_material(std::istream &is)
{
	SubMaterialList *submaterials = new SubMaterialList;
	std::string materialname;
	
	char data[1024];
	memset(data, 0, sizeof(data));
	int level = 1;
	size_t index = 0;

	while (is.getline(data, sizeof(data) - 1)) {

		std::istringstream line(data);
		std::string word;
		line >> word;

		if ((level == 1) && (word.compare("*MATERIAL_NAME") == 0)) {
			// read material name
			std::string n;
			char c;
			while ((line.get(c)) && (c != '"'));
			while ((line.get(c)) && (c != '"'))
				n += c;

			materialname.assign(n);
			
		} else if ((level == 1) && (word.compare("*SUBMATERIAL") == 0)) {
				
			if ((line >> index) && (line >> word) && (word.compare("{") == 0)) {
				// this material defines submaterials, we ignore the parent definition
				materialname.clear();

				// add material to the submaterial list
				(*submaterials)[index] = read_submaterial(is);
			}
			
		} else {
			do {
				if (word.compare("{") == 0) {
					level++;
				} else if (word.compare("}") == 0) {
					level--;
				}
			} while (line >> word);
		}

		if (level == 0) {
			
			if (materialname.size()) {
				// no submaterials, add a single material to the submaterial list
				Material *material = Material::find(materialname);
				if (!material) {
					material = new Material(materialname);
					Material::add(material);
					material->set_texture(materialname);
				}
				(*submaterials)[0] = material;
			}
			
			return submaterials;
		}
	}

	return submaterials;
}

bool ASEFile::read_material_list(std::istream &is)
{
	char data[1024];
	memset(data, 0, sizeof(data));
	int level = 1;
	
	while (is.getline(data, sizeof(data) - 1)) {

		std::istringstream line(data);
		std::string word;
		size_t index;
		line >> word;

		if ((level == 1) && (word.compare("*MATERIAL") == 0)) {
			if ((line >> index) && (line >> word) && (word.compare("{") == 0)) {
				// add material to the ase material list
				ase_materials[index] = read_material(is);
				//con_debug << "  " << name() << " " << "*MATERIAL " << index << " " << ase_materials[index]->name() << std::endl;
			}

		} else {
			do {
				if (word.compare("{") == 0) {
					level++;
				} else if (word.compare("}") == 0) {
					level--;
				}
			} while (line >> word);
		}

		if (level == 0)
			return true;
	}

	return false;
}

bool ASEFile::read_mesh_vertex_list(std::istream &is)
{
	char data[1024];
	memset(data, 0, sizeof(data));

	while (is.getline(data, sizeof(data) - 1)) {
		std::istringstream line(data);

		std::string firstword;
		line >> firstword;

		if (firstword.compare("}") == 0) {
			return true;

		} else if (firstword.compare("*MESH_VERTEX") == 0) {
			size_t index;
			float x, y, z;
			if (line >> index >> x >> y >> z) {
				math::Vector3f *v = new math::Vector3f(x, y, z);
				ase_vertexlist[index] = v;

				for (size_t i = 0; i < 3; i++) {
					if ((*v)[i] > ase_maxbbox[i]) {
						ase_maxbbox[i] = (*v)[i];
					}
					if ((*v)[i] < ase_minbbox[i]) {
						ase_minbbox[i] = (*v)[i];
					}
				}
				ase_vertexcount++;
			}
		}
	}

	return false;
}

bool ASEFile::read_mesh_face_list(std::istream &is)
{
	char data[1024];
	memset(data, 0, sizeof(data));

	while (is.getline(data, sizeof(data) - 1)) {
		std::istringstream line(data);

		std::string word;
		line >> word;

		if (word.compare("}") == 0) {
			return true;

		} else if (word.compare("*MESH_FACE") == 0) {
			std::string facestr;
			size_t a, b, c;
			if ((line >> facestr) &&
					(line >> word) && (line >> a) &&
					(line >> word) && (line >> b) &&
					(line >> word) && (line >> c)) {
				
				size_t index;
			
				if (facestr.size() && facestr[facestr.size()-1] == ':') {
					facestr.erase(facestr.size() - 1);
				}
				
				std::istringstream faceindexstr(facestr);
				if (faceindexstr >> index) {
					Triangle *triangle = new Triangle(*ase_vertexlist[a], *ase_vertexlist[b], *ase_vertexlist[c]);
					ase_facelist[index] = triangle;

					while (line >> word) {
						if (word.compare("*MESH_MTLID") == 0) {
							size_t submaterial_index = 0;
							if (line >> submaterial_index) {
								triangle->set_material_index(submaterial_index);
							}
						}
					}
				}
				
			}
			ase_facecount++;
		}
	}

	return false;
}

bool ASEFile::read_mesh_normals(std::istream &is)
{
	char data[1024];
	memset(data, 0, sizeof(data));

	size_t count = 0;

	size_t index = 0;
	size_t vertindex = 0;
	float x, y, z;
	FaceList::iterator it;

	while (is.getline(data, sizeof(data) - 1)) {
		std::istringstream line(data);

		std::string firstword;
		line >> firstword;

		if (firstword.compare("}") == 0) {
			//con_debug << "    " << count << " face normals" << std::endl;
			return true;

		} else if (firstword.compare("*MESH_FACENORMAL") == 0) {
			if (line >> index >> x >> y >> z) {
				it = ase_facelist.find(index);
				if (it != ase_facelist.end()) {
					(*it).second->normal().assign(x, y, z);
					count++;
				} else {
					con_warn << "  could not find face " << index << std::endl;
				}
				vertindex = 0;
			} else {
				it = ase_facelist.end();
			}
		} else if (firstword.compare("*MESH_VERTEXNORMAL") == 0) {

			if ((it != ase_facelist.end()) && (line >> index >> x >> y >> z)) {

				if (vertindex == 0) {
					(*it).second->n0().assign(x, y, z);
				} else if (vertindex == 1) {
					(*it).second->n1().assign(x, y, z);
				} else if (vertindex == 2) {
					(*it).second->n2().assign(x, y, z);
				}

				vertindex++;
			}
		}
	}

	return false;
}
bool ASEFile::read_mesh_tvertex_list(std::istream &is)
{
	size_t count = 0;

	char data[1024];
	memset(data, 0, sizeof(data));

	while (is.getline(data, sizeof(data) - 1)) {
		std::istringstream line(data);

		std::string firstword;
		line >> firstword;

		if (firstword.compare("}") == 0) {
			//con_debug << "    " << count << " texture vertices" << std::endl;
			return true;

		} else if (firstword.compare("*MESH_TVERT") == 0) {
			size_t index;
			float x, y, z;
			if (line >> index >> x >> y >> z) {
				math::Vector3f *v = new math::Vector3f(x, y, z);
				ase_tvertexlist[index] = v;
				count++;
			}
		}
	}

	return false;
}

bool ASEFile::read_mesh_tface_list(std::istream &is)
{
	size_t count = 0;

	char data[1024];
	memset(data, 0, sizeof(data));

	while (is.getline(data, sizeof(data) - 1)) {
		std::istringstream line(data);

		std::string firstword;
		line >> firstword;

		if (firstword.compare("}") == 0) {
			//con_debug << "    " << count << " face texture coordinates" << std::endl;
			return true;

		} else if (firstword.compare("*MESH_TFACE") == 0) {
			// face texture coordinates
			size_t index, a, b, c;
			if (line >> index >> a >> b >> c) {
				Triangle *triangle = ase_facelist[index];
				math::Vector3f *t0 = ase_tvertexlist[a];
				math::Vector3f *t1 = ase_tvertexlist[b];
				math::Vector3f *t2 = ase_tvertexlist[c];
				if (triangle) {
					if (t0)
						triangle->t0().assign(t0->x(), 1.0f - t0->y());
					if (t1)
						triangle->t1().assign(t1->x(), 1.0f - t1->y());
					if (t2)
						triangle->t2().assign(t2->x(), 1.0f - t2->y());
				}
				count++;
			}
		}
	}

	return false;
}

bool ASEFile::read_mesh(std::istream &is)
{
	char data[1024];
	memset(data, 0, sizeof(data));
	int level = 1;

	ase_vertexlist.clear();

	while (is.getline(data, sizeof(data) - 1)) {
		std::istringstream line(data);

		std::string word;
		line >> word;

		if ((level == 1) && (word.compare("*MESH_VERTEX_LIST") == 0)) {
			if ((line >> word) && (word.compare("{") == 0)) {
				//con_debug << "  " << name() << " *MESH_VERTEX_LIST" << std::endl;
				read_mesh_vertex_list(is);
			}

		} else if ((level == 1) && (word.compare("*MESH_FACE_LIST") == 0)) {
			if ((line >> word) && (word.compare("{") == 0)) {
				//con_debug << "  " << name() << " *MESH_FACE_LIST" << std::endl;
				read_mesh_face_list(is);
			}

		} else if ((level == 1) && (word.compare("*MESH_NORMALS") == 0)) {
			if ((line >> word) && (word.compare("{") == 0)) {
				//con_debug << "  " << name() << " *MESH_NORMALS" << std::endl;
				read_mesh_normals(is);
			}

		} else if ((level == 1) && (word.compare("*MESH_TVERTLIST") == 0)) {
			if ((line >> word) && (word.compare("{") == 0)) {
				//con_debug << "  " << name() << " *MESH_TVERTLIST" << std::endl;
				read_mesh_tvertex_list(is);
			}

		} else if ((level == 1) && (word.compare("*MESH_TFACELIST") == 0)) {
			if ((line >> word) && (word.compare("{") == 0)) {
				//con_debug << "  " << name() << " *MESH_TFACELIST" << std::endl;
				read_mesh_tface_list(is);
			}

		} else {

			do {
				if (word.compare("{") == 0) {
					level++;
				} else if (word.compare("}") == 0) {
					level--;
				}
			} while (line >> word);
		}

		if (level == 0)
			return true;
	}

	return false;
}

bool ASEFile::clear_geom()
{
	// delete texture vertices
	for (VertexList::iterator it = ase_tvertexlist.begin(); it != ase_tvertexlist.end(); it++) {
		delete(*it).second;
	}
	ase_tvertexlist.clear();

	// delete geometry vertices
	for (VertexList::iterator it = ase_vertexlist.begin(); it != ase_vertexlist.end(); it++) {
		delete(*it).second;
	}
	ase_vertexlist.clear();

	// delete faces
	for (FaceList::iterator it = ase_facelist.begin(); it != ase_facelist.end(); it++) {
		delete(*it).second;
	}
	ase_facelist.clear();

	return true;
}

bool ASEFile::read_geom(std::istream &is)
{
	char data[1024];
	memset(data, 0, sizeof(data));
	int level = 1;
	SubMaterialList *submateriallist = 0;

	while (is.getline(data, sizeof(data) - 1)) {

		std::istringstream line(data);
		std::string word;
		size_t index;
		line >> word;

		if ((level == 1) && (word.compare("*MESH") == 0)) {
			if ((line >> word) && (word.compare("{") == 0)) {
				//con_debug << "  " << name() << " " << "*MESH" << std::endl;
				read_mesh(is);
			}
		} else if ((level == 1) && (word.compare("*MATERIAL_REF") == 0)) {
			// the material to use for this GEOMOBJECT
			if (line >> index) {
				//con_debug << "  " << name() << " " << "*MATERIAL_REF " << index << std::endl;
				
				// find the Material for index
				MaterialList::iterator it = ase_materials.find(index);
				if (it != ase_materials.end()) {
					submateriallist = (*it).second;
				} else {
					submateriallist = 0;
				}
			}
		} else {
			do {
				if (word.compare("{") == 0) {
					level++;
				} else if (word.compare("}") == 0) {
					level--;
				}
			} while (line >> word);
		}

		if (level == 0) {
			
			// import only if the material is defined and there actually are triangles to import
			if (submateriallist && ase_facelist.size()) {
				
				// iterate submaterial list, create a fragment per material
				for (SubMaterialList::iterator smit = submateriallist->begin(); smit != submateriallist->end(); smit++) {
					size_t submaterial_index = (*smit).first;
					Material *material = (*smit).second;

					if ((material->flags() & Material::Ignore) != Material::Ignore ) {
						
						// load GEOMOBJECT triangles with matching material into the fragment
						Fragment *fragment = new Fragment(Fragment::Triangles, material);
						for (FaceList::iterator it = ase_facelist.begin(); it != ase_facelist.end(); it++) {
							
							Triangle *triangle = (*it).second;
							if (triangle->material_index() == submaterial_index) {
								fragment->add_vertex((triangle->v0() * SCALE) , triangle->n0(), triangle->t0(), false);
								fragment->add_vertex((triangle->v1() * SCALE) , triangle->n1(), triangle->t1(), false);
								fragment->add_vertex((triangle->v2() * SCALE) , triangle->n2(), triangle->t2(), false);
							}
						}
						
						if (fragment->structural_size() + fragment->detail_size() > 0) {
							ase_fragmentgroup->add_fragment(fragment);
						} else {
							delete fragment;
						}
					}
				}
			}

			// clear geometry data
			clear_geom();
			return true;
		}
	}

	clear_geom();
	return false;
}

bool ASEFile::read()
{
	if (!read_header(asefile_ifs)) {
		con_warn << name() << ": not a valid ASE file!\n";
		return 0;
	}

	char data[1024];
	memset(data, 0, sizeof(data));

	while (asefile_ifs.getline(data, sizeof(data) - 1)) {
		std::istringstream line(data);

		std::string word;
		line >> word;

		if (word.compare("*GEOMOBJECT") == 0) {

			if ((line >> word) && (word.compare("{") == 0)) {
				//con_debug << "  " << name() << " " << "*GEOMOBJECT" << std::endl;
				read_geom(asefile_ifs);
			}
		} else if (word.compare("*MATERIAL_LIST") == 0) {

			if ((line >> word) && (word.compare("{") == 0)) {
				//con_debug << "  " << name() << " " << "*MATERIAL_LIST" << std::endl;
				read_material_list(asefile_ifs);
			}
		}
	}

	return true;
}

Model *ASEFile::load(const std::string &name)
{
	ASEFile asefile(name);

	if (!asefile.is_open()) {
		return 0;
	}

	if (!asefile.read()) {
		return 0;
	}

	if (!asefile.fragmentgroup()->size())
		return 0;
	
	// create a new model
	Model *model = new Model(name);

	// set bounding box properties
	math::Vector3f origin(0.0f, 0.0f, 0.0f);
	model->set_origin(origin);
	model->model_minbbox.assign (asefile.ase_minbbox * SCALE);
	model->model_maxbbox.assign (asefile.ase_maxbbox * SCALE);
	model->set_radius(math::max(model->model_minbbox.length(), model->model_maxbbox.length()));
	
	for (FragmentGroup::Fragments::const_iterator fit = asefile.fragmentgroup()->fragments().begin(); fit != asefile.fragmentgroup()->fragments().end(); fit++) {
		
		const Fragment *fragment = (*fit);
		model->model_tris_count += fragment->structural_size() + fragment->detail_size();
	}
	model->add_group(asefile.fragmentgroup());

	con_debug << "  " << asefile.name() << " " << asefile.vertexcount() << " vertices " << asefile.facecount() << " faces " << std::endl;
	
	return model;
}

} // namespace model