import random import time import wx #---------------------------------------------------------------------- colours = [ "BLACK", "BLUE", "BLUE VIOLET", "BROWN", "CYAN", "DARK GREY", "DARK GREEN", "GOLD", "GREY", "GREEN", "MAGENTA", "NAVY", "PINK", "RED", "SKY BLUE", "VIOLET", "YELLOW", ] #---------------------------------------------------------------------- def makeRandomPoints(num, w, h): pnts = [] for i in range(num): x = random.randint(0, w) y = random.randint(0, h) pnts.append( (x,y) ) return pnts def makeRandomLines(num, w, h): pnts = [] for i in range(num): x1 = random.randint(0, w) y1 = random.randint(0, h) x2 = random.randint(0, w) y2 = random.randint(0, h) pnts.append( (x1,y1, x2,y2) ) return pnts def makeRandomRectangles(num, W, H): rects = [] for i in range(num): w = random.randint(10, W/2) h = random.randint(10, H/2) x = random.randint(0, W - w) y = random.randint(0, H - h) rects.append( (x, y, w, h) ) return rects def makeRandomText(num): Np = 8 # number of characters in text text = [] for i in range(num): word = [] for i in range(Np): c = chr( random.randint(48, 122) ) word.append( c ) text.append( "".join(word) ) return text def makeRandomPolygons(num, W, H): Np = 8 # number of points per polygon polys = [] for i in range(num): poly = [] for i in range(Np): x = random.randint(0, W) y = random.randint(0, H) poly.append( (x,y) ) polys.append( poly ) return polys def makeRandomPens(num, cache): pens = [] for i in range(num): c = random.choice(colours) t = random.randint(1, 4) if not cache.has_key( (c, t) ): cache[(c, t)] = wx.Pen(c, t) pens.append( cache[(c, t)] ) return pens def makeRandomBrushes(num, cache): brushes = [] for i in range(num): c = random.choice(colours) if not cache.has_key(c): cache[c] = wx.Brush(c) brushes.append( cache[c] ) return brushes def makeRandomColors(num): colors = [] for i in range(num): c = random.choice(colours) colors.append(wx.NamedColour(c)) return colors pencache = {} brushcache = {} points = None lines = None rectangles = None polygons = None text = None pens = None brushes = None colors1 = None colors2 = None def Init(w, h, n): global pencache global brushcache global points global lines global rectangles global polygons global text global pens global brushes global colors1 global colors2 # make some lists of random shapes points = makeRandomPoints(n, w, h) try: import Numeric Apoints = Numeric.array(points) except: pass lines = makeRandomLines(n, w, h) rectangles = makeRandomRectangles(n, w, h) polygons = makeRandomPolygons(n, w, h) text = makeRandomText(n) # make some random pens and brushes pens = makeRandomPens(n, pencache) brushes = makeRandomBrushes(n, brushcache) # make some random color lists colors1 = makeRandomColors(n) colors2 = makeRandomColors(n) def TestPoints(dc,log): dc.BeginDrawing() start = time.time() dc.SetPen(wx.Pen("BLACK", 4)) dc.DrawPointList(points) dc.DrawPointList(points, wx.Pen("RED", 2)) dc.DrawPointList(points, pens) dc.EndDrawing() log.write("DrawTime: %s seconds with DrawPointList\n" % (time.time() - start)) def TestArrayPoints(dc,log): try: import Numeric dc.BeginDrawing() start = time.time() dc.SetPen(wx.Pen("BLACK", 1)) for i in range(1): dc.DrawPointList(Apoints) #dc.DrawPointList(Apoints, wx.Pen("RED", 2)) #dc.DrawPointList(Apoints, pens) dc.EndDrawing() log.write("DrawTime: %s seconds with DrawPointList an Numpy Array\n" % (time.time() - start)) except ImportError: log.write("Couldn't import Numeric") pass def TestLines(dc,log): dc.BeginDrawing() start = time.time() dc.SetPen(wx.Pen("BLACK", 2)) dc.DrawLineList(lines) dc.DrawLineList(lines, wx.Pen("RED", 2)) dc.DrawLineList(lines, pens) dc.EndDrawing() log.write("DrawTime: %s seconds with DrawLineList\n" % (time.time() - start)) def TestRectangles(dc,log): dc.BeginDrawing() start = time.time() dc.SetPen( wx.Pen("BLACK",1) ) dc.SetBrush( wx.Brush("RED") ) dc.DrawRectangleList(rectangles) dc.DrawRectangleList(rectangles,pens) dc.DrawRectangleList(rectangles,pens[0],brushes) dc.DrawRectangleList(rectangles,pens,brushes[0]) dc.DrawRectangleList(rectangles,None,brushes) ## for i in range(10): ## #dc.DrawRectangleList(rectangles,pens,brushes) ## dc.DrawRectangleList(rectangles) dc.EndDrawing() log.write("DrawTime: %s seconds with DrawRectanglesList\n" % (time.time() - start)) def TestEllipses(dc,log): dc.BeginDrawing() start = time.time() dc.SetPen( wx.Pen("BLACK",1) ) dc.SetBrush( wx.Brush("RED") ) dc.DrawEllipseList(rectangles) dc.DrawEllipseList(rectangles,pens) dc.DrawEllipseList(rectangles,pens[0],brushes) dc.DrawEllipseList(rectangles,pens,brushes[0]) dc.DrawEllipseList(rectangles,None,brushes) dc.DrawEllipseList(rectangles,pens,brushes) dc.EndDrawing() log.write("DrawTime: %s seconds with DrawEllipsesList\n" % (time.time() - start)) def TestRectanglesArray(dc,log): try: import Numeric Apoints = Numeric.array(rectangles) dc.BeginDrawing() start = time.time() dc.SetPen(wx.Pen("BLACK", 1)) dc.DrawRectangleList(rectangles) dc.DrawRectangleList(rectangles,pens) dc.DrawRectangleList(rectangles,pens[0],brushes) dc.DrawRectangleList(rectangles,pens,brushes[0]) dc.DrawRectangleList(rectangles,None,brushes) ## for i in range(10): ## #dc.DrawRectangleList(rectangles,pens,brushes) ## dc.DrawRectangleList(rectangles) dc.EndDrawing() log.write("DrawTime: %s seconds with DrawRectangleList and Numpy Array\n" % (time.time() - start)) except ImportError: log.write("Couldn't import Numeric") pass def TestRectanglesLoop(dc,log): dc.BeginDrawing() start = time.time() dc.DrawRectangleList(rectangles,pens,brushes) log.write("DrawTime: %s seconds with DrawRectanglesList\n" % (time.time() - start)) start = time.time() for i in range(len(rectangles)): dc.SetPen( pens[i] ) dc.SetBrush( brushes[i] ) dc.DrawRectangle(rectangles[i][0],rectangles[i][1],rectangles[i][2],rectangles[i][3]) dc.EndDrawing() log.write("DrawTime: %s seconds with Python loop\n" % (time.time() - start)) def TestPolygons(dc,log): dc.BeginDrawing() start = time.time() dc.SetPen(wx.Pen("BLACK", 1)) dc.DrawPolygonList(polygons) dc.DrawPolygonList(polygons,pens) dc.DrawPolygonList(polygons,pens[0],brushes) dc.DrawPolygonList(polygons,pens,brushes[0]) dc.DrawPolygonList(polygons,None,brushes) log.write("DrawTime: %s seconds with DrawPolygonList\n" % (time.time() - start)) dc.EndDrawing() def TestText(dc,log): dc.BeginDrawing() start = time.time() # NOTE: you need to set BackgroundMode for the background colors to be used. dc.SetBackgroundMode(wx.SOLID) foreground = colors1 background = colors2 dc.DrawTextList(text, points, foreground, background) log.write("DrawTime: %s seconds with DrawTextList\n" % (time.time() - start)) dc.EndDrawing() class TestNB(wx.Notebook): def __init__(self, parent, id, log): style = wx.NB_BOTTOM if wx.Platform == "__WXMAC__": style = 0 wx.Notebook.__init__(self, parent, id, style=style) self.log = log # Initialize our various samples and add them to the notebook. win = DrawPanel(self, TestEllipses, log) self.AddPage(win, 'Ellipses') win = DrawPanel(self, TestText, log) self.AddPage(win, 'Text') win = DrawPanel(self, TestPolygons, log) self.AddPage(win, 'Polygons') win = DrawPanel(self, TestPoints, log) self.AddPage(win, 'Points') win = DrawPanel(self, TestLines, log) self.AddPage(win, 'Lines') win = DrawPanel(self, TestRectangles, log) self.AddPage(win, 'Rectangles') # Class used for all the various sample pages; the mechanics are the same # for each one with regards to the notebook. The only difference is # the function we use to draw on it. class DrawPanel(wx.Panel): def __init__(self, parent, drawFun, log): wx.Panel.__init__(self, parent, -1) self.SetBackgroundColour(wx.WHITE) self.log = log self.drawFun = drawFun self.Bind(wx.EVT_PAINT, self.OnPaint) def OnPaint(self, evt): dc = wx.PaintDC(self) dc.Clear() self.drawFun(dc,self.log) #---------------------------------------------------------------------- def runTest(frame, nb, log): w = nb.GetClientSize().width h = nb.GetClientSize().height if w < 600: w = 600 if h < 400: h = 400 Init(w, h, 200) win = TestNB(nb, -1, log) return win #---------------------------------------------------------------------- overview = """\ Some methods have been added to wx.DC to help with optimization of drawing routines. Currently they are:
```    DrawPointList(sequence, pens=None)
```
Where sequence is a tuple, list, whatever of 2 element tuples (x, y) and pens is either None, a single pen or a list of pens.
```    DrawLineList(sequence, pens=None)
```
Where sequence is a tuple, list, whatever of 4 element tuples (x1,y1, x2,y2) and pens is either None, a single pen or a list of pens.
```    DrawRectangleList(rectangles, pens=None, brushes=None)
```
```    DrawEllipseList(ellipses, pens=None, brushes=None)
```
```    DrawPolygonList(polygons, pens=None, brushes=None)
```
```    DrawTextList(textList, coords, foregrounds = None, backgrounds = None)
```
""" if __name__ == '__main__': import sys,os import run run.main(['', os.path.basename(sys.argv[0])] + sys.argv[1:])