#include <gtk/gtk.h>


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

#include "smoke.h"
#include "palette.h"
#include "smoke.h"  /* colormap and smoke are siblings */
#include "colormap.h"

#include "gtk_colormap.h"


static gboolean select_colormap(GtkComboBox *combo, gpointer data)
{
  colormap_set_colormap(gtk_combo_box_get_active(combo));

  return TRUE;
}

static gboolean select_num_colors(GtkRange *adjustment, gpointer data)
{
  colormap_set_num_colors((int)gtk_range_get_value(adjustment));

  return TRUE;
}

static gboolean set_scaling(GtkToggleButton *button, gpointer data) {
  colormap_set_scaling(gtk_toggle_button_get_active(button));

  return TRUE;
}

static gboolean set_clamping(GtkToggleButton *button, gpointer data) {
  colormap_set_clamping(gtk_toggle_button_get_active(button));

  return TRUE;
}

static gboolean reset_scaling(GtkWidget *button, gpointer data) {
  colormap_reset_scaling();

  return TRUE;
}

static gboolean reset_clamping(GtkWidget *button, gpointer data) {
  colormap_reset_clamping();

  return TRUE;
}

static gboolean select_smoke_sort(GtkComboBox *combo, gpointer data)
{
  smoke_set_sort(gtk_combo_box_get_active(combo));

  return TRUE;
}

static gboolean select_alpha(GtkRange *adjustment, gpointer data)
{
  colormap_set_alpha((float)gtk_range_get_value(adjustment));

  return TRUE;
}

static gboolean adjust_autoscaling(GtkToggleButton *button, gpointer data) {
  colormap_set_autoscaling(gtk_toggle_button_get_active(button));

  return TRUE;
}

static gboolean set_dataset(GtkToggleButton *button, gpointer data)
{
  if (gtk_toggle_button_get_active(button)) {
    smoke_set_dataset((int)data);
  }

  return TRUE;
}


GtkWidget *create_colormap_page(void)
{
  /* in reality a page is really just a vbox filled with other widgets
   */
  GtkWidget *page;
  GtkWidget *frame;
  GtkWidget *box, *box2;
  GtkWidget *label;
  GtkWidget *button;
  GtkWidget *combo;
  GtkAdjustment *color_adjustment;
  GtkWidget *color_scale;
  GSList *dataset_group;

  page = gtk_vbox_new(FALSE, 0);

    frame = gtk_frame_new("Scaling & Clamping");

      box = gtk_vbox_new(FALSE, 0);

        box2 = gtk_hbox_new(FALSE, 0);

          button = gtk_check_button_new_with_label("Scaling");
            gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), colormap_get_scaling());
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(set_scaling), NULL);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          button = gtk_check_button_new_with_label("Clamping");
            gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), colormap_get_clamping());
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(set_clamping), 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);

          button = gtk_check_button_new_with_label("AutoScale");
            gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), colormap_get_autoscaling());
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(adjust_autoscaling), NULL);
            gtk_box_pack_start(GTK_BOX(box), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

        box2 = gtk_hbox_new(FALSE, 0);
   
          button = gtk_button_new_with_label("Reset scaling");
          g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_scaling), NULL);
          gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);      
          gtk_widget_show(button);
   
          button = gtk_button_new_with_label("Reset clamping");
          g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(reset_clamping), 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("Dataset");

      box = gtk_vbox_new(FALSE, 0);

        box2 = gtk_hbox_new(FALSE, 0);
        
          button = gtk_radio_button_new_with_label (NULL, "Rho");
            if (smoke_get_dataset() == DATASET_RHO) {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), TRUE);
            } else {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), FALSE);            
            }
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(set_dataset), (gpointer)DATASET_RHO);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          dataset_group = gtk_radio_button_get_group(GTK_RADIO_BUTTON(button));

          button = gtk_radio_button_new_with_label(dataset_group, "Velocity");
            if (smoke_get_dataset() == DATASET_VEL) {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), TRUE);
            } else {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), FALSE);            
            }
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(set_dataset), (gpointer)DATASET_VEL);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          button = gtk_radio_button_new_with_label_from_widget(GTK_RADIO_BUTTON(button), "Force");
            if (smoke_get_dataset() == DATASET_FORCE) {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), TRUE);
            } else {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), FALSE);            
            }
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(set_dataset), (gpointer)DATASET_FORCE);
            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_hbox_new(FALSE, 0);
  
          dataset_group = gtk_radio_button_get_group(GTK_RADIO_BUTTON(button));

          button = gtk_radio_button_new_with_label(dataset_group, "Velocity divergence");
            if (smoke_get_dataset() == DATASET_DIVV) {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), TRUE);
            } else {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), FALSE);            
            }
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(set_dataset), (gpointer)DATASET_DIVV);
            gtk_box_pack_start(GTK_BOX(box2), button, TRUE, TRUE, 0);
          gtk_widget_show(button);

          button = gtk_radio_button_new_with_label_from_widget(GTK_RADIO_BUTTON(button), "Force divergence");
            if (smoke_get_dataset() == DATASET_DIVF) {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), TRUE);
            } else {
              gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(button), FALSE);            
            }
            g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(set_dataset), (gpointer)DATASET_DIVF);
            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("Colors");

    box = gtk_vbox_new(FALSE, 0);
      

      label = gtk_label_new("Colormap");

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


      combo = gtk_combo_box_new_text();

        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_BLACKWHITE);
        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_RAINBOW);
        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_BANDS);
        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_GREY_BANDS);
        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_WILRIK);
        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_OLIVER);
        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_RED);
        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_GREEN);
        gtk_combo_box_append_text(GTK_COMBO_BOX(combo), PALETTE_NAME_BLUE);

        gtk_combo_box_set_active(GTK_COMBO_BOX(combo), colormap_get_colormap());
        g_signal_connect(combo, "changed", G_CALLBACK(select_colormap), (gpointer)NULL);

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


      label = gtk_label_new("Number of Colors");

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


      color_adjustment = GTK_ADJUSTMENT(gtk_adjustment_new(
            colormap_get_num_colors(), 1, PALETTE_MAXCOLORS, 0.5, 5, 0.1));
      color_scale = gtk_hscale_new(color_adjustment);
			  gtk_scale_set_digits(GTK_SCALE(color_scale), 0);
        g_signal_connect(GTK_RANGE(color_scale), "value-changed", G_CALLBACK(select_num_colors), NULL);

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


      label = gtk_label_new("Alpha");

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

      color_adjustment = GTK_ADJUSTMENT(gtk_adjustment_new(
            colormap_get_alpha(), 0, 1, 0.05, 0.05, 0.01));
      color_scale = gtk_hscale_new(color_adjustment);
			  gtk_scale_set_digits(GTK_SCALE(color_scale), 1);
        g_signal_connect(GTK_RANGE(color_scale), "value-changed", G_CALLBACK(select_alpha), NULL);

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


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

        label = gtk_label_new("Smoke visualization");
        gtk_box_pack_start(GTK_BOX(box), label, FALSE, TRUE, 0);
        gtk_widget_show(label);

        combo = gtk_combo_box_new_text();

          gtk_combo_box_append_text(GTK_COMBO_BOX(combo), "Fluid smoke");
          gtk_combo_box_append_text(GTK_COMBO_BOX(combo), "Grid smoke");

          gtk_combo_box_set_active(GTK_COMBO_BOX(combo), smoke_get_sort());
          g_signal_connect(combo, "changed", G_CALLBACK(select_smoke_sort), (gpointer)NULL);

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

  gtk_widget_show(frame);

  return page;
}