#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <stdarg.h>
#include <time.h>

#include "dive.h"
#include "display.h"
#include "divelist.h"

static int cylinder_changed;
static GtkComboBox *cylinder_description;
static GtkSpinButton *cylinder_size, *cylinder_pressure, *nitrox_value;

static void cylinder_cb(GtkComboBox *combo_box, gpointer data)
{
	GtkTreeIter iter;
	GtkTreeModel *model = gtk_combo_box_get_model(combo_box);
	GValue value1 = {0, }, value2 = {0,};
	int volume, pressure;
	cylinder_t *cyl = current_dive->cylinder + 0;

	/* Did the user set it to some non-standard value? */
	if (!gtk_combo_box_get_active_iter(combo_box, &iter)) {
		cylinder_changed = 1;
		return;
	}

	/*
	 * We get "change" signal callbacks just because we set
	 * the description by hand. Whatever. So ignore them if
	 * they are no-ops.
	 */
	if (!cylinder_changed && cyl->type.description) {
		int same;
		char *desc = gtk_combo_box_get_active_text(combo_box);

		same = !strcmp(desc, cyl->type.description);
		g_free(desc);
		if (same)
			return;
	}
	cylinder_changed = 1;

	gtk_tree_model_get_value(model, &iter, 1, &value1);
	volume = g_value_get_int(&value1);
	gtk_tree_model_get_value(model, &iter, 2, &value2);
	pressure = g_value_get_int(&value2);

	gtk_spin_button_set_value(cylinder_size,
			volume / 1000.0);
	gtk_spin_button_set_value(cylinder_pressure,
			pressure / 1000.0);
}

static gboolean match_cylinder(GtkTreeModel *model,
				GtkTreePath *path,
				GtkTreeIter *iter,
				gpointer data)
{
	const char *name;
	const char *desc = data;
	GValue value = {0, };

	gtk_tree_model_get_value(model, iter, 0, &value);
	name = g_value_get_string(&value);
	if (strcmp(desc, name))
		return FALSE;
	gtk_combo_box_set_active_iter(cylinder_description, iter);
	return TRUE;
}

void show_dive_equipment(struct dive *dive)
{
	cylinder_t *cyl = &dive->cylinder[0];
	const char *desc = cyl->type.description;
	GtkTreeModel *model = gtk_combo_box_get_model(cylinder_description);
	double o2;

	if (desc)
		gtk_tree_model_foreach(model, match_cylinder, (gpointer)desc);
	gtk_spin_button_set_value(cylinder_size,
			cyl->type.size.mliter / 1000.0);
	gtk_spin_button_set_value(cylinder_pressure,
			cyl->type.workingpressure.mbar / 1000.0);
	o2 = cyl->gasmix.o2.permille / 10.0;
	if (!o2)
		o2 = 21.0;
	gtk_spin_button_set_value(nitrox_value, o2);
}

static GtkWidget *create_spinbutton(GtkWidget *vbox, const char *name, double min, double max, double incr)
{
	GtkWidget *frame, *button;

	frame = gtk_frame_new(name);
	gtk_container_add(GTK_CONTAINER(vbox), frame);

	button = gtk_spin_button_new_with_range(min, max, incr);
	gtk_container_add(GTK_CONTAINER(frame), button);

	gtk_spin_button_set_update_policy(GTK_SPIN_BUTTON(button), GTK_UPDATE_IF_VALID);

	return button;
}

static void fill_cylinder_info(cylinder_t *cyl, const char *desc, int mliter, int mbar, int o2)
{
	if (o2 < 211)
		o2 = 0;
	cyl->type.description = desc;
	cyl->type.size.mliter = mliter;
	cyl->type.workingpressure.mbar = mbar;
	cyl->gasmix.o2.permille = o2;
}

static void record_cylinder_changes(struct dive *dive)
{
	const gchar *desc;
	GtkComboBox *box = cylinder_description;
	int volume, pressure, o2;

	desc = gtk_combo_box_get_active_text(box);
	volume = gtk_spin_button_get_value(cylinder_size) * 1000 + 0.5;
	pressure = gtk_spin_button_get_value(cylinder_pressure) * 1000 + 0.5;
	o2 = gtk_spin_button_get_value(nitrox_value)*10 + 0.5;
	fill_cylinder_info(dive->cylinder+0, desc, volume, pressure, o2);
}

void flush_dive_equipment_changes(struct dive *dive)
{
	record_cylinder_changes(dive);
}

/* We should take these from the dive list instead */
static struct tank_info {
	const char *name;
	int size;	/* cuft or mliter depending on psi */
	int psi;	/* If zero, size is in mliter */
} tank_info[] = {
	{ "None", 0, 0 },
	{ "10.0 l", 10000 },
	{ "11.1 l", 11100 },
	{ "AL72", 72, 3000 },
	{ "AL80", 80, 3000 },
	{ "LP85", 85, 2400 },
	{ "LP95", 95, 2400 },
	{ "LP85+", 85, 2640 },
	{ "LP95+", 95, 2640 },
	{ "HP100", 100, 3442 },
	{ "HP119", 119, 3442 },
	{ NULL, }
};

static void fill_tank_list(GtkListStore *store)
{
	GtkTreeIter iter;

	struct tank_info *info = tank_info;

	while (info->name) {
		int size = info->size;
		int psi = info->psi;
		int mbar = 0, ml = size;

		/* Is it in cuft and psi? */
		if (psi) {
			double bar = 0.0689475729 * psi;
			double airvolume = 28316.8466 * size;
			double atm = bar / 1.01325;

			ml = airvolume / atm + 0.5;
			mbar = bar*1000 + 0.5;
		}

		gtk_list_store_append(store, &iter);
		gtk_list_store_set(store, &iter,
			0, info->name,
			1, ml,
			2, mbar,
			-1);
		info++;
	}
}

static void cylinder_widget(GtkWidget *box, int nr, GtkListStore *model)
{
	GtkWidget *frame, *hbox;
	GtkWidget *widget;
	char buffer[80];

	snprintf(buffer, sizeof(buffer), "Cylinder %d", nr);
	frame = gtk_frame_new(buffer);
	gtk_box_pack_start(GTK_BOX(box), frame, FALSE, FALSE, 0);

	hbox = gtk_hbox_new(TRUE, 3);
	gtk_container_add(GTK_CONTAINER(frame), hbox);

	frame = gtk_frame_new("Description");
	gtk_box_pack_start(GTK_BOX(hbox), frame, FALSE, FALSE, 0);

	widget = gtk_combo_box_entry_new_with_model(GTK_TREE_MODEL(model), 0);
	gtk_container_add(GTK_CONTAINER(frame), widget);

	cylinder_description = GTK_COMBO_BOX(widget);
	g_signal_connect(widget, "changed", G_CALLBACK(cylinder_cb), NULL);

	widget = create_spinbutton(hbox, "Size", 0, 200, 0.1);
	cylinder_size = GTK_SPIN_BUTTON(widget);

	widget = create_spinbutton(hbox, "Working Pressure", 0, 5000, 1);
	cylinder_pressure = GTK_SPIN_BUTTON(widget);

	widget = create_spinbutton(hbox, "Nitrox", 21, 100, 0.1);
	nitrox_value = GTK_SPIN_BUTTON(widget);
	gtk_spin_button_set_range(nitrox_value, 21.0, 100.0);
}

static GtkListStore *create_tank_size_model(void)
{
	GtkListStore *model;

	model = gtk_list_store_new(3,
		G_TYPE_STRING,		/* Tank name */
		G_TYPE_INT,		/* Tank size in mliter */
		G_TYPE_INT,		/* Tank working pressure in mbar */
		-1);

	fill_tank_list(model);
	return model;
}

GtkWidget *equipment_widget(void)
{
	GtkWidget *vbox;
	GtkListStore *model;

	vbox = gtk_vbox_new(FALSE, 3);

	model = create_tank_size_model();
	cylinder_widget(vbox, 0, model);

	return vbox;
}