12 |
""" |
""" |
13 |
|
|
14 |
__version__ = "$Revision$" |
__version__ = "$Revision$" |
15 |
|
# $Source$ |
16 |
__ThubanVersion__ = "0.8" #"$THUBAN_0_2$" |
# $Id$ |
|
#__BuildDate__ = "$Date$" |
|
17 |
|
|
18 |
import os |
import os |
19 |
import copy |
import copy |
20 |
|
|
21 |
from wxPython.wx import * |
from wxPython.wx import * |
|
from wxPython.wx import __version__ as wxPython_version |
|
22 |
|
|
23 |
import Thuban |
import Thuban |
|
import Thuban.version |
|
24 |
|
|
25 |
from Thuban import _ |
from Thuban import _ |
26 |
|
from Thuban.Model.messages import TITLE_CHANGED |
27 |
from Thuban.Model.session import create_empty_session |
from Thuban.Model.session import create_empty_session |
28 |
from Thuban.Model.layer import Layer, RasterLayer |
from Thuban.Model.layer import Layer, RasterLayer |
29 |
|
from Thuban.Model.postgisdb import PostGISShapeStore, has_postgis_support |
30 |
# XXX: replace this by |
# XXX: replace this by |
31 |
# from wxPython.lib.dialogs import wxMultipleChoiceDialog |
# from wxPython.lib.dialogs import wxMultipleChoiceDialog |
32 |
# when Thuban does not support wxPython 2.4.0 any more. |
# when Thuban does not support wxPython 2.4.0 any more. |
41 |
|
|
42 |
from context import Context |
from context import Context |
43 |
from command import registry, Command, ToolCommand |
from command import registry, Command, ToolCommand |
44 |
from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION |
from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, \ |
45 |
|
MAP_REPLACED |
46 |
from about import About |
from about import About |
47 |
|
|
48 |
from Thuban.UI.dock import DockFrame |
from Thuban.UI.dock import DockFrame |
49 |
from Thuban.UI.join import JoinDialog |
from Thuban.UI.join import JoinDialog |
50 |
|
from Thuban.UI.dbdialog import DBFrame, DBDialog, ChooseDBTableDialog |
51 |
import resource |
import resource |
52 |
import Thuban.Model.resource |
import Thuban.Model.resource |
53 |
|
|
62 |
# actually come from. This delegation is implemented in the |
# actually come from. This delegation is implemented in the |
63 |
# Subscribe and unsubscribed methods |
# Subscribe and unsubscribed methods |
64 |
delegated_messages = {LAYER_SELECTED: "canvas", |
delegated_messages = {LAYER_SELECTED: "canvas", |
65 |
SHAPES_SELECTED: "canvas"} |
SHAPES_SELECTED: "canvas", |
66 |
|
MAP_REPLACED: "canvas"} |
67 |
|
|
68 |
# Methods delegated to some instance variables. The delegation is |
# Methods delegated to some instance variables. The delegation is |
69 |
# implemented in the __getattr__ method. |
# implemented in the __getattr__ method. |
102 |
canvas.Subscribe(VIEW_POSITION, self.view_position_changed) |
canvas.Subscribe(VIEW_POSITION, self.view_position_changed) |
103 |
canvas.Subscribe(SHAPES_SELECTED, self.identify_view_on_demand) |
canvas.Subscribe(SHAPES_SELECTED, self.identify_view_on_demand) |
104 |
self.canvas = canvas |
self.canvas = canvas |
105 |
|
self.canvas.Subscribe(TITLE_CHANGED, self.title_changed) |
106 |
|
|
107 |
self.SetMainWindow(self.canvas) |
self.SetMainWindow(self.canvas) |
108 |
|
|
134 |
""" |
""" |
135 |
if channel in self.delegated_messages: |
if channel in self.delegated_messages: |
136 |
object = getattr(self, self.delegated_messages[channel]) |
object = getattr(self, self.delegated_messages[channel]) |
137 |
object.Unsubscribe(channel, *args) |
try: |
138 |
|
object.Unsubscribe(channel, *args) |
139 |
|
except wxPyDeadObjectError: |
140 |
|
# The object was a wxObject and has already been |
141 |
|
# destroyed. Hopefully it has unsubscribed all its |
142 |
|
# subscribers already so that it's OK if we do nothing |
143 |
|
# here |
144 |
|
pass |
145 |
|
|
146 |
def __getattr__(self, attr): |
def __getattr__(self, attr): |
147 |
"""If attr is one of the delegated methods return that method |
"""If attr is one of the delegated methods return that method |
372 |
result = wxID_NO |
result = wxID_NO |
373 |
return result |
return result |
374 |
|
|
|
def prepare_new_session(self): |
|
|
for d in self.dialogs.values(): |
|
|
if not isinstance(d, tree.SessionTreeView): |
|
|
d.Close() |
|
|
|
|
375 |
def NewSession(self): |
def NewSession(self): |
376 |
if self.save_modified_session() != wxID_CANCEL: |
if self.save_modified_session() != wxID_CANCEL: |
|
self.prepare_new_session() |
|
377 |
self.application.SetSession(create_empty_session()) |
self.application.SetSession(create_empty_session()) |
378 |
|
|
379 |
def OpenSession(self): |
def OpenSession(self): |
382 |
"Thuban Session File (*.thuban)|*.thuban", |
"Thuban Session File (*.thuban)|*.thuban", |
383 |
wxOPEN) |
wxOPEN) |
384 |
if dlg.ShowModal() == wxID_OK: |
if dlg.ShowModal() == wxID_OK: |
385 |
self.prepare_new_session() |
self.application.OpenSession(dlg.GetPath(), |
386 |
self.application.OpenSession(dlg.GetPath()) |
self.run_db_param_dialog) |
387 |
dlg.Destroy() |
dlg.Destroy() |
388 |
|
|
389 |
|
def run_db_param_dialog(self, parameters, message): |
390 |
|
dlg = DBDialog(self, _("DB Connection Parameters"), parameters, |
391 |
|
message) |
392 |
|
return dlg.RunDialog() |
393 |
|
|
394 |
def SaveSession(self): |
def SaveSession(self): |
395 |
if self.application.session.filename == None: |
if self.application.session.filename == None: |
396 |
self.SaveSessionAs() |
self.SaveSessionAs() |
426 |
|
|
427 |
def SetMap(self, map): |
def SetMap(self, map): |
428 |
self.canvas.SetMap(map) |
self.canvas.SetMap(map) |
429 |
self.__SetTitle(map.Title()) |
self.update_title() |
430 |
|
|
431 |
dialog = self.FindRegisteredDock("legend") |
dialog = self.FindRegisteredDock("legend") |
432 |
if dialog is not None: |
if dialog is not None: |
457 |
dlg.ShowModal() |
dlg.ShowModal() |
458 |
dlg.Destroy() |
dlg.Destroy() |
459 |
|
|
460 |
|
def DatabaseManagement(self): |
461 |
|
name = "dbmanagement" |
462 |
|
dialog = self.get_open_dialog(name) |
463 |
|
if dialog is None: |
464 |
|
map = self.canvas.Map() |
465 |
|
dialog = DBFrame(self, name, self.application.Session()) |
466 |
|
self.add_dialog(name, dialog) |
467 |
|
dialog.Show() |
468 |
|
dialog.Raise() |
469 |
|
|
470 |
def AddLayer(self): |
def AddLayer(self): |
471 |
dlg = wxFileDialog(self, _("Select a data file"), ".", "", "*.*", |
dlg = wxFileDialog(self, _("Select one or more data files"), ".", "", |
472 |
wxOPEN) |
_("Shapefiles (*.shp)") + "|*.shp;*.SHP|" + |
473 |
|
_("All Files (*.*)") + "|*.*", |
474 |
|
wxOPEN | wxMULTIPLE) |
475 |
if dlg.ShowModal() == wxID_OK: |
if dlg.ShowModal() == wxID_OK: |
476 |
filename = dlg.GetPath() |
filenames = dlg.GetPaths() |
477 |
title = os.path.splitext(os.path.basename(filename))[0] |
for filename in filenames: |
478 |
map = self.canvas.Map() |
title = os.path.splitext(os.path.basename(filename))[0] |
479 |
has_layers = map.HasLayers() |
map = self.canvas.Map() |
480 |
try: |
has_layers = map.HasLayers() |
481 |
store = self.application.Session().OpenShapefile(filename) |
try: |
482 |
except IOError: |
store = self.application.Session().OpenShapefile(filename) |
483 |
# the layer couldn't be opened |
except IOError: |
484 |
self.RunMessageBox(_("Add Layer"), |
# the layer couldn't be opened |
485 |
_("Can't open the file '%s'.") % filename) |
self.RunMessageBox(_("Add Layer"), |
486 |
else: |
_("Can't open the file '%s'.")%filename) |
487 |
layer = Layer(title, store) |
else: |
488 |
map.AddLayer(layer) |
layer = Layer(title, store) |
489 |
if not has_layers: |
map.AddLayer(layer) |
490 |
# if we're adding a layer to an empty map, fit the |
if not has_layers: |
491 |
# new map to the window |
# if we're adding a layer to an empty map, fit the |
492 |
self.canvas.FitMapToWindow() |
# new map to the window |
493 |
|
self.canvas.FitMapToWindow() |
494 |
dlg.Destroy() |
dlg.Destroy() |
495 |
|
|
496 |
def AddRasterLayer(self): |
def AddRasterLayer(self): |
515 |
self.canvas.FitMapToWindow() |
self.canvas.FitMapToWindow() |
516 |
dlg.Destroy() |
dlg.Destroy() |
517 |
|
|
518 |
|
def AddDBLayer(self): |
519 |
|
"""Add a layer read from a database""" |
520 |
|
session = self.application.Session() |
521 |
|
dlg = ChooseDBTableDialog(self, self.application.Session()) |
522 |
|
|
523 |
|
if dlg.ShowModal() == wxID_OK: |
524 |
|
dbconn, dbtable = dlg.GetTable() |
525 |
|
try: |
526 |
|
title = str(dbtable) |
527 |
|
|
528 |
|
# Chose the correct Interface for the database type |
529 |
|
store = PostGISShapeStore(dbconn, dbtable) |
530 |
|
session.AddShapeStore(store) |
531 |
|
layer = Layer(title, store) |
532 |
|
except: |
533 |
|
# Some error occured while initializing the layer |
534 |
|
self.RunMessageBox(_("Add Layer from database"), |
535 |
|
_("Can't open the database table '%s'") |
536 |
|
% dbtable) |
537 |
|
|
538 |
|
map = self.canvas.Map() |
539 |
|
|
540 |
|
has_layers = map.HasLayers() |
541 |
|
map.AddLayer(layer) |
542 |
|
if not has_layers: |
543 |
|
self.canvas.FitMapToWindow() |
544 |
|
|
545 |
|
dlg.Destroy() |
546 |
|
|
547 |
def RemoveLayer(self): |
def RemoveLayer(self): |
548 |
layer = self.current_layer() |
layer = self.current_layer() |
549 |
if layer is not None: |
if layer is not None: |
712 |
else: |
else: |
713 |
dialog.Show(not dialog.IsShown()) |
dialog.Show(not dialog.IsShown()) |
714 |
|
|
|
self.canvas.FitMapToWindow() |
|
|
|
|
715 |
def LegendShown(self): |
def LegendShown(self): |
716 |
"""Return true iff the legend is currently open""" |
"""Return true iff the legend is currently open""" |
717 |
dialog = self.FindRegisteredDock("legend") |
dialog = self.FindRegisteredDock("legend") |
865 |
title = dlg.GetValue() |
title = dlg.GetValue() |
866 |
if title != "": |
if title != "": |
867 |
self.Map().SetTitle(title) |
self.Map().SetTitle(title) |
|
self.__SetTitle(title) |
|
868 |
|
|
869 |
dlg.Destroy() |
dlg.Destroy() |
870 |
|
|
906 |
# FIXME: bring dialog to front? |
# FIXME: bring dialog to front? |
907 |
pass |
pass |
908 |
|
|
909 |
def __SetTitle(self, title): |
def title_changed(self, map): |
910 |
self.SetTitle("Thuban - " + title) |
"""Subscribed to the canvas' TITLE_CHANGED messages""" |
911 |
|
self.update_title() |
912 |
|
|
913 |
|
def update_title(self): |
914 |
|
"""Update the window's title according to it's current state. |
915 |
|
|
916 |
|
In this default implementation the title is 'Thuban - ' followed |
917 |
|
by the map's title or simply 'Thuban' if there is not map. |
918 |
|
Derived classes should override this method to get different |
919 |
|
titles. |
920 |
|
|
921 |
|
This method is called automatically by other methods when the |
922 |
|
title may have to change. For the methods implemented in this |
923 |
|
class this usually only means that a different map has been set |
924 |
|
or the current map's title has changed. |
925 |
|
""" |
926 |
|
map = self.Map() |
927 |
|
if map is not None: |
928 |
|
title = _("Thuban - %s") % (map.Title(),) |
929 |
|
else: |
930 |
|
title = _("Thuban") |
931 |
|
self.SetTitle(title) |
932 |
|
|
933 |
|
|
934 |
# |
# |
935 |
# Define all the commands available in the main window |
# Define all the commands available in the main window |
1001 |
"""Return True if the GDAL is available""" |
"""Return True if the GDAL is available""" |
1002 |
return Thuban.Model.resource.has_gdal_support() |
return Thuban.Model.resource.has_gdal_support() |
1003 |
|
|
1004 |
|
def _has_dbconnections(context): |
1005 |
|
"""Return whether the the session has database connections""" |
1006 |
|
return context.session.HasDBConnections() |
1007 |
|
|
1008 |
|
def _has_postgis_support(context): |
1009 |
|
return has_postgis_support() |
1010 |
|
|
1011 |
|
|
1012 |
# File menu |
# File menu |
1013 |
_method_command("new_session", _("&New Session"), "NewSession", |
_method_command("new_session", _("&New Session"), "NewSession", |
1014 |
helptext = _("Start a new session")) |
helptext = _("Start a new session")) |
1024 |
_method_command("toggle_legend", _("Legend"), "ToggleLegend", |
_method_command("toggle_legend", _("Legend"), "ToggleLegend", |
1025 |
checked = _has_legend_shown, |
checked = _has_legend_shown, |
1026 |
helptext = _("Toggle Legend on/off")) |
helptext = _("Toggle Legend on/off")) |
1027 |
|
_method_command("database_management", _("&Database Connections..."), |
1028 |
|
"DatabaseManagement", |
1029 |
|
sensitive = _has_postgis_support) |
1030 |
_method_command("exit", _("E&xit"), "Exit", |
_method_command("exit", _("E&xit"), "Exit", |
1031 |
helptext = _("Finish working with Thuban")) |
helptext = _("Finish working with Thuban")) |
1032 |
|
|
1076 |
_method_command("rasterlayer_add", _("&Add Image Layer..."), "AddRasterLayer", |
_method_command("rasterlayer_add", _("&Add Image Layer..."), "AddRasterLayer", |
1077 |
helptext = _("Add a new image layer to the map"), |
helptext = _("Add a new image layer to the map"), |
1078 |
sensitive = _has_gdal_support) |
sensitive = _has_gdal_support) |
1079 |
|
_method_command("layer_add_db", _("Add &Database Layer..."), "AddDBLayer", |
1080 |
|
helptext = _("Add a new database layer to active map"), |
1081 |
|
sensitive = _has_dbconnections) |
1082 |
_method_command("layer_remove", _("&Remove Layer"), "RemoveLayer", |
_method_command("layer_remove", _("&Remove Layer"), "RemoveLayer", |
1083 |
helptext = _("Remove selected layer"), |
helptext = _("Remove selected layer"), |
1084 |
sensitive = _can_remove_layer) |
sensitive = _can_remove_layer) |
1154 |
helptext = _("Join two tables creating a new one")) |
helptext = _("Join two tables creating a new one")) |
1155 |
|
|
1156 |
# Export only under Windows ... |
# Export only under Windows ... |
1157 |
map_menu = ["layer_add", "rasterlayer_add", "layer_remove", |
map_menu = ["layer_add", "layer_add_db", "rasterlayer_add", "layer_remove", |
1158 |
None, |
None, |
1159 |
"map_rename", |
"map_rename", |
1160 |
"map_projection", |
"map_projection", |
1161 |
None, |
None, |
1162 |
"map_zoom_in_tool", "map_zoom_out_tool", |
"map_zoom_in_tool", "map_zoom_out_tool", |
1163 |
"map_pan_tool", |
"map_pan_tool", |
1164 |
"map_full_extent", |
"map_full_extent", |
1165 |
"layer_full_extent", |
"layer_full_extent", |
1166 |
"selected_full_extent", |
"selected_full_extent", |
1167 |
None, |
None, |
1178 |
[Menu("file", _("&File"), |
[Menu("file", _("&File"), |
1179 |
["new_session", "open_session", None, |
["new_session", "open_session", None, |
1180 |
"save_session", "save_session_as", None, |
"save_session", "save_session_as", None, |
1181 |
|
"database_management", None, |
1182 |
"toggle_session_tree", None, |
"toggle_session_tree", None, |
1183 |
"exit"]), |
"exit"]), |
1184 |
Menu("map", _("&Map"), map_menu), |
Menu("map", _("&Map"), map_menu), |