/*
 * Copyright 2005, Jérôme Duval. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Inspired by SoundCapture from Be newsletter (Media Kit Basics: Consumers 
 * and Producers)
 */
#include "UpDownButton.h"
#include "icon_button.h"

UpDownButton::UpDownButton(BRect _rect, BMessage *msg, uint32 resizeFlags)
        : BControl(_rect, "button", NULL, msg, resizeFlags, B_WILL_DRAW),
        fLastValue(B_CONTROL_ON)
{
        BRect rect = BRect(0, 0, kUpDownButtonWidth - 1, kUpDownButtonHeight - 1);
        fBitmapUp = new BBitmap(rect, B_CMAP8);
        fBitmapUp->SetBits(kButtonUpBits, kUpDownButtonWidth * kUpDownButtonHeight,
                0, B_CMAP8);
        fBitmapDown = new BBitmap(rect, B_CMAP8);
        fBitmapDown->SetBits(kButtonDownBits, kUpDownButtonWidth 
                * kUpDownButtonHeight, 0, B_CMAP8);
        fBitmapMiddle = new BBitmap(rect, B_CMAP8);
        fBitmapMiddle->SetBits(kButtonMiddleBits, kUpDownButtonWidth 
                * kUpDownButtonHeight, 0, B_CMAP8);
}


UpDownButton::~UpDownButton()
{
        delete fBitmapUp;
        delete fBitmapDown;
        delete fBitmapMiddle;
}


void
UpDownButton::Draw(BRect updateRect)
{
        SetDrawingMode(B_OP_OVER);
        
        if(IsTracking()) {
                if((Bounds().top + Bounds().Height()/2) > (fTrackingY + 3))
                        DrawBitmap(fBitmapUp);
                else if((Bounds().top + Bounds().Height()/2) < (fTrackingY - 3))
                        DrawBitmap(fBitmapDown);
                else
                        DrawBitmap(fBitmapMiddle);
        } else {
                if(Value()==B_CONTROL_OFF)
                        DrawBitmap(fBitmapUp);
                else
                        DrawBitmap(fBitmapDown);        
        }
        
        SetDrawingMode(B_OP_COPY);
}


void
UpDownButton::MouseDown(BPoint point)
{
        if(!IsEnabled())
                return;
        
        fLastValue = Value();
        fTrackingY = (Bounds().top + Bounds().Height()/2);
        
        SetTracking(true);
        SetMouseEventMask(B_POINTER_EVENTS, B_LOCK_WINDOW_FOCUS);
        
        SetValue(Value() == B_CONTROL_ON ? B_CONTROL_OFF : B_CONTROL_ON);       
}


void 
UpDownButton::MouseMoved(BPoint point, uint32 transit, const BMessage *message)
{
        if (!IsTracking())
                return;
                
        fTrackingY = point.y;
        Draw(Bounds());
        Flush();
}


void
UpDownButton::MouseUp(BPoint point)
{
        if (!IsTracking())
                return;
        
        if((Bounds().top + Bounds().Height()/2) > (fTrackingY + 3))
                SetValue(B_CONTROL_ON);
        else if((Bounds().top + Bounds().Height()/2) < (fTrackingY - 3))
                SetValue(B_CONTROL_OFF);
        
        if(Value()!=fLastValue)
                Invoke();
        SetTracking(false);
        Draw(Bounds());
        Flush();
        fLastValue = Value();
}