/*
	TE4 - T-Engine 4
	Copyright (C) 2009 - 2015 Nicolas Casalini

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program.  If not, see <http://www.gnu.org/licenses/>.

	Nicolas Casalini "DarkGod"
	darkgod@te4.org
*/

extern "C" {
#include "lua.h"
#include "lauxlib.h"
#include "display.h"
#include "types.h"
#include "physfs.h"
#include "physfsrwops.h"
#include "main.h"
}

#include "displayobjects/TextObject.hpp"

int DOText::addCharQuad(const char *str, size_t len, font_style style, int bx, int by, float r, float g, float b, float a) {
	int x = 0, y = by;
	ssize_t off = 1;
	int32_t c;
	float italic = 0;
	if (style == FONT_STYLE_ITALIC) { style = FONT_STYLE_NORMAL; italic = 0.2; }
	while (off > 0) {
		off = utf8proc_iterate((const uint8_t*)str, len, &c);
		str += off;
		len -= off;

		if (c > 0 && c < MAX_ATLAS_DATA) {
			font_atlas_data_style *d = &font->atlas_data[c].data[style];
			if (!d->w) font_add_atlas(font, c, style);
			if (d->w) {
				addQuad(
					d->w * italic + bx + x,		y,		d->tx1, d->ty1,
					d->w * italic + bx + x + d->w,	y,		d->tx2, d->ty1,
					bx + x + d->w,			y + d->h,	d->tx2, d->ty2,
					bx + x,				y + d->h,	d->tx1, d->ty2,
					r, g, b, a
				);
				x += d->w;
			}
		}
	}
	return x;
}

int DOText::getTextChunkSize(const char *str, size_t len, font_style style) {
	int x = 0, y = 0;
	ssize_t off = 1;
	int32_t c;
	float italic = 0;
	if (style == FONT_STYLE_ITALIC) { style = FONT_STYLE_NORMAL; italic = 0.2; }
	while (off > 0) {
		off = utf8proc_iterate((const uint8_t*)str, len, &c);
		str += off;
		len -= off;

		if (c > 0 && c < MAX_ATLAS_DATA) {
			font_atlas_data_style *d = &font->atlas_data[c].data[style];
			if (!d->w) font_add_atlas(font, c, style);
			if (d->w) {
				x += d->w;
			}
		}
	}
	return x;
}

void DOText::parseText() {
	font_type *f = font;
	if (!f) return;
	if (!f->atlas) font_make_atlas(f, 0, 0);
	size_t len = strlen(text);
	if (!len) return;
	const char *str = text;
	float r = 1, g = 1, b = 1, a = 1;
	float lr = 1, lg = 1, lb = 1, la = 1;
	int max_width = line_max_width;
	int bx = 0, by = 0;
	bool no_linefeed = lua_toboolean(L, 11);

	setTexture(f->atlas_tex, LUA_NOREF);
	clear();

	// Update VO size once, we are allocating a few more than neede in case of utf8 or control sequences, but we dont care
	vertices.reserve(len * 4);

	int font_h = TTF_FontHeight(f->font);
	int id_dduid = 1;
	int nb_lines = 1;
	int id_real_line = 1;
	char *line_data = NULL;
	int line_data_size = 0;
	char *start = (char*)str, *stop = (char*)str, *next = (char*)str;
	int max_size = 0;
	int size = 0;
	bool is_separator = false;
	int i;
	bool force_nl = false;
	font_style style = FONT_STYLE_NORMAL;
	while (true)
	{
		if ((*next == '\n') || (*next == ' ') || (*next == '\0') || (*next == '#'))
		{
			bool inced = false;
			if (*next == ' ' && *(next+1))
			{
				inced = true;
				stop = next;
				next++;
			}
			else stop = next - 1;

			// Make a surface for the word
			int len = next - start;
			int future_size = getTextChunkSize(start, len, style);

			// If we must do a newline, flush the previous word and the start the new line
			if (!no_linefeed && (force_nl || (future_size && max_width && (size + future_size > max_width))))
			{
				// Push it & reset the surface
				id_dduid = 1;
				is_separator = false;
//				printf("Ending previous line at size %d\n", size);
				if (size > max_size) max_size = size;
				size = 0;
				nb_lines++;
				if (force_nl)
				{
					id_real_line++;
					if (line_data) { line_data = NULL; }
				}
				force_nl = false;
			}

			if (len)
			{
				// Detect separators
				if ((*start == '-') && (*(start+1) == '-') && (*(start+2) == '-') && !(*(start+3))) is_separator = true;

//				printf("Drawing word '%s'\n", start);
				size += addCharQuad(start, len, style, bx + size, by + (nb_lines-1) * font_h, r, g, b, a);
			}
			if (inced) next--;
			start = next + 1;

			// Force a linefeed
			if (*next == '\n') force_nl = true;

			// Handle special codes
			else if (*next == '#')
			{
				char *codestop = next + 1;
				while (*codestop && *codestop != '#') codestop++;
				// Font style
				if (*(next+1) == '{') {
					if (*(next+2) == 'n') style = FONT_STYLE_NORMAL;
					else if (*(next+2) == 'b') style = FONT_STYLE_BOLD;
					else if (*(next+2) == 'i') style = FONT_STYLE_ITALIC;
					else if (*(next+2) == 'u') style = FONT_STYLE_UNDERLINED;
				}
				// Entity UID
				else if ((codestop - (next+1) > 4) && (*(next+1) == 'U') && (*(next+2) == 'I') && (*(next+3) == 'D') && (*(next+4) == ':')) {
// 					lua_getglobal(L, "__get_uid_entity");
// 					char *colon = next + 5;
// 					while (*colon && *colon != ':') colon++;
// 					lua_pushlstring(L, next+5, colon - (next+5));
// 					lua_call(L, 1, 1);
// 					if (lua_istable(L, -1))
// 					{
// //							printf("DirectDrawUID in font:draw %d : %d\n", size, h);
// 						lua_createtable(L, 0, 4);

// 						lua_pushliteral(L, "e");
// 						lua_pushvalue(L, -3);
// 						lua_rawset(L, -3);

// 						lua_pushliteral(L, "x");
// 						lua_pushnumber(L, size);
// 						lua_rawset(L, -3);

// 						lua_pushliteral(L, "w");
// 						lua_pushnumber(L, h);
// 						lua_rawset(L, -3);

// 						lua_rawseti(L, -4, id_dduid++); // __dduids

// 						size += h;
// 					}
// 					lua_pop(L, 1);
				}
				// Extra data
				else if (*(next+1) == '&') {
					line_data = next + 2;
					line_data_size = codestop - (next+2);
				}
				// Color
				else {
					if ((codestop - (next+1) == 4) && (*(next+1) == 'L') && (*(next+2) == 'A') && (*(next+3) == 'S') && (*(next+4) == 'T'))
					{
						r = lr;
						g = lg;
						b = lb;
						a = la;
					}

					lua_getglobal(L, "colors");
					lua_pushlstring(L, next+1, codestop - (next+1));
					lua_rawget(L, -2);
					if (lua_istable(L, -1)) {
						lr = r;
						lg = g;
						lb = b;
						la = a;

						lua_pushliteral(L, "r");
						lua_rawget(L, -2);
						r = lua_tonumber(L, -1) / 255;
						lua_pushliteral(L, "g");
						lua_rawget(L, -3);
						g = lua_tonumber(L, -1) / 255;
						lua_pushliteral(L, "b");
						lua_rawget(L, -4);
						b = lua_tonumber(L, -1) / 255;
						lua_pop(L, 3);
						a = 1;
					}
					// Hexacolor
					else if (codestop - (next+1) == 6)
					{
						lr = r;
						lg = g;
						lb = b;
						la = a;

						int rh = 0, gh = 0, bh = 0;

						if ((*(next+1) >= '0') && (*(next+1) <= '9')) rh += 16 * (*(next+1) - '0');
						else if ((*(next+1) >= 'a') && (*(next+1) <= 'f')) rh += 16 * (10 + *(next+1) - 'a');
						else if ((*(next+1) >= 'A') && (*(next+1) <= 'F')) rh += 16 * (10 + *(next+1) - 'A');
						if ((*(next+2) >= '0') && (*(next+2) <= '9')) rh += (*(next+2) - '0');
						else if ((*(next+2) >= 'a') && (*(next+2) <= 'f')) rh += (10 + *(next+2) - 'a');
						else if ((*(next+2) >= 'A') && (*(next+2) <= 'F')) rh += (10 + *(next+2) - 'A');

						if ((*(next+3) >= '0') && (*(next+3) <= '9')) gh += 16 * (*(next+3) - '0');
						else if ((*(next+3) >= 'a') && (*(next+3) <= 'f')) gh += 16 * (10 + *(next+3) - 'a');
						else if ((*(next+3) >= 'A') && (*(next+3) <= 'F')) gh += 16 * (10 + *(next+3) - 'A');
						if ((*(next+4) >= '0') && (*(next+4) <= '9')) gh += (*(next+4) - '0');
						else if ((*(next+4) >= 'a') && (*(next+4) <= 'f')) gh += (10 + *(next+4) - 'a');
						else if ((*(next+4) >= 'A') && (*(next+4) <= 'F')) gh += (10 + *(next+4) - 'A');

						if ((*(next+5) >= '0') && (*(next+5) <= '9')) bh += 16 * (*(next+5) - '0');
						else if ((*(next+5) >= 'a') && (*(next+5) <= 'f')) bh += 16 * (10 + *(next+5) - 'a');
						else if ((*(next+5) >= 'A') && (*(next+5) <= 'F')) bh += 16 * (10 + *(next+5) - 'A');
						if ((*(next+6) >= '0') && (*(next+6) <= '9')) bh += (*(next+6) - '0');
						else if ((*(next+6) >= 'a') && (*(next+6) <= 'f')) bh += (10 + *(next+6) - 'a');
						else if ((*(next+6) >= 'A') && (*(next+6) <= 'F')) bh += (10 + *(next+6) - 'A');

						r = (float)rh / 255;
						g = (float)gh / 255;
						b = (float)bh / 255;
						a = 1;
					}
					lua_pop(L, 2);
				}

				char old = *codestop;
				*codestop = '\0';
//				printf("Found code: %s\n", next+1);
				*codestop = old;

				start = codestop + 1;
				next = codestop; // The while will increment it, so we dont so it here
			}
		}
		if (*next == '\0') break;
		next++;
	}

	id_dduid = 1;
	if (size > max_size) max_size = size;

	this->nb_lines = nb_lines;
	this->w = max_size;
	this->h = nb_lines * font_h;
}

void DOText::setText(const char *text) {
	free((void*)this->text);
	this->text = strdup(text);
	parseText();
}