/[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 1620 by bh, Wed Aug 20 13:14:22 2003 UTC revision 2186 by jan, Sun Apr 18 20:37:01 2004 UTC
# Line 1  Line 1 
1  # Copyright (C) 2001, 2002, 2003 by Intevation GmbH  # Copyright (C) 2001, 2002, 2003, 2004 by Intevation GmbH
2  # Authors:  # Authors:
3  # Jan-Oliver Wagner <[email protected]>  # Jan-Oliver Wagner <[email protected]>
4  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
# Line 12  The main window Line 12  The main window
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  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 49  from about import About Line 47  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, ChooseDBTableDialog  from Thuban.UI.dbdialog import DBFrame, DBDialog, ChooseDBTableDialog
51  import resource  import resource
52  import Thuban.Model.resource  import Thuban.Model.resource
53    
# Line 104  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 135  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 372  class MainWindow(DockFrame): Line 378  class MainWindow(DockFrame):
378    
379      def OpenSession(self):      def OpenSession(self):
380          if self.save_modified_session() != wxID_CANCEL:          if self.save_modified_session() != wxID_CANCEL:
381              dlg = wxFileDialog(self, _("Open Session"), ".", "",              dlg = wxFileDialog(self, _("Open Session"),
382                                   self.application.Path("data"), "",
383                                 "Thuban Session File (*.thuban)|*.thuban",                                 "Thuban Session File (*.thuban)|*.thuban",
384                                 wxOPEN)                                 wxOPEN)
385              if dlg.ShowModal() == wxID_OK:              if dlg.ShowModal() == wxID_OK:
386                  self.application.OpenSession(dlg.GetPath())                  self.application.OpenSession(dlg.GetPath(),
387                                                 self.run_db_param_dialog)
388                    self.application.SetPath("data", dlg.GetPath())
389              dlg.Destroy()              dlg.Destroy()
390    
391        def run_db_param_dialog(self, parameters, message):
392            dlg = DBDialog(self, _("DB Connection Parameters"), parameters,
393                           message)
394            return dlg.RunDialog()
395    
396      def SaveSession(self):      def SaveSession(self):
397          if self.application.session.filename == None:          if self.application.session.filename == None:
398              self.SaveSessionAs()              self.SaveSessionAs()
# Line 386  class MainWindow(DockFrame): Line 400  class MainWindow(DockFrame):
400              self.application.SaveSession()              self.application.SaveSession()
401    
402      def SaveSessionAs(self):      def SaveSessionAs(self):
403          dlg = wxFileDialog(self, _("Save Session As"), ".", "",          dlg = wxFileDialog(self, _("Save Session As"),
404                               self.application.Path("data"), "",
405                             "Thuban Session File (*.thuban)|*.thuban",                             "Thuban Session File (*.thuban)|*.thuban",
406                             wxSAVE|wxOVERWRITE_PROMPT)                             wxSAVE|wxOVERWRITE_PROMPT)
407          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
408              self.application.session.SetFilename(dlg.GetPath())              self.application.session.SetFilename(dlg.GetPath())
409              self.application.SaveSession()              self.application.SaveSession()
410                self.application.SetPath("data",dlg.GetPath())
411          dlg.Destroy()          dlg.Destroy()
412    
413      def Exit(self):      def Exit(self):
# Line 414  class MainWindow(DockFrame): Line 430  class MainWindow(DockFrame):
430    
431      def SetMap(self, map):      def SetMap(self, map):
432          self.canvas.SetMap(map)          self.canvas.SetMap(map)
433          self.__SetTitle(map.Title())          self.update_title()
434    
435          dialog = self.FindRegisteredDock("legend")          dialog = self.FindRegisteredDock("legend")
436          if dialog is not None:          if dialog is not None:
# Line 456  class MainWindow(DockFrame): Line 472  class MainWindow(DockFrame):
472          dialog.Raise()          dialog.Raise()
473    
474      def AddLayer(self):      def AddLayer(self):
475          dlg = wxFileDialog(self, _("Select a data file"), ".", "", "*.*",          dlg = wxFileDialog(self, _("Select one or more data files"),
476                             wxOPEN)                             self.application.Path("data"), "",
477                               _("Shapefiles (*.shp)") + "|*.shp;*.SHP|" +
478                               _("All Files (*.*)") + "|*.*",
479                               wxOPEN | wxMULTIPLE)
480          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
481              filename = dlg.GetPath()              filenames = dlg.GetPaths()
482              title = os.path.splitext(os.path.basename(filename))[0]              for filename in filenames:
483              map = self.canvas.Map()                  title = os.path.splitext(os.path.basename(filename))[0]
484              has_layers = map.HasLayers()                  map = self.canvas.Map()
485              try:                  has_layers = map.HasLayers()
486                  store = self.application.Session().OpenShapefile(filename)                  try:
487              except IOError:                      store = self.application.Session().OpenShapefile(filename)
488                  # the layer couldn't be opened                  except IOError:
489                  self.RunMessageBox(_("Add Layer"),                      # the layer couldn't be opened
490                                     _("Can't open the file '%s'.") % filename)                      self.RunMessageBox(_("Add Layer"),
491              else:                                         _("Can't open the file '%s'.")%filename)
492                  layer = Layer(title, store)                  else:
493                  map.AddLayer(layer)                      layer = Layer(title, store)
494                  if not has_layers:                      map.AddLayer(layer)
495                      # if we're adding a layer to an empty map, fit the                      if not has_layers:
496                      # new map to the window                          # if we're adding a layer to an empty map, fit the
497                      self.canvas.FitMapToWindow()                          # new map to the window
498                            self.canvas.FitMapToWindow()
499                        self.application.SetPath("data",filename)
500          dlg.Destroy()          dlg.Destroy()
501    
502      def AddRasterLayer(self):      def AddRasterLayer(self):
503          dlg = wxFileDialog(self, _("Select an image file"), ".", "", "*.*",          dlg = wxFileDialog(self, _("Select an image file"),
504                               self.application.Path("data"), "", "*.*",
505                             wxOPEN)                             wxOPEN)
506          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
507              filename = dlg.GetPath()              filename = dlg.GetPath()
# Line 498  class MainWindow(DockFrame): Line 520  class MainWindow(DockFrame):
520                      # if we're adding a layer to an empty map, fit the                      # if we're adding a layer to an empty map, fit the
521                      # new map to the window                      # new map to the window
522                      self.canvas.FitMapToWindow()                      self.canvas.FitMapToWindow()
523                    self.application.SetPath("data", filename)
524          dlg.Destroy()          dlg.Destroy()
525    
526      def AddDBLayer(self):      def AddDBLayer(self):
527          """Add a layer read from a database"""          """Add a layer read from a database"""
528          session = self.application.Session()          session = self.application.Session()
529          dlg = ChooseDBTableDialog(self.application.Session(), self,-1, "")          dlg = ChooseDBTableDialog(self, self.application.Session())
530    
531          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
532              dbconn, dbtable = dlg.GetTable()              dbconn, dbtable, id_column, geo_column = dlg.GetTable()
533              try:              try:
534                  title = str(dbtable)                  title = str(dbtable)
535    
536                  # Chose the correct Interface for the database type                  # Chose the correct Interface for the database type
537                  store = PostGISShapeStore(dbconn, dbtable)                  store = session.OpenDBShapeStore(dbconn, dbtable,
538                  session.AddShapeStore(store)                                                   id_column = id_column,
539                                                     geometry_column = geo_column)
540                  layer = Layer(title, store)                  layer = Layer(title, store)
541              except:              except:
542                  # Some error occured while initializing the layer                  # Some error occured while initializing the layer
543                  self.RunMessageBox(_("Add Layer from database"),                  self.RunMessageBox(_("Add Layer from database"),
544                                     _("Can't open the database table '%s'")                                     _("Can't open the database table '%s'")
545                                     % dbtable)                                     % dbtable)
546                    return
547    
548              map = self.canvas.Map()              map = self.canvas.Map()
549    
# Line 547  class MainWindow(DockFrame): Line 572  class MainWindow(DockFrame):
572              return self.canvas.Map().CanRemoveLayer(layer)              return self.canvas.Map().CanRemoveLayer(layer)
573          return False          return False
574    
575        def LayerToTop(self):
576            layer = self.current_layer()
577            if layer is not None:
578                self.canvas.Map().MoveLayerToTop(layer)
579    
580      def RaiseLayer(self):      def RaiseLayer(self):
581          layer = self.current_layer()          layer = self.current_layer()
582          if layer is not None:          if layer is not None:
# Line 557  class MainWindow(DockFrame): Line 587  class MainWindow(DockFrame):
587          if layer is not None:          if layer is not None:
588              self.canvas.Map().LowerLayer(layer)              self.canvas.Map().LowerLayer(layer)
589    
590        def LayerToBottom(self):
591            layer = self.current_layer()
592            if layer is not None:
593                self.canvas.Map().MoveLayerToBottom(layer)
594    
595      def current_layer(self):      def current_layer(self):
596          """Return the currently selected layer.          """Return the currently selected layer.
597    
# Line 568  class MainWindow(DockFrame): Line 603  class MainWindow(DockFrame):
603          """Return true if a layer is currently selected"""          """Return true if a layer is currently selected"""
604          return self.canvas.HasSelectedLayer()          return self.canvas.HasSelectedLayer()
605    
606        def has_selected_shape_layer(self):
607            """Return true if a shape layer is currently selected"""
608            return isinstance(self.current_layer(), Layer)
609    
610      def has_selected_shapes(self):      def has_selected_shapes(self):
611          """Return true if a shape is currently selected"""          """Return true if a shape is currently selected"""
612          return self.canvas.HasSelectedShapes()          return self.canvas.HasSelectedShapes()
# Line 575  class MainWindow(DockFrame): Line 614  class MainWindow(DockFrame):
614      def HideLayer(self):      def HideLayer(self):
615          layer = self.current_layer()          layer = self.current_layer()
616          if layer is not None:          if layer is not None:
617              layer.SetVisible(0)              layer.SetVisible(False)
618    
619      def ShowLayer(self):      def ShowLayer(self):
620          layer = self.current_layer()          layer = self.current_layer()
621          if layer is not None:          if layer is not None:
622              layer.SetVisible(1)              layer.SetVisible(True)
623    
624        def ToggleLayerVisibility(self):
625            layer = self.current_layer()
626            layer.SetVisible(not layer.Visible())
627    
628      def DuplicateLayer(self):      def DuplicateLayer(self):
629          """Ceate a new layer above the selected layer with the same shapestore          """Ceate a new layer above the selected layer with the same shapestore
# Line 600  class MainWindow(DockFrame): Line 643  class MainWindow(DockFrame):
643          return layer is not None and hasattr(layer, "ShapeStore")          return layer is not None and hasattr(layer, "ShapeStore")
644    
645      def LayerShowTable(self):      def LayerShowTable(self):
646            """
647            Present a TableView Window for the current layer.
648            In case the window is already open, bring it to the front.
649            In case, there is no active layer, do nothing.
650            In case, the layer has no ShapeStore, do nothing.
651            """
652          layer = self.current_layer()          layer = self.current_layer()
653          if layer is not None:          if layer is not None:
654                if not hasattr(layer, "ShapeStore"):
655                    return
656              table = layer.ShapeStore().Table()              table = layer.ShapeStore().Table()
657              name = "table_view" + str(id(table))              name = "table_view" + str(id(table))
658              dialog = self.get_open_dialog(name)              dialog = self.get_open_dialog(name)
# Line 612  class MainWindow(DockFrame): Line 663  class MainWindow(DockFrame):
663                  self.add_dialog(name, dialog)                  self.add_dialog(name, dialog)
664                  dialog.Show(True)                  dialog.Show(True)
665              else:              else:
666                  # FIXME: bring dialog to front here                  dialog.Raise()
                 pass  
667    
668      def MapProjection(self):      def MapProjection(self):
669    
# Line 703  class MainWindow(DockFrame): Line 753  class MainWindow(DockFrame):
753          return dialog is not None and dialog.IsShown()          return dialog is not None and dialog.IsShown()
754    
755      def TableOpen(self):      def TableOpen(self):
756          dlg = wxFileDialog(self, _("Open Table"), ".", "",          dlg = wxFileDialog(self, _("Open Table"),
757                               self.application.Path("data"), "",
758                             _("DBF Files (*.dbf)") + "|*.dbf|" +                             _("DBF Files (*.dbf)") + "|*.dbf|" +
759                             #_("CSV Files (*.csv)") + "|*.csv|" +                             #_("CSV Files (*.csv)") + "|*.csv|" +
760                             _("All Files (*.*)") + "|*.*",                             _("All Files (*.*)") + "|*.*",
# Line 719  class MainWindow(DockFrame): Line 770  class MainWindow(DockFrame):
770                                     _("Can't open the file '%s'.") % filename)                                     _("Can't open the file '%s'.") % filename)
771              else:              else:
772                  self.ShowTableView(table)                  self.ShowTableView(table)
773                    self.application.SetPath("data",filename)
774    
775      def TableClose(self):      def TableClose(self):
776          tables = self.application.session.UnreferencedTables()          tables = self.application.session.UnreferencedTables()
# Line 795  class MainWindow(DockFrame): Line 847  class MainWindow(DockFrame):
847    
848          # Second, let the user rename the layers          # Second, let the user rename the layers
849          for table in to_rename:          for table in to_rename:
850              dlg = wxTextEntryDialog(self, "Table Title: ", "Rename Table",              dlg = wxTextEntryDialog(self, _("Table Title:"), _("Rename Table"),
851                                      table.Title())                                      table.Title())
852              try:              try:
853                  if dlg.ShowModal() == wxID_OK:                  if dlg.ShowModal() == wxID_OK:
# Line 844  class MainWindow(DockFrame): Line 896  class MainWindow(DockFrame):
896          self.canvas.Print()          self.canvas.Print()
897    
898      def RenameMap(self):      def RenameMap(self):
899          dlg = wxTextEntryDialog(self, "Map Title: ", "Rename Map",          dlg = wxTextEntryDialog(self, _("Map Title:"), _("Rename Map"),
900                                  self.Map().Title())                                  self.Map().Title())
901          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
902              title = dlg.GetValue()              title = dlg.GetValue()
903              if title != "":              if title != "":
904                  self.Map().SetTitle(title)                  self.Map().SetTitle(title)
                 self.__SetTitle(title)  
905    
906          dlg.Destroy()          dlg.Destroy()
907    
# Line 858  class MainWindow(DockFrame): Line 909  class MainWindow(DockFrame):
909          """Let the user rename the currently selected layer"""          """Let the user rename the currently selected layer"""
910          layer = self.current_layer()          layer = self.current_layer()
911          if layer is not None:          if layer is not None:
912              dlg = wxTextEntryDialog(self, "Layer Title: ", "Rename Layer",              dlg = wxTextEntryDialog(self, _("Layer Title:"), _("Rename Layer"),
913                                      layer.Title())                                      layer.Title())
914              try:              try:
915                  if dlg.ShowModal() == wxID_OK:                  if dlg.ShowModal() == wxID_OK:
# Line 892  class MainWindow(DockFrame): Line 943  class MainWindow(DockFrame):
943                  # FIXME: bring dialog to front?                  # FIXME: bring dialog to front?
944                  pass                  pass
945    
946      def __SetTitle(self, title):      def title_changed(self, map):
947          self.SetTitle("Thuban - " + title)          """Subscribed to the canvas' TITLE_CHANGED messages"""
948            self.update_title()
949    
950        def update_title(self):
951            """Update the window's title according to it's current state.
952    
953            In this default implementation the title is 'Thuban - ' followed
954            by the map's title or simply 'Thuban' if there is not map.
955            Derived classes should override this method to get different
956            titles.
957    
958            This method is called automatically by other methods when the
959            title may have to change. For the methods implemented in this
960            class this usually only means that a different map has been set
961            or the current map's title has changed.
962            """
963            map = self.Map()
964            if map is not None:
965                title = _("Thuban - %s") % (map.Title(),)
966            else:
967                title = _("Thuban")
968            self.SetTitle(title)
969    
970    
971  #  #
972  # Define all the commands available in the main window  # Define all the commands available in the main window
# Line 935  def _has_selected_layer(context): Line 1008  def _has_selected_layer(context):
1008      """Return true if a layer is selected in the context"""      """Return true if a layer is selected in the context"""
1009      return context.mainwindow.has_selected_layer()      return context.mainwindow.has_selected_layer()
1010    
1011    def _has_selected_layer_visible(context):
1012        """Return true if a layer is selected in the context which is
1013        visible."""
1014        if context.mainwindow.has_selected_layer():
1015            layer = context.mainwindow.current_layer()
1016            if layer.Visible(): return True
1017        return False
1018    
1019    def _has_selected_shape_layer(context):
1020        """Return true if a shape layer is selected in the context"""
1021        return context.mainwindow.has_selected_shape_layer()
1022    
1023  def _has_selected_shapes(context):  def _has_selected_shapes(context):
1024      """Return true if a layer is selected in the context"""      """Return true if a layer is selected in the context"""
1025      return context.mainwindow.has_selected_shapes()      return context.mainwindow.has_selected_shapes()
# Line 954  def _has_visible_map(context): Line 1039  def _has_visible_map(context):
1039      if map is not None:      if map is not None:
1040          for layer in map.Layers():          for layer in map.Layers():
1041              if layer.Visible():              if layer.Visible():
1042                  return 1                  return True
1043      return 0      return False
1044    
1045  def _has_legend_shown(context):  def _has_legend_shown(context):
1046      """Return true if the legend window is shown"""      """Return true if the legend window is shown"""
# Line 969  def _has_dbconnections(context): Line 1054  def _has_dbconnections(context):
1054      """Return whether the the session has database connections"""      """Return whether the the session has database connections"""
1055      return context.session.HasDBConnections()      return context.session.HasDBConnections()
1056    
1057    def _has_postgis_support(context):
1058        return has_postgis_support()
1059    
1060    
1061  # File menu  # File menu
1062  _method_command("new_session", _("&New Session"), "NewSession",  _method_command("new_session", _("&New Session"), "NewSession",
1063                  helptext = _("Start a new session"))                  helptext = _("Start a new session"))
# Line 985  _method_command("toggle_legend", _("Lege Line 1074  _method_command("toggle_legend", _("Lege
1074                  checked = _has_legend_shown,                  checked = _has_legend_shown,
1075                  helptext = _("Toggle Legend on/off"))                  helptext = _("Toggle Legend on/off"))
1076  _method_command("database_management", _("&Database Connections..."),  _method_command("database_management", _("&Database Connections..."),
1077                  "DatabaseManagement")                  "DatabaseManagement",
1078                    sensitive = _has_postgis_support)
1079  _method_command("exit", _("E&xit"), "Exit",  _method_command("exit", _("E&xit"), "Exit",
1080                  helptext = _("Finish working with Thuban"))                  helptext = _("Finish working with Thuban"))
1081    
# Line 1066  _method_command("layer_hide", _("&Hide") Line 1156  _method_command("layer_hide", _("&Hide")
1156                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
1157  _method_command("layer_show_table", _("Show Ta&ble"), "LayerShowTable",  _method_command("layer_show_table", _("Show Ta&ble"), "LayerShowTable",
1158                  helptext = _("Show the selected layer's table"),                  helptext = _("Show the selected layer's table"),
1159                  sensitive = _has_selected_layer)                  sensitive = _has_selected_shape_layer)
1160  _method_command("layer_properties", _("&Properties..."), "LayerEditProperties",  _method_command("layer_properties", _("&Properties..."), "LayerEditProperties",
1161                  sensitive = _has_selected_layer,                  sensitive = _has_selected_layer,
1162                  helptext = _("Edit the properties of the selected layer"))                  helptext = _("Edit the properties of the selected layer"))
1163  _method_command("layer_jointable", _("&Join Table..."), "LayerJoinTable",  _method_command("layer_jointable", _("&Join Table..."), "LayerJoinTable",
1164                  sensitive = _has_selected_layer,                  sensitive = _has_selected_shape_layer,
1165                  helptext = _("Join and attach a table to the selected layer"))                  helptext = _("Join and attach a table to the selected layer"))
1166    
1167    # further layer methods:
1168    _method_command("layer_to_top", _("&Top"), "LayerToTop",
1169                    helptext = _("Put selected layer to the top"),
1170                    sensitive = _has_selected_layer)
1171    _method_command("layer_to_bottom", _("&Bottom"), "LayerToBottom",
1172                    helptext = _("Put selected layer to the bottom"),
1173                    sensitive = _has_selected_layer)
1174    _method_command("layer_visibility", _("&Visible"), "ToggleLayerVisibility",
1175                    checked = _has_selected_layer_visible,
1176                    helptext = _("Toggle visibility of selected layer"),
1177                    sensitive = _has_selected_layer)
1178    
1179  def _can_unjoin(context):  def _can_unjoin(context):
1180      """Return whether the Layer/Unjoin command can be executed.      """Return whether the Layer/Unjoin command can be executed.
1181    
Legend:
Removed from v.1620  
changed lines
  Added in v.2186
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26