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


from wxPython.wx       import *
try:
    from wxPython.glcanvas import *
    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 = wxMessageDialog(frame, 'The wxGLCanvas has not been included with this build of wxPython!',
                          'Sorry', wxOK | wxICON_INFORMATION)
        dlg.ShowModal()
        dlg.Destroy()

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


else:
    def runTest(frame, nb, log):
        win = wxFrame(frame, -1, "GL Demos", wxDefaultPosition, wxSize(300,300))
        CubeCanvas(win)
        #MySplitter(win)
        frame.otherWin = win
        win.Show(true)
        return None


    class MySplitter(wxSplitterWindow):
        def __init__(self, parent):
            wxSplitterWindow.__init__(self, parent, -1)
            cube = CubeCanvas(self)
            cone = ConeCanvas(self)

            self.SplitVertically(cube, cone)
            self.SetSashPosition(300)


    class CubeCanvas(wxGLCanvas):
        def __init__(self, parent):
            wxGLCanvas.__init__(self, parent, -1) #,
                                #attribList=[GL_RED_BITS, 4, GL_DOUBLEBUFFER] )
            EVT_ERASE_BACKGROUND(self, self.OnEraseBackground)
            EVT_SIZE(self, self.OnSize)
            EVT_PAINT(self, self.OnPaint)
            self.init = false

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

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


        def OnPaint(self, event):
            dc = wxPaintDC(self)

            self.SetCurrent()

            if not self.init:
                self.InitGL()
                self.init = true

            # 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()

            self.SwapBuffers()


        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(30.0, 1.0, 0.0, 0.0);
            glRotatef(30.0, 0.0, 1.0, 0.0);

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


    class ConeCanvas(wxGLCanvas):
        def __init__(self, parent):
            wxGLCanvas.__init__(self, parent, -1)
            EVT_ERASE_BACKGROUND(self, self.OnEraseBackground)
            EVT_SIZE(self, self.OnSize)
            EVT_PAINT(self, self.OnPaint)
            self.init = false

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

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

        def GLInit( 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)


        def OnPaint( self, event ):
            dc = wxPaintDC(self)
            if not self.init:
                self.GLInit()
                self.init = true

            ### Tell system to use _this_ glcanvas for all commands
            self.SetCurrent()
            # clear color and depth buffers
            glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
            # position viewer
            glMatrixMode(GL_MODELVIEW);
            # 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);

            ### From cone.py
            glTranslate(0, -1, 0)
            glRotate(250, 1, 0, 0)
            glutSolidCone(1, 2, 50, 10)
            glPopMatrix()
            # push into visible buffer
            self.SwapBuffers()


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


overview = """\
"""


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

def _test():
    class MyApp(wxApp):
        def OnInit(self):
            frame = wxFrame(None, -1, "GL Demos", wxDefaultPosition, wxSize(600,300))
            #win = ConeCanvas(frame)
            MySplitter(frame)
            frame.Show(TRUE)
            self.SetTopWindow(frame)
            return TRUE

    app = MyApp(0)
    app.MainLoop()

if __name__ == '__main__':
    _test()
Commit	Line	Data
cf694132 RD	1
	2	from wxPython.wx import *
	3	try:
	4	from wxPython.glcanvas import *
	5	haveGLCanvas = true
	6	except ImportError:
	7	haveGLCanvas = false
	8
54b96882 RD	9	try:
54b96882 RD	10	# The Python OpenGL package can be found at
c368d904	11	# http://PyOpenGL.sourceforge.net/
54b96882 RD	12	from OpenGL.GL import *
	13	from OpenGL.GLUT import *
	14	haveOpenGL = true
	15	except ImportError:
	16	haveOpenGL = false
	17
cf694132 RD	18	#----------------------------------------------------------------------
	19
	20	if not haveGLCanvas:
	21	def runTest(frame, nb, log):
	22	dlg = wxMessageDialog(frame, 'The wxGLCanvas has not been included with this build of wxPython!',
	23	'Sorry', wxOK \| wxICON_INFORMATION)
	24	dlg.ShowModal()
	25	dlg.Destroy()
	26
54b96882 RD	27	elif not haveOpenGL:
	28	def runTest(frame, nb, log):
	29	dlg = wxMessageDialog(frame,
	30	'The OpenGL package was not found. You can get it at\n'
c368d904	31	'http://PyOpenGL.sourceforge.net/',
54b96882 RD	32	'Sorry', wxOK \| wxICON_INFORMATION)
	33	dlg.ShowModal()
	34	dlg.Destroy()
cf694132 RD	35
cf694132 RD	36
54b96882	37	else:
cf694132	38	def runTest(frame, nb, log):
d2103cf2 RD	39	win = wxFrame(frame, -1, "GL Demos", wxDefaultPosition, wxSize(300,300))
	40	CubeCanvas(win)
	41	#MySplitter(win)
6bddd8c5 RD	42	frame.otherWin = win
	43	win.Show(true)
	44	return None
cf694132 RD	45
cf694132 RD	46
d2103cf2 RD	47	class MySplitter(wxSplitterWindow):
	48	def __init__(self, parent):
	49	wxSplitterWindow.__init__(self, parent, -1)
	50	cube = CubeCanvas(self)
	51	cone = ConeCanvas(self)
	52
	53	self.SplitVertically(cube, cone)
	54	self.SetSashPosition(300)
	55
cf694132	56
d2103cf2 RD	57
d2103cf2 RD	58	class CubeCanvas(wxGLCanvas):
cf694132	59	def __init__(self, parent):
c368d904 RD	60	wxGLCanvas.__init__(self, parent, -1) #,
c368d904 RD	61	#attribList=[GL_RED_BITS, 4, GL_DOUBLEBUFFER] )
cf694132	62	EVT_ERASE_BACKGROUND(self, self.OnEraseBackground)
f6bcfd97 BP	63	EVT_SIZE(self, self.OnSize)
f6bcfd97 BP	64	EVT_PAINT(self, self.OnPaint)
cf694132 RD	65	self.init = false
	66
	67	def OnEraseBackground(self, event):
	68	pass # Do nothing, to avoid flashing.
	69
cf694132 RD	70	def OnSize(self, event):
cf694132 RD	71	size = self.GetClientSize()
d2103cf2	72	if self.GetContext():
cf694132 RD	73	self.SetCurrent()
	74	glViewport(0, 0, size.width, size.height)
	75
	76
	77	def OnPaint(self, event):
	78	dc = wxPaintDC(self)
	79
cf694132 RD	80	self.SetCurrent()
cf694132 RD	81
cf694132 RD	82	if not self.init:
	83	self.InitGL()
	84	self.init = true
	85
	86	# clear color and depth buffers
	87	glClear(GL_COLOR_BUFFER_BIT \| GL_DEPTH_BUFFER_BIT);
	88
	89	# draw six faces of a cube
	90	glBegin(GL_QUADS)
	91	glNormal3f( 0.0, 0.0, 1.0)
	92	glVertex3f( 0.5, 0.5, 0.5)
	93	glVertex3f(-0.5, 0.5, 0.5)
	94	glVertex3f(-0.5,-0.5, 0.5)
	95	glVertex3f( 0.5,-0.5, 0.5)
	96
	97	glNormal3f( 0.0, 0.0,-1.0)
	98	glVertex3f(-0.5,-0.5,-0.5)
	99	glVertex3f(-0.5, 0.5,-0.5)
	100	glVertex3f( 0.5, 0.5,-0.5)
	101	glVertex3f( 0.5,-0.5,-0.5)
	102
	103	glNormal3f( 0.0, 1.0, 0.0)
	104	glVertex3f( 0.5, 0.5, 0.5)
	105	glVertex3f( 0.5, 0.5,-0.5)
	106	glVertex3f(-0.5, 0.5,-0.5)
	107	glVertex3f(-0.5, 0.5, 0.5)
	108
	109	glNormal3f( 0.0,-1.0, 0.0)
	110	glVertex3f(-0.5,-0.5,-0.5)
	111	glVertex3f( 0.5,-0.5,-0.5)
	112	glVertex3f( 0.5,-0.5, 0.5)
	113	glVertex3f(-0.5,-0.5, 0.5)
	114
	115	glNormal3f( 1.0, 0.0, 0.0)
	116	glVertex3f( 0.5, 0.5, 0.5)
	117	glVertex3f( 0.5,-0.5, 0.5)
	118	glVertex3f( 0.5,-0.5,-0.5)
	119	glVertex3f( 0.5, 0.5,-0.5)
	120
	121	glNormal3f(-1.0, 0.0, 0.0)
	122	glVertex3f(-0.5,-0.5,-0.5)
	123	glVertex3f(-0.5,-0.5, 0.5)
	124	glVertex3f(-0.5, 0.5, 0.5)
	125	glVertex3f(-0.5, 0.5,-0.5)
	126	glEnd()
	127
	128	self.SwapBuffers()
	129
	130
	131	def InitGL(self):
	132	# set viewing projection
	133	glMatrixMode(GL_PROJECTION);
	134	glFrustum(-0.5, 0.5, -0.5, 0.5, 1.0, 3.0);
	135
	136	# position viewer
	137	glMatrixMode(GL_MODELVIEW);
	138	glTranslatef(0.0, 0.0, -2.0);
	139
	140	# position object
	141	glRotatef(30.0, 1.0, 0.0, 0.0);
	142	glRotatef(30.0, 0.0, 1.0, 0.0);
	143
	144	glEnable(GL_DEPTH_TEST);
	145	glEnable(GL_LIGHTING);
146	glEnable(GL_LIGHT0);
147
148
149
d2103cf2 RD	150	class ConeCanvas(wxGLCanvas):
	151	def __init__(self, parent):
	152	wxGLCanvas.__init__(self, parent, -1)
	153	EVT_ERASE_BACKGROUND(self, self.OnEraseBackground)
f6bcfd97 BP	154	EVT_SIZE(self, self.OnSize)
f6bcfd97 BP	155	EVT_PAINT(self, self.OnPaint)
d2103cf2 RD	156	self.init = false
	157
	158	def OnEraseBackground(self, event):
	159	pass # Do nothing, to avoid flashing.
	160
	161	def OnSize(self, event):
	162	size = self.GetClientSize()
	163	if self.GetContext():
	164	self.SetCurrent()
	165	glViewport(0, 0, size.width, size.height)
	166
	167	def GLInit( self ):
	168	glMatrixMode(GL_PROJECTION);
	169	# camera frustrum setup
	170	glFrustum(-0.5, 0.5, -0.5, 0.5, 1.0, 3.0);
	171	glMaterial(GL_FRONT, GL_AMBIENT, [0.2, 0.2, 0.2, 1.0])
	172	glMaterial(GL_FRONT, GL_DIFFUSE, [0.8, 0.8, 0.8, 1.0])
	173	glMaterial(GL_FRONT, GL_SPECULAR, [1.0, 0.0, 1.0, 1.0])
	174	glMaterial(GL_FRONT, GL_SHININESS, 50.0)
	175	glLight(GL_LIGHT0, GL_AMBIENT, [0.0, 1.0, 0.0, 1.0])
	176	glLight(GL_LIGHT0, GL_DIFFUSE, [1.0, 1.0, 1.0, 1.0])
	177	glLight(GL_LIGHT0, GL_SPECULAR, [1.0, 1.0, 1.0, 1.0])
	178	glLight(GL_LIGHT0, GL_POSITION, [1.0, 1.0, 1.0, 0.0]);
	179	glLightModel(GL_LIGHT_MODEL_AMBIENT, [0.2, 0.2, 0.2, 1.0])
	180	glEnable(GL_LIGHTING)
	181	glEnable(GL_LIGHT0)
	182	glDepthFunc(GL_LESS)
	183	glEnable(GL_DEPTH_TEST)
	184	glClear(GL_COLOR_BUFFER_BIT \| GL_DEPTH_BUFFER_BIT)
	185
	186
	187	def OnPaint( self, event ):
	188	dc = wxPaintDC(self)
	189	if not self.init:
	190	self.GLInit()
	191	self.init = true
	192
	193	### Tell system to use _this_ glcanvas for all commands
	194	self.SetCurrent()
	195	# clear color and depth buffers
	196	glClear(GL_COLOR_BUFFER_BIT \| GL_DEPTH_BUFFER_BIT);
	197	# position viewer
	198	glMatrixMode(GL_MODELVIEW);
	199	# use a fresh transformation matrix
	200	glPushMatrix()
	201	# position object
	202	glTranslate(0.0, 0.0, -2.0);
	203	glRotate(30.0, 1.0, 0.0, 0.0);
	204	glRotate(30.0, 0.0, 1.0, 0.0);
	205
	206	### From cone.py
	207	glTranslate(0, -1, 0)
	208	glRotate(250, 1, 0, 0)
	209	glutSolidCone(1, 2, 50, 10)
	210	glPopMatrix()
	211	# push into visible buffer
	212	self.SwapBuffers()
cf694132 RD	213
cf694132 RD	214
e166644c	215	#----------------------------------------------------------------------
cf694132 RD	216
	217
	218
	219
	220
	221
	222
	223
	224	overview = """\
	225	"""
	226
	227
	228
	229
	230
	231	#----------------------------------------------------------------------
	232
	233	def _test():
	234	class MyApp(wxApp):
	235	def OnInit(self):
493f1553	236	frame = wxFrame(None, -1, "GL Demos", wxDefaultPosition, wxSize(600,300))
d2103cf2 RD	237	#win = ConeCanvas(frame)
d2103cf2 RD	238	MySplitter(frame)
cf694132 RD	239	frame.Show(TRUE)
	240	self.SetTopWindow(frame)
	241	return TRUE
	242
	243	app = MyApp(0)
	244	app.MainLoop()
	245
	246	if __name__ == '__main__':
	247	_test()