/*
    KindlePDFViewer: FreeType font rastering for UI
    Copyright (C) 2011 Hans-Werner Hilse <hilse@web.de>

    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/>.
*/
#include <string.h>
//#include <stdio.h>
#include <stdint.h>
#include <ft2build.h>
#include FT_FREETYPE_H
#include "blitbuffer.h"

/* for font access: */
#include <fitz/fitz.h>
#include <pdf/mupdf.h>

#include "ft.h"

FT_Library freetypelib;

static int newFace(lua_State *L) {
	const char *filename = luaL_checkstring(L, 1);
	int pxsize = luaL_optint(L, 2, 16*64);

	FT_Face *face = (FT_Face*) lua_newuserdata(L, sizeof(FT_Face));
	luaL_getmetatable(L, "ft_face");
	lua_setmetatable(L, -2);

	FT_Error error = FT_New_Face(freetypelib, filename, 0, face);
	if(error) {
		return luaL_error(L, "freetype error");
	}

	error = FT_Set_Pixel_Sizes(*face, 0, pxsize);
	if(error) {
		error = FT_Done_Face(*face);
		return luaL_error(L, "freetype error");
	}

	if((*face)->charmap == NULL) {
		//TODO
		//fprintf(stderr, "no unicode charmap found, to be implemented.\n");
	}
	return 1;
}

static int newBuiltinFace(lua_State *L) {
	const char *fontname = luaL_checkstring(L, 1);
	int pxsize = luaL_optint(L, 2, 16*64);
	char *fontdata = NULL;

	unsigned int size;
	/* we use compiled-in font data from mupdf build */
	if(!strcmp("mono", fontname)) {
		fontdata = pdf_find_substitute_font(1, 0, 0, 0, &size);
	} else if(!strcmp("sans", fontname)) {
		fontdata = pdf_find_substitute_font(0, 0, 0, 0, &size);
	} else if(!strcmp("cjk", fontname)) {
		fontdata = pdf_find_substitute_cjk_font(0, 0, &size);
	} else {
		fontdata = pdf_find_builtin_font(fontname, &size);
	}
	if(fontdata == NULL) {
		return luaL_error(L, "no such built-in font");
	}

	FT_Face *face = (FT_Face*) lua_newuserdata(L, sizeof(FT_Face));
	luaL_getmetatable(L, "ft_face");
	lua_setmetatable(L, -2);

	FT_Error error = FT_New_Memory_Face(freetypelib, (FT_Byte*)fontdata, size, 0, face);
	if(error) {
		return luaL_error(L, "freetype error");
	}

	error = FT_Set_Pixel_Sizes(*face, 0, pxsize);
	if(error) {
		error = FT_Done_Face(*face);
		return luaL_error(L, "freetype error");
	}

	if((*face)->charmap == NULL) {
		//TODO
		//fprintf(stderr, "no unicode charmap found, to be implemented.\n");
	}
	return 1;
}


static int renderGlyph(lua_State *L) {
	FT_Face *face = (FT_Face*) luaL_checkudata(L, 1, "ft_face");
	int ch = luaL_checkint(L, 2);
	FT_Error error = FT_Load_Char(*face, ch, FT_LOAD_RENDER);
	if(error) {
		return luaL_error(L, "freetype error");
	}

	int w = (*face)->glyph->bitmap.width;
	int h = (*face)->glyph->bitmap.rows;

	lua_newtable(L);

	BlitBuffer *bb;
	int result = newBlitBufferNative(L, w, h, &bb);
	if(result != 1) {
		return result;
	}

	lua_setfield(L, -2, "bb");

	uint8_t *dst = bb->data;
	int y;
	int x;
	for(y = 0; y < h; y++) {
		uint8_t *src = (*face)->glyph->bitmap.buffer + y * (*face)->glyph->bitmap.pitch;
		for(x = 0; x < (w/2); x++) {
			*dst = (src[0] & 0xF0) | (src[1] >> 4);
			src+=2;
			dst++;
		}
		if(w & 1) {
			*dst = *src & 0xF0;
			dst++;
		}
	}

	lua_pushinteger(L, (*face)->glyph->bitmap_left);
	lua_setfield(L, -2, "l");
	lua_pushinteger(L, (*face)->glyph->bitmap_top);
	lua_setfield(L, -2, "t");
	lua_pushinteger(L, (*face)->glyph->advance.x >> 6);
	lua_setfield(L, -2, "ax");
	lua_pushinteger(L, (*face)->glyph->advance.y >> 6);
	lua_setfield(L, -2, "ay");

	return 1;
}

static int hasKerning(lua_State *L) {
	FT_Face *face = (FT_Face*) luaL_checkudata(L, 1, "ft_face");
	if(FT_HAS_KERNING((*face))) {
		lua_pushinteger(L, 1);
	} else {
		lua_pushinteger(L, 0);
	}
	return 1;
}

static int getKerning(lua_State *L) {
	FT_Face *face = (FT_Face*) luaL_checkudata(L, 1, "ft_face");
	int left = FT_Get_Char_Index(*face, luaL_checkint(L, 2));
	int right = FT_Get_Char_Index(*face, luaL_checkint(L, 3));
	FT_Vector kerning;
	FT_Error error = FT_Get_Kerning(*face, left, right, FT_KERNING_DEFAULT, &kerning);
	if(error) {
		return luaL_error(L, "freetype error when getting kerning (l=%d, r=%d)", left, right);
	}
	lua_pushinteger(L, kerning.x >> 6);
	return 1;
}

static int getHeight(lua_State *L) {
	FT_Face *face = (FT_Face*) luaL_checkudata(L, 1, "ft_face");

	double pixels_height,pixels_ascender;
	double em_size, y_scale;

	/* compute floating point scale factors */
	em_size = 1.0 * (*face)->units_per_EM;
	y_scale = (*face)->size->metrics.y_ppem / em_size;

	/* convert design distances to floating point pixels */
	pixels_height = (*face)->height * y_scale;
	pixels_ascender = (*face)->ascender * y_scale;

	lua_pushnumber(L, pixels_height);
	lua_pushnumber(L, pixels_ascender);
	return 2;
}

static int doneFace(lua_State *L) {
	FT_Face *face = (FT_Face*) luaL_checkudata(L, 1, "ft_face");
	if(*face != NULL) {
		FT_Error error = FT_Done_Face(*face);
		if(error) {
			return luaL_error(L, "freetype error when freeing face");
		}
		*face = NULL;
	}
	return 0;
}

static const struct luaL_reg ft_face_meth[] = {
	{"renderGlyph", renderGlyph},
	{"hasKerning", hasKerning},
	{"getKerning", getKerning},
	{"getHeight", getHeight},
	{"done", doneFace},
	{"__gc", doneFace},
	{NULL, NULL}
};

static const struct luaL_reg ft_func[] = {
	{"newFace", newFace},
	{"newBuiltinFace", newBuiltinFace},
	{NULL, NULL}
};

int luaopen_ft(lua_State *L) {
	int error = FT_Init_FreeType(&freetypelib);
	if(error) {
		return luaL_error(L, "freetype error on initialization");
	}

	luaL_newmetatable(L, "ft_face");
	lua_pushstring(L, "__index");
	lua_pushvalue(L, -2);
	lua_settable(L, -3);
	luaL_register(L, NULL, ft_face_meth);

	luaL_register(L, "freetype", ft_func);
	return 1;
}