#!/usr/bin/env python
# ----------------------------------------------------------------------------
# Pyglet GLSL Demo Texture Shader on http://www.pythonstuff.org
# pythonstuff_at_gmx_dot_at (c) 2010
#
# based on the "graphics.py" batch/VBO demo by
# pyglet
# Copyright (c) 2006-2008 Alex Holkner
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in
# the documentation and/or other materials provided with the
# distribution.
# * Neither the name of pyglet nor the names of its
# contributors may be used to endorse or promote products
# derived from this software without specific prior written
# permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
# FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
# COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
# ----------------------------------------------------------------------------
'''This expands the previous example by adding a nice texture
Find more GLSL-examples http://www.pythonstuff.org
'''
html = '''
Pyglet GLSL Texture Demo
ENTER = Shader on/off
R = Reset
Q, Esc = Quit
F = Toggle Figure
T = Toggle Texture
W, S, A, D = Up, Down, Left, Right
Space = Move/Stop
Arrows = Move Light 0
H = This Help
'''
from math import pi, sin, cos, sqrt
from euclid import *
import pyglet
from pyglet.gl import *
from pyglet.window import key
from pyglet import image, resource
from shader import Shader
resource.path.append('textures')
resource.reindex()
texturecnt = 1
try:
# Try and create a window with multisampling (antialiasing)
config = Config(sample_buffers=1, samples=4,
depth_size=16, double_buffer=True,)
window = pyglet.window.Window(resizable=True, config=config, vsync=False) # "vsync=False" to check the framerate
except pyglet.window.NoSuchConfigException:
# Fall back to no multisampling for old hardware
window = pyglet.window.Window(resizable=True)
label = pyglet.text.HTMLLabel(html, # location=location,
width=window.width//2,
multiline=True, anchor_x='center', anchor_y='center')
fps_display = pyglet.clock.ClockDisplay() # see programming guide pg 48
@window.event
def on_resize(width, height):
if height==0: height=1
# Keep text vertically centered in the window
label.y = window.height // 2
# Override the default on_resize handler to create a 3D projection
glViewport(0, 0, width, height)
glMatrixMode(GL_PROJECTION)
glLoadIdentity()
gluPerspective(60., width / float(height), .1, 1000.)
glMatrixMode(GL_MODELVIEW)
return pyglet.event.EVENT_HANDLED
def update(dt):
global autorotate
global rot
if autorotate:
rot += Vector3(0.1, 12, 5) * dt
rot.x %= 360
rot.y %= 360
rot.z %= 360
pyglet.clock.schedule(update)
def dismiss_dialog(dt):
global showdialog
showdialog = False
pyglet.clock.schedule_once(dismiss_dialog, 10.0)
# Define a simple function to create ctypes arrays of floats:
def vec(*args):
return (GLfloat * len(args))(*args)
@window.event
def on_draw():
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
glLoadIdentity()
glLoadIdentity()
glTranslatef(0.0, 0.0, -3.5);
glRotatef(rot.x, 0, 0, 1)
glRotatef(rot.y, 0, 1, 0)
glRotatef(rot.z, 1, 0, 0)
glPolygonMode(GL_FRONT, GL_FILL)
if shaderon:
# bind our shader
shader.bind()
shader.uniformi('toggletexture', toggletexture )
for i in range(texturecnt):
glActiveTexture(GL_TEXTURE0+i)
glEnable(GL_TEXTURE_2D)
glBindTexture(GL_TEXTURE_2D, texture[i].id)
shader.uniformi('my_color_texture[' + str(i) + ']',i )
if togglefigure:
batch1.draw()
else:
batch2.draw()
for i in range(texturecnt):
glActiveTexture(GL_TEXTURE0+i)
glDisable(GL_TEXTURE_2D)
shader.unbind()
else:
if togglefigure:
batch1.draw()
else:
batch2.draw()
glActiveTexture(GL_TEXTURE0)
glEnable(GL_TEXTURE_2D)
glDisable(GL_LIGHTING)
glDisable(GL_DEPTH_TEST)
if showdialog:
glLoadIdentity()
glTranslatef(0, -200, -450)
label.draw()
glLoadIdentity()
glTranslatef(250, -290, -500)
fps_display.draw()
glEnable(GL_DEPTH_TEST)
glEnable(GL_LIGHTING)
glDisable(GL_TEXTURE_2D)
@window.event
def on_key_press(symbol, modifiers):
global autorotate
global rot
global togglefigure
global toggletexture
global light0pos
global light1pos
global showdialog
global shaderon
if symbol == key.R:
print 'Reset'
rot = Vector3(0, 0, 0)
elif symbol == key.ESCAPE or symbol == key.Q:
print 'Good Bye !' # ESC would do it anyway, but not "Q"
pyglet.app.exit()
return pyglet.event.EVENT_HANDLED
elif symbol == key.H:
showdialog = not showdialog
elif symbol == key.ENTER:
print 'Shader toggle'
shaderon = not shaderon
elif symbol == key.SPACE:
print 'Toggle autorotate'
autorotate = not autorotate
elif symbol == key.F:
togglefigure = not togglefigure
print 'Toggle Figure ', togglefigure
elif symbol == key.T:
toggletexture = not toggletexture
print 'Toggle Texture ', toggletexture
elif symbol == key.A:
print 'Stop left'
if autorotate:
autorotate = False
else:
rot.y += -rotstep
rot.y %= 360
elif symbol == key.S:
print 'Stop down'
if autorotate:
autorotate = False
else:
rot.z += rotstep
rot.z %= 360
elif symbol == key.W:
print 'Stop up'
if autorotate:
autorotate = False
else:
rot.z += -rotstep
rot.z %= 360
elif symbol == key.D:
print 'Stop right'
if autorotate:
autorotate = False
else:
rot.y += rotstep
rot.y %= 360
elif symbol == key.LEFT:
print 'Light0 rotate left'
tmp = light0pos[0]
light0pos[0] = tmp * cos( lightstep ) - light0pos[2] * sin( lightstep )
light0pos[2] = light0pos[2] * cos( lightstep ) + tmp * sin( lightstep )
glLoadIdentity()
glLightfv(GL_LIGHT0, GL_POSITION, vec(*light0pos))
elif symbol == key.RIGHT:
print 'Light0 rotate right'
tmp = light0pos[0]
light0pos[0] = tmp * cos( -lightstep ) - light0pos[2] * sin( -lightstep )
light0pos[2] = light0pos[2] * cos( -lightstep ) + tmp * sin( -lightstep )
glLoadIdentity()
glLightfv(GL_LIGHT0, GL_POSITION, vec(*light0pos))
elif symbol == key.UP:
print 'Light0 up'
tmp = light0pos[1]
light0pos[1] = tmp * cos( -lightstep ) - light0pos[2] * sin( -lightstep )
light0pos[2] = light0pos[2] * cos( -lightstep ) + tmp * sin( -lightstep )
glLoadIdentity()
glLightfv(GL_LIGHT0, GL_POSITION, vec(*light0pos))
elif symbol == key.DOWN:
print 'Light0 down'
tmp = light0pos[1]
light0pos[1] = tmp * cos( lightstep ) - light0pos[2] * sin( lightstep )
light0pos[2] = light0pos[2] * cos( lightstep ) + tmp * sin( lightstep )
glLoadIdentity()
glLightfv(GL_LIGHT0, GL_POSITION, vec(*light0pos))
else:
print 'OTHER KEY'
def setup():
# One-time GL setup
global light0pos
global light1pos
global toggletexture
global texture
light0pos = [20.0, 20.0, 20.0, 1.0] # positional light !
light1pos = [-20.0, -20.0, 20.0, 0.0] # infinitely away light !
glClearColor(1, 1, 1, 1)
glColor4f(1.0, 0.0, 0.0, 0.5 )
glEnable(GL_DEPTH_TEST)
glEnable(GL_CULL_FACE)
texture = []
for i in range (texturecnt):
texturefile = 'Texturemap' + str(i) + '.jpg'
print "Loading Texture", texturefile
textureSurface = pyglet.resource.texture(texturefile)
texture.append( textureSurface.get_texture() )
glBindTexture(texture[i].target, texture[i].id)
print "Texture ", i, " bound to ", texture[i].id
# Uncomment this line for a wireframe view
#glPolygonMode(GL_FRONT_AND_BACK, GL_LINE)
glEnable(GL_BLEND)
glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
# Simple light setup. On Windows GL_LIGHT0 is enabled by default,
# but this is not the case on Linux or Mac, so remember to always
# include it.
glEnable(GL_LIGHTING)
glEnable(GL_LIGHT0)
glEnable(GL_LIGHT1)
glLightfv(GL_LIGHT0, GL_POSITION, vec(*light0pos))
glLightfv(GL_LIGHT0, GL_AMBIENT, vec(0.3, 0.3, 0.3, 1.0))
glLightfv(GL_LIGHT0, GL_DIFFUSE, vec(0.9, 0.9, 0.9, 1.0))
glLightfv(GL_LIGHT0, GL_SPECULAR, vec(1.0, 1.0, 1.0, 1.0))
glLightfv(GL_LIGHT1, GL_POSITION, vec(*light1pos))
glLightfv(GL_LIGHT1, GL_DIFFUSE, vec(.6, .6, .6, 1.0))
glLightfv(GL_LIGHT1, GL_SPECULAR, vec(1.0, 1.0, 1.0, 1.0))
glMaterialfv(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE, vec(0.8, 0.5, 0.5, 1.0))
glMaterialfv(GL_FRONT_AND_BACK, GL_SPECULAR, vec(1, 1, 1, 1))
glMaterialf(GL_FRONT_AND_BACK, GL_SHININESS, 50)
# create our Phong Shader by Jerome GUINOT aka 'JeGX' - jegx [at] ozone3d [dot] net
# see http://www.ozone3d.net/tutorials/glsl_lighting_phong.php
shader = Shader(['''
varying vec3 normal, lightDir0, lightDir1, eyeVec;
void main()
{
normal = gl_NormalMatrix * gl_Normal;
vec3 vVertex = vec3(gl_ModelViewMatrix * gl_Vertex);
lightDir0 = vec3(gl_LightSource[0].position.xyz - vVertex);
lightDir1 = vec3(gl_LightSource[1].position.xyz - vVertex);
eyeVec = -vVertex;
gl_Position = ftransform();
gl_TexCoord[0] = gl_TextureMatrix[0] * gl_MultiTexCoord0;
}
'''], ['''
varying vec3 normal, lightDir0, lightDir1, eyeVec;
uniform sampler2D my_color_texture['''+str(texturecnt)+''']; //0 = ColorMap
uniform int toggletexture; // false/true
void main (void)
{
vec4 texColor = texture2D(my_color_texture[0], gl_TexCoord[0].st);
vec4 final_color;
if ( toggletexture == 0 ) texColor = gl_FrontMaterial.ambient;
final_color = (gl_FrontLightModelProduct.sceneColor * vec4(texColor.rgb,1.0)) +
(gl_LightSource[0].ambient * vec4(texColor.rgb,1.0)) +
(gl_LightSource[1].ambient * vec4(texColor.rgb,1.0));
vec3 N = normalize(normal);
vec3 L0 = normalize(lightDir0);
vec3 L1 = normalize(lightDir1);
float lambertTerm0 = dot(N,L0);
float lambertTerm1 = dot(N,L1);
if(lambertTerm0 > 0.0)
{
final_color += gl_LightSource[0].diffuse *
gl_FrontMaterial.diffuse *
lambertTerm0;
vec3 E = normalize(eyeVec);
vec3 R = reflect(-L0, N);
float specular = pow( max(dot(R, E), 0.0),
gl_FrontMaterial.shininess );
final_color += gl_LightSource[0].specular *
gl_FrontMaterial.specular *
specular;
}
if(lambertTerm1 > 0.0)
{
final_color += gl_LightSource[1].diffuse *
gl_FrontMaterial.diffuse *
lambertTerm1;
vec3 E = normalize(eyeVec);
vec3 R = reflect(-L1, N);
float specular = pow( max(dot(R, E), 0.0),
gl_FrontMaterial.shininess );
final_color += gl_LightSource[1].specular *
gl_FrontMaterial.specular *
specular;
}
gl_FragColor = final_color;
}
'''])
class Torus(object):
list = None
def __init__(self, radius, inner_radius, slices, inner_slices,
batch, group=None):
# Create the vertex and normal arrays.
vertices = []
normals = []
textureuvw = []
tangents = []
u_step = 2 * pi / (slices - 1)
v_step = 2 * pi / (inner_slices - 1)
u = 0.
for i in range(slices):
cos_u = cos(u)
sin_u = sin(u)
v = 0.
for j in range(inner_slices):
cos_v = cos(v)
sin_v = sin(v)
d = (radius + inner_radius * cos_v)
x = d * cos_u
y = d * sin_u
z = inner_radius * sin_v
nx = cos_u * cos_v
ny = sin_u * cos_v
nz = sin_v
vertices.extend([x, y, z])
normals.extend([nx, ny, nz])
textureuvw.extend([u / (2 * pi), v / (2 * pi), 0.0])
tangents.extend([ int(round(255 * (0.5 - 0.5 * sin_u))),
int(round(255 * (0.5 + 0.5 * cos_u))),
0 ])
v += v_step
u += u_step
# Create a list of triangle indices.
indices = []
for i in range(slices - 1):
for j in range(inner_slices - 1):
p = i * inner_slices + j
indices.extend([p, p + inner_slices, p + inner_slices + 1])
indices.extend([p, p + inner_slices + 1, p + 1])
self.vertex_list = batch.add_indexed(len(vertices)//3,
GL_TRIANGLES,
group,
indices,
('v3f/static', vertices),
('n3f/static', normals),
('t3f/static', textureuvw),
('c3B/static', tangents))
def delete(self):
self.vertex_list.delete()
class Sphere(object):
vv = [] # vertex vectors
vcount = 0
vertices = []
normals = []
textureuvw = []
tangents = []
indices = []
def splitTriangle(self, i1, i2, i3):
'''
Interpolates and Normalizes 3 Vectors p1, p2, p3.
Result is an Array of 4 Triangles
'''
p12 = self.vv[i1] + self.vv[i2]
p13 = self.vv[i1] + self.vv[i3]
p23 = self.vv[i2] + self.vv[i3]
self.vv.append( p12.normalize() )
self.vv.append( p13.normalize() )
self.vv.append( p23.normalize() )
rslt = []
rslt.append([i1, self.vcount+0, self.vcount+1])
rslt.append([self.vcount+0, i2, self.vcount+2])
rslt.append([self.vcount+0, self.vcount+2, self.vcount+1])
rslt.append([self.vcount+1, self.vcount+2, i3])
self.vcount += 3
return rslt
def recurseTriangle(self, i1, i2, i3, level):
if level > 0: # split in 4 triangles
p1, p2, p3, p4 = self.splitTriangle( i1, i2, i3 )
self.recurseTriangle( *p1, level=level-1 )
self.recurseTriangle( *p2, level=level-1 )
self.recurseTriangle( *p3, level=level-1 )
self.recurseTriangle( *p4, level=level-1 )
else:
self.indices.extend( [i1, i2, i3] ) # just MAKE the triangle
def flatten(self, x):
"""flatten(sequence) -> list
Returns a single, flat list which contains all elements retrieved
from the sequence and all recursively contained sub-sequences
(iterables).
Examples:
>>> [1, 2, [3,4], (5,6)]
[1, 2, [3, 4], (5, 6)]
>>> flatten([[[1,2,3], (42,None)], [4,5], [6], 7, MyVector(8,9,10)])
[1, 2, 3, 42, None, 4, 5, 6, 7, 8, 9, 10]"""
result = []
for el in x:
#if isinstance(el, (list, tuple)):
if hasattr(el, "__iter__") and not isinstance(el, basestring):
result.extend(self.flatten(el))
else:
result.append(el)
return result
def __init__(self, radius, slices, batch, group=None):
# Create the vertex and normal arrays.
self.vv.append( Vector3(0.0, 0.0, 1.0) ) # North
self.vv.append( Vector3(0.0, 0.0, -1.0) ) # South
self.vv.append( Vector3(1.0, 0.0, 0.0) ) # A
self.vv.append( Vector3(0.0, 1.0, 0.0) ) # B
self.vv.append( Vector3(-1.0, 0.0, 0.0) ) # C
self.vv.append( Vector3(0.0, -1.0, 0.0) ) # D
self.vcount = 6
self.recurseTriangle( 0, 2, 3, slices )
self.recurseTriangle( 0, 3, 4, slices )
self.recurseTriangle( 0, 4, 5, slices )
self.recurseTriangle( 0, 5, 2, slices )
self.recurseTriangle( 1, 3, 2, slices )
self.recurseTriangle( 1, 4, 3, slices )
self.recurseTriangle( 1, 5, 4, slices )
self.recurseTriangle( 1, 2, 5, slices )
for v in range(self.vcount):
self.normals.extend(self.vv[v][:])
# equal area projection, see http://www.uwgb.edu/dutchs/structge/sphproj.htm
uv = Vector2(self.vv[v][0], self.vv[v][1])
if abs(uv) > 1E-5:
uv = uv.normalized() * abs( self.vv[v] + Vector3(0, 0, -1))
self.textureuvw.extend([uv[0],uv[1], 0.0])
uvw = Vector3( self.vv[v][1], -self.vv[v][0], 0.0 ) # does not completely fit the ea-projection !
if abs( uvw ) > 1E-5:
uvw.normalize()
self.tangents.extend([ int(round(255 * (0.5 - 0.5 * uvw[0]))),
int(round(255 * (0.5 - 0.5 * uvw[1]))),
int(round(255 * (0.5 - 0.5 * uvw[2]))) ])
self.vv = [(x * radius) for x in self.vv]
self.vertices = [x[:] for x in self.vv]
self.vertices = self.flatten( self.vertices )
self.vertex_list = batch.add_indexed(len(self.vertices)//3,
GL_TRIANGLES,
group,
self.indices,
('v3f/static', self.vertices),
('n3f/static', self.normals),
('t3f/static', self.textureuvw),
('c3B/static', self.tangents))
def delete(self):
self.vertex_list.delete()
rot = Vector3(0, 0, 0)
autorotate = True
rotstep = 10
lightstep = 10 * pi/180
togglefigure = False
toggletexture = True
showdialog = True
shaderon = True
setup()
batch1 = pyglet.graphics.Batch()
torus = Torus(1, 0.3, 80, 25, batch=batch1)
batch2 = pyglet.graphics.Batch()
sphere = Sphere(1.2, 4, batch=batch2)
pyglet.app.run()
#thats all