1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
from ctypes import *
import sys
import pygame
from pygame.locals import *
try:
# For OpenGL-ctypes
from OpenGL import platform
gl = platform.OpenGL
except ImportError:
# For PyOpenGL
gl = cdll.LoadLibrary('libGL.so')
from OpenGL.GL import *
from OpenGL.GLU import *
from OpenGL.GLUT import *
GL_FRAGMENT_SHADER_ARB = 0x8B30
GL_VERTEX_SHADER_ARB = 0x8B31
GL_OBJECT_COMPILE_STATUS_ARB= 0x8B81
GL_OBJECT_LINK_STATUS_ARB = 0x8B82
GL_INFO_LOG_LENGTH_ARB = 0x8B84
glCreateShaderObjectARB = gl.glCreateShaderObjectARB
glShaderSourceARB = gl.glShaderSourceARB
glShaderSourceARB.argtypes = [c_int, c_int, POINTER(c_char_p), POINTER(c_int)]
glCompileShaderARB = gl.glCompileShaderARB
glGetObjectParameterivARB = gl.glGetObjectParameterivARB
glGetObjectParameterivARB.argtypes = [c_int, c_int, POINTER(c_int)]
glCreateProgramObjectARB = gl.glCreateProgramObjectARB
glGetInfoLogARB = gl.glGetShaderInfoLog
glGetInfoLogARB.argtypes = [c_int, c_int, POINTER(c_int), c_char_p]
glAttachObjectARB = gl.glAttachObjectARB
glLinkProgramARB = gl.glLinkProgramARB
glDeleteObjectARB = gl.glDeleteObjectARB
glGetError = gl.glGetError
glUseProgramObjectARB = gl.glUseProgramObjectARB
def print_log(shader):
length = c_int()
glGetObjectParameterivARB(shader, GL_INFO_LOG_LENGTH, byref(length))
if length.value > 0:
log = create_string_buffer(length.value)
glGetInfoLogARB(shader, length, byref(length), log)
print >> sys.stderr, log.value
def compile_shader(source, shader_type):
shader = glCreateShaderObjectARB(shader_type)
source = c_char_p(source)
length = c_int(-1)
glShaderSourceARB(shader, 1, byref(source), byref(length))
glCompileShaderARB(shader)
status = c_int()
glGetObjectParameterivARB(shader, GL_OBJECT_COMPILE_STATUS_ARB,
byref(status))
if (not status.value):
print_log(shader)
raise SystemExit
return shader
def compile_program(vertex_source, fragment_source):
vertex_shader = None
fragment_shader = None
program = glCreateProgramObjectARB()
if vertex_source:
vertex_shader = compile_shader(vertex_source, GL_VERTEX_SHADER_ARB)
glAttachObjectARB(program, vertex_shader)
if fragment_source:
fragment_shader = compile_shader(fragment_source,
GL_FRAGMENT_SHADER_ARB)
glAttachObjectARB(program, fragment_shader)
glLinkProgramARB(program)
if vertex_shader:
glDeleteObjectARB(vertex_shader)
if fragment_shader:
glDeleteObjectARB(fragment_shader)
return program
def toon_program():
return compile_program('''
varying vec3 normal;
void main() {
normal = gl_NormalMatrix * gl_Normal;
gl_Position = gl_ModelViewProjectionMatrix * gl_Vertex;
gl_FrontColor = gl_Color;
}
''', '''
varying vec3 normal;
void main() {
float intensity;
vec3 n = normalize(normal);
vec3 l = normalize(gl_LightSource[0].position).xyz;
// quantize to 5 steps (.4, .6, .8 and 1)
intensity = dot(l, n);
if (intensity < .15) intensity = .5;
else if (intensity >= .15 && intensity < .5) intensity = .7;
else if (intensity >= .5 && intensity < .75) intensity = .9;
else intensity = 1;
gl_FragColor = gl_Color * intensity;
}
''')
def outline_program():
return compile_program('''
uniform vec3 camPos;
uniform float outlineThreshold;
uniform float edgeThreshold;
varying vec3 norm;
void main(){
vec4 pos = gl_Vertex;
vec3 dir = camPos - gl_Vertex.xyz;
pos.w = float(
dot(dir, gl_MultiTexCoord0.xyz) *
dot(dir, gl_MultiTexCoord1.xyz) < outlineThreshold ||
dot(gl_MultiTexCoord0.xyz, gl_MultiTexCoord1.xyz) < edgeThreshold);
gl_Position = gl_ModelViewProjectionMatrix * pos;
}
''', '''
varying vec3 norm;
void main(){
gl_FragColor = vec4(0.0);
}
''')
if __name__ == '__main__':
glutInit(sys.argv)
width, height = 640, 480
pygame.init()
pygame.display.set_mode((width, height), OPENGL | DOUBLEBUF)
program = toon_program()
glMatrixMode(GL_PROJECTION)
glLoadIdentity()
gluPerspective(90.0, width/float(height), 1.0, 100.0)
glMatrixMode(GL_MODELVIEW)
glEnable(GL_DEPTH_TEST)
quit = False
angle = 0
while not quit:
for e in pygame.event.get():
if e.type in (QUIT, KEYDOWN):
quit = True
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
glLoadIdentity()
glTranslate(0.0, 0.0, -2.5)
glRotate(angle, 0.0, 1.0, 0.0)
glUseProgramObjectARB(program)
glColor(1.0, .5, .5, 1.0)
glutSolidTeapot(1.0)
angle += 0.1
pygame.display.flip()
|