/* ASE - Allegro Sprite Editor
 * Copyright (C) 2001-2010  David Capello
 *
 * 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 2 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, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "config.h"

#include <allegro.h>
#include <stdio.h>
#include <string.h>
#include <vector>

#include "base/bind.h"
#include "gui/jinete.h"

#include "app.h"
#include "app/color.h"
#include "commands/command.h"
#include "commands/params.h"
#include "core/cfg.h"
#include "dialogs/filesel.h"
#include "gfx/hsv.h"
#include "gfx/rgb.h"
#include "gfx/size.h"
#include "modules/editors.h"
#include "modules/gui.h"
#include "modules/palettes.h"
#include "raster/image.h"
#include "raster/palette.h"
#include "raster/quantization.h"
#include "raster/sprite.h"
#include "raster/stock.h"
#include "raster/undo.h"
#include "sprite_wrappers.h"
#include "ui_context.h"
#include "widgets/color_bar.h"
#include "widgets/colview.h"
#include "widgets/editor.h"
#include "widgets/paledit.h"
#include "widgets/statebar.h"

using namespace gfx;

static Frame* window = NULL;
static int redraw_timer_id = -1;
static bool redraw_all = false;

// Slot for App::Exit signal 
static void on_exit_delete_this_widget()
{
  ASSERT(window != NULL);

  jmanager_remove_timer(redraw_timer_id);
  redraw_timer_id = -1;

  jwidget_free(window);
}

//////////////////////////////////////////////////////////////////////
// palette_editor

class PaletteEditorCommand : public Command
{
public:
  PaletteEditorCommand();
  Command* clone() { return new PaletteEditorCommand(*this); }

protected:
  void onLoadParams(Params* params);
  void onExecute(Context* context);

private:
  bool m_open;
  bool m_close;
  bool m_switch;
  bool m_background;
};

// #define get_sprite(wgt) (*(const SpriteReader*)(wgt->getRoot())->user_data[0])

static Widget *R_label, *G_label, *B_label;
static Widget *H_label, *S_label, *V_label;
static Slider *R_slider, *G_slider, *B_slider;
static Slider *H_slider, *S_slider, *V_slider;
static Widget *R_entry, *G_entry, *B_entry;
static Widget *H_entry, *S_entry, *V_entry;
static Widget *hex_entry;
static PalEdit* palette_editor;
static Widget* more_options = NULL;
static bool disable_colorbar_signals = false;

static bool window_msg_proc(JWidget widget, JMessage msg);
static bool window_close_hook(JWidget widget, void *data);
static void load_command(JWidget widget);
static void save_command(JWidget widget);
static void ramp_command(JWidget widget);
static void sort_command(JWidget widget);
static void quantize_command(JWidget widget);

static void sliderRGB_change_hook(Slider* widget);
static void sliderHSV_change_hook(Slider* widget);
static bool entryRGB_change_hook(JWidget widget, void *data);
static bool entryHSV_change_hook(JWidget widget, void *data);
static bool hex_entry_change_hook(JWidget widget, void *data);
static void update_entries_from_sliders();
static void update_sliders_from_entries();
static void update_hex_entry();
static void update_current_sprite_palette(const char* operationName);
static void update_colorbar();
static bool palette_editor_change_hook(JWidget widget, void *data);
static bool select_rgb_hook(JWidget widget, void *data);
static bool select_hsv_hook(JWidget widget, void *data);
static bool expand_button_select_hook(JWidget widget, void *data);
static void modify_rgb_of_selected_entries(const Rgb& dst_rgb, bool set_r, bool set_g, bool set_b);
static void modify_hsv_of_selected_entries(const Hsv& dst_hsv, bool set_h, bool set_s, bool set_v);
static void on_color_changed(const Color& color);

static void set_new_palette(Palette *palette, const char* operationName);

PaletteEditorCommand::PaletteEditorCommand()
  : Command("palette_editor",
	    "PaletteEditor",
	    CmdRecordableFlag)
{
  m_open = true;
  m_close = false;
  m_switch = false;
  m_background = false;
}

void PaletteEditorCommand::onLoadParams(Params* params)
{
  std::string target = params->get("target");
  if (target == "foreground") m_background = false;
  else if (target == "background") m_background = true;

  std::string open_str = params->get("open");
  if (open_str == "true") m_open = true;
  else m_open = false;

  std::string close_str = params->get("close");
  if (close_str == "true") m_close = true;
  else m_close = false;

  std::string switch_str = params->get("switch");
  if (switch_str == "true") m_switch = true;
  else m_switch = false;
}

void PaletteEditorCommand::onExecute(Context* context)
{
  Widget* palette_editor_view;
  Widget* select_rgb;
  Widget* select_hsv;
  Button* expand_button;
  Button* button_load;
  Button* button_save;
  Button* button_ramp;
  Button* button_sort;
  Button* button_quantize;
  bool first_time = false;

  // If the window was never loaded yet, load it
  if (!window) {
    if (m_close)
      return;			// Do nothing (the user want to close and inexistent window)

    // Load the palette editor window
    window = static_cast<Frame*>(load_widget("palette_editor.xml", "palette_editor"));
    redraw_timer_id = jmanager_add_timer(window, 250);

    first_time = true;

    // Append hooks
    window->Close.connect(Bind<bool>(&window_close_hook, (JWidget)window, (void*)0));

    // Hook fg/bg color changes (by eyedropper mainly)
    app_get_colorbar()->FgColorChange.connect(&on_color_changed);
    app_get_colorbar()->BgColorChange.connect(&on_color_changed);

    // Hook App::Exit signal
    App::instance()->Exit.connect(&on_exit_delete_this_widget);
  }
  // If the window is opened, close it (only in "switch" mode)
  else if (window->isVisible() && (m_switch || m_close)) {
    window->closeWindow(NULL);
    return;
  }

  get_widgets(window,
	      "R_label",  &R_label,
	      "R_slider", &R_slider,
	      "R_entry",  &R_entry,
	      "G_label",  &G_label,
	      "G_slider", &G_slider,
	      "G_entry",  &G_entry,
	      "B_label",  &B_label,
	      "B_slider", &B_slider,
	      "B_entry",  &B_entry,
	      "H_label",  &H_label,
	      "H_slider", &H_slider,
	      "H_entry",  &H_entry,
	      "S_label",  &S_label,
	      "S_slider", &S_slider,
	      "S_entry",  &S_entry,
	      "V_label",  &V_label,
	      "V_slider", &V_slider,
	      "V_entry",  &V_entry,
	      "hex_entry", &hex_entry,
	      "select_rgb", &select_rgb,
	      "select_hsv", &select_hsv,
	      "expand", &expand_button,
	      "more_options", &more_options,
	      "load", &button_load,
	      "save", &button_save,
	      "ramp", &button_ramp,
	      "sort", &button_sort,
	      "quantize", &button_quantize,
	      "palette_editor", &palette_editor_view, NULL);
      
  // Custom widgets
  if (first_time) {
    palette_editor = new PalEdit(true);
    palette_editor->setBoxSize(4*jguiscale());

    jview_attach(palette_editor_view, palette_editor);
    jview_maxsize(palette_editor_view);

    // Set palette editor columns
    palette_editor->setColumns(16);
  
    // Hook signals
    jwidget_add_hook(window, -1, window_msg_proc, NULL);
    R_slider->Change.connect(Bind<void>(&sliderRGB_change_hook, R_slider));
    G_slider->Change.connect(Bind<void>(&sliderRGB_change_hook, G_slider));
    B_slider->Change.connect(Bind<void>(&sliderRGB_change_hook, B_slider));
    H_slider->Change.connect(Bind<void>(&sliderHSV_change_hook, H_slider));
    S_slider->Change.connect(Bind<void>(&sliderHSV_change_hook, S_slider));
    V_slider->Change.connect(Bind<void>(&sliderHSV_change_hook, V_slider));
    HOOK(R_entry, JI_SIGNAL_ENTRY_CHANGE, entryRGB_change_hook, 0);
    HOOK(G_entry, JI_SIGNAL_ENTRY_CHANGE, entryRGB_change_hook, 0);
    HOOK(B_entry, JI_SIGNAL_ENTRY_CHANGE, entryRGB_change_hook, 0);
    HOOK(H_entry, JI_SIGNAL_ENTRY_CHANGE, entryHSV_change_hook, 0);
    HOOK(S_entry, JI_SIGNAL_ENTRY_CHANGE, entryHSV_change_hook, 0);
    HOOK(V_entry, JI_SIGNAL_ENTRY_CHANGE, entryHSV_change_hook, 0);
    HOOK(hex_entry, JI_SIGNAL_ENTRY_CHANGE, hex_entry_change_hook, 0);
    HOOK(palette_editor, SIGNAL_PALETTE_EDITOR_CHANGE, palette_editor_change_hook, 0);
    HOOK(select_rgb, JI_SIGNAL_RADIO_CHANGE, select_rgb_hook, 0);
    HOOK(select_hsv, JI_SIGNAL_RADIO_CHANGE, select_hsv_hook, 0);
    HOOK(expand_button, JI_SIGNAL_BUTTON_SELECT, expand_button_select_hook, 0);
    
    setup_mini_look(select_rgb);
    setup_mini_look(select_hsv);

    // Hide (or show) the "More Options" depending the saved value in .cfg file
    more_options->setVisible(get_config_bool("PaletteEditor", "ShowMoreOptions", false));

    button_load->Click.connect(Bind<void>(&load_command, button_load));
    button_save->Click.connect(Bind<void>(&save_command, button_save));
    button_ramp->Click.connect(Bind<void>(&ramp_command, button_ramp));
    button_sort->Click.connect(Bind<void>(&sort_command, button_sort));
    button_quantize->Click.connect(Bind<void>(&quantize_command, button_quantize));

    select_rgb_hook(NULL, NULL);
  }

  // Show the specified target color
  {
    Color color =
      (m_background ? context->getSettings()->getBgColor():
  		      context->getSettings()->getFgColor());

    on_color_changed(color);
  }

  if (m_switch || m_open) {
    if (!window->isVisible()) {
      // Default bounds
      window->remap_window();

      int width = MAX(jrect_w(window->rc), JI_SCREEN_W/2);
      window->setBounds(Rect(JI_SCREEN_W - width - jrect_w(app_get_toolbar()->rc),
			     JI_SCREEN_H - jrect_h(window->rc) - jrect_h(app_get_statusbar()->rc),
			     width, jrect_h(window->rc)));

      // Load window configuration
      load_window_pos(window, "PaletteEditor");
    }

    // Run the window in background
    window->open_window_bg();
  }
}

static bool window_msg_proc(JWidget widget, JMessage msg)
{
  if (msg->type == JM_TIMER &&
      msg->timer.timer_id == redraw_timer_id) {
    // Redraw all editors
    if (redraw_all) {
      redraw_all = false;
      jmanager_stop_timer(redraw_timer_id);

      try {
	const CurrentSpriteReader sprite(UIContext::instance());
	update_editors_with_sprite(sprite);
      }
      catch (...) {
	// Do nothing
      }
    }
    // Redraw just the current editor
    else {
      redraw_all = true;
      current_editor->editor_update();
    }
  }
  return false;
}

static bool window_close_hook(JWidget widget, void *data)
{
  // Save window configuration
  save_window_pos(window, "PaletteEditor");
  return false;
}

static void load_command(JWidget widget)
{
  Palette *palette;
  base::string filename = ase_file_selector("Load Palette", "", "png,pcx,bmp,tga,lbm,col");
  if (!filename.empty()) {
    palette = Palette::load(filename.c_str());
    if (!palette) {
      jalert("Error<<Loading palette file||&Close");
    }
    else {
      set_new_palette(palette, "Load Palette");
      delete palette;
    }
  }
}

static void save_command(JWidget widget)
{
  base::string filename;
  int ret;

 again:
  filename = ase_file_selector("Save Palette", "", "png,pcx,bmp,tga,col");
  if (!filename.empty()) {
    if (exists(filename.c_str())) {
      ret = jalert("Warning<<File exists, overwrite it?<<%s||&Yes||&No||&Cancel",
		   get_filename(filename.c_str()));

      if (ret == 2)
	goto again;
      else if (ret != 1)
	return;
    }

    Palette* palette = get_current_palette();
    if (!palette->save(filename.c_str())) {
      jalert("Error<<Saving palette file||&Close");
    }
  }
}

static void ramp_command(JWidget widget)
{
  int range_type = palette_editor->getRangeType();
  int i1 = palette_editor->get1stColor();
  int i2 = palette_editor->get2ndColor();
  Palette* src_palette = get_current_palette();
  Palette* dst_palette = new Palette(0, 256);
  bool array[256];

  palette_editor->getSelectedEntries(array);
  src_palette->copyColorsTo(dst_palette);

  if ((i1 >= 0) && (i2 >= 0)) {
    // Make the ramp
    if (range_type == PALETTE_EDITOR_RANGE_LINEAL) {
      // Lineal ramp
      dst_palette->makeHorzRamp(i1, i2);
    }
    else if (range_type == PALETTE_EDITOR_RANGE_RECTANGULAR) {
      // Rectangular ramp
      dst_palette->makeRectRamp(i1, i2, palette_editor->getColumns());
    }
  }

  set_new_palette(dst_palette, "Color Ramp");
  delete dst_palette;
}

//////////////////////////////////////////////////////////////////////
// Sort Options Begin

struct SortDlgData
{
  Widget* available_criteria;
  Widget* selected_criteria;
  Widget* insert_criteria;
  Widget* remove_criteria;
  Widget* asc;
  Widget* des;
  Widget* first;
  Widget* last;
  Widget* ok_button;
};

static bool insert_criteria_hook(Widget* widget, void* data);
static bool remove_criteria_hook(Widget* widget, void* data);
static bool sort_by_criteria(Palette* palette, int from, int to, JList selected_listitems, std::vector<int>& mapping);

static void sort_command(JWidget widget)
{
  if (jalert("ASE Beta<<Sort command is not available in this beta version.||&OK")) // TODO remove this
    return;

  SortDlgData data;

  try {
    // Load the sort criteria window
    FramePtr dlg(load_widget("palette_editor.xml", "sort_criteria"));

    get_widgets(dlg,
		"available_criteria", &data.available_criteria,
		"selected_criteria", &data.selected_criteria,
		"insert_criteria", &data.insert_criteria,
		"remove_criteria", &data.remove_criteria,
		"asc", &data.asc,
		"des",  &data.des,
		"first", &data.first,
		"last",  &data.last,
		"ok_button", &data.ok_button, NULL);

    // Selected Ascending by default
    data.asc->setSelected(true);

    // Range to sort
    int i1 = palette_editor->get1stColor();
    int i2 = palette_editor->get2ndColor();
    if (i1 == i2) {		// Sort all palette entries
      i1 = 0;
      i2 = get_current_palette()->size()-1;
    }
    else if (i1 > i2) {
      std::swap(i1, i2);
    }
    data.first->setTextf("%d", i1);
    data.last->setTextf("%d", i2);

    HOOK(data.insert_criteria, JI_SIGNAL_BUTTON_SELECT, insert_criteria_hook, &data);
    HOOK(data.remove_criteria, JI_SIGNAL_BUTTON_SELECT, remove_criteria_hook, &data);

    // If there is a selected <listitem> in available criteria
    // <listbox>, insert it as default criteria to sort colors
    if (jlistbox_get_selected_child(data.available_criteria))
      insert_criteria_hook(data.insert_criteria, (void*)&data);

    // Open the window
    dlg->open_window_fg();

    if (dlg->get_killer() == data.ok_button) {
      Palette* palette = new Palette(*get_current_palette());
      int from = data.first->getTextInt();
      int to = data.last->getTextInt();

      from = MID(0, from, palette->size()-1);
      to = MID(from, to, palette->size()-1);

      std::vector<int> mapping;
      sort_by_criteria(palette, from, to, data.selected_criteria->children, mapping);

      if (UIContext::instance()->get_current_sprite()) {
	// Remap all colors
	if (mapping.size() > 0) {
	  CurrentSpriteWriter sprite(UIContext::instance());
	  Palette* frame_palette = sprite->getCurrentPalette();
	  int frame_begin = 0;
	  int frame_end = 0;
	  int frame = 0;
	  while (frame < sprite->getTotalFrames()) {
	    if (sprite->getPalette(frame) == frame_palette) {
	      frame_begin = frame;
	      break;
	    }
	    ++frame;
	  }
	  while (frame < sprite->getTotalFrames()) {
	    if (sprite->getPalette(frame) != frame_palette)
	      break;
	    ++frame;
	  }
	  frame_end = frame;

	  //////////////////////////////////////////////////////////////////////
	  // TODO The following code is unreadable, move this to Undoable class

	  if (sprite->getUndo()->isEnabled()) {
	    sprite->getUndo()->setLabel("Sort Palette");
	    sprite->getUndo()->undo_open();

	    // Remove the current palette in the current frame
	    sprite->getUndo()->undo_remove_palette(sprite, frame_palette);
	  }

	  // Delete the current palette
	  sprite->deletePalette(frame_palette);

	  // Setup the new palette in the sprite
	  palette->setFrame(frame_begin);
	  sprite->setPalette(palette, true);

	  if (sprite->getUndo()->isEnabled()) {
	    // Add undo information about the new added palette
	    sprite->getUndo()->undo_add_palette(sprite, sprite->getPalette(frame_begin));

	    // Add undo information about image remapping
	    sprite->getUndo()->undo_remap_palette(sprite, frame_begin, frame_end-1, mapping);
	    sprite->getUndo()->undo_close();
	  }

	  // Remap images (to the new palette indexes)
	  sprite->remapImages(frame_begin, frame_end-1, mapping);
	}
      }

      // Set the new palette in the sprite
      set_new_palette(palette, "Sort Palette");

      delete palette;
    }
  }
  catch (ase_exception& e) {
    e.show();
  }
}

static bool insert_criteria_hook(Widget* widget, void* _data)
{
  SortDlgData* data = (SortDlgData*)_data;

  // Move the selected item to the 
  Widget* item = jlistbox_get_selected_child(data->available_criteria);
  if (item) {
    std::string new_criteria(item->getText());
    new_criteria += " - ";
    new_criteria += (data->asc->isSelected() ? data->asc->getText():
					       data->des->getText());

    // Remove the criteria
    int removed_index = jlistbox_get_selected_index(data->available_criteria);
    jwidget_remove_child(data->available_criteria, item);

    int count = jlistbox_get_items_count(data->available_criteria);
    if (count > 0) {
      jlistbox_select_index(data->available_criteria,
			    removed_index < count ? removed_index: count-1);
    }

    // Add to the selected criteria
    item->setText(new_criteria.c_str());
    jwidget_add_child(data->selected_criteria, item);
    jlistbox_select_child(data->selected_criteria, item);

    // Relayout
    data->available_criteria->setBounds(data->available_criteria->getBounds()); // TODO layout()
    data->selected_criteria->setBounds(data->selected_criteria->getBounds()); // TODO layout()
    data->available_criteria->dirty();
    data->selected_criteria->dirty();
  }

  return true;
}

static bool remove_criteria_hook(Widget* widget, void* _data)
{
  SortDlgData* data = (SortDlgData*)_data;

  // Move the selected item to the 
  Widget* item = jlistbox_get_selected_child(data->selected_criteria);
  if (item) {
    std::string criteria_text(item->getText());
    int index = criteria_text.find('-');
    criteria_text = criteria_text.substr(0, index-1);

    // Remove from the selected criteria
    int removed_index = jlistbox_get_selected_index(data->selected_criteria);
    jwidget_remove_child(data->selected_criteria, item);

    int count = jlistbox_get_items_count(data->selected_criteria);
    if (count > 0) {
      jlistbox_select_index(data->selected_criteria,
			    removed_index < count ? removed_index: count-1);
    }

    // Add to the available criteria
    item->setText(criteria_text.c_str());
    jwidget_add_child(data->available_criteria, item);
    jlistbox_select_child(data->available_criteria, item);

    // Relayout
    data->available_criteria->setBounds(data->available_criteria->getBounds()); // TODO layout()
    data->selected_criteria->setBounds(data->selected_criteria->getBounds()); // TODO layout()
    data->available_criteria->dirty();
    data->selected_criteria->dirty();
  }

  return true;
}

static bool sort_by_criteria(Palette* palette, int from, int to, JList selected_listitems, std::vector<int>& mapping)
{
  SortPalette* sort_palette = NULL;
  JLink link;

  JI_LIST_FOR_EACH(selected_listitems, link) {
    Widget* item = (Widget*)link->data;
    std::string item_text = item->getText();
    SortPalette::Channel channel = SortPalette::YUV_Luma;
    bool ascending = false;

    if (item_text.find("RGB") != std::string::npos) {
      if (item_text.find("Red") != std::string::npos) {
	channel = SortPalette::RGB_Red;
      }
      else if (item_text.find("Green") != std::string::npos) {
	channel = SortPalette::RGB_Green;
      }
      else if (item_text.find("Blue") != std::string::npos) {
	channel = SortPalette::RGB_Blue;
      }
      else
	ASSERT(false);
    }
    else if (item_text.find("HSV") != std::string::npos) {
      if (item_text.find("Hue") != std::string::npos) {
	channel = SortPalette::HSV_Hue;
      }
      else if (item_text.find("Saturation") != std::string::npos) {
	channel = SortPalette::HSV_Saturation;
      }
      else if (item_text.find("Value") != std::string::npos) {
	channel = SortPalette::HSV_Value;
      }
      else
	ASSERT(false);
    }
    else if (item_text.find("HSL") != std::string::npos) {
      if (item_text.find("Lightness") != std::string::npos) {
	channel = SortPalette::HSL_Lightness;
      }
      else
	ASSERT(false);
    }
    else if (item_text.find("YUV") != std::string::npos) {
      if (item_text.find("Luma") != std::string::npos) {
	channel = SortPalette::YUV_Luma;
      }
      else
	ASSERT(false);
    }
    else
      ASSERT(false);

    if (item_text.find("Ascending") != std::string::npos)
      ascending = true;
    else if (item_text.find("Descending") != std::string::npos)
      ascending = false;
    else
      ASSERT(false);

    SortPalette* chain = new SortPalette(channel, ascending);
    if (sort_palette)
      sort_palette->addChain(chain);
    else
      sort_palette = chain;
  }

  if (sort_palette) {
    palette->sort(from, to, sort_palette, mapping);
    delete sort_palette;
  }

  return false;
}

// Sort Options End
//////////////////////////////////////////////////////////////////////

static void quantize_command(JWidget widget)
{
  Palette* palette = NULL;

  {
    const CurrentSpriteReader& sprite(UIContext::instance());

    if (sprite == NULL) {
      jalert("Error<<There is no sprite selected to quantize.||&OK");
      return;
    }

    if (sprite->getImgType() != IMAGE_RGB) {
      jalert("Error<<You can use this command only for RGB sprites||&OK");
      return;
    }

    palette = quantization::create_palette_from_rgb(sprite);
  }

  set_new_palette(palette, "Quantize Palette");
  delete palette;
}

static void sliderRGB_change_hook(Slider* widget)
{
  int r = R_slider->getValue();
  int g = G_slider->getValue();
  int b = B_slider->getValue();
  Rgb rgb(r, g, b);
  Hsv hsv(rgb);

  H_slider->setValue(hsv.hueInt());
  S_slider->setValue(hsv.saturationInt());
  V_slider->setValue(hsv.valueInt());

  modify_rgb_of_selected_entries(rgb,
				 widget == R_slider,
				 widget == G_slider,
				 widget == B_slider);

  update_entries_from_sliders();
  update_hex_entry();
  update_current_sprite_palette("Color Change");
  update_colorbar();
}

static void sliderHSV_change_hook(Slider* widget)
{
  int h = H_slider->getValue();
  int s = S_slider->getValue();
  int v = V_slider->getValue();

  Hsv hsv(double(h), double(s) / 100.0, double(v) / 100.0);
  Rgb rgb(hsv);

  R_slider->setValue(rgb.red());
  G_slider->setValue(rgb.green());
  B_slider->setValue(rgb.blue());

  modify_hsv_of_selected_entries(hsv,
				 widget == H_slider,
				 widget == S_slider,
				 widget == V_slider);

  update_entries_from_sliders();
  update_hex_entry();
  update_current_sprite_palette("Color Change");
  update_colorbar();
}

static bool entryRGB_change_hook(JWidget widget, void *data)
{
  int r = R_entry->getTextInt();
  int g = G_entry->getTextInt();
  int b = B_entry->getTextInt();
  r = MID(0, r, 255);
  g = MID(0, g, 255);
  b = MID(0, b, 255);

  Rgb rgb(r, g, b);
  Hsv hsv(rgb);

  H_entry->setTextf("%d", hsv.hueInt());
  S_entry->setTextf("%d", hsv.saturationInt());
  V_entry->setTextf("%d", hsv.valueInt());

  modify_rgb_of_selected_entries(rgb,
				 widget == R_slider,
				 widget == G_slider,
				 widget == B_slider);

  update_sliders_from_entries();
  update_hex_entry();
  update_current_sprite_palette("Color Change");
  update_colorbar();
  return false;
}

static bool entryHSV_change_hook(JWidget widget, void *data)
{
  int h = H_entry->getTextInt();
  int s = S_entry->getTextInt();
  int v = V_entry->getTextInt();
  h = MID(0, h, 360);
  s = MID(0, s, 100);
  v = MID(0, v, 100);

  Hsv hsv(double(h), double(s) / 100.0, double(v) / 100.0);
  Rgb rgb(hsv);

  R_entry->setTextf("%d", rgb.red());
  G_entry->setTextf("%d", rgb.green());
  B_entry->setTextf("%d", rgb.blue());

  modify_hsv_of_selected_entries(hsv,
				 widget == H_entry,
				 widget == S_entry,
				 widget == V_entry);

  update_sliders_from_entries();
  update_hex_entry();
  update_current_sprite_palette("Color Change");
  update_colorbar();
  return false;
}

static bool hex_entry_change_hook(JWidget widget, void *data)
{
  Palette* palette = get_current_palette();
  std::string text = hex_entry->getText();
  int r, g, b;
  bool array[256];
  int c;

  // Fill with zeros at the end of the text
  while (text.size() < 6)
    text.push_back('0');

  // Convert text (Base 16) to integer
  int hex = strtol(text.c_str(), NULL, 16);

  R_slider->setValue(r = ((hex & 0xff0000) >> 16));
  G_slider->setValue(g = ((hex & 0xff00) >> 8));
  B_slider->setValue(b = ((hex & 0xff)));

  Hsv hsv(Rgb(r, g, b));

  palette_editor->getSelectedEntries(array);
  for (c=0; c<256; c++) {
    if (array[c]) {
      palette->setEntry(c, _rgba(r, g, b, 255));
    }
  }

  H_slider->setValue(hsv.hueInt());
  V_slider->setValue(hsv.saturationInt());
  S_slider->setValue(hsv.valueInt());

  update_entries_from_sliders();
  update_current_sprite_palette("Color Change");
  update_colorbar();
  return false;
}

static void update_entries_from_sliders()
{
  R_entry->setTextf("%d", R_slider->getValue());
  G_entry->setTextf("%d", G_slider->getValue());
  B_entry->setTextf("%d", B_slider->getValue());

  H_entry->setTextf("%d", H_slider->getValue());
  S_entry->setTextf("%d", S_slider->getValue());
  V_entry->setTextf("%d", V_slider->getValue());
}

static void update_sliders_from_entries()
{
  R_slider->setValue(R_entry->getTextInt());
  G_slider->setValue(G_entry->getTextInt());
  B_slider->setValue(B_entry->getTextInt());

  H_slider->setValue(H_entry->getTextInt());
  S_slider->setValue(S_entry->getTextInt());
  V_slider->setValue(V_entry->getTextInt());
}

static void update_hex_entry()
{
  hex_entry->setTextf("%02x%02x%02x",
		      R_slider->getValue(),
		      G_slider->getValue(),
		      B_slider->getValue());
}

static void update_current_sprite_palette(const char* operationName)
{
  if (UIContext::instance()->get_current_sprite()) {
    try {
      CurrentSpriteWriter sprite(UIContext::instance());
      Palette* newPalette = get_current_palette(); // System current pal
      Palette* currentSpritePalette = sprite->getPalette(sprite->getCurrentFrame()); // Sprite current pal
      int from, to;

      // Check differences between current sprite palette and current system palette
      from = to = -1;
      currentSpritePalette->countDiff(newPalette, &from, &to);

      if (from >= 0 && to >= from) {
	// Add undo information to save the range of pal entries that will be modified.
	if (sprite->getUndo()->isEnabled()) {
	  sprite->getUndo()->setLabel(operationName);
	  sprite->getUndo()->undo_set_palette_colors(sprite, currentSpritePalette, from, to);
	}

	// Change the sprite palette
	sprite->setPalette(newPalette, false);
      }
    }
    catch (ase_exception& e) {
      e.show();
    }
  }

  jwidget_dirty(palette_editor);

  if (!jmanager_timer_is_running(redraw_timer_id))
    jmanager_start_timer(redraw_timer_id);
  redraw_all = false;
}

static void update_colorbar()
{
  app_get_colorbar()->dirty();
}

static void update_sliders_from_color(const Color& color)
{
  R_slider->setValue(color.getRed());
  G_slider->setValue(color.getGreen());
  B_slider->setValue(color.getBlue());
  H_slider->setValue(color.getHue());
  S_slider->setValue(color.getSaturation());
  V_slider->setValue(color.getValue());
}

static bool palette_editor_change_hook(JWidget widget, void *data)
{
  Color color = Color::fromIndex(palette_editor->get2ndColor());

  // colorviewer_set_color(colorviewer, color);

  {
    disable_colorbar_signals = true;

    if (jmouse_b(0) & 2)
      app_get_colorbar()->setBgColor(color);
    else
      app_get_colorbar()->setFgColor(color);

    disable_colorbar_signals = false;
  }

  update_sliders_from_color(color); // Update sliders
  update_entries_from_sliders();    // Update entries
  update_hex_entry();		    // Update hex field
  return false;
}

static bool select_rgb_hook(JWidget widget, void *data)
{
  R_label->setVisible(true);
  R_slider->setVisible(true);
  R_entry->setVisible(true);
  G_label->setVisible(true);
  G_slider->setVisible(true);
  G_entry->setVisible(true);
  B_label->setVisible(true);
  B_slider->setVisible(true);
  B_entry->setVisible(true);

  H_label->setVisible(false);
  H_slider->setVisible(false);
  H_entry->setVisible(false);
  S_label->setVisible(false);
  S_slider->setVisible(false);
  S_entry->setVisible(false);
  V_label->setVisible(false);
  V_slider->setVisible(false);
  V_entry->setVisible(false);

  window->setBounds(window->getBounds());
  window->dirty();

  return true;
}

static bool select_hsv_hook(JWidget widget, void *data)
{
  R_label->setVisible(false);
  R_slider->setVisible(false);
  R_entry->setVisible(false);
  G_label->setVisible(false);
  G_slider->setVisible(false);
  G_entry->setVisible(false);
  B_label->setVisible(false);
  B_slider->setVisible(false);
  B_entry->setVisible(false);

  H_label->setVisible(true);
  H_slider->setVisible(true);
  H_entry->setVisible(true);
  S_label->setVisible(true);
  S_slider->setVisible(true);
  S_entry->setVisible(true);
  V_label->setVisible(true);
  V_slider->setVisible(true);
  V_entry->setVisible(true);

  window->setBounds(window->getBounds());
  window->dirty();

  return true;
}

static bool expand_button_select_hook(JWidget widget, void *data)
{
  Size reqSize;

  if (more_options->isVisible()) {
    set_config_bool("PaletteEditor", "ShowMoreOptions", false);
    more_options->setVisible(false);

    // Get the required size of the "More options" panel
    reqSize = more_options->getPreferredSize();
    reqSize.h += 4;

    // Remove the space occupied by the "More options" panel
    {
      JRect rect = jrect_new(window->rc->x1, window->rc->y1,
			     window->rc->x2, window->rc->y2 - reqSize.h);
      window->move_window(rect);
      jrect_free(rect);
    }
  }
  else {
    set_config_bool("PaletteEditor", "ShowMoreOptions", true);
    more_options->setVisible(true);

    // Get the required size of the whole window
    reqSize = window->getPreferredSize();

    // Add space for the "more_options" panel
    if (jrect_h(window->rc) < reqSize.h) {
      JRect rect = jrect_new(window->rc->x1, window->rc->y1,
			     window->rc->x2, window->rc->y1 + reqSize.h);

      // Show the expanded area inside the screen
      if (rect->y2 > JI_SCREEN_H)
	jrect_displace(rect, 0, JI_SCREEN_H - rect->y2);
      
      window->move_window(rect);
      jrect_free(rect);
    }
    else
      window->setBounds(window->getBounds()); // TODO layout() method is missing
  }

  // Redraw the window
  window->dirty();
  return true;
}

static void modify_rgb_of_selected_entries(const Rgb& dst_rgb, bool set_r, bool set_g, bool set_b)
{
  bool array[256];
  palette_editor->getSelectedEntries(array);
  ase_uint32 src_color;
  int r, g, b;

  Palette* palette = get_current_palette();
  for (int c=0; c<256; c++) {
    if (array[c]) {
      // Get the current color entry (RGB components)
      src_color = palette->getEntry(c);

      // Setup the new RGB values depending the desired values in set_X component.
      r = (set_r ? dst_rgb.red():   _rgba_getr(src_color));
      g = (set_g ? dst_rgb.green(): _rgba_getg(src_color));
      b = (set_b ? dst_rgb.blue():  _rgba_getb(src_color));

      palette->setEntry(c, _rgba(r, g, b, 255));
    }
  }
}

static void modify_hsv_of_selected_entries(const Hsv& dst_hsv, bool set_h, bool set_s, bool set_v)
{
  bool array[256];
  palette_editor->getSelectedEntries(array);
  ase_uint32 src_color;

  Palette* palette = get_current_palette();
  for (int c=0; c<256; c++) {
    if (array[c]) {
      src_color = palette->getEntry(c);

      // Get the current RGB values of the palette entry
      Rgb rgb(_rgba_getr(src_color),
	      _rgba_getg(src_color),
	      _rgba_getb(src_color));

      // Convert RGB to HSV
      Hsv hsv(rgb);

      // Only modify the desired HSV components
      if (set_h) hsv.hue(dst_hsv.hue());
      if (set_s) hsv.saturation(dst_hsv.saturation());
      if (set_v) hsv.value(dst_hsv.value());

      // Convert HSV to RGB
      rgb = Rgb(hsv);

      // Update the palette entry
      palette->setEntry(c, _rgba(rgb.red(), rgb.green(), rgb.blue(), 255));
    }
  }
}

static void on_color_changed(const Color& color)
{
  if (disable_colorbar_signals)
    return;

  if (color.isValid() && color.getType() == Color::IndexType) {
    int index = color.getIndex();
    palette_editor->selectColor(index);

    update_sliders_from_color(color); // Update sliders
    update_entries_from_sliders();    // Update entries
    update_hex_entry();		    // Update hex field

    jwidget_flush_redraw(window);
  }
}

static void set_new_palette(Palette* palette, const char* operationName)
{
  // Copy the palette
  palette->copyColorsTo(get_current_palette());

  // Set the palette calling the hooks
  set_current_palette(palette, false);

  // Update the sprite palette
  update_current_sprite_palette(operationName);

  // Redraw the entire screen
  jmanager_refresh_screen();
}

//////////////////////////////////////////////////////////////////////
// CommandFactory

Command* CommandFactory::create_palette_editor_command()
{
  return new PaletteEditorCommand;
}