/*
 * Copyright 2004-2020, Haiku, Inc.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Adrien Destugues, pulkomandy@pulkomandy.tk
 *		Michael Wilber
 */


#include "IconView.h"

#include <new>
#include <stdio.h>
#include <string.h>

#include <ControlLook.h>
#include <Entry.h>
#include <IconUtils.h>
#include <Node.h>
#include <NodeInfo.h>


using std::nothrow;


IconView::IconView(icon_size iconSize)
	:
	BView("IconView", B_WILL_DRAW),
	fIconSize(iconSize),
	fIconBitmap(NULL),
	fDrawIcon(false)
{
	SetViewUIColor(B_PANEL_BACKGROUND_COLOR);
}


IconView::~IconView()
{
	delete fIconBitmap;
	fIconBitmap = NULL;
}


status_t
IconView::SetIcon(const BPath& path, icon_size iconSize)
{
	fDrawIcon = false;

	if (iconSize != fIconSize || fIconBitmap == NULL) {
		BBitmap* bitmap = new BBitmap(BRect(BPoint(0, 0),
			be_control_look->ComposeIconSize(iconSize)), B_RGBA32);
		if (bitmap == NULL)
			return B_NO_MEMORY;

		delete fIconBitmap;
		fIconBitmap = bitmap;
		fIconSize = iconSize;
	}

	status_t status = fIconBitmap->InitCheck();
	if (status != B_OK)
		return status;

	BEntry entry(path.Path());
	BNode node(&entry);
	BNodeInfo info(&node);

	status = info.GetTrackerIcon(fIconBitmap,
		(icon_size)fIconBitmap->Bounds().IntegerWidth());
	if (status != B_OK)
		return status;

	if (!fIconBitmap->IsValid())
		return fIconBitmap->InitCheck();

	_SetSize();

	fDrawIcon = true;
	Invalidate();
	return B_OK;
}


status_t
IconView::SetIcon(const uint8_t* data, size_t size, icon_size iconSize)
{
	fDrawIcon = false;

	if (iconSize != fIconSize || fIconBitmap == NULL) {
		BBitmap* bitmap = new BBitmap(BRect(BPoint(0, 0),
			be_control_look->ComposeIconSize(iconSize)), B_RGBA32);
		if (bitmap == NULL)
			return B_NO_MEMORY;

		delete fIconBitmap;
		fIconBitmap = bitmap;
		fIconSize = iconSize;
	}

	status_t status = fIconBitmap->InitCheck();
	if (status != B_OK)
		return status;

	status = BIconUtils::GetVectorIcon(data, size, fIconBitmap);
	if (status != B_OK)
		return status;

	if (!fIconBitmap->IsValid())
		return fIconBitmap->InitCheck();

	_SetSize();

	fDrawIcon = true;
	Invalidate();
	return B_OK;
}


status_t
IconView::SetIcon(const BBitmap* icon)
{
	if (icon == NULL) {
		fDrawIcon = false;
		return B_OK;
	}

	delete fIconBitmap;
	fIconBitmap = new BBitmap(icon);
	if (fIconBitmap == NULL)
		return B_NO_MEMORY;

	status_t status = fIconBitmap->InitCheck();
	if (status != B_OK)
		return status;

	fIconSize = (icon_size)-1;
	_SetSize();

	fDrawIcon = true;
	Invalidate();
	return B_OK;
}


void
IconView::DrawIcon(bool draw)
{
	if (draw == fDrawIcon)
		return;

	fDrawIcon = draw;
	Invalidate();
}


void
IconView::Draw(BRect area)
{
	if (fDrawIcon && fIconBitmap != NULL) {
		SetDrawingMode(B_OP_ALPHA);
		SetBlendingMode(B_PIXEL_ALPHA, B_ALPHA_OVERLAY);
		DrawBitmap(fIconBitmap);
		SetDrawingMode(B_OP_COPY);
	} else
		BView::Draw(area);
}


status_t
IconView::InitCheck() const
{
	if (fIconBitmap == NULL)
		return B_NO_MEMORY;

	return fIconBitmap->InitCheck();
}


void
IconView::_SetSize()
{
	SetExplicitMinSize(fIconBitmap->Bounds().Size());
	SetExplicitMaxSize(fIconBitmap->Bounds().Size());
	SetExplicitPreferredSize(fIconBitmap->Bounds().Size());
}