X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/5e1796efd30885c38cdf3d4779ad9ce026111649..628eae0bcd7916c57847b7eab739d4cc32534ccf:/wxPython/demo/FloatCanvas.py diff --git a/wxPython/demo/FloatCanvas.py b/wxPython/demo/FloatCanvas.py index 0c1f670d1f..7f1d2d0d00 100644 --- a/wxPython/demo/FloatCanvas.py +++ b/wxPython/demo/FloatCanvas.py @@ -1,101 +1,59 @@ #!/usr/bin/env python2.3 + +import wx + +##First, make sure Numeric or numarray can be imported. try: import Numeric import RandomArray haveNumeric = True except ImportError: + # Numeric isn't there, let's try numarray try: import numarray as Numeric import numarray.random_array as RandomArray haveNumeric = True except ImportError: + # numarray isn't there either haveNumeric = False + errorText = ( + "The FloatCanvas requires either the Numeric or numarray module\n\n" + "You can get them at:\n" + "http://sourceforge.net/projects/numpy\n\n" + "NOTE: The Numeric module is substantially faster than numarray for this\n" + "purpose, if you have lots of objects\n" + ) + +#--------------------------------------------------------------------------- -if not haveNumeric: - errorText = """ -The FloatCanvas requires either the Numeric or Numarray module: -You can get them at: - http://sourceforge.net/projects/numpy - -NOTE: The Numeric module is substantially faster than numarray for this -purpose, if you have lot's of objects -""" - - def runTest(frame, nb, log): - dlg = wx.MessageDialog(frame, errorText, 'Sorry', wx.OK | - wx.ICON_INFORMATION) - dlg.ShowModal() - dlg.Destroy() - - overview = "" - -else: - StartUpDemo = "all" - if __name__ == "__main__": # parse options if run stand-alone - # check options: - import sys, getopt - optlist, args = getopt.getopt(sys.argv[1:],'l',["local","all","text","map","stext","hit","hitf","animate","speed","temp","props"]) - - for opt in optlist: - if opt[0] == "--all": - StartUpDemo = "all" - elif opt[0] == "--text": - StartUpDemo = "text" - elif opt[0] == "--map": - StartUpDemo = "map" - elif opt[0] == "--stext": - StartUpDemo = "stext" - elif opt[0] == "--hit": - StartUpDemo = "hit" - elif opt[0] == "--hitf": - StartUpDemo = "hitf" - elif opt[0] == "--animate": - StartUpDemo = "animate" - elif opt[0] == "--speed": - StartUpDemo = "speed" - elif opt[0] == "--temp": - StartUpDemo = "temp" - elif opt[0] == "--props": - StartUpDemo = "props" - import wx - import time, random - - def runTest(frame, nb, log): - """ - This method is used by the wxPython Demo Framework for integrating - this demo with the rest. - """ - win = DrawFrame(None, -1, "FloatCanvas Drawing Window",wx.DefaultPosition,(500,500)) - frame.otherWin = win - win.Show(True) - win.DrawTest() - +def BuildDrawFrame(): # this gets called when needed, rather than on import try: from floatcanvas import NavCanvas, FloatCanvas except ImportError: # if it's not there locally, try the wxPython lib. from wx.lib.floatcanvas import NavCanvas, FloatCanvas - import wxPython.lib.colourdb - + import wx.lib.colourdb + import time, random + class DrawFrame(wx.Frame): - + """ A frame used for the FloatCanvas Demo - + """ - - + + def __init__(self,parent, id,title,position,size): wx.Frame.__init__(self,parent, id,title,position, size) - + ## Set up the MenuBar MenuBar = wx.MenuBar() - + file_menu = wx.Menu() item = file_menu.Append(-1, "&Close","Close this frame") self.Bind(wx.EVT_MENU, self.OnQuit, item) MenuBar.Append(file_menu, "&File") - + draw_menu = wx.Menu() item = draw_menu.Append(-1, "&Draw Test","Run a test of drawing random components") @@ -103,7 +61,7 @@ else: item = draw_menu.Append(-1, "&Line Test","Run a test of drawing random lines") self.Bind(wx.EVT_MENU, self.LineTest, item) - + item = draw_menu.Append(-1, "Draw &Map","Run a test of drawing a map") self.Bind(wx.EVT_MENU, self.DrawMap, item) item = draw_menu.Append(-1, "&Text Test","Run a test of text drawing") @@ -122,21 +80,24 @@ else: self.Bind(wx.EVT_MENU, self.SpeedTest, item) item = draw_menu.Append(-1, "Change &Properties","Run a test of Changing Object Properties") self.Bind(wx.EVT_MENU, self.PropertiesChangeTest, item) + item = draw_menu.Append(-1, "&Arrows","Run a test of Arrows") + self.Bind(wx.EVT_MENU, self.ArrowTest, item) + MenuBar.Append(draw_menu, "&Tests") - + view_menu = wx.Menu() item = view_menu.Append(-1, "Zoom to &Fit","Zoom to fit the window") self.Bind(wx.EVT_MENU, self.ZoomToFit, item) MenuBar.Append(view_menu, "&View") - + help_menu = wx.Menu() item = help_menu.Append(-1, "&About", "More information About this program") self.Bind(wx.EVT_MENU, self.OnAbout, item) MenuBar.Append(help_menu, "&Help") - + self.SetMenuBar(MenuBar) - + self.CreateStatusBar() # Add the Canvas self.Canvas = NavCanvas.NavCanvas(self, @@ -153,8 +114,8 @@ else: self.EventsAreBound = False ## getting all the colors and linestyles for random objects - wxPython.lib.colourdb.updateColourDB() - self.colors = wxPython.lib.colourdb.getColourList() + wx.lib.colourdb.updateColourDB() + self.colors = wx.lib.colourdb.getColourList() #self.LineStyles = FloatCanvas.DrawObject.LineStyleList.keys() @@ -179,7 +140,20 @@ else: self.EventsAreBound = True def UnBindAllMouseEvents(self): - ## Here is how you catch FloatCanvas mouse events + ## Here is how you unbind FloatCanvas mouse events + FloatCanvas.EVT_LEFT_DOWN(self.Canvas, None ) + FloatCanvas.EVT_LEFT_UP(self.Canvas, None ) + FloatCanvas.EVT_LEFT_DCLICK(self.Canvas, None) + + FloatCanvas.EVT_MIDDLE_DOWN(self.Canvas, None ) + FloatCanvas.EVT_MIDDLE_UP(self.Canvas, None ) + FloatCanvas.EVT_MIDDLE_DCLICK(self.Canvas, None ) + + FloatCanvas.EVT_RIGHT_DOWN(self.Canvas, None ) + FloatCanvas.EVT_RIGHT_UP(self.Canvas, None ) + FloatCanvas.EVT_RIGHT_DCLICK(self.Canvas, None ) + + FloatCanvas.EVT_MOUSEWHEEL(self.Canvas, None ) FloatCanvas.EVT_LEFT_DOWN(self.Canvas, None ) FloatCanvas.EVT_LEFT_UP(self.Canvas, None ) FloatCanvas.EVT_LEFT_DCLICK(self.Canvas, None) @@ -239,41 +213,48 @@ else: def OnWheel(self, event): print "Mouse Wheel Moved in DrawFrame" self.PrintCoords(event) - + Rot = event.GetWheelRotation() + print "Wheel Rotation is:", Rot + print "Wheel Delta is:", event.GetWheelDelta() + Rot = Rot / abs(Rot) * 0.1 + if event.ControlDown(): # move left-right + self.Canvas.MoveImage( (Rot, 0), "Panel" ) + else: # move up-down + self.Canvas.MoveImage( (0, Rot), "Panel" ) + def OnMove(self, event): """ - Updates the staus bar with the world coordinates + Updates the status bar with the world coordinates """ self.SetStatusText("%.2f, %.2f"%tuple(event.Coords)) def OnAbout(self, event): print "OnAbout called" - + dlg = wx.MessageDialog(self, "This is a small program to demonstrate\n" "the use of the FloatCanvas\n", "About Me", wx.OK | wx.ICON_INFORMATION) dlg.ShowModal() dlg.Destroy() - + def ZoomToFit(self,event): self.Canvas.ZoomToBB() - + def Clear(self,event = None): self.UnBindAllMouseEvents() self.Canvas.ClearAll() self.Canvas.SetProjectionFun(None) self.Canvas.Draw() - + def OnQuit(self,event): self.Close(True) - + def OnCloseWindow(self, event): self.Destroy() - + def DrawTest(self,event=None): wx.GetApp().Yield() -# import random -# import RandomArray + Range = (-10,10) colors = self.colors @@ -283,8 +264,8 @@ else: Canvas.ClearAll() Canvas.SetProjectionFun(None) - ## Random tests of everything: - + ## Random tests of everything: + # Rectangles for i in range(3): x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) @@ -293,7 +274,7 @@ else: h = random.randint(1,5) w = random.randint(1,5) Canvas.AddRectangle(x,y,w,h,LineWidth = lw,FillColor = colors[cf]) - + # Ellipses for i in range(3): x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) @@ -302,15 +283,14 @@ else: h = random.randint(1,5) w = random.randint(1,5) Canvas.AddEllipse(x,y,h,w,LineWidth = lw,FillColor = colors[cf]) - + # Points for i in range(5): x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) D = random.randint(1,50) cf = random.randint(0,len(colors)-1) Canvas.AddPoint((x,y), Color = colors[cf], Diameter = D) - - # Circles + # Circles for i in range(5): x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) D = random.randint(1,5) @@ -319,8 +299,7 @@ else: cl = random.randint(0,len(colors)-1) Canvas.AddCircle(x,y,D,LineWidth = lw,LineColor = colors[cl],FillColor = colors[cf]) Canvas.AddText("Circle # %i"%(i),x,y,Size = 12,BackgroundColor = None,Position = "cc") - - # Lines + # Lines for i in range(5): points = [] for j in range(random.randint(2,10)): @@ -330,8 +309,7 @@ else: cf = random.randint(0,len(colors)-1) cl = random.randint(0,len(colors)-1) Canvas.AddLine(points, LineWidth = lw, LineColor = colors[cl]) - - # Polygons + # Polygons for i in range(3): points = [] for j in range(random.randint(2,6)): @@ -345,7 +323,7 @@ else: LineColor = colors[cl], FillColor = colors[cf], FillStyle = 'Solid') - + ## Pointset for i in range(4): points = [] @@ -353,7 +331,7 @@ else: cf = random.randint(0,len(colors)-1) D = random.randint(1,4) Canvas.AddPointSet(points, Color = colors[cf], Diameter = D) - + # Text String = "Unscaled text" for i in range(3): @@ -370,6 +348,17 @@ else: x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) Canvas.AddScaledText(String, x, y, Size = ts, Color = colors[cf], Position = "cc") + # Arrows + N = 5 + Points = RandomArray.uniform(Range[0], Range[1], (N,2) ) + for i in range(N): + Canvas.AddArrow(Points[i], + random.uniform(20,100), + Direction = random.uniform(0,360), + LineWidth = random.uniform(1,5), + LineColor = colors[random.randint(0,len(colors)-1)], + ArrowHeadAngle = random.uniform(20,90)) + Canvas.ZoomToBB() def TestAnimation(self,event=None): @@ -392,7 +381,7 @@ else: Canvas.ClearAll() Canvas.SetProjectionFun(None) - ## Random tests of everything: + ## Random tests of everything: colors = self.colors # Rectangles for i in range(3): @@ -402,7 +391,7 @@ else: h = random.randint(1,5) w = random.randint(1,5) Canvas.AddRectangle(x,y,h,w,LineWidth = lw,FillColor = colors[cf]) - + # Ellipses for i in range(3): x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) @@ -411,7 +400,7 @@ else: h = random.randint(1,5) w = random.randint(1,5) Canvas.AddEllipse(x,y,h,w,LineWidth = lw,FillColor = colors[cf]) - + # Circles for i in range(5): x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) @@ -421,7 +410,7 @@ else: cl = random.randint(0,len(colors)-1) Canvas.AddCircle(x,y,D,LineWidth = lw,LineColor = colors[cl],FillColor = colors[cf]) Canvas.AddText("Circle # %i"%(i),x,y,Size = 12,BackgroundColor = None,Position = "cc") - + # Lines for i in range(5): points = [] @@ -432,7 +421,7 @@ else: cf = random.randint(0,len(colors)-1) cl = random.randint(0,len(colors)-1) Canvas.AddLine(points, LineWidth = lw, LineColor = colors[cl]) - + # Polygons for i in range(3): points = [] @@ -447,7 +436,7 @@ else: LineColor = colors[cl], FillColor = colors[cf], FillStyle = 'Solid') - + # Scaled Text String = "Scaled text" for i in range(3): @@ -465,7 +454,7 @@ else: self.Timer = wx.PyTimer(self.ShowFrame) self.FrameDelay = 50 # milliseconds - + Canvas.ZoomToBB() def ShowFrame(self): @@ -484,7 +473,7 @@ else: wx.GetApp().Yield(True) else: self.Timer.Stop() - + def MoveMe(self, Object): self.MovingObject = Object @@ -497,7 +486,7 @@ else: self.TimeStep = 1 self.Timer.Start(self.FrameDelay) #print "Did %i frames in %f seconds"%(N, (time.time() - start) ) - + def TestHitTest(self,event=None): wx.GetApp().Yield() @@ -507,18 +496,18 @@ else: Canvas.ClearAll() Canvas.SetProjectionFun(None) - #Add a HitAble rectangle + #Add a Hit-able rectangle w, h = 60, 20 dx = 80 dy = 40 x,y = 20, 20 FontSize = 8 - + #Add one that is not HitAble Canvas.AddRectangle(x, y, w, h, LineWidth = 2) Canvas.AddText("Not Hit-able", x, y, Size = FontSize, Position = "bl") - + x += dx R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2) @@ -724,7 +713,7 @@ else: dx = 80 dy = 40 x,y = 20, 20 - + color = "Red" R = Canvas.AddRectangle(x, y, w, h, LineWidth = 2, FillColor = color, InForeground = False) R.Name = color + "Rectangle" @@ -734,10 +723,10 @@ else: Canvas.AddText(R.Name, x, y+h, Position = "tl") ## A set of Rectangles that move together - + ## NOTE: In a real app, it might be better to create a new ## custom FloatCanvas DrawObject - + self.MovingRects = [] x += dx color = "LightBlue" @@ -774,16 +763,16 @@ else: def RectMoveLeft(self,Object): self.MoveRects("left") - + def RectMoveRight(self,Object): self.MoveRects("right") - + def RectMoveUp(self,Object): self.MoveRects("up") - + def RectMoveDown(self,Object): self.MoveRects("down") - + def MoveRects(self, Dir): for Object in self.MovingRects: X,Y = Object.XY @@ -793,8 +782,8 @@ else: elif Dir == "down": Y -= 10 Object.SetXY(X,Y) self.Canvas.Draw() - - + + def PointSetGotHit(self, Object): print Object.Name, "Got Hit\n" @@ -837,19 +826,19 @@ else: Canvas.AddPointSet((x,y), Color = "White", Diameter = 2) x,y = (0, 2) - + Canvas.AddPointSet((x,y), Color = "White", Diameter = 2) self.Canvas.AddText("Top Center",x,y,Size = 14,Color = "Black",Position = "tc") self.Canvas.AddText("Bottom Center",x,y,Size = 14,Color = "White",Position = "bc") x,y = (0, 4) - + Canvas.AddPointSet((x,y), Color = "White", Diameter = 2) self.Canvas.AddText("Center Right",x,y,Size = 14,Color = "Black",Position = "cr") self.Canvas.AddText("Center Left",x,y,Size = 14,Color = "Black",Position = "cl") x,y = (0, -2) - + Canvas.AddPointSet((x,y), Color = "White", Diameter = 2) self.Canvas.AddText("Center Center",x,y,Size = 14,Color = "Black",Position = "cc") @@ -888,13 +877,13 @@ else: x,y = (0, 20) - + Canvas.AddScaledText("Top Center",x,y,Size = 7,Color = "Black",Position = "tc") Canvas.AddScaledText("Bottom Center",x,y,Size = 7,Color = "White",Position = "bc") Canvas.AddPointSet((x,y), Color = "White", Diameter = 4) x,y = (0, -20) - + Canvas.AddScaledText("Center Right",x,y,Size = 9,Color = "Black",Position = "cr") Canvas.AddScaledText("Center Left",x,y,Size = 9,Color = "Black",Position = "cl") Canvas.AddPointSet((x,y), Color = "White", Diameter = 4) @@ -920,17 +909,17 @@ else: T = self.Canvas.AddScaledText("Bookman Font", x, y, Size = 8, Font = Font) self.Canvas.ZoomToBB() - + def DrawMap(self,event = None): wx.GetApp().Yield() import os, time self.BindAllMouseEvents() - + ## Test of Actual Map Data self.Canvas.ClearAll() self.Canvas.SetProjectionFun("FlatEarth") #start = time.clock() - Shorelines = Read_MapGen(os.path.join("data",'world.dat'),stats = 0) + Shorelines = self.Read_MapGen(os.path.join("data",'world.dat'),stats = 0) #print "It took %f seconds to load %i shorelines"%(time.clock() - start,len(Shorelines) ) #start = time.clock() for segment in Shorelines: @@ -939,8 +928,8 @@ else: #start = time.clock() self.Canvas.ZoomToBB() #print "It took %f seconds to draw %i shorelines"%(time.clock() - start,len(Shorelines) ) - - + + def LineTest(self,event = None): wx.GetApp().Yield() import os, time @@ -1018,7 +1007,7 @@ else: w = random.randint(1,5) self.Rectangle = Canvas.AddRectangle(x,y,w,h,LineWidth = lw,FillColor = colors[cf]) self.ColorObjectsAll.append(self.Rectangle) - + # Ellipse x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) lw = random.randint(1,5) @@ -1027,7 +1016,7 @@ else: w = random.randint(1,5) self.Ellipse = Canvas.AddEllipse(x,y,h,w,LineWidth = lw,FillColor = colors[cf]) self.ColorObjectsAll.append(self.Ellipse) - + # Point xy = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) D = random.randint(1,50) @@ -1035,7 +1024,7 @@ else: cf = random.randint(0,len(colors)-1) cl = random.randint(0,len(colors)-1) self.ColorObjectsColor.append(Canvas.AddPoint(xy, colors[cf], D)) - + # Circle x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) D = random.randint(1,5) @@ -1054,7 +1043,7 @@ else: cf = random.randint(0,len(colors)-1) cl = random.randint(0,len(colors)-1) self.ColorObjectsLine.append(Canvas.AddLine(points, LineWidth = lw, LineColor = colors[cl])) - + # Polygon ## points = [] ## for j in range(random.randint(2,6)): @@ -1069,21 +1058,21 @@ else: LineColor = colors[cl], FillColor = colors[cf], FillStyle = 'Solid')) - + ## Pointset points = RandomArray.uniform(Range[0],Range[1],(100,2)) cf = random.randint(0,len(colors)-1) D = random.randint(1,4) self.PointSet = Canvas.AddPointSet(points, Color = colors[cf], Diameter = D) self.ColorObjectsColor.append(self.PointSet) - + ## Point point = RandomArray.uniform(Range[0],Range[1],(2,)) cf = random.randint(0,len(colors)-1) D = random.randint(1,4) self.Point = Canvas.AddPoint(point, Color = colors[cf], Diameter = D) self.ColorObjectsColor.append(self.Point) - + # Text String = "Unscaled text" ts = random.randint(10,40) @@ -1136,7 +1125,53 @@ else: x,y = (random.uniform(Range[0],Range[1]),random.uniform(Range[0],Range[1])) w,h = random.randint(1,5), random.randint(1,5) Object.SetShape(x,y,w,h) - + self.Canvas.Draw(Force = True) + + def ArrowTest(self,event=None): + wx.GetApp().Yield() + self.UnBindAllMouseEvents() + Canvas = self.Canvas + + Canvas.ClearAll() + Canvas.SetProjectionFun(None) + + # put in a rectangle to get a bounding box + Canvas.AddRectangle(0,0,20,20,LineColor = None) + + # Draw some Arrows + Canvas.AddArrow((10,10),Length = 40, Direction = 0) + Canvas.AddArrow((10,10),Length = 50, Direction = 45 ,LineWidth = 2, LineColor = "Black", ArrowHeadAngle = 20) + Canvas.AddArrow((10,10),Length = 60, Direction = 90 ,LineWidth = 3, LineColor = "Red", ArrowHeadAngle = 30) + Canvas.AddArrow((10,10),Length = 70, Direction = 135,LineWidth = 4, LineColor = "Red", ArrowHeadAngle = 40) + Canvas.AddArrow((10,10),Length = 80, Direction = 180,LineWidth = 5, LineColor = "Blue", ArrowHeadAngle = 50) + Canvas.AddArrow((10,10),Length = 90, Direction = 225,LineWidth = 4, LineColor = "Blue", ArrowHeadAngle = 60) + Canvas.AddArrow((10,10),Length = 100,Direction = 270,LineWidth = 3, LineColor = "Green", ArrowHeadAngle = 70) + Canvas.AddArrow((10,10),Length = 110,Direction = 315,LineWidth = 2, LineColor = "Green", ArrowHeadAngle = 90 ) + + Canvas.AddText("Clickable Arrow",4,18,Position = "bc") + Arrow = Canvas.AddArrow((4,18), 80, Direction = 90 ,LineWidth = 3, LineColor = "Red", ArrowHeadAngle = 30) + Arrow.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.ArrowClicked) + + Canvas.AddText("Changable Arrow",16,4,Position = "cc") + self.RotArrow = Canvas.AddArrow((16,4), 80, Direction = 0 ,LineWidth = 3, LineColor = "Green", ArrowHeadAngle = 30) + self.RotArrow.Bind(FloatCanvas.EVT_FC_LEFT_DOWN, self.RotateArrow) + + + + Canvas.ZoomToBB() + + def ArrowClicked(self,event): + print "The Arrow was Clicked" + + def RotateArrow(self,event): + print "The Changeable Arrow was Clicked" + ## You can do them either one at a time, or both at once + ## Doing them both at once prevents the arrow points from being calculated twice + #self.RotArrow.SetDirection(self.RotArrow.Direction + random.uniform(-90,90)) + #self.RotArrow.SetLength(self.RotArrow.Length + random.randint(-20,20)) + self.RotArrow.SetLengthDirection(self.RotArrow.Length + random.randint(-20,20), + self.RotArrow.Direction + random.uniform(-90,90) ) + self.Canvas.Draw(Force = True) def TempTest(self, event= None): @@ -1165,7 +1200,7 @@ else: self.SelectedPoly = None self.SelectPoints = [] self.SelectedPoint = None - + Canvas.ZoomToBB() def SelectPoly(self, Object): @@ -1191,65 +1226,154 @@ else: print "Point Num: %i Hit"%Point.VerticeNum self.SelectedPoint = Point - - + def Read_MapGen(self, filename, stats = 0,AllLines=0): + """ + This function reads a MapGen Format file, and + returns a list of NumPy arrays with the line segments in them. + + Each NumPy array in the list is an NX2 array of Python Floats. + + The demo should have come with a file, "world.dat" that is the + shorelines of the whole world, in MapGen format. + + """ + import string + file = open(filename,'rt') + data = file.readlines() + data = map(string.strip,data) + + Shorelines = [] + segment = [] + for line in data: + if line: + if line == "# -b": #New segment beginning + if segment: Shorelines.append(Numeric.array(segment)) + segment = [] + else: + segment.append(map(float,string.split(line))) + if segment: Shorelines.append(Numeric.array(segment)) + + if stats: + NumSegments = len(Shorelines) + NumPoints = 0 + for segment in Shorelines: + NumPoints = NumPoints + len(segment) + AvgPoints = NumPoints / NumSegments + print "Number of Segments: ", NumSegments + print "Average Number of Points per segment: ",AvgPoints + if AllLines: + Lines = [] + for segment in Shorelines: + Lines.append(segment[0]) + for point in segment[1:-1]: + Lines.append(point) + Lines.append(point) + Lines.append(segment[-1]) + return Lines + else: + return Shorelines + return DrawFrame + +#--------------------------------------------------------------------------- + +if __name__ == "__main__": + # check options: + import sys, getopt + optlist, args = getopt.getopt(sys.argv[1:],'l',["local","all","text","map","stext","hit","hitf","animate","speed","temp","props","arrow"]) + + if not haveNumeric: + raise ImportError(errorText) + StartUpDemo = "all" # the default + for opt in optlist: + if opt[0] == "--all": + StartUpDemo = "all" + elif opt[0] == "--text": + StartUpDemo = "text" + elif opt[0] == "--map": + StartUpDemo = "map" + elif opt[0] == "--stext": + StartUpDemo = "stext" + elif opt[0] == "--hit": + StartUpDemo = "hit" + elif opt[0] == "--hitf": + StartUpDemo = "hitf" + elif opt[0] == "--animate": + StartUpDemo = "animate" + elif opt[0] == "--speed": + StartUpDemo = "speed" + elif opt[0] == "--temp": + StartUpDemo = "temp" + elif opt[0] == "--props": + StartUpDemo = "props" + elif opt[0] == "--arrow": + StartUpDemo = "arrow" + try: + import fixdc + except ImportError: + print ("\n *************Notice*************\n" + "The fixdc module fixes the DC API for wxPython2.5.1.6 You can get the module from:\n" + "http://prdownloads.sourceforge.net/wxpython/fixdc.py?download\n" + " It will set up the DC methods to match both older and upcoming versions\n") + raise + class DemoApp(wx.App): """ How the demo works: - + Under the Draw menu, there are three options: - + *Draw Test: will put up a picture of a bunch of randomly generated objects, of each kind supported. - + *Draw Map: will draw a map of the world. Be patient, it is a big map, with a lot of data, and will take a while to load and draw (about 10 sec on my 450Mhz PIII). Redraws take about 2 sec. This demonstrates how the performance is not very good for large drawings. - + *Clear: Clears the Canvas. - + Once you have a picture drawn, you can zoom in and out and move about the picture. There is a tool bar with three tools that can be selected. - + The magnifying glass with the plus is the zoom in tool. Once selected, if you click the image, it will zoom in, centered on where you clicked. If you click and drag the mouse, you will get a rubber band box, and the image will zoom to fit that box when you release it. - + The magnifying glass with the minus is the zoom out tool. Once selected, if you click the image, it will zoom out, centered on where you clicked. (note that this takes a while when you are looking at the map, as it has a LOT of lines to be drawn. The image is double buffered, so you don't see the drawing in progress) - + The hand is the move tool. Once selected, if you click and drag on the image, it will move so that the part you clicked on ends up where you release the mouse. Nothing is changed while you are dragging. The drawing is too slow for that. - + I'd like the cursor to change as you change tools, but the stock wxCursors didn't include anything I liked, so I stuck with the pointer. Please let me know if you have any nice cursor images for me to use. - - + + Any bugs, comments, feedback, questions, and especially code are welcome: - + -Chris Barker - + Chris.Barker@noaa.gov - + """ def __init__(self, *args, **kwargs): wx.App.__init__(self, *args, **kwargs) - + def OnInit(self): wx.InitAllImageHandlers() + DrawFrame = BuildDrawFrame() frame = DrawFrame(None, -1, "FloatCanvas Demo App",wx.DefaultPosition,(700,700)) - + self.SetTopWindow(frame) frame.Show() @@ -1279,84 +1403,72 @@ else: elif StartUpDemo == "props": "starting PropertiesChange Test" frame.PropertiesChangeTest() - - return True - - def Read_MapGen(filename,stats = 0,AllLines=0): - """ - This function reads a MapGen Format file, and - returns a list of NumPy arrays with the line segments in them. - - Each NumPy array in the list is an NX2 array of Python Floats. - - The demo should have come with a file, "world.dat" that is the - shorelines of the whole world, in MapGen format. - - """ - import string - file = open(filename,'rt') - data = file.readlines() - data = map(string.strip,data) - - Shorelines = [] - segment = [] - for line in data: - if line: - if line == "# -b": #New segment beginning - if segment: Shorelines.append(Numeric.array(segment)) - segment = [] - else: - segment.append(map(float,string.split(line))) - if segment: Shorelines.append(Numeric.array(segment)) - - if stats: - NumSegments = len(Shorelines) - NumPoints = 0 - for segment in Shorelines: - NumPoints = NumPoints + len(segment) - AvgPoints = NumPoints / NumSegments - print "Number of Segments: ", NumSegments - print "Average Number of Points per segment: ",AvgPoints - if AllLines: - Lines = [] - for segment in Shorelines: - Lines.append(segment[0]) - for point in segment[1:-1]: - Lines.append(point) - Lines.append(point) - Lines.append(segment[-1]) - return Lines - else: - return Shorelines - - ## for the wxPython demo: - try: - import floatcanvas - except ImportError: # if it's not there locally, try the wxPython lib. - from wx.lib import floatcanvas - - overview = floatcanvas.__doc__ - -if __name__ == "__main__": - if not haveNumeric: - print errorText - else: - app = DemoApp(False)# put in True if you want output to go to it's own window. - app.MainLoop() - - - - - - - - + elif StartUpDemo == "arrow": + "starting arrow Test" + frame.ArrowTest() + return True + app = DemoApp(False)# put in True if you want output to go to it's own window. + app.MainLoop() +else: + # It's not running stand-alone, set up for wxPython demo. + if not haveNumeric: + ## TestPanel and runTest used for integration into wxPython Demo + class TestPanel(wx.Panel): + def __init__(self, parent, log): + self.log = log + wx.Panel.__init__(self, parent, -1) + + import images + + note1 = wx.StaticText(self, -1, errorText) + note2 = wx.StaticText(self, -1, "This is what the FloatCanvas can look like:") + S = wx.BoxSizer(wx.VERTICAL) + S.Add((10, 10), 1) + S.Add(note1, 0, wx.ALIGN_CENTER) + S.Add(note2, 0, wx.ALIGN_CENTER | wx.BOTTOM, 4) + S.Add(wx.StaticBitmap(self,-1,images.getFloatCanvasBitmap()),0,wx.ALIGN_CENTER) + S.Add((10, 10), 1) + self.SetSizer(S) + self.Layout() + else: + ## TestPanel and runTest used for integration into wxPython Demo + class TestPanel(wx.Panel): + def __init__(self, parent, log): + self.log = log + wx.Panel.__init__(self, parent, -1) + note1 = wx.StaticText(self, -1, "The FloatCanvas Demo needs") + note2 = wx.StaticText(self, -1, "a separate frame") + b = wx.Button(self, -1, "Open Demo Frame Now") + b.Bind(wx.EVT_BUTTON, self.OnButton) + + S = wx.BoxSizer(wx.VERTICAL) + S.Add((10, 10), 1) + S.Add(note1, 0, wx.ALIGN_CENTER) + S.Add(note2, 0, wx.ALIGN_CENTER | wx.BOTTOM, 5) + S.Add(b, 0, wx.ALIGN_CENTER | wx.ALL, 5) + S.Add((10, 10), 1) + self.SetSizer(S) + self.Layout() + + def OnButton(self, evt): + DrawFrame = BuildDrawFrame() + frame = DrawFrame(None, -1, "FloatCanvas Drawing Window",wx.DefaultPosition,(500,500)) + + #win = wx.lib.plot.TestFrame(self, -1, "PlotCanvas Demo") + frame.Show() + frame.DrawTest() + def runTest(frame, nb, log): + win = TestPanel(nb, log) + return win + # import to get the doc + from wx.lib import floatcanvas + overview = floatcanvas.__doc__