1 |
# Copyright (c) 2001, 2002, 2003 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 |
|
|
44 |
from Thuban.Lib.connector import Publisher |
from Thuban.Lib.connector import Publisher |
45 |
from Thuban.Model.color import Color |
from Thuban.Model.color import Color |
46 |
|
|
47 |
|
import resource |
48 |
|
|
49 |
from selection import Selection |
from selection import Selection |
50 |
from renderer import ScreenRenderer, PrinterRender |
from renderer import ScreenRenderer, ExportRenderer, PrinterRenderer |
51 |
|
|
52 |
import labeldialog |
import labeldialog |
53 |
|
|
54 |
from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION |
from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, \ |
55 |
|
SCALE_CHANGED |
56 |
|
|
57 |
|
|
58 |
# |
# |
232 |
self.view.LabelShapeAt(event.m_x, event.m_y) |
self.view.LabelShapeAt(event.m_x, event.m_y) |
233 |
|
|
234 |
|
|
|
|
|
|
|
|
235 |
class MapPrintout(wx.wxPrintout): |
class MapPrintout(wx.wxPrintout): |
236 |
|
|
237 |
""" |
""" |
238 |
wxPrintout class for printing Thuban maps |
wxPrintout class for printing Thuban maps |
239 |
""" |
""" |
240 |
|
|
241 |
def __init__(self, map): |
def __init__(self, canvas, map, region, selected_layer, selected_shapes): |
242 |
wx.wxPrintout.__init__(self) |
wx.wxPrintout.__init__(self) |
243 |
|
self.canvas = canvas |
244 |
self.map = map |
self.map = map |
245 |
|
self.region = region |
246 |
|
self.selected_layer = selected_layer |
247 |
|
self.selected_shapes = selected_shapes |
248 |
|
|
249 |
def GetPageInfo(self): |
def GetPageInfo(self): |
250 |
return (1, 1, 1, 1) |
return (1, 1, 1, 1) |
258 |
|
|
259 |
def draw_on_dc(self, dc): |
def draw_on_dc(self, dc): |
260 |
width, height = self.GetPageSizePixels() |
width, height = self.GetPageSizePixels() |
261 |
llx, lly, urx, ury = self.map.ProjectedBoundingBox() |
scale, offset, mapregion = OutputTransform(self.canvas.scale, |
262 |
scalex = width / (urx - llx) |
self.canvas.offset, |
263 |
scaley = height / (ury - lly) |
self.canvas.GetSizeTuple(), |
264 |
scale = min(scalex, scaley) |
self.GetPageSizePixels()) |
|
offx = 0.5 * (width - (urx + llx) * scale) |
|
|
offy = 0.5 * (height + (ury + lly) * scale) |
|
|
|
|
265 |
resx, resy = self.GetPPIPrinter() |
resx, resy = self.GetPPIPrinter() |
266 |
renderer = PrinterRender(dc, scale, (offx, offy), resolution = resx) |
renderer = PrinterRenderer(dc, scale, offset, resolution = resy) |
267 |
renderer.RenderMap(self.map) |
x, y, width, height = self.region |
268 |
|
canvas_scale = self.canvas.scale |
269 |
|
renderer.RenderMap(self.map, |
270 |
|
(0,0, |
271 |
|
(width/canvas_scale)*scale, |
272 |
|
(height/canvas_scale)*scale), |
273 |
|
mapregion, |
274 |
|
self.selected_layer, self.selected_shapes) |
275 |
return wx.true |
return wx.true |
276 |
|
|
277 |
|
|
280 |
"""A widget that displays a map and offers some interaction""" |
"""A widget that displays a map and offers some interaction""" |
281 |
|
|
282 |
# Some messages that can be subscribed/unsubscribed directly through |
# Some messages that can be subscribed/unsubscribed directly through |
283 |
# the MapCanvas come in fact from other objects. This is a map to |
# the MapCanvas come in fact from other objects. This is a dict |
284 |
# map those messages to the names of the instance variables they |
# mapping those messages to the names of the instance variables they |
285 |
# actually come from. This delegation is implemented in the |
# actually come from. The delegation is implemented in the Subscribe |
286 |
# Subscribe and unsubscribed methods |
# and Unsubscribe methods |
287 |
delegated_messages = {LAYER_SELECTED: "selection", |
delegated_messages = {LAYER_SELECTED: "selection", |
288 |
SHAPES_SELECTED: "selection"} |
SHAPES_SELECTED: "selection"} |
289 |
|
|
292 |
delegated_methods = {"SelectLayer": "selection", |
delegated_methods = {"SelectLayer": "selection", |
293 |
"SelectShapes": "selection", |
"SelectShapes": "selection", |
294 |
"SelectedLayer": "selection", |
"SelectedLayer": "selection", |
295 |
"HasSelectedLayer": "selection"} |
"HasSelectedLayer": "selection", |
296 |
|
"HasSelectedShapes": "selection", |
297 |
|
"SelectedShapes": "selection"} |
298 |
|
|
299 |
def __init__(self, parent, winid): |
def __init__(self, parent, winid): |
300 |
wxWindow.__init__(self, parent, winid) |
wxWindow.__init__(self, parent, winid) |
376 |
|
|
377 |
def OnPaint(self, event): |
def OnPaint(self, event): |
378 |
dc = wxPaintDC(self) |
dc = wxPaintDC(self) |
379 |
if self.map is not None and self.map.HasLayers(): |
clear = self.map is None or not self.map.HasLayers() |
380 |
self.do_redraw() |
|
381 |
else: |
#wxBeginBusyCursor() |
382 |
|
|
383 |
|
if not clear: |
384 |
|
try: |
385 |
|
self.do_redraw() |
386 |
|
except: |
387 |
|
print "Error during drawing:", sys.exc_info()[0] |
388 |
|
clear = True |
389 |
|
|
390 |
|
if clear: |
391 |
# If we've got no map or if the map is empty, simply clear |
# If we've got no map or if the map is empty, simply clear |
392 |
# the screen. |
# the screen. |
393 |
|
|
399 |
dc.Clear() |
dc.Clear() |
400 |
dc.EndDrawing() |
dc.EndDrawing() |
401 |
|
|
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. |
406 |
|
|
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 |
|
dc.Clear() |
425 |
|
|
426 |
selected_layer = self.selection.SelectedLayer() |
selected_layer = self.selection.SelectedLayer() |
427 |
selected_shapes = self.selection.SelectedShapes() |
selected_shapes = self.selection.SelectedShapes() |
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 hasattr(self, "export_path"): |
451 |
|
export_path = self.export_path |
452 |
|
else: |
453 |
|
export_path="." |
454 |
|
dlg = wxFileDialog(self, _("Export Map"), export_path, "", |
455 |
|
"Enhanced Metafile (*.wmf)|*.wmf", |
456 |
|
wxSAVE|wxOVERWRITE_PROMPT) |
457 |
|
if dlg.ShowModal() == wxID_OK: |
458 |
|
self.export_path = os.path.dirname(dlg.GetPath()) |
459 |
|
dc = wxMetaFileDC(dlg.GetPath()) |
460 |
|
|
461 |
|
scale, offset, mapregion = OutputTransform(self.scale, |
462 |
|
self.offset, |
463 |
|
self.GetSizeTuple(), |
464 |
|
dc.GetSizeTuple()) |
465 |
|
|
466 |
|
selected_layer = self.selection.SelectedLayer() |
467 |
|
selected_shapes = self.selection.SelectedShapes() |
468 |
|
|
469 |
|
renderer = ExportRenderer(dc, scale, offset) |
470 |
|
|
471 |
|
# Pass the entire bitmap as update region to the renderer. |
472 |
|
# We're redrawing the whole bitmap, after all. |
473 |
|
width, height = self.GetSizeTuple() |
474 |
|
renderer.RenderMap(self.map, |
475 |
|
(0,0, |
476 |
|
(width/self.scale)*scale, |
477 |
|
(height/self.scale)*scale), |
478 |
|
mapregion, |
479 |
|
selected_layer, selected_shapes) |
480 |
|
dc.EndDrawing() |
481 |
|
dc.Close() |
482 |
|
dlg.Destroy() |
483 |
|
|
484 |
def Print(self): |
def Print(self): |
485 |
printer = wx.wxPrinter() |
printer = wx.wxPrinter() |
486 |
printout = MapPrintout(self.map) |
width, height = self.GetSizeTuple() |
487 |
|
selected_layer = self.selection.SelectedLayer() |
488 |
|
selected_shapes = self.selection.SelectedShapes() |
489 |
|
|
490 |
|
printout = MapPrintout(self, self.map, (0, 0, width, height), |
491 |
|
selected_layer, selected_shapes) |
492 |
printer.Print(self, printout, wx.true) |
printer.Print(self, printout, wx.true) |
493 |
printout.Destroy() |
printout.Destroy() |
494 |
|
|
531 |
self.scale = scale |
self.scale = scale |
532 |
self.offset = offset |
self.offset = offset |
533 |
self.full_redraw() |
self.full_redraw() |
534 |
|
self.issue(SCALE_CHANGED, scale) |
535 |
|
|
536 |
def proj_to_win(self, x, y): |
def proj_to_win(self, x, y): |
537 |
"""\ |
"""\ |
556 |
width, height = self.GetSizeTuple() |
width, height = self.GetSizeTuple() |
557 |
llx, lly, urx, ury = rect |
llx, lly, urx, ury = rect |
558 |
if llx == urx or lly == ury: |
if llx == urx or lly == ury: |
559 |
# zero with or zero height. Do Nothing |
# zero width or zero height. Do Nothing |
560 |
return |
return |
561 |
scalex = width / (urx - llx) |
scalex = width / (urx - llx) |
562 |
scaley = height / (ury - lly) |
scaley = height / (ury - lly) |
575 |
if bbox is not None: |
if bbox is not None: |
576 |
self.FitRectToWindow(bbox) |
self.FitRectToWindow(bbox) |
577 |
|
|
578 |
|
def FitLayerToWindow(self, layer): |
579 |
|
"""Fit the given layer to the window. |
580 |
|
|
581 |
|
Set the scale so that the layer fits exactly into the window and |
582 |
|
center it in the window. |
583 |
|
""" |
584 |
|
|
585 |
|
bbox = layer.LatLongBoundingBox() |
586 |
|
if bbox is not None: |
587 |
|
proj = self.map.GetProjection() |
588 |
|
if proj is not None: |
589 |
|
bbox = proj.ForwardBBox(bbox) |
590 |
|
|
591 |
|
if bbox is not None: |
592 |
|
self.FitRectToWindow(bbox) |
593 |
|
|
594 |
|
def FitSelectedToWindow(self): |
595 |
|
layer = self.selection.SelectedLayer() |
596 |
|
shapes = self.selection.SelectedShapes() |
597 |
|
|
598 |
|
bbox = layer.ShapesBoundingBox(shapes) |
599 |
|
if bbox is not None: |
600 |
|
proj = self.map.GetProjection() |
601 |
|
if proj is not None: |
602 |
|
bbox = proj.ForwardBBox(bbox) |
603 |
|
|
604 |
|
if bbox is not None: |
605 |
|
self.FitRectToWindow(bbox) |
606 |
|
|
607 |
def ZoomFactor(self, factor, center = None): |
def ZoomFactor(self, factor, center = None): |
608 |
"""Multiply the zoom by factor and center on center. |
"""Multiply the zoom by factor and center on center. |
609 |
|
|
667 |
def PanTool(self): |
def PanTool(self): |
668 |
"""Start the pan tool""" |
"""Start the pan tool""" |
669 |
self.SelectTool(PanTool(self)) |
self.SelectTool(PanTool(self)) |
670 |
|
#img = resource.GetImageResource("pan", wxBITMAP_TYPE_XPM) |
671 |
|
#bmp = resource.GetBitmapResource("pan", wxBITMAP_TYPE_XPM) |
672 |
|
#print bmp |
673 |
|
#img = wxImageFromBitmap(bmp) |
674 |
|
#print img |
675 |
|
#cur = wxCursor(img) |
676 |
|
#print cur |
677 |
|
#self.SetCursor(cur) |
678 |
|
|
679 |
def IdentifyTool(self): |
def IdentifyTool(self): |
680 |
"""Start the identify tool""" |
"""Start the identify tool""" |
753 |
# Even when the window becomes larger some parts of the bitmap |
# Even when the window becomes larger some parts of the bitmap |
754 |
# could be reused. |
# could be reused. |
755 |
self.full_redraw() |
self.full_redraw() |
756 |
|
pass |
757 |
|
|
758 |
def shape_selected(self, layer, shape): |
def shape_selected(self, layer, shape): |
759 |
"""Receiver for the SHAPES_SELECTED messages. Redraw the map.""" |
"""Receiver for the SHAPES_SELECTED messages. Redraw the map.""" |
856 |
continue |
continue |
857 |
|
|
858 |
filled = layer.GetClassification().GetDefaultFill() \ |
filled = layer.GetClassification().GetDefaultFill() \ |
859 |
is not Color.None |
is not Color.Transparent |
860 |
stroked = layer.GetClassification().GetDefaultLineColor() \ |
stroked = layer.GetClassification().GetDefaultLineColor() \ |
861 |
is not Color.None |
is not Color.Transparent |
862 |
|
|
863 |
layer_proj = layer.projection |
layer_proj = layer.projection |
864 |
if layer_proj is not None: |
if layer_proj is not None: |
999 |
valign = ALIGN_CENTER |
valign = ALIGN_CENTER |
1000 |
label_layer.AddLabel(x, y, text, |
label_layer.AddLabel(x, y, text, |
1001 |
halign = halign, valign = valign) |
halign = halign, valign = valign) |
1002 |
|
|
1003 |
|
def OutputTransform(canvas_scale, canvas_offset, canvas_size, device_extend): |
1004 |
|
"""Calculate dimensions to transform canvas content to output device.""" |
1005 |
|
width, height = device_extend |
1006 |
|
|
1007 |
|
# Only 80 % of the with are available for the map |
1008 |
|
width = width * 0.8 |
1009 |
|
|
1010 |
|
# Define the distance of the map from DC border |
1011 |
|
distance = 20 |
1012 |
|
|
1013 |
|
if height < width: |
1014 |
|
# landscape |
1015 |
|
map_height = height - 2*distance |
1016 |
|
map_width = map_height |
1017 |
|
else: |
1018 |
|
# portrait, recalibrate width (usually the legend width is too |
1019 |
|
# small |
1020 |
|
width = width * 0.9 |
1021 |
|
map_height = width - 2*distance |
1022 |
|
map_width = map_height |
1023 |
|
|
1024 |
|
mapregion = (distance, distance, |
1025 |
|
distance+map_width, distance+map_height) |
1026 |
|
|
1027 |
|
canvas_width, canvas_height = canvas_size |
1028 |
|
|
1029 |
|
scalex = map_width / (canvas_width/canvas_scale) |
1030 |
|
scaley = map_height / (canvas_height/canvas_scale) |
1031 |
|
scale = min(scalex, scaley) |
1032 |
|
canvas_offx, canvas_offy = canvas_offset |
1033 |
|
offx = scale*canvas_offx/canvas_scale |
1034 |
|
offy = scale*canvas_offy/canvas_scale |
1035 |
|
|
1036 |
|
return scale, (offx, offy), mapregion |