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
|
#! /usr/bin/python
from OpenGL.GLUT import *
from OpenGL.GLU import *
from OpenGL.GL import *
import sys
from readmmp import readmmp
from bondage import *
from RandomArray import *
MOUSE_WHEEL_UP = 3
MOUSE_WHEEL_DOWN = 4
name = 'Developmental Simulator'
def glutsetup():
glutInit(sys.argv)
glutInitDisplayMode(GLUT_DOUBLE | GLUT_RGB)
glutInitWindowSize(800,600)
glutCreateWindow(name)
glClearColor(0.,0.,0.,1.)
glShadeModel(GL_SMOOTH)
glEnable(GL_CULL_FACE)
glEnable(GL_DEPTH_TEST)
glEnable(GL_LIGHTING)
lightZeroPosition = [10.,4.,10.,1.]
lightZeroColor = [0.8,1.0,0.8,1.0] #green tinged
glLightfv(GL_LIGHT0, GL_POSITION, lightZeroPosition)
glLightfv(GL_LIGHT0, GL_DIFFUSE, lightZeroColor)
glLightf(GL_LIGHT0, GL_CONSTANT_ATTENUATION, 0.1)
glLightf(GL_LIGHT0, GL_LINEAR_ATTENUATION, 0.05)
glEnable(GL_LIGHT0)
glutDisplayFunc(display)
glMatrixMode(GL_PROJECTION)
gluPerspective(40.,1.,1.,40.)
glMatrixMode(GL_MODELVIEW)
gluLookAt(0,0,10,
0,0,0,
0,1,0)
def display():
glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT)
glMaterialfv(GL_FRONT,GL_DIFFUSE,[1.0,1.0,1.0,1.])
glBegin(GL_LINES)
for (a,b,o) in bonds:
glVertex(pos[:,a])
glVertex(pos[:,b])
glEnd()
glMaterialfv(GL_FRONT,GL_DIFFUSE,[1.0,0.5,0.5,1.])
glBegin(GL_LINES)
for (a,o1,b,o2,c) in bends:
glVertex(pos[:,a])
glVertex(pos[:,c])
glEnd()
for i in range(len(elt)):
glPushMatrix()
glTranslate(pos[0,i],pos[1,i],pos[2,i])
glutSolidSphere(0.7,40,20)
glPopMatrix()
glutSwapBuffers()
return
def step():
global new, pos, old
old=pos
pos=new
new=2*pos-old+massacc*force(pos)
display()
def main():
global elt, pos, new, bonds, bends, massacc
elt, pos, bonds = readmmp(sys.argv[1])
bends = bondsetup(bonds)
dt=1e-16
massacc=array([dt*dt/elmass[x] for x in elt])
new=pos+massacc*455.0*standard_normal(shape(pos))
glutsetup()
glutIdleFunc(step)
glutMainLoop()
if __name__ == '__main__': main()
|
