/*
 * Copyright 2000, Georges-Edouard Berenger. All rights reserved.
 * Copyright 2022, Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 */
#include "IconMenuItem.h"

#include <ControlLook.h>
#include <Bitmap.h>


IconMenuItem::IconMenuItem(BBitmap* icon, const char* title,
	BMessage* msg, bool drawText, bool purge)
	:
	BMenuItem(title, msg),
	fIcon(icon),
	fDrawText(drawText),
	fPurge(purge)
{
}


IconMenuItem::IconMenuItem(BBitmap* icon, BMenu* menu, bool drawText, bool purge)
	:
	BMenuItem(menu),
	fIcon(icon),
	fDrawText(drawText),
	fPurge(purge)
{
}


IconMenuItem::~IconMenuItem()
{
	if (fPurge)
		delete fIcon;
}


void
IconMenuItem::Reset(BBitmap* icon, bool purge)
{
	if (fPurge)
		delete fIcon;

	fPurge = purge;
	fIcon = icon;
}


void
IconMenuItem::DrawContent()
{
	DrawIcon();

	if (fDrawText) {
		BPoint loc = ContentLocation();
		loc.x += ceilf(be_control_look->DefaultLabelSpacing() * 3.3f);
		Menu()->MovePenTo(loc);
		BMenuItem::DrawContent();
	}
}


void
IconMenuItem::Highlight(bool hilited)
{
	BMenuItem::Highlight(hilited);
	DrawIcon();
}


void
IconMenuItem::DrawIcon()
{
	if (fIcon == NULL)
		return;

	BPoint loc = ContentLocation();
	BRect frame = Frame();

	loc.y = frame.top + (frame.bottom - frame.top - fIcon->Bounds().Height()) / 2;

	BMenu* menu = Menu();

	if (fIcon->ColorSpace() == B_RGBA32) {
		menu->SetDrawingMode(B_OP_ALPHA);
		menu->SetBlendingMode(B_PIXEL_ALPHA, B_ALPHA_OVERLAY);
	} else
		menu->SetDrawingMode(B_OP_OVER);

	menu->DrawBitmap(fIcon, loc);

	menu->SetDrawingMode(B_OP_COPY);
}


void
IconMenuItem::GetContentSize(float* width, float* height)
{
	BMenuItem::GetContentSize(width, height);
	if (fIcon == NULL)
		return;

	const float limit = ceilf(fIcon->Bounds().Height() +
		(be_control_look->DefaultLabelSpacing() / 3.0f));
	if (*height < limit)
		*height = limit;
	if (fDrawText)
		*width += fIcon->Bounds().Width() + be_control_look->DefaultLabelSpacing();
	else
		*width = fIcon->Bounds().Width() + 1;
}