1 |
# Copyright (c) 2001, 2002 by Intevation GmbH |
# Copyright (c) 2001, 2002, 2003 by Intevation GmbH |
2 |
# Authors: |
# Authors: |
3 |
# Bernhard Herzog <[email protected]> |
# Bernhard Herzog <[email protected]> |
4 |
|
# Frank Koormann <[email protected]> |
5 |
# |
# |
6 |
# This program is free software under the GPL (>=v2) |
# This program is free software under the GPL (>=v2) |
7 |
# Read the file COPYING coming with Thuban for details. |
# Read the file COPYING coming with Thuban for details. |
12 |
|
|
13 |
__version__ = "$Revision$" |
__version__ = "$Revision$" |
14 |
|
|
15 |
|
from Thuban import _ |
16 |
|
|
17 |
|
import sys |
18 |
|
import os.path |
19 |
|
|
20 |
from math import hypot |
from math import hypot |
21 |
|
|
22 |
from wxPython.wx import wxWindow,\ |
from wxPython.wx import wxWindow,\ |
23 |
wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\ |
wxPaintDC, wxColour, wxClientDC, wxINVERT, wxTRANSPARENT_BRUSH, wxFont,\ |
24 |
EVT_PAINT, EVT_LEFT_DOWN, EVT_LEFT_UP, EVT_MOTION, EVT_LEAVE_WINDOW |
EVT_PAINT, EVT_LEFT_DOWN, EVT_LEFT_UP, EVT_MOTION, EVT_LEAVE_WINDOW, \ |
25 |
|
wxBITMAP_TYPE_XPM, wxBeginBusyCursor, wxEndBusyCursor, wxCursor, \ |
26 |
|
wxImageFromBitmap, wxPlatform |
27 |
|
|
28 |
|
# Export related stuff |
29 |
|
if wxPlatform == '__WXMSW__': |
30 |
|
from wxPython.wx import wxMetaFileDC |
31 |
|
from wxPython.wx import wxFileDialog, wxSAVE, wxOVERWRITE_PROMPT, wxID_OK |
32 |
|
|
33 |
from wxPython import wx |
from wxPython import wx |
34 |
|
|
35 |
from wxproj import point_in_polygon_shape, shape_centroid |
from wxproj import point_in_polygon_shape, shape_centroid |
36 |
|
|
|
|
|
37 |
from Thuban.Model.messages import MAP_PROJECTION_CHANGED, \ |
from Thuban.Model.messages import MAP_PROJECTION_CHANGED, \ |
38 |
LAYERS_CHANGED, LAYER_LEGEND_CHANGED, LAYER_VISIBILITY_CHANGED |
MAP_LAYERS_CHANGED, LAYER_CHANGED, LAYER_VISIBILITY_CHANGED |
39 |
from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \ |
from Thuban.Model.layer import SHAPETYPE_POLYGON, SHAPETYPE_ARC, \ |
40 |
SHAPETYPE_POINT |
SHAPETYPE_POINT |
41 |
from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \ |
from Thuban.Model.label import ALIGN_CENTER, ALIGN_TOP, ALIGN_BOTTOM, \ |
42 |
ALIGN_LEFT, ALIGN_RIGHT |
ALIGN_LEFT, ALIGN_RIGHT |
43 |
from Thuban.Lib.connector import Publisher |
from Thuban.Lib.connector import Publisher |
44 |
|
from Thuban.Model.color import Color |
45 |
|
|
46 |
|
import resource |
47 |
|
|
48 |
from renderer import ScreenRenderer, PrinterRender |
from selection import Selection |
49 |
|
from renderer import ScreenRenderer, ExportRenderer, PrinterRenderer |
50 |
|
|
51 |
import labeldialog |
import labeldialog |
52 |
|
|
53 |
from messages import SELECTED_SHAPE, VIEW_POSITION |
from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, \ |
54 |
|
SCALE_CHANGED |
55 |
|
|
56 |
|
|
57 |
# |
# |
231 |
self.view.LabelShapeAt(event.m_x, event.m_y) |
self.view.LabelShapeAt(event.m_x, event.m_y) |
232 |
|
|
233 |
|
|
|
|
|
|
|
|
234 |
class MapPrintout(wx.wxPrintout): |
class MapPrintout(wx.wxPrintout): |
235 |
|
|
236 |
""" |
""" |
237 |
wxPrintout class for printing Thuban maps |
wxPrintout class for printing Thuban maps |
238 |
""" |
""" |
239 |
|
|
240 |
def __init__(self, map): |
def __init__(self, canvas, map, region, selected_layer, selected_shapes): |
241 |
wx.wxPrintout.__init__(self) |
wx.wxPrintout.__init__(self) |
242 |
|
self.canvas = canvas |
243 |
self.map = map |
self.map = map |
244 |
|
self.region = region |
245 |
|
self.selected_layer = selected_layer |
246 |
|
self.selected_shapes = selected_shapes |
247 |
|
|
248 |
def GetPageInfo(self): |
def GetPageInfo(self): |
249 |
return (1, 1, 1, 1) |
return (1, 1, 1, 1) |
257 |
|
|
258 |
def draw_on_dc(self, dc): |
def draw_on_dc(self, dc): |
259 |
width, height = self.GetPageSizePixels() |
width, height = self.GetPageSizePixels() |
260 |
llx, lly, urx, ury = self.map.ProjectedBoundingBox() |
scale, offset, mapregion = OutputTransform(self.canvas.scale, |
261 |
scalex = width / (urx - llx) |
self.canvas.offset, |
262 |
scaley = height / (ury - lly) |
self.canvas.GetSizeTuple(), |
263 |
scale = min(scalex, scaley) |
self.GetPageSizePixels()) |
|
offx = 0.5 * (width - (urx + llx) * scale) |
|
|
offy = 0.5 * (height + (ury + lly) * scale) |
|
|
|
|
264 |
resx, resy = self.GetPPIPrinter() |
resx, resy = self.GetPPIPrinter() |
265 |
renderer = PrinterRender(dc, scale, (offx, offy), resolution = resx) |
renderer = PrinterRenderer(dc, scale, offset, resolution = resy) |
266 |
renderer.RenderMap(self.map) |
x, y, width, height = self.region |
267 |
return wx.true |
canvas_scale = self.canvas.scale |
268 |
|
renderer.RenderMap(self.map, |
269 |
|
(0,0, |
270 |
|
(width/canvas_scale)*scale, |
271 |
|
(height/canvas_scale)*scale), |
272 |
|
mapregion, |
273 |
|
self.selected_layer, self.selected_shapes) |
274 |
|
return True |
275 |
|
|
276 |
|
|
277 |
class MapCanvas(wxWindow, Publisher): |
class MapCanvas(wxWindow, Publisher): |
278 |
|
|
279 |
"""A widget that displays a map and offers some interaction""" |
"""A widget that displays a map and offers some interaction""" |
280 |
|
|
281 |
def __init__(self, parent, winid, interactor): |
# Some messages that can be subscribed/unsubscribed directly through |
282 |
|
# the MapCanvas come in fact from other objects. This is a dict |
283 |
|
# mapping those messages to the names of the instance variables they |
284 |
|
# actually come from. The delegation is implemented in the Subscribe |
285 |
|
# and Unsubscribe methods |
286 |
|
delegated_messages = {LAYER_SELECTED: "selection", |
287 |
|
SHAPES_SELECTED: "selection"} |
288 |
|
|
289 |
|
# Methods delegated to some instance variables. The delegation is |
290 |
|
# implemented in the __getattr__ method. |
291 |
|
delegated_methods = {"SelectLayer": "selection", |
292 |
|
"SelectShapes": "selection", |
293 |
|
"SelectedLayer": "selection", |
294 |
|
"HasSelectedLayer": "selection", |
295 |
|
"HasSelectedShapes": "selection", |
296 |
|
"SelectedShapes": "selection"} |
297 |
|
|
298 |
|
def __init__(self, parent, winid): |
299 |
wxWindow.__init__(self, parent, winid) |
wxWindow.__init__(self, parent, winid) |
300 |
self.SetBackgroundColour(wxColour(255, 255, 255)) |
self.SetBackgroundColour(wxColour(255, 255, 255)) |
301 |
|
|
318 |
# if the mouse is outside the window. |
# if the mouse is outside the window. |
319 |
self.current_position = None |
self.current_position = None |
320 |
|
|
|
# If true, OnIdle will call do_redraw to do the actual |
|
|
# redrawing. Set by OnPaint to avoid some unnecessary redraws. |
|
|
# To force a redraw call full_redraw(). |
|
|
self.redraw_on_idle = 0 |
|
|
|
|
321 |
# the bitmap serving as backing store |
# the bitmap serving as backing store |
322 |
self.bitmap = None |
self.bitmap = None |
323 |
|
|
324 |
# the interactor |
# the selection |
325 |
self.interactor = interactor |
self.selection = Selection() |
326 |
self.interactor.Subscribe(SELECTED_SHAPE, self.shape_selected) |
self.selection.Subscribe(SHAPES_SELECTED , self.shape_selected) |
327 |
|
|
328 |
# keep track of which layers/shapes are selected to make sure we |
# keep track of which layers/shapes are selected to make sure we |
329 |
# only redraw when necessary |
# only redraw when necessary |
337 |
EVT_MOTION(self, self.OnMotion) |
EVT_MOTION(self, self.OnMotion) |
338 |
EVT_LEAVE_WINDOW(self, self.OnLeaveWindow) |
EVT_LEAVE_WINDOW(self, self.OnLeaveWindow) |
339 |
wx.EVT_SIZE(self, self.OnSize) |
wx.EVT_SIZE(self, self.OnSize) |
|
wx.EVT_IDLE(self, self.OnIdle) |
|
340 |
|
|
341 |
def __del__(self): |
def __del__(self): |
342 |
wxWindow.__del__(self) |
wxWindow.__del__(self) |
343 |
Publisher.__del__(self) |
Publisher.__del__(self) |
344 |
|
|
345 |
|
def Subscribe(self, channel, *args): |
346 |
|
"""Extend the inherited method to handle delegated messages. |
347 |
|
|
348 |
|
If channel is one of the delegated messages call the appropriate |
349 |
|
object's Subscribe method. Otherwise just call the inherited |
350 |
|
method. |
351 |
|
""" |
352 |
|
if channel in self.delegated_messages: |
353 |
|
object = getattr(self, self.delegated_messages[channel]) |
354 |
|
object.Subscribe(channel, *args) |
355 |
|
else: |
356 |
|
Publisher.Subscribe(self, channel, *args) |
357 |
|
|
358 |
|
def Unsubscribe(self, channel, *args): |
359 |
|
"""Extend the inherited method to handle delegated messages. |
360 |
|
|
361 |
|
If channel is one of the delegated messages call the appropriate |
362 |
|
object's Unsubscribe method. Otherwise just call the inherited |
363 |
|
method. |
364 |
|
""" |
365 |
|
if channel in self.delegated_messages: |
366 |
|
object = getattr(self, self.delegated_messages[channel]) |
367 |
|
object.Unsubscribe(channel, *args) |
368 |
|
else: |
369 |
|
Publisher.Unsubscribe(self, channel, *args) |
370 |
|
|
371 |
|
def __getattr__(self, attr): |
372 |
|
if attr in self.delegated_methods: |
373 |
|
return getattr(getattr(self, self.delegated_methods[attr]), attr) |
374 |
|
raise AttributeError(attr) |
375 |
|
|
376 |
def OnPaint(self, event): |
def OnPaint(self, event): |
377 |
dc = wxPaintDC(self) |
dc = wxPaintDC(self) |
378 |
if self.map is not None and self.map.HasLayers(): |
clear = self.map is None or not self.map.HasLayers() |
379 |
# We have a non-empty map. Redraw it in idle time |
|
380 |
self.redraw_on_idle = 1 |
wxBeginBusyCursor() |
381 |
else: |
try: |
382 |
# If we've got no map or if the map is empty, simply clear |
if not clear: |
383 |
# the screen. |
self.do_redraw() |
384 |
|
try: |
385 |
# XXX it's probably possible to get rid of this. The |
pass |
386 |
# background color of the window is already white and the |
except: |
387 |
# only thing we may have to do is to call self.Refresh() |
print "Error during drawing:", sys.exc_info()[0] |
388 |
# with a true argument in the right places. |
clear = True |
389 |
dc.BeginDrawing() |
|
390 |
dc.Clear() |
if clear: |
391 |
dc.EndDrawing() |
# If we've got no map or if the map is empty, simply clear |
392 |
|
# the screen. |
393 |
|
|
394 |
|
# XXX it's probably possible to get rid of this. The |
395 |
|
# background color of the window is already white and the |
396 |
|
# only thing we may have to do is to call self.Refresh() |
397 |
|
# with a true argument in the right places. |
398 |
|
dc.BeginDrawing() |
399 |
|
dc.Clear() |
400 |
|
dc.EndDrawing() |
401 |
|
finally: |
402 |
|
wxEndBusyCursor() |
403 |
|
|
404 |
def do_redraw(self): |
def do_redraw(self): |
405 |
# This should only be called if we have a non-empty map. |
# This should only be called if we have a non-empty map. |
417 |
dc.BeginDrawing() |
dc.BeginDrawing() |
418 |
|
|
419 |
# clear the background |
# clear the background |
420 |
dc.SetBrush(wx.wxWHITE_BRUSH) |
#dc.SetBrush(wx.wxWHITE_BRUSH) |
421 |
dc.SetPen(wx.wxTRANSPARENT_PEN) |
#dc.SetPen(wx.wxTRANSPARENT_PEN) |
422 |
dc.DrawRectangle(0, 0, width, height) |
#dc.DrawRectangle(0, 0, width, height) |
423 |
|
dc.SetBackground(wx.wxWHITE_BRUSH) |
424 |
if 1: #self.interactor.selected_map is self.map: |
dc.Clear() |
425 |
selected_layer = self.interactor.selected_layer |
|
426 |
selected_shape = self.interactor.selected_shape |
selected_layer = self.selection.SelectedLayer() |
427 |
else: |
selected_shapes = self.selection.SelectedShapes() |
|
selected_layer = None |
|
|
selected_shape = None |
|
428 |
|
|
429 |
# draw the map into the bitmap |
# draw the map into the bitmap |
430 |
renderer = ScreenRenderer(dc, self.scale, self.offset) |
renderer = ScreenRenderer(dc, self.scale, self.offset) |
432 |
# Pass the entire bitmap as update region to the renderer. |
# Pass the entire bitmap as update region to the renderer. |
433 |
# We're redrawing the whole bitmap, after all. |
# We're redrawing the whole bitmap, after all. |
434 |
renderer.RenderMap(self.map, (0, 0, width, height), |
renderer.RenderMap(self.map, (0, 0, width, height), |
435 |
selected_layer, selected_shape) |
selected_layer, selected_shapes) |
436 |
|
|
437 |
dc.EndDrawing() |
dc.EndDrawing() |
438 |
dc.SelectObject(wx.wxNullBitmap) |
dc.SelectObject(wx.wxNullBitmap) |
446 |
clientdc.Blit(0, 0, width, height, dc, 0, 0) |
clientdc.Blit(0, 0, width, height, dc, 0, 0) |
447 |
clientdc.EndDrawing() |
clientdc.EndDrawing() |
448 |
|
|
449 |
|
def Export(self): |
450 |
|
if self.scale == 0: |
451 |
|
return |
452 |
|
|
453 |
|
if hasattr(self, "export_path"): |
454 |
|
export_path = self.export_path |
455 |
|
else: |
456 |
|
export_path="." |
457 |
|
dlg = wxFileDialog(self, _("Export Map"), export_path, "", |
458 |
|
"Enhanced Metafile (*.wmf)|*.wmf", |
459 |
|
wxSAVE|wxOVERWRITE_PROMPT) |
460 |
|
if dlg.ShowModal() == wxID_OK: |
461 |
|
self.export_path = os.path.dirname(dlg.GetPath()) |
462 |
|
dc = wxMetaFileDC(dlg.GetPath()) |
463 |
|
|
464 |
|
scale, offset, mapregion = OutputTransform(self.scale, |
465 |
|
self.offset, |
466 |
|
self.GetSizeTuple(), |
467 |
|
dc.GetSizeTuple()) |
468 |
|
|
469 |
|
selected_layer = self.selection.SelectedLayer() |
470 |
|
selected_shapes = self.selection.SelectedShapes() |
471 |
|
|
472 |
|
renderer = ExportRenderer(dc, scale, offset) |
473 |
|
|
474 |
|
# Pass the entire bitmap as update region to the renderer. |
475 |
|
# We're redrawing the whole bitmap, after all. |
476 |
|
width, height = self.GetSizeTuple() |
477 |
|
renderer.RenderMap(self.map, |
478 |
|
(0,0, |
479 |
|
(width/self.scale)*scale, |
480 |
|
(height/self.scale)*scale), |
481 |
|
mapregion, |
482 |
|
selected_layer, selected_shapes) |
483 |
|
dc.EndDrawing() |
484 |
|
dc.Close() |
485 |
|
dlg.Destroy() |
486 |
|
|
487 |
def Print(self): |
def Print(self): |
488 |
printer = wx.wxPrinter() |
printer = wx.wxPrinter() |
489 |
printout = MapPrintout(self.map) |
width, height = self.GetSizeTuple() |
490 |
printer.Print(self, printout, wx.true) |
selected_layer = self.selection.SelectedLayer() |
491 |
|
selected_shapes = self.selection.SelectedShapes() |
492 |
|
|
493 |
|
printout = MapPrintout(self, self.map, (0, 0, width, height), |
494 |
|
selected_layer, selected_shapes) |
495 |
|
printer.Print(self, printout, True) |
496 |
printout.Destroy() |
printout.Destroy() |
497 |
|
|
498 |
def SetMap(self, map): |
def SetMap(self, map): |
499 |
redraw_channels = (LAYERS_CHANGED, LAYER_LEGEND_CHANGED, |
redraw_channels = (MAP_LAYERS_CHANGED, LAYER_CHANGED, |
500 |
LAYER_VISIBILITY_CHANGED) |
LAYER_VISIBILITY_CHANGED) |
501 |
if self.map is not None: |
if self.map is not None: |
502 |
for channel in redraw_channels: |
for channel in redraw_channels: |
504 |
self.map.Unsubscribe(MAP_PROJECTION_CHANGED, |
self.map.Unsubscribe(MAP_PROJECTION_CHANGED, |
505 |
self.projection_changed) |
self.projection_changed) |
506 |
self.map = map |
self.map = map |
507 |
|
self.selection.ClearSelection() |
508 |
if self.map is not None: |
if self.map is not None: |
509 |
for channel in redraw_channels: |
for channel in redraw_channels: |
510 |
self.map.Subscribe(channel, self.full_redraw) |
self.map.Subscribe(channel, self.full_redraw) |
531 |
self.full_redraw() |
self.full_redraw() |
532 |
|
|
533 |
def set_view_transform(self, scale, offset): |
def set_view_transform(self, scale, offset): |
534 |
|
# width/height of the projected bbox |
535 |
|
llx, lly, urx, ury = bbox = self.map.ProjectedBoundingBox() |
536 |
|
pwidth = float(urx - llx) |
537 |
|
pheight = float(ury - lly) |
538 |
|
|
539 |
|
# width/height of the window |
540 |
|
wwidth, wheight = self.GetSizeTuple() |
541 |
|
|
542 |
|
# The window's center in projected coordinates assuming the new |
543 |
|
# scale/offset |
544 |
|
pcenterx = (wwidth/2 - offset[0]) / scale |
545 |
|
pcentery = (offset[1] - wheight/2) / scale |
546 |
|
|
547 |
|
# The window coordinates used when drawing the shapes must fit |
548 |
|
# into 16bit signed integers. |
549 |
|
max_len = max(pwidth, pheight) |
550 |
|
if max_len: |
551 |
|
max_scale = 32000.0 / max_len |
552 |
|
else: |
553 |
|
# FIXME: What to do in this case? The bbox is effectively |
554 |
|
# empty so any scale should work. |
555 |
|
max_scale = scale |
556 |
|
|
557 |
|
# The minimal scale is somewhat arbitrarily set to half that of |
558 |
|
# the bbox fit into the window |
559 |
|
scales = [] |
560 |
|
if pwidth: |
561 |
|
scales.append(wwidth / pwidth) |
562 |
|
if pheight: |
563 |
|
scales.append(wheight / pheight) |
564 |
|
if scales: |
565 |
|
min_scale = 0.5 * min(scales) |
566 |
|
else: |
567 |
|
min_scale = scale |
568 |
|
|
569 |
|
if scale > max_scale: |
570 |
|
scale = max_scale |
571 |
|
elif scale < min_scale: |
572 |
|
scale = min_scale |
573 |
|
|
574 |
self.scale = scale |
self.scale = scale |
575 |
self.offset = offset |
|
576 |
|
# determine new offset to preserve the center |
577 |
|
self.offset = (wwidth/2 - scale * pcenterx, |
578 |
|
wheight/2 + scale * pcentery) |
579 |
self.full_redraw() |
self.full_redraw() |
580 |
|
self.issue(SCALE_CHANGED, scale) |
581 |
|
|
582 |
def proj_to_win(self, x, y): |
def proj_to_win(self, x, y): |
583 |
"""\ |
"""\ |
584 |
Return the point in window coords given by projected coordinates x y |
Return the point in window coords given by projected coordinates x y |
585 |
""" |
""" |
586 |
|
if self.scale == 0: |
587 |
|
return (0, 0) |
588 |
|
|
589 |
offx, offy = self.offset |
offx, offy = self.offset |
590 |
return (self.scale * x + offx, -self.scale * y + offy) |
return (self.scale * x + offx, -self.scale * y + offy) |
591 |
|
|
593 |
"""\ |
"""\ |
594 |
Return the point in projected coordinates given by window coords x y |
Return the point in projected coordinates given by window coords x y |
595 |
""" |
""" |
596 |
|
if self.scale == 0: |
597 |
|
return (0, 0) |
598 |
|
|
599 |
offx, offy = self.offset |
offx, offy = self.offset |
600 |
return ((x - offx) / self.scale, (offy - y) / self.scale) |
return ((x - offx) / self.scale, (offy - y) / self.scale) |
601 |
|
|
602 |
def FitRectToWindow(self, rect): |
def FitRectToWindow(self, rect): |
603 |
"""Fit the rectangular region given by rect into the window. |
"""Fit the rectangular region given by rect into the window. |
604 |
|
|
605 |
Set scale so that rect (in projected coordinates) just fits into |
Set scale so that rect (in projected coordinates) just fits into |
606 |
the window and center it. |
the window and center it. |
607 |
""" |
""" |
608 |
width, height = self.GetSizeTuple() |
width, height = self.GetSizeTuple() |
609 |
llx, lly, urx, ury = rect |
llx, lly, urx, ury = rect |
610 |
if llx == urx or lly == ury: |
if llx == urx or lly == ury: |
611 |
# zero with or zero height. Do Nothing |
# zero width or zero height. Do Nothing |
612 |
return |
return |
613 |
scalex = width / (urx - llx) |
scalex = width / (urx - llx) |
614 |
scaley = height / (ury - lly) |
scaley = height / (ury - lly) |
619 |
|
|
620 |
def FitMapToWindow(self): |
def FitMapToWindow(self): |
621 |
"""Fit the map to the window |
"""Fit the map to the window |
622 |
|
|
623 |
Set the scale so that the map fits exactly into the window and |
Set the scale so that the map fits exactly into the window and |
624 |
center it in the window. |
center it in the window. |
625 |
""" |
""" |
627 |
if bbox is not None: |
if bbox is not None: |
628 |
self.FitRectToWindow(bbox) |
self.FitRectToWindow(bbox) |
629 |
|
|
630 |
|
def FitLayerToWindow(self, layer): |
631 |
|
"""Fit the given layer to the window. |
632 |
|
|
633 |
|
Set the scale so that the layer fits exactly into the window and |
634 |
|
center it in the window. |
635 |
|
""" |
636 |
|
|
637 |
|
bbox = layer.LatLongBoundingBox() |
638 |
|
if bbox is not None: |
639 |
|
proj = self.map.GetProjection() |
640 |
|
if proj is not None: |
641 |
|
bbox = proj.ForwardBBox(bbox) |
642 |
|
|
643 |
|
if bbox is not None: |
644 |
|
self.FitRectToWindow(bbox) |
645 |
|
|
646 |
|
def FitSelectedToWindow(self): |
647 |
|
layer = self.selection.SelectedLayer() |
648 |
|
shapes = self.selection.SelectedShapes() |
649 |
|
|
650 |
|
bbox = layer.ShapesBoundingBox(shapes) |
651 |
|
if bbox is not None: |
652 |
|
proj = self.map.GetProjection() |
653 |
|
if proj is not None: |
654 |
|
bbox = proj.ForwardBBox(bbox) |
655 |
|
|
656 |
|
if bbox is not None: |
657 |
|
if len(shapes) == 1 and layer.ShapeType() == SHAPETYPE_POINT: |
658 |
|
self.ZoomFactor(1, self.proj_to_win(bbox[0], bbox[1])) |
659 |
|
else: |
660 |
|
self.FitRectToWindow(bbox) |
661 |
|
|
662 |
def ZoomFactor(self, factor, center = None): |
def ZoomFactor(self, factor, center = None): |
663 |
"""Multiply the zoom by factor and center on center. |
"""Multiply the zoom by factor and center on center. |
664 |
|
|
666 |
that should be centered. If it is omitted, it defaults to the |
that should be centered. If it is omitted, it defaults to the |
667 |
center of the window |
center of the window |
668 |
""" |
""" |
669 |
width, height = self.GetSizeTuple() |
if self.scale > 0: |
670 |
scale = self.scale * factor |
width, height = self.GetSizeTuple() |
671 |
offx, offy = self.offset |
scale = self.scale * factor |
672 |
if center is not None: |
offx, offy = self.offset |
673 |
cx, cy = center |
if center is not None: |
674 |
else: |
cx, cy = center |
675 |
cx = width / 2 |
else: |
676 |
cy = height / 2 |
cx = width / 2 |
677 |
offset = (factor * (offx - cx) + width / 2, |
cy = height / 2 |
678 |
factor * (offy - cy) + height / 2) |
offset = (factor * (offx - cx) + width / 2, |
679 |
self.set_view_transform(scale, offset) |
factor * (offy - cy) + height / 2) |
680 |
|
self.set_view_transform(scale, offset) |
681 |
|
|
682 |
def ZoomOutToRect(self, rect): |
def ZoomOutToRect(self, rect): |
683 |
"""Zoom out to fit the currently visible region into rect. |
"""Zoom out to fit the currently visible region into rect. |
704 |
offx, offy = self.offset |
offx, offy = self.offset |
705 |
self.set_view_transform(self.scale, (offx + dx, offy + dy)) |
self.set_view_transform(self.scale, (offx + dx, offy + dy)) |
706 |
|
|
707 |
|
def SelectTool(self, tool): |
708 |
|
"""Make tool the active tool. |
709 |
|
|
710 |
|
The parameter should be an instance of Tool or None to indicate |
711 |
|
that no tool is active. |
712 |
|
""" |
713 |
|
self.tool = tool |
714 |
|
|
715 |
def ZoomInTool(self): |
def ZoomInTool(self): |
716 |
"""Start the zoom in tool""" |
"""Start the zoom in tool""" |
717 |
self.tool = ZoomInTool(self) |
self.SelectTool(ZoomInTool(self)) |
718 |
|
|
719 |
def ZoomOutTool(self): |
def ZoomOutTool(self): |
720 |
"""Start the zoom out tool""" |
"""Start the zoom out tool""" |
721 |
self.tool = ZoomOutTool(self) |
self.SelectTool(ZoomOutTool(self)) |
722 |
|
|
723 |
def PanTool(self): |
def PanTool(self): |
724 |
"""Start the pan tool""" |
"""Start the pan tool""" |
725 |
self.tool = PanTool(self) |
self.SelectTool(PanTool(self)) |
726 |
|
#img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM) |
727 |
|
#bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM) |
728 |
|
#print bmp |
729 |
|
#img = wxImageFromBitmap(bmp) |
730 |
|
#print img |
731 |
|
#cur = wxCursor(img) |
732 |
|
#print cur |
733 |
|
#self.SetCursor(cur) |
734 |
|
|
735 |
def IdentifyTool(self): |
def IdentifyTool(self): |
736 |
"""Start the identify tool""" |
"""Start the identify tool""" |
737 |
self.tool = IdentifyTool(self) |
self.SelectTool(IdentifyTool(self)) |
738 |
|
|
739 |
def LabelTool(self): |
def LabelTool(self): |
740 |
"""Start the label tool""" |
"""Start the label tool""" |
741 |
self.tool = LabelTool(self) |
self.SelectTool(LabelTool(self)) |
742 |
|
|
743 |
def CurrentTool(self): |
def CurrentTool(self): |
744 |
"""Return the name of the current tool or None if no tool is active""" |
"""Return the name of the current tool or None if no tool is active""" |
784 |
self.set_current_position(event) |
self.set_current_position(event) |
785 |
if self.dragging: |
if self.dragging: |
786 |
self.ReleaseMouse() |
self.ReleaseMouse() |
787 |
self.tool.Hide(self.drag_dc) |
try: |
788 |
self.tool.MouseUp(event) |
self.tool.Hide(self.drag_dc) |
789 |
self.drag_dc = None |
self.tool.MouseUp(event) |
790 |
self.dragging = 0 |
finally: |
791 |
|
self.drag_dc = None |
792 |
|
self.dragging = 0 |
793 |
|
|
794 |
def OnMotion(self, event): |
def OnMotion(self, event): |
795 |
self.set_current_position(event) |
self.set_current_position(event) |
801 |
def OnLeaveWindow(self, event): |
def OnLeaveWindow(self, event): |
802 |
self.set_current_position(None) |
self.set_current_position(None) |
803 |
|
|
|
def OnIdle(self, event): |
|
|
if self.redraw_on_idle: |
|
|
self.do_redraw() |
|
|
self.redraw_on_idle = 0 |
|
|
|
|
804 |
def OnSize(self, event): |
def OnSize(self, event): |
805 |
# the window's size has changed. We have to get a new bitmap. If |
# the window's size has changed. We have to get a new bitmap. If |
806 |
# we want to be clever we could try to get by without throwing |
# we want to be clever we could try to get by without throwing |
809 |
# Even when the window becomes larger some parts of the bitmap |
# Even when the window becomes larger some parts of the bitmap |
810 |
# could be reused. |
# could be reused. |
811 |
self.full_redraw() |
self.full_redraw() |
812 |
|
pass |
813 |
|
|
814 |
def shape_selected(self, layer, shape): |
def shape_selected(self, layer, shape): |
815 |
"""Redraw the map. |
"""Receiver for the SHAPES_SELECTED messages. Redraw the map.""" |
816 |
|
# The selection object takes care that it only issues |
817 |
Receiver for the SELECTED_SHAPE messages. Try to redraw only |
# SHAPES_SELECTED messages when the set of selected shapes has |
818 |
when necessary. |
# actually changed, so we can do a full redraw unconditionally. |
819 |
""" |
# FIXME: We should perhaps try to limit the redraw to the are |
820 |
# A redraw is necessary when the display has to change, which |
# actually covered by the shapes before and after the selection |
821 |
# means that either the status changes from having no selection |
# change. |
822 |
# to having a selection shape or vice versa, or when the fact |
self.full_redraw() |
|
# whether there is a selection at all doesn't change, when the |
|
|
# shape which is selected has changed (which means that layer or |
|
|
# shapeid changes). |
|
|
if ((shape is not None or self.last_selected_shape is not None) |
|
|
and (shape != self.last_selected_shape |
|
|
or layer != self.last_selected_layer)): |
|
|
self.full_redraw() |
|
|
|
|
|
# remember the selection so we can compare when it changes again. |
|
|
self.last_selected_layer = layer |
|
|
self.last_selected_shape = shape |
|
823 |
|
|
824 |
def unprojected_rect_around_point(self, x, y): |
def unprojected_rect_around_point(self, x, y, dist): |
825 |
"""return a rect a few pixels around (x, y) in unprojected corrdinates |
"""return a rect dist pixels around (x, y) in unprojected corrdinates |
826 |
|
|
827 |
The return value is a tuple (minx, miny, maxx, maxy) suitable a |
The return value is a tuple (minx, miny, maxx, maxy) suitable a |
828 |
parameter to a layer's ShapesInRegion method. |
parameter to a layer's ShapesInRegion method. |
836 |
xs = [] |
xs = [] |
837 |
ys = [] |
ys = [] |
838 |
for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)): |
for dx, dy in ((-1, -1), (1, -1), (1, 1), (-1, 1)): |
839 |
px, py = self.win_to_proj(x + dx, y + dy) |
px, py = self.win_to_proj(x + dist * dx, y + dist * dy) |
840 |
if inverse: |
if inverse: |
841 |
px, py = inverse(px, py) |
px, py = inverse(px, py) |
842 |
xs.append(px) |
xs.append(px) |
863 |
forward = None |
forward = None |
864 |
|
|
865 |
scale = self.scale |
scale = self.scale |
|
offx, offy = self.offset |
|
866 |
|
|
867 |
box = self.unprojected_rect_around_point(px, py) |
if scale == 0: |
868 |
|
return None, None |
869 |
|
|
870 |
|
offx, offy = self.offset |
871 |
|
|
872 |
if select_labels: |
if select_labels: |
873 |
labels = self.map.LabelLayer().Labels() |
labels = self.map.LabelLayer().Labels() |
915 |
if not layer.Visible(): |
if not layer.Visible(): |
916 |
continue |
continue |
917 |
|
|
918 |
filled = layer.fill is not None |
filled = layer.GetClassification().GetDefaultFill() \ |
919 |
stroked = layer.stroke is not None |
is not Color.Transparent |
920 |
|
stroked = layer.GetClassification().GetDefaultLineColor() \ |
921 |
|
is not Color.Transparent |
922 |
|
|
923 |
layer_proj = layer.projection |
layer_proj = layer.projection |
924 |
if layer_proj is not None: |
if layer_proj is not None: |
930 |
|
|
931 |
select_shape = -1 |
select_shape = -1 |
932 |
|
|
933 |
|
# Determine the ids of the shapes that overlap a tiny area |
934 |
|
# around the point. For layers containing points we have to |
935 |
|
# choose a larger size of the box we're testing agains so |
936 |
|
# that we take the size of the markers into account |
937 |
|
# FIXME: Once the markers are more flexible this part has to |
938 |
|
# become more flexible too, of course |
939 |
|
if shapetype == SHAPETYPE_POINT: |
940 |
|
box = self.unprojected_rect_around_point(px, py, 5) |
941 |
|
else: |
942 |
|
box = self.unprojected_rect_around_point(px, py, 1) |
943 |
shape_ids = layer.ShapesInRegion(box) |
shape_ids = layer.ShapesInRegion(box) |
944 |
shape_ids.reverse() |
shape_ids.reverse() |
945 |
|
|
946 |
if shapetype == SHAPETYPE_POLYGON: |
if shapetype == SHAPETYPE_POLYGON: |
947 |
for i in shape_ids: |
for i in shape_ids: |
948 |
result = point_in_polygon_shape(layer.shapefile.cobject(), |
shapefile = layer.ShapeStore().Shapefile().cobject() |
949 |
i, |
result = point_in_polygon_shape(shapefile, i, |
950 |
filled, stroked, |
filled, stroked, |
951 |
map_proj, layer_proj, |
map_proj, layer_proj, |
952 |
scale, -scale, offx, offy, |
scale, -scale, offx, offy, |
956 |
break |
break |
957 |
elif shapetype == SHAPETYPE_ARC: |
elif shapetype == SHAPETYPE_ARC: |
958 |
for i in shape_ids: |
for i in shape_ids: |
959 |
result = point_in_polygon_shape(layer.shapefile.cobject(), |
shapefile = layer.ShapeStore().Shapefile().cobject() |
960 |
|
result = point_in_polygon_shape(shapefile, |
961 |
i, 0, 1, |
i, 0, 1, |
962 |
map_proj, layer_proj, |
map_proj, layer_proj, |
963 |
scale, -scale, offx, offy, |
scale, -scale, offx, offy, |
998 |
# to deselect the currently selected layer, so we simply select |
# to deselect the currently selected layer, so we simply select |
999 |
# the already selected layer again. |
# the already selected layer again. |
1000 |
if layer is None: |
if layer is None: |
1001 |
layer = self.interactor.SelectedLayer() |
layer = self.selection.SelectedLayer() |
1002 |
self.interactor.SelectLayerAndShape(layer, shape) |
shapes = [] |
1003 |
|
else: |
1004 |
|
shapes = [shape] |
1005 |
|
self.selection.SelectShapes(layer, shapes) |
1006 |
return result |
return result |
1007 |
|
|
1008 |
def LabelShapeAt(self, x, y): |
def LabelShapeAt(self, x, y): |
1019 |
# a label was selected |
# a label was selected |
1020 |
label_layer.RemoveLabel(shape_index) |
label_layer.RemoveLabel(shape_index) |
1021 |
elif layer is not None: |
elif layer is not None: |
1022 |
text = labeldialog.run_label_dialog(self, layer.table, shape_index) |
text = labeldialog.run_label_dialog(self, |
1023 |
|
layer.ShapeStore().Table(), |
1024 |
|
shape_index) |
1025 |
if text: |
if text: |
1026 |
proj = self.map.projection |
proj = self.map.projection |
1027 |
if proj is not None: |
if proj is not None: |
1036 |
|
|
1037 |
shapetype = layer.ShapeType() |
shapetype = layer.ShapeType() |
1038 |
if shapetype == SHAPETYPE_POLYGON: |
if shapetype == SHAPETYPE_POLYGON: |
1039 |
x, y = shape_centroid(layer.shapefile.cobject(), |
shapefile = layer.ShapeStore().Shapefile().cobject() |
1040 |
shape_index, |
x, y = shape_centroid(shapefile, shape_index, |
1041 |
map_proj, layer_proj, 1, 1, 0, 0) |
map_proj, layer_proj, 1, 1, 0, 0) |
1042 |
if map_proj is not None: |
if map_proj is not None: |
1043 |
x, y = map_proj.Inverse(x, y) |
x, y = map_proj.Inverse(x, y) |
1062 |
valign = ALIGN_CENTER |
valign = ALIGN_CENTER |
1063 |
label_layer.AddLabel(x, y, text, |
label_layer.AddLabel(x, y, text, |
1064 |
halign = halign, valign = valign) |
halign = halign, valign = valign) |
1065 |
|
|
1066 |
|
def OutputTransform(canvas_scale, canvas_offset, canvas_size, device_extend): |
1067 |
|
"""Calculate dimensions to transform canvas content to output device.""" |
1068 |
|
width, height = device_extend |
1069 |
|
|
1070 |
|
# Only 80 % of the with are available for the map |
1071 |
|
width = width * 0.8 |
1072 |
|
|
1073 |
|
# Define the distance of the map from DC border |
1074 |
|
distance = 20 |
1075 |
|
|
1076 |
|
if height < width: |
1077 |
|
# landscape |
1078 |
|
map_height = height - 2*distance |
1079 |
|
map_width = map_height |
1080 |
|
else: |
1081 |
|
# portrait, recalibrate width (usually the legend width is too |
1082 |
|
# small |
1083 |
|
width = width * 0.9 |
1084 |
|
map_height = width - 2*distance |
1085 |
|
map_width = map_height |
1086 |
|
|
1087 |
|
mapregion = (distance, distance, |
1088 |
|
distance+map_width, distance+map_height) |
1089 |
|
|
1090 |
|
canvas_width, canvas_height = canvas_size |
1091 |
|
|
1092 |
|
scalex = map_width / (canvas_width/canvas_scale) |
1093 |
|
scaley = map_height / (canvas_height/canvas_scale) |
1094 |
|
scale = min(scalex, scaley) |
1095 |
|
canvas_offx, canvas_offy = canvas_offset |
1096 |
|
offx = scale*canvas_offx/canvas_scale |
1097 |
|
offy = scale*canvas_offy/canvas_scale |
1098 |
|
|
1099 |
|
return scale, (offx, offy), mapregion |