#include "DrawButton.h"

DrawButton::DrawButton(BRect frame, const char *name, BBitmap *up, BBitmap *down,
						BMessage *msg, int32 resize, int32 flags)
	: BButton(frame, name, "", msg, resize, flags)
{
	fUp=up;
	fDown=down;
	fDisabled=NULL;
}

DrawButton::~DrawButton(void)
{
	delete fUp;
	delete fDown;
	delete fDisabled;
}

void
DrawButton::SetBitmaps(BBitmap *up, BBitmap *down)
{
	delete fUp;
	delete fDown;
	
	fUp=up;
	fDown=down;
	
	if(IsEnabled())
		Invalidate();
}

void
DrawButton::SetDisabled(BBitmap *disabled)
{
	delete fDisabled;
	
	fDisabled=disabled;
	
	if(!IsEnabled())
		Invalidate();
}

void
DrawButton::Draw(BRect update)
{
	if(!IsEnabled())
	{
		if(fDisabled)
			DrawBitmap(fDisabled, BPoint(0,0));
		else
			StrokeRect(Bounds());
		return;
	}
	
	if(Value() == B_CONTROL_ON)
	{
		if(fDown)
			DrawBitmap(fDown, BPoint(0,0));
		else
			StrokeRect(Bounds());
	}
	else
	{
		if(fUp)
			DrawBitmap(fUp, BPoint(0,0));
		else
			StrokeRect(Bounds());
	}
}

void
DrawButton::ResizeToPreferred(void)
{
	if(fUp)
		ResizeTo(fUp->Bounds().Width(),fUp->Bounds().Height());
	else
	if(fDown)
		ResizeTo(fDown->Bounds().Width(),fDown->Bounds().Height());
	else
	if(fDisabled)
		ResizeTo(fDisabled->Bounds().Width(),fDisabled->Bounds().Height());
}