354 lines
10 KiB
C++
354 lines
10 KiB
C++
/********************************************************************
|
|
KWin - the KDE window manager
|
|
This file is part of the KDE project.
|
|
|
|
Copyright (C) 2009 Martin Gräßlin <kde@martin-graesslin.com>
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
Based on Compiz Fusion cube gear plugin by Dennis Kasprzyk:
|
|
http://gitweb.compiz-fusion.org/?p=fusion/plugins/gears;a=blob;f=gears.c;hb=HEAD
|
|
Which is based on glxgears.c by Brian Paul:
|
|
http://cvsweb.xfree86.org/cvsweb/xc/programs/glxgears/glxgears.c
|
|
*********************************************************************/
|
|
|
|
#include "gears.h"
|
|
#include "../cube/cube_proxy.h"
|
|
|
|
#include <math.h>
|
|
|
|
#include <GL/gl.h>
|
|
|
|
namespace KWin
|
|
{
|
|
|
|
KWIN_EFFECT( gears, GearsEffect )
|
|
KWIN_EFFECT_SUPPORTED( gears, GearsEffect::supported() )
|
|
|
|
GearsEffect::GearsEffect()
|
|
: CubeInsideEffect()
|
|
, m_active( false )
|
|
, m_contentRotation( 0.0f )
|
|
, m_angle( 0.0f )
|
|
{
|
|
CubeEffectProxy* proxy =
|
|
static_cast<CubeEffectProxy*>( effects->getProxy( "cube" ) );
|
|
if( proxy )
|
|
proxy->registerCubeInsideEffect( this );
|
|
}
|
|
|
|
GearsEffect::~GearsEffect()
|
|
{
|
|
CubeEffectProxy* proxy =
|
|
static_cast<CubeEffectProxy*>( effects->getProxy( "cube" ) );
|
|
if( proxy )
|
|
proxy->unregisterCubeInsideEffect( this );
|
|
}
|
|
|
|
bool GearsEffect::supported()
|
|
{
|
|
return effects->compositingType() == OpenGLCompositing;
|
|
}
|
|
|
|
void GearsEffect::prePaintScreen( ScreenPrePaintData& data, int time )
|
|
{
|
|
if( m_active )
|
|
{
|
|
m_contentRotation += time * 360.0f / 20000.0f;
|
|
if( m_contentRotation > 360.0f )
|
|
m_contentRotation -= 360.0f;
|
|
m_angle += time * 360.0f / 8000.0f;
|
|
if( m_angle > 360.0f )
|
|
m_angle -= 360.0f;
|
|
}
|
|
effects->prePaintScreen( data, time );
|
|
}
|
|
|
|
void GearsEffect::postPaintScreen()
|
|
{
|
|
if( m_active )
|
|
{
|
|
effects->addRepaintFull();
|
|
}
|
|
effects->postPaintScreen();
|
|
}
|
|
|
|
|
|
void GearsEffect::paint()
|
|
{
|
|
if( m_active )
|
|
{
|
|
paintGears();
|
|
}
|
|
}
|
|
|
|
void GearsEffect::setActive( bool active )
|
|
{
|
|
m_active = active;
|
|
if( active )
|
|
initGears();
|
|
else
|
|
endGears();
|
|
}
|
|
|
|
void GearsEffect::gear( float inner_radius, float outer_radius, float width, int teeth, float tooth_depth )
|
|
{
|
|
GLint i;
|
|
GLfloat r0, r1, r2, maxr2, minr2;
|
|
GLfloat angle, da;
|
|
GLfloat u, v, len;
|
|
|
|
r0 = inner_radius;
|
|
r1 = outer_radius - tooth_depth / 2.0;
|
|
maxr2 = r2 = outer_radius + tooth_depth / 2.0;
|
|
minr2 = r2;
|
|
|
|
da = 2.0 * M_PI / teeth / 4.0;
|
|
|
|
glShadeModel( GL_FLAT );
|
|
|
|
glNormal3f( 0.0, 0.0, 1.0 );
|
|
|
|
/* draw front face */
|
|
glBegin( GL_QUAD_STRIP );
|
|
|
|
for( i = 0; i <= teeth; ++i )
|
|
{
|
|
angle = i * 2.0 * M_PI / teeth;
|
|
glVertex3f( r0 * cos( angle ), r0 * sin( angle ), width * 0.5 );
|
|
glVertex3f( r1 * cos( angle ), r1 * sin( angle ), width * 0.5 );
|
|
|
|
if( i < teeth )
|
|
{
|
|
glVertex3f( r0 * cos( angle ), r0 * sin( angle ), width * 0.5 );
|
|
glVertex3f( r1 * cos( angle + 3 * da ), r1 * sin( angle + 3 * da ), width * 0.5 );
|
|
}
|
|
}
|
|
|
|
glEnd();
|
|
|
|
/* draw front sides of teeth */
|
|
glBegin( GL_QUADS );
|
|
|
|
for( i = 0; i < teeth; ++i )
|
|
{
|
|
angle = i * 2.0 * M_PI / teeth;
|
|
|
|
glVertex3f( r1 * cos( angle ), r1 * sin( angle ), width * 0.5 );
|
|
glVertex3f( r2 * cos( angle + da ), r2 * sin( angle + da ), width * 0.5 );
|
|
glVertex3f( r2 * cos( angle + 2 * da ), r2 * sin( angle + 2 * da ), width * 0.5 );
|
|
glVertex3f( r1 * cos( angle + 3 * da ), r1 * sin( angle + 3 * da ), width * 0.5 );
|
|
r2 = minr2;
|
|
}
|
|
|
|
r2 = maxr2;
|
|
|
|
glEnd();
|
|
|
|
glNormal3f( 0.0, 0.0, -1.0 );
|
|
|
|
/* draw back face */
|
|
glBegin( GL_QUAD_STRIP );
|
|
|
|
for( i = 0; i <= teeth; ++i )
|
|
{
|
|
angle = i * 2.0 * M_PI / teeth;
|
|
glVertex3f( r1 * cos( angle ), r1 * sin( angle ), -width * 0.5 );
|
|
glVertex3f( r0 * cos( angle ), r0 * sin( angle ), -width * 0.5 );
|
|
|
|
if( i < teeth )
|
|
{
|
|
glVertex3f( r1 * cos( angle + 3 * da ), r1 * sin( angle + 3 * da ), -width * 0.5 );
|
|
glVertex3f( r0 * cos( angle ), r0 * sin( angle ), -width * 0.5 );
|
|
}
|
|
}
|
|
|
|
glEnd();
|
|
|
|
/* draw back sides of teeth */
|
|
glBegin( GL_QUADS );
|
|
da = 2.0 * M_PI / teeth / 4.0;
|
|
|
|
for( i = 0; i < teeth; ++i )
|
|
{
|
|
angle = i * 2.0 * M_PI / teeth;
|
|
|
|
glVertex3f( r1 * cos( angle + 3 * da ), r1 * sin( angle + 3 * da ), -width * 0.5 );
|
|
glVertex3f( r2 * cos( angle + 2 * da ), r2 * sin( angle + 2 * da ), -width * 0.5 );
|
|
glVertex3f( r2 * cos (angle + da), r2 * sin (angle + da), -width * 0.5 );
|
|
glVertex3f( r1 * cos (angle), r1 * sin (angle), -width * 0.5 );
|
|
r2 = minr2;
|
|
}
|
|
|
|
r2 = maxr2;
|
|
|
|
glEnd();
|
|
|
|
/* draw outward faces of teeth */
|
|
glBegin( GL_QUAD_STRIP );
|
|
|
|
for( i = 0; i < teeth; ++i )
|
|
{
|
|
angle = i * 2.0 * M_PI / teeth;
|
|
|
|
glVertex3f( r1 * cos( angle ), r1 * sin( angle ), width * 0.5 );
|
|
glVertex3f( r1 * cos( angle ), r1 * sin( angle ), -width * 0.5 );
|
|
u = r2 * cos( angle + da ) - r1 * cos( angle );
|
|
v = r2 * sin( angle + da ) - r1 * sin( angle );
|
|
len = sqrt( u * u + v * v );
|
|
u /= len;
|
|
v /= len;
|
|
glNormal3f( v, -u, 0.0 );
|
|
glVertex3f( r2 * cos( angle + da ), r2 * sin( angle + da ), width * 0.5 );
|
|
glVertex3f( r2 * cos( angle + da ), r2 * sin( angle + da ), -width * 0.5 );
|
|
glNormal3f( cos( angle + 1.5 * da ), sin( angle + 1.5 * da ), 0.0 );
|
|
glVertex3f( r2 * cos( angle + 2 * da ), r2 * sin( angle + 2 * da ), width * 0.5 );
|
|
glVertex3f( r2 * cos( angle + 2 * da ), r2 * sin( angle + 2 * da ), -width * 0.5 );
|
|
u = r1 * cos( angle + 3 * da ) - r2 * cos( angle + 2 * da );
|
|
v = r1 * sin( angle + 3 * da ) - r2 * sin( angle + 2 * da );
|
|
glNormal3f( v, -u, 0.0 );
|
|
glVertex3f( r1 * cos( angle + 3 * da ), r1 * sin( angle + 3 * da ), width * 0.5 );
|
|
glVertex3f( r1 * cos( angle + 3 * da ), r1 * sin( angle + 3 * da ), -width * 0.5 );
|
|
glNormal3f( cos( angle + 3.5 * da ), sin( angle + 3.5 * da ), 0.0 );
|
|
r2 = minr2;
|
|
}
|
|
|
|
r2 = maxr2;
|
|
|
|
glVertex3f( r1 * cos( 0 ), r1 * sin( 0 ), width * 0.5 );
|
|
glVertex3f( r1 * cos( 0 ), r1 * sin( 0 ), -width * 0.5 );
|
|
|
|
glEnd();
|
|
|
|
glShadeModel( GL_SMOOTH );
|
|
|
|
/* draw inside radius cylinder */
|
|
glBegin( GL_QUAD_STRIP );
|
|
|
|
for( i = 0; i <= teeth; ++i )
|
|
{
|
|
angle = i * 2.0 * M_PI / teeth;
|
|
glNormal3f( -cos( angle ), -sin( angle ), 0.0 );
|
|
glVertex3f( r0 * cos( angle ), r0 * sin( angle ), -width * 0.5 );
|
|
glVertex3f( r0 * cos( angle ), r0 * sin( angle ), width * 0.5 );
|
|
}
|
|
glEnd();
|
|
}
|
|
|
|
void GearsEffect::paintGears()
|
|
{
|
|
static GLfloat white[4] = { 1.0, 1.0, 1.0, 1.0 };
|
|
QRect rect = effects->clientArea( FullArea, effects->activeScreen(), effects->currentDesktop() );
|
|
|
|
glPushMatrix();
|
|
// invert scale
|
|
float fovy = 60.0f;
|
|
float zNear = 0.1f;
|
|
float ymax = zNear * tan( fovy * M_PI / 360.0f );
|
|
float ymin = -ymax;
|
|
float xmin = ymin;
|
|
float xmax = ymax;
|
|
float scaleFactor = 1.1 * tan( fovy * M_PI / 360.0f )/ymax;
|
|
glScalef( 1.0f/((xmax-xmin)*scaleFactor/displayWidth()), 1.0f/(-(ymax-ymin)*scaleFactor/displayHeight()), 1000.0f );
|
|
|
|
glPushAttrib( GL_COLOR_BUFFER_BIT | GL_TEXTURE_BIT | GL_ENABLE_BIT | GL_LIGHTING_BIT );
|
|
|
|
glDisable( GL_BLEND );
|
|
|
|
glPushMatrix();
|
|
|
|
// content rotation requires depth test - so disabled
|
|
// glRotatef( m_contentRotation, 0.0, 1.0, 0.0 );
|
|
|
|
glScalef( 0.05, 0.05, 0.05 );
|
|
|
|
glEnable( GL_NORMALIZE );
|
|
glEnable( GL_LIGHTING );
|
|
glEnable( GL_LIGHT1 );
|
|
glDisable( GL_COLOR_MATERIAL );
|
|
|
|
glTexEnvi( GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_MODULATE );
|
|
|
|
glPushMatrix();
|
|
glTranslatef( -3.0, -2.0, 0.0 );
|
|
glRotatef( m_angle, 0.0, 0.0, 1.0 );
|
|
glCallList( m_gear1 );
|
|
glPopMatrix();
|
|
|
|
glPushMatrix();
|
|
glTranslatef( 3.1, -2.0, 0.0 );
|
|
glRotatef( -2.0 * m_angle - 9.0, 0.0, 0.0, 1.0 );
|
|
glCallList( m_gear2 );
|
|
glPopMatrix();
|
|
|
|
glPushMatrix();
|
|
glTranslatef( -3.1, 4.2, 0.0 );
|
|
glRotatef( -2.0 * m_angle - 25.0, 0.0, 0.0, 1.0 );
|
|
glCallList( m_gear3 );
|
|
glPopMatrix();
|
|
|
|
glMaterialfv( GL_FRONT, GL_AMBIENT_AND_DIFFUSE, white );
|
|
|
|
glPopMatrix();
|
|
|
|
glDisable( GL_LIGHT1 );
|
|
glDisable( GL_NORMALIZE );
|
|
glEnable( GL_COLOR_MATERIAL );
|
|
|
|
glDisable( GL_LIGHTING );
|
|
|
|
glPopMatrix();
|
|
glPopAttrib();
|
|
}
|
|
|
|
void GearsEffect::initGears()
|
|
{
|
|
static GLfloat pos[4] = { 5.0f, 5.0f, 10.0f, 0.0f };
|
|
static GLfloat red[4] = { 0.8f, 0.1f, 0.0f, 1.0f };
|
|
static GLfloat green[4] = { 0.0f, 0.8f, 0.2f, 1.0f };
|
|
static GLfloat blue[4] = { 0.2f, 0.2f, 1.0f, 1.0f };
|
|
static GLfloat ambientLight[] = { 0.3f, 0.3f, 0.3f, 0.3f };
|
|
static GLfloat diffuseLight[] = { 0.5f, 0.5f, 0.5f, 0.5f };
|
|
|
|
glLightfv( GL_LIGHT1, GL_AMBIENT, ambientLight );
|
|
glLightfv( GL_LIGHT1, GL_DIFFUSE, diffuseLight );
|
|
glLightfv( GL_LIGHT1, GL_POSITION, pos );
|
|
|
|
m_gear1 = glGenLists( 1 );
|
|
glNewList( m_gear1, GL_COMPILE );
|
|
glMaterialfv( GL_FRONT, GL_AMBIENT_AND_DIFFUSE, red );
|
|
gear( 1.0f, 4.0f, 1.0f, 20, 0.7f );
|
|
glEndList();
|
|
|
|
m_gear2 = glGenLists( 1 );
|
|
glNewList( m_gear2, GL_COMPILE );
|
|
glMaterialfv( GL_FRONT, GL_AMBIENT_AND_DIFFUSE, green );
|
|
gear( 0.5f, 2.0f, 2.0f, 10, 0.7f );
|
|
glEndList();
|
|
|
|
m_gear3 = glGenLists( 1 );
|
|
glNewList( m_gear3, GL_COMPILE );
|
|
glMaterialfv( GL_FRONT, GL_AMBIENT_AND_DIFFUSE, blue );
|
|
gear( 1.3f, 2.0f, 0.5f, 10, 0.7f );
|
|
glEndList();
|
|
}
|
|
|
|
void GearsEffect::endGears()
|
|
{
|
|
glDeleteLists( m_gear1, 1 );
|
|
glDeleteLists( m_gear2, 1 );
|
|
glDeleteLists( m_gear3, 1 );
|
|
}
|
|
|
|
} // namespace
|