[wxWidgets.git] / wxPython / demo / GLCanvas.py


import wx

try:
    from wx import glcanvas
    haveGLCanvas = True
except ImportError:
    haveGLCanvas = False

try:
    # The Python OpenGL package can be found at
    # http://PyOpenGL.sourceforge.net/
    from OpenGL.GL import *
    from OpenGL.GLUT import *
    haveOpenGL = True
except ImportError:
    haveOpenGL = False

#----------------------------------------------------------------------

if not haveGLCanvas:
    def runTest(frame, nb, log):
        dlg = wx.MessageDialog(frame, 'The GLCanvas class has not been included with this build of wxPython!',
                          'Sorry', wx.OK | wx.ICON_INFORMATION)
        dlg.ShowModal()
        dlg.Destroy()

elif not haveOpenGL:
    def runTest(frame, nb, log):
        dlg = wx.MessageDialog(frame,
                              'The OpenGL package was not found.  You can get it at\n'
                              'http://PyOpenGL.sourceforge.net/',
                          'Sorry', wx.OK | wx.ICON_INFORMATION)
        dlg.ShowModal()
        dlg.Destroy()


else:
    buttonDefs = {
        wx.NewId() : ('CubeCanvas',      'Cube'),
        wx.NewId() : ('ConeCanvas',      'Cone'),
        }

    class ButtonPanel(wx.Panel):
        def __init__(self, parent, log):
            wx.Panel.__init__(self, parent, -1)
            self.log = log

            box = wx.BoxSizer(wx.VERTICAL)
            box.Add((20, 30))
            keys = buttonDefs.keys()
            keys.sort()
            for k in keys:
                text = buttonDefs[k][1]
                btn = wx.Button(self, k, text)
                box.Add(btn, 0, wx.ALIGN_CENTER|wx.ALL, 15)
                self.Bind(wx.EVT_BUTTON, self.OnButton, btn)

            #** Enable this to show putting a GLCanvas on the wx.Panel
            if 0:
                c = CubeCanvas(self)
                c.SetSize((200, 200))
                box.Add(c, 0, wx.ALIGN_CENTER|wx.ALL, 15)

            self.SetAutoLayout(True)
            self.SetSizer(box)


        def OnButton(self, evt):
            canvasClassName = buttonDefs[evt.GetId()][0]
            canvasClass = eval(canvasClassName)
            frame = wx.Frame(None, -1, canvasClassName, size=(400,400))
            canvas = canvasClass(frame)
            frame.Show(True)


    def runTest(frame, nb, log):
        win = ButtonPanel(nb, log)
        return win


    class MyCanvasBase(glcanvas.GLCanvas):
        def __init__(self, parent):
            glcanvas.GLCanvas.__init__(self, parent, -1)
            self.init = False
            # initial mouse position
            self.lastx = self.x = 30
            self.lasty = self.y = 30
            self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)
            self.Bind(wx.EVT_SIZE, self.OnSize)
            self.Bind(wx.EVT_PAINT, self.OnPaint)
            self.Bind(wx.EVT_LEFT_DOWN, self.OnMouseDown)
            self.Bind(wx.EVT_LEFT_UP, self.OnMouseUp)
            self.Bind(wx.EVT_MOTION, self.OnMouseMotion)


        def OnEraseBackground(self, event):
            pass # Do nothing, to avoid flashing on MSW.


        def OnSize(self, event):
            size = self.GetClientSize()
            if self.GetContext():
                self.SetCurrent()
                glViewport(0, 0, size.width, size.height)
            event.Skip()
            

        def OnPaint(self, event):
            dc = wx.PaintDC(self)
            self.SetCurrent()
            if not self.init:
                self.InitGL()
                self.init = True
            self.OnDraw()


        def OnMouseDown(self, evt):
            self.CaptureMouse()


        def OnMouseUp(self, evt):
            self.ReleaseMouse()


        def OnMouseMotion(self, evt):
            if evt.Dragging() and evt.LeftIsDown():
                self.x, self.y = self.lastx, self.lasty
                self.x, self.y = evt.GetPosition()
                self.Refresh(False)


    class CubeCanvas(MyCanvasBase):
        def InitGL(self):
            # set viewing projection
            glMatrixMode(GL_PROJECTION);
            glFrustum(-0.5, 0.5, -0.5, 0.5, 1.0, 3.0);

            # position viewer
            glMatrixMode(GL_MODELVIEW);
            glTranslatef(0.0, 0.0, -2.0);

            # position object
            glRotatef(self.y, 1.0, 0.0, 0.0);
            glRotatef(self.x, 0.0, 1.0, 0.0);

            glEnable(GL_DEPTH_TEST);
            glEnable(GL_LIGHTING);
            glEnable(GL_LIGHT0);


        def OnDraw(self):
            # clear color and depth buffers
            glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);

            # draw six faces of a cube
            glBegin(GL_QUADS)
            glNormal3f( 0.0, 0.0, 1.0)
            glVertex3f( 0.5, 0.5, 0.5)
            glVertex3f(-0.5, 0.5, 0.5)
            glVertex3f(-0.5,-0.5, 0.5)
            glVertex3f( 0.5,-0.5, 0.5)

            glNormal3f( 0.0, 0.0,-1.0)
            glVertex3f(-0.5,-0.5,-0.5)
            glVertex3f(-0.5, 0.5,-0.5)
            glVertex3f( 0.5, 0.5,-0.5)
            glVertex3f( 0.5,-0.5,-0.5)

            glNormal3f( 0.0, 1.0, 0.0)
            glVertex3f( 0.5, 0.5, 0.5)
            glVertex3f( 0.5, 0.5,-0.5)
            glVertex3f(-0.5, 0.5,-0.5)
            glVertex3f(-0.5, 0.5, 0.5)

            glNormal3f( 0.0,-1.0, 0.0)
            glVertex3f(-0.5,-0.5,-0.5)
            glVertex3f( 0.5,-0.5,-0.5)
            glVertex3f( 0.5,-0.5, 0.5)
            glVertex3f(-0.5,-0.5, 0.5)

            glNormal3f( 1.0, 0.0, 0.0)
            glVertex3f( 0.5, 0.5, 0.5)
            glVertex3f( 0.5,-0.5, 0.5)
            glVertex3f( 0.5,-0.5,-0.5)
            glVertex3f( 0.5, 0.5,-0.5)

            glNormal3f(-1.0, 0.0, 0.0)
            glVertex3f(-0.5,-0.5,-0.5)
            glVertex3f(-0.5,-0.5, 0.5)
            glVertex3f(-0.5, 0.5, 0.5)
            glVertex3f(-0.5, 0.5,-0.5)
            glEnd()

            glRotatef((self.lasty - self.y)/100., 1.0, 0.0, 0.0);
            glRotatef((self.lastx - self.x)/100., 0.0, 1.0, 0.0);

            self.SwapBuffers()


    class ConeCanvas(MyCanvasBase):
        def InitGL( self ):
            glMatrixMode(GL_PROJECTION);
            # camera frustrum setup
            glFrustum(-0.5, 0.5, -0.5, 0.5, 1.0, 3.0);
            glMaterial(GL_FRONT, GL_AMBIENT, [0.2, 0.2, 0.2, 1.0])
            glMaterial(GL_FRONT, GL_DIFFUSE, [0.8, 0.8, 0.8, 1.0])
            glMaterial(GL_FRONT, GL_SPECULAR, [1.0, 0.0, 1.0, 1.0])
            glMaterial(GL_FRONT, GL_SHININESS, 50.0)
            glLight(GL_LIGHT0, GL_AMBIENT, [0.0, 1.0, 0.0, 1.0])
            glLight(GL_LIGHT0, GL_DIFFUSE, [1.0, 1.0, 1.0, 1.0])
            glLight(GL_LIGHT0, GL_SPECULAR, [1.0, 1.0, 1.0, 1.0])
            glLight(GL_LIGHT0, GL_POSITION, [1.0, 1.0, 1.0, 0.0]);
            glLightModel(GL_LIGHT_MODEL_AMBIENT, [0.2, 0.2, 0.2, 1.0])
            glEnable(GL_LIGHTING)
            glEnable(GL_LIGHT0)
            glDepthFunc(GL_LESS)
            glEnable(GL_DEPTH_TEST)
            glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)
            # position viewer
            glMatrixMode(GL_MODELVIEW);


        def OnDraw(self):
            # clear color and depth buffers
            glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
            # use a fresh transformation matrix
            glPushMatrix()
            # position object
            glTranslate(0.0, 0.0, -2.0);
            glRotate(30.0, 1.0, 0.0, 0.0);
            glRotate(30.0, 0.0, 1.0, 0.0);

            glTranslate(0, -1, 0)
            glRotate(250, 1, 0, 0)
            glutSolidCone(0.5, 1, 30, 5)
            glPopMatrix()
            glRotatef((self.lasty - self.y)/100., 0.0, 0.0, 1.0);
            glRotatef(0.0, (self.lastx - self.x)/100., 1.0, 0.0);
            # push into visible buffer
            self.SwapBuffers()


#----------------------------------------------------------------------


overview = """\
"""


if __name__ == '__main__':
    import sys,os
    import run
    run.main(['', os.path.basename(sys.argv[0])])


#----------------------------------------------------------------------
Commit	Line	Data
8fa876ca	1
41378d34	2	import wx
8fa876ca	3
cf694132	4	try:
41378d34	5	from wx import glcanvas
1e4a197e	6	haveGLCanvas = True
cf694132	7	except ImportError:
1e4a197e	8	haveGLCanvas = False
cf694132	9
54b96882 RD	10	try:
54b96882 RD	11	# The Python OpenGL package can be found at
c368d904	12	# http://PyOpenGL.sourceforge.net/
41378d34 RD	13	from OpenGL.GL import *
41378d34 RD	14	from OpenGL.GLUT import *
1e4a197e	15	haveOpenGL = True
54b96882	16	except ImportError:
1e4a197e	17	haveOpenGL = False
54b96882	18
cf694132 RD	19	#----------------------------------------------------------------------
	20
	21	if not haveGLCanvas:
	22	def runTest(frame, nb, log):
41378d34 RD	23	dlg = wx.MessageDialog(frame, 'The GLCanvas class has not been included with this build of wxPython!',
41378d34 RD	24	'Sorry', wx.OK \| wx.ICON_INFORMATION)
cf694132 RD	25	dlg.ShowModal()
	26	dlg.Destroy()
	27
54b96882 RD	28	elif not haveOpenGL:
54b96882 RD	29	def runTest(frame, nb, log):
41378d34 RD	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)
54b96882 RD	34	dlg.ShowModal()
54b96882 RD	35	dlg.Destroy()
cf694132 RD	36
cf694132 RD	37
41378d34 RD	38
41378d34 RD	39
54b96882	40	else:
5b119149	41	buttonDefs = {
8fa876ca RD	42	wx.NewId() : ('CubeCanvas', 'Cube'),
8fa876ca RD	43	wx.NewId() : ('ConeCanvas', 'Cone'),
5b119149 RD	44	}
5b119149 RD	45
8fa876ca	46	class ButtonPanel(wx.Panel):
5b119149	47	def __init__(self, parent, log):
8fa876ca	48	wx.Panel.__init__(self, parent, -1)
5b119149 RD	49	self.log = log
5b119149 RD	50
8fa876ca RD	51	box = wx.BoxSizer(wx.VERTICAL)
8fa876ca RD	52	box.Add((20, 30))
5b119149 RD	53	keys = buttonDefs.keys()
	54	keys.sort()
	55	for k in keys:
	56	text = buttonDefs[k][1]
8fa876ca RD	57	btn = wx.Button(self, k, text)
8fa876ca RD	58	box.Add(btn, 0, wx.ALIGN_CENTER\|wx.ALL, 15)
41378d34	59	self.Bind(wx.EVT_BUTTON, self.OnButton, btn)
5b119149	60
41378d34	61	#** Enable this to show putting a GLCanvas on the wx.Panel
4ece0694 RD	62	if 0:
	63	c = CubeCanvas(self)
	64	c.SetSize((200, 200))
8fa876ca	65	box.Add(c, 0, wx.ALIGN_CENTER\|wx.ALL, 15)
4ece0694	66
1e4a197e	67	self.SetAutoLayout(True)
5b119149 RD	68	self.SetSizer(box)
5b119149 RD	69
4ece0694	70
5b119149 RD	71	def OnButton(self, evt):
	72	canvasClassName = buttonDefs[evt.GetId()][0]
	73	canvasClass = eval(canvasClassName)
8fa876ca	74	frame = wx.Frame(None, -1, canvasClassName, size=(400,400))
5b119149	75	canvas = canvasClass(frame)
1e4a197e	76	frame.Show(True)
cf694132 RD	77
cf694132 RD	78
d2103cf2	79
5b119149 RD	80	def runTest(frame, nb, log):
	81	win = ButtonPanel(nb, log)
	82	return win
d2103cf2	83
cf694132	84
d2103cf2	85
d58a80fa	86
8fa876ca	87	class MyCanvasBase(glcanvas.GLCanvas):
cf694132	88	def __init__(self, parent):
8fa876ca	89	glcanvas.GLCanvas.__init__(self, parent, -1)
1e4a197e	90	self.init = False
5b119149 RD	91	# initial mouse position
	92	self.lastx = self.x = 30
	93	self.lasty = self.y = 30
8fa876ca RD	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)
41378d34	97	self.Bind(wx.EVT_LEFT_DOWN, self.OnMouseDown)
8fa876ca RD	98	self.Bind(wx.EVT_LEFT_UP, self.OnMouseUp)
8fa876ca RD	99	self.Bind(wx.EVT_MOTION, self.OnMouseMotion)
cf694132	100
01f44840	101
cf694132	102	def OnEraseBackground(self, event):
5b119149	103	pass # Do nothing, to avoid flashing on MSW.
cf694132	104
01f44840	105
cf694132 RD	106	def OnSize(self, event):
cf694132 RD	107	size = self.GetClientSize()
d2103cf2	108	if self.GetContext():
cf694132	109	self.SetCurrent()
41378d34	110	glViewport(0, 0, size.width, size.height)
01f44840 RD	111	event.Skip()
01f44840 RD	112
cf694132	113
cf694132	114	def OnPaint(self, event):
8fa876ca	115	dc = wx.PaintDC(self)
cf694132	116	self.SetCurrent()
cf694132 RD	117	if not self.init:
cf694132 RD	118	self.InitGL()
1e4a197e	119	self.init = True
5b119149 RD	120	self.OnDraw()
5b119149 RD	121
01f44840	122
5b119149 RD	123	def OnMouseDown(self, evt):
	124	self.CaptureMouse()
	125
01f44840	126
5b119149 RD	127	def OnMouseUp(self, evt):
	128	self.ReleaseMouse()
	129
01f44840	130
5b119149 RD	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()
1e4a197e	135	self.Refresh(False)
cf694132	136
5b119149	137
41378d34 RD	138
41378d34 RD	139
5b119149 RD	140	class CubeCanvas(MyCanvasBase):
	141	def InitGL(self):
	142	# set viewing projection
41378d34 RD	143	glMatrixMode(GL_PROJECTION);
41378d34 RD	144	glFrustum(-0.5, 0.5, -0.5, 0.5, 1.0, 3.0);
5b119149 RD	145
5b119149 RD	146	# position viewer
41378d34 RD	147	glMatrixMode(GL_MODELVIEW);
41378d34 RD	148	glTranslatef(0.0, 0.0, -2.0);
5b119149 RD	149
5b119149 RD	150	# position object
41378d34 RD	151	glRotatef(self.y, 1.0, 0.0, 0.0);
41378d34 RD	152	glRotatef(self.x, 0.0, 1.0, 0.0);
5b119149	153
41378d34 RD	154	glEnable(GL_DEPTH_TEST);
	155	glEnable(GL_LIGHTING);
	156	glEnable(GL_LIGHT0);
5b119149 RD	157
	158
	159	def OnDraw(self):
cf694132	160	# clear color and depth buffers
41378d34	161	glClear(GL_COLOR_BUFFER_BIT \| GL_DEPTH_BUFFER_BIT);
cf694132 RD	162
cf694132 RD	163	# draw six faces of a cube
41378d34 RD	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);
cf694132	204
5b119149	205	self.SwapBuffers()
cf694132	206
cf694132	207
41378d34 RD	208
	209
	210
5b119149 RD	211	class ConeCanvas(MyCanvasBase):
5b119149 RD	212	def InitGL( self ):
41378d34	213	glMatrixMode(GL_PROJECTION);
d2103cf2	214	# camera frustrum setup
41378d34 RD	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)
d58a80fa	230	# position viewer
41378d34	231	glMatrixMode(GL_MODELVIEW);
d2103cf2 RD	232
d2103cf2 RD	233
5b119149	234	def OnDraw(self):
d2103cf2	235	# clear color and depth buffers
41378d34	236	glClear(GL_COLOR_BUFFER_BIT \| GL_DEPTH_BUFFER_BIT);
d2103cf2	237	# use a fresh transformation matrix
41378d34	238	glPushMatrix()
d2103cf2	239	# position object
41378d34 RD	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);
d2103cf2 RD	250	# push into visible buffer
d2103cf2 RD	251	self.SwapBuffers()
cf694132 RD	252
cf694132 RD	253
5b119149 RD	254
5b119149 RD	255
e166644c	256	#----------------------------------------------------------------------
cf694132 RD	257
	258
	259
41378d34	260
cf694132 RD	261	overview = """\
	262	"""
	263
	264
cf694132	265
41378d34 RD	266	if __name__ == '__main__':
	267	import sys,os
	268	import run
	269	run.main(['', os.path.basename(sys.argv[0])])
cf694132	270
cf694132	271
41378d34 RD	272
	273
	274	#----------------------------------------------------------------------
d58a80fa	275