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: |
583 |
"""Return true if a layer is currently selected""" |
"""Return true if a layer is currently selected""" |
584 |
return self.canvas.HasSelectedLayer() |
return self.canvas.HasSelectedLayer() |
585 |
|
|
586 |
|
def has_selected_shape_layer(self): |
587 |
|
"""Return true if a shape layer is currently selected""" |
588 |
|
return isinstance(self.current_layer(), Layer) |
589 |
|
|
590 |
def has_selected_shapes(self): |
def has_selected_shapes(self): |
591 |
"""Return true if a shape is currently selected""" |
"""Return true if a shape is currently selected""" |
592 |
return self.canvas.HasSelectedShapes() |
return self.canvas.HasSelectedShapes() |
869 |
title = dlg.GetValue() |
title = dlg.GetValue() |
870 |
if title != "": |
if title != "": |
871 |
self.Map().SetTitle(title) |
self.Map().SetTitle(title) |
|
self.__SetTitle(title) |
|
872 |
|
|
873 |
dlg.Destroy() |
dlg.Destroy() |
874 |
|
|
910 |
# FIXME: bring dialog to front? |
# FIXME: bring dialog to front? |
911 |
pass |
pass |
912 |
|
|
913 |
def __SetTitle(self, title): |
def title_changed(self, map): |
914 |
self.SetTitle("Thuban - " + title) |
"""Subscribed to the canvas' TITLE_CHANGED messages""" |
915 |
|
self.update_title() |
916 |
|
|
917 |
|
def update_title(self): |
918 |
|
"""Update the window's title according to it's current state. |
919 |
|
|
920 |
|
In this default implementation the title is 'Thuban - ' followed |
921 |
|
by the map's title or simply 'Thuban' if there is not map. |
922 |
|
Derived classes should override this method to get different |
923 |
|
titles. |
924 |
|
|
925 |
|
This method is called automatically by other methods when the |
926 |
|
title may have to change. For the methods implemented in this |
927 |
|
class this usually only means that a different map has been set |
928 |
|
or the current map's title has changed. |
929 |
|
""" |
930 |
|
map = self.Map() |
931 |
|
if map is not None: |
932 |
|
title = _("Thuban - %s") % (map.Title(),) |
933 |
|
else: |
934 |
|
title = _("Thuban") |
935 |
|
self.SetTitle(title) |
936 |
|
|
937 |
|
|
938 |
# |
# |
939 |
# Define all the commands available in the main window |
# Define all the commands available in the main window |
975 |
"""Return true if a layer is selected in the context""" |
"""Return true if a layer is selected in the context""" |
976 |
return context.mainwindow.has_selected_layer() |
return context.mainwindow.has_selected_layer() |
977 |
|
|
978 |
|
def _has_selected_shape_layer(context): |
979 |
|
"""Return true if a shape layer is selected in the context""" |
980 |
|
return context.mainwindow.has_selected_shape_layer() |
981 |
|
|
982 |
def _has_selected_shapes(context): |
def _has_selected_shapes(context): |
983 |
"""Return true if a layer is selected in the context""" |
"""Return true if a layer is selected in the context""" |
984 |
return context.mainwindow.has_selected_shapes() |
return context.mainwindow.has_selected_shapes() |
1009 |
"""Return True if the GDAL is available""" |
"""Return True if the GDAL is available""" |
1010 |
return Thuban.Model.resource.has_gdal_support() |
return Thuban.Model.resource.has_gdal_support() |
1011 |
|
|
1012 |
|
def _has_dbconnections(context): |
1013 |
|
"""Return whether the the session has database connections""" |
1014 |
|
return context.session.HasDBConnections() |
1015 |
|
|
1016 |
|
def _has_postgis_support(context): |
1017 |
|
return has_postgis_support() |
1018 |
|
|
1019 |
|
|
1020 |
# File menu |
# File menu |
1021 |
_method_command("new_session", _("&New Session"), "NewSession", |
_method_command("new_session", _("&New Session"), "NewSession", |
1022 |
helptext = _("Start a new session")) |
helptext = _("Start a new session")) |
1032 |
_method_command("toggle_legend", _("Legend"), "ToggleLegend", |
_method_command("toggle_legend", _("Legend"), "ToggleLegend", |
1033 |
checked = _has_legend_shown, |
checked = _has_legend_shown, |
1034 |
helptext = _("Toggle Legend on/off")) |
helptext = _("Toggle Legend on/off")) |
1035 |
|
_method_command("database_management", _("&Database Connections..."), |
1036 |
|
"DatabaseManagement", |
1037 |
|
sensitive = _has_postgis_support) |
1038 |
_method_command("exit", _("E&xit"), "Exit", |
_method_command("exit", _("E&xit"), "Exit", |
1039 |
helptext = _("Finish working with Thuban")) |
helptext = _("Finish working with Thuban")) |
1040 |
|
|
1084 |
_method_command("rasterlayer_add", _("&Add Image Layer..."), "AddRasterLayer", |
_method_command("rasterlayer_add", _("&Add Image Layer..."), "AddRasterLayer", |
1085 |
helptext = _("Add a new image layer to the map"), |
helptext = _("Add a new image layer to the map"), |
1086 |
sensitive = _has_gdal_support) |
sensitive = _has_gdal_support) |
1087 |
|
_method_command("layer_add_db", _("Add &Database Layer..."), "AddDBLayer", |
1088 |
|
helptext = _("Add a new database layer to active map"), |
1089 |
|
sensitive = _has_dbconnections) |
1090 |
_method_command("layer_remove", _("&Remove Layer"), "RemoveLayer", |
_method_command("layer_remove", _("&Remove Layer"), "RemoveLayer", |
1091 |
helptext = _("Remove selected layer"), |
helptext = _("Remove selected layer"), |
1092 |
sensitive = _can_remove_layer) |
sensitive = _can_remove_layer) |
1115 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_layer) |
1116 |
_method_command("layer_show_table", _("Show Ta&ble"), "LayerShowTable", |
_method_command("layer_show_table", _("Show Ta&ble"), "LayerShowTable", |
1117 |
helptext = _("Show the selected layer's table"), |
helptext = _("Show the selected layer's table"), |
1118 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_shape_layer) |
1119 |
_method_command("layer_properties", _("&Properties..."), "LayerEditProperties", |
_method_command("layer_properties", _("&Properties..."), "LayerEditProperties", |
1120 |
sensitive = _has_selected_layer, |
sensitive = _has_selected_layer, |
1121 |
helptext = _("Edit the properties of the selected layer")) |
helptext = _("Edit the properties of the selected layer")) |
1122 |
_method_command("layer_jointable", _("&Join Table..."), "LayerJoinTable", |
_method_command("layer_jointable", _("&Join Table..."), "LayerJoinTable", |
1123 |
sensitive = _has_selected_layer, |
sensitive = _has_selected_shape_layer, |
1124 |
helptext = _("Join and attach a table to the selected layer")) |
helptext = _("Join and attach a table to the selected layer")) |
1125 |
|
|
1126 |
def _can_unjoin(context): |
def _can_unjoin(context): |
1162 |
helptext = _("Join two tables creating a new one")) |
helptext = _("Join two tables creating a new one")) |
1163 |
|
|
1164 |
# Export only under Windows ... |
# Export only under Windows ... |
1165 |
map_menu = ["layer_add", "rasterlayer_add", "layer_remove", |
map_menu = ["layer_add", "layer_add_db", "rasterlayer_add", "layer_remove", |
1166 |
None, |
None, |
1167 |
"map_rename", |
"map_rename", |
1168 |
"map_projection", |
"map_projection", |
1169 |
None, |
None, |
1170 |
"map_zoom_in_tool", "map_zoom_out_tool", |
"map_zoom_in_tool", "map_zoom_out_tool", |
1171 |
"map_pan_tool", |
"map_pan_tool", |
1172 |
"map_full_extent", |
"map_full_extent", |
1173 |
"layer_full_extent", |
"layer_full_extent", |
1174 |
"selected_full_extent", |
"selected_full_extent", |
1175 |
None, |
None, |
1186 |
[Menu("file", _("&File"), |
[Menu("file", _("&File"), |
1187 |
["new_session", "open_session", None, |
["new_session", "open_session", None, |
1188 |
"save_session", "save_session_as", None, |
"save_session", "save_session_as", None, |
1189 |
|
"database_management", None, |
1190 |
"toggle_session_tree", None, |
"toggle_session_tree", None, |
1191 |
"exit"]), |
"exit"]), |
1192 |
Menu("map", _("&Map"), map_menu), |
Menu("map", _("&Map"), map_menu), |