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

Copyright (C) 2007 Lubos Lunak <l.lunak@kde.org>

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

#include "drunken.h"

#include <math.h>

namespace KWin
{

void DrunkenEffect::prePaintScreen( int* mask, QRegion* region, int time )
    {
    if( !windows.isEmpty())
        *mask |= Scene::PAINT_SCREEN_WITH_TRANSFORMED_WINDOWS;
    effects->prePaintScreen( mask, region, time );
    }

void DrunkenEffect::prePaintWindow( EffectWindow* w, int* mask, QRegion* paint, QRegion* clip, int time )
    {
    if( windows.contains( w ))
        {
        windows[ w ] += time / 1000.;
        if( windows[ w ] < 1 )
            *mask |= Scene::PAINT_WINDOW_TRANSFORMED;
        else
            windows.remove( w );
        }
    effects->prePaintWindow( w, mask, paint, clip, time );
    }

void DrunkenEffect::paintWindow( EffectWindow* w, int mask, QRegion region, WindowPaintData& data )
    {
    if( !windows.contains( w ))
        {
        effects->paintWindow( w, mask, region, data );
        return;
        }
    WindowPaintData d1 = data;
    // 4 cycles, decreasing amplitude
    int diff = int( sin( windows[ w ] * 8 * M_PI ) * ( 1 - windows[ w ] ) * 10 );
    d1.xTranslate -= diff;
    d1.opacity *= 0.5;
    effects->paintWindow( w, mask, region, d1 );
    WindowPaintData d2 = data;
    d2.xTranslate += diff;
    d2.opacity *= 0.5;
    effects->paintWindow( w, mask, region, d2 );
    }

void DrunkenEffect::postPaintWindow( EffectWindow* w )
    {
    if( windows.contains( w ))
        w->window()->addRepaintFull();
    effects->postPaintWindow( w );
    }

void DrunkenEffect::windowAdded( EffectWindow* w )
    {
    windows[ w ] = 0;
    w->window()->addRepaintFull();
    }

void DrunkenEffect::windowClosed( EffectWindow* w )
    {
    windows.remove( w );
    }

} // namespace