wxPython/demo/DrawXXXList.py

   1
   2 from wxPython.wx import *
   3 import whrandom, time
   4
   5 #----------------------------------------------------------------------
   6
   7 colours = [
   8     "BLACK",
   9     "BLUE",
  10     "BLUE VIOLET",
  11     "BROWN",
  12     "CYAN",
  13     "DARK GREY",
  14     "DARK GREEN",
  15     "GOLD",
  16     "GREY",
  17     "GREEN",
  18     "MAGENTA",
  19     "NAVY",
  20     "PINK",
  21     "RED",
  22     "SKY BLUE",
  23     "VIOLET",
  24     "YELLOW",
  25     ]
  26
  27 #----------------------------------------------------------------------
  28
  29 def makeRandomPoints(num, w, h):
  30     pnts = []
  31     for i in range(num):
  32         x = whrandom.randint(0, w)
  33         y = whrandom.randint(0, h)
  34         pnts.append( (x,y) )
  35     return pnts
  36
  37
  38 def makeRandomLines(num, w, h):
  39     pnts = []
  40     for i in range(num):
  41         x1 = whrandom.randint(0, w)
  42         y1 = whrandom.randint(0, h)
  43         x2 = whrandom.randint(0, w)
  44         y2 = whrandom.randint(0, h)
  45         pnts.append( (x1,y1, x2,y2) )
  46     return pnts
  47
  48
  49 def makeRandomPens(num, cache):
  50     pens = []
  51     for i in range(num):
  52         c = whrandom.choice(colours)
  53         t = whrandom.randint(1, 4)
  54         if not cache.has_key( (c, t) ):
  55             cache[(c, t)] = wxPen(c, t)
  56         pens.append( cache[(c, t)] )
  57     return pens
  58
  59
  60 class TestPanel(wxPanel):
  61     def __init__(self, parent, size, log):
  62         wxPanel.__init__(self, parent, -1, size=size)
  63         self.log = log
  64         self.SetBackgroundColour(wxWHITE)
  65
  66         w = size.width
  67         h = size.height
  68         pencache = {}
  69
  70         # make some lists of random points
  71         self.pnts1 = makeRandomPoints(1000, w, h)
  72         self.pnts2 = makeRandomPoints(1000, w, h)
  73         self.pnts3 = makeRandomPoints(1000, w, h)
  74         self.pens1  = makeRandomPens(1000, pencache)
  75
  76         # and now some lines
  77         self.lines1 = makeRandomLines(500, w, h)
  78         self.lines2 = makeRandomLines(500, w, h)
  79         self.lines3 = makeRandomLines(500, w, h)
  80         self.pens2  = makeRandomPens(500, pencache)
  81
  82         EVT_PAINT(self, self.OnPaint)
  83
  84
  85     def OnPaint(self, evt):
  86         dc = wxPaintDC(self)
  87         dc.BeginDrawing()
  88         start = time.time()
  89
  90         dc.SetPen(wxPen("BLACK", 1))
  91         dc.DrawPointList(self.pnts1)
  92         dc.DrawPointList(self.pnts2, wxPen("RED", 2))
  93         dc.DrawPointList(self.pnts3, self.pens1)
  94
  95         dc.SetPen(wxPen("BLACK", 1))
  96         dc.DrawLineList(self.lines1)
  97         dc.DrawLineList(self.lines2, wxPen("RED", 2))
  98         dc.DrawLineList(self.lines3, self.pens2)
  99
 100         dc.EndDrawing()
 101         self.log.write("DrawTime: %s seconds\n" % (time.time() - start))
 102         self.log.write("GetBoundingBox: %s\n" % (dc.GetBoundingBox(), ))
 103
 104 #----------------------------------------------------------------------
 105
 106 def runTest(frame, nb, log):
 107     w = nb.GetClientSize().width
 108     h = nb.GetClientSize().height
 109     if w < 300: w = 300
 110     if h < 300: h = 300
 111     win = TestPanel(nb, wxSize(w, h), log)
 112     return win
 113
 114 #----------------------------------------------------------------------
 115
 116
 117 overview = """\
 118
 119 Some methods have been added to wxDC to help with optimization of
 120 drawing routines.  Currently they are:
 121
 122 <pre>
 123     DrawPointList(sequence, pens=None)
 124 </pre>
 125     Where sequence is a tuple, list, whatever of 2 element tuples
 126     (x, y) and pens is either None, a single pen or a list of pens.
 127
 128 <pre>
 129     DrawLineList(sequence, pens=None)
 130 </pre>
 131      Where sequence is a tuple, list, whatever of 4 element tuples
 132      (x1,y1, x2,y2) andd pens is either None, a single pen or a list
 133      of pens.
 134
 135 """