#include <gtk/gtk.h>


#include <rfftw.h>
#include "fluids.h"


#include "gtk_main.h"
#include "renderer_gl.h"
#include "normals.h"
#include "smoke.h"
#include "colormap.h"
#include "glyphs.h"
#include "divergence.h"
#include "isolines.h"
#include "heightplots.h"
#include "streamlines.h"
#include "flowvis.h"



static gboolean select_render_grid(GtkWidget *button, gpointer data) {
  renderer_set_render_grid(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button)));

  return TRUE;
}


static gboolean select_render_smoke(GtkWidget *button, gpointer data) {
  smoke_set_render(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button)));

  return TRUE;
}


static gboolean select_render_glyphs(GtkWidget *button, gpointer data) {
  glyphs_set_render(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button)));

  return TRUE;
}

static gboolean select_render_isolines(GtkWidget *button, gpointer data) {
  isolines_set_render(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button)));

  return TRUE;
}

static gboolean select_render_streamlines(GtkWidget *button, gpointer data) {
  streamlines_set_render(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button)));

  return TRUE;
}

static gboolean select_render_flowvis(GtkWidget *button, gpointer data) {
  flowvis_set_render(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button)));

  return TRUE;
}

static gboolean select_calculate(GtkWidget *button, gpointer data) {
  fluids_set_calculate(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button)));

  return TRUE;
}

static gboolean select_normals(GtkWidget *button, gpointer data) {
  normals_set_render(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(button)));

  return TRUE;
}

static gboolean reset_zoom(GtkWidget *button, gpointer data) {
  renderer_reset_zoom();

  return TRUE;
}

static gboolean reset_pos(GtkWidget *button, gpointer data) {
  renderer_reset_pos();

  return TRUE;
}

static gboolean reset_yaw(GtkWidget *button, gpointer data) {
  renderer_reset_yaw();

  return TRUE;
}

static gboolean reset_pitch(GtkWidget *button, gpointer data) {
  renderer_reset_pitch();

  return TRUE;
}

static gboolean reset_roll(GtkWidget *button, gpointer data) {
  renderer_reset_roll();

  return TRUE;
}

static gboolean reset_rotate(GtkWidget *button, gpointer data) {
  renderer_reset_yaw();
  renderer_reset_pitch();
  renderer_reset_roll();

  return TRUE;
}

static gboolean reset_all(GtkWidget *button, gpointer data) {
  renderer_reset_yaw();
  renderer_reset_pitch();
  renderer_reset_roll();
  renderer_reset_zoom();
  renderer_reset_pos();

  return TRUE;
}

static gboolean select_zoom_speed(GtkRange *adjustment, gpointer data)
{
  renderer_set_zoomspeed((int)gtk_range_get_value(adjustment));

  return TRUE;
}

static gboolean select_cell_size(GtkRange *adjustment, gpointer data)
{
  renderer_set_grid_cell_size((int)gtk_range_get_value(adjustment));

  return TRUE;
}

static gboolean reset_simulation(GtkWidget *button, gpointer data) {
  fluids_reset_simulation();

  return TRUE;
}


GtkWidget *create_main_page(void)
{
  /* in reality a page is really just a vbox filled with other widgets
   */
  GtkWidget *page;
  GtkWidget *frame;
  GtkWidget *box, *box2;
  GtkWidget *button;
  GtkAdjustment *adjustment;
  GtkWidget *scale;

  page = gtk_vbox_new(FALSE, 0);

    frame = gtk_frame_new("Simulation");

      box = gtk_vbox_new(FALSE, 0);
          button = gtk_check_button_new_with_label("Draw grid");
          gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), renderer_get_render_grid());
          g_signal_connect (G_OBJECT(button), "clicked", G_CALLBACK(select_render_grid), NULL);
          gtk_box_pack_start (GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);
          
          button = gtk_check_button_new_with_label("Draw smoke");
          gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), smoke_get_render());
          g_signal_connect (G_OBJECT(button), "clicked", G_CALLBACK(select_render_smoke), NULL);
          gtk_box_pack_start (GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);
          
          button = gtk_check_button_new_with_label("Draw glyphs");
          gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), glyphs_get_render());
          g_signal_connect (G_OBJECT(button), "clicked", G_CALLBACK(select_render_glyphs), NULL);
          gtk_box_pack_start (GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          button = gtk_check_button_new_with_label("Draw isolines");
          gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), isolines_get_render());
          g_signal_connect (G_OBJECT(button), "clicked", G_CALLBACK(select_render_isolines), NULL);
          gtk_box_pack_start (GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);
   
          button = gtk_check_button_new_with_label("Draw streamlines");
          gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), streamlines_get_render());
          g_signal_connect (G_OBJECT(button), "clicked", G_CALLBACK(select_render_streamlines), NULL);
          gtk_box_pack_start (GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          button = gtk_check_button_new_with_label("Draw smoke history");
          gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), flowvis_get_render());
          g_signal_connect (G_OBJECT(button), "clicked", G_CALLBACK(select_render_flowvis), NULL);
          gtk_box_pack_start (GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          button = gtk_check_button_new_with_label("Toggle calculation");
          gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), fluids_get_calculate());
          g_signal_connect (G_OBJECT(button), "clicked", G_CALLBACK(select_calculate), NULL);
          gtk_box_pack_start (GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          button = gtk_check_button_new_with_label("Draw normal vectors");
          gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), normals_get_render());
          g_signal_connect (G_OBJECT(button), "clicked", G_CALLBACK(select_normals), NULL);
          gtk_box_pack_start (GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          button = gtk_button_new_with_label("Reset simulation");
          g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_simulation), NULL);
          gtk_box_pack_start(GTK_BOX(box), button, TRUE, TRUE, 0);      
          gtk_widget_show(button);

        gtk_container_add(GTK_CONTAINER(frame), box);
      gtk_widget_show(box);

      gtk_box_pack_start (GTK_BOX(page), frame, FALSE, TRUE, 0);
    gtk_widget_show(frame);

    frame = gtk_frame_new("View");

      box = gtk_hbox_new(FALSE, 0);

        button = gtk_button_new_with_label("Reset all");
          g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_all), NULL);
          gtk_box_pack_start(GTK_BOX(box), button, TRUE, TRUE, 0);      
        gtk_widget_show(button);

        box2 = gtk_vbox_new(FALSE, 0);

          button = gtk_button_new_with_label("Reset zoom");
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_zoom), NULL);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);      
          gtk_widget_show(button);

          button = gtk_button_new_with_label("Reset position");
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_pos), NULL);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);      
          gtk_widget_show(button);

          button = gtk_button_new_with_label("Reset rotate");
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_rotate), NULL);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);      
          gtk_widget_show(button);
          
        gtk_box_pack_start(GTK_BOX(box), box2, TRUE, TRUE, 0);
        gtk_widget_show(box2);

        box2 = gtk_vbox_new(FALSE, 0);

          button = gtk_button_new_with_label("Reset yaw");
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_yaw), NULL);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);      
          gtk_widget_show(button);

          button = gtk_button_new_with_label("Reset pitch");
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_pitch), NULL);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);      
          gtk_widget_show(button);

          button = gtk_button_new_with_label("Reset roll");
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_roll), NULL);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);      
          gtk_widget_show(button);

        gtk_box_pack_start(GTK_BOX(box), box2, TRUE, TRUE, 0);
        gtk_widget_show(box2);

        gtk_container_add(GTK_CONTAINER(frame), box);
      gtk_widget_show(box);

      gtk_box_pack_start (GTK_BOX(page), frame, FALSE, TRUE, 0);
    gtk_widget_show(frame);

    frame = gtk_frame_new("Zoom speed");

      box = gtk_vbox_new(FALSE, 0);

       adjustment = GTK_ADJUSTMENT(gtk_adjustment_new(
            renderer_get_zoomspeed(), 1, 256, 5, 5, 0.1));
       scale = gtk_hscale_new(adjustment);
			 gtk_scale_set_digits(GTK_SCALE(scale), 0);
       g_signal_connect(GTK_RANGE(scale), "value-changed", G_CALLBACK(select_zoom_speed), NULL);

      gtk_box_pack_start(GTK_BOX(box), scale, FALSE, TRUE, 0);
      gtk_widget_show(scale);

      gtk_widget_show(box);
      gtk_container_add(GTK_CONTAINER(frame), box);

      gtk_box_pack_start (GTK_BOX(page), frame, FALSE, TRUE, 0);
    gtk_widget_show(frame);


    frame = gtk_frame_new("Grid cell size");

      box = gtk_vbox_new(FALSE, 0);

       adjustment = GTK_ADJUSTMENT(gtk_adjustment_new(
            renderer_get_grid_cell_size(), 16, 128, 1, 8, 0.0));
       scale = gtk_hscale_new(adjustment);
			 gtk_scale_set_digits(GTK_SCALE(scale), 0);
       g_signal_connect(GTK_RANGE(scale), "value-changed", G_CALLBACK(select_cell_size), NULL);

      gtk_box_pack_start(GTK_BOX(box), scale, FALSE, TRUE, 0);
      gtk_widget_show(scale);

      gtk_widget_show(box);
      gtk_container_add(GTK_CONTAINER(frame), box);

      gtk_box_pack_start (GTK_BOX(page), frame, FALSE, TRUE, 0);
    gtk_widget_show(frame);


  return page;
}