/[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 829 by jonathan, Tue May 6 12:06:38 2003 UTC revision 1084 by jan, Wed May 28 10:25:09 2003 UTC
# Line 2  Line 2 
2  # Authors:  # Authors:
3  # Jan-Oliver Wagner <[email protected]>  # Jan-Oliver Wagner <[email protected]>
4  # Bernhard Herzog <[email protected]>  # Bernhard Herzog <[email protected]>
5    # Frank Koormann <[email protected]>
6  #  #
7  # This program is free software under the GPL (>=v2)  # This program is free software under the GPL (>=v2)
8  # Read the file COPYING coming with Thuban for details.  # Read the file COPYING coming with Thuban for details.
# Line 18  __ThubanVersion__ = "0.2" #"$THUBAN_0_2$ Line 19  __ThubanVersion__ = "0.2" #"$THUBAN_0_2$
19  import os  import os
20    
21  from wxPython.wx import *  from wxPython.wx import *
22    from wxPython.wx import __version__ as wxPython_version
23    
24    from wxPython.lib.dialogs import wxMultipleChoiceDialog
25    
26  import Thuban  import Thuban
27    import Thuban.version
28    
29  from Thuban import _  from Thuban import _
30  from Thuban.Model.session import create_empty_session  from Thuban.Model.session import create_empty_session
31  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer, RasterLayer
 from Thuban.Model.color import Color  
 from Thuban.Model.proj import Projection  
32    
33  import view  import view
34  import tree  import tree
 import proj4dialog  
35  import tableview, identifyview  import tableview, identifyview
36  from Thuban.UI.classifier import Classifier  from Thuban.UI.classifier import Classifier
37  import legend  import legend
# Line 39  from command import registry, Command, T Line 42  from command import registry, Command, T
42  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION  from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION
43    
44  from Thuban.UI.dock import DockFrame  from Thuban.UI.dock import DockFrame
45    from Thuban.UI.join import JoinDialog
46    
47  import resource  import resource
48    
# Line 60  class MainWindow(DockFrame): Line 64  class MainWindow(DockFrame):
64      # implemented in the __getattr__ method.      # implemented in the __getattr__ method.
65      delegated_methods = {"SelectLayer": "canvas",      delegated_methods = {"SelectLayer": "canvas",
66                           "SelectShapes": "canvas",                           "SelectShapes": "canvas",
67                             "SelectedLayer": "canvas",
68                             "SelectedShapes": "canvas",
69                           }                           }
70    
71      def __init__(self, parent, ID, title, application, interactor,      def __init__(self, parent, ID, title, application, interactor,
# Line 98  class MainWindow(DockFrame): Line 104  class MainWindow(DockFrame):
104    
105          self.init_dialogs()          self.init_dialogs()
106    
107          EVT_CLOSE(self, self._OnClose)          EVT_CLOSE(self, self.OnClose)
108    
109      def Subscribe(self, channel, *args):      def Subscribe(self, channel, *args):
110          """Subscribe a function to a message channel.          """Subscribe a function to a message channel.
# Line 357  class MainWindow(DockFrame): Line 363  class MainWindow(DockFrame):
363                  d.Close()                  d.Close()
364    
365      def NewSession(self):      def NewSession(self):
366          self.save_modified_session()          if self.save_modified_session() != wxID_CANCEL:
367          self.prepare_new_session()              self.prepare_new_session()
368          self.application.SetSession(create_empty_session())              self.application.SetSession(create_empty_session())
369    
370      def OpenSession(self):      def OpenSession(self):
371          self.save_modified_session()          if self.save_modified_session() != wxID_CANCEL:
372          dlg = wxFileDialog(self, _("Open Session"), ".", "", "*.thuban", wxOPEN)              dlg = wxFileDialog(self, _("Open Session"), ".", "",
373          if dlg.ShowModal() == wxID_OK:                                 "Thuban Session File (*.thuban)|*.thuban",
374              self.prepare_new_session()                                 wxOPEN)
375              self.application.OpenSession(dlg.GetPath())              if dlg.ShowModal() == wxID_OK:
376          dlg.Destroy()                  self.prepare_new_session()
377                    self.application.OpenSession(dlg.GetPath())
378                dlg.Destroy()
379    
380      def SaveSession(self):      def SaveSession(self):
381          if self.application.session.filename == None:          if self.application.session.filename == None:
# Line 377  class MainWindow(DockFrame): Line 385  class MainWindow(DockFrame):
385    
386      def SaveSessionAs(self):      def SaveSessionAs(self):
387          dlg = wxFileDialog(self, _("Save Session As"), ".", "",          dlg = wxFileDialog(self, _("Save Session As"), ".", "",
388                             "*.thuban", wxSAVE|wxOVERWRITE_PROMPT)                             "Thuban Session File (*.thuban)|*.thuban",
389                               wxSAVE|wxOVERWRITE_PROMPT)
390          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
391              self.application.session.SetFilename(dlg.GetPath())              self.application.session.SetFilename(dlg.GetPath())
392              self.application.SaveSession()              self.application.SaveSession()
# Line 386  class MainWindow(DockFrame): Line 395  class MainWindow(DockFrame):
395      def Exit(self):      def Exit(self):
396          self.Close(False)          self.Close(False)
397    
398      def _OnClose(self, event):      def OnClose(self, event):
399          result = self.save_modified_session(can_veto = event.CanVeto())          result = self.save_modified_session(can_veto = event.CanVeto())
400          if result == wxID_CANCEL:          if result == wxID_CANCEL:
401              event.Veto()              event.Veto()
# Line 395  class MainWindow(DockFrame): Line 404  class MainWindow(DockFrame):
404              # wx's destroy event, but that isn't implemented for wxGTK              # wx's destroy event, but that isn't implemented for wxGTK
405              # yet.              # yet.
406              self.canvas.Unsubscribe(VIEW_POSITION, self.view_position_changed)              self.canvas.Unsubscribe(VIEW_POSITION, self.view_position_changed)
407              DockFrame._OnClose(self, event)              DockFrame.OnClose(self, event)
408                for dlg in self.dialogs.values():
409                    dlg.Destroy()
410                self.canvas.Destroy()
411              self.Destroy()              self.Destroy()
412    
413      def SetMap(self, map):      def SetMap(self, map):
# Line 428  class MainWindow(DockFrame): Line 440  class MainWindow(DockFrame):
440    
441      def About(self):      def About(self):
442          self.RunMessageBox(_("About"),          self.RunMessageBox(_("About"),
443                             _("Thuban v%s\n"                             _("Thuban %s\n"
444                              #"Build Date: %s\n"                              #"Build Date: %s\n"
445                              "\n"                              "using:\n"
446                                "  %s\n"
447                                "  %s\n\n"
448                              "Thuban is a program for\n"                              "Thuban is a program for\n"
449                              "exploring geographic data.\n"                              "exploring geographic data.\n"
450                              "Copyright (C) 2001-2003 Intevation GmbH.\n"                              "Copyright (C) 2001-2003 Intevation GmbH.\n"
451                              "Thuban is licensed under the GNU GPL"                              "Thuban is licensed under the GNU GPL"
452                             % __ThubanVersion__), #__BuildDate__)),                              % (Thuban.version.longversion,
453                                   "wxPython %s" % wxPython_version,
454                                   "Python %d.%d.%d" % sys.version_info[:3]
455                                  )),
456    #                           % __ThubanVersion__), #__BuildDate__)),
457                             wxOK | wxICON_INFORMATION)                             wxOK | wxICON_INFORMATION)
458    
459      def AddLayer(self):      def AddLayer(self):
# Line 444  class MainWindow(DockFrame): Line 462  class MainWindow(DockFrame):
462          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
463              filename = dlg.GetPath()              filename = dlg.GetPath()
464              title = os.path.splitext(os.path.basename(filename))[0]              title = os.path.splitext(os.path.basename(filename))[0]
             store = self.application.Session().OpenShapefile(filename)  
             layer = Layer(title, store)  
465              map = self.canvas.Map()              map = self.canvas.Map()
466              has_layers = map.HasLayers()              has_layers = map.HasLayers()
467              try:              try:
468                  map.AddLayer(layer)                  store = self.application.Session().OpenShapefile(filename)
469              except IOError:              except IOError:
470                  # the layer couldn't be opened                  # the layer couldn't be opened
471                  self.RunMessageBox(_("Add Layer"),                  self.RunMessageBox(_("Add Layer"),
472                                     _("Can't open the file '%s'.") % filename)                                     _("Can't open the file '%s'.") % filename)
473              else:              else:
474                    layer = Layer(title, store)
475                    map.AddLayer(layer)
476                    if not has_layers:
477                        # if we're adding a layer to an empty map, fit the
478                        # new map to the window
479                        self.canvas.FitMapToWindow()
480            dlg.Destroy()
481    
482        def AddRasterLayer(self):
483            dlg = wxFileDialog(self, _("Select an image file"), ".", "", "*.*",
484                               wxOPEN)
485            if dlg.ShowModal() == wxID_OK:
486                filename = dlg.GetPath()
487                title = os.path.splitext(os.path.basename(filename))[0]
488                map = self.canvas.Map()
489                has_layers = map.HasLayers()
490                try:
491                    layer = RasterLayer(title, filename)
492                except IOError:
493                    # the layer couldn't be opened
494                    self.RunMessageBox(_("Add Image Layer"),
495                                       _("Can't open the file '%s'.") % filename)
496                else:
497                    map.AddLayer(layer)
498                  if not has_layers:                  if not has_layers:
499                      # if we're adding a layer to an empty map, fit the                      # if we're adding a layer to an empty map, fit the
500                      # new map to the window                      # new map to the window
# Line 504  class MainWindow(DockFrame): Line 544  class MainWindow(DockFrame):
544          """Return true if a shape is currently selected"""          """Return true if a shape is currently selected"""
545          return self.canvas.HasSelectedShapes()          return self.canvas.HasSelectedShapes()
546    
     def choose_color(self):  
         """Run the color selection dialog and return the selected color.  
   
         If the user cancels, return None.  
         """  
         dlg = wxColourDialog(self)  
         color = None  
         if dlg.ShowModal() == wxID_OK:  
             data = dlg.GetColourData()  
             wxc = data.GetColour()  
             color = Color(wxc.Red() / 255.0,  
                           wxc.Green() / 255.0,  
                           wxc.Blue() / 255.0)  
         dlg.Destroy()  
         return color  
   
547      def HideLayer(self):      def HideLayer(self):
548          layer = self.current_layer()          layer = self.current_layer()
549          if layer is not None:          if layer is not None:
# Line 538  class MainWindow(DockFrame): Line 562  class MainWindow(DockFrame):
562              dialog = self.get_open_dialog(name)              dialog = self.get_open_dialog(name)
563              if dialog is None:              if dialog is None:
564                  dialog = tableview.LayerTableFrame(self, name,                  dialog = tableview.LayerTableFrame(self, name,
565                                                 _("Table: %s") % layer.Title(),                                           _("Layer Table: %s") % layer.Title(),
566                                                     layer, table)                                           layer, table)
567                  self.add_dialog(name, dialog)                  self.add_dialog(name, dialog)
568                  dialog.Show(true)                  dialog.Show(True)
569              else:              else:
570                  # FIXME: bring dialog to front here                  # FIXME: bring dialog to front here
571                  pass                  pass
# Line 595  class MainWindow(DockFrame): Line 619  class MainWindow(DockFrame):
619              dialog.Show()              dialog.Show()
620          dialog.Raise()          dialog.Raise()
621    
622        def LayerJoinTable(self):
623            layer = self.canvas.SelectedLayer()
624            if layer is not None:
625                dlg = JoinDialog(self, _("Join Layer with Table"),
626                                 self.application.session,
627                                 layer = layer)
628                dlg.ShowModal()
629    
630        def LayerUnjoinTable(self):
631            layer = self.canvas.SelectedLayer()
632            if layer is not None:
633                orig_store = layer.ShapeStore().OrigShapeStore()
634                if orig_store:
635                    layer.SetShapeStore(orig_store)
636    
637      def ShowLegend(self):      def ShowLegend(self):
638          if not self.LegendShown():          if not self.LegendShown():
# Line 619  class MainWindow(DockFrame): Line 657  class MainWindow(DockFrame):
657          dialog = self.FindRegisteredDock("legend")          dialog = self.FindRegisteredDock("legend")
658          return dialog is not None and dialog.IsShown()          return dialog is not None and dialog.IsShown()
659    
660        def TableOpen(self):
661            dlg = wxFileDialog(self, _("Open Table"), ".", "",
662                               _("DBF Files (*.dbf)") + "|*.dbf|" +
663                               #_("CSV Files (*.csv)") + "|*.csv|" +
664                               _("All Files (*.*)") + "|*.*",
665                               wxOPEN)
666            if dlg.ShowModal() == wxID_OK:
667                filename = dlg.GetPath()
668                dlg.Destroy()
669                try:
670                    table = self.application.session.OpenTableFile(filename)
671                except IOError:
672                    # the layer couldn't be opened
673                    self.RunMessageBox(_("Open Table"),
674                                       _("Can't open the file '%s'.") % filename)
675                else:
676                    self.ShowTableView(table)
677    
678        def TableClose(self):
679            tables = self.application.session.UnreferencedTables()
680    
681            lst = [(t.Title(), t) for t in tables]
682            lst.sort()
683            titles = [i[0] for i in lst]
684            dlg = wxMultipleChoiceDialog(self, _("Pick the tables to close:"),
685                                         _("Close Table"), titles,
686                                         size = (400, 300),
687                                         style = wxDEFAULT_DIALOG_STYLE |
688                                                 wxRESIZE_BORDER)
689            if dlg.ShowModal() == wxID_OK:
690                for i in dlg.GetValue():
691                    self.application.session.RemoveTable(lst[i][1])
692    
693    
694        def TableShow(self):
695            """Offer a multi-selection dialog for tables to be displayed
696    
697            The windows for the selected tables are opened or brought to
698            the front.
699            """
700            tables = self.application.session.Tables()
701    
702            lst = [(t.Title(), t) for t in tables]
703            lst.sort()
704            titles = [i[0] for i in lst]
705            dlg = wxMultipleChoiceDialog(self, _("Pick the table to show:"),
706                                         _("Show Table"), titles,
707                                         size = (400,300),
708                                         style = wxDEFAULT_DIALOG_STYLE |
709                                                 wxRESIZE_BORDER)
710            if (dlg.ShowModal() == wxID_OK):
711                for i in dlg.GetValue():
712                    # XXX: if the table belongs to a layer, open a
713                    # LayerTableFrame instead of QueryTableFrame
714                    self.ShowTableView(lst[i][1])
715    
716        def TableJoin(self):
717            dlg = JoinDialog(self, _("Join Tables"), self.application.session)
718            dlg.ShowModal()
719    
720        def ShowTableView(self, table):
721            """Open a table view for the table and optionally"""
722            name = "table_view%d" % id(table)
723            dialog = self.get_open_dialog(name)
724            if dialog is None:
725                dialog = tableview.QueryTableFrame(self, name,
726                                                   _("Table: %s") % table.Title(),
727                                                   table)
728                self.add_dialog(name, dialog)
729                dialog.Show(True)
730            # FIXME: else bring dialog to front
731    
732      def ZoomInTool(self):      def ZoomInTool(self):
733          self.canvas.ZoomInTool()          self.canvas.ZoomInTool()
734    
# Line 644  class MainWindow(DockFrame): Line 754  class MainWindow(DockFrame):
754      def FullSelectionExtent(self):      def FullSelectionExtent(self):
755          self.canvas.FitSelectedToWindow()          self.canvas.FitSelectedToWindow()
756    
757        def ExportMap(self):
758            self.canvas.Export()
759    
760      def PrintMap(self):      def PrintMap(self):
761          self.canvas.Print()          self.canvas.Print()
762    
# Line 795  _method_command("layer_full_extent", _(" Line 908  _method_command("layer_full_extent", _("
908  _method_command("selected_full_extent", _("&Full selection extent"), "FullSelectionExtent",  _method_command("selected_full_extent", _("&Full selection extent"), "FullSelectionExtent",
909                 helptext = _("Full Selection Extent"), icon = "fullselextent",                 helptext = _("Full Selection Extent"), icon = "fullselextent",
910                sensitive = _has_selected_shapes)                sensitive = _has_selected_shapes)
911    _method_command("map_export", _("E&xport"), "ExportMap",
912                        helptext = _("Export the map to file"))
913  _method_command("map_print", _("Prin&t"), "PrintMap",  _method_command("map_print", _("Prin&t"), "PrintMap",
914                  helptext = _("Print the map"))                  helptext = _("Print the map"))
915  _method_command("map_rename", _("&Rename..."), "RenameMap",  _method_command("map_rename", _("&Rename..."), "RenameMap",
916                  helptext = _("Rename the map"))                  helptext = _("Rename the map"))
917  _method_command("layer_add", _("&Add Layer..."), "AddLayer",  _method_command("layer_add", _("&Add Layer..."), "AddLayer",
918                  helptext = _("Add a new layer to active map"))                  helptext = _("Add a new layer to active map"))
919    _method_command("rasterlayer_add", _("&Add Image Layer..."), "AddRasterLayer",
920                    helptext = _("Add a new image layer to active map"))
921  _method_command("layer_remove", _("&Remove Layer"), "RemoveLayer",  _method_command("layer_remove", _("&Remove Layer"), "RemoveLayer",
922                  helptext = _("Remove selected layer(s)"),                  helptext = _("Remove selected layer(s)"),
923                  sensitive = _can_remove_layer)                  sensitive = _can_remove_layer)
# Line 825  _method_command("layer_show_table", _("S Line 942  _method_command("layer_show_table", _("S
942                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
943  _method_command("layer_properties", _("&Properties..."), "LayerEditProperties",  _method_command("layer_properties", _("&Properties..."), "LayerEditProperties",
944                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
945    _method_command("layer_jointable", _("&Join Table..."), "LayerJoinTable",
946                    sensitive = _has_selected_layer)
947    
948  # the menu structure  def _can_unjoin(context):
949  main_menu = Menu("<main>", "<main>",      """Return whether the Layer/Unjoin command can be executed.
950                   [Menu("file", _("&File"),  
951                         ["new_session", "open_session", None,      This is the case if a layer is selected and that layer has a
952                          "save_session", "save_session_as", None,      shapestore that has an original shapestore.
953                          "toggle_session_tree", None,      """
954                          "exit"]),      layer = context.mainwindow.SelectedLayer()
955                    Menu("map", _("&Map"),      if layer is None:
956                         ["layer_add", "layer_remove", "map_rename",          return 0
957        getstore = getattr(layer, "ShapeStore", None)
958        if getstore is not None:
959            return getstore().OrigShapeStore() is not None
960        else:
961            return 0
962    _method_command("layer_unjointable", _("&Unjoin Table..."), "LayerUnjoinTable",
963                    sensitive = _can_unjoin)
964    
965    # Table menu
966    _method_command("table_open", _("&Open..."), "TableOpen")
967    _method_command("table_close", _("&Close"), "TableClose",
968           sensitive = lambda context: bool(context.session.UnreferencedTables()))
969    _method_command("table_show", _("&Show"), "TableShow")
970    _method_command("table_join", _("&Join..."), "TableJoin")
971    
972    #  Export only under Windows ...
973    map_menu = ["layer_add", "rasterlayer_add", "layer_remove", "map_rename",
974                          None,                          None,
975                          "map_projection",                          "map_projection",
976                          None,                          None,
# Line 847  main_menu = Menu("<main>", "<main>", Line 983  main_menu = Menu("<main>", "<main>",
983                          "map_identify_tool", "map_label_tool",                          "map_identify_tool", "map_label_tool",
984                          None,                          None,
985                          "toggle_legend",                          "toggle_legend",
986                          None,                          None]
987                          "map_print"]),  if wxPlatform == '__WXMSW__':
988        map_menu.append("map_export")
989    map_menu.append("map_print")
990    
991    # the menu structure
992    main_menu = Menu("<main>", "<main>",
993                     [Menu("file", _("&File"),
994                           ["new_session", "open_session", None,
995                            "save_session", "save_session_as", None,
996                            "toggle_session_tree", None,
997                            "exit"]),
998                      Menu("map", _("&Map"), map_menu),
999                    Menu("layer", _("&Layer"),                    Menu("layer", _("&Layer"),
1000                          ["layer_raise", "layer_lower",                          ["layer_raise", "layer_lower",
1001                          None,                          None,
1002                          "layer_show", "layer_hide",                          "layer_show", "layer_hide",
1003                          None,                          None,
1004                            "layer_projection",
1005                            None,
1006                          "layer_show_table",                          "layer_show_table",
1007                            "layer_jointable",
1008                            "layer_unjointable",
1009                          None,                          None,
                         "layer_projection",  
1010                          "layer_properties"]),                          "layer_properties"]),
1011                      Menu("table", _("&Table"),
1012                           ["table_open", "table_close",
1013                           None,
1014                           "table_show",
1015                           None,
1016                           "table_join"]),
1017                    Menu("help", _("&Help"),                    Menu("help", _("&Help"),
1018                         ["help_about"])])                         ["help_about"])])
1019    
Legend:
Removed from v.829  
changed lines
  Added in v.1084
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26