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

#include "IconOptionsControl.h"

#include <stdio.h>

#include <Window.h>

#include "IconButton.h"

#define LABEL_DIST 8.0

// constructor
IconOptionsControl::IconOptionsControl(const char* name,
                                                                           const char* label,
                                                                           BMessage* message,
                                                                           BHandler* target)
        : MView(),
          BControl(BRect(0.0, 0.0, 10.0, 10.0), name, label, message,
                           B_FOLLOW_NONE, B_WILL_DRAW | B_FRAME_EVENTS),
          fTargetCache(target)
{
        if (Label()) {
                labelwidth = StringWidth(Label()) + LABEL_DIST;
        } else {
                labelwidth = 0.0;
        }
}

// destructor
IconOptionsControl::~IconOptionsControl()
{
}

// layoutprefs
minimax
IconOptionsControl::layoutprefs()
{
        // sanity checks
        if (rolemodel)
                labelwidth = rolemodel->LabelWidth();
        if (labelwidth < LabelWidth())
                labelwidth = LabelWidth();

        mpm.mini.x = 0.0 + labelwidth + 5.0;
        mpm.mini.y = 0.0;

        for (int32 i = 0; IconButton* button = _FindIcon(i); i++) {
                minimax childPrefs = button->layoutprefs();
                mpm.mini.x += childPrefs.mini.x + 1;
                if (mpm.mini.y < childPrefs.mini.y + 1)
                        mpm.mini.y = childPrefs.mini.y + 1;
        }

        mpm.maxi.x = mpm.mini.x;
        mpm.maxi.y = mpm.mini.y;

        mpm.weight = 1.0;

        return mpm;
}

// layout
BRect
IconOptionsControl::layout(BRect frame)
{
        if (frame.Width() < mpm.mini.x + 1)
                frame.right = frame.left + mpm.mini.x + 1;

        MoveTo(frame.LeftTop());
        ResizeTo(frame.Width(), frame.Height());

        return Frame();
}

// LabelWidth
float
IconOptionsControl::LabelWidth()
{
        float width = ceilf(StringWidth(Label()));
        if (width > 0.0)
                width += LABEL_DIST;
        return width;
}

// SetLabel
void
IconOptionsControl::SetLabel(const char* label)
{
        BControl::SetLabel(label);
        float width = LabelWidth();
        if (rolemodel)
                labelwidth = rolemodel->LabelWidth() > labelwidth ?
                                                rolemodel->LabelWidth() : labelwidth;

        labelwidth = width > labelwidth ? width : labelwidth;

        _TriggerRelayout();
}

// AttachedToWindow
void
IconOptionsControl::AttachedToWindow()
{
        BControl::AttachedToWindow();

        SetViewColor(B_TRANSPARENT_32_BIT);
        SetLowColor(ui_color(B_PANEL_BACKGROUND_COLOR));
}

// AllAttached
void
IconOptionsControl::AllAttached()
{
        for (int32 i = 0; IconButton* button = _FindIcon(i); i++)
                button->SetTarget(this);
        if (fTargetCache)
                SetTarget(fTargetCache);
}

// Draw
void
IconOptionsControl::Draw(BRect updateRect)
{
        FillRect(updateRect, B_SOLID_LOW);

        if (Label()) {
                if (!IsEnabled())
                        SetHighColor(tint_color(LowColor(), B_DISABLED_LABEL_TINT));
                else
                        SetHighColor(tint_color(LowColor(), B_DARKEN_MAX_TINT));

                font_height fh;
                GetFontHeight(&fh);
                BPoint p(Bounds().LeftTop());
                p.y += floorf(Bounds().Height() / 2.0 + (fh.ascent + fh.descent) / 2.0) - 2.0;
                DrawString(Label(), p);
        }
}

// FrameResized
void
IconOptionsControl::FrameResized(float width, float height)
{
        _LayoutIcons(Bounds());
}

// SetValue
void
IconOptionsControl::SetValue(int32 value)
{
        if (IconButton* valueButton = _FindIcon(value)) {
                for (int32 i = 0; IconButton* button = _FindIcon(i); i++) {
                        button->SetPressed(button == valueButton);
                }
        }
}

// Value
int32
IconOptionsControl::Value() const
{
        for (int32 i = 0; IconButton* button = _FindIcon(i); i++) {
                if (button->IsPressed())
                        return i;
        }
        return -1;
}

// SetEnabled
void
IconOptionsControl::SetEnabled(bool enable)
{
        for (int32 i = 0; IconButton* button = _FindIcon(i); i++) {
                button->SetEnabled(enable);
        }
        BControl::SetEnabled(enable);
        Invalidate();
}

// MessageReceived
void
IconOptionsControl::MessageReceived(BMessage* message)
{
        // catch a message from the attached IconButtons to
        // handle switching the pressed icon
        BView* source;
        if (message->FindPointer("be:source", (void**)&source) >= B_OK) {
                if (IconButton* sourceIcon = dynamic_cast<IconButton*>(source)) {
                        for (int32 i = 0; IconButton* button = _FindIcon(i); i++) {
                                if (button == sourceIcon) {
                                        SetValue(i);
                                        break;
                                }
                        }
                        // forward the message
                        Invoke(message);
                        return;
                }
        }
        BControl::MessageReceived(message);
}

// Invoke
status_t
IconOptionsControl::Invoke(BMessage* message)
{
        return BInvoker::Invoke(message);
}

// AddOption
void
IconOptionsControl::AddOption(IconButton* icon)
{
        if (icon) {
                if (!_FindIcon(0)) {
                        // first icon added, mark it
                        icon->SetPressed(true);
                }
                AddChild(icon);
                icon->SetTarget(this);
                layoutprefs();
                _TriggerRelayout();
        }
}

// _FindIcon
IconButton*
IconOptionsControl::_FindIcon(int32 index) const
{
        if (BView* view = ChildAt(index))
                return dynamic_cast<IconButton*>(view);
        return NULL;
}

// _TriggerRelayout
void
IconOptionsControl::_TriggerRelayout()
{
        if (!Parent())
                return;

        MView* mParent = dynamic_cast<MView*>(Parent());
        if (mParent) {
                if (BWindow* window = Window()) {
                        window->PostMessage(M_RECALCULATE_SIZE);
                }
        } else {
                _LayoutIcons(Bounds());
        }
}

// _LayoutIcons
void
IconOptionsControl::_LayoutIcons(BRect frame)
{
        BPoint lt = frame.LeftTop();

        // sanity checks
        if (rolemodel)
                labelwidth = rolemodel->LabelWidth();
        if (labelwidth < LabelWidth())
                labelwidth = LabelWidth();

        lt.x += labelwidth;

        for (int32 i = 0; IconButton* button = _FindIcon(i); i++) {
                if (i == 0) {
                        lt.y = ceilf((frame.top + frame.bottom - button->mpm.mini.y) / 2.0);
                }
                button->MoveTo(lt);
                button->ResizeTo(button->mpm.mini.x, button->mpm.mini.y);
                lt = button->Frame().RightTop() + BPoint(1.0, 0.0);
        }
}