/******************************************************************************
 * QSkinny - Copyright (C) 2016 Uwe Rathmann
 * This file may be used under the terms of the QSkinny License, Version 1.0
 *****************************************************************************/

#include "QskPageIndicatorSkinlet.h"
#include "QskPageIndicator.h"

#include "QskBoxNode.h"
#include "QskSGNode.h"

QskPageIndicatorSkinlet::QskPageIndicatorSkinlet( QskSkin* skin )
    : QskSkinlet( skin )
{
    setNodeRoles( { PanelRole, BulletsRole } );
}

QskPageIndicatorSkinlet::~QskPageIndicatorSkinlet()
{
}

QRectF QskPageIndicatorSkinlet::subControlRect( const QskSkinnable* skinnable,
    const QRectF& contentsRect, QskAspect::Subcontrol subControl ) const
{
    if ( subControl == QskPageIndicator::Panel )
    {
        return contentsRect;
    }

    return Inherited::subControlRect( skinnable, contentsRect, subControl );
}

QSGNode* QskPageIndicatorSkinlet::updateSubNode(
    const QskSkinnable* skinnable, quint8 nodeRole, QSGNode* node ) const
{
    const auto indicator = static_cast< const QskPageIndicator* >( skinnable );

    switch ( nodeRole )
    {
        case PanelRole:
        {
            return updateBoxNode( indicator, node, QskPageIndicator::Panel );
        }

        case BulletsRole:
        {
            return updateBulletsNode( indicator, node );
        }
    }

    return Inherited::updateSubNode( skinnable, nodeRole, node );
}

QRectF QskPageIndicatorSkinlet::bulletRect(
    const QskPageIndicator* indicator, const QRectF& rect, int index ) const
{
    using Q = QskPageIndicator;

    const qreal szNormal = indicator->strutSizeHint( Q::Bullet ).width();
    const qreal szHighlighted = indicator->strutSizeHint( Q::Highlighted ).width();

    const auto currentIndex = indicator->currentIndex();

    // scale bullet size if we are in between a transition:
    qreal indexDiff = qAbs( currentIndex - index );
    if ( indexDiff > ( indicator->count() - 1 ) )
        indexDiff = ( indicator->count() - currentIndex ); // wrapping

    const qreal sz0 = ( indexDiff < 1 ) ?
        ( 1 - indexDiff ) * szHighlighted + indexDiff * szNormal : szNormal;

    const qreal spacing = indicator->spacingHint( Q::Panel );
    const bool horizontal = ( indicator->orientation() == Qt::Horizontal );

    qreal w, h;
    if ( horizontal )
    {
        w = ( indicator->count() - 1 ) * ( szNormal + spacing ) + szHighlighted;
        h = rect.height();
    }
    else
    {
        w = rect.width();
        h = ( indicator->count() - 1 ) * ( szNormal + spacing ) + szHighlighted;
    }

    QRectF r( 0, 0, w, h );
    r.moveCenter( rect.center() );

    qreal s2;

    {
        const qreal s = ( index > currentIndex ) ? szHighlighted : szNormal;
        if ( indexDiff < 1 && index >= currentIndex )
        {
            // scrolling from or to this bullet:
            s2 = szNormal + qAbs( szHighlighted - szNormal ) * indexDiff;
        }
        else if ( ( currentIndex > ( indicator->count() - 1 ) &&
            index > ( currentIndex - indicator->count() + 1 ) ) )
        {
            // wrapping case:
            qreal wrappingDiff = indexDiff;
            while ( wrappingDiff > 1 )
                wrappingDiff -= 1;

            s2 = szNormal + qAbs( szHighlighted - szNormal ) * wrappingDiff;
        }
        else
        {
            s2 = s;
        }
    }

    const qreal x = r.left() + s2 + spacing + ( index - 1 ) * ( szNormal + spacing );

    qreal adjust = ( currentIndex == index )
        ? ( szNormal - szHighlighted ) : ( szHighlighted - szNormal );
    adjust = 0.5 * qMax( 0.0, adjust );

    if ( indexDiff < 1 )
        adjust *= indexDiff;

    QRectF bulletRect( 0.0, 0.0, sz0, sz0 );

    if ( horizontal )
        bulletRect.moveTo( x, r.top() + adjust );
    else
        bulletRect.moveTo( r.left() + adjust, x );
    
    return bulletRect;
}

QSGNode* QskPageIndicatorSkinlet::updateBulletsNode(
    const QskPageIndicator* indicator, QSGNode* node ) const
{
    if ( indicator->count() == 0 )
        return nullptr;

    if ( node == nullptr )
        node = new QSGNode();

    QRectF rect = indicator->subControlRect( QskPageIndicator::Panel );
    rect = indicator->innerBox( QskPageIndicator::Panel, rect );

    // index of the highlighted bullet
    int currentBullet = qRound( indicator->currentIndex() );
    if ( currentBullet >= indicator->count() )
        currentBullet = 0;

    auto bulletNode = node->firstChild();
    for ( int i = 0; i < indicator->count(); i++ )
    {
        using Q = QskPageIndicator;

        if ( i > 0 )
            bulletNode = bulletNode->nextSibling();

        if ( bulletNode == nullptr )
            bulletNode = new QskBoxNode();

        updateBoxNode( indicator, bulletNode, bulletRect( indicator, rect, i ),
            ( i == currentBullet ) ? Q::Highlighted : Q::Bullet );

        if ( bulletNode->parent() != node )
            node->appendChildNode( bulletNode );
    }

    // if count has decreased we need to remove superfluous nodes
    QskSGNode::removeAllChildNodesAfter( node, bulletNode );

    return node;
}

QSizeF QskPageIndicatorSkinlet::sizeHint( const QskSkinnable* skinnable,
    Qt::SizeHint which, const QSizeF& ) const
{
    using Q = QskPageIndicator;

    if ( which != Qt::PreferredSize )
        return QSizeF();

    const auto indicator = static_cast< const QskPageIndicator* >( skinnable );

    const auto bulletSize = indicator->strutSizeHint( Q::Bullet );

    const auto maxSize = bulletSize.expandedTo(
        indicator->strutSizeHint( Q::Highlighted ) );

    const qreal spacing = indicator->spacingHint( Q::Panel );

    const int n = indicator->count();

    qreal w = 0;
    qreal h = 0;

    if ( indicator->orientation() == Qt::Horizontal )
    {
        if ( n > 0 )
        {
            w += maxSize.width();

            if ( n > 1 )
                w += ( n - 1 ) * ( bulletSize.width() + spacing );
        }

        h = maxSize.height();
    }
    else
    {
        if ( n > 0 )
        {
            h += maxSize.height();

            if ( n > 1 )
                h += ( n - 1 ) * ( bulletSize.height() + spacing );
        }

        w = maxSize.width();
    }

    const auto hint = indicator->outerBoxSize( Q::Panel, QSizeF( w, h ) );
    return hint.expandedTo( indicator->strutSizeHint( Q::Panel ) );
}

#include "moc_QskPageIndicatorSkinlet.cpp"