1 from Numeric
import array
,Float
,cos
,pi
,sum,minimum
,maximum
,Int32
3 from time
import clock
, sleep
4 from wxPython
.wx
import *
8 ID_ZOOM_IN_BUTTON
= wxNewId()
9 ID_ZOOM_OUT_BUTTON
= wxNewId()
10 ID_ZOOM_TO_FIT_BUTTON
= wxNewId()
11 ID_MOVE_MODE_BUTTON
= wxNewId()
12 ID_TEST_BUTTON
= wxNewId()
14 ID_ABOUT_MENU
= wxNewId()
15 ID_EXIT_MENU
= wxNewId()
16 ID_ZOOM_TO_FIT_MENU
= wxNewId()
17 ID_DRAWTEST_MENU
= wxNewId()
18 ID_DRAWMAP_MENU
= wxNewId()
19 ID_CLEAR_MENU
= wxNewId()
24 ### These are some functions for bitmaps of icons.
28 return cPickle
.loads(zlib
.decompress(
29 'x\xda\xd3\xc8)0\xe4\nV72T\x00!\x05Cu\xae\xc4`u=\x85d\x05\xa7\x9c\xc4\xe4l0O\
30 \x01\xc8S\xb6t\x06A(\x1f\x0b\xa0\xa9\x8c\x9e\x1e6\x19\xa0\xa8\x1e\x88\xd4C\
31 \x97\xd1\x83\xe8\x80 \x9c2zh\xa6\xc1\x11X\n\xab\x8c\x02\x8a\x0cD!\x92\x12\
32 \x98\x8c\x1e\x8a\x8b\xd1d\x14\xf4\x90%\x90LC\xf6\xbf\x1e\xba\xab\x91%\xd0\
33 \xdc\x86C\x06\xd9m\xe8!\xaa\x87S\x86\x1a1\xa7\x07\x00v\x0f[\x17' ))
36 return wxBitmapFromXPMData(GetHandData())
38 #----------------------------------------------------------------------
40 return cPickle
.loads(zlib
.decompress(
41 'x\xda\xd3\xc8)0\xe4\nV72T\x00!\x05Cu\xae\xc4`u=\x85d\x05\xa7\x9c\xc4\xe4l0O\
42 \x01\xc8S\xb6t\x06A(\x1f\x0b RF\x0f\x08\xb0\xc9@D\xe1r\x08\x19\xb8j=l2`\r\
43 \xe82HF\xe9a\xc8\xe8\xe9A\x9c@\x8a\x0c\x0e\xd3p\xbb\x00\x8f\xab\xe1>\xd5\xd3\
44 \xc3\x15:P)l!\n\x91\xc2\x1a\xd6`)\xec\xb1\x00\x92\xc2\x11?\xb8e\x88\x8fSt\
45 \x19=\x00\x82\x16[\xf7' ))
48 return wxBitmapFromXPMData(GetPlusData())
50 #----------------------------------------------------------------------
52 return cPickle
.loads(zlib
.decompress(
53 'x\xda\xd3\xc8)0\xe4\nV72T\x00!\x05Cu\xae\xc4`u=\x85d\x05\xa7\x9c\xc4\xe4l0O\
54 \x01\xc8S\xb6t\x06A(\x1f\x0b RF\x0f\x08\xb0\xc9@D\xe1r\x08\x19\xb8j=\xa2e\
55 \x10\x16@\x99\xc82zz\x10\'\x90"\x83\xc34r\xdc\x86\xf0\xa9\x9e\x1e\xae\xd0\
56 \x81Ja\x0bQ\x88\x14\xd6\xb0\x06Ka\x8f\x05\x90\x14\x8e\xf8\xc1-C|\x9c\xa2\xcb\
57 \xe8\x01\x00\xed\x0f[\x87' ))
60 return wxBitmapFromXPMData(GetMinusData())
62 ## This is a bunch of stuff for implimenting interactive use: catching
63 ## when objects are clicked on by the mouse, etc. I've made a start, so if
64 ## you are interesed in making that work, let me know, I may have gotten
67 #### I probably want a full set of events someday:
68 ## #### mouse over, right click, left click mouse up etc, etc.
69 ## ##FLOATCANVAS_EVENT_LEFT_DOWN = wxNewEventType()
70 ## ##FLOATCANVAS_EVENT_LEFT_UP = wxNewEventType()
71 ## ##FLOATCANVAS_EVENT_RIGHT_DOWN = wxNewEventType()
72 ## ##FLOATCANVAS_EVENT_RIGHT_UP = wxNewEventType()
73 ## ##FLOATCANVAS_EVENT_MOUSE_OVER = wxNewEventType()
75 ##WXFLOATCANVASEVENT = wxNewEventType()
77 ##def EVT_WXFLOATCANVASEVENT( window, function ):
79 ## """Your documentation here"""
81 ## window.Connect( -1, -1, WXFLOATCANVASEVENT, function )
83 ##class wxFloatCanvasObjectEvent(wxPyCommandEvent):
84 ## def __init__(self, WindowID,Object):
85 ## wxPyCommandEvent.__init__(self, WXFLOATCANVASEVENT, WindowID)
86 ## self.Object = Object
89 ## self.__class__( self.GetId() )
91 ##class ColorGenerator:
93 ## """ An instance of this class generates a unique color each time
94 ## GetNextColor() is called. Someday I will use a proper Python
95 ## generator for this class.
97 ## The point of this generator is for the hit-test bitmap, each object
98 ## needs to be a unique color. Also, each system can be running a
99 ## different number of colors, and it doesn't appear to be possible to
100 ## have a wxMemDC with a different colordepth as the screen so this
101 ## generates colors far enough apart that they can be distinguished on
102 ## a 16bit screen. Anything less than 16bits won't work.
105 ## def __init__(self,depth = 16):
114 ## raise "ColorGenerator does not work with depth = %s"%depth
116 ## def GetNextColor(self):
118 ## ##r,g,b = self.r,self.g,self.b
127 ## ## fixme: this should be a derived exception
128 ## raise "Too many objects for HitTest"
129 ## return (self.r,self.g,self.b)
134 This is the base class for all the objects that can be drawn.
136 each object has the following properties; (incomplete)
138 BoundingBox : is of the form: array((min_x,min_y),(max_x,max_y))
144 def __init__(self
,Foreground
= 0):
145 self
.Foreground
= Foreground
149 # I pre-define all these as class variables to provide an easier
150 # interface, and perhaps speed things up by caching all the Pens
151 # and Brushes, although that may not help, as I think wx now
152 # does that on it's own. Send me a note if you know!
155 ( None,"Transparent") : wxTRANSPARENT_BRUSH
,
156 ("Blue","Solid") : wxBLUE_BRUSH
,
157 ("Green","Solid") : wxGREEN_BRUSH
,
158 ("White","Solid") : wxWHITE_BRUSH
,
159 ("Black","Solid") : wxBLACK_BRUSH
,
160 ("Grey","Solid") : wxGREY_BRUSH
,
161 ("MediumGrey","Solid") : wxMEDIUM_GREY_BRUSH
,
162 ("LightGrey","Solid") : wxLIGHT_GREY_BRUSH
,
163 ("Cyan","Solid") : wxCYAN_BRUSH
,
164 ("Red","Solid") : wxRED_BRUSH
167 (None,"Transparent",1) : wxTRANSPARENT_PEN
,
168 ("Green","Solid",1) : wxGREEN_PEN
,
169 ("White","Solid",1) : wxWHITE_PEN
,
170 ("Black","Solid",1) : wxBLACK_PEN
,
171 ("Grey","Solid",1) : wxGREY_PEN
,
172 ("MediumGrey","Solid",1) : wxMEDIUM_GREY_PEN
,
173 ("LightGrey","Solid",1) : wxLIGHT_GREY_PEN
,
174 ("Cyan","Solid",1) : wxCYAN_PEN
,
175 ("Red","Solid",1) : wxRED_PEN
179 "Transparent" : wxTRANSPARENT
,
181 "BiDiagonalHatch": wxBDIAGONAL_HATCH
,
182 "CrossDiagHatch" : wxCROSSDIAG_HATCH
,
183 "FDiagonal_Hatch": wxFDIAGONAL_HATCH
,
184 "CrossHatch" : wxCROSS_HATCH
,
185 "HorizontalHatch": wxHORIZONTAL_HATCH
,
186 "VerticalHatch" : wxVERTICAL_HATCH
191 "Transparent": wxTRANSPARENT
,
193 "LongDash" : wxLONG_DASH
,
194 "ShortDash" : wxSHORT_DASH
,
195 "DotDash" : wxDOT_DASH
,
198 def SetBrush(self
,FillColor
,FillStyle
):
199 if FillColor
is None or FillStyle
is None:
200 self
.Brush
= wxTRANSPARENT_BRUSH
201 self
.FillStyle
= "Transparent"
203 if not self
.BrushList
.has_key((FillColor
,FillStyle
)):
204 self
.BrushList
[(FillColor
,FillStyle
)] = wxBrush(FillColor
,self
.FillStyleList
[FillStyle
])
205 self
.Brush
= self
.BrushList
[(FillColor
,FillStyle
)]
207 def SetPen(self
,LineColor
,LineStyle
,LineWidth
):
208 if (LineColor
is None) or (LineStyle
is None):
209 self
.Pen
= wxTRANSPARENT_PEN
210 self
.LineStyle
= 'Transparent'
212 if not self
.PenList
.has_key((LineColor
,LineStyle
,LineWidth
)):
213 self
.PenList
[(LineColor
,LineStyle
,LineWidth
)] = wxPen(LineColor
,LineWidth
,self
.LineStyleList
[LineStyle
])
214 self
.Pen
= self
.PenList
[(LineColor
,LineStyle
,LineWidth
)]
216 def SetPens(self
,LineColors
,LineStyles
,LineWidths
):
218 This method used when an object could have a list of pens, rather than just one
219 It is used for LineSet, and perhaps others in the future.
221 fixme: this is really kludgy, there has got to be a better way!
226 if type(LineColors
) == types
.ListType
:
227 length
= len(LineColors
)
229 LineColors
= [LineColors
]
231 if type(LineStyles
) == types
.ListType
:
232 length
= len(LineStyles
)
234 LineStyles
= [LineStyles
]
236 if type(LineWidths
) == types
.ListType
:
237 length
= len(LineWidths
)
239 LineWidths
= [LineWidths
]
242 if len(LineColors
) == 1:
243 LineColors
= LineColors
*length
244 if len(LineStyles
) == 1:
245 LineStyles
= LineStyles
*length
246 if len(LineWidths
) == 1:
247 LineWidths
= LineWidths
*length
250 for (LineColor
,LineStyle
,LineWidth
) in zip(LineColors
,LineStyles
,LineWidths
):
251 if LineColor
is None or LineStyle
is None:
252 self
.Pens
.append(wxTRANSPARENT_PEN
)
253 # what's this for?> self.LineStyle = 'Transparent'
254 if not self
.PenList
.has_key((LineColor
,LineStyle
,LineWidth
)):
255 Pen
= wxPen(LineColor
,LineWidth
,self
.LineStyleList
[LineStyle
])
256 self
.Pens
.append(Pen
)
258 self
.Pens
.append(self
.PenList
[(LineColor
,LineStyle
,LineWidth
)])
260 self
.Pens
= self
.Pens
[0]
262 def PutInBackground(self
):
263 if self
._Canvas
and self
.Foreground
:
264 self
._Canvas
._TopDrawList
.remove(self
)
265 self
._Canvas
._DrawList
.append(self
)
266 self
._Canvas
._BackgroundDirty
= 1
269 def PutInForeground(self
):
270 if self
._Canvas
and (not self
.Foreground
):
271 self
._Canvas
._TopDrawList
.append(self
)
272 self
._Canvas
._DrawList
.remove(self
)
273 self
._Canvas
._BackgroundDirty
= 1
277 class Polygon(draw_object
):
281 The Polygon class takes a list of 2-tuples, or a NX2 NumPy array of
282 point coordinates. so that Points[N][0] is the x-coordinate of
283 point N and Points[N][1] is the y-coordinate or Points[N,0] is the
284 x-coordinate of point N and Points[N,1] is the y-coordinate for
288 def __init__(self
,Points
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
= 0):
289 draw_object
.__init
__(self
,Foreground
)
290 self
.Points
= array(Points
,Float
)
291 self
.BoundingBox
= array(((min(self
.Points
[:,0]),min(self
.Points
[:,1])),(max(self
.Points
[:,0]),max(self
.Points
[:,1]))),Float
)
293 self
.LineColor
= LineColor
294 self
.LineStyle
= LineStyle
295 self
.LineWidth
= LineWidth
296 self
.FillColor
= FillColor
297 self
.FillStyle
= FillStyle
299 self
.SetPen(LineColor
,LineStyle
,LineWidth
)
300 self
.SetBrush(FillColor
,FillStyle
)
303 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
304 Points
= WorldToPixel(self
.Points
)
306 dc
.SetBrush(self
.Brush
)
307 #dc.DrawPolygon(map(lambda x: (x[0],x[1]), Points.tolist()))
308 dc
.DrawPolygon(Points
)
310 class PolygonSet(draw_object
):
312 The PolygonSet class takes a Geometry.Polygon object.
313 so that Points[N] = (x1,y1) and Points[N+1] = (x2,y2). N must be an even number!
315 it creates a set of line segments, from (x1,y1) to (x2,y2)
319 def __init__(self
,PolySet
,LineColors
,LineStyles
,LineWidths
,FillColors
,FillStyles
,Foreground
= 0):
320 draw_object
.__init
__(self
, Foreground
)
322 ##fixme: there should be some error checking for everything being the right length.
325 self
.Points
= array(Points
,Float
)
326 self
.BoundingBox
= array(((min(self
.Points
[:,0]),min(self
.Points
[:,1])),(max(self
.Points
[:,0]),max(self
.Points
[:,1]))),Float
)
328 self
.LineColors
= LineColors
329 self
.LineStyles
= LineStyles
330 self
.LineWidths
= LineWidths
331 self
.FillColors
= FillColors
332 self
.FillStyles
= FillStyles
334 self
.SetPens(LineColors
,LineStyles
,LineWidths
)
336 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
337 Points
= WorldToPixel(self
.Points
)
338 Points
.shape
= (-1,4)
339 dc
.DrawLineList(Points
,self
.Pens
)
342 class Line(draw_object
):
344 The Line class takes a list of 2-tuples, or a NX2 NumPy array of point coordinates.
345 so that Points[N][0] is the x-coordinate of point N and Points[N][1] is the y-coordinate
346 or Points[N,0] is the x-coordinate of point N and Points[N,1] is the y-coordinate for arrays.
348 It will draw a straight line if there are two points, and a polyline if there are more than two.
351 def __init__(self
,Points
,LineColor
,LineStyle
,LineWidth
,Foreground
= 0):
352 draw_object
.__init
__(self
, Foreground
)
354 self
.Points
= array(Points
,Float
)
355 self
.BoundingBox
= array(((min(self
.Points
[:,0]),min(self
.Points
[:,1])),(max(self
.Points
[:,0]),max(self
.Points
[:,1]))),Float
)
357 self
.LineColor
= LineColor
358 self
.LineStyle
= LineStyle
359 self
.LineWidth
= LineWidth
361 self
.SetPen(LineColor
,LineStyle
,LineWidth
)
363 def SetPoints(self
,Points
):
365 self
.BoundingBox
= array(((min(self
.Points
[:,0]),min(self
.Points
[:,1])),(max(self
.Points
[:,0]),max(self
.Points
[:,1]))),Float
)
367 # It looks like this shouldn't be private
368 self
._Canvas
.BoundingBoxDirty
= 1
370 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
371 Points
= WorldToPixel(self
.Points
)
373 #dc.DrawLines(map(lambda x: (x[0],x[1]), Points.tolist()))
377 class LineSet(draw_object
):
379 The LineSet class takes a list of 2-tuples, or a NX2 NumPy array of point coordinates.
380 so that Points[N] = (x1,y1) and Points[N+1] = (x2,y2). N must be an even number!
382 it creates a set of line segments, from (x1,y1) to (x2,y2)
386 def __init__(self
,Points
,LineColors
,LineStyles
,LineWidths
,Foreground
= 0):
387 draw_object
.__init
__(self
, Foreground
)
389 NumLines
= len(Points
) / 2
390 ##fixme: there should be some error checking for everything being the right length.
393 self
.Points
= array(Points
,Float
)
394 self
.BoundingBox
= array(((min(self
.Points
[:,0]),min(self
.Points
[:,1])),(max(self
.Points
[:,0]),max(self
.Points
[:,1]))),Float
)
396 self
.LineColors
= LineColors
397 self
.LineStyles
= LineStyles
398 self
.LineWidths
= LineWidths
400 self
.SetPens(LineColors
,LineStyles
,LineWidths
)
402 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
403 Points
= WorldToPixel(self
.Points
)
404 Points
.shape
= (-1,4)
405 dc
.DrawLineList(Points
,self
.Pens
)
408 class PointSet(draw_object
):
410 The PointSet class takes a list of 2-tuples, or a NX2 NumPy array of point coordinates.
411 so that Points[N][0] is the x-coordinate of point N and Points[N][1] is the y-coordinate
412 or Points[N,0] is the x-coordinate of point N and Points[N,1] is the y-coordinate for arrays.
414 Each point will be drawn the same color and Diameter. The Diameter is in screen points,
415 not world coordinates.
418 def __init__(self
,Points
,Color
,Diameter
,Foreground
= 0):
419 draw_object
.__init
__(self
,Foreground
)
421 self
.Points
= array(Points
,Float
)
422 self
.Points
.shape
= (-1,2) # Make sure it is a NX2 array, even if there is only one point
423 self
.BoundingBox
= array(((min(self
.Points
[:,0]),min(self
.Points
[:,1])),(max(self
.Points
[:,0]),max(self
.Points
[:,1]))),Float
)
426 self
.Diameter
= Diameter
428 self
.SetPen(Color
,"Solid",1)
429 self
.SetBrush(Color
,"Solid")
431 def SetPoints(self
,Points
):
433 self
.BoundingBox
= array(((min(self
.Points
[:,0]),min(self
.Points
[:,1])),(max(self
.Points
[:,0]),max(self
.Points
[:,1]))),Float
)
435 # It looks like this shouldn't be private
436 self
._Canvas
.BoundingBoxDirty
= 1
438 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
440 Points
= WorldToPixel(self
.Points
)
441 if self
.Diameter
<= 1:
442 dc
.DrawPointList(Points
)
443 elif self
.Diameter
<= 2:
444 # A Little optimization for a diameter2 - point
445 dc
.DrawPointList(Points
)
446 dc
.DrawPointList(Points
+ (1,0))
447 dc
.DrawPointList(Points
+ (0,1))
448 dc
.DrawPointList(Points
+ (1,1))
450 dc
.SetBrush(self
.Brush
)
451 radius
= int(round(self
.Diameter
/2))
453 dc
.DrawEllipse((x
- radius
), (y
- radius
), self
.Diameter
, self
.Diameter
)
457 class Dot(draw_object
):
459 The Dot class takes an x.y coordinate pair, and the Diameter of the circle.
460 The Diameter is in pixels, so it won't change with zoom.
462 Also Fill and line data
465 def __init__(self
,x
,y
,Diameter
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
= 0):
466 draw_object
.__init
__(self
,Foreground
)
470 self
.Diameter
= Diameter
471 # NOTE: the bounding box does not include the diameter of the dot, as that is in pixel coords.
472 # If this is a problem, perhaps you should use a circle, instead!
473 self
.BoundingBox
= array(((x
,y
),(x
,y
)),Float
)
475 self
.LineColor
= LineColor
476 self
.LineStyle
= LineStyle
477 self
.LineWidth
= LineWidth
478 self
.FillColor
= FillColor
479 self
.FillStyle
= FillStyle
481 self
.SetPen(LineColor
,LineStyle
,LineWidth
)
482 self
.SetBrush(FillColor
,FillStyle
)
484 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
486 dc
.SetBrush(self
.Brush
)
487 radius
= int(round(self
.Diameter
/2))
488 (X
,Y
) = WorldToPixel((self
.X
,self
.Y
))
489 dc
.DrawEllipse((X
- radius
), (Y
- radius
), self
.Diameter
, self
.Diameter
)
492 class Rectangle(draw_object
):
493 def __init__(self
,x
,y
,width
,height
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
= 0):
494 draw_object
.__init
__(self
,Foreground
)
500 self
.BoundingBox
= array(((x
,y
),(x
+width
,y
+height
)),Float
)
501 self
.LineColor
= LineColor
502 self
.LineStyle
= LineStyle
503 self
.LineWidth
= LineWidth
504 self
.FillColor
= FillColor
505 self
.FillStyle
= FillStyle
507 self
.SetPen(LineColor
,LineStyle
,LineWidth
)
508 self
.SetBrush(FillColor
,FillStyle
)
510 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
511 (X
,Y
) = WorldToPixel((self
.X
,self
.Y
))
512 (Width
,Height
) = ScaleFunction((self
.Width
,self
.Height
))
515 dc
.SetBrush(self
.Brush
)
516 dc
.DrawRectangle(X
,Y
,Width
,Height
)
518 class Ellipse(draw_object
):
519 def __init__(self
,x
,y
,width
,height
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
= 0):
520 draw_object
.__init
__(self
,Foreground
)
526 self
.BoundingBox
= array(((x
,y
),(x
+width
,y
+height
)),Float
)
527 self
.LineColor
= LineColor
528 self
.LineStyle
= LineStyle
529 self
.LineWidth
= LineWidth
530 self
.FillColor
= FillColor
531 self
.FillStyle
= FillStyle
533 self
.SetPen(LineColor
,LineStyle
,LineWidth
)
534 self
.SetBrush(FillColor
,FillStyle
)
536 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
537 (X
,Y
) = WorldToPixel((self
.X
,self
.Y
))
538 (Width
,Height
) = ScaleFunction((self
.Width
,self
.Height
))
541 dc
.SetBrush(self
.Brush
)
542 dc
.DrawEllipse(X
,Y
,Width
,Height
)
544 class Circle(draw_object
):
545 def __init__(self
,x
,y
,Diameter
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
= 0):
546 draw_object
.__init
__(self
,Foreground
)
550 self
.Diameter
= Diameter
551 self
.BoundingBox
= array(((x
-Diameter
/2,y
-Diameter
/2),(x
+Diameter
/2,y
+Diameter
/2)),Float
)
552 self
.LineColor
= LineColor
553 self
.LineStyle
= LineStyle
554 self
.LineWidth
= LineWidth
555 self
.FillColor
= FillColor
556 self
.FillStyle
= FillStyle
558 self
.SetPen(LineColor
,LineStyle
,LineWidth
)
559 self
.SetBrush(FillColor
,FillStyle
)
561 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
562 (X
,Y
) = WorldToPixel((self
.X
,self
.Y
))
563 (Diameter
,dummy
) = ScaleFunction((self
.Diameter
,self
.Diameter
))
566 dc
.SetBrush(self
.Brush
)
567 dc
.DrawEllipse(X
-Diameter
/2,Y
-Diameter
/2,Diameter
,Diameter
)
569 class Text(draw_object
):
572 This class creates a text object, placed at the coordinates,
573 x,y. the "Position" argument is a two charactor string, indicating
574 where in relation to the coordinates the string should be oriented.
576 The first letter is: t, c, or b, for top, center and bottom
577 The second letter is: l, c, or r, for left, center and right
579 I've provided arguments for Family, Style, and Weight of font, but
580 have not yet implimented them, so all text is: wxSWISS, wxNORMAL, wxNORMAL.
581 I'd love it if someone would impliment that!
583 The size is fixed, and does not scale with the drawing.
587 def __init__(self
,String
,x
,y
,Size
,ForeGround
,BackGround
,Family
,Style
,Weight
,Underline
,Position
,Foreground
= 0):
588 draw_object
.__init
__(self
,Foreground
)
593 self
.ForeGround
= ForeGround
594 if BackGround
is None:
595 self
.BackGround
= None
597 self
.BackGround
= BackGround
601 self
.Underline
= Underline
602 self
.Position
= Position
603 # fixme: this should use the passed in parameters!
604 self
.Font
= wxFont(Size
, wxMODERN
, wxNORMAL
, wxNORMAL
, Underline
, )
606 self
.BoundingBox
= array(((x
,y
),(x
,y
)),Float
)
613 def _Draw(self
,dc
,WorldToPixel
,ScaleFunction
):
614 (X
,Y
) = WorldToPixel((self
.X
,self
.Y
))
615 dc
.SetFont(self
.Font
)
616 dc
.SetTextForeground(self
.ForeGround
)
618 dc
.SetBackgroundMode(wxSOLID
)
619 dc
.SetTextBackground(self
.BackGround
)
621 dc
.SetBackgroundMode(wxTRANSPARENT
)
623 # compute the shift, and adjust the coordinates, if neccesary
624 # This had to be put in here, becsuse there is no wxDC during __init__
625 if self
.x_shift
is None or self
.y_shift
is None:
626 if self
.Position
== 'tl':
627 x_shift
,y_shift
= 0,0
629 (w
,h
) = dc
.GetTextExtent(self
.String
)
630 if self
.Position
[0] == 't':
632 elif self
.Position
[0] == 'c':
634 elif self
.Position
[0] == 'b':
637 ##fixme: this should be a real derived exception
638 raise "Invalid value for Text Object Position Attribute"
639 if self
.Position
[1] == 'l':
641 elif self
.Position
[1] == 'c':
643 elif self
.Position
[1] == 'r':
646 ##fixme: this should be a real derived exception
647 raise "Invalid value for Text Object Position Attribute"
648 self
.x_shift
= x_shift
649 self
.y_shift
= y_shift
650 dc
.DrawText(self
.String
, X
-self
.x_shift
, Y
-self
.y_shift
)
653 #---------------------------------------------------------------------------
655 class FloatCanvas(wxPanel
):
659 This is a high level window for drawing maps and anything else in an
660 arbitrary coordinate system.
662 The goal is to provide a convenient way to draw stuff on the screen
663 without having to deal with handling OnPaint events, converting to pixel
664 coordinates, knowing about wxWindows brushes, pens, and colors, etc. It
665 also provides virtually unlimited zooming and scrolling
667 I am using it for two things:
668 1) general purpose drawing in floating point coordinates
669 2) displaying map data in Lat-long coordinates
671 If the projection is set to None, it will draw in general purpose
672 floating point coordinates. If the projection is set to 'FlatEarth', it
673 will draw a FlatEarth projection, centered on the part of the map that
674 you are viewing. You can also pass in your own projection function.
676 It is double buffered, so re-draws after the window is uncovered by something
679 It relies on NumPy, which is needed for speed
681 Bugs and Limitations:
682 Lots: patches, fixes welcome
684 For Map drawing: It ignores the fact that the world is, in fact, a
685 sphere, so it will do strange things if you are looking at stuff near
686 the poles or the date line. so far I don't have a need to do that, so I
687 havn't bothered to add any checks for that yet.
690 I have set no zoom limits. What this means is that if you zoom in really
691 far, you can get integer overflows, and get wierd results. It
692 doesn't seem to actually cause any problems other than wierd output, at
693 least when I have run it.
696 I have done a couple of things to improve speed in this app. The one
697 thing I have done is used NumPy Arrays to store the coordinates of the
698 points of the objects. This allowed me to use array oriented functions
699 when doing transformations, and should provide some speed improvement
700 for objects with a lot of points (big polygons, polylines, pointsets).
702 The real slowdown comes when you have to draw a lot of objects, because
703 you have to call the wxDC.DrawSomething call each time. This is plenty
704 fast for tens of objects, OK for hundreds of objects, but pretty darn
705 slow for thousands of objects.
707 The solution is to be able to pass some sort of object set to the DC
708 directly. I've used DC.DrawPointList(Points), and it helped a lot with
709 drawing lots of points. I havn't got a LineSet type object, so I havn't
710 used DC.DrawLineList yet. I'd like to get a full set of DrawStuffList()
711 methods implimented, and then I'd also have a full set of Object sets
712 that could take advantage of them. I hope to get to it some day.
714 Copyright: Christopher Barker
716 License: Same as wxPython
718 Please let me know if you're using this!!!
722 Chris.Barker@noaa.gov
726 def __init__(self
, parent
, id = -1,
727 size
= wxDefaultSize
,
728 ProjectionFun
= None,
729 BackgroundColor
= "WHITE",
731 EnclosingFrame
= None,
736 wxPanel
.__init
__( self
, parent
, id, wxDefaultPosition
, size
)
738 if ProjectionFun
== 'FlatEarth':
739 self
.ProjectionFun
= self
.FlatEarthProjection
740 elif type(ProjectionFun
) == types
.FunctionType
:
741 self
.ProjectionFun
= ProjectionFun
742 elif ProjectionFun
is None:
743 self
.ProjectionFun
= lambda x
=None: array( (1,1), Float
)
745 raise('Projectionfun must be either: "FlatEarth", None, or a function that takes the ViewPortCenter and returns a MapProjectionVector')
747 self
.UseBackground
= UseBackground
748 self
.UseHitTest
= UseHitTest
750 self
.NumBetweenBlits
= 40
752 ## you can have a toolbar with buttons for zoom-in, zoom-out and
753 ## move. If you don't use the toolbar, you should provide your
754 ## own way of navigating the canvas
756 ## Create the vertical sizer for the toolbar and Panel
757 box
= wxBoxSizer(wxVERTICAL
)
758 box
.Add(self
.BuildToolbar(), 0, wxALL | wxALIGN_LEFT | wxGROW
, 4)
760 self
.DrawPanel
= wxWindow(self
,-1,wxDefaultPosition
,wxDefaultSize
,wxSUNKEN_BORDER
)
761 box
.Add(self
.DrawPanel
,1,wxGROW
)
764 self
.SetAutoLayout(True)
767 self
.DrawPanel
= self
769 self
.DrawPanel
.BackgroundBrush
= wxBrush(BackgroundColor
,wxSOLID
)
773 self
.EnclosingFrame
= EnclosingFrame
775 EVT_PAINT(self
.DrawPanel
, self
.OnPaint
)
776 EVT_SIZE(self
.DrawPanel
, self
.OnSize
)
778 EVT_LEFT_DOWN(self
.DrawPanel
, self
.LeftButtonEvent
)
779 EVT_LEFT_UP(self
.DrawPanel
, self
.LeftButtonEvent
)
780 EVT_RIGHT_DOWN(self
.DrawPanel
, self
.RightButtonEvent
)
781 EVT_MOTION(self
.DrawPanel
, self
.LeftButtonEvent
)
785 if self
.UseBackground
:
786 self
._TopDrawList
= []
787 self
.BoundingBox
= None
788 self
.BoundingBoxDirty
= 0
789 self
.ViewPortCenter
= array( (0,0), Float
)
791 self
.MapProjectionVector
= array( (1,1), Float
) # No Projection to start!
792 self
.TransformVector
= array( (1,-1), Float
) # default Transformation
797 self
.StartRBBox
= None
798 self
.PrevRBBox
= None
799 self
.StartMove
= None
800 self
.PrevMoveBox
= None
801 # called just to make sure everything is initialized
805 def BuildToolbar(self
):
806 tb
= wxToolBar(self
,-1)
809 tb
.SetToolBitmapSize((23,23))
811 tb
.AddTool(ID_ZOOM_IN_BUTTON
, GetPlusBitmap(), isToggle
=true
,shortHelpString
= "Zoom In")
812 EVT_TOOL(self
, ID_ZOOM_IN_BUTTON
, self
.SetMode
)
814 tb
.AddTool(ID_ZOOM_OUT_BUTTON
, GetMinusBitmap(), isToggle
=true
,shortHelpString
= "Zoom Out")
815 EVT_TOOL(self
, ID_ZOOM_OUT_BUTTON
, self
.SetMode
)
817 tb
.AddTool(ID_MOVE_MODE_BUTTON
, GetHandBitmap(), isToggle
=true
,shortHelpString
= "Move")
818 EVT_TOOL(self
, ID_MOVE_MODE_BUTTON
, self
.SetMode
)
822 tb
.AddControl(wxButton(tb
, ID_ZOOM_TO_FIT_BUTTON
, "Zoom To Fit",wxDefaultPosition
, wxDefaultSize
))
823 EVT_BUTTON(self
, ID_ZOOM_TO_FIT_BUTTON
, self
.ZoomToFit
)
828 def SetMode(self
,event
):
829 for id in [ID_ZOOM_IN_BUTTON
,ID_ZOOM_OUT_BUTTON
,ID_MOVE_MODE_BUTTON
]:
830 self
.ToolBar
.ToggleTool(id,0)
831 self
.ToolBar
.ToggleTool(event
.GetId(),1)
832 if event
.GetId() == ID_ZOOM_IN_BUTTON
:
833 self
.SetGUIMode("ZoomIn")
834 elif event
.GetId() == ID_ZOOM_OUT_BUTTON
:
835 self
.SetGUIMode("ZoomOut")
836 elif event
.GetId() == ID_MOVE_MODE_BUTTON
:
837 self
.SetGUIMode("Move")
840 def SetGUIMode(self
,Mode
):
841 if Mode
in ["ZoomIn","ZoomOut","Move",None]:
844 raise "Not a valid Mode"
846 def FlatEarthProjection(self
,CenterPoint
):
847 return array((cos(pi
*CenterPoint
[1]/180),1),Float
)
849 def LeftButtonEvent(self
,event
):
850 if self
.EnclosingFrame
:
852 position
= self
.PixelToWorld((event
.GetX(),event
.GetY()))
853 self
.EnclosingFrame
.SetStatusText("%8.3f, %8.3f"%tuple(position
))
855 if self
.GUIMode
== "ZoomIn":
857 self
.StartRBBox
= (event
.GetX(),event
.GetY())
858 self
.PrevRBBox
= None
859 elif event
.Dragging() and event
.LeftIsDown() and self
.StartRBBox
:
860 x0
,y0
= self
.StartRBBox
861 x1
,y1
= event
.GetX(),event
.GetY()
862 w
, h
= abs(x1
-x0
),abs(y1
-y0
)
863 w
= max(w
,int(h
*self
.AspectRatio
))
864 h
= int(w
/self
.AspectRatio
)
865 x_c
, y_c
= (x0
+x1
)/2 , (y0
+y1
)/2
866 dc
= wxClientDC(self
.DrawPanel
)
868 dc
.SetPen(wxPen('WHITE', 2,wxSHORT_DASH
))
869 dc
.SetBrush(wxTRANSPARENT_BRUSH
)
870 dc
.SetLogicalFunction(wxXOR
)
872 dc
.DrawRectangle(*self
.PrevRBBox
)
873 dc
.DrawRectangle(x_c
-w
/2,y_c
-h
/2,w
,h
)
874 self
.PrevRBBox
= (x_c
-w
/2,y_c
-h
/2,w
,h
)
877 elif event
.LeftUp() and self
.StartRBBox
:
878 self
.PrevRBBox
= None
879 EndRBBox
= (event
.GetX(),event
.GetY())
880 StartRBBox
= self
.StartRBBox
881 # if mouse has moved less that ten pixels, don't use the box.
882 if abs(StartRBBox
[0] - EndRBBox
[0]) > 10 and abs(StartRBBox
[1] - EndRBBox
[1]) > 10:
883 EndRBBox
= self
.PixelToWorld(EndRBBox
)
884 StartRBBox
= self
.PixelToWorld(StartRBBox
)
885 BB
= array(((min(EndRBBox
[0],StartRBBox
[0]), min(EndRBBox
[1],StartRBBox
[1])),
886 (max(EndRBBox
[0],StartRBBox
[0]), max(EndRBBox
[1],StartRBBox
[1]))),Float
)
889 Center
= self
.PixelToWorld(StartRBBox
)
890 self
.Zoom(1.5,Center
)
891 self
.StartRBBox
= None
893 if self
.GUIMode
== "ZoomOut":
895 Center
= self
.PixelToWorld((event
.GetX(),event
.GetY()))
896 self
.Zoom(1/1.5,Center
)
897 elif self
.GUIMode
== "Move":
899 self
.StartMove
= array((event
.GetX(),event
.GetY()))
900 self
.PrevMoveBox
= None
901 elif event
.Dragging() and event
.LeftIsDown() and self
.StartMove
:
902 x_1
,y_1
= event
.GetX(),event
.GetY()
903 w
, h
= self
.PanelSize
904 x_tl
, y_tl
= x_1
- self
.StartMove
[0], y_1
- self
.StartMove
[1]
905 dc
= wxClientDC(self
.DrawPanel
)
907 dc
.SetPen(wxPen('WHITE', 1,))
908 dc
.SetBrush(wxTRANSPARENT_BRUSH
)
909 dc
.SetLogicalFunction(wxXOR
)
911 dc
.DrawRectangle(*self
.PrevMoveBox
)
912 dc
.DrawRectangle(x_tl
,y_tl
,w
,h
)
913 self
.PrevMoveBox
= (x_tl
,y_tl
,w
,h
)
916 elif event
.LeftUp() and self
.StartMove
:
917 self
.PrevMoveBox
= None
918 StartMove
= self
.StartMove
919 EndMove
= array((event
.GetX(),event
.GetY()))
920 if sum((StartMove
-EndMove
)**2) > 16:
921 self
.Move(StartMove
-EndMove
,'Pixel')
922 self
.StartMove
= None
924 def RightButtonEvent(self
,event
):
926 if self
.GUIMode
== "ZoomIn":
927 Center
= self
.PixelToWorld((event
.GetX(),event
.GetY()))
928 self
.Zoom(1/1.5,Center
)
929 elif self
.GUIMode
== "ZoomOut":
930 Center
= self
.PixelToWorld((event
.GetX(),event
.GetY()))
931 self
.Zoom(1.5,Center
)
936 def MakeNewBuffers(self
):
937 # Make new offscreen bitmap:
938 self
._Buffer
= wxEmptyBitmap(self
.PanelSize
[0],self
.PanelSize
[1])
939 if self
.UseBackground
:
940 self
._BackBuffer
= wxEmptyBitmap(self
.PanelSize
[0],self
.PanelSize
[1])
941 self
._BackgroundDirty
= 1
945 def OnSize(self
,event
):
946 self
.PanelSize
= array(self
.DrawPanel
.GetClientSizeTuple(),Int32
)
948 self
.AspectRatio
= self
.PanelSize
[0]/self
.PanelSize
[1]
949 except ZeroDivisionError:
950 self
.AspectRatio
= 1.0
951 self
.MakeNewBuffers()
954 def OnPaint(self
, event
):
955 #dc = wxBufferedPaintDC(self.DrawPanel, self._Buffer)
956 dc
= wxPaintDC(self
.DrawPanel
)
957 dc
.DrawBitmap(self
._Buffer
,0,0)
961 The Draw method gets pretty complicated because of all the buffers
963 There is a main buffer set up to double buffer the screen, so
964 you can get quick re-draws when the window gets uncovered.
966 If self.UseBackground is set, and an object is set up with the
967 "ForeGround" flag, then it gets drawn to the screen after blitting
968 the background. This is done so that you can have a complicated
969 background, but have something changing on the foreground,
970 without having to wait for the background to get re-drawn. This
971 can be used to support simple animation, for instance.
974 if self
.Debug
: start
= clock()
975 ScreenDC
= wxClientDC(self
.DrawPanel
)
976 ViewPortWorld
= ( self
.PixelToWorld((0,0)), self
.PixelToWorld(self
.PanelSize
) )
977 ViewPortBB
= array( ( minimum
.reduce(ViewPortWorld
), maximum
.reduce(ViewPortWorld
) ) )
978 if self
.UseBackground
:
980 dc
.SelectObject(self
._BackBuffer
)
981 dc
.SetBackground(self
.DrawPanel
.BackgroundBrush
)
983 if self
._BackgroundDirty
:
987 for Object
in self
._DrawList
:
988 if self
.BBCheck(Object
.BoundingBox
,ViewPortBB
):
989 #print "object is in Bounding Box"
991 Object
._Draw
(dc
,self
.WorldToPixel
,self
.ScaleFunction
)
992 if i
% self
.NumBetweenBlits
== 0:
993 ScreenDC
.Blit(0, 0, self
.PanelSize
[0],self
.PanelSize
[1], dc
, 0, 0)
997 self
._BackgroundDirty
= 0
998 dc
.SelectObject(self
._Buffer
)
1000 ##Draw Background on Main Buffer:
1001 dc
.DrawBitmap(self
._BackBuffer
,0,0)
1002 #Draw the OnTop stuff
1004 for Object
in self
._TopDrawList
:
1006 Object
._Draw
(dc
,self
.WorldToPixel
,self
.ScaleFunction
)
1007 if i
% self
.NumBetweenBlits
== 0:
1008 ScreenDC
.Blit(0, 0, self
.PanelSize
[0],self
.PanelSize
[1], dc
, 0, 0)
1010 else: # not using a Background DC
1012 dc
.SelectObject(self
._Buffer
)
1013 dc
.SetBackground(self
.DrawPanel
.BackgroundBrush
)
1018 for Object
in self
._DrawList
:
1019 if self
.BBCheck(Object
.BoundingBox
,ViewPortBB
):
1020 #print "object is in Bounding Box"
1022 Object
._Draw
(dc
,self
.WorldToPixel
,self
.ScaleFunction
)
1023 if i
% self
.NumBetweenBlits
== 0:
1024 ScreenDC
.Blit(0, 0, self
.PanelSize
[0],self
.PanelSize
[1], dc
, 0, 0)
1028 # now refresh the screen
1029 #ScreenDC.DrawBitmap(self._Buffer,0,0) #NOTE: uisng DrawBitmap didn't work right on MSW
1030 ScreenDC
.Blit(0, 0, self
.PanelSize
[0],self
.PanelSize
[1], dc
, 0, 0)
1032 # If the canvas is in the middle of a zoom or move, the Rubber Band box needs to be re-drawn
1034 ScreenDC
.SetPen(wxPen('WHITE', 2,wxSHORT_DASH
))
1035 ScreenDC
.SetBrush(wxTRANSPARENT_BRUSH
)
1036 ScreenDC
.SetLogicalFunction(wxXOR
)
1037 ScreenDC
.DrawRectangle(*self
.PrevRBBox
)
1038 elif self
.PrevMoveBox
:
1039 ScreenDC
.SetPen(wxPen('WHITE', 1,))
1040 ScreenDC
.SetBrush(wxTRANSPARENT_BRUSH
)
1041 ScreenDC
.SetLogicalFunction(wxXOR
)
1042 ScreenDC
.DrawRectangle(*self
.PrevMoveBox
)
1043 if self
.Debug
: print "Drawing took %f seconds of CPU time"%(clock()-start
)
1045 def BBCheck(self
, BB1
, BB2
):
1048 BBCheck(BB1, BB2) returns True is the Bounding boxes intesect, False otherwise
1051 if ( (BB1
[1,0] > BB2
[0,0]) and (BB1
[0,0] < BB2
[1,0]) and
1052 (BB1
[1,1] > BB2
[0,1]) and (BB1
[0,1] < BB2
[1,1]) ):
1057 def Move(self
,shift
,CoordType
):
1059 move the image in the window.
1061 shift is an (x,y) tuple, specifying the amount to shift in each direction
1063 It can be in any of three coordinates: Panel, Pixel, World,
1064 specified by the CoordType parameter
1066 Panel coordinates means you want to shift the image by some
1067 fraction of the size of the displaed image
1069 Pixel coordinates means you want to shift the image by some number of pixels
1071 World coordinates meand you want to shift the image by an amount
1072 in Floating point world coordinates
1076 shift
= array(shift
,Float
)
1077 if CoordType
== 'Panel':# convert from panel coordinates
1078 shift
= shift
* array((-1,1),Float
) *self
.PanelSize
/self
.TransformVector
1079 elif CoordType
== 'Pixel': # convert from pixel coordinates
1080 shift
= shift
/self
.TransformVector
1081 elif CoordType
== 'World': # No conversion
1084 raise 'CoordType must be either "Panel", "Pixel", or "World"'
1086 self
.ViewPortCenter
= self
.ViewPortCenter
+ shift
1087 self
.MapProjectionVector
= self
.ProjectionFun(self
.ViewPortCenter
)
1088 self
.TransformVector
= array((self
.Scale
,-self
.Scale
),Float
)* self
.MapProjectionVector
1089 self
._BackgroundDirty
= 1
1092 def Zoom(self
,factor
,center
= None):
1095 Zoom(factor, center) changes the amount of zoom of the image by factor.
1096 If factor is greater than one, the image gets larger.
1097 If factor is less than one, the image gets smaller.
1099 Center is a tuple of (x,y) coordinates of the center of the viewport, after zooming.
1100 If center is not given, the center will stay the same.
1103 self
.Scale
= self
.Scale
*factor
1105 self
.ViewPortCenter
= array(center
,Float
)
1106 self
.MapProjectionVector
= self
.ProjectionFun(self
.ViewPortCenter
)
1107 self
.TransformVector
= array((self
.Scale
,-self
.Scale
),Float
)* self
.MapProjectionVector
1108 self
._BackgroundDirty
= 1
1111 def ZoomToFit(self
,event
):
1114 def ZoomToBB(self
,NewBB
= None,DrawFlag
= 1):
1118 Zooms the image to the bounding box given, or to the bounding
1119 box of all the objects on the canvas, if none is given.
1126 if self
.BoundingBoxDirty
:
1127 self
._ResetBoundingBox
()
1128 BoundingBox
= self
.BoundingBox
1130 self
.ViewPortCenter
= array(((BoundingBox
[0,0]+BoundingBox
[1,0])/2,
1131 (BoundingBox
[0,1]+BoundingBox
[1,1])/2 ),Float
)
1132 self
.MapProjectionVector
= self
.ProjectionFun(self
.ViewPortCenter
)
1133 # Compute the new Scale
1134 BoundingBox
= BoundingBox
* self
.MapProjectionVector
1136 self
.Scale
= min((self
.PanelSize
[0] / (BoundingBox
[1,0]-BoundingBox
[0,0])),
1137 (self
.PanelSize
[1] / (BoundingBox
[1,1]-BoundingBox
[0,1])))*0.95
1138 except ZeroDivisionError: # this will happen if the BB has zero width or height
1140 self
.Scale
= (self
.PanelSize
[0] / (BoundingBox
[1,0]-BoundingBox
[0,0]))*0.95
1141 except ZeroDivisionError:
1143 self
.Scale
= (self
.PanelSize
[1] / (BoundingBox
[1,1]-BoundingBox
[0,1]))*0.95
1144 except ZeroDivisionError: #zero size! (must be a single point)
1147 self
.TransformVector
= array((self
.Scale
,-self
.Scale
),Float
)* self
.MapProjectionVector
1149 self
._BackgroundDirty
= 1
1152 def RemoveObjects(self
,Objects
):
1153 for Object
in Objects
:
1154 self
.RemoveObject(Object
,ResetBB
= 0)
1155 self
.BoundingBoxDirty
= 1
1157 def RemoveObject(self
,Object
,ResetBB
= 1):
1158 if Object
.Foreground
:
1159 self
._TopDrawList
.remove(Object
)
1161 self
._DrawList
.remove(Object
)
1162 self
._BackgroundDirty
= 1
1165 self
.BoundingBoxDirty
= 1
1167 def Clear(self
, ResetBB
= True):
1169 if self
.UseBackground
:
1170 self
._TopDrawList
= []
1171 self
._BackgroundDirty
= 1
1173 self
._ResetBoundingBox
()
1175 def _AddBoundingBox(self
,NewBB
):
1176 if self
.BoundingBox
is None:
1177 self
.BoundingBox
= NewBB
1178 self
.ZoomToBB(NewBB
,DrawFlag
= 0)
1180 self
.BoundingBox
= array(((min(self
.BoundingBox
[0,0],NewBB
[0,0]),
1181 min(self
.BoundingBox
[0,1],NewBB
[0,1])),
1182 (max(self
.BoundingBox
[1,0],NewBB
[1,0]),
1183 max(self
.BoundingBox
[1,1],NewBB
[1,1]))),Float
)
1184 def _ResetBoundingBox(self
):
1185 # NOTE: could you remove an item without recomputing the entire bounding box?
1186 self
.BoundingBox
= None
1188 self
.BoundingBox
= self
._DrawList
[0].BoundingBox
1189 for Object
in self
._DrawList
[1:]:
1190 self
._AddBoundingBox
(Object
.BoundingBox
)
1191 if self
.UseBackground
:
1192 for Object
in self
._TopDrawList
:
1193 self
._AddBoundingBox
(Object
.BoundingBox
)
1194 if self
.BoundingBox
is None:
1195 self
.ViewPortCenter
= array( (0,0), Float
)
1196 self
.TransformVector
= array( (1,-1), Float
)
1197 self
.MapProjectionVector
= array( (1,1), Float
)
1199 self
.BoundingBoxDirty
= 0
1201 def PixelToWorld(self
,Points
):
1203 Converts coordinates from Pixel coordinates to world coordinates.
1205 Points is a tuple of (x,y) coordinates, or a list of such tuples, or a NX2 Numpy array of x,y coordinates.
1208 return (((array(Points
,Float
) - (self
.PanelSize
/2))/self
.TransformVector
) + self
.ViewPortCenter
)
1210 def WorldToPixel(self
,Coordinates
):
1212 This function will get passed to the drawing functions of the objects,
1213 to transform from world to pixel coordinates.
1214 Coordinates should be a NX2 array of (x,y) coordinates, or
1215 a 2-tuple, or sequence of 2-tuples.
1217 return (((array(Coordinates
,Float
) - self
.ViewPortCenter
)*self
.TransformVector
)+(self
.PanelSize
/2)).astype('i')
1219 def ScaleFunction(self
,Lengths
):
1221 This function will get passed to the drawing functions of the objects,
1222 to Change a length from world to pixel coordinates.
1224 Lengths should be a NX2 array of (x,y) coordinates, or
1225 a 2-tuple, or sequence of 2-tuples.
1227 return (array(Lengths
,Float
)*self
.TransformVector
).astype('i')
1230 ## This is a set of methods that add objects to the Canvas. It kind
1231 ## of seems like a lot of duplication, but I wanted to be able to
1232 ## instantiate the draw objects separatley form adding them, but
1233 ## also to be able to do add one oin one step. I'm open to better
1235 def AddRectangle(self
,x
,y
,width
,height
,
1236 LineColor
= "Black",
1237 LineStyle
= "Solid",
1240 FillStyle
= "Solid",
1242 Object
= Rectangle(x
,y
,width
,height
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
)
1243 self
.AddObject(Object
)
1246 def AddEllipse(self
,x
,y
,width
,height
,
1247 LineColor
= "Black",
1248 LineStyle
= "Solid",
1251 FillStyle
= "Solid",
1253 Object
= Ellipse(x
,y
,width
,height
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
)
1254 self
.AddObject(Object
)
1257 def AddCircle(self
,x
,y
,Diameter
,
1258 LineColor
= "Black",
1259 LineStyle
= "Solid",
1262 FillStyle
= "Solid",
1264 Object
= Circle(x
,y
,Diameter
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
)
1265 self
.AddObject(Object
)
1268 def AddDot(self
,x
,y
,Diameter
,
1269 LineColor
= "Black",
1270 LineStyle
= "Solid",
1273 FillStyle
= "Solid",
1275 Object
= Dot(x
,y
,Diameter
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
)
1276 self
.AddObject(Object
)
1279 def AddPolygon(self
,Points
,
1280 LineColor
= "Black",
1281 LineStyle
= "Solid",
1284 FillStyle
= "Solid",
1287 Object
= Polygon(Points
,LineColor
,LineStyle
,LineWidth
,FillColor
,FillStyle
,Foreground
)
1288 self
.AddObject(Object
)
1291 def AddLine(self
,Points
,
1292 LineColor
= "Black",
1293 LineStyle
= "Solid",
1297 Object
= Line(Points
,LineColor
,LineStyle
,LineWidth
,Foreground
)
1298 self
.AddObject(Object
)
1301 def AddLineSet(self
,Points
,
1302 LineColors
= "Black",
1303 LineStyles
= "Solid",
1307 Object
= LineSet(Points
,LineColors
,LineStyles
,LineWidths
,Foreground
)
1308 self
.AddObject(Object
)
1311 def AddPointSet(self
,Points
,
1316 Object
= PointSet(Points
,Color
,Diameter
,Foreground
)
1317 self
.AddObject(Object
)
1320 def AddText(self
,String
,x
,y
,
1322 ForeGround
= 'Black',
1330 Object
= Text(String
,x
,y
,Size
,ForeGround
,BackGround
,Family
,Style
,Weight
,Underline
,Position
,Foreground
)
1331 self
.AddObject(Object
)
1334 def AddObject(self
,obj
):
1335 # put in a reference to the Canvas, so remove and other stuff can work
1337 if obj
.Foreground
and self
.UseBackground
:
1338 self
._TopDrawList
.append(obj
)
1340 self
._DrawList
.append(obj
)
1341 self
._backgrounddirty
= 1
1342 self
._AddBoundingBox
(obj
.BoundingBox
)