File: runme.py

package info (click to toggle)
swig1.3 1.3.24-1
  • links: PTS
  • area: main
  • in suites: sarge
  • size: 19,336 kB
  • ctags: 10,604
  • sloc: cpp: 27,917; ansic: 24,160; yacc: 4,412; python: 4,255; java: 4,156; makefile: 3,735; sh: 3,552; cs: 2,250; ruby: 2,150; lisp: 1,605; tcl: 1,136; perl: 980; php: 879; ml: 825
file content (64 lines) | stat: -rw-r--r-- 1,371 bytes parent folder | download | duplicates (4)
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
# Plot a 3D function
# This example uses the low-level C interface.

from _gifplot import *
from math import *

# Here is the function to plot
def func(x,y):
	return 5*cos(2*sqrt(x*x+y*y))*exp(-0.3*sqrt(x*x+y*y))

# Here are some plotting parameters
xmin = -5.0
xmax =  5.0
ymin = -5.0
ymax =  5.0
zmin = -5.0
zmax =  5.0

# Grid resolution
nxpoints =  60
nypoints =  60

cmap  = new_ColorMap("cmap")
frame = new_FrameBuffer(500,500)
FrameBuffer_clear(frame,BLACK)

p3    = new_Plot3D(frame,xmin,ymin,zmin,xmax,ymax,zmax)
Plot3D_lookat(p3,2*max([xmax-xmin,ymax-ymin,zmax-zmin]))
Plot3D_autoperspective(p3,40)
Plot3D_rotu(p3,60)
Plot3D_rotr(p3,30)
Plot3D_rotd(p3,10)

def drawsolid():
	Plot3D_clear(p3,BLACK)
	Plot3D_start(p3)
	dx = 1.0*(xmax-xmin)/nxpoints
	dy = 1.0*(ymax-ymin)/nypoints
	cscale = 240.0/(zmax-zmin)
	x = xmin
	for i in xrange(0,nxpoints):
		y = ymin
		for j in xrange(0,nypoints):
			z1 = func(x,y)
			z2 = func(x+dx,y)
			z3 = func(x+dx,y+dy)
			z4 = func(x,y+dy)
			c1 = cscale*(z1-zmin)
			c2 = cscale*(z2-zmin)
			c3 = cscale*(z3-zmin)
			c4 = cscale*(z4-zmin)
			c = int((c1+c2+c3+c4)/4)
			if (c < 0) : c = 0
			if c > 239 : c = 239
			Plot3D_solidquad(p3,x,y,z1,x+dx,y,z2,x+dx,y+dy,z3,x,y+dy,z4,c+16)
			y = y + dy		
		x = x + dx

print "Making a nice 3D plot..."
drawsolid()

FrameBuffer_writeGIF(frame,cmap,"image.gif")
print "Wrote image.gif"