247 |
def __init__(self, parent, winid, interactor): |
def __init__(self, parent, winid, interactor): |
248 |
wxWindow.__init__(self, parent, winid) |
wxWindow.__init__(self, parent, winid) |
249 |
self.SetBackgroundColour(wxColour(255, 255, 255)) |
self.SetBackgroundColour(wxColour(255, 255, 255)) |
250 |
|
|
251 |
|
# the map displayed in this canvas. Set with SetMap() |
252 |
self.map = None |
self.map = None |
253 |
|
|
254 |
|
# scale and offset describe the transformation from projected |
255 |
|
# coordinates to window coordinates. |
256 |
self.scale = 1.0 |
self.scale = 1.0 |
257 |
self.offset = (0, 0) |
self.offset = (0, 0) |
258 |
|
|
259 |
|
# whether the user is currently dragging the mouse, i.e. moving |
260 |
|
# the mouse while pressing a mouse button |
261 |
self.dragging = 0 |
self.dragging = 0 |
262 |
|
|
263 |
|
# the currently active tool |
264 |
self.tool = None |
self.tool = None |
265 |
self.redraw_on_idle = 0 |
|
266 |
|
# The current mouse position of the last OnMotion event or None |
267 |
|
# if the mouse is outside the window. |
268 |
self.current_position = None |
self.current_position = None |
269 |
|
|
270 |
|
|
271 |
|
# If true, OnIdle will call do_redraw to do the actual |
272 |
|
# redrawing. Set by OnPaint to avoid some unnecessary redraws. |
273 |
|
# To force a redraw call full_redraw(). |
274 |
|
self.redraw_on_idle = 0 |
275 |
|
|
276 |
|
# the bitmap serving as backing store |
277 |
|
self.bitmap = None |
278 |
|
|
279 |
|
# the interactor |
280 |
|
self.interactor = interactor |
281 |
|
self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected) |
282 |
|
|
283 |
|
# subscribe the WX events we're interested in |
284 |
EVT_PAINT(self, self.OnPaint) |
EVT_PAINT(self, self.OnPaint) |
285 |
EVT_LEFT_DOWN(self, self.OnLeftDown) |
EVT_LEFT_DOWN(self, self.OnLeftDown) |
286 |
EVT_LEFT_UP(self, self.OnLeftUp) |
EVT_LEFT_UP(self, self.OnLeftUp) |
287 |
EVT_MOTION(self, self.OnMotion) |
EVT_MOTION(self, self.OnMotion) |
288 |
EVT_LEAVE_WINDOW(self, self.OnLeaveWindow) |
EVT_LEAVE_WINDOW(self, self.OnLeaveWindow) |
289 |
|
wx.EVT_SIZE(self, self.OnSize) |
290 |
wx.EVT_IDLE(self, self.OnIdle) |
wx.EVT_IDLE(self, self.OnIdle) |
|
self.interactor = interactor |
|
|
self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected) |
|
291 |
|
|
292 |
def __del__(self): |
def __del__(self): |
293 |
wxWindow.__del__(self) |
wxWindow.__del__(self) |
313 |
def do_redraw(self): |
def do_redraw(self): |
314 |
# This should only be called if we have a non-empty map. We draw |
# This should only be called if we have a non-empty map. We draw |
315 |
# it into a memory DC and then blit it to the screen. |
# it into a memory DC and then blit it to the screen. |
316 |
|
|
317 |
width, height = self.GetSizeTuple() |
width, height = self.GetSizeTuple() |
|
bitmap = wx.wxEmptyBitmap(width, height) |
|
|
dc = wx.wxMemoryDC() |
|
|
dc.SelectObject(bitmap) |
|
|
dc.BeginDrawing() |
|
318 |
|
|
319 |
# clear the background |
# If self.bitmap's still there, reuse it. Otherwise redraw it |
320 |
dc.SetBrush(wx.wxWHITE_BRUSH) |
if self.bitmap is not None: |
321 |
dc.SetPen(wx.wxTRANSPARENT_PEN) |
bitmap = self.bitmap |
|
dc.DrawRectangle(0, 0, width, height) |
|
|
|
|
|
if 1: #self.interactor.selected_map is self.map: |
|
|
selected_layer = self.interactor.selected_layer |
|
|
selected_shape = self.interactor.selected_shape |
|
322 |
else: |
else: |
323 |
selected_layer = None |
bitmap = wx.wxEmptyBitmap(width, height) |
324 |
selected_shape = None |
dc = wx.wxMemoryDC() |
325 |
|
dc.SelectObject(bitmap) |
326 |
|
dc.BeginDrawing() |
327 |
|
|
328 |
# draw the map into the bitmap |
# clear the background |
329 |
renderer = ScreenRenderer(dc, self.scale, self.offset) |
dc.SetBrush(wx.wxWHITE_BRUSH) |
330 |
renderer.RenderMap(self.map, selected_layer, selected_shape) |
dc.SetPen(wx.wxTRANSPARENT_PEN) |
331 |
|
dc.DrawRectangle(0, 0, width, height) |
332 |
|
|
333 |
|
if 1: #self.interactor.selected_map is self.map: |
334 |
|
selected_layer = self.interactor.selected_layer |
335 |
|
selected_shape = self.interactor.selected_shape |
336 |
|
else: |
337 |
|
selected_layer = None |
338 |
|
selected_shape = None |
339 |
|
|
340 |
|
# draw the map into the bitmap |
341 |
|
renderer = ScreenRenderer(dc, self.scale, self.offset) |
342 |
|
renderer.RenderMap(self.map, selected_layer, selected_shape) |
343 |
|
|
344 |
dc.EndDrawing() |
dc.EndDrawing() |
345 |
|
dc.SelectObject(wx.wxNullBitmap) |
346 |
|
self.bitmap = bitmap |
347 |
|
|
348 |
# blit the bitmap to the screen |
# blit the bitmap to the screen |
349 |
|
dc = wx.wxMemoryDC() |
350 |
|
dc.SelectObject(bitmap) |
351 |
clientdc = wxClientDC(self) |
clientdc = wxClientDC(self) |
352 |
clientdc.BeginDrawing() |
clientdc.BeginDrawing() |
353 |
clientdc.Blit(0, 0, width, height, dc, 0, 0) |
clientdc.Blit(0, 0, width, height, dc, 0, 0) |
364 |
LAYER_VISIBILITY_CHANGED) |
LAYER_VISIBILITY_CHANGED) |
365 |
if self.map is not None: |
if self.map is not None: |
366 |
for channel in redraw_channels: |
for channel in redraw_channels: |
367 |
self.map.Unsubscribe(channel, self.redraw) |
self.map.Unsubscribe(channel, self.full_redraw) |
368 |
self.map.Unsubscribe(MAP_PROJECTION_CHANGED, |
self.map.Unsubscribe(MAP_PROJECTION_CHANGED, |
369 |
self.projection_changed) |
self.projection_changed) |
370 |
self.map = map |
self.map = map |
371 |
if self.map is not None: |
if self.map is not None: |
372 |
for channel in redraw_channels: |
for channel in redraw_channels: |
373 |
self.map.Subscribe(channel, self.redraw) |
self.map.Subscribe(channel, self.full_redraw) |
374 |
self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed) |
self.map.Subscribe(MAP_PROJECTION_CHANGED, self.projection_changed) |
375 |
self.FitMapToWindow() |
self.FitMapToWindow() |
376 |
# force a redraw. If map is not empty, it's already been called |
# force a redraw. If map is not empty, it's already been called |
377 |
# by FitMapToWindow but if map is empty it hasn't been called |
# by FitMapToWindow but if map is empty it hasn't been called |
378 |
# yet so we have to explicitly call it. |
# yet so we have to explicitly call it. |
379 |
self.redraw() |
self.full_redraw() |
380 |
|
|
381 |
def Map(self): |
def Map(self): |
382 |
return self.map |
return self.map |
384 |
def redraw(self, *args): |
def redraw(self, *args): |
385 |
self.Refresh(0) |
self.Refresh(0) |
386 |
|
|
387 |
|
def full_redraw(self, *args): |
388 |
|
self.bitmap = None |
389 |
|
self.redraw() |
390 |
|
|
391 |
def projection_changed(self, *args): |
def projection_changed(self, *args): |
392 |
self.FitMapToWindow() |
self.FitMapToWindow() |
393 |
self.redraw() |
self.full_redraw() |
394 |
|
|
395 |
def set_view_transform(self, scale, offset): |
def set_view_transform(self, scale, offset): |
396 |
self.scale = scale |
self.scale = scale |
397 |
self.offset = offset |
self.offset = offset |
398 |
self.redraw() |
self.full_redraw() |
399 |
|
|
400 |
def proj_to_win(self, x, y): |
def proj_to_win(self, x, y): |
401 |
"""\ |
"""\ |
552 |
self.do_redraw() |
self.do_redraw() |
553 |
self.redraw_on_idle = 0 |
self.redraw_on_idle = 0 |
554 |
|
|
555 |
|
def OnSize(self, event): |
556 |
|
# the window's size has changed. We have to get a new bitmap. If |
557 |
|
# we want to be clever we could try to get by without throwing |
558 |
|
# everything away. E.g. when the window gets smaller, we could |
559 |
|
# either keep the bitmap or create the new one from the old one. |
560 |
|
# Even when the window becomes larger some parts of the bitmap |
561 |
|
# could be reused. |
562 |
|
self.full_redraw() |
563 |
|
|
564 |
def shape_selected(self, layer, shape): |
def shape_selected(self, layer, shape): |
565 |
self.redraw() |
self.full_redraw() |
566 |
|
|
567 |
def find_shape_at(self, px, py, select_labels = 0, selected_layer = 1): |
def find_shape_at(self, px, py, select_labels = 0, selected_layer = 1): |
568 |
"""Determine the shape at point px, py in window coords |
"""Determine the shape at point px, py in window coords |