/*****************************************************************
 KWin - the KDE window manager
 This file is part of the KDE project.

Copyright (C) 2007 Philip Falkner <philip.falkner@gmail.com>

You can Freely distribute this program under the GNU General Public
License. See the file "COPYING" for the exact licensing terms.
******************************************************************/

#include "boxswitch.h"

#include "tabbox.h"
#include "client.h"
#include "scene_opengl.h"
#include <QSize>
#include <QPainter>
#include <QPixmap>

#ifdef HAVE_OPENGL
#include <GL/gl.h>
#endif

namespace KWinInternal
{

BoxSwitchEffect::BoxSwitchEffect()
    : mActivated( 0 )
    , mMode( 0 )
    , painting_desktop( 0 )
    {
    frame_margin = 10;
    highlight_margin = 5;
    }

BoxSwitchEffect::~BoxSwitchEffect()
    {
    }

void BoxSwitchEffect::prePaintScreen( int* mask, QRegion* region, int time )
    {
    effects->prePaintScreen( mask, region, time );
    }

void BoxSwitchEffect::prePaintWindow( EffectWindow* w, int* mask, QRegion* region, int time )
    {
    if( mActivated )
        {
        if( mMode == TabBox::WindowsMode )
            {
            if( windows.contains( w ) && w != selected_window )
                {
                *mask |= Scene::PAINT_WINDOW_TRANSLUCENT;
                *mask &= ~Scene::PAINT_WINDOW_OPAQUE;
                }
            }
        else
            {
            if( painting_desktop )
                {
                if( w->isOnDesktop( painting_desktop ))
                    w->enablePainting( Scene::Window::PAINT_DISABLED_BY_DESKTOP );
                else
                    w->disablePainting( Scene::Window::PAINT_DISABLED_BY_DESKTOP );
                }
            }
        }
    effects->prePaintWindow( w, mask, region, time );
    }

void BoxSwitchEffect::paintScreen( int mask, QRegion region, ScreenPaintData& data )
    {
    effects->paintScreen( mask, region, data );
    if( mActivated )
        {
        if( mMode == TabBox::WindowsMode )
            {
            paintFrame();

            foreach( EffectWindow* w, windows.keys())
                {
                if( w == selected_window )
                    {
                    paintHighlight( windows[ w ].area,
                        static_cast< Client* >( w->window())->caption());
                    }
                paintWindowThumbnail( w );
                }
            }
        else
            {
            if( !painting_desktop )
                {
                paintFrame();

                for( painting_desktop = 1; painting_desktop <= desktops.count(); painting_desktop++ )
                    {
                    if( painting_desktop == selected_desktop )
                        {
                        paintHighlight( desktops[ painting_desktop ].area,
                            Workspace::self()->desktopName( painting_desktop ));
                        }

                    paintDesktopThumbnail( painting_desktop );
                    }
                painting_desktop = 0;
                }
            }
        }
    }

void BoxSwitchEffect::paintWindow( EffectWindow* w, int mask, QRegion region, WindowPaintData& data )
    {
    if( mActivated )
        {
        if( mMode == TabBox::WindowsMode )
            {
            if( windows.contains( w ) && w != selected_window )
                {
                data.opacity *= 0.1;
                }
            }
        }
    effects->paintWindow( w, mask, region, data );
    }

void BoxSwitchEffect::windowInputMouseEvent( Window w, QEvent* e )
    {
    assert( w == mInput );
    if( e->type() != QEvent::MouseButtonPress )
        return;
    QPoint pos = static_cast< QMouseEvent* >( e )->pos();
    pos += frame_area.topLeft();

    // determine which item was clicked
    if( mMode == TabBox::WindowsMode )
        {
        foreach( EffectWindow* w, windows.keys())
            {
            if( windows[ w ].clickable.contains( pos ))
                {
                if( Client* c = static_cast< Client* >( w->window()))
                    {
                    Workspace::self()->activateClient( c );
                    if( c->isShade() && options->shadeHover )
                        c->setShade( ShadeActivated );
                    Workspace::self()->unrefTabBox();
                    setInactive();
                    }
                }
            }
        }
    else
        {
        for( int i = 1; i <= desktops.count(); i++ )
            {
            if( desktops[ i ].clickable.contains( pos ))
                {
                Workspace::self()->setCurrentDesktop( i );
                Workspace::self()->unrefTabBox();
                setInactive();
                }
            }
        }
    }

void BoxSwitchEffect::windowDamaged( EffectWindow* w, const QRect& damage )
    {
    if( mActivated )
        {
        if( mMode == TabBox::WindowsMode )
            {
            if( windows.contains( w ))
                {
                workspace()->addRepaint( windows[ w ].area );
                }
            }
        else
            {
            if( w->isOnAllDesktops())
                {
                foreach( ItemInfo info, desktops )
                    workspace()->addRepaint( info.area );
                }
            else
                {
                workspace()->addRepaint( desktops[ w->desktop() ].area );
                }
            }
        }
    }

void BoxSwitchEffect::windowGeometryShapeChanged( EffectWindow* w, const QRect& old )
    {
    if( mActivated )
        {
        if( mMode == TabBox::WindowsMode )
            {
            if( windows.contains( w ) && w->size() != old.size())
                {
                workspace()->addRepaint( windows[ w ].area );
                }
            }
        else
            {
            if( w->isOnAllDesktops())
                {
                foreach( ItemInfo info, desktops )
                    workspace()->addRepaint( info.area );
                }
            else
                {
                workspace()->addRepaint( desktops[ w->desktop() ].area );
                }
            }
        }
    }

void BoxSwitchEffect::tabBoxAdded( int mode )
    {
    if( !mActivated )
        {
        if( mode == TabBox::WindowsMode )
            {
            if( Workspace::self()->currentTabBoxClientList().count() > 1 )
                {
                mMode = mode;
                Workspace::self()->refTabBox();
                setActive();
                }
            }
        else
            { // DesktopMode or DesktopListMode
            if( Workspace::self()->numberOfDesktops() > 1 )
                {
                mMode = mode;
                painting_desktop = 0;
                Workspace::self()->refTabBox();
                setActive();
                }
            }
        }
    }

void BoxSwitchEffect::tabBoxClosed()
    {
    if( mActivated )
        setInactive();
    }

void BoxSwitchEffect::tabBoxUpdated()
    {
    if( mActivated )
        {
        if( mMode == TabBox::WindowsMode )
            {
            if( selected_window != NULL )
                {
                if( windows.contains( selected_window ))
                    workspace()->addRepaint( windows.value( selected_window ).area );
                selected_window->window()->addRepaintFull();
                }
            selected_window = Workspace::self()->currentTabBoxClient()->effectWindow();
            if( windows.contains( selected_window ))
                workspace()->addRepaint( windows.value( selected_window ).area );
            selected_window->window()->addRepaintFull();
            if( Workspace::self()->currentTabBoxClientList() == tab_clients )
                return;
            tab_clients = Workspace::self()->currentTabBoxClientList();
            }
        else
            {
            if( desktops.contains( selected_desktop ))
                workspace()->addRepaint( desktops.value( selected_desktop ).area );
            selected_desktop = Workspace::self()->currentTabBoxDesktop();
            if( desktops.contains( selected_desktop ))
                workspace()->addRepaint( desktops.value( selected_desktop ).area );
            if( Workspace::self()->numberOfDesktops() == desktops.count())
                return;
            }
        workspace()->addRepaint( frame_area );
        calculateFrameSize();
        calculateItemSizes();
        moveResizeInputWindow( frame_area.x(), frame_area.y(), frame_area.width(), frame_area.height());
        workspace()->addRepaint( frame_area );
        }
    }

void BoxSwitchEffect::setActive()
    {
    mActivated = true;
    if( mMode == TabBox::WindowsMode )
        {
        tab_clients = Workspace::self()->currentTabBoxClientList();
        selected_window = Workspace::self()->currentTabBoxClient()->effectWindow();
        }
    else
        {
        selected_desktop = Workspace::self()->currentTabBoxDesktop();
        }
    calculateFrameSize();
    calculateItemSizes();
    mInput = effects->createInputWindow( this, frame_area.x(), frame_area.y(),
        frame_area.width(), frame_area.height(), Qt::ArrowCursor );
    workspace()->addRepaint( frame_area );
    if( mMode == TabBox::WindowsMode )
        {
        foreach( EffectWindow* w, windows.keys())
            {
            if( w != selected_window )
                w->window()->addRepaintFull();
            }
        }
    }

void BoxSwitchEffect::setInactive()
    {
    mActivated = false;
    if( mInput != None )
        {
        effects->destroyInputWindow( mInput );
        mInput = None;
        }
    if( mMode == TabBox::WindowsMode )
        {
        foreach( EffectWindow* w, windows.keys())
            {
            if( w != selected_window )
                w->window()->addRepaintFull();
            }
        }
    workspace()->addRepaint( frame_area );
    frame_area = QRect();
    }

void BoxSwitchEffect::moveResizeInputWindow( int x, int y, int width, int height )
    {
    XMoveWindow( display(), mInput, x, y );
    XResizeWindow( display(), mInput, width, height );
    }

void BoxSwitchEffect::calculateFrameSize()
    {
    int itemcount;

    if( mMode == TabBox::WindowsMode )
        {
        itemcount = tab_clients.count();
        item_max_size.setWidth( 200 );
        item_max_size.setHeight( 200 );
        }
    else
        {
        itemcount = Workspace::self()->numberOfDesktops();
        item_max_size.setWidth( 200 );
        item_max_size.setHeight( 200 );
        }
    // Shrink the size until all windows/desktops can fit onscreen
    frame_area.setWidth( frame_margin * 2 + itemcount * item_max_size.width());
    while( frame_area.width() > displayWidth())
        {
        item_max_size /= 2;
        frame_area.setWidth( frame_margin * 2 + itemcount * item_max_size.width());
        }
    frame_area.setHeight( frame_margin * 2 + item_max_size.height() + 15 );
    frame_area.moveTo( ( displayWidth() - frame_area.width()) / 2, ( displayHeight() - frame_area.height()) / 2 );
    }

void BoxSwitchEffect::calculateItemSizes()
    {
    if( mMode == TabBox::WindowsMode )
        {
        windows.clear();
        for( int i = 0; i < tab_clients.count(); i++ )
            {
            EffectWindow* w = tab_clients.at( i )->effectWindow();

            windows[ w ].area = QRect( frame_area.x() + frame_margin
                + i * item_max_size.width(),
                frame_area.y() + frame_margin,
                item_max_size.width(), item_max_size.height());
            windows[ w ].clickable = windows[ w ].area;
            windows[ w ].icon = static_cast< Client* >( w->window())->icon();
/* #ifdef HAVE_OPENGL
            windows[ w ].glIcon.bindPixmap( windows[ w ].icon.handle(), windows[ w ].icon.width(), windows[ w ].icon.height(), windows[ w ].icon.depth());
            windows[ w ].glIcon.glFilter = GL_LINEAR;
#endif */
            }
        }
    else
        {
        desktops.clear();
        int iDesktop = ( mMode == TabBox::DesktopMode ) ? Workspace::self()->currentDesktop() : 1;
        for( int i = 1; i <= Workspace::self()->numberOfDesktops(); i++ )
            {
            desktops[ iDesktop ].area = QRect( frame_area.x() + frame_margin
                + ( i - 1 ) * item_max_size.width(),
                frame_area.y() + frame_margin,
                item_max_size.width(), item_max_size.height());
            desktops[ iDesktop ].clickable = desktops[ iDesktop ].area;

            if( mMode == TabBox::DesktopMode )
                iDesktop = Workspace::self()->nextDesktopFocusChain( iDesktop );
            else
                iDesktop++;
            }
        }
    }

void BoxSwitchEffect::paintFrame()
    {
#ifdef HAVE_OPENGL
    glPushAttrib( GL_CURRENT_BIT | GL_ENABLE_BIT );
    glEnable( GL_BLEND );
    glBlendFunc( GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA );
    glColor4f( 0, 0, 0, 0.75 );
    glBegin( GL_QUADS );
    glVertex2i( frame_area.x(), frame_area.y());
    glVertex2i( frame_area.x() + frame_area.width(), frame_area.y());
    glVertex2i( frame_area.x() + frame_area.width(), frame_area.y() + frame_area.height());
    glVertex2i( frame_area.x(), frame_area.y() + frame_area.height());
    glEnd();
    glPopAttrib();
#endif
    }

void BoxSwitchEffect::paintHighlight( QRect area, QString text )
    {
#ifdef HAVE_OPENGL
    glPushAttrib( GL_CURRENT_BIT | GL_ENABLE_BIT );
    glEnable( GL_BLEND );
    glBlendFunc( GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA );
    glColor4f( 1, 1, 1, 0.75 );
    glBegin( GL_QUADS );
    glVertex2i( area.x(), area.y());
    glVertex2i( area.x() + area.width(), area.y());
    glVertex2i( area.x() + area.width(), area.y() + area.height());
    glVertex2i( area.x(), area.y() + area.height());
    glEnd();
    glPopAttrib();
#endif
//    kDebug() << text << endl; // TODO draw this nicely on screen
    }

void BoxSwitchEffect::paintDesktopThumbnail( int iDesktop )
    {
    if( !desktops.contains( iDesktop ))
        return;

    ScreenPaintData data;
    QRect region;
    QRect r = desktops[ iDesktop ].area.adjusted( highlight_margin, highlight_margin,
        -highlight_margin, -highlight_margin );
    QSize size = QSize( displayWidth(), displayHeight());

    size.scale( r.size(), Qt::KeepAspectRatio );
    data.xScale = size.width() / double( displayWidth());
    data.yScale = size.height() / double( displayHeight());
    int width = int( displayWidth() * data.xScale );
    int height = int( displayHeight() * data.yScale );
    int x = r.x() + ( r.width() - width ) / 2;
    int y = r.y() + ( r.height() - height ) / 2;
    region = QRect( x, y, width, height );
    data.xTranslate = x;
    data.yTranslate = y;

    effects->paintScreen( Scene::PAINT_SCREEN_TRANSFORMED | Scene::PAINT_SCREEN_BACKGROUND_FIRST,
        region, data );
    }

void BoxSwitchEffect::paintWindowThumbnail( EffectWindow* w )
    {
    if( !windows.contains( w ))
        return;
    WindowPaintData data;

    setPositionTransformations( data,
        windows[ w ].thumbnail, w,
        windows[ w ].area.adjusted( highlight_margin, highlight_margin, -highlight_margin, -highlight_margin ),
        Qt::KeepAspectRatio );

    effects->drawWindow( w,
        Scene::PAINT_WINDOW_OPAQUE | Scene::PAINT_WINDOW_TRANSFORMED,
        windows[ w ].thumbnail, data );
    }

void BoxSwitchEffect::paintWindowIcon( EffectWindow* w )
    {
    if( !windows.contains( w ))
        return;
/*    if( windows[ w ].icon.serialNumber()
        != static_cast< Client* >( w->window())->icon().serialNumber())
        { // icon changed
        windows[ w ].icon = static_cast< Client* >( w->window())->icon();
#ifdef HAVE_OPENGL
        windows[ w ].glIcon.bindPixmap( windows[ w ].icon.handle(),
            windows[ w ].icon.width(), windows[ w ].icon.height(),
            windows[ w ].icon.depth(), windows[ w ].icon.rect());
#endif
        }
#ifdef HAVE_OPENGL
    glPushMatrix();
    glPushAttrib( GL_ENABLE_BIT );
    windows[ w ].glIcon.enable();
    windows[ w ].glIcon.preparePaint();
    int x = windows[ w ].area.x() + ( windows[ w ].area.width() - windows[ w ].icon.width()) / 2
    int y = windows[ w ].area.y() + ( windows[ w ].area.height() - windows[ w ].icon.height()) / 2
    glBegin( GL_QUADS );
    glTexCoord2i( 0, 0 );
    glVertex2i( x, y );
    glTexCoord2i( windows[ w ].icon.width(), 0 );
    glVertex2i( x + windows[ w ].icon.width(), y );
    glTexCoord2i( windows[ w ].icon.width(), windows[ w ].icon.height());
    glVertex2i( x + windows[ w ].icon.width(), y + windows[ w ].icon.height());
    glTexCoord2i( 0, windows[ w ].icon.height());
    glVertex2i( x, y + windows[ w ].icon.height());
    glEnd();
    windows[ w ].glIcon.finishPaint();
    glPopAttrib();
    glPopMatrix();
    windows[ w ].glIcon.disable();
#endif */
    }

BoxSwitchEffect::ItemInfo::~ItemInfo()
    {
/* #ifdef HAVE_OPENGL
    glIcon.discard();
#endif */
    }

} // namespace