]> git.saurik.com Git - wxWidgets.git/blame_incremental - wxPython/demo/GLCanvas.py
Fixed OOR typo
[wxWidgets.git] / wxPython / demo / GLCanvas.py
... / ...
CommitLineData
1
2import wx
3
4try:
5 from wx import glcanvas
6 haveGLCanvas = True
7except ImportError:
8 haveGLCanvas = False
9
10try:
11 # The Python OpenGL package can be found at
12 # http://PyOpenGL.sourceforge.net/
13 from OpenGL.GL import *
14 from OpenGL.GLUT import *
15 haveOpenGL = True
16except ImportError:
17 haveOpenGL = False
18
19#----------------------------------------------------------------------
20
21if not haveGLCanvas:
22 def runTest(frame, nb, log):
23 dlg = wx.MessageDialog(frame, 'The GLCanvas class has not been included with this build of wxPython!',
24 'Sorry', wx.OK | wx.ICON_INFORMATION)
25 dlg.ShowModal()
26 dlg.Destroy()
27
28elif not haveOpenGL:
29 def runTest(frame, nb, log):
30 dlg = wx.MessageDialog(frame,
31 'The OpenGL package was not found. You can get it at\n'
32 'http://PyOpenGL.sourceforge.net/',
33 'Sorry', wx.OK | wx.ICON_INFORMATION)
34 dlg.ShowModal()
35 dlg.Destroy()
36
37
38
39
40else:
41 buttonDefs = {
42 wx.NewId() : ('CubeCanvas', 'Cube'),
43 wx.NewId() : ('ConeCanvas', 'Cone'),
44 }
45
46 class ButtonPanel(wx.Panel):
47 def __init__(self, parent, log):
48 wx.Panel.__init__(self, parent, -1)
49 self.log = log
50
51 box = wx.BoxSizer(wx.VERTICAL)
52 box.Add((20, 30))
53 keys = buttonDefs.keys()
54 keys.sort()
55 for k in keys:
56 text = buttonDefs[k][1]
57 btn = wx.Button(self, k, text)
58 box.Add(btn, 0, wx.ALIGN_CENTER|wx.ALL, 15)
59 self.Bind(wx.EVT_BUTTON, self.OnButton, btn)
60
61 #** Enable this to show putting a GLCanvas on the wx.Panel
62 if 0:
63 c = CubeCanvas(self)
64 c.SetSize((200, 200))
65 box.Add(c, 0, wx.ALIGN_CENTER|wx.ALL, 15)
66
67 self.SetAutoLayout(True)
68 self.SetSizer(box)
69
70
71 def OnButton(self, evt):
72 canvasClassName = buttonDefs[evt.GetId()][0]
73 canvasClass = eval(canvasClassName)
74 frame = wx.Frame(None, -1, canvasClassName, size=(400,400))
75 canvas = canvasClass(frame)
76 frame.Show(True)
77
78
79
80 def runTest(frame, nb, log):
81 win = ButtonPanel(nb, log)
82 return win
83
84
85
86
87 class MyCanvasBase(glcanvas.GLCanvas):
88 def __init__(self, parent):
89 glcanvas.GLCanvas.__init__(self, parent, -1)
90 self.init = False
91 # initial mouse position
92 self.lastx = self.x = 30
93 self.lasty = self.y = 30
94 self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)
95 self.Bind(wx.EVT_SIZE, self.OnSize)
96 self.Bind(wx.EVT_PAINT, self.OnPaint)
97 self.Bind(wx.EVT_LEFT_DOWN, self.OnMouseDown)
98 self.Bind(wx.EVT_LEFT_UP, self.OnMouseUp)
99 self.Bind(wx.EVT_MOTION, self.OnMouseMotion)
100
101
102 def OnEraseBackground(self, event):
103 pass # Do nothing, to avoid flashing on MSW.
104
105
106 def OnSize(self, event):
107 size = self.GetClientSize()
108 if self.GetContext():
109 self.SetCurrent()
110 glViewport(0, 0, size.width, size.height)
111 event.Skip()
112
113
114 def OnPaint(self, event):
115 dc = wx.PaintDC(self)
116 self.SetCurrent()
117 if not self.init:
118 self.InitGL()
119 self.init = True
120 self.OnDraw()
121
122
123 def OnMouseDown(self, evt):
124 self.CaptureMouse()
125
126
127 def OnMouseUp(self, evt):
128 self.ReleaseMouse()
129
130
131 def OnMouseMotion(self, evt):
132 if evt.Dragging() and evt.LeftIsDown():
133 self.x, self.y = self.lastx, self.lasty
134 self.x, self.y = evt.GetPosition()
135 self.Refresh(False)
136
137
138
139
140 class CubeCanvas(MyCanvasBase):
141 def InitGL(self):
142 # set viewing projection
143 glMatrixMode(GL_PROJECTION);
144 glFrustum(-0.5, 0.5, -0.5, 0.5, 1.0, 3.0);
145
146 # position viewer
147 glMatrixMode(GL_MODELVIEW);
148 glTranslatef(0.0, 0.0, -2.0);
149
150 # position object
151 glRotatef(self.y, 1.0, 0.0, 0.0);
152 glRotatef(self.x, 0.0, 1.0, 0.0);
153
154 glEnable(GL_DEPTH_TEST);
155 glEnable(GL_LIGHTING);
156 glEnable(GL_LIGHT0);
157
158
159 def OnDraw(self):
160 # clear color and depth buffers
161 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
162
163 # draw six faces of a cube
164 glBegin(GL_QUADS)
165 glNormal3f( 0.0, 0.0, 1.0)
166 glVertex3f( 0.5, 0.5, 0.5)
167 glVertex3f(-0.5, 0.5, 0.5)
168 glVertex3f(-0.5,-0.5, 0.5)
169 glVertex3f( 0.5,-0.5, 0.5)
170
171 glNormal3f( 0.0, 0.0,-1.0)
172 glVertex3f(-0.5,-0.5,-0.5)
173 glVertex3f(-0.5, 0.5,-0.5)
174 glVertex3f( 0.5, 0.5,-0.5)
175 glVertex3f( 0.5,-0.5,-0.5)
176
177 glNormal3f( 0.0, 1.0, 0.0)
178 glVertex3f( 0.5, 0.5, 0.5)
179 glVertex3f( 0.5, 0.5,-0.5)
180 glVertex3f(-0.5, 0.5,-0.5)
181 glVertex3f(-0.5, 0.5, 0.5)
182
183 glNormal3f( 0.0,-1.0, 0.0)
184 glVertex3f(-0.5,-0.5,-0.5)
185 glVertex3f( 0.5,-0.5,-0.5)
186 glVertex3f( 0.5,-0.5, 0.5)
187 glVertex3f(-0.5,-0.5, 0.5)
188
189 glNormal3f( 1.0, 0.0, 0.0)
190 glVertex3f( 0.5, 0.5, 0.5)
191 glVertex3f( 0.5,-0.5, 0.5)
192 glVertex3f( 0.5,-0.5,-0.5)
193 glVertex3f( 0.5, 0.5,-0.5)
194
195 glNormal3f(-1.0, 0.0, 0.0)
196 glVertex3f(-0.5,-0.5,-0.5)
197 glVertex3f(-0.5,-0.5, 0.5)
198 glVertex3f(-0.5, 0.5, 0.5)
199 glVertex3f(-0.5, 0.5,-0.5)
200 glEnd()
201
202 glRotatef((self.lasty - self.y)/100., 1.0, 0.0, 0.0);
203 glRotatef((self.lastx - self.x)/100., 0.0, 1.0, 0.0);
204
205 self.SwapBuffers()
206
207
208
209
210
211 class ConeCanvas(MyCanvasBase):
212 def InitGL( self ):
213 glMatrixMode(GL_PROJECTION);
214 # camera frustrum setup
215 glFrustum(-0.5, 0.5, -0.5, 0.5, 1.0, 3.0);
216 glMaterial(GL_FRONT, GL_AMBIENT, [0.2, 0.2, 0.2, 1.0])
217 glMaterial(GL_FRONT, GL_DIFFUSE, [0.8, 0.8, 0.8, 1.0])
218 glMaterial(GL_FRONT, GL_SPECULAR, [1.0, 0.0, 1.0, 1.0])
219 glMaterial(GL_FRONT, GL_SHININESS, 50.0)
220 glLight(GL_LIGHT0, GL_AMBIENT, [0.0, 1.0, 0.0, 1.0])
221 glLight(GL_LIGHT0, GL_DIFFUSE, [1.0, 1.0, 1.0, 1.0])
222 glLight(GL_LIGHT0, GL_SPECULAR, [1.0, 1.0, 1.0, 1.0])
223 glLight(GL_LIGHT0, GL_POSITION, [1.0, 1.0, 1.0, 0.0]);
224 glLightModel(GL_LIGHT_MODEL_AMBIENT, [0.2, 0.2, 0.2, 1.0])
225 glEnable(GL_LIGHTING)
226 glEnable(GL_LIGHT0)
227 glDepthFunc(GL_LESS)
228 glEnable(GL_DEPTH_TEST)
229 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
230 # position viewer
231 glMatrixMode(GL_MODELVIEW);
232
233
234 def OnDraw(self):
235 # clear color and depth buffers
236 glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
237 # use a fresh transformation matrix
238 glPushMatrix()
239 # position object
240 glTranslate(0.0, 0.0, -2.0);
241 glRotate(30.0, 1.0, 0.0, 0.0);
242 glRotate(30.0, 0.0, 1.0, 0.0);
243
244 glTranslate(0, -1, 0)
245 glRotate(250, 1, 0, 0)
246 glutSolidCone(0.5, 1, 30, 5)
247 glPopMatrix()
248 glRotatef((self.lasty - self.y)/100., 0.0, 0.0, 1.0);
249 glRotatef(0.0, (self.lastx - self.x)/100., 1.0, 0.0);
250 # push into visible buffer
251 self.SwapBuffers()
252
253
254
255
256#----------------------------------------------------------------------
257
258
259
260
261overview = """\
262"""
263
264
265
266if __name__ == '__main__':
267 import sys,os
268 import run
269 run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])
270
271
272
273
274#----------------------------------------------------------------------
275