/*
 *  2ooM: The Master of Orion II Reverse Engineering Project
 *  Graphical tool for inspecting LBX archives.
 *  Copyright (C) 2010 Nick Bowler
 *
 *  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 <config.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

#include <gtk/gtk.h>

#include "lbx.h"
#include "image.h"
#include "render.h"

#include "bg.xbm"

extern char lbxgui_xml[];

static GtkTreeStore *archives;
static GtkBuilder *builder;

static GtkWidget *canvas;
static GdkGC     *bg_gc;

static LBXimg *image;

static GdkPixbuf *framebuf;

void play_toggled(GtkToggleButton *button, gpointer data)
{
	gboolean active = gtk_toggle_button_get_active(button);
	GtkWidget *spin;

	if (active && !image) {
		gtk_toggle_button_set_active(button, FALSE);
		return;
	}

	spin = GTK_WIDGET(gtk_builder_get_object(builder, "framespin"));
	gtk_widget_set_sensitive(spin, !active);

	if (active) {
		gtk_spin_button_set_value(GTK_SPIN_BUTTON(spin), 0);
	}
}

void set_frame(GtkSpinButton *spin, gpointer data)
{
	unsigned frame = gtk_spin_button_get_value_as_int(spin);

	if (image)
		render_to_pixbuf(image, framebuf, frame);
	gdk_window_invalidate_rect(canvas->window, NULL, FALSE);
}

static void redraw_image(void)
{
	GtkSpinButton *spin;

	spin = GTK_SPIN_BUTTON(gtk_builder_get_object(builder, "framespin"));
	set_frame(spin, NULL);
}

static void tick(void *p, double delta)
{
	static double elapsed = 0;
	double seconds_per_frame = 1.0/15;
	struct lbx_imginfo info;
	GtkSpinButton   *spin;
	GtkToggleButton *play;
	unsigned frame;

	if (!image)
		return;

	lbximg_getinfo(image, &info);
	spin = GTK_SPIN_BUTTON(gtk_builder_get_object(builder, "framespin"));
	play = GTK_TOGGLE_BUTTON(gtk_builder_get_object(builder, "playbutton"));

	frame = gtk_spin_button_get_value_as_int(spin);
	if (!gtk_toggle_button_get_active(play))
		return;

	elapsed += delta;
	while (elapsed > seconds_per_frame) {
		elapsed -= seconds_per_frame;

		if (++frame >= info.nframes) {
			if (!info.looping) {
				gtk_toggle_button_set_active(play, FALSE);
				break;
			}

			frame = info.loopstart;
		}

		gtk_spin_button_set_value(spin, frame);
	}
}

static gboolean timeout(gpointer data)
{
	static double lasttime = INFINITY;
	static GTimer *timer;

	if (isinf(lasttime)) {
		timer = g_timer_new();
		if (timer) {
			lasttime = 0;
		}
	} else {
		double time = g_timer_elapsed(timer, NULL);

		tick(data, time - lasttime);
		lasttime = time;
	}

	return TRUE;
}

gboolean canvas_expose(GtkWidget *canvas, GdkEventExpose *event, gpointer data)
{
	if (!framebuf)
		return FALSE;

	gdk_draw_rectangle(canvas->window, bg_gc, TRUE,
		event->area.x, event->area.y,
		event->area.width, event->area.height);

	gdk_draw_pixbuf(canvas->window, NULL, framebuf,
		event->area.x, event->area.y, event->area.x, event->area.y,
		event->area.width, event->area.height,
		GDK_RGB_DITHER_NORMAL, 0, 0);

	return TRUE;
}

static int alloc_framebuffer(LBXimg *image)
{
	struct lbx_imginfo info;
	GtkSpinButton *spin;

	if (framebuf)
		g_object_unref(framebuf);

	lbximg_getinfo(image, &info);
	framebuf = gdk_pixbuf_new(GDK_COLORSPACE_RGB, TRUE, 8,
		info.width, info.height);
	g_return_val_if_fail(framebuf, -1);

	spin = GTK_SPIN_BUTTON(gtk_builder_get_object(builder, "framespin"));
	gtk_spin_button_set_range(spin, 0, info.nframes-1);
	gtk_spin_button_set_value(spin, 0);

	gtk_widget_set_size_request(canvas, info.width, info.height);
	return 0;
}

static int close_image(void *handle)
{
	lbx_file_close(handle);
	return 0;
}

static LBXimg *load_lbx_image(LBX *archive, unsigned index)
{
	LBXfile *file;
	LBXimg *image;

	file = lbx_file_open(archive, index);
	g_return_val_if_fail(file, NULL);

	image = lbximg_open(file, &lbx_arch_fops, close_image);
	if (!image)
		lbx_file_close(file);
	return image;
}

void set_override(GtkComboBox *combo)
{
	GtkTreeIter iter;
	gpointer lbx;
	guint    index;
	LBXimg *img;

	memset(palette_override, 0, sizeof palette_override);
	if (!gtk_combo_box_get_active_iter(combo, &iter))
		return;

	gtk_tree_model_get(GTK_TREE_MODEL(archives), &iter,
		1, &lbx,
		2, &index,
		-1);

	img = load_lbx_image(lbx, index);
	if (img) {
		lbximg_getpalette(img, palette_override);
		lbximg_close(img);
		redraw_image();
	}
}

void set_palette(GtkComboBox *combo)
{
	GtkTreeIter iter;
	gpointer lbx;
	guint    index;
	LBXfile  *f;

	memset(palette_external, 0, sizeof palette_external);
	if (!gtk_combo_box_get_active_iter(combo, &iter))
		return;

	gtk_tree_model_get(GTK_TREE_MODEL(archives), &iter,
		1, &lbx,
		2, &index,
		-1);

	f = lbx_file_open(lbx, index);
	if (f) {
		if (lbximg_loadpalette(f, &lbx_arch_fops, palette_external))
			memset(palette_external, 0, sizeof palette_external);
		lbx_file_close(f);
	}

	redraw_image();
}

void set_image(GtkComboBox *combo)
{
	GtkTreeIter iter;
	gpointer lbx;
	guint    index;
	LBXimg *img;

	if (image) {
		lbximg_close(image);
		image = NULL;
	}

	gtk_widget_set_size_request(canvas, -1, -1);
	memset(palette_internal, 0, sizeof palette_internal);
	if (!gtk_combo_box_get_active_iter(combo, &iter))
		return;

	gtk_tree_model_get(GTK_TREE_MODEL(archives), &iter,
		1, &lbx,
		2, &index,
		-1);

	img = load_lbx_image(lbx, index);
	if (img) {
		if (lbximg_getpalette(img, palette_internal) == -1) {
			puts("crap");
			lbximg_close(img);
		}

		if (alloc_framebuffer(img) == -1) {
			puts("crap");
			lbximg_close(img);
		}
	}

	image = img;
	redraw_image();
}

void set_background(GtkCheckMenuItem *item, gpointer data)
{
	GdkColor black = { .red = 0,      .green = 0,      .blue = 0      };
	GdkColor dark  = { .red = 0x6666, .green = 0x6666, .blue = 0x6666 };
	GdkColor light = { .red = 0x9999, .green = 0x9999, .blue = 0x9999 };

	if (gtk_check_menu_item_get_active(item)) {
		/* Enable stippled background. */
		gdk_gc_set_rgb_fg_color(bg_gc, &dark);
		gdk_gc_set_rgb_bg_color(bg_gc, &light);
		gdk_gc_set_fill(bg_gc, GDK_OPAQUE_STIPPLED);
	} else {
		gdk_gc_set_rgb_fg_color(bg_gc, &black);
		gdk_gc_set_fill(bg_gc, GDK_SOLID);
	}
	redraw_image();
}

void show_about(GtkWidget *widget)
{
	GtkWindow *parent = GTK_WINDOW(gtk_widget_get_toplevel(widget));

	gtk_show_about_dialog(parent,
		"name", g_get_application_name(),
		"version", PACKAGE_VERSION,
		"copyright", "Copyright \xc2\xa9 2010 Nick Bowler",
		"website", "http://toom.sourceforge.net/",
		NULL);
}

static int load_archive(const char *path)
{
	gchar *basename;
	GtkTreeIter iter1, iter2;
	unsigned nfiles;
	LBX *lbx;

	lbx = lbx_fopen(path);
	if (!lbx)
		return -1;

	basename = g_path_get_basename(path);
	gtk_tree_store_append(archives, &iter1, NULL);
	gtk_tree_store_set(archives, &iter1,
		0, basename,
		1, (gpointer)lbx,
		-1);
	g_free(basename);

	nfiles = lbx_numfiles(lbx);
	for (unsigned i = 0; i < nfiles; i++) {
		struct lbx_statbuf stat;

		lbx_file_stat(lbx, i, &stat);
		gtk_tree_store_append(archives, &iter2, &iter1);
		gtk_tree_store_set(archives, &iter2,
			0, stat.name,
			1, (gpointer)lbx,
			2, (guint)i,
			-1);
	}

	return 0;
}

static void load_error(const char *name, GtkWindow *window)
{
	GtkWidget *dialog;

	dialog = gtk_message_dialog_new(window,
		GTK_DIALOG_MODAL,
		GTK_MESSAGE_ERROR,
		GTK_BUTTONS_OK,
		"Failed to open %s.  Decent error messages coming eventually.",
		name);
	gtk_dialog_run(GTK_DIALOG(dialog));
	gtk_widget_destroy(dialog);
}

void open_archive(GtkWidget *widget)
{
	GtkWindow *window = GTK_WINDOW(gtk_widget_get_toplevel(widget));
	GtkFileFilter *lbxfilter, *allfilter;
	GtkFileChooser *chooser;
	GtkTreeIter iter;

	chooser = GTK_FILE_CHOOSER(gtk_file_chooser_dialog_new(
		"Load Archive",
		window,
		GTK_FILE_CHOOSER_ACTION_OPEN,
		GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
		GTK_STOCK_OPEN, GTK_RESPONSE_ACCEPT,
		NULL));

	lbxfilter = gtk_file_filter_new();
	gtk_file_filter_set_name(lbxfilter, "LBX Archives (*.lbx)");
	gtk_file_filter_add_pattern(lbxfilter, "*.lbx");
	gtk_file_filter_add_pattern(lbxfilter, "*.LBX");

	allfilter = gtk_file_filter_new();
	gtk_file_filter_set_name(allfilter, "All Files");
	gtk_file_filter_add_pattern(allfilter, "*");

	gtk_file_chooser_add_filter(chooser, lbxfilter);
	gtk_file_chooser_add_filter(chooser, allfilter);

	while (gtk_dialog_run(GTK_DIALOG(chooser)) == GTK_RESPONSE_ACCEPT) {
		gchar *name = gtk_file_chooser_get_filename(chooser);

		if (load_archive(name) == 0) {
			g_free(name);
			break;
		}

		load_error(name, GTK_WINDOW(chooser));
		g_free(name);
	}

	gtk_widget_destroy(GTK_WIDGET(chooser));
}

static void clear_combobox(GtkEntry *entry, GtkEntryIconPosition pos,
                           GdkEvent *event, gpointer data)
{
	GtkComboBox *combo = data;

	gtk_entry_set_text(entry, "");
	gtk_combo_box_set_active(combo, -1);
	redraw_image();
}

static void init_combobox(GtkBuilder *builder, const char *name)
{
	GtkComboBox *combo;
	GtkEntry *entry;

	combo = GTK_COMBO_BOX(gtk_builder_get_object(builder, name));
	gtk_combo_box_set_model(combo, GTK_TREE_MODEL(archives));
	gtk_combo_box_entry_set_text_column(GTK_COMBO_BOX_ENTRY(combo), 0);

	entry = GTK_ENTRY(gtk_bin_get_child(GTK_BIN(combo)));
	g_signal_connect(G_OBJECT(entry), "icon-press",
		G_CALLBACK(clear_combobox), combo);
	gtk_entry_set_icon_from_stock(entry, GTK_ENTRY_ICON_SECONDARY,
		GTK_STOCK_CLEAR);
}

static void init_interface(void)
{
	GtkTreeViewColumn *column;
	GtkCellRenderer *renderer;
	GtkTreeView *tree;

	renderer = gtk_cell_renderer_text_new();
	column = gtk_tree_view_column_new();
	gtk_tree_view_column_pack_start(column, renderer, TRUE);
	gtk_tree_view_column_add_attribute(column, renderer, "text", 0);

	tree = GTK_TREE_VIEW(gtk_builder_get_object(builder, "treeview"));
	gtk_tree_view_set_model(tree, GTK_TREE_MODEL(archives));
	gtk_tree_view_append_column(tree, column);

	init_combobox(builder, "palettechooser");
	init_combobox(builder, "imagechooser");
	init_combobox(builder, "overchooser");
}

static void init_background(GdkDrawable *drawable)
{
	GtkWidget *check;
	GdkBitmap *bitmap;

	bg_gc = gdk_gc_new(canvas->window);

	bitmap = gdk_bitmap_create_from_data(drawable, (gchar *)bg_bits,
		bg_width, bg_height);

	gdk_gc_set_stipple(bg_gc, bitmap);
	gdk_gc_set_ts_origin(bg_gc, 0, 0);

	check = GTK_WIDGET(gtk_builder_get_object(builder, "menu-background"));
	gtk_check_menu_item_set_active(GTK_CHECK_MENU_ITEM(check), TRUE);
	set_background(GTK_CHECK_MENU_ITEM(check), NULL);
}

int main(int argc, char **argv)
{
	GtkWidget *window;
	GError *err = NULL;

	if (!gtk_init_with_args(&argc, &argv, NULL, NULL, NULL, &err)) {
		fprintf(stderr, "%s\n", err->message);
		return EXIT_FAILURE;
	}

	archives = gtk_tree_store_new(3,
		G_TYPE_STRING, G_TYPE_POINTER, G_TYPE_UINT);
	while (*argv)
		load_archive(*argv++);

	builder = gtk_builder_new();
	if (!gtk_builder_add_from_string(builder, lbxgui_xml, -1, &err)) {
		fprintf(stderr, "%s\n", err->message);
		return EXIT_FAILURE;
	}

	window = GTK_WIDGET(gtk_builder_get_object(builder, "mainwindow"));
	canvas = GTK_WIDGET(gtk_builder_get_object(builder, "canvas"));

	init_interface();
	gtk_builder_connect_signals(builder, window);

	g_timeout_add(10, timeout, NULL);
	gtk_widget_show_all(window);

	init_background(canvas->window);
	gtk_main();
}