#include "SidebarToolbox.h"


void SidebarToolbox::toolAction(GtkWidget *button, Tool *tool) {
  tool->action();
}

void SidebarToolbox::sizeAllocate(GtkWidget *widget, GtkAllocation *allocation, SidebarToolbox *toolbox) {
  toolbox->updateRows(false);
}

SidebarToolbox::SidebarToolbox() {
  
  cols = 1; rows = 0;
  
  widget = gtk_table_new(1, 1, TRUE);
  g_object_ref_sink(G_OBJECT(widget));
  
  g_signal_connect(G_OBJECT(widget), "size-allocate", G_CALLBACK(sizeAllocate), this);
  
  gtk_widget_show(widget);
}

SidebarToolbox::~SidebarToolbox() {
  for(std::map<Tool*, GtkWidget*>::iterator button = buttons.begin(); button != buttons.end(); button++) {
    g_object_unref(G_OBJECT(button->second));
  }
  
  g_object_unref(G_OBJECT(widget));
}

void SidebarToolbox::addTool(Tool *tool) {
  tools.push_back(tool);
  
  GtkWidget *button = gtk_button_new();
  g_object_ref_sink(G_OBJECT(button));
  gtk_button_set_image(GTK_BUTTON(button), tool->getImage());
  gtk_widget_show(button);
  g_signal_connect(G_OBJECT(button), "clicked", G_CALLBACK(toolAction), tool);
  
  buttons.insert(std::pair<Tool*, GtkWidget*>(tool, button));
  
  updateRows(true);
  update();
}

void SidebarToolbox::removeTool(Tool *tool) {
  tools.remove(tool);
  
  g_object_unref(G_OBJECT(buttons[tool]));
  buttons.erase(tool);
  
  updateRows(true);
}

void SidebarToolbox::updateRows(bool changed) {
  int cols, rows;
  GtkRequisition requisition = {0, 0};
  
  if(tools.empty()) {
    cols = 1; rows = 0;
  }
  else {
    gtk_widget_size_request(buttons[tools.front()], &requisition);
    
    cols = MAX(1, widget->allocation.width/requisition.width);
    rows = (tools.size()+cols-1)/cols;
  }
  
  if(this->cols != cols || this->rows != rows || changed) {
    this->cols = cols;
    this->rows = rows;
    
    gtk_widget_set_size_request(widget, -1, rows*requisition.height);
    
    for(std::map<Tool*, GtkWidget*>::iterator button = buttons.begin(); button != buttons.end(); button++) {
      if(gtk_widget_get_parent(button->second))
        gtk_container_remove(GTK_CONTAINER(widget), button->second);
    }
    
    gtk_table_resize(GTK_TABLE(widget), MAX(1, cols), MAX(1, rows));
    
    int iCol = 0, iRow = 0;
    for(std::list<Tool*>::iterator tool = tools.begin(); tool != tools.end(); tool++) {
      gtk_table_attach(GTK_TABLE(widget), buttons[*tool], iCol, iCol+1, iRow, iRow+1, GTK_FILL, GTK_FILL, 0, 0);
      
      iCol++;
      if(iCol >= cols) {
        iRow++;
        iCol = 0;
      }
    }
  }
}

void SidebarToolbox::update() {
  for(std::list<Tool*>::iterator tool = tools.begin(); tool != tools.end(); tool++) {
    gtk_widget_set_tooltip_text(buttons[*tool], (*tool)->getName());
    gtk_widget_set_sensitive(buttons[*tool], (*tool)->isSensitive());
  }
}