#include <rtgui/dc.h>
#include <rtgui/rtgui_theme.h>
#include <rtgui/widgets/checkbox.h>

static void _rtgui_checkbox_constructor(rtgui_checkbox_t *box)
{
	/* init widget and set event handler */
	RTGUI_WIDGET(box)->flag |= RTGUI_WIDGET_FLAG_FOCUSABLE;
	rtgui_object_set_event_handler(RTGUI_OBJECT(box), rtgui_checkbox_event_handler);

	/* set status */
	box->status_down = RTGUI_CHECKBOX_STATUS_UNCHECKED;
	box->on_button = RT_NULL;

	/* set default gc */
	RTGUI_WIDGET_TEXTALIGN(RTGUI_WIDGET(box)) = RTGUI_ALIGN_LEFT | RTGUI_ALIGN_CENTER_VERTICAL;
}

DEFINE_CLASS_TYPE(checkbox, "checkbox", 
	RTGUI_LABEL_TYPE,
	_rtgui_checkbox_constructor,
	RT_NULL,
	sizeof(struct rtgui_checkbox));

void rtgui_checkbox_set_onbutton(rtgui_checkbox_t* checkbox, rtgui_onbutton_func_t func)
{
	RT_ASSERT(checkbox != RT_NULL);

	checkbox->on_button = func;
}

rt_bool_t rtgui_checkbox_event_handler(struct rtgui_object* object, struct rtgui_event* event)
{
	struct rtgui_widget *widget;
	struct rtgui_checkbox *box = (struct rtgui_checkbox*)widget;

	RT_ASSERT(widget != RT_NULL);
	RT_ASSERT(event != RT_NULL);

	widget = RTGUI_WIDGET(object);
	box = RTGUI_CHECKBOX(object);

	switch (event->type)
	{
	case RTGUI_EVENT_PAINT:
#ifndef RTGUI_USING_SMALL_SIZE
		if (widget->on_draw != RT_NULL)
		{
			return widget->on_draw(RTGUI_OBJECT(widget), event);
		}
		else
#endif
			rtgui_theme_draw_checkbox(box);
		break;

	case RTGUI_EVENT_MOUSE_BUTTON:
		{
			if (RTGUI_WIDGET_IS_ENABLE(widget) && !RTGUI_WIDGET_IS_HIDE(widget))
			{
				struct rtgui_event_mouse* emouse = (struct rtgui_event_mouse*)event;
				if (emouse->button & RTGUI_MOUSE_BUTTON_LEFT &&
					emouse->button & RTGUI_MOUSE_BUTTON_UP)
				{
					/* set focus */
					rtgui_widget_focus(widget);

					if (box->status_down & RTGUI_CHECKBOX_STATUS_UNCHECKED)
					{
						/* check it */
						box->status_down = RTGUI_CHECKBOX_STATUS_CHECKED;
					}
					else
					{
						/* un-check it */
						box->status_down = RTGUI_CHECKBOX_STATUS_UNCHECKED;
					}
				}

				/* draw checkbox */
				rtgui_theme_draw_checkbox(box);

#ifndef RTGUI_USING_SMALL_SIZE
				/* call user callback */
				if (widget->on_mouseclick != RT_NULL)
				{
					return widget->on_mouseclick(RTGUI_OBJECT(widget), event);
				}
#endif
				if (box->on_button != RT_NULL)
				{
					box->on_button(RTGUI_OBJECT(widget), event);
					return RT_TRUE;
				}
			}

			return RT_TRUE;
		}
	}

	return RT_FALSE;
}

struct rtgui_checkbox* rtgui_checkbox_create(const char* text, rt_bool_t checked)
{
    struct rtgui_checkbox* box;

    box = (struct rtgui_checkbox*) rtgui_widget_create (RTGUI_CHECKBOX_TYPE);
    if (box != RT_NULL)
    {
		rtgui_rect_t rect;

		/* set default rect */
		rtgui_font_get_metrics(rtgui_font_default(), text, &rect);
		rect.x2 += RTGUI_BORDER_DEFAULT_WIDTH + 5 + (RTGUI_BORDER_DEFAULT_WIDTH << 1);
		rect.y2 += (RTGUI_BORDER_DEFAULT_WIDTH << 1);

		rtgui_widget_set_rect(RTGUI_WIDGET(box), &rect);
		rtgui_label_set_text(RTGUI_LABEL(box), text);
		
		if (checked == RT_TRUE)
			box->status_down = RTGUI_CHECKBOX_STATUS_CHECKED;
		else
			box->status_down = RTGUI_CHECKBOX_STATUS_UNCHECKED;
	}

	return box;
}

void rtgui_checkbox_destroy(rtgui_checkbox_t* box)
{
	rtgui_widget_destroy(RTGUI_WIDGET(box));
}

void rtgui_checkbox_set_checked(rtgui_checkbox_t* checkbox, rt_bool_t checked)
{
	RT_ASSERT(checkbox != RT_NULL);
	if (checked == RT_TRUE)
		checkbox->status_down = RTGUI_CHECKBOX_STATUS_CHECKED;
	else
		checkbox->status_down = RTGUI_CHECKBOX_STATUS_UNCHECKED;
		
}

rt_bool_t rtgui_checkbox_get_checked(rtgui_checkbox_t* checkbox)
{
	RT_ASSERT(checkbox != RT_NULL);

	if (checkbox->status_down == RTGUI_CHECKBOX_STATUS_CHECKED)
		return RT_TRUE;

	return RT_FALSE;
}