// MIT License, Copyright (c) 2020 Marvin Borner
// Some GFX functions
// TODO: Better support for bpp < 32
// TODO: Use efficient redrawing

#include <assert.h>
#include <bmp.h>
#include <gfx.h>
#include <mem.h>
#include <msg.h>
#include <png.h>
#include <psf.h>
#include <str.h>
#include <sys.h>
#include <vesa.h>

// TODO: Move to some global config file
#define FONT_COUNT 6
#define FONT_8_PATH "/font/spleen-5x8.psfu"
#define FONT_12_PATH "/font/spleen-6x12.psfu"
#define FONT_16_PATH "/font/spleen-8x16.psfu"
#define FONT_24_PATH "/font/spleen-12x24.psfu"
#define FONT_32_PATH "/font/spleen-16x32.psfu"
#define FONT_64_PATH "/font/spleen-32x64.psfu"

struct font *fonts[FONT_COUNT];

static void load_font(enum font_type font_type)
{
	if (fonts[font_type])
		return;

	const char *path = NULL;

	switch (font_type) {
	case FONT_8:
		path = FONT_8_PATH;
		break;
	case FONT_12:
		path = FONT_12_PATH;
		break;
	case FONT_16:
		path = FONT_16_PATH;
		break;
	case FONT_24:
		path = FONT_24_PATH;
		break;
	case FONT_32:
		path = FONT_32_PATH;
		break;
	case FONT_64:
		path = FONT_64_PATH;
		break;
	default:
		return;
	}

	fonts[font_type] = psf_parse(sread(path));
	assert(fonts[font_type]);
}

static void write_char(struct context *ctx, int x, int y, struct font *font, u32 c, char ch)
{
	int bypp = ctx->bpp >> 3;

	int pos = x * bypp + y * ctx->pitch;
	char *draw = (char *)&ctx->fb[pos];

	u32 stride = font->char_size / font->height;
	for (int cy = 0; cy < font->height; cy++) {
		for (int cx = 0; cx < font->width; cx++) {
			u8 bits = font->chars[ch * font->char_size + cy * stride + cx / 8];
			u8 bit = bits >> (7 - cx % 8) & 1;
			if (bit) {
				draw[bypp * cx] = GET_BLUE(c);
				draw[bypp * cx + 1] = GET_GREEN(c);
				draw[bypp * cx + 2] = GET_RED(c);
				draw[bypp * cx + 3] = GET_ALPHA(c);
			}
		}
		draw += ctx->pitch;
	}
}

static void draw_rectangle(struct context *ctx, int x1, int y1, int x2, int y2, u32 c)
{
	int bypp = ctx->bpp >> 3;
	u8 *draw = &ctx->fb[x1 * bypp + y1 * ctx->pitch];
	for (int i = 0; i < y2 - y1; i++) {
		for (int j = 0; j < x2 - x1; j++) {
			draw[bypp * j] = GET_BLUE(c);
			draw[bypp * j + 1] = GET_GREEN(c);
			draw[bypp * j + 2] = GET_RED(c);
			draw[bypp * j + 3] = GET_ALPHA(c);
		}
		draw += ctx->pitch;
	}
}

struct context *gfx_new_ctx(struct context *ctx)
{
	struct message msg = { 0 };
	msg_send(pidof(WM_PATH), GFX_NEW_CONTEXT, ctx);
	msg_receive(&msg);
	memcpy(ctx, msg.data, sizeof(*ctx));
	return ctx;
}

// On-demand font loading
struct font *gfx_resolve_font(enum font_type font_type)
{
	if (!fonts[font_type])
		load_font(font_type);
	return fonts[font_type];
}

void gfx_write_char(struct context *ctx, int x, int y, enum font_type font_type, u32 c, char ch)
{
	struct font *font = gfx_resolve_font(font_type);
	write_char(ctx, x, y, font, c, ch);
	/* gfx_redraw(); */
}

void gfx_write(struct context *ctx, int x, int y, enum font_type font_type, u32 c, const char *text)
{
	struct font *font = gfx_resolve_font(font_type);
	u32 cnt = 0;
	for (u32 i = 0; i < strlen(text); i++) {
		// TODO: Should this be here?
		if (text[i] == '\r') {
			cnt = 0;
		} else if (text[i] == '\n') {
			cnt = 0;
			y += font->height;
		} else if (text[i] == '\t') {
			cnt += 4;
		} else {
			// TODO: Overflow on single line input
			if ((cnt + 1) * font->width > ctx->width) {
				cnt = 0;
				y += font->height;
			}
			write_char(ctx, x + cnt * font->width, y, font, c, text[i]);
			cnt++;
		}
	}
	/* gfx_redraw(); */
}

void gfx_load_image(struct context *ctx, const char *path, int x, int y)
{
	// TODO: Support x, y
	// TODO: Detect image type
	// struct bmp *bmp = bmp_load(path);
	struct bmp *bmp = png_load(path);
	assert(bmp && bmp->width + x <= ctx->width);
	assert(bmp && bmp->height + y <= ctx->height);

	// TODO: Fix reversed png in decoder
	int bypp = bmp->bpp >> 3;
	// u8 *srcfb = &bmp->data[bypp + (bmp->height - 1) * bmp->pitch];
	u8 *srcfb = bmp->data;
	u8 *destfb = &ctx->fb[bypp];
	for (u32 cy = 0; cy < bmp->height; cy++) {
		memcpy(destfb, srcfb, bmp->pitch);
		// srcfb -= bmp->pitch;
		srcfb += bmp->pitch;
		destfb += ctx->pitch;
	}
	/* gfx_redraw(); */
}

void gfx_load_wallpaper(struct context *ctx, const char *path)
{
	gfx_load_image(ctx, path, 0, 0);
}

void gfx_copy(struct context *dest, struct context *src, int x, int y, u32 width, u32 height)
{
	int bypp = dest->bpp >> 3;
	u8 *srcfb = &src->fb[x * bypp + y * src->pitch];
	u8 *destfb = &dest->fb[x * bypp + y * dest->pitch];
	for (u32 cy = 0; cy < height; cy++) {
		memcpy(destfb, srcfb, width * bypp);
		srcfb += src->pitch;
		destfb += dest->pitch;
	}
}

// TODO: Support alpha values other than 0x0 and 0xff (blending)
// TODO: Optimize!
void gfx_ctx_on_ctx(struct context *dest, struct context *src, int x, int y)
{
	if (src->width == dest->width && src->height == dest->height && src->x == 0 &&
	    dest->y == 0) {
		memcpy(dest->fb, src->fb, dest->pitch * dest->height);
		return;
	}

	if (src->width > dest->width || src->height > dest->height)
		return;

	// TODO: Negative x and y
	int bypp = dest->bpp >> 3;
	u8 *srcfb = src->fb;
	u8 *destfb = &dest->fb[x * bypp + y * dest->pitch];
	for (u32 cy = 0; cy < src->height && cy + y < dest->height; cy++) {
		int diff = 0;
		for (u32 cx = 0; cx < src->width && cx + x < dest->width; cx++) {
			if (srcfb[3]) {
				destfb[0] = srcfb[0];
				destfb[1] = srcfb[1];
				destfb[2] = srcfb[2];
				destfb[3] = srcfb[3];
			}

			srcfb += bypp;
			destfb += bypp;
			diff += bypp;
		}
		srcfb += src->pitch - diff;
		destfb += dest->pitch - diff;
	}
}

void gfx_draw_rectangle(struct context *ctx, int x1, int y1, int x2, int y2, u32 c)
{
	draw_rectangle(ctx, x1, y1, x2, y2, c);
	/* gfx_redraw(); */
}

void gfx_fill(struct context *ctx, u32 c)
{
	draw_rectangle(ctx, 0, 0, ctx->width, ctx->height, c);
	/* gfx_redraw(); */
}

void gfx_border(struct context *ctx, u32 c, u32 width)
{
	if (width <= 0)
		return;

	int bypp = ctx->bpp >> 3;
	u8 *draw = ctx->fb;
	for (u32 i = 0; i < ctx->height; i++) {
		for (u32 j = 0; j < ctx->width; j++) {
			if (j <= width - 1 || i <= width - 1 ||
			    j - ctx->width + width + 1 <= width ||
			    i - ctx->height + width <= width) {
				draw[bypp * j + 0] = GET_BLUE(c);
				draw[bypp * j + 1] = GET_GREEN(c);
				draw[bypp * j + 2] = GET_RED(c);
				draw[bypp * j + 3] = GET_ALPHA(c);
			}
		}
		draw += ctx->pitch;
	}
	/* gfx_redraw(); */
}

int gfx_font_height(enum font_type font_type)
{
	struct font *font = gfx_resolve_font(font_type);
	return font->height;
}

int gfx_font_width(enum font_type font_type)
{
	struct font *font = gfx_resolve_font(font_type);
	return font->width;
}