/***************************************************************************
  
  gbutton.cpp
  
  (c) 2004-2006 - Daniel Campos Fernández <dcamposf@gmail.com>

  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, 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., 51 Franklin Street, Fifth Floor, Boston,
  MA 02110-1301, USA.
  
***************************************************************************/

#include "widgets.h"
#include "gapplication.h"
#include "gmainwindow.h"
#include "gdesktop.h"
#include "gbutton.h"

#include <unistd.h>

static void cb_click(GtkButton *object, gButton *data)
{
	if (data->disable)
	{
		data->disable=false;
		return;
	}
		
	data->unsetOtherRadioButtons();

	if (data->type == gButton::Tool)
	{
		if (!data->isToggle())
		{
			data->disable = true;
			gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(data->widget), false);
		}
	}

	CB_button_click(data);
}

static void cb_click_radio(GtkButton *object, gControl *data)
{
	if (gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(object)))
		CB_button_click(data);
	return;
}

static void cb_click_check(GtkButton *object, gButton *data)
{
	if (data->isTristate() && !data->locked())
	{
		data->lock();
		if (data->inconsistent())
		{
			data->setInconsistent(false);
			data->setValue(false);
		}
		else if (!data->value())
			data->setInconsistent(true);
		data->unlock();
	}

	CB_button_click(data);
}

#ifdef GTK3
static gboolean button_draw(GtkWidget *wid, cairo_t *cr, gButton *data)
#else
static gboolean button_expose(GtkWidget *wid, GdkEventExpose *e, gButton *data)
#endif
{
	GdkPixbuf *img;
	GdkRectangle rect;
	GtkCellRendererState state;
	bool rtl;
	int x, w, wt, wp, hp;
	int d = gDesktop::scale() / 2;
	gColor fg = GB_COLOR_DEFAULT;
	int size;
	
	rtl = gtk_widget_get_direction(wid) == GTK_TEXT_DIR_RTL;
	
	#ifdef GTK3
		GtkStateFlags f = gtk_widget_get_state_flags(data->widget);
		rect.x = rect.y = 0;
		rect.width = data->width();
		rect.height = data->height();
	#else
		GtkStateType f = (GtkStateType)GTK_WIDGET_STATE(data->widget);
		rect = wid->allocation;
	#endif

#ifdef GTK3
	if (data->getBorder() && (f & GTK_STATE_FLAG_ACTIVE))
#else
	if (data->getBorder() && (f == GTK_STATE_ACTIVE))
#endif
	{
		int d = 1 + gDesktop::scale() / 16;
		rect.x += d;
		rect.y += d;
	}

	wp = 0;
	wt = 0;
	
	if (data->hasText())
	{
		gt_set_cell_renderer_text_from_font((GtkCellRendererText *)data->rendtxt, data->font());

		fg = data->realForeground();

		if (fg == GB_COLOR_DEFAULT)
		{
			g_object_set(G_OBJECT(data->rendtxt),
				"foreground-set", FALSE,
				(void *)NULL);
		}
		else
		{
#ifdef GTK3
			GdkRGBA col;
			gt_from_color(fg, &col);
			g_object_set(G_OBJECT(data->rendtxt),
				"foreground-set", TRUE,
				"foreground-rgba", &col,
				(void *)NULL);
#else
			GdkColor col;
			fill_gdk_color(&col, fg);
			g_object_set(G_OBJECT(data->rendtxt),
				"foreground-set", TRUE,
				"foreground-gdk", &col,
				(void *)NULL);
#endif
		}

		wt = data->font()->width(data->text(), strlen(data->text())) + 4;
	}
	
	if (data->rendpix)
	{
		#ifdef GTK3
		if (f & GTK_STATE_FLAG_INSENSITIVE)
		#else
		if (f == GTK_STATE_INSENSITIVE)
		#endif
		{
		  if (!data->rendinc)
		    data->rendinc = gt_pixbuf_create_disabled(data->rendpix);
		  img = data->rendinc;
    }
		else
		  img = data->rendpix;

		wp = gdk_pixbuf_get_width(img);
		hp = gdk_pixbuf_get_height(img);
		
		size = Min(data->width(), data->height()) - d * 2;
		if (size <= 0)
			wp = hp = 0;
		else if (Max(wp, hp) > size)
		{
			if (wp > hp)
			{
				wp = size * wp / hp;
				hp = size;
			}
			else
			{
				hp = size * hp / wp;
				wp = size;
			}
		}
	}

	w = wt + wp;
	if (wt && wp)
		w += d;
	
	if (wp)
	{
		x = rect.x + (rect.width - w) / 2;
		if (rtl && wt)
			x += wt + d;

		#ifndef GTK3
			cairo_t *cr = gdk_cairo_create(wid->window);
			gdk_cairo_region(cr, e->region);
			cairo_clip(cr);
		#endif
		
		gt_cairo_draw_pixbuf(cr, img, x, rect.y + (rect.height - hp) / 2, wp, hp, 1.0, NULL);
		
		#ifndef GTK3
			cairo_destroy(cr);
		#endif
	}
	
	if (wt)
	{
		x = rect.x + (rect.width - w) / 2;
		if (!rtl && wp)
			x += wp + d;
		
		#ifdef GTK3
		
			g_object_set(G_OBJECT(data->rendtxt), "sensitive", !(f & GTK_STATE_INSENSITIVE), (void *)NULL);
			
			if (fg != GB_COLOR_DEFAULT)
				state = (GtkCellRendererState)0;
			else if (f & GTK_STATE_SELECTED)
				state = GTK_CELL_RENDERER_SELECTED;
			else if (f & GTK_STATE_INSENSITIVE)
				state = GTK_CELL_RENDERER_INSENSITIVE;
			else
				state = (GtkCellRendererState)0;

		#else
			
			g_object_set(G_OBJECT(data->rendtxt), "sensitive", true, (void *)NULL);
			
			switch (f)
			{
				case GTK_STATE_PRELIGHT: 
					state = GTK_CELL_RENDERER_PRELIT;
					break;
				//case GTK_STATE_PRELIGHT: state=GTK_CELL_RENDERER_PRELIT; break;
				case GTK_STATE_SELECTED: 
					state = GTK_CELL_RENDERER_SELECTED; 
					break;
				
				case GTK_STATE_INSENSITIVE: 
					state = GTK_CELL_RENDERER_INSENSITIVE; 
					g_object_set(G_OBJECT(data->rendtxt), "sensitive", false, (void *)NULL); 
					break;
					
				default:
					state = (GtkCellRendererState)0; 
					break;
			}
			
		#endif

		rect.x = x;
		rect.width = wt;
		gtk_cell_renderer_set_fixed_size(data->rendtxt, rect.width, rect.height);
		#ifdef GTK3
			gtk_cell_renderer_render(data->rendtxt, cr, wid, &rect, &rect, state);
		#else
			gtk_cell_renderer_render(data->rendtxt, wid->window, wid, &rect, &rect, &e->area, state);
		#endif
	}

	return FALSE;
}

#ifdef GTK3
static void cb_state(GtkWidget *widget, GtkStateFlags state, gButton *data)
#else
static void cb_state(GtkWidget *widget, GtkStateType state, gButton *data)
#endif
{
	data->refresh();
}


gButton::gButton(gContainer *par, Type typ) : gControl(par)
{
	gContainer *ct;

	disable = false;
	_toggle = false;
	_radio = false;
	_animated = false;
	_stretch = true;
	_tristate = false;
	_autoresize = false;
	bufText = NULL;
	rendtxt = NULL;
	rendpix = NULL;
	rendinc = NULL;
	_label = NULL;
	pic = NULL;
	shortcut = 0;
	_is_button = TRUE;
	
	switch(typ)
	{
		case Check:
			border = gtk_check_button_new();
			break;
			
		case Radio:
			ct = parent();
			if (!ct->radiogroup) 
			{
				ct->radiogroup = gtk_radio_button_new(NULL);
				g_object_ref(ct->radiogroup);
				border = gtk_radio_button_new_from_widget(GTK_RADIO_BUTTON(ct->radiogroup));
				gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(border),true);
			}
			else 
			{
				border = gtk_radio_button_new_from_widget(GTK_RADIO_BUTTON(ct->radiogroup));
			}
			break;
		
		case Toggle:
			_no_background = true;
			rendtxt = gtk_cell_renderer_text_new();
			border = gtk_toggle_button_new();
			break;

		case Tool:
			_no_background = true;
			rendtxt = gtk_cell_renderer_text_new();
			border = gtk_toggle_button_new();
			gt_set_focus_on_click(border, false);
			break;
		
		default:
			_no_background = true;
			border = gtk_button_new();
			rendtxt = gtk_cell_renderer_text_new();
			typ = Button;
			break;
	}

  widget = border;
	
  type = typ;

	if (rendtxt) 
	{
		g_object_set(G_OBJECT(rendtxt), "xalign", 0.5, "yalign", 0.5, "xpad", 0, "ypad", 0, NULL);
		ON_DRAW(widget, this, button_expose, button_draw);
	}
	
	realize();
	
	gtk_widget_add_events(widget, GDK_POINTER_MOTION_MASK);
	
	if (type == Radio)
		g_signal_connect(G_OBJECT(widget),"clicked",G_CALLBACK(cb_click_radio),(gpointer)this);
	else if (type == Check)
		g_signal_connect(G_OBJECT(widget), "clicked", G_CALLBACK(cb_click_check), (gpointer)this);	
	else
	{
		g_signal_connect(G_OBJECT(widget),"clicked",G_CALLBACK(cb_click),(gpointer)this);	
		setColorButton();
	}
	
	#ifdef GTK3
		g_signal_connect(G_OBJECT(widget), "state-flags-changed", G_CALLBACK(cb_state), (gpointer)this);	
	#else
		g_signal_connect(G_OBJECT(widget), "state-changed", G_CALLBACK(cb_state), (gpointer)this);	
	#endif
	
	setText(NULL);

	if (type == Tool) 
    setBorder(false);
}

gButton::~gButton()
{
	setDefault(false);
	setCancel(false);
  setPicture(0);
  g_free(bufText);  
}

void gButton::setInconsistent(bool vl)
{
	if (type != Check) return;

	gtk_toggle_button_set_inconsistent (GTK_TOGGLE_BUTTON(widget),vl);
}

bool gButton::inconsistent() const
{
	gboolean vl=false;

	if (type != Check) return false;
	g_object_get (G_OBJECT(widget),"inconsistent",&vl,(void *)NULL);
	return vl;	
}

void gButton::setText(const char *st)
{
	GtkAccelGroup *accel;
	char *buf;
	
	accel = window()->accel;
		
	if (bufText) 
	{
		if (shortcut)
			gtk_widget_remove_accelerator(widget, accel, (guint)shortcut, GDK_MOD1_MASK);
		g_free(bufText);
	}

 	bufText = st ? g_strdup(st) : NULL;

	if (rendtxt)
	{
		if (hasText())
		{
			shortcut = (int)gMnemonic_correctMarkup(bufText, &buf);

			if (shortcut)
				gtk_widget_add_accelerator(widget, "clicked", accel, (guint)shortcut, GDK_MOD1_MASK, (GtkAccelFlags)0);

			if (rendtxt)
				g_object_set(G_OBJECT(rendtxt), "markup", buf, (void *)NULL);

			g_free(buf);
		}
		else
		{
			g_object_set(G_OBJECT(rendtxt), "markup", "", (void *)NULL);
		}

		refresh();
	}
	else
	{
		if (hasText())
		{
			gMnemonic_correctText((char*)st, &buf);
			gtk_button_set_use_underline(GTK_BUTTON(widget), TRUE);
			gtk_button_set_label(GTK_BUTTON(widget), buf);
			g_free(buf);
		}
		else
			gtk_button_set_label(GTK_BUTTON(widget), "");

		_label = gtk_bin_get_child(GTK_BIN(widget));
		updateDirection();
		#ifndef GTK3
		set_gdk_fg_color(_label, realForeground());
		#endif
	}

	updateFont();
}


gPicture* gButton::picture() const
{
	if ( (type == Check) || (type == Radio) ) 
    return NULL;
	
	return pic;	
}

void gButton::setPicture(gPicture *npic)
{
	GdkPixbuf *new_rendpix = NULL;
	
	if ((type == Check) || (type == Radio)) return;
	
	gPicture::assign(&pic, npic);
	
	if (pic)
	{
   	new_rendpix = pic->getPixbuf();
    if (new_rendpix)
    	g_object_ref(new_rendpix);
  }
  
	if (rendpix) { g_object_unref(G_OBJECT(rendpix)); rendpix = NULL; }
	if (rendinc) { g_object_unref(G_OBJECT(rendinc)); rendinc = NULL; }

	rendpix = new_rendpix;
	
	updateSize();
	refresh();
}

bool gButton::getBorder() const
{
	switch(gtk_button_get_relief(GTK_BUTTON(widget)))
	{
		case GTK_RELIEF_NORMAL:
		case GTK_RELIEF_HALF:
			return true;
		default: 
			return false;
	}
}

void gButton::setBorder(bool vl)
{
	gtk_button_set_relief (GTK_BUTTON(widget), vl ? GTK_RELIEF_NORMAL : GTK_RELIEF_NONE);
}

bool gButton::isDefault() const
{
	gMainWindow *win = window();	
	return win ? win->_default == this : false;
}

void gButton::setDefault(bool vl)
{
	gMainWindow *win = window();
	
	if (type != Button || !win)
		return;
	
	if (vl)
	{
		win->_default = this;
#if GTK_CHECK_VERSION(2, 18, 0)
		gtk_widget_set_can_default(widget, true);
#else
		GTK_WIDGET_SET_FLAGS (widget, GTK_CAN_DEFAULT);
#endif
		//gtk_widget_grab_default (widget);
	}
	else
	{
#if GTK_CHECK_VERSION(2, 18, 0)
		gtk_widget_set_can_default(widget, false);
#else
		GTK_WIDGET_UNSET_FLAGS(widget, GTK_CAN_DEFAULT);
#endif
		if (win->_default == this)
			win->_default = NULL;
	}
}

bool gButton::isCancel() const
{
	gMainWindow *win = window();	
	return win ? win->_cancel == this : false;
}

void gButton::setCancel(bool vl)
{
	gMainWindow *win = window();
	
	if (type != Button || !win) 
    return;
	
	if (vl)
		win->_cancel = this;
	else if (win->_cancel == this)
		win->_cancel = NULL;
}

bool gButton::value() const
{
  if (type == Button)
    return false;
  else
		return gtk_toggle_button_get_active (GTK_TOGGLE_BUTTON(widget));
}

void gButton::setValue(bool vl)
{
  if (type == Button)
  {
    if (vl) gtk_button_clicked(GTK_BUTTON(widget));
  }
  else
    gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(widget), vl);
}

void gButton::setToggle(bool vl)
{
	if (type != Tool) return;
	_toggle = vl;
}

bool gButton::isToggle() const
{
	return type == Toggle || type == Check || type == Radio || _toggle;
}

void gButton::animateClick(bool on)
{
	if (type != Button) return;
	
	if (!on && !_animated)
	{
#ifdef GTK3
		gtk_widget_set_state_flags(widget, GTK_STATE_FLAG_ACTIVE, FALSE);
#else
		gtk_widget_set_state(widget, GTK_STATE_ACTIVE);
#endif
		refresh();
		_animated = true;
	}
	else if (on && _animated)
	{
		_animated = false;
#ifdef GTK3
		gtk_widget_unset_state_flags(widget, GTK_STATE_FLAG_ACTIVE);
#else
		gtk_widget_set_state(widget, GTK_STATE_NORMAL);
#endif
		refresh();
		gtk_button_clicked(GTK_BUTTON(widget));
	}
}

int gButton::autoHeight() const
{
	int mh = 0;
	
	if (hasText())
	{
		if (type == Button || type == Toggle || type == Tool)
			mh = font()->height() + gDesktop::scale();
		else
			mh = font()->height() + 2;
	}
	
	if (pic && (pic->height() > mh))
		mh = pic->height();
	
	return mh;
}

void gButton::setRadio(bool vl)
{
	_radio = vl;
	if (value())
		unsetOtherRadioButtons();
}

bool gButton::isRadio() const
{
	return type == Radio || _radio;
}

void gButton::unsetOtherRadioButtons()
{
	gContainer *pr = parent();
	gControl *child;
	gButton *button;
	int i;
	
	if (type == Radio || type == Button || !isRadio() || !isToggle())
		return;
	
	for (i = 0; i < pr->childCount(); i++)
	{
		child = pr->child(i);
		if (!child->isButton())
			continue;
			
		button = (gButton *)child;
		
		if (button == this)
		{
			if (!value())
			{
				button->disable = true;
				button->setValue(true);
			}
		}
		else if (button->type == type && button->isRadio() && button->isToggle() && button->value())
		{
			button->disable = true;
			button->setValue(false);
		}
	}
}

bool gButton::hasShortcut() const
{
	return isDefault() || isCancel() || shortcut;
}


void gButton::setStretch(bool vl)
{
	_stretch = vl;
}

void gButton::setRealForeground(gColor color)
{
	gControl::setRealForeground(color);

#ifndef GTK3
	if (_label)
		set_gdk_fg_color(_label, color);
#endif
}

void gButton::setTristate(bool vl)
{
	_tristate = vl;
	if (!_tristate)
		setInconsistent(false);
}

void gButton::setAutoResize(bool vl)
{
	_autoresize = vl;
	updateSize();
}

void gButton::updateSize()
{
	int mw, mh;
	
	if (!_autoresize)
		return;
	
	mh = autoHeight();
	mw = 0;
	
	if (hasText())
	{
		if (type == Check || type == Radio)
		{
#ifdef GTK3
			int indicator_size, indicator_spacing, focus_width, focus_pad;
			gtk_widget_style_get(widget,
				"indicator-size", &indicator_size,
				"indicator-spacing", &indicator_spacing,
				"focus-line-width", &focus_width,
        "focus-padding", &focus_pad,
				(char *)NULL);
			mw += (indicator_spacing + indicator_size + indicator_spacing + 2 * (focus_width + focus_pad)) + font()->width(bufText, strlen(bufText));
#else
			GtkRequisition req;
			g_signal_emit_by_name(border, "size-request",	&req);
			mw += req.width;
#endif
		}
		else
			mw += font()->width(bufText, strlen(bufText)) + gDesktop::scale();
	}
	
	if (pic)
	{
		if (mw) mw += gDesktop::scale() / 2;
		mw += pic->width();
	}
	
	mw += gDesktop::scale();
	
	if (mh < height())
		mh = height();

	resize(mw, mh);
}

void gButton::updateDirection()
{
	gControl::updateDirection();
	if (_label)
		gtk_widget_set_direction(_label, gtk_widget_get_direction(widget));
}

gColor gButton::defaultBackground() const
{
	if (type == Check || type == Radio)
		return gControl::defaultBackground();
	else
		return gDesktop::getColor(COLOR_BUTTON_BACKGROUND, !isEnabled());
}
