mirror of https://github.com/python/cpython.git
427 lines
12 KiB
Python
Executable File
427 lines
12 KiB
Python
Executable File
#! /usr/bin/env python
|
|
# Simulate the artwork in the hall.
|
|
# Jack Jansen, Feb 91.
|
|
|
|
from gl import *
|
|
from GL import *
|
|
from math import *
|
|
from DEVICE import *
|
|
import sys
|
|
import __main__
|
|
main_dict = __main__.__dict__
|
|
|
|
SPOTDIRECTION = 103
|
|
SPOTLIGHT = 104
|
|
|
|
#
|
|
# Make a cylinder paralel with the Z axis with center (X,Y,0)
|
|
# and radius 1
|
|
def mkcyl(nslice, nparts, docircle):
|
|
cyl = []
|
|
step = 2.0 / float(nslice)
|
|
z = -1.0
|
|
for i in range(nslice):
|
|
cyl.append(mkslice(z, z+step, nparts, docircle))
|
|
z = z + step
|
|
return drawcylinder(cyl)
|
|
#
|
|
# Make one part of a cylinder
|
|
#
|
|
def mkslice(z1, z2, nparts, docircle):
|
|
if docircle:
|
|
w1 = z1
|
|
w2 = z2
|
|
w1 = sqrt(1.0-w1*w1)
|
|
w2 = sqrt(1.0-w2*w2)
|
|
normalz = 1.0
|
|
else:
|
|
w1 = 1.0
|
|
w2 = 1.0
|
|
normalz = 0.0
|
|
slice = []
|
|
step = (2.0*pi)/float(nparts)
|
|
angle = 0.0
|
|
for i in range(nparts+1):
|
|
vx = cos(angle)
|
|
vy = sin(angle)
|
|
slice.append( ((vx*w1,vy*w1,z1), (vx*w1, vy*w1, z1*normalz)) )
|
|
slice.append( ((vx*w2,vy*w2,z2), (vx*w2, vy*w2, z2*normalz)) )
|
|
angle = angle + step
|
|
return slice
|
|
#
|
|
# Drawcylinder : draw the cylinder
|
|
#
|
|
class struct: pass
|
|
curobj = struct()
|
|
curobj.curobj = 1
|
|
def drawcylinder(cyl):
|
|
obj = curobj.curobj
|
|
curobj.curobj = curobj.curobj+1
|
|
makeobj(obj)
|
|
for slice in cyl:
|
|
bgntmesh()
|
|
vnarray(slice)
|
|
endtmesh()
|
|
closeobj()
|
|
return obj
|
|
#
|
|
def drawnormals(cyl):
|
|
for slice in cyl:
|
|
for triang in slice:
|
|
bgnline()
|
|
v3f(triang[0])
|
|
v3f(triang[0][0] + triang[1][0], triang[0][1] + triang[1][1], triang[0][2] + triang[1][2])
|
|
endline()
|
|
def drawfloors():
|
|
obj = curobj.curobj
|
|
curobj.curobj = curobj.curobj+1
|
|
makeobj(obj)
|
|
bgnpolygon()
|
|
v3i(4,6,-6)
|
|
v3i(-6,6,-6)
|
|
v3i(-6,-6,-6)
|
|
v3i(4,-6,-6)
|
|
endpolygon()
|
|
for floor in range(3):
|
|
pos = -1 + 5*floor
|
|
bgnpolygon()
|
|
v3i(4,4,pos)
|
|
v3i(-6,4,pos)
|
|
v3i(-6,6,pos)
|
|
v3i(4,6,pos)
|
|
endpolygon()
|
|
bgnpolygon()
|
|
v3i(-4,4,pos)
|
|
v3i(-4,-4,pos)
|
|
v3i(-6,-4,pos)
|
|
v3i(-6,4,pos)
|
|
endpolygon()
|
|
bgnpolygon()
|
|
v3i(-6,-4,pos)
|
|
v3i(-6,-6,pos)
|
|
v3i(4,-6,pos)
|
|
v3i(4,-4,pos)
|
|
endpolygon()
|
|
closeobj()
|
|
return obj
|
|
def drawdoors():
|
|
obj = curobj.curobj
|
|
curobj.curobj = curobj.curobj+1
|
|
makeobj(obj)
|
|
for floor in range(3):
|
|
pos = -1+5*floor
|
|
bgnpolygon()
|
|
v3i(-2,6,pos)
|
|
v3i(-2,6,pos+3)
|
|
v3i(0,6,pos+3)
|
|
v3i(0,6,pos)
|
|
endpolygon()
|
|
closeobj()
|
|
return obj
|
|
def drawrailing():
|
|
obj = curobj.curobj
|
|
curobj.curobj = curobj.curobj+1
|
|
makeobj(obj)
|
|
for floor in range(3):
|
|
pos = -1 + 5*floor
|
|
bgnpolygon()
|
|
v3i(4,4,pos)
|
|
v3i(4,4,pos-1)
|
|
v3i(-4,4,pos-1)
|
|
v3i(-4,4,pos)
|
|
endpolygon()
|
|
bgnpolygon()
|
|
v3i(-4,4,pos)
|
|
v3i(-4,4,pos-1)
|
|
v3i(-4,-4,pos-1)
|
|
v3i(-4,-4,pos)
|
|
endpolygon()
|
|
bgnpolygon()
|
|
v3i(-4,-4,pos)
|
|
v3i(-4,-4,pos-1)
|
|
v3i(4,-4,pos-1)
|
|
v3i(4,-4,pos)
|
|
endpolygon()
|
|
closeobj()
|
|
return obj
|
|
def drawwalls():
|
|
obj = curobj.curobj
|
|
curobj.curobj = curobj.curobj+1
|
|
makeobj(obj)
|
|
bgnpolygon()
|
|
v3i(4,6,-6)
|
|
v3i(4,6,18)
|
|
v3i(-6,6,18)
|
|
v3i(-6,6,-6)
|
|
endpolygon()
|
|
bgnpolygon()
|
|
v3i(-6,6,-6)
|
|
v3i(-6,6,18)
|
|
v3i(-6,-6,18)
|
|
v3i(-6,-6,-6)
|
|
endpolygon()
|
|
bgnpolygon()
|
|
v3i(-6,-6,-6)
|
|
v3i(-6,-6,18)
|
|
v3i(4,-6,18)
|
|
v3i(4,-6,-6)
|
|
endpolygon()
|
|
bgnpolygon()
|
|
v3i(4,-6,-6)
|
|
v3i(4,-6,18)
|
|
v3i(4,4,18)
|
|
v3i(4,4,-6)
|
|
endpolygon()
|
|
closeobj()
|
|
return obj
|
|
def axis():
|
|
bgnline()
|
|
cpack(0xff0000)
|
|
v3i(-1,0,0)
|
|
v3i(1,0,0)
|
|
v3f(1.0, 0.1, 0.1)
|
|
endline()
|
|
bgnline()
|
|
cpack(0xff00)
|
|
v3i(0,-1,0)
|
|
v3i(0,1,0)
|
|
v3f(0.1, 1.0, 0.1)
|
|
endline()
|
|
bgnline()
|
|
cpack(0xff)
|
|
v3i(0,0,-1)
|
|
v3i(0,0,1)
|
|
v3f(0.1,0.1,1.0)
|
|
endline()
|
|
#
|
|
green_velvet = [ DIFFUSE, 0.05, 0.4, 0.05, LMNULL]
|
|
silver = [ DIFFUSE, 0.3, 0.3, 0.3, SPECULAR, 0.9, 0.9, 0.95, \
|
|
SHININESS, 40.0, LMNULL]
|
|
floormat = [ AMBIENT, 0.5, 0.25, 0.15, DIFFUSE, 0.5, 0.25, 0.15, SPECULAR, 0.6, 0.3, 0.2, SHININESS, 20.0, LMNULL]
|
|
wallmat = [ DIFFUSE, 0.4, 0.2, 0.1, AMBIENT, 0.4, 0.20, 0.10, SPECULAR, 0.0, 0.0, 0.0, SHININESS, 20.0, LMNULL]
|
|
offwhite = [ DIFFUSE, 0.8, 0.8, 0.6, AMBIENT, 0.8, 0.8, 0.6, SPECULAR, 0.9, 0.9, 0.9, SHININESS, 30.0, LMNULL]
|
|
doormat = [ DIFFUSE, 0.1, 0.2, 0.5, AMBIENT, 0.2, 0.4, 1.0, SPECULAR, 0.2, 0.4, 1.0, SHININESS, 60.0, LMNULL]
|
|
|
|
toplight = [ LCOLOR, 1.0, 1.0, 0.5, \
|
|
POSITION, 0.0, 0.0, 11.0, 1.0, LMNULL]
|
|
floor1light = [ LCOLOR, 1.0, 1.0, 1.0, POSITION, 3.9, -3.9, 0.0, 1.0, \
|
|
SPOTDIRECTION, 1.0, 1.0, 0.0, SPOTLIGHT, 10.0, 90.0, LMNULL]
|
|
|
|
lmodel = [ AMBIENT, 0.92, 0.8, 0.5, LOCALVIEWER, 1.0, LMNULL]
|
|
#
|
|
def lighting():
|
|
lmdef(DEFMATERIAL, 1, green_velvet)
|
|
lmdef(DEFMATERIAL, 2, silver)
|
|
lmdef(DEFMATERIAL, 3, floormat)
|
|
lmdef(DEFMATERIAL, 4, wallmat)
|
|
lmdef(DEFMATERIAL, 5, offwhite)
|
|
lmdef(DEFMATERIAL, 6, doormat)
|
|
lmdef(DEFLIGHT, 1, toplight)
|
|
lmdef(DEFLIGHT, 2, floor1light)
|
|
lmdef(DEFLMODEL, 1, lmodel)
|
|
lmbind(MATERIAL, 1)
|
|
lmbind(LIGHT0, 1)
|
|
lmbind(LIGHT1, 2)
|
|
lmbind(LMODEL, 1)
|
|
IdMat=[1.0,0.0,0.0,0.0, 0.0,1.0,0.0,0.0, 0.0,0.0,1.0,0.0, 0.0,0.0,0.0,1.0]
|
|
#
|
|
def defun(axis):
|
|
done = 0
|
|
while not done:
|
|
print 'F'+axis+'(t) = ',
|
|
s = sys.stdin.readline(100)
|
|
print
|
|
try:
|
|
s = 'def f'+axis+'(t): return '+s
|
|
exec(s, main_dict)
|
|
done = 1
|
|
except RuntimeError:
|
|
print 'Sorry, there is a syntax error in your expression'
|
|
def getfunctions():
|
|
print 'Welcome to the CWI art simulator. You can now enter X, Y and Z'
|
|
print 'coordinates as a function of t.'
|
|
print 'Normal trig functions are available. Please use floating point'
|
|
print 'values only (so 0.0 for 0). Comments to jack@cwi.nl'
|
|
defun('x')
|
|
defun('y')
|
|
defun('z')
|
|
print 'Ok, here you go. Use mouse+right button to move up/down,'
|
|
print 'mouse+middle to speed up/slow down time. type ESC to quit simulation'
|
|
def main():
|
|
getfunctions()
|
|
foreground()
|
|
prefposition(100,600,100,600)
|
|
void = winopen('cyl')
|
|
qdevice(ESCKEY)
|
|
qdevice(MOUSE1)
|
|
qdevice(MOUSE2)
|
|
qdevice(PKEY)
|
|
RGBmode()
|
|
doublebuffer()
|
|
gconfig()
|
|
zbuffer(1)
|
|
mmode(MVIEWING)
|
|
perspective(400, 1.0, 1.0, 20.0)
|
|
loadmatrix(IdMat)
|
|
vx = 0.0
|
|
vy = -6.0
|
|
vz = 0.0
|
|
lookat(0.0, -6.0, 0.0, 0.0, 0.0, 0.0, 0)
|
|
lighting()
|
|
t = -1.0
|
|
step = 1.0
|
|
bol = mkcyl(12,24, 1)
|
|
cable = mkcyl(1, 6, 0)
|
|
floors = drawfloors()
|
|
walls = drawwalls()
|
|
pillar = mkcyl(1,4,0)
|
|
railing = drawrailing()
|
|
doors = drawdoors()
|
|
shademodel(GOURAUD)
|
|
mousing = -1
|
|
pausing = 0
|
|
while 1:
|
|
#
|
|
# Check for some user input
|
|
#
|
|
if qtest():
|
|
dev, value = qread()
|
|
if dev == PKEY and value == 1:
|
|
pausing = 1
|
|
if dev == ESCKEY:
|
|
break
|
|
elif (dev==MOUSE1 or dev==MOUSE2) and value == 1:
|
|
if mousing > 0:
|
|
vx = 0.0
|
|
vy = -6.0
|
|
vz = 0.0
|
|
mousing = dev
|
|
oldx = getvaluator(MOUSEX)
|
|
oldy = getvaluator(MOUSEY)
|
|
elif (dev==MOUSE1 or dev==MOUSE2):
|
|
mousing = -1
|
|
if mousing >= 0:
|
|
newx = getvaluator(MOUSEX)
|
|
newy = getvaluator(MOUSEY)
|
|
if newy <> oldy and mousing==MOUSE1:
|
|
vz = vz + float(newy - oldy)/100.0
|
|
dist = sqrt(vx*vx + vy*vy + vz*vz)
|
|
perspective(400, 1.0, 1.0, dist+16.0)
|
|
loadmatrix(IdMat)
|
|
if vz < 0.0:
|
|
lookat(vx, vy, vz, 0.0, 0.0, 0.0, 1800)
|
|
else:
|
|
lookat(vx, vy, vz, 0.0, 0.0, 0.0, 0)
|
|
if newy <> oldy and mousing==MOUSE2:
|
|
step = step * exp(float(newy-oldy)/400.0)
|
|
if getbutton(CTRLKEY) == 0:
|
|
t = t + step
|
|
else:
|
|
t = t - step
|
|
if getbutton(LEFTSHIFTKEY) == 0:
|
|
shademodel(GOURAUD)
|
|
else:
|
|
shademodel(FLAT)
|
|
#
|
|
# Draw background and axis
|
|
cpack(0x105090)
|
|
clear()
|
|
zclear()
|
|
cpack(0x905010)
|
|
axis()
|
|
#
|
|
# Draw object
|
|
#
|
|
bolx = fx(t)
|
|
boly = fy(t)
|
|
bolz = fz(t)
|
|
err = ''
|
|
if bolx < -4.0 or bolx > 4.0:
|
|
err = 'X('+`bolx`+') out of range [-4,4]'
|
|
if boly < -4.0 or boly > 4.0:
|
|
err = 'Y('+`boly`+') out of range [-4,4]'
|
|
if bolz < -4.0 or bolz > 8.0:
|
|
err = 'Z('+`bolz`+') out of range [-4,8]'
|
|
if not err:
|
|
pushmatrix()
|
|
translate(bolx, boly, bolz)
|
|
scale(0.3, 0.3, 0.3)
|
|
lmbind(MATERIAL, 2)
|
|
callobj(bol)
|
|
popmatrix()
|
|
#
|
|
# Draw the cables
|
|
#
|
|
bolz = bolz + 0.3
|
|
pushmatrix()
|
|
linesmooth(SML_ON)
|
|
bgnline()
|
|
v3i(-4,-4,9)
|
|
v3f(bolx, boly, bolz)
|
|
endline()
|
|
bgnline()
|
|
v3i(-4,4,9)
|
|
v3f(bolx, boly, bolz)
|
|
endline()
|
|
bgnline()
|
|
v3i(4,-4,9)
|
|
v3f(bolx, boly, bolz)
|
|
endline()
|
|
bgnline()
|
|
v3i(4,4,9)
|
|
v3f(bolx, boly, bolz)
|
|
endline()
|
|
popmatrix()
|
|
#
|
|
# draw the floors
|
|
#
|
|
lmbind(MATERIAL, 3)
|
|
callobj(floors)
|
|
lmbind(MATERIAL, 4)
|
|
callobj(walls)
|
|
lmbind(MATERIAL, 5)
|
|
pushmatrix()
|
|
translate(-4.5,4.5,3.0)
|
|
scale(0.2,0.2,9.0)
|
|
rotate(450,'z')
|
|
callobj(pillar)
|
|
popmatrix()
|
|
callobj(railing)
|
|
lmbind(MATERIAL, 6)
|
|
pushmatrix()
|
|
translate(0.0, -0.01, 0.0)
|
|
callobj(doors)
|
|
popmatrix()
|
|
if mousing == MOUSE2 or err:
|
|
cpack(0xff0000)
|
|
cmov(0.0, 0.0, 0.4)
|
|
charstr('t='+`t`)
|
|
if mousing == MOUSE2:
|
|
cpack(0xff0000)
|
|
cmov(0.0, 0.0, 0.2)
|
|
charstr('delta-t='+`step`)
|
|
if err:
|
|
cpack(0xff00)
|
|
cmov(0.0, 0.0, 0.2)
|
|
print err
|
|
charstr(err)
|
|
pausing = 1
|
|
if pausing:
|
|
cpack(0xff00)
|
|
cmov(0.0, 0.0, 0.0)
|
|
charstr('Pausing, type P to continue')
|
|
swapbuffers()
|
|
if pausing:
|
|
while 1:
|
|
dv=qread()
|
|
if dv==(PKEY,1):
|
|
break
|
|
if dv==(ESCKEY,1):
|
|
sys.exit(0)
|
|
pausing = 0
|
|
#
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
sys.exit(1)
|