/****************************************************************************
 * libwiigui
 *
 * Tantric 2009
 *
 * gui_button.cpp
 *
 * GUI class definitions
 ***************************************************************************/

#include "gui.h"

static int scrollison;

/**
 * Constructor for the GuiButton class.
 */

GuiButton::GuiButton( int w, int h )
{
    width = w;
    height = h;
    image = NULL;
    imageOver = NULL;
    imageHold = NULL;
    imageClick = NULL;
    icon = NULL;
    iconOver = NULL;
    iconHold = NULL;
    iconClick = NULL;
    toolTip = NULL;

    for ( int i = 0; i < 3; i++ )
    {
        label[i] = NULL;
        labelOver[i] = NULL;
        labelHold[i] = NULL;
        labelClick[i] = NULL;
    }

    soundOver = NULL;
    soundHold = NULL;
    soundClick = NULL;
    selectable = true;
    holdable = false;
    clickable = true;

    time1 = time2 = 0;
}

GuiButton::GuiButton( GuiImage* img, GuiImage* imgOver, int hor, int vert, int x, int y, GuiTrigger* trig, GuiSound* sndOver, GuiSound* sndClick, u8 grow )
{
    width = img->GetWidth();
    height = img->GetHeight();
    image = img;
    image->SetParent( this );
    imageOver = imgOver;
    if ( imageOver ) imageOver->SetParent( this );
    imageHold = NULL;
    imageClick = NULL;
    icon = NULL;
    iconOver = NULL;
    iconHold = NULL;
    iconClick = NULL;
    toolTip = NULL;
    alignmentHor = hor;
    alignmentVert = vert;
    xoffset = x;
    yoffset = y;
    trigger[0] = trig;

    for ( int i = 0; i < 3; i++ )
    {
        label[i] = NULL;
        labelOver[i] = NULL;
        labelHold[i] = NULL;
        labelClick[i] = NULL;
    }

    soundOver = sndOver;
    soundHold = NULL;
    soundClick = sndClick;
    selectable = true;
    holdable = false;
    clickable = true;

    if ( grow == 1 )
    {
        effectsOver |= EFFECT_SCALE;
        effectAmountOver = 4;
        effectTargetOver = 110;
    }
    time1 = time2 = 0;
}

GuiButton::GuiButton( GuiImage* img, GuiImage* imgOver, int hor, int vert, int x, int y, GuiTrigger* trig, GuiSound* sndOver, GuiSound* sndClick, u8 grow, GuiTooltip* tt, int ttx, int tty, int h_align, int v_align )
{
    width = img->GetWidth();
    height = img->GetHeight();
    image = img;
    image->SetParent( this );
    imageOver = imgOver;
    if ( imageOver ) imageOver->SetParent( this );
    imageHold = NULL;
    imageClick = NULL;
    icon = NULL;
    iconOver = NULL;
    iconHold = NULL;
    iconClick = NULL;
    toolTip = NULL;
    alignmentHor = hor;
    alignmentVert = vert;
    xoffset = x;
    yoffset = y;
    trigger[0] = trig;

    for ( int i = 0; i < 3; i++ )
    {
        label[i] = NULL;
        labelOver[i] = NULL;
        labelHold[i] = NULL;
        labelClick[i] = NULL;
    }

    soundOver = sndOver;
    soundHold = NULL;
    soundClick = sndClick;
    selectable = true;
    holdable = false;
    clickable = true;

    if ( grow == 1 )
    {
        effectsOver |= EFFECT_SCALE;
        effectAmountOver = 4;
        effectTargetOver = 110;
    }

    toolTip = tt;
    toolTip->SetParent( this );
    toolTip->SetAlignment( h_align, v_align );
    toolTip->SetPosition( ttx, tty );

    time1 = time2 = 0;
}

/**
 * Destructor for the GuiButton class.
 */
GuiButton::~GuiButton()
{
}

void GuiButton::SetImage( GuiImage* img )
{
    LOCK( this );
    image = img;
    if ( img ) img->SetParent( this );
}
void GuiButton::SetImageOver( GuiImage* img )
{
    LOCK( this );
    imageOver = img;
    if ( img ) img->SetParent( this );
}
void GuiButton::SetImageHold( GuiImage* img )
{
    LOCK( this );
    imageHold = img;
    if ( img ) img->SetParent( this );
}
void GuiButton::SetImageClick( GuiImage* img )
{
    LOCK( this );
    imageClick = img;
    if ( img ) img->SetParent( this );
}
void GuiButton::SetIcon( GuiImage* img )
{
    LOCK( this );
    icon = img;
    if ( img ) img->SetParent( this );
}
void GuiButton::SetIconOver( GuiImage* img )
{
    LOCK( this );
    iconOver = img;
    if ( img ) img->SetParent( this );
}
void GuiButton::SetIconHold( GuiImage* img )
{
    LOCK( this );
    iconHold = img;
    if ( img ) img->SetParent( this );
}
void GuiButton::SetIconClick( GuiImage* img )
{
    LOCK( this );
    iconClick = img;
    if ( img ) img->SetParent( this );
}
void GuiButton::SetLabel( GuiText* txt, int n )
{
    LOCK( this );
    label[n] = txt;
    if ( txt ) txt->SetParent( this );
}
void GuiButton::SetLabelOver( GuiText* txt, int n )
{
    LOCK( this );
    labelOver[n] = txt;
    if ( txt ) txt->SetParent( this );
}
void GuiButton::SetLabelHold( GuiText* txt, int n )
{
    LOCK( this );
    labelHold[n] = txt;
    if ( txt ) txt->SetParent( this );
}
void GuiButton::SetLabelClick( GuiText* txt, int n )
{
    LOCK( this );
    labelClick[n] = txt;
    if ( txt ) txt->SetParent( this );
}
void GuiButton::SetSoundOver( GuiSound * snd )
{
    LOCK( this );
    soundOver = snd;
}
void GuiButton::SetSoundHold( GuiSound * snd )
{
    LOCK( this );
    soundHold = snd;
}
void GuiButton::SetSoundClick( GuiSound * snd )
{
    LOCK( this );
    soundClick = snd;
}

void GuiButton::SetToolTip( GuiTooltip* tt, int x, int y, int h_align, int v_align )
{
    LOCK( this );
    if ( tt )
    {
        toolTip = tt;
        toolTip->SetParent( this );
        toolTip->SetAlignment( h_align, v_align );
        toolTip->SetPosition( x, y );

    }
}

void GuiButton::RemoveToolTip()
{
    LOCK( this );
    toolTip = NULL;
}

void GuiButton::RemoveSoundOver()
{
    LOCK( this );
    soundOver = NULL;
}
void GuiButton::RemoveSoundClick()
{
    LOCK( this );
    soundClick = NULL;
}
void GuiButton::SetSkew( int XX1, int YY1, int XX2, int YY2, int XX3, int YY3, int XX4, int YY4 )
{
    if ( image )
    {
        image->xx1 = XX1;
        image->yy1 = YY1;
        image->xx2 = XX2;
        image->yy2 = YY2;
        image->xx3 = XX3;
        image->yy3 = YY3;
        image->xx4 = XX4;
        image->yy4 = YY4;
    }
}

void GuiButton::SetSkew( int *skew )
{
    if ( image )
        image->SetSkew( skew );
}


/**
 * Draw the button on screen
 */
void GuiButton::Draw()
{
    LOCK( this );
    if ( !this->IsVisible() )
        return;

    // draw image
    if ( ( state == STATE_SELECTED || state == STATE_HELD ) && imageOver )
        imageOver->Draw();
    else if ( image )
        image->Draw();
    // draw icon
    if ( ( state == STATE_SELECTED || state == STATE_HELD ) && iconOver )
        iconOver->Draw();
    else if ( icon )
        icon->Draw();
    // draw text
    for ( int i = 0; i < 3; i++ )
    {
        if ( ( state == STATE_SELECTED || state == STATE_HELD ) && labelOver[i] )
            labelOver[i]->Draw();
        else if ( label[i] )
            label[i]->Draw();
    }

    this->UpdateEffects();
}
void GuiButton::DrawTooltip()
{
    LOCK( this );
    if ( this->IsVisible() && state == STATE_SELECTED && toolTip )
    {
        if ( time2 == 0 )
        {
            time( &time1 );
            time2 = time1;
        }
        if ( time1 != 0 )  // timer läuft
            time( &time1 );

        if ( time1 == 0 || difftime( time1, time2 ) >= 2 )
        {
            if ( time1 != 0 )  // timer gerade abgelaufen
                toolTip->SetEffect( EFFECT_FADE, 20 );
            time1 = 0;
            toolTip->Draw();
            return;
        }
    }
    else
    {
        if ( time2 != 0 && time1 == 0 ) // timer abgelaufen, gerade DESELECT
            if ( toolTip ) toolTip->SetEffect( EFFECT_FADE, -20 );
        time2 = 0;
    }
    if ( toolTip && toolTip->GetEffect() )
        toolTip->Draw();
}
void GuiButton::ScrollIsOn( int f )
{
    LOCK( this );
    scrollison = f;
}
void GuiButton::Update( GuiTrigger * t )
{
    LOCK( this );
    if ( !this->IsVisible() || state == STATE_CLICKED || state == STATE_DISABLED || !t )
        return;
    else if ( parentElement && parentElement->GetState() == STATE_DISABLED )
        return;

#ifdef HW_RVL
    // cursor
    if ( t->wpad.ir.valid )
    {
        if ( this->IsInside( t->wpad.ir.x, t->wpad.ir.y ) )
        {
            if ( state == STATE_DEFAULT ) // we weren't on the button before!
            {
                if ( scrollison == 0 )
                {
                    this->SetState( STATE_SELECTED, t->chan );
                }

                if ( this->Rumble() && scrollison == 0 )
                    rumbleRequest[t->chan] = 1;

                if ( soundOver && scrollison == 0 )
                    soundOver->Play();

                if ( effectsOver && !effects && scrollison == 0 )
                {
                    // initiate effects
                    effects = effectsOver;
                    effectAmount = effectAmountOver;
                    effectTarget = effectTargetOver;
                }
            }
        }
        else
        {
            if ( state == STATE_SELECTED && ( stateChan == t->chan || stateChan == -1 ) )
                this->ResetState();

            if ( effectTarget == effectTargetOver && effectAmount == effectAmountOver )
            {
                // initiate effects (in reverse)
                effects = effectsOver;
                effectAmount = -effectAmountOver;
                effectTarget = 100;
            }
        }
    }
#else

    if ( state == STATE_SELECTED && ( stateChan == t->chan || stateChan == -1 ) )
        this->ResetState();

    if ( effectTarget == effectTargetOver && effectAmount == effectAmountOver )
    {
        // initiate effects (in reverse)
        effects = effectsOver;
        effectAmount = -effectAmountOver;
        effectTarget = 100;
    }


#endif

    // button triggers
    if ( this->IsClickable() && scrollison == 0 )
    {
        s32 wm_btns, wm_btns_trig, cc_btns, cc_btns_trig;
        for ( int i = 0; i < 6; i++ )
        {
            if ( trigger[i] && ( trigger[i]->chan == -1 || trigger[i]->chan == t->chan ) )
            {
                // higher 16 bits only (wiimote)
                wm_btns = t->wpad.btns_d << 16;
                wm_btns_trig = trigger[i]->wpad.btns_d << 16;

                // lower 16 bits only (classic controller)
                cc_btns = t->wpad.btns_d >> 16;
                cc_btns_trig = trigger[i]->wpad.btns_d >> 16;

                if (
                    ( ( t->wpad.btns_d > 0 &&
                        wm_btns == wm_btns_trig ) ||
                      ( cc_btns == cc_btns_trig && t->wpad.exp.type == EXP_CLASSIC ) ) ||
                    ( t->pad.btns_d == trigger[i]->pad.btns_d && t->pad.btns_d > 0 ) )
                {
                    if ( t->chan == stateChan || stateChan == -1 )
                    {
                        if ( state == STATE_SELECTED )
                        {
                            this->SetState( STATE_CLICKED, t->chan );

                            if ( soundClick )
                                soundClick->Play();
                        }
                        else if ( trigger[i]->type == TRIGGER_BUTTON_ONLY )
                        {
                            this->SetState( STATE_CLICKED, t->chan );
                            if ( soundClick )
                                soundClick->Play();
                        }
                        else if ( trigger[i]->type == TRIGGER_BUTTON_ONLY_IN_FOCUS &&
                                  parentElement->IsFocused() )
                        {
                            this->SetState( STATE_CLICKED, t->chan );
                            if ( soundClick )
                                soundClick->Play();
                        }
                    }
                }
            }
        }
    }

    if ( this->IsHoldable() )
    {
        bool held = false;
        s32 wm_btns, wm_btns_h, wm_btns_trig, cc_btns, cc_btns_h, cc_btns_trig;

        for ( int i = 0; i < 6; i++ )
        {
            if ( trigger[i] && ( trigger[i]->chan == -1 || trigger[i]->chan == t->chan ) )
            {
                // higher 16 bits only (wiimote)
                wm_btns = t->wpad.btns_d << 16;
                wm_btns_h = t->wpad.btns_h << 16;
                wm_btns_trig = trigger[i]->wpad.btns_h << 16;

                // lower 16 bits only (classic controller)
                cc_btns = t->wpad.btns_d >> 16;
                cc_btns_h = t->wpad.btns_h >> 16;
                cc_btns_trig = trigger[i]->wpad.btns_h >> 16;

                if (
                    ( ( t->wpad.btns_d > 0 &&
                        wm_btns == wm_btns_trig ) ||
                      ( cc_btns == cc_btns_trig && t->wpad.exp.type == EXP_CLASSIC ) ) ||
                    ( t->pad.btns_d == trigger[i]->pad.btns_h && t->pad.btns_d > 0 ) )
                {
                    if ( trigger[i]->type == TRIGGER_HELD && state == STATE_SELECTED &&
                            ( t->chan == stateChan || stateChan == -1 ) )
                        this->SetState( STATE_CLICKED, t->chan );
                }

                if (
                    ( ( t->wpad.btns_h > 0 &&
                        wm_btns_h == wm_btns_trig ) ||
                      ( cc_btns_h == cc_btns_trig && t->wpad.exp.type == EXP_CLASSIC ) ) ||
                    ( t->pad.btns_h == trigger[i]->pad.btns_h && t->pad.btns_h > 0 ) )
                {
                    if ( trigger[i]->type == TRIGGER_HELD )
                        held = true;
                }

                if ( !held && state == STATE_HELD && stateChan == t->chan )
                {
                    this->ResetState();
                }
                else if ( held && state == STATE_CLICKED && stateChan == t->chan )
                {
                    this->SetState( STATE_HELD, t->chan );
                }
            }
        }
    }

    if ( updateCB )
        updateCB( this );
}