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