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


#include "SymbolButton.h"

#include <GradientLinear.h>
#include <LayoutUtils.h>
#include <Shape.h>


static const rgb_color kGreen = (rgb_color){ 116, 224, 0, 255 };


// constructor
SymbolButton::SymbolButton(const char* name, BShape* symbolShape,
                BMessage* message, uint32 borders)
        :
        BButton(name, NULL, message),
        fSymbol(symbolShape),
        fBorders(borders)
{
}


SymbolButton::~SymbolButton()
{
        delete fSymbol;
}


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

        if (fBorders != 0) {
                be_control_look->DrawButtonFrame(this, bounds, updateRect, base,
                        base, flags & ~BControlLook::B_DISABLED, fBorders);
                be_control_look->DrawButtonBackground(this, bounds, updateRect, base,
                        flags);
        }

        if (fSymbol == NULL)
                return;

        if (IsEnabled()) {
                if (Value() == B_CONTROL_ON)
                        base = tint_color(base, (B_DARKEN_4_TINT + B_DARKEN_MAX_TINT) / 2);
                else
                        base = tint_color(base, B_DARKEN_4_TINT);
        } else {
                if (Value() == B_CONTROL_ON)
                        base = tint_color(base, B_DARKEN_2_TINT);
                else
                        base = tint_color(base, B_DARKEN_1_TINT);
        }

        BPoint offset;
        offset.x = (bounds.left + bounds.right) / 2;
        offset.y = (bounds.top + bounds.bottom) / 2;
        offset.x -= fSymbol->Bounds().Width() / 2;
        offset.y -= fSymbol->Bounds().Height() / 2;
        offset.x = floorf(offset.x - fSymbol->Bounds().left);
        offset.y = ceilf(offset.y - fSymbol->Bounds().top);

        MovePenTo(offset);
        BGradientLinear gradient;
        gradient.AddColor(tint_color(base, B_DARKEN_1_TINT), 0);
        gradient.AddColor(base, 255);
        gradient.SetStart(offset);
        offset.y += fSymbol->Bounds().Height();
        gradient.SetEnd(offset);
        FillShape(fSymbol, gradient);
}


BSize
SymbolButton::MinSize()
{
        if (fSymbol == NULL)
                return BButton::MinSize();

        float scale = fBorders != 0 ? 2.5f : 1.0f;

        BSize size;
        size.width = ceilf(fSymbol->Bounds().Width() * scale);
        size.height = ceilf(fSymbol->Bounds().Height() * scale);
        return BLayoutUtils::ComposeSize(ExplicitMinSize(), size);
}


BSize
SymbolButton::MaxSize()
{
        BSize size(MinSize());
        if (fBorders != 0)
                size.width = ceilf(size.width * 1.5f);
        return BLayoutUtils::ComposeSize(ExplicitMaxSize(), size);
}


void
SymbolButton::SetSymbol(BShape* symbolShape)
{
        BSize oldSize = MinSize();

        delete fSymbol;
        fSymbol = symbolShape;

        if (MinSize() != oldSize)
                InvalidateLayout();

        Invalidate();
}