/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/UI/mainwindow.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/UI/mainwindow.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1155 by jan, Thu Jun 12 12:17:11 2003 UTC revision 2040 by bh, Tue Dec 23 11:37:04 2003 UTC
# Line 12  The main window Line 12  The main window
12  """  """
13    
14  __version__ = "$Revision$"  __version__ = "$Revision$"
15    # $Source$
16  __ThubanVersion__ = "0.2" #"$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.
# Line 43  from menu import Menu Line 41  from menu import Menu
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
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
53    
54  import projdialog  import projdialog
55    
56    
   
57  class MainWindow(DockFrame):  class MainWindow(DockFrame):
58    
59      # Some messages that can be subscribed/unsubscribed directly through      # Some messages that can be subscribed/unsubscribed directly through
# Line 62  class MainWindow(DockFrame): Line 62  class MainWindow(DockFrame):
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.
# Line 101  class MainWindow(DockFrame): Line 102  class MainWindow(DockFrame):
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    
# Line 108  class MainWindow(DockFrame): Line 110  class MainWindow(DockFrame):
110    
111          self.init_dialogs()          self.init_dialogs()
112    
113            self.ShowLegend()
114    
115          EVT_CLOSE(self, self.OnClose)          EVT_CLOSE(self, self.OnClose)
116    
117      def Subscribe(self, channel, *args):      def Subscribe(self, channel, *args):
# Line 130  class MainWindow(DockFrame): Line 134  class MainWindow(DockFrame):
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
# Line 361  class MainWindow(DockFrame): Line 372  class MainWindow(DockFrame):
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):
# Line 377  class MainWindow(DockFrame): Line 382  class MainWindow(DockFrame):
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()
# Line 416  class MainWindow(DockFrame): Line 426  class MainWindow(DockFrame):
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:
# Line 443  class MainWindow(DockFrame): Line 453  class MainWindow(DockFrame):
453          return self.get_open_dialog("session_tree") is not None          return self.get_open_dialog("session_tree") is not None
454    
455      def About(self):      def About(self):
456          self.RunMessageBox(_("About"),          dlg = About(self)
457                             _("Thuban %s\n"          dlg.ShowModal()
458                              #"Build Date: %s\n"          dlg.Destroy()
459                              "using:\n"  
460                              "  %s\n"      def DatabaseManagement(self):
461                              "  %s\n\n"          name = "dbmanagement"
462                              "Thuban is a program for\n"          dialog = self.get_open_dialog(name)
463                              "exploring geographic data.\n"          if dialog is None:
464                              "Copyright (C) 2001-2003 Intevation GmbH.\n"              map = self.canvas.Map()
465                              "Thuban is licensed under the GNU GPL"              dialog = DBFrame(self, name, self.application.Session())
466                              % (Thuban.version.longversion,              self.add_dialog(name, dialog)
467                                 "wxPython %s" % wxPython_version,              dialog.Show()
468                                 "Python %d.%d.%d" % sys.version_info[:3]          dialog.Raise()
                               )),  
 #                           % __ThubanVersion__), #__BuildDate__)),  
                            wxOK | wxICON_INFORMATION)  
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):
# Line 505  class MainWindow(DockFrame): Line 515  class MainWindow(DockFrame):
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:
# Line 544  class MainWindow(DockFrame): Line 583  class MainWindow(DockFrame):
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()
# Line 578  class MainWindow(DockFrame): Line 621  class MainWindow(DockFrame):
621      def LayerShowTable(self):      def LayerShowTable(self):
622          layer = self.current_layer()          layer = self.current_layer()
623          if layer is not None:          if layer is not None:
624              table = layer.table              table = layer.ShapeStore().Table()
625              name = "table_view" + str(id(table))              name = "table_view" + str(id(table))
626              dialog = self.get_open_dialog(name)              dialog = self.get_open_dialog(name)
627              if dialog is None:              if dialog is None:
# Line 673  class MainWindow(DockFrame): Line 716  class MainWindow(DockFrame):
716          else:          else:
717              dialog.Show(not dialog.IsShown())              dialog.Show(not dialog.IsShown())
718    
         self.canvas.FitMapToWindow()  
   
719      def LegendShown(self):      def LegendShown(self):
720          """Return true iff the legend is currently open"""          """Return true iff the legend is currently open"""
721          dialog = self.FindRegisteredDock("legend")          dialog = self.FindRegisteredDock("legend")
# Line 750  class MainWindow(DockFrame): Line 791  class MainWindow(DockFrame):
791                                                 table)                                                 table)
792              self.add_dialog(name, dialog)              self.add_dialog(name, dialog)
793              dialog.Show(True)              dialog.Show(True)
794          # FIXME: else bring dialog to front          dialog.Raise()
795    
796      def TableRename(self):      def TableRename(self):
797          """Let the user rename a table"""          """Let the user rename a table"""
# Line 773  class MainWindow(DockFrame): Line 814  class MainWindow(DockFrame):
814    
815          # Second, let the user rename the layers          # Second, let the user rename the layers
816          for table in to_rename:          for table in to_rename:
817              dlg = wxTextEntryDialog(self, "Table Title: ", "Rename Table",              dlg = wxTextEntryDialog(self, _("Table Title:"), _("Rename Table"),
818                                      table.Title())                                      table.Title())
819              try:              try:
820                  if dlg.ShowModal() == wxID_OK:                  if dlg.ShowModal() == wxID_OK:
# Line 822  class MainWindow(DockFrame): Line 863  class MainWindow(DockFrame):
863          self.canvas.Print()          self.canvas.Print()
864    
865      def RenameMap(self):      def RenameMap(self):
866          dlg = wxTextEntryDialog(self, "Map Title: ", "Rename Map",          dlg = wxTextEntryDialog(self, _("Map Title:"), _("Rename Map"),
867                                  self.Map().Title())                                  self.Map().Title())
868          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
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    
# Line 836  class MainWindow(DockFrame): Line 876  class MainWindow(DockFrame):
876          """Let the user rename the currently selected layer"""          """Let the user rename the currently selected layer"""
877          layer = self.current_layer()          layer = self.current_layer()
878          if layer is not None:          if layer is not None:
879              dlg = wxTextEntryDialog(self, "Layer Title: ", "Rename Layer",              dlg = wxTextEntryDialog(self, _("Layer Title:"), _("Rename Layer"),
880                                      layer.Title())                                      layer.Title())
881              try:              try:
882                  if dlg.ShowModal() == wxID_OK:                  if dlg.ShowModal() == wxID_OK:
# Line 870  class MainWindow(DockFrame): Line 910  class MainWindow(DockFrame):
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
# Line 913  def _has_selected_layer(context): Line 975  def _has_selected_layer(context):
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()
# Line 939  def _has_legend_shown(context): Line 1005  def _has_legend_shown(context):
1005      """Return true if the legend window is shown"""      """Return true if the legend window is shown"""
1006      return context.mainwindow.LegendShown()      return context.mainwindow.LegendShown()
1007    
1008    def _has_gdal_support(context):
1009        """Return True if the GDAL is available"""
1010        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",
# Line 955  _method_command("toggle_session_tree", _ Line 1032  _method_command("toggle_session_tree", _
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    
# Line 1002  _method_command("map_rename", _("&Rename Line 1082  _method_command("map_rename", _("&Rename
1082  _method_command("layer_add", _("&Add Layer..."), "AddLayer",  _method_command("layer_add", _("&Add Layer..."), "AddLayer",
1083                  helptext = _("Add a new layer to the map"))                  helptext = _("Add a new layer to the map"))
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)
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)
# Line 1031  _method_command("layer_hide", _("&Hide") Line 1115  _method_command("layer_hide", _("&Hide")
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):
# Line 1078  _method_command("table_join", _("&Join.. Line 1162  _method_command("table_join", _("&Join..
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_rename",  map_menu = ["layer_add", "layer_add_db", "rasterlayer_add", "layer_remove",
1166                          None,                          None,
1167                            "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,
# Line 1101  main_menu = Menu("<main>", "<main>", Line 1186  main_menu = Menu("<main>", "<main>",
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),
# Line 1136  main_toolbar = Menu("<toolbar>", "<toolb Line 1222  main_toolbar = Menu("<toolbar>", "<toolb
1222                       "selected_full_extent",                       "selected_full_extent",
1223                       None,                       None,
1224                       "map_identify_tool", "map_label_tool"])                       "map_identify_tool", "map_label_tool"])
1225    
Legend:
Removed from v.1155  
changed lines
  Added in v.2040
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26