/*
 * Copyright 2010 Stephan Aßmus <superstippi@gmx.de>. All rights reserved.
 * Distributed under the terms of the MIT License.
 */

#include "BitmapButton.h"

#include <string.h>

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


static const float kFrameInset = 2;


BBitmapButton::BBitmapButton(const char* resourceName, BMessage* message)
        :
        BButton("", message),
        fBitmap(BTranslationUtils::GetBitmap(resourceName)),
        fBackgroundMode(BUTTON_BACKGROUND)
{
}


BBitmapButton::BBitmapButton(const uint8* bits, uint32 width, uint32 height,
                color_space format, BMessage* message)
        :
        BButton("", message),
        fBitmap(new BBitmap(BRect(0, 0, width - 1, height - 1), 0, format)),
        fBackgroundMode(BUTTON_BACKGROUND)
{
        memcpy(fBitmap->Bits(), bits, fBitmap->BitsLength());
}


BBitmapButton::~BBitmapButton()
{
        delete fBitmap;
}


BSize
BBitmapButton::MinSize()
{
        BSize min(0, 0);
        if (fBitmap) {
                min.width = fBitmap->Bounds().Width();
                min.height = fBitmap->Bounds().Height();
        }
        min.width += kFrameInset * 2;
        min.height += kFrameInset * 2;
        return min;
}


BSize
BBitmapButton::MaxSize()
{
        return BSize(B_SIZE_UNLIMITED, B_SIZE_UNLIMITED);
}


BSize
BBitmapButton::PreferredSize()
{
        return MinSize();
}


void
BBitmapButton::Draw(BRect updateRect)
{
        BRect bounds(Bounds());
        rgb_color base = ui_color(B_PANEL_BACKGROUND_COLOR);
        uint32 flags = be_control_look->Flags(this);

        if (fBackgroundMode == BUTTON_BACKGROUND || Value() == B_CONTROL_ON) {
                be_control_look->DrawButtonBackground(this, bounds, updateRect, base,
                        flags);
        } else {
                SetHighColor(tint_color(base, B_DARKEN_2_TINT));
                StrokeLine(bounds.LeftBottom(), bounds.RightBottom());
                bounds.bottom--;
                be_control_look->DrawMenuBarBackground(this, bounds, updateRect, base,
                        flags);
        }

        if (fBitmap == NULL)
                return;

        SetDrawingMode(B_OP_ALPHA);

        if (!IsEnabled()) {
                SetBlendingMode(B_CONSTANT_ALPHA, B_ALPHA_OVERLAY);
                SetHighColor(0, 0, 0, 120);
        }

        BRect bitmapBounds(fBitmap->Bounds());
        BPoint bitmapLocation(
                floorf((bounds.left + bounds.right
                        - (bitmapBounds.left + bitmapBounds.right)) / 2 + 0.5f),
                floorf((bounds.top + bounds.bottom
                        - (bitmapBounds.top + bitmapBounds.bottom)) / 2 + 0.5f));

        DrawBitmap(fBitmap, bitmapLocation);
}


void
BBitmapButton::SetBackgroundMode(uint32 mode)
{
        if (fBackgroundMode != mode) {
                fBackgroundMode = mode;
                Invalidate();
        }
}