wxPython/demo/FloatCanvas.py

   1
   2 try:
   3     import Numeric
   4     import RandomArray
   5     haveNumeric = True
   6 except ImportError:
   7     try:
   8         import numarray as Numeric
   9         import numarray.random_array as RandomArray
  10         haveNumeric = True
  11     except ImportError:
  12         haveNumeric = False
  13
  14 if not haveNumeric:
  15     errorText = """
  16 The FloatCanvas requires either the Numeric or Numarray module:
  17 You can get them at:
  18      http://sourceforge.net/projects/numpy
  19
  20 NOTE: The Numeric module is substantially faster than numarray for this
  21 purpose, if you have lot's of objects
  22 """
  23
  24     def runTest(frame, nb, log):
  25         dlg = wx.MessageDialog(frame, errorText, 'Sorry', wx.OK |
  26                                wx.ICON_INFORMATION)
  27         dlg.ShowModal()
  28         dlg.Destroy()
  29
  30     overview = ""
  31
  32 else:
  33     StartUpDemo = "all"
  34     if __name__ == "__main__": # parse options if run stand-alone
  35         # check options:
  36         import sys, getopt
  37         optlist, args = getopt.getopt(sys.argv[1:],'l',["local","all","text","map","stext","hit","hitf","animate","speed","temp"])
  38
  39         for opt in optlist:
  40             if opt[0] == "--all":
  41                 StartUpDemo = "all"
  42             elif opt[0] == "--text":
  43                 StartUpDemo = "text"
  44             elif opt[0] == "--map":
  45                 StartUpDemo = "map"
  46             elif opt[0] == "--stext":
  47                 StartUpDemo = "stext"
  48             elif opt[0] == "--hit":
  49                 StartUpDemo = "hit"
  50             elif opt[0] == "--hitf":
  51                 StartUpDemo = "hitf"
  52             elif opt[0] == "--animate":
  53                 StartUpDemo = "animate"
  54             elif opt[0] == "--speed":
  55                 StartUpDemo = "speed"
  56             elif opt[0] == "--temp":
  57                 StartUpDemo = "temp"
  58     import wx
  59     import time, random
  60
  61     def runTest(frame, nb, log):
  62         """
  63         This method is used by the wxPython Demo Framework for integrating
  64         this demo with the rest.
  65         """
  66         win = DrawFrame(None, -1, "FloatCanvas Drawing Window",wx.DefaultPosition,(500,500))
  67         frame.otherWin = win
  68         win.Show(True)
  69         win.DrawTest()
  70
  71     try:
  72         from floatcanvas import NavCanvas, FloatCanvas
  73     except ImportError: # if it's not there locally, try the wxPython lib.
  74         from wx.lib.floatcanvas import NavCanvas, FloatCanvas
  75
  76     import wxPython.lib.colourdb
  77
  78     class DrawFrame(wx.Frame):
  79
  80         """
  81         A frame used for the FloatCanvas Demo
  82
  83         """
  84
  85
  86         def __init__(self,parent, id,title,position,size):
  87             wx.Frame.__init__(self,parent, id,title,position, size)
  88
  89             ## Set up the MenuBar
  90             MenuBar = wx.MenuBar()
  91
  92             file_menu = wx.Menu()
  93             item = file_menu.Append(-1, "&Close","Close this frame")
  94             self.Bind(wx.EVT_MENU, self.OnQuit, item)
  95             MenuBar.Append(file_menu, "&File")
  96
  97             draw_menu = wx.Menu()
  98
  99             item = draw_menu.Append(-1, "&Draw Test","Run a test of drawing random components")
 100             self.Bind(wx.EVT_MENU, self.DrawTest, item)
 101
 102             item = draw_menu.Append(-1, "&Line Test","Run a test of drawing random lines")
 103             self.Bind(wx.EVT_MENU, self.LineTest, item)
 104
 105             item = draw_menu.Append(-1, "Draw &Map","Run a test of drawing a map")
 106             self.Bind(wx.EVT_MENU, self.DrawMap, item)
 107             item = draw_menu.Append(-1, "&Text Test","Run a test of text drawing")
 108             self.Bind(wx.EVT_MENU, self.TestText, item)
 109             item = draw_menu.Append(-1, "&ScaledText Test","Run a test of text drawing")
 110             self.Bind(wx.EVT_MENU, self.TestScaledText, item)
 111             item = draw_menu.Append(-1, "&Clear","Clear the Canvas")
 112             self.Bind(wx.EVT_MENU, self.Clear, item)
 113             item = draw_menu.Append(-1, "&Hit Test","Run a test of the hit test code")
 114             self.Bind(wx.EVT_MENU, self.TestHitTest, item)
 115             item = draw_menu.Append(-1, "&Hit Test Foreground","Run a test of the hit test code with a foreground Object")
 116             self.Bind(wx.EVT_MENU, self.TestHitTestForeground, item)
 117             item = draw_menu.Append(-1, "&Animation","Run a test of Animation")
 118             self.Bind(wx.EVT_MENU, self.TestAnimation, item)
 119             item = draw_menu.Append(-1, "&Speed","Run a test of Drawing Speed")
 120             self.Bind(wx.EVT_MENU, self.SpeedTest, item)
 121             MenuBar.Append(draw_menu, "&Tests")
 122
 123             view_menu = wx.Menu()
 124             item = view_menu.Append(-1, "Zoom to &Fit","Zoom to fit the window")
 125             self.Bind(wx.EVT_MENU, self.ZoomToFit, item)
 126             MenuBar.Append(view_menu, "&View")
 127
 128             help_menu = wx.Menu()
 129             item = help_menu.Append(-1, "&About",
 130                                     "More information About this program")
 131             self.Bind(wx.EVT_MENU, self.OnAbout, item)
 132             MenuBar.Append(help_menu, "&Help")
 133
 134             self.SetMenuBar(MenuBar)
 135
 136             self.CreateStatusBar()
 137             # Add the Canvas
 138             self.Canvas = NavCanvas.NavCanvas(self,
 139                                               -1,
 140                                               (500,500),
 141                                               Debug = 1,
 142                                               BackgroundColor = "DARK SLATE BLUE")
 143
 144             wx.EVT_CLOSE(self, self.OnCloseWindow)
 145
 146             FloatCanvas.EVT_MOTION(self.Canvas, self.OnMove )
 147             #FloatCanvas.EVT_LEFT_UP(self.Canvas, self.OnLeftUp )
 148
 149             self.EventsAreBound = False
 150
 151             ## getting all the colors and linestyles  for random objects
 152             wxPython.lib.colourdb.updateColourDB()
 153             self.colors = wxPython.lib.colourdb.getColourList()
 154             #self.LineStyles = FloatCanvas.DrawObject.LineStyleList.keys()
 155
 156
 157             return None
 158
 159         def BindAllMouseEvents(self):
 160             if not self.EventsAreBound:
 161                 ## Here is how you catch FloatCanvas mouse events
 162                 FloatCanvas.EVT_LEFT_DOWN(self.Canvas, self.OnLeftDown )
 163                 FloatCanvas.EVT_LEFT_UP(self.Canvas, self.OnLeftUp )
 164                 FloatCanvas.EVT_LEFT_DCLICK(self.Canvas, self.OnLeftDouble )
 165
 166                 FloatCanvas.EVT_MIDDLE_DOWN(self.Canvas, self.OnMiddleDown )
 167                 FloatCanvas.EVT_MIDDLE_UP(self.Canvas, self.OnMiddleUp )
 168                 FloatCanvas.EVT_MIDDLE_DCLICK(self.Canvas, self.OnMiddleDouble )
 169
 170                 FloatCanvas.EVT_RIGHT_DOWN(self.Canvas, self.OnRightDown )
 171                 FloatCanvas.EVT_RIGHT_UP(self.Canvas, self.OnRightUp )
 172                 FloatCanvas.EVT_RIGHT_DCLICK(self.Canvas, self.OnRightDouble )
 173
 174                 FloatCanvas.EVT_MOUSEWHEEL(self.Canvas, self.OnWheel )
 175             self.EventsAreBound = True
 176
 177         def UnBindAllMouseEvents(self):
 178             ## Here is how you catch FloatCanvas mouse events
 179             FloatCanvas.EVT_LEFT_DOWN(self.Canvas, None )
 180             FloatCanvas.EVT_LEFT_UP(self.Canvas, None )
 181             FloatCanvas.EVT_LEFT_DCLICK(self.Canvas, None)
 182
 183             FloatCanvas.EVT_MIDDLE_DOWN(self.Canvas, None )
 184             FloatCanvas.EVT_MIDDLE_UP(self.Canvas, None )
 185             FloatCanvas.EVT_MIDDLE_DCLICK(self.Canvas, None )
 186
 187             FloatCanvas.EVT_RIGHT_DOWN(self.Canvas, None )
 188             FloatCanvas.EVT_RIGHT_UP(self.Canvas, None )
 189             FloatCanvas.EVT_RIGHT_DCLICK(self.Canvas, None )
 190
 191             FloatCanvas.EVT_MOUSEWHEEL(self.Canvas, None )
 192
 193             self.EventsAreBound = False
 194
 195         def PrintCoords(self,event):
 196             print "coords are: %s"%(event.Coords,)
 197             print "pixel coords are: %s\n"%(event.GetPosition(),)
 198
 199         def OnLeftDown(self, event):
 200             print "Left Button has been clicked in DrawFrame"
 201             self.PrintCoords(event)
 202
 203         def OnLeftUp(self, event):
 204             print "Left up in DrawFrame"
 205             self.PrintCoords(event)
 206
 207         def OnLeftDouble(self, event):
 208             print "Left Double Click in DrawFrame"
 209             self.PrintCoords(event)
 210
 211         def OnMiddleDown(self, event):
 212             print "Middle Button clicked in DrawFrame"
 213             self.PrintCoords(event)
 214
 215         def OnMiddleUp(self, event):
 216             print "Middle Button Up in DrawFrame"
 217             self.PrintCoords(event)
 218
 219         def OnMiddleDouble(self, event):
 220             print "Middle Button Double clicked in DrawFrame"
 221             self.PrintCoords(event)
 222
 223         def OnRightDown(self, event):
 224             print "Right Button has been clicked in DrawFrame"
 225             self.PrintCoords(event)
 226
 227         def OnRightUp(self, event):
 228             print "Right Button Up in DrawFrame"
 229             self.PrintCoords(event)
 230
 231         def OnRightDouble(self, event):
 232             print "Right Button Double clicked in DrawFrame"
 233             self.PrintCoords(event)
 234
 235         def OnWheel(self, event):
 236             print "Mouse Wheel Moved in DrawFrame"
 237             self.PrintCoords(event)
 238
 239         def OnMove(self, event):
 240             """
 241             Updates the staus bar with the world coordinates
 242             """
 243             self.SetStatusText("%.2f, %.2f"%tuple(event.Coords))
 244
 245         def OnAbout(self, event):
 246             print "OnAbout called"
 247
 248             dlg = wx.MessageDialog(self, "This is a small program to demonstrate\n"
 249                                                       "the use of the FloatCanvas\n",
 250                                                       "About Me", wx.OK | wx.ICON_INFORMATION)
 251             dlg.ShowModal()
 252             dlg.Destroy()
 253
 254         def ZoomToFit(self,event):
 255             self.Canvas.ZoomToBB()
 256
 257         def Clear(self,event = None):
 258             self.UnBindAllMouseEvents()
 259             self.Canvas.ClearAll()
 260             self.Canvas.SetProjectionFun(None)
 261             self.Canvas.Draw()
 262
 263         def OnQuit(self,event):
 264             self.Close(True)
 265
 266         def OnCloseWindow(self, event):
 267             self.Destroy()
 268
 269         def DrawTest(self,event=None):
 270             wx.GetApp().Yield(True)
 271 #            import random
 272 #            import RandomArray
 273             Range = (-10,10)
 274             colors = self.colors
 275
 276             self.BindAllMouseEvents()
 277             Canvas = self.Canvas
 278
 279             Canvas.ClearAll()
 280             Canvas.SetProjectionFun(None)
 281
 282             ##          Random tests of everything:
 283
 284             # Rectangles
 285             for i in range(3):
 286                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 287                 lw = random.randint(1,5)
 288                 cf = random.randint(0,len(colors)-1)
 289                 h = random.randint(1,5)
 290                 w = random.randint(1,5)
 291                 Canvas.AddRectangle(x,y,h,w,LineWidth = lw,FillColor = colors[cf])
 292
 293             # Ellipses
 294             for i in range(3):
 295                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 296                 lw = random.randint(1,5)
 297                 cf = random.randint(0,len(colors)-1)
 298                 h = random.randint(1,5)
 299                 w = random.randint(1,5)
 300                 Canvas.AddEllipse(x,y,h,w,LineWidth = lw,FillColor = colors[cf])
 301
 302 ##            # Dots -- Does anyone need this?
 303 ##            for i in range(5):
 304 ##                x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 305 ##                D = random.randint(1,50)
 306 ##                lw = random.randint(1,5)
 307 ##                cf = random.randint(0,len(colors)-1)
 308 ##                cl = random.randint(0,len(colors)-1)
 309 ##                Canvas.AddDot(x,y,D,LineWidth = lw,LineColor = colors[cl],FillColor = colors[cf])
 310
 311                 # Circles
 312             for i in range(5):
 313                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 314                 D = random.randint(1,5)
 315                 lw = random.randint(1,5)
 316                 cf = random.randint(0,len(colors)-1)
 317                 cl = random.randint(0,len(colors)-1)
 318                 Canvas.AddCircle(x,y,D,LineWidth = lw,LineColor = colors[cl],FillColor = colors[cf])
 319                 Canvas.AddText("Circle # %i"%(i),x,y,Size = 12,BackgroundColor = None,Position = "cc")
 320
 321                 # Lines
 322             for i in range(5):
 323                 points = []
 324                 for j in range(random.randint(2,10)):
 325                     point = (random.randint(Range[0],Range[1]),random.randint(Range[0],Range[1]))
 326                     points.append(point)
 327                 lw = random.randint(1,10)
 328                 cf = random.randint(0,len(colors)-1)
 329                 cl = random.randint(0,len(colors)-1)
 330                 Canvas.AddLine(points, LineWidth = lw, LineColor = colors[cl])
 331
 332                 # Polygons
 333             for i in range(3):
 334                 points = []
 335                 for j in range(random.randint(2,6)):
 336                     point = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 337                     points.append(point)
 338                 lw = random.randint(1,6)
 339                 cf = random.randint(0,len(colors)-1)
 340                 cl = random.randint(0,len(colors)-1)
 341                 Canvas.AddPolygon(points,
 342                                        LineWidth = lw,
 343                                        LineColor = colors[cl],
 344                                        FillColor = colors[cf],
 345                                        FillStyle = 'Solid')
 346
 347             ## Pointset
 348             for i in range(4):
 349                 points = []
 350                 points = RandomArray.uniform(Range[0],Range[1],(100,2))
 351                 cf = random.randint(0,len(colors)-1)
 352                 D = random.randint(1,4)
 353                 Canvas.AddPointSet(points, Color = colors[cf], Diameter = D)
 354
 355             # Text
 356             String = "Unscaled text"
 357             for i in range(3):
 358                 ts = random.randint(10,40)
 359                 cf = random.randint(0,len(colors)-1)
 360                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 361                 Canvas.AddText(String, x, y, Size = ts, Color = colors[cf], Position = "cc")
 362
 363             # Scaled Text
 364             String = "Scaled text"
 365             for i in range(3):
 366                 ts = random.random()*3 + 0.2
 367                 cf = random.randint(0,len(colors)-1)
 368                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 369                 Canvas.AddScaledText(String, x, y, Size = ts, Color = colors[cf], Position = "cc")
 370
 371             Canvas.ZoomToBB()
 372
 373         def TestAnimation(self,event=None):
 374             """
 375
 376             In this test, a relatively complex background is drawn, and
 377             a simple object placed in the foreground is moved over
 378             it. This demonstrates how to use the InForeground attribute
 379             to make an object in the foregorund draw fast, without
 380             having to re-draw the whole background.
 381
 382             """
 383             print "Running TestAnimation"
 384             wx.GetApp().Yield(True)
 385             Range = (-10,10)
 386             self.Range = Range
 387
 388             self.UnBindAllMouseEvents()
 389             Canvas = self.Canvas
 390
 391             Canvas.ClearAll()
 392             Canvas.SetProjectionFun(None)
 393
 394             ##          Random tests of everything:
 395             colors = self.colors
 396             # Rectangles
 397             for i in range(3):
 398                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 399                 lw = random.randint(1,5)
 400                 cf = random.randint(0,len(colors)-1)
 401                 h = random.randint(1,5)
 402                 w = random.randint(1,5)
 403                 Canvas.AddRectangle(x,y,h,w,LineWidth = lw,FillColor = colors[cf])
 404
 405             # Ellipses
 406             for i in range(3):
 407                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 408                 lw = random.randint(1,5)
 409                 cf = random.randint(0,len(colors)-1)
 410                 h = random.randint(1,5)
 411                 w = random.randint(1,5)
 412                 Canvas.AddEllipse(x,y,h,w,LineWidth = lw,FillColor = colors[cf])
 413
 414             # Circles
 415             for i in range(5):
 416                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 417                 D = random.randint(1,5)
 418                 lw = random.randint(1,5)
 419                 cf = random.randint(0,len(colors)-1)
 420                 cl = random.randint(0,len(colors)-1)
 421                 Canvas.AddCircle(x,y,D,LineWidth = lw,LineColor = colors[cl],FillColor = colors[cf])
 422                 Canvas.AddText("Circle # %i"%(i),x,y,Size = 12,BackgroundColor = None,Position = "cc")
 423
 424             # Lines
 425             for i in range(5):
 426                 points = []
 427                 for j in range(random.randint(2,10)):
 428                     point = (random.randint(Range[0],Range[1]),random.randint(Range[0],Range[1]))
 429                     points.append(point)
 430                 lw = random.randint(1,10)
 431                 cf = random.randint(0,len(colors)-1)
 432                 cl = random.randint(0,len(colors)-1)
 433                 Canvas.AddLine(points, LineWidth = lw, LineColor = colors[cl])
 434
 435             # Polygons
 436             for i in range(3):
 437                 points = []
 438                 for j in range(random.randint(2,6)):
 439                     point = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 440                     points.append(point)
 441                 lw = random.randint(1,6)
 442                 cf = random.randint(0,len(colors)-1)
 443                 cl = random.randint(0,len(colors)-1)
 444                 Canvas.AddPolygon(points,
 445                                        LineWidth = lw,
 446                                        LineColor = colors[cl],
 447                                        FillColor = colors[cf],
 448                                        FillStyle = 'Solid')
 449
 450             # Scaled Text
 451             String = "Scaled text"
 452             for i in range(3):
 453                 ts = random.random()*3 + 0.2
 454                 cf = random.randint(0,len(colors)-1)
 455                 x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 456                 Canvas.AddScaledText(String, x, y, Size = ts, Color = colors[cf], Position = "cc")
 457
 458
 459             # Now the Foreground Object:
 460             C = Canvas.AddCircle(0,0,7,LineWidth = 2,LineColor = "Black",FillColor = "Red", InForeground = True)
 461             T = Canvas.AddScaledText("Click to Move",0,0, Size = 0.8, Position = 'cc', InForeground = True)
 462             C.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.MoveMe)
 463             C.Text = T
 464
 465             self.Timer = wx.PyTimer(self.ShowFrame)
 466             self.FrameDelay = 50 # milliseconds
 467
 468             Canvas.ZoomToBB()
 469
 470         def ShowFrame(self):
 471             Object = self.MovingObject
 472             Range = self.Range
 473             if  self.TimeStep < self.NumTimeSteps:
 474                 x,y = Object.XY
 475                 if x > Range[1] or x < Range[0]:
 476                     self.dx = -self.dx
 477                 if y > Range[1] or y < Range[0]:
 478                     self.dy = -self.dy
 479                 Object.Move( (self.dx,self.dy) )
 480                 Object.Text.Move( (self.dx,self.dy))
 481                 self.Canvas.Draw()
 482                 self.TimeStep += 1
 483                 wx.GetApp().Yield(True)
 484             else:
 485                 self.Timer.Stop()
 486
 487
 488         def MoveMe(self, Object):
 489             self.MovingObject = Object
 490             Range = self.Range
 491             self.dx = random.uniform(Range[0]/4,Range[1]/4)
 492             self.dy = random.uniform(Range[0]/4,Range[1]/4)
 493             #import time
 494             #start = time.time()
 495             self.NumTimeSteps = 500
 496             self.TimeStep = 1
 497             self.Timer.Start(self.FrameDelay)
 498             #print "Did %i frames in %f seconds"%(N, (time.time() - start) )
 499
 500         def TestHitTest(self,event=None):
 501             wx.GetApp().Yield(True)
 502
 503             self.UnBindAllMouseEvents()
 504             Canvas = self.Canvas
 505
 506             Canvas.ClearAll()
 507             Canvas.SetProjectionFun(None)
 508
 509             #Add a HitAble rectangle
 510             w, h = 60, 20
 511
 512             dx = 80
 513             dy = 40
 514             x,y = 20, 20
 515
 516             #Add one that is not HitAble
 517             Canvas.AddRectangle(x, y, w, h, LineWidth = 2)
 518             Canvas.AddText("Not Hit-able", x, y, Position = "bl")
 519
 520
 521             x += dx
 522             R = Canvas.AddRectangle(x, y, w, h,LineWidth = 2)
 523             R.Name = "Line Rectangle"
 524             R.HitFill = False
 525             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectGotHit)
 526             Canvas.AddText("Left Click Line", x, y, Position = "bl")
 527             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 528
 529
 530             x += dx
 531             color = "Red"
 532             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 533             R.Name = color + "Rectangle"
 534             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectGotHit)
 535             Canvas.AddText("Left Click Fill", x, y, Position = "bl")
 536             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 537
 538             x = 20
 539             y += dy
 540             color = "LightBlue"
 541             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 542             R.Name = color + " Rectangle"
 543             R.Bind(FloatCanvas.EVT_FC_RIGHT_DOWN, self.RectGotHit)
 544             Canvas.AddText("Right Click Fill", x, y, Position = "bl")
 545             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 546
 547             x += dx
 548             color = "Grey"
 549             R = Canvas.AddEllipse(x, y, w, h,LineWidth = 2,FillColor = color)
 550             R.Name = color +" Ellipse"
 551             R.Bind(FloatCanvas.EVT_FC_RIGHT_DOWN, self.RectGotHit)
 552             Canvas.AddText("Right Click Fill", x, y, Position = "bl")
 553             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 554
 555             x += dx
 556             color = "Brown"
 557             R = Canvas.AddCircle(x+dx/2, y+dy/2, dx/4, LineWidth = 2, FillColor = color)
 558             R.Name = color + " Circle"
 559             R.HitFill = True
 560             R.Bind(FloatCanvas.EVT_FC_LEFT_DCLICK, self.RectGotHit)
 561             Canvas.AddText("Left D-Click Fill", x, y, Position = "bl")
 562             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 563
 564             x = 20
 565             y += dy
 566             color = "Pink"
 567             R = Canvas.AddCircle(x+dx/2, y+dy/2, dx/4, LineWidth = 2,FillColor = color)
 568             R.Name = color +  " Circle"
 569             R.Bind(FloatCanvas.EVT_FC_LEFT_UP, self.RectGotHit)
 570             Canvas.AddText("Left Up Fill", x, y, Position = "bl")
 571             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 572
 573             x += dx
 574             color = "White"
 575             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 576             R.Name = color + " Rectangle"
 577             R.Bind(FloatCanvas.EVT_FC_MIDDLE_DOWN, self.RectGotHit)
 578             Canvas.AddText("Middle Down", x, y, Position = "bl")
 579             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 580
 581             x += dx
 582             color = "AQUAMARINE"
 583             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 584             R.Name = color + " Rectangle"
 585             R.Bind(FloatCanvas.EVT_FC_MIDDLE_UP, self.RectGotHit)
 586             Canvas.AddText("Middle Up", x, y, Position = "bl")
 587             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 588
 589             x = 20
 590             y += dy
 591             color = "CORAL"
 592             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 593             R.Name = color + " Rectangle"
 594             R.Bind(FloatCanvas.EVT_FC_MIDDLE_DCLICK, self.RectGotHit)
 595             Canvas.AddText("Middle DoubleClick", x, y, Position = "bl")
 596             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 597
 598             x += dx
 599             color = "CYAN"
 600             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 601             R.Name = color + " Rectangle"
 602             R.Bind(FloatCanvas.EVT_FC_RIGHT_UP, self.RectGotHit)
 603             Canvas.AddText("Right Up", x, y, Position = "bl")
 604             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 605
 606             x += dx
 607             color = "LIME GREEN"
 608             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 609             R.Name = color + " Rectangle"
 610             R.Bind(FloatCanvas.EVT_FC_RIGHT_DCLICK, self.RectGotHit)
 611             Canvas.AddText("Right Double Click", x, y, Position = "bl")
 612             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 613
 614             x = 20
 615             y += dy
 616             color = "MEDIUM GOLDENROD"
 617             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 618             R.Name = color
 619             R.Bind(FloatCanvas.EVT_FC_RIGHT_DOWN, self.RectGotHitRight)
 620             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectGotHitLeft)
 621             Canvas.AddText("L and R Click", x, y, Position = "bl")
 622             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 623
 624             x += dx
 625             color = "SALMON"
 626             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 627             R.Name = color + " Rectangle"
 628             R.Bind(FloatCanvas.EVT_FC_ENTER_OBJECT, self.RectMouseOver)
 629             Canvas.AddText("Mouse Enter", x, y, Position = "bl")
 630             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 631
 632             x += dx
 633             color = "MEDIUM VIOLET RED"
 634             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 635             R.Name = color
 636             R.Bind(FloatCanvas.EVT_FC_LEAVE_OBJECT, self.RectMouseLeave)
 637             Canvas.AddText("Mouse Leave", x, y, Position = "bl")
 638             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 639
 640             x = 20
 641             y += dy
 642             color = "SKY BLUE"
 643             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color)
 644             R.Name = color
 645             R.Bind(FloatCanvas.EVT_FC_ENTER_OBJECT, self.RectMouseOver)
 646             R.Bind(FloatCanvas.EVT_FC_LEAVE_OBJECT, self.RectMouseLeave)
 647             Canvas.AddText("Enter and Leave", x, y, Position = "bl")
 648             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 649
 650             x += dx
 651             color = "WHEAT"
 652             R = Canvas.AddRectangle(x, y, w+12, h, LineColor = None, FillColor = color)
 653             R.Name = color
 654             R.Bind(FloatCanvas.EVT_FC_ENTER_OBJECT, self.RectMouseOver)
 655             R.Bind(FloatCanvas.EVT_FC_LEAVE_OBJECT, self.RectMouseLeave)
 656             Canvas.AddText("Mouse Enter&Leave", x, y, Position = "bl")
 657             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 658
 659             x += dx
 660             color = "KHAKI"
 661             R = Canvas.AddRectangle(x-12, y, w+12, h, LineColor = None, FillColor = color)
 662             R.Name = color
 663             R.Bind(FloatCanvas.EVT_FC_ENTER_OBJECT, self.RectMouseOver)
 664             R.Bind(FloatCanvas.EVT_FC_LEAVE_OBJECT, self.RectMouseLeave)
 665             Canvas.AddText("Mouse ENter&Leave", x, y, Position = "bl")
 666             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 667
 668             x = 20
 669             y += dy
 670             L = Canvas.AddLine(( (x, y), (x+10, y+10), (x+w, y+h) ), LineWidth = 2, LineColor = "Red")
 671             L.Name = "A Line"
 672             L.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectGotHitLeft)
 673             Canvas.AddText("Left Down", x, y, Position = "bl")
 674             Canvas.AddText(L.Name, x, y+h, Position = "tl")
 675
 676             x += dx
 677             color = "SEA GREEN"
 678             Points = Numeric.array(( (x, y), (x, y+2.*h/3), (x+w, y+h), (x+w, y+h/2.), (x + 2.*w/3, y+h/2.), (x + 2.*w/3,y) ), Numeric.Float)
 679             R = Canvas.AddPolygon(Points,  LineWidth = 2, FillColor = color)
 680             R.Name = color + " Polygon"
 681             R.Bind(FloatCanvas.EVT_FC_RIGHT_DOWN, self.RectGotHitRight)
 682             Canvas.AddText("RIGHT_DOWN", x, y, Position = "bl")
 683             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 684
 685             x += dx
 686             color = "Red"
 687             Points = Numeric.array(( (x, y), (x, y+2.*h/3), (x+w, y+h), (x+w, y+h/2.), (x + 2.*w/3, y+h/2.), (x + 2.*w/3,y) ), Numeric.Float)
 688             R = Canvas.AddPointSet(Points,  Diameter = 4, Color = color)
 689             R.Name = "PointSet"
 690             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.PointSetGotHit)
 691             Canvas.AddText("LEFT_DOWN", x, y, Position = "bl")
 692             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 693
 694             x = 20
 695             y += dy
 696             T = Canvas.AddText("Hit-able Text", x, y, Size = 15, Color = "Red", Position = 'tl')
 697             T.Name = "Hit-able Text"
 698             T.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectGotHitLeft)
 699             Canvas.AddText("Left Down", x, y, Position = "bl")
 700
 701             x += dx
 702             T = Canvas.AddScaledText("Scaled Text", x, y, Size = 1./2*h, Color = "Pink", Position = 'bl')
 703             Canvas.AddPointSet( (x, y), Diameter = 3)
 704             T.Name = "Scaled Text"
 705             T.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectGotHitLeft)
 706             Canvas.AddText("Left Down", x, y, Position = "tl")
 707
 708             self.Canvas.ZoomToBB()
 709
 710         def TestHitTestForeground(self,event=None):
 711             print "Running: TestHitTestForeground"
 712             wx.GetApp().Yield(True)
 713
 714             self.UnBindAllMouseEvents()
 715             Canvas = self.Canvas
 716
 717             Canvas.ClearAll()
 718             Canvas.SetProjectionFun(None)
 719
 720             #Add a Hitable rectangle
 721             w, h = 60, 20
 722
 723             dx = 80
 724             dy = 40
 725             x,y = 20, 20
 726
 727             color = "Red"
 728             R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color, InForeground = False)
 729             R.Name = color + "Rectangle"
 730             R.HitFill = True
 731             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectGotHit)
 732             Canvas.AddText("Left Click Fill", x, y, Position = "bl")
 733             Canvas.AddText(R.Name, x, y+h, Position = "tl")
 734
 735             ## A set of Rectangles that move together
 736
 737             ## NOTE: In a real app, it might be better to create a new
 738             ## custom FloatCanvas DrawObject
 739
 740             self.MovingRects = []
 741             x += dx
 742             color = "LightBlue"
 743             R = Canvas.AddRectangle(x, y, w/2, h/2, LineWidth = 2, FillColor = color, InForeground = True)
 744             R.HitFill = True
 745             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectMoveLeft)
 746             L = Canvas.AddText("Left", x + w/4, y + h/4, Position = "cc", InForeground = True)
 747             self.MovingRects.extend( (R,L) )
 748
 749             x += w/2
 750             R = Canvas.AddRectangle(x, y, w/2, h/2, LineWidth = 2, FillColor = color, InForeground = True)
 751             R.HitFill = True
 752             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectMoveRight)
 753             L = Canvas.AddText("Right", x + w/4, y + h/4, Position = "cc", InForeground = True)
 754             self.MovingRects.extend( (R,L) )
 755
 756             x -= w/2
 757             y += h/2
 758             R = Canvas.AddRectangle(x, y, w/2, h/2, LineWidth = 2, FillColor = color, InForeground = True)
 759             R.HitFill = True
 760             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectMoveUp)
 761             L = Canvas.AddText("Up", x + w/4, y + h/4, Position = "cc", InForeground = True)
 762             self.MovingRects.extend( (R,L) )
 763
 764
 765             x += w/2
 766             R = Canvas.AddRectangle(x, y, w/2, h/2, LineWidth = 2, FillColor = color, InForeground = True)
 767             R.HitFill = True
 768             R.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RectMoveDown)
 769             L = Canvas.AddText("Down", x + w/4, y + h/4, Position = "cc", InForeground = True)
 770             self.MovingRects.extend( (R,L) )
 771
 772             self.Canvas.ZoomToBB()
 773
 774         def RectMoveLeft(self,Object):
 775             self.MoveRects("left")
 776
 777         def RectMoveRight(self,Object):
 778             self.MoveRects("right")
 779
 780         def RectMoveUp(self,Object):
 781             self.MoveRects("up")
 782
 783         def RectMoveDown(self,Object):
 784             self.MoveRects("down")
 785
 786         def MoveRects(self, Dir):
 787             for Object in self.MovingRects:
 788                 X,Y = Object.XY
 789                 if Dir == "left": X -= 10
 790                 elif Dir == "right": X += 10
 791                 elif Dir == "up": Y += 10
 792                 elif Dir == "down": Y -= 10
 793                 Object.SetXY(X,Y)
 794             self.Canvas.Draw()
 795
 796
 797         def PointSetGotHit(self, Object):
 798             print Object.Name, "Got Hit\n"
 799
 800         def RectGotHit(self, Object):
 801             print Object.Name, "Got Hit\n"
 802
 803         def RectGotHitRight(self, Object):
 804             print Object.Name, "Got Hit With Right\n"
 805
 806         def RectGotHitLeft(self, Object):
 807             print Object.Name, "Got Hit with Left\n"
 808
 809         def RectMouseOver(self, Object):
 810             print "Mouse entered:", Object.Name
 811
 812         def RectMouseLeave(self, Object):
 813             print "Mouse left ", Object.Name
 814
 815
 816         def TestText(self, event= None):
 817             wx.GetApp().Yield(True)
 818
 819             self.BindAllMouseEvents()
 820             Canvas = self.Canvas
 821             Canvas.ClearAll()
 822             Canvas.SetProjectionFun(None)
 823
 824             x,y  = (0, 0)
 825
 826             ## Add a non-visible rectangle, just to get a Bounding Box
 827             ## Text objects have a zero-size bounding box, because it changes with zoom
 828             Canvas.AddRectangle(-10,-10,20,20,LineWidth = 1, LineColor = None)
 829
 830             # Text
 831             String = "Some text"
 832 ##            for i in range(10):
 833 ##                ts = random.randint(10,40)
 834 ##                cf = random.randint(0,len(colors)-1)
 835 ##                x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
 836             self.Canvas.AddText("Top Left",x,y,Size = 14,Color = "Yellow",BackgroundColor = "Blue", Position = "tl")
 837             self.Canvas.AddText("Bottom Left",x,y,Size = 14,Color = "Cyan",BackgroundColor = "Black",Position = "bl")
 838             self.Canvas.AddText("Top Right",x,y,Size = 14,Color = "Black",BackgroundColor = "Cyan",Position = "tr")
 839             self.Canvas.AddText("Bottom Right",x,y,Size = 14,Color = "Blue",BackgroundColor = "Yellow",Position = "br")
 840             Canvas.AddPointSet((x,y), Color = "White", Diameter = 2)
 841
 842             x,y  = (0, 2)
 843
 844             Canvas.AddPointSet((x,y), Color = "White", Diameter = 2)
 845             self.Canvas.AddText("Top Center",x,y,Size = 14,Color = "Black",Position = "tc")
 846             self.Canvas.AddText("Bottom Center",x,y,Size = 14,Color = "White",Position = "bc")
 847
 848             x,y  = (0, 4)
 849
 850             Canvas.AddPointSet((x,y), Color = "White", Diameter = 2)
 851             self.Canvas.AddText("Center Right",x,y,Size = 14,Color = "Black",Position = "cr")
 852             self.Canvas.AddText("Center Left",x,y,Size = 14,Color = "Black",Position = "cl")
 853
 854             x,y  = (0, -2)
 855
 856             Canvas.AddPointSet((x,y), Color = "White", Diameter = 2)
 857             self.Canvas.AddText("Center Center",x,y,Size = 14,Color = "Black",Position = "cc")
 858
 859             self.Canvas.AddText("40 Pixels",-10,8,Size = 40)
 860             self.Canvas.AddText("20 Pixels",-10,5,Size = 20)
 861             self.Canvas.AddText("10 Pixels",-10,3,Size = 10)
 862
 863             self.Canvas.AddText("MODERN Font", -10, 0, Family = wx.MODERN)
 864             self.Canvas.AddText("DECORATIVE Font", -10, -1, Family = wx.DECORATIVE)
 865             self.Canvas.AddText("ROMAN Font", -10, -2, Family = wx.ROMAN)
 866             self.Canvas.AddText("SCRIPT Font", -10, -3, Family = wx.SCRIPT)
 867             self.Canvas.AddText("ROMAN BOLD Font", -10, -4, Family = wx.ROMAN, Weight=wx.BOLD)
 868             self.Canvas.AddText("ROMAN ITALIC BOLD Font", -10, -5, Family = wx.ROMAN, Weight=wx.BOLD, Style=wx.ITALIC)
 869
 870             # NOTE: this font exists on my Linux box..who knows were else you'll find it!
 871             Font = wx.Font(20, wx.DEFAULT, wx.ITALIC, wx.NORMAL, False, "zapf chancery")
 872             self.Canvas.AddText("zapf chancery Font", -10, -6, Font = Font)
 873
 874             self.Canvas.ZoomToBB()
 875
 876         def TestScaledText(self, event= None):
 877             wx.GetApp().Yield(True)
 878
 879             self.BindAllMouseEvents()
 880             Canvas = self.Canvas
 881             Canvas.ClearAll()
 882             Canvas.SetProjectionFun(None)
 883
 884             x,y  = (0, 0)
 885
 886             T = Canvas.AddScaledText("Top Left",x,y,Size = 5,Color = "Yellow",BackgroundColor = "Blue", Position = "tl")
 887             T = Canvas.AddScaledText("Bottom Left",x,y,Size = 5,Color = "Cyan",BackgroundColor = "Black",Position = "bl")
 888             T = Canvas.AddScaledText("Top Right",x,y,Size = 5,Color = "Black",BackgroundColor = "Cyan",Position = "tr")
 889             T = Canvas.AddScaledText("Bottom Right",x,y,Size = 5,Color = "Blue",BackgroundColor = "Yellow",Position = "br")
 890             Canvas.AddPointSet((x,y), Color = "Red", Diameter = 4)
 891
 892
 893             x,y  = (0, 20)
 894
 895             Canvas.AddScaledText("Top Center",x,y,Size = 7,Color = "Black",Position = "tc")
 896             Canvas.AddScaledText("Bottom Center",x,y,Size = 7,Color = "White",Position = "bc")
 897             Canvas.AddPointSet((x,y), Color = "White", Diameter = 4)
 898
 899             x,y  = (0, -20)
 900
 901             Canvas.AddScaledText("Center Right",x,y,Size = 9,Color = "Black",Position = "cr")
 902             Canvas.AddScaledText("Center Left",x,y,Size = 9,Color = "Black",Position = "cl")
 903             Canvas.AddPointSet((x,y), Color = "White", Diameter = 4)
 904
 905             x = -200
 906
 907             self.Canvas.AddScaledText("MODERN Font", x, 0, Size = 7, Family = wx.MODERN, Color = (0,0,0))
 908             self.Canvas.AddScaledText("DECORATIVE Font", x, -10, Size = 7, Family = wx.DECORATIVE, Color = (0,0,1))
 909             self.Canvas.AddScaledText("ROMAN Font", x, -20, Size = 7, Family = wx.ROMAN)
 910             self.Canvas.AddScaledText("SCRIPT Font", x, -30, Size = 7, Family = wx.SCRIPT)
 911             self.Canvas.AddScaledText("ROMAN BOLD Font", x, -40, Size = 7, Family = wx.ROMAN, Weight=wx.BOLD)
 912             self.Canvas.AddScaledText("ROMAN ITALIC BOLD Font", x, -50, Size = 7, Family = wx.ROMAN, Weight=wx.BOLD, Style=wx.ITALIC)
 913             Canvas.AddPointSet((x,0), Color = "White", Diameter = 4)
 914
 915
 916             # NOTE: this font exists on my Linux box..who knows were else you'll find it!
 917             x,y = (-100, 50)
 918             Font = wx.Font(12, wx.DEFAULT, wx.ITALIC, wx.NORMAL, False, "zapf chancery")
 919             T = self.Canvas.AddScaledText("zapf chancery Font", x, y, Size = 20, Font = Font, Position = 'bc')
 920
 921             x,y = (-50, -50)
 922             Font = wx.Font(12, wx.DEFAULT, wx.ITALIC, wx.NORMAL, False, "bookman")
 923             T = self.Canvas.AddScaledText("Bookman Font", x, y, Size = 8, Font = Font)
 924
 925             self.Canvas.ZoomToBB()
 926
 927         def DrawMap(self,event = None):
 928             wx.GetApp().Yield(True)
 929             import os, time
 930             self.BindAllMouseEvents()
 931
 932         ## Test of Actual Map Data
 933             self.Canvas.ClearAll()
 934             self.Canvas.SetProjectionFun("FlatEarth")
 935             #start = time.clock()
 936             Shorelines = Read_MapGen(os.path.join("data",'world.dat'),stats = 0)
 937             #print "It took %f seconds to load %i shorelines"%(time.clock() - start,len(Shorelines) )
 938             #start = time.clock()
 939             for segment in Shorelines:
 940                 self.Canvas.AddLine(segment)
 941             #print "It took %f seconds to add %i shorelines"%(time.clock() - start,len(Shorelines) )
 942             #start = time.clock()
 943             self.Canvas.ZoomToBB()
 944             #print "It took %f seconds to draw %i shorelines"%(time.clock() - start,len(Shorelines) )
 945
 946
 947         def LineTest(self,event = None):
 948             wx.GetApp().Yield(True)
 949             import os, time
 950 #            import random
 951             colors = self.colors
 952             Range = (-10,10)
 953         ## Test of drawing lots of lines
 954             Canvas = self.Canvas
 955             Canvas.ClearAll()
 956             Canvas.SetProjectionFun(None)
 957             #start = time.clock()
 958             linepoints = []
 959             linecolors = []
 960             linewidths = []
 961             for i in range(2000):
 962                 points = (random.randint(Range[0],Range[1]),
 963                          random.randint(Range[0],Range[1]),
 964                          random.randint(Range[0],Range[1]),
 965                          random.randint(Range[0],Range[1]))
 966                 linepoints.append(points)
 967                 linewidths.append(random.randint(1,10) )
 968                 linecolors.append(random.randint(0,len(colors)-1) )
 969             for (points,color,width) in zip(linepoints,linecolors,linewidths):
 970                 Canvas.AddLine((points[0:2],points[2:4]), LineWidth = width, LineColor = colors[color])
 971             #print "It took %f seconds to add %i lines"%(time.clock() - start,len(linepoints) )
 972             #start = time.clock()
 973             Canvas.ZoomToBB()
 974             #print "It took %f seconds to draw %i lines"%(time.clock() - start,len(linepoints) )
 975
 976         def SpeedTest(self,event=None):
 977             wx.GetApp().Yield(True)
 978 #            import random
 979 #            import RandomArray
 980             BigRange = (-1000,1000)
 981             colors = self.colors
 982
 983             self.UnBindAllMouseEvents()
 984             Canvas = self.Canvas
 985
 986             Canvas.ClearAll()
 987             Canvas.SetProjectionFun(None)
 988
 989             # Lots of Text
 990             String = "Unscaled text"
 991             coords = []
 992             text = []
 993             for i in range(5000):
 994                 x,y = (random.uniform(BigRange[0],BigRange[1]),random.uniform(BigRange[0],BigRange[1]))
 995                 coords.append( (x,y) )
 996             print "Drawing the Numbers"
 997             start = time.clock()
 998             for i in xrange( len(coords) ):
 999                 Canvas.AddText("%i"%(i),
1000                                coords[i][0],
1001                                coords[i][1],
1002                                Size = 12,
1003                                Position = "cc",
1004                                BackgroundColor = "White")
1005             print "It took %s seconds to add the numbers"%(time.clock() - start)
1006
1007
1008 ##            ObjectList = []
1009
1010 ##            print "Building a list of lots of random objects"
1011 ##            ##                Random tests of everything:
1012 ##            def MakeRange():
1013 ##                while True:
1014 ##                    Range = ( random.randint(BigRange[0],BigRange[1]), random.randint(BigRange[0],BigRange[1]) )
1015 ##                    if abs (Range[0] -  Range[1]) < 10:
1016 ##                        continue
1017 ##                    if Range[0] > Range[1]:
1018 ##                        Range = ( Range[1], Range[0] )
1019 ##                    break
1020 ##                return Range
1021
1022 ##            # Rectangles
1023 ##            for i in range(300):
1024 ##                Range = MakeRange()
1025 ##                x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
1026 ##                lw = random.randint(1,5)
1027 ##                cf = random.randint(0,len(colors)-1)
1028 ##                h = random.randint(1, Range[1] - Range[0])
1029 ##                w = random.randint(1, Range[1] - Range[0])
1030 ##                ObjectList.append(FloatCanvas.Rectangle(x,y,h,w,LineWidth = lw,FillColor = colors[cf]))
1031
1032 ##            # Ellipses
1033 ##            for i in range(300):
1034 ##                Range = MakeRange()
1035 ##                x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
1036 ##                lw = random.randint(1,5)
1037 ##                cf = random.randint(0,len(colors)-1)
1038 ##                h = random.randint(1, Range[1] - Range[0])
1039 ##                w = random.randint(1, Range[1] - Range[0])
1040 ##                ObjectList.append(FloatCanvas.Ellipse(x,y,h,w,LineWidth = lw,FillColor = colors[cf]))
1041
1042
1043 ##            # Circles
1044 ##            for i in range(500):
1045 ##                Range = MakeRange()
1046 ##                x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
1047 ##                D = random.randint(1, (Range[1] - Range[0]) / 2)
1048 ##                lw = random.randint(1,5)
1049 ##                cf = random.randint(0,len(colors)-1)
1050 ##                cl = random.randint(0,len(colors)-1)
1051 ##                ObjectList.append(FloatCanvas.Circle(x,y,D,LineWidth = lw,LineColor = colors[cl],FillColor = colors[cf]))
1052 ##                #ObjectList.append(FloatCanvas.Text("Circle # %i"%(i),x,y,Size = 12,BackgroundColor = None,Position = "cc"))
1053
1054 ##                # Lines
1055 ##            for i in range(500):
1056 ##                Range = MakeRange()
1057 ##                points = []
1058 ##                for j in range(random.randint(2,100)):
1059 ##                    point = (random.randint(Range[0],Range[1]),random.randint(Range[0],Range[1]))
1060 ##                    points.append(point)
1061 ##                lw = random.randint(1,10)
1062 ##                cf = random.randint(0,len(colors)-1)
1063 ##                cl = random.randint(0,len(colors)-1)
1064 ##                ObjectList.append(FloatCanvas.Line(points, LineWidth = lw, LineColor = colors[cl]) )
1065
1066 ##                # Polygons
1067 ##            for i in range(300):
1068 ##                Range = MakeRange()
1069 ##                points = []
1070 ##                for j in range(random.randint(2,60)):
1071 ##                    point = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
1072 ##                    points.append(point)
1073 ##                lw = random.randint(1,6)
1074 ##                cf = random.randint(0,len(colors)-1)
1075 ##                cl = random.randint(0,len(colors)-1)
1076 ##                ObjectList.append(FloatCanvas.Polygon(points,
1077 ##                                       LineWidth = lw,
1078 ##                                       LineColor = colors[cl],
1079 ##                                       FillColor = colors[cf],
1080 ##                                       FillStyle = 'Solid') )
1081 ##            random.shuffle(ObjectList)
1082 ##            print "Adding lots of random objects"
1083 ##            start = time.clock()
1084 ##            for Object in ObjectList:
1085 ##                Canvas.AddObject(Object)
1086 ##            print "It took %s Seconds to add %i objects "%(time.clock() - start, len(ObjectList) )
1087
1088 ##            ## Pointset
1089 ##            for i in range(100):
1090 ##                points = []
1091 ##                points = RandomArray.uniform(Range[0],Range[1],(100,2))
1092 ##                cf = random.randint(0,len(colors)-1)
1093 ##                D = random.randint(1,4)
1094 ##                Canvas.AddPointSet(points, Color = colors[cf], Diameter = D)
1095
1096
1097 ##            # Scaled Text
1098 ##            String = "Scaled text"
1099 ##            for i in range(30):
1100 ##                ts = random.random()*3 + 0.2
1101 ##                cf = random.randint(0,len(colors)-1)
1102 ##                x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
1103 ##                Canvas.AddScaledText(String, x, y, Size = ts, Color = colors[cf], Position = "cc")
1104
1105             Canvas.ZoomToBB()
1106
1107
1108         def TempTest(self, event= None):
1109             "Running the Temporary test"
1110             wx.GetApp().Yield(True)
1111
1112             self.UnBindAllMouseEvents()
1113             Canvas = self.Canvas
1114             Canvas.ClearAll()
1115             Canvas.SetProjectionFun(None)
1116
1117 #            import random
1118 #            import RandomArray
1119             Range = (-10,10)
1120
1121             # Create a random Polygon
1122             points = []
1123             for j in range(6):
1124                 point = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1]))
1125                 points.append(point)
1126             Poly = Canvas.AddPolygon(points,
1127                                      LineWidth = 2,
1128                                      LineColor = "Black",
1129                                      FillColor = "LightBlue",
1130                                      FillStyle = 'Solid')
1131
1132             Poly.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.SelectPoly)
1133
1134             self.SelectedPoly = None
1135             self.SelectPoints = []
1136             self.SelectedPoint = None
1137
1138             Canvas.ZoomToBB()
1139
1140         def SelectPoly(self, Object):
1141             print "In SelectPoly"
1142             Canvas = self.Canvas
1143             if Object is self.SelectedPoly:
1144                 pass
1145             else:
1146                 #fixme: Do something to unselect the old one
1147                 self.SelectedPoly = Object
1148                 Canvas.RemoveObjects(self.SelectPoints)
1149                 self.SelectPoints = []
1150             # Draw points on the Vertices of the Selected Poly:
1151             for i, point in enumerate(Object.Points):
1152                 P = Canvas.AddPointSet(point, Diameter = 6, Color = "Red")
1153                 P.VerticeNum = i
1154                 P.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.SelectPointHit)
1155                 self.SelectPoints.append(P)
1156             #Canvas.ZoomToBB()
1157             Canvas.Draw()
1158
1159         def SelectPointHit(self, Point):
1160             print "Point Num: %i Hit"%Point.VerticeNum
1161             self.SelectedPoint = Point
1162
1163
1164
1165     class DemoApp(wx.App):
1166         """
1167         How the demo works:
1168
1169         Under the Draw menu, there are three options:
1170
1171         *Draw Test: will put up a picture of a bunch of randomly generated
1172         objects, of each kind supported.
1173
1174         *Draw Map: will draw a map of the world. Be patient, it is a big map,
1175         with a lot of data, and will take a while to load and draw (about 10 sec
1176         on my 450Mhz PIII). Redraws take about 2 sec. This demonstrates how the
1177         performance is not very good for large drawings.
1178
1179         *Clear: Clears the Canvas.
1180
1181         Once you have a picture drawn, you can zoom in and out and move about
1182         the picture. There is a tool bar with three tools that can be
1183         selected.
1184
1185         The magnifying glass with the plus is the zoom in tool. Once selected,
1186         if you click the image, it will zoom in, centered on where you
1187         clicked. If you click and drag the mouse, you will get a rubber band
1188         box, and the image will zoom to fit that box when you release it.
1189
1190         The magnifying glass with the minus is the zoom out tool. Once selected,
1191         if you click the image, it will zoom out, centered on where you
1192         clicked. (note that this takes a while when you are looking at the map,
1193         as it has a LOT of lines to be drawn. The image is double buffered, so
1194         you don't see the drawing in progress)
1195
1196         The hand is the move tool. Once selected, if you click and drag on the
1197         image, it will move so that the part you clicked on ends up where you
1198         release the mouse. Nothing is changed while you are dragging. The
1199         drawing is too slow for that.
1200
1201         I'd like the cursor to change as you change tools, but the stock
1202         wxCursors didn't include anything I liked, so I stuck with the
1203         pointer. Please let me know if you have any nice cursor images for me to
1204         use.
1205
1206
1207         Any bugs, comments, feedback, questions, and especially code are welcome:
1208
1209         -Chris Barker
1210
1211         Chris.Barker@noaa.gov
1212
1213         """
1214
1215         def __init__(self, *args, **kwargs):
1216             wx.App.__init__(self, *args, **kwargs)
1217
1218         def OnInit(self):
1219             wx.InitAllImageHandlers()
1220             frame = DrawFrame(None, -1, "FloatCanvas Demo App",wx.DefaultPosition,(700,700))
1221
1222             self.SetTopWindow(frame)
1223             frame.Show()
1224
1225             ## check to see if the demo is set to start in a particular mode.
1226             if StartUpDemo == "text":
1227                 frame.TestText()
1228             if StartUpDemo == "stext":
1229                 frame.TestScaledText()
1230             elif StartUpDemo == "all":
1231                 frame.DrawTest()
1232             elif StartUpDemo == "map":
1233                 frame.DrawMap()
1234             elif StartUpDemo == "hit":
1235                 frame.TestHitTest()
1236             elif StartUpDemo == "hitf":
1237                 "starting TestHitTestForeground"
1238                 frame.TestHitTestForeground()
1239             elif StartUpDemo == "animate":
1240                 "starting TestAnimation"
1241                 frame.TestAnimation()
1242             elif StartUpDemo == "speed":
1243                 "starting SpeedTest"
1244                 frame.SpeedTest()
1245             elif StartUpDemo == "temp":
1246                 "starting temp Test"
1247                 frame.TempTest()
1248
1249             return True
1250
1251     def Read_MapGen(filename,stats = 0,AllLines=0):
1252         """
1253         This function reads a MapGen Format file, and
1254         returns a list of NumPy arrays with the line segments in them.
1255
1256         Each NumPy array in the list is an NX2 array of Python Floats.
1257
1258         The demo should have come with a file, "world.dat" that is the
1259         shorelines of the whole world, in MapGen format.
1260
1261         """
1262         import string
1263         file = open(filename,'rt')
1264         data = file.readlines()
1265         data = map(string.strip,data)
1266
1267         Shorelines = []
1268         segment = []
1269         for line in data:
1270             if line:
1271                 if line == "# -b": #New segment beginning
1272                     if segment: Shorelines.append(Numeric.array(segment))
1273                     segment = []
1274                 else:
1275                     segment.append(map(float,string.split(line)))
1276         if segment: Shorelines.append(Numeric.array(segment))
1277
1278         if stats:
1279             NumSegments = len(Shorelines)
1280             NumPoints = 0
1281             for segment in Shorelines:
1282                 NumPoints = NumPoints + len(segment)
1283             AvgPoints = NumPoints / NumSegments
1284             print "Number of Segments: ", NumSegments
1285             print "Average Number of Points per segment: ",AvgPoints
1286         if AllLines:
1287             Lines = []
1288             for segment in Shorelines:
1289                 Lines.append(segment[0])
1290                 for point in segment[1:-1]:
1291                     Lines.append(point)
1292                     Lines.append(point)
1293                 Lines.append(segment[-1])
1294             return Lines
1295         else:
1296             return Shorelines
1297
1298     ## for the wxPython demo:
1299     try:
1300         import floatcanvas
1301     except ImportError: # if it's not there locally, try the wxPython lib.
1302         from wx.lib import floatcanvas
1303
1304     overview = floatcanvas.__doc__
1305
1306 if __name__ == "__main__":
1307     if not haveNumeric:
1308         print errorText
1309     else:
1310         app = DemoApp(False)# put in True if you want output to go to it's own window.
1311         app.MainLoop()
1312
1313
1314
1315
1316
1317
1318
1319
1320
1321
1322
1323
1324
1325
1326
1327
1328
1329
1330
1331
1332
1333