#include "global.h"

void export_obj_write_string(u8_array_t *out, char *str) {
	for (i32 i = 0; i < string_length(str); ++i) {
		u8_array_push(out, char_code_at(str, i));
	}
}

void export_obj_run(char *path, mesh_object_t_array_t *paint_objects, bool apply_disp) {
	u8_array_t *o = u8_array_create_from_raw((u8[]){}, 0);
	export_obj_write_string(o, "# armorpaint.org\n");

	i32 poff = 0;
	i32 noff = 0;
	i32 toff = 0;
	for (i32 i = 0; i < paint_objects->length; ++i) {
		mesh_object_t *p    = paint_objects->buffer[i];
		mesh_data_t   *mesh = p->data;
		f32            inv  = 1 / 32767.0;
		f32            sc   = p->data->scale_pos * inv;
		i16_array_t   *posa = mesh->vertex_arrays->buffer[0]->values;
		i16_array_t   *nora = mesh->vertex_arrays->buffer[1]->values;
		i16_array_t   *texa = mesh->vertex_arrays->buffer[2]->values;
		i32            len  = math_floor(posa->length / 4.0);

		// Merge shared vertices and remap indices
		i16_array_t *posa2  = i16_array_create(len * 3);
		i16_array_t *nora2  = i16_array_create(len * 3);
		i16_array_t *texa2  = i16_array_create(len * 2);
		i32_array_t *posmap = i32_array_create(len);
		i32_array_t *normap = i32_array_create(len);
		i32_array_t *texmap = i32_array_create(len);

		i32 pi = 0;
		i32 ni = 0;
		i32 ti = 0;
		for (i32 i = 0; i < len; ++i) {
			bool found = false;
			for (i32 j = 0; j < pi; ++j) {
				if (posa2->buffer[j * 3] == posa->buffer[i * 4] && posa2->buffer[j * 3 + 1] == posa->buffer[i * 4 + 1] &&
				    posa2->buffer[j * 3 + 2] == posa->buffer[i * 4 + 2]) {
					posmap->buffer[i] = j;
					found             = true;
					break;
				}
			}
			if (!found) {
				posmap->buffer[i]         = pi;
				posa2->buffer[pi * 3]     = posa->buffer[i * 4];
				posa2->buffer[pi * 3 + 1] = posa->buffer[i * 4 + 1];
				posa2->buffer[pi * 3 + 2] = posa->buffer[i * 4 + 2];
				pi++;
			}

			found = false;
			for (i32 j = 0; j < ni; ++j) {
				if (nora2->buffer[j * 3] == nora->buffer[i * 2] && nora2->buffer[j * 3 + 1] == nora->buffer[i * 2 + 1] &&
				    nora2->buffer[j * 3 + 2] == posa->buffer[i * 4 + 3]) {
					normap->buffer[i] = j;
					found             = true;
					break;
				}
			}
			if (!found) {
				normap->buffer[i]         = ni;
				nora2->buffer[ni * 3]     = nora->buffer[i * 2];
				nora2->buffer[ni * 3 + 1] = nora->buffer[i * 2 + 1];
				nora2->buffer[ni * 3 + 2] = posa->buffer[i * 4 + 3];
				ni++;
			}

			found = false;
			for (i32 j = 0; j < ti; ++j) {
				if (texa2->buffer[j * 2] == texa->buffer[i * 2] && texa2->buffer[j * 2 + 1] == texa->buffer[i * 2 + 1]) {
					texmap->buffer[i] = j;
					found             = true;
					break;
				}
			}
			if (!found) {
				texmap->buffer[i]         = ti;
				texa2->buffer[ti * 2]     = texa->buffer[i * 2];
				texa2->buffer[ti * 2 + 1] = texa->buffer[i * 2 + 1];
				ti++;
			}
		}
		if (apply_disp) {
			// let height: buffer_t = gpu_get_texture_pixels(layers[0].texpaint_pack);
			// let res: i32 = layers[0].texpaint_pack.width;
			// let strength: f32 = 0.1;
			// for (let i: i32 = 0; i < len; ++i) {
			// 	let x: i32 = math_floor(texa2[i * 2    ] / 32767 * res);
			// 	let y: i32 = math_floor((1.0 - texa2[i * 2 + 1] / 32767) * res);
			// 	let h: f32 = (1.0 - height.get((y * res + x) * 4 + 3) / 255) * strength;
			// 	posa2[i * 3    ] -= math_floor(nora2[i * 3    ] * inv * h / sc);
			// 	posa2[i * 3 + 1] -= math_floor(nora2[i * 3 + 1] * inv * h / sc);
			// 	posa2[i * 3 + 2] -= math_floor(nora2[i * 3 + 2] * inv * h / sc);
			// }
		}

		export_obj_write_string(o, string("o %s\n", p->base->name));
		for (i32 i = 0; i < pi; ++i) {
			export_obj_write_string(o, "v ");
			f32 f = posa2->buffer[i * 3] * sc;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = posa2->buffer[i * 3 + 2] * sc;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = -posa2->buffer[i * 3 + 1] * sc;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, "\n");
		}
		for (i32 i = 0; i < ni; ++i) {
			export_obj_write_string(o, "vn ");
			f32 f = nora2->buffer[i * 3] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = nora2->buffer[i * 3 + 2] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = -nora2->buffer[i * 3 + 1] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, "\n");
		}
		for (i32 i = 0; i < ti; ++i) {
			export_obj_write_string(o, "vt ");
			f32 f = texa2->buffer[i * 2] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = 1.0 - texa2->buffer[i * 2 + 1] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, "\n");
		}

		u32_array_t *inda = mesh->index_array;
		for (i32 i = 0; i < math_floor(inda->length / 3.0); ++i) {
			i32 pi1 = posmap->buffer[inda->buffer[i * 3]] + 1 + poff;
			i32 pi2 = posmap->buffer[inda->buffer[i * 3 + 1]] + 1 + poff;
			i32 pi3 = posmap->buffer[inda->buffer[i * 3 + 2]] + 1 + poff;
			i32 ni1 = normap->buffer[inda->buffer[i * 3]] + 1 + noff;
			i32 ni2 = normap->buffer[inda->buffer[i * 3 + 1]] + 1 + noff;
			i32 ni3 = normap->buffer[inda->buffer[i * 3 + 2]] + 1 + noff;
			i32 ti1 = texmap->buffer[inda->buffer[i * 3]] + 1 + toff;
			i32 ti2 = texmap->buffer[inda->buffer[i * 3 + 1]] + 1 + toff;
			i32 ti3 = texmap->buffer[inda->buffer[i * 3 + 2]] + 1 + toff;
			export_obj_write_string(o, "f ");
			export_obj_write_string(o, i32_to_string(pi1));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ti1));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ni1));
			export_obj_write_string(o, " ");
			export_obj_write_string(o, i32_to_string(pi2));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ti2));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ni2));
			export_obj_write_string(o, " ");
			export_obj_write_string(o, i32_to_string(pi3));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ti3));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ni3));
			export_obj_write_string(o, "\n");
		}
		poff += pi;
		noff += ni;
		toff += ti;
	}

	if (!ends_with(path, ".obj")) {
		path = string("%s.obj", path);
	}
	iron_file_save_bytes(path, o, 0);
}

void export_obj_run_fast(char *path, mesh_object_t_array_t *paint_objects) {
	// Skips merging shared vertices

	u8_array_t *o = u8_array_create_from_raw((u8[]){}, 0);
	export_obj_write_string(o, "# armorpaint.org\n");

	i32 poff = 0;
	i32 noff = 0;
	i32 toff = 0;
	for (i32 i = 0; i < paint_objects->length; ++i) {
		mesh_object_t *p    = paint_objects->buffer[i];
		mesh_data_t   *mesh = p->data;
		f32            inv  = 1 / 32767.0;
		f32            sc   = p->data->scale_pos * inv;
		i16_array_t   *posa = mesh->vertex_arrays->buffer[0]->values;
		i16_array_t   *nora = mesh->vertex_arrays->buffer[1]->values;
		i16_array_t   *texa = mesh->vertex_arrays->buffer[2]->values;

		i32 pi = posa->length / 4.0;
		i32 ni = pi;
		i32 ti = pi;

		export_obj_write_string(o, string("o %s\n", p->base->name));
		for (i32 i = 0; i < pi; ++i) {
			export_obj_write_string(o, "v ");
			f32 f = posa->buffer[i * 4] * sc;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = posa->buffer[i * 4 + 2] * sc;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = -posa->buffer[i * 4 + 1] * sc;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, "\n");
		}
		for (i32 i = 0; i < ni; ++i) {
			export_obj_write_string(o, "vn ");
			f32 f = nora->buffer[i * 2] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = posa->buffer[i * 4 + 3] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = -nora->buffer[i * 2 + 1] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, "\n");
		}
		for (i32 i = 0; i < ti; ++i) {
			export_obj_write_string(o, "vt ");
			f32 f = texa->buffer[i * 2] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, " ");
			f = 1.0 - texa->buffer[i * 2 + 1] * inv;
			export_obj_write_string(o, f32_to_string(f));
			export_obj_write_string(o, "\n");
		}

		u32_array_t *inda = mesh->index_array;
		for (i32 i = 0; i < math_floor(inda->length / 3.0); ++i) {
			i32 pi1 = inda->buffer[i * 3] + 1 + poff;
			i32 pi2 = inda->buffer[i * 3 + 1] + 1 + poff;
			i32 pi3 = inda->buffer[i * 3 + 2] + 1 + poff;
			i32 ni1 = inda->buffer[i * 3] + 1 + noff;
			i32 ni2 = inda->buffer[i * 3 + 1] + 1 + noff;
			i32 ni3 = inda->buffer[i * 3 + 2] + 1 + noff;
			i32 ti1 = inda->buffer[i * 3] + 1 + toff;
			i32 ti2 = inda->buffer[i * 3 + 1] + 1 + toff;
			i32 ti3 = inda->buffer[i * 3 + 2] + 1 + toff;
			export_obj_write_string(o, "f ");
			export_obj_write_string(o, i32_to_string(pi1));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ti1));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ni1));
			export_obj_write_string(o, " ");
			export_obj_write_string(o, i32_to_string(pi2));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ti2));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ni2));
			export_obj_write_string(o, " ");
			export_obj_write_string(o, i32_to_string(pi3));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ti3));
			export_obj_write_string(o, "/");
			export_obj_write_string(o, i32_to_string(ni3));
			export_obj_write_string(o, "\n");
		}
		poff += pi;
		noff += ni;
		toff += ti;
	}

	if (!ends_with(path, ".obj")) {
		path = string("%s.obj", path);
	}
	iron_file_save_bytes(path, o, 0);
}

void export_obj_run_sculpt(char *path, mesh_object_t_array_t *paint_objects) {
	slot_layer_t *l = NULL;
	for (i32 i = 0; i < project_layers->length; ++i) {
		if (project_layers->buffer[i]->texpaint_sculpt != NULL) {
			l = project_layers->buffer[i];
			break;
		}
	}
	if (l == NULL) {
		return;
	}

	u8_array_t *o = u8_array_create_from_raw((u8[]){}, 0);
	export_obj_write_string(o, "# armorpaint.org\n");

	mesh_object_t *p    = paint_objects->buffer[0];
	mesh_data_t   *mesh = p->data;
	f32            sc   = mesh->scale_pos;
	i16_array_t   *texa = mesh->vertex_arrays->buffer[2]->values;
	i32            len  = math_floor(mesh->index_array->length);
	i32            tris = math_floor(len / 3.0);
	f32            inv  = 1.0 / 32767.0;

	buffer_t *pixels = gpu_get_texture_pixels(l->texpaint_sculpt);

	export_obj_write_string(o, string("o %s\n", p->base->name));

	for (i32 i = 0; i < len; ++i) {
		f32 x = buffer_get_f32(pixels, i * 16)     * sc;
		f32 y = buffer_get_f32(pixels, i * 16 + 4) * sc;
		f32 z = buffer_get_f32(pixels, i * 16 + 8) * sc;
		export_obj_write_string(o, "v ");
		export_obj_write_string(o, f32_to_string(x));
		export_obj_write_string(o, " ");
		export_obj_write_string(o, f32_to_string(z));
		export_obj_write_string(o, " ");
		export_obj_write_string(o, f32_to_string(-y));
		export_obj_write_string(o, "\n");
	}

	for (i32 t = 0; t < tris; ++t) {
		i32 i0 = t * 3, i1 = t * 3 + 1, i2 = t * 3 + 2;
		f32 x0 = buffer_get_f32(pixels, i0 * 16)     * sc;
		f32 y0 = buffer_get_f32(pixels, i0 * 16 + 4) * sc;
		f32 z0 = buffer_get_f32(pixels, i0 * 16 + 8) * sc;
		f32 x1 = buffer_get_f32(pixels, i1 * 16)     * sc;
		f32 y1 = buffer_get_f32(pixels, i1 * 16 + 4) * sc;
		f32 z1 = buffer_get_f32(pixels, i1 * 16 + 8) * sc;
		f32 x2 = buffer_get_f32(pixels, i2 * 16)     * sc;
		f32 y2 = buffer_get_f32(pixels, i2 * 16 + 4) * sc;
		f32 z2 = buffer_get_f32(pixels, i2 * 16 + 8) * sc;
		f32 e1x = x1 - x0, e1y = y1 - y0, e1z = z1 - z0;
		f32 e2x = x2 - x0, e2y = y2 - y0, e2z = z2 - z0;
		f32 nx  = e1y * e2z - e1z * e2y;
		f32 ny  = e1z * e2x - e1x * e2z;
		f32 nz  = e1x * e2y - e1y * e2x;
		f32 nl  = math_sqrt(nx * nx + ny * ny + nz * nz);
		if (nl > 0.0) { nx /= nl; ny /= nl; nz /= nl; }
		export_obj_write_string(o, "vn ");
		export_obj_write_string(o, f32_to_string(nx));
		export_obj_write_string(o, " ");
		export_obj_write_string(o, f32_to_string(nz));
		export_obj_write_string(o, " ");
		export_obj_write_string(o, f32_to_string(-ny));
		export_obj_write_string(o, "\n");
	}

	for (i32 i = 0; i < len; ++i) {
		f32 u = texa->buffer[i * 2]     * inv;
		f32 v = 1.0 - texa->buffer[i * 2 + 1] * inv;
		export_obj_write_string(o, "vt ");
		export_obj_write_string(o, f32_to_string(u));
		export_obj_write_string(o, " ");
		export_obj_write_string(o, f32_to_string(v));
		export_obj_write_string(o, "\n");
	}

	for (i32 t = 0; t < tris; ++t) {
		i32 b = t * 3 + 1;
		export_obj_write_string(o, "f ");
		export_obj_write_string(o, i32_to_string(b));
		export_obj_write_string(o, "/");
		export_obj_write_string(o, i32_to_string(b));
		export_obj_write_string(o, "/");
		export_obj_write_string(o, i32_to_string(t + 1));
		export_obj_write_string(o, " ");
		export_obj_write_string(o, i32_to_string(b + 1));
		export_obj_write_string(o, "/");
		export_obj_write_string(o, i32_to_string(b + 1));
		export_obj_write_string(o, "/");
		export_obj_write_string(o, i32_to_string(t + 1));
		export_obj_write_string(o, " ");
		export_obj_write_string(o, i32_to_string(b + 2));
		export_obj_write_string(o, "/");
		export_obj_write_string(o, i32_to_string(b + 2));
		export_obj_write_string(o, "/");
		export_obj_write_string(o, i32_to_string(t + 1));
		export_obj_write_string(o, "\n");
	}

	if (!ends_with(path, ".obj")) {
		path = string("%s.obj", path);
	}
	iron_file_save_bytes(path, o, 0);
}