/[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 923 by frank, Mon May 19 09:12:25 2003 UTC revision 1108 by jonathan, Fri May 30 06:31:21 2003 UTC
# Line 17  __ThubanVersion__ = "0.2" #"$THUBAN_0_2$ Line 17  __ThubanVersion__ = "0.2" #"$THUBAN_0_2$
17  #__BuildDate__ = "$Date$"  #__BuildDate__ = "$Date$"
18    
19  import os  import os
20    import copy
21    
22  from wxPython.wx import *  from wxPython.wx import *
23  from wxPython.wx import __version__ as wxPython_version  from wxPython.wx import __version__ as wxPython_version
24    
25    from wxPython.lib.dialogs import wxMultipleChoiceDialog
26    
27  import Thuban  import Thuban
28  import Thuban.version  import Thuban.version
29    
30  from Thuban import _  from Thuban import _
31  from Thuban.Model.session import create_empty_session  from Thuban.Model.session import create_empty_session
32  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  
33    
34  import view  import view
35  import tree  import tree
 import proj4dialog  
36  import tableview, identifyview  import tableview, identifyview
37  from Thuban.UI.classifier import Classifier  from Thuban.UI.classifier import Classifier
38  import legend  import legend
# Line 65  class MainWindow(DockFrame): Line 65  class MainWindow(DockFrame):
65      # implemented in the __getattr__ method.      # implemented in the __getattr__ method.
66      delegated_methods = {"SelectLayer": "canvas",      delegated_methods = {"SelectLayer": "canvas",
67                           "SelectShapes": "canvas",                           "SelectShapes": "canvas",
68                             "SelectedLayer": "canvas",
69                           "SelectedShapes": "canvas",                           "SelectedShapes": "canvas",
70                           }                           }
71    
# Line 104  class MainWindow(DockFrame): Line 105  class MainWindow(DockFrame):
105    
106          self.init_dialogs()          self.init_dialogs()
107    
108          EVT_CLOSE(self, self._OnClose)          EVT_CLOSE(self, self.OnClose)
109    
110      def Subscribe(self, channel, *args):      def Subscribe(self, channel, *args):
111          """Subscribe a function to a message channel.          """Subscribe a function to a message channel.
# Line 363  class MainWindow(DockFrame): Line 364  class MainWindow(DockFrame):
364                  d.Close()                  d.Close()
365    
366      def NewSession(self):      def NewSession(self):
367          self.save_modified_session()          if self.save_modified_session() != wxID_CANCEL:
368          self.prepare_new_session()              self.prepare_new_session()
369          self.application.SetSession(create_empty_session())              self.application.SetSession(create_empty_session())
370    
371      def OpenSession(self):      def OpenSession(self):
372          self.save_modified_session()          if self.save_modified_session() != wxID_CANCEL:
373          dlg = wxFileDialog(self, _("Open Session"), ".", "",              dlg = wxFileDialog(self, _("Open Session"), ".", "",
374                             "Thuban Session File (*.thuban)|*.thuban", wxOPEN)                                 "Thuban Session File (*.thuban)|*.thuban",
375          if dlg.ShowModal() == wxID_OK:                                 wxOPEN)
376              self.prepare_new_session()              if dlg.ShowModal() == wxID_OK:
377              self.application.OpenSession(dlg.GetPath())                  self.prepare_new_session()
378          dlg.Destroy()                  self.application.OpenSession(dlg.GetPath())
379                dlg.Destroy()
380    
381      def SaveSession(self):      def SaveSession(self):
382          if self.application.session.filename == None:          if self.application.session.filename == None:
# Line 394  class MainWindow(DockFrame): Line 396  class MainWindow(DockFrame):
396      def Exit(self):      def Exit(self):
397          self.Close(False)          self.Close(False)
398    
399      def _OnClose(self, event):      def OnClose(self, event):
400          result = self.save_modified_session(can_veto = event.CanVeto())          result = self.save_modified_session(can_veto = event.CanVeto())
401          if result == wxID_CANCEL:          if result == wxID_CANCEL:
402              event.Veto()              event.Veto()
# Line 403  class MainWindow(DockFrame): Line 405  class MainWindow(DockFrame):
405              # wx's destroy event, but that isn't implemented for wxGTK              # wx's destroy event, but that isn't implemented for wxGTK
406              # yet.              # yet.
407              self.canvas.Unsubscribe(VIEW_POSITION, self.view_position_changed)              self.canvas.Unsubscribe(VIEW_POSITION, self.view_position_changed)
408              DockFrame._OnClose(self, event)              DockFrame.OnClose(self, event)
409                for dlg in self.dialogs.values():
410                    dlg.Destroy()
411                self.canvas.Destroy()
412              self.Destroy()              self.Destroy()
413    
414      def SetMap(self, map):      def SetMap(self, map):
# Line 458  class MainWindow(DockFrame): Line 463  class MainWindow(DockFrame):
463          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
464              filename = dlg.GetPath()              filename = dlg.GetPath()
465              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)  
466              map = self.canvas.Map()              map = self.canvas.Map()
467              has_layers = map.HasLayers()              has_layers = map.HasLayers()
468              try:              try:
469                  map.AddLayer(layer)                  store = self.application.Session().OpenShapefile(filename)
470              except IOError:              except IOError:
471                  # the layer couldn't be opened                  # the layer couldn't be opened
472                  self.RunMessageBox(_("Add Layer"),                  self.RunMessageBox(_("Add Layer"),
473                                     _("Can't open the file '%s'.") % filename)                                     _("Can't open the file '%s'.") % filename)
474              else:              else:
475                    layer = Layer(title, store)
476                    map.AddLayer(layer)
477                    if not has_layers:
478                        # if we're adding a layer to an empty map, fit the
479                        # new map to the window
480                        self.canvas.FitMapToWindow()
481            dlg.Destroy()
482    
483        def AddRasterLayer(self):
484            dlg = wxFileDialog(self, _("Select an image file"), ".", "", "*.*",
485                               wxOPEN)
486            if dlg.ShowModal() == wxID_OK:
487                filename = dlg.GetPath()
488                title = os.path.splitext(os.path.basename(filename))[0]
489                map = self.canvas.Map()
490                has_layers = map.HasLayers()
491                try:
492                    layer = RasterLayer(title, filename)
493                except IOError:
494                    # the layer couldn't be opened
495                    self.RunMessageBox(_("Add Image Layer"),
496                                       _("Can't open the file '%s'.") % filename)
497                else:
498                    map.AddLayer(layer)
499                  if not has_layers:                  if not has_layers:
500                      # if we're adding a layer to an empty map, fit the                      # if we're adding a layer to an empty map, fit the
501                      # new map to the window                      # new map to the window
# Line 522  class MainWindow(DockFrame): Line 549  class MainWindow(DockFrame):
549          layer = self.current_layer()          layer = self.current_layer()
550          if layer is not None:          if layer is not None:
551              layer.SetVisible(0)              layer.SetVisible(0)
552            
553      def ShowLayer(self):      def ShowLayer(self):
554          layer = self.current_layer()          layer = self.current_layer()
555          if layer is not None:          if layer is not None:
556              layer.SetVisible(1)              layer.SetVisible(1)
557    
558        def DuplicateLayer(self):
559            """Ceate a new layer above the selected layer with the same shapestore
560            """
561            layer = self.current_layer()
562            if layer is not None and hasattr(layer, "ShapeStore"):
563                new_layer = Layer(_("Copy of `%s'") % layer.Title(),
564                                  layer.ShapeStore(),
565                                  projection = layer.GetProjection())
566                new_classification = copy.deepcopy(layer.GetClassification())
567                new_layer.SetClassification(new_classification)
568                self.Map().AddLayer(new_layer)
569    
570        def CanDuplicateLayer(self):
571            """Return whether the DuplicateLayer method can create a duplicate"""
572            layer = self.current_layer()
573            return layer is not None and hasattr(layer, "ShapeStore")
574    
575      def LayerShowTable(self):      def LayerShowTable(self):
576          layer = self.current_layer()          layer = self.current_layer()
577          if layer is not None:          if layer is not None:
# Line 536  class MainWindow(DockFrame): Line 580  class MainWindow(DockFrame):
580              dialog = self.get_open_dialog(name)              dialog = self.get_open_dialog(name)
581              if dialog is None:              if dialog is None:
582                  dialog = tableview.LayerTableFrame(self, name,                  dialog = tableview.LayerTableFrame(self, name,
583                                                 _("Table: %s") % layer.Title(),                                           _("Layer Table: %s") % layer.Title(),
584                                                     layer, table)                                           layer, table)
585                  self.add_dialog(name, dialog)                  self.add_dialog(name, dialog)
586                  dialog.Show(true)                  dialog.Show(True)
587              else:              else:
588                  # FIXME: bring dialog to front here                  # FIXME: bring dialog to front here
589                  pass                  pass
# Line 594  class MainWindow(DockFrame): Line 638  class MainWindow(DockFrame):
638          dialog.Raise()          dialog.Raise()
639    
640      def LayerJoinTable(self):      def LayerJoinTable(self):
641          print "LayerJoinTable"          layer = self.canvas.SelectedLayer()
642            if layer is not None:
643                dlg = JoinDialog(self, _("Join Layer with Table"),
644                                 self.application.session,
645                                 layer = layer)
646                dlg.ShowModal()
647    
648      def LayerUnjoinTable(self):      def LayerUnjoinTable(self):
649          print "LayerUnjoinTable"          layer = self.canvas.SelectedLayer()
650            if layer is not None:
651                orig_store = layer.ShapeStore().OrigShapeStore()
652                if orig_store:
653                    layer.SetShapeStore(orig_store)
654    
655      def ShowLegend(self):      def ShowLegend(self):
656          if not self.LegendShown():          if not self.LegendShown():
# Line 617  class MainWindow(DockFrame): Line 670  class MainWindow(DockFrame):
670          else:          else:
671              dialog.Show(not dialog.IsShown())              dialog.Show(not dialog.IsShown())
672    
673            self.canvas.FitMapToWindow()
674    
675      def LegendShown(self):      def LegendShown(self):
676          """Return true iff the legend is currently open"""          """Return true iff the legend is currently open"""
677          dialog = self.FindRegisteredDock("legend")          dialog = self.FindRegisteredDock("legend")
678          return dialog is not None and dialog.IsShown()          return dialog is not None and dialog.IsShown()
679    
680      def TableOpen(self):      def TableOpen(self):
         print "TableOpen"  
681          dlg = wxFileDialog(self, _("Open Table"), ".", "",          dlg = wxFileDialog(self, _("Open Table"), ".", "",
682                             "DBF Files (*.dbf)|*.dbf|" +                             _("DBF Files (*.dbf)") + "|*.dbf|" +
683                             "CSV Files (*.csv)|*.csv|" +                             #_("CSV Files (*.csv)") + "|*.csv|" +
684                             "All Files (*.*)|*.*",                             _("All Files (*.*)") + "|*.*",
685                             wxOPEN)                             wxOPEN)
686          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
687              #self.application.session.OpenTable(dlg.GetPath())              filename = dlg.GetPath()
688              pass              dlg.Destroy()
689                try:
690          dlg.Destroy()                  table = self.application.session.OpenTableFile(filename)
691                except IOError:
692                    # the layer couldn't be opened
693                    self.RunMessageBox(_("Open Table"),
694                                       _("Can't open the file '%s'.") % filename)
695                else:
696                    self.ShowTableView(table)
697    
698      def TableClose(self):      def TableClose(self):
699          print "TableClose"          tables = self.application.session.UnreferencedTables()
700    
701            lst = [(t.Title(), t) for t in tables]
702            lst.sort()
703            titles = [i[0] for i in lst]
704            dlg = wxMultipleChoiceDialog(self, _("Pick the tables to close:"),
705                                         _("Close Table"), titles,
706                                         size = (400, 300),
707                                         style = wxDEFAULT_DIALOG_STYLE |
708                                                 wxRESIZE_BORDER)
709            if dlg.ShowModal() == wxID_OK:
710                for i in dlg.GetValue():
711                    self.application.session.RemoveTable(lst[i][1])
712    
713    
714      def TableShow(self):      def TableShow(self):
715          print "TableShow"          """Offer a multi-selection dialog for tables to be displayed
716    
717            The windows for the selected tables are opened or brought to
718            the front.
719            """
720            tables = self.application.session.Tables()
721    
722      def TableHide(self):          lst = [(t.Title(), t) for t in tables]
723          print "TableHide"          lst.sort()
724            titles = [i[0] for i in lst]
725            dlg = wxMultipleChoiceDialog(self, _("Pick the table to show:"),
726                                         _("Show Table"), titles,
727                                         size = (400,300),
728                                         style = wxDEFAULT_DIALOG_STYLE |
729                                                 wxRESIZE_BORDER)
730            if (dlg.ShowModal() == wxID_OK):
731                for i in dlg.GetValue():
732                    # XXX: if the table belongs to a layer, open a
733                    # LayerTableFrame instead of QueryTableFrame
734                    self.ShowTableView(lst[i][1])
735    
736      def TableJoin(self):      def TableJoin(self):
         print "TableJoin"  
737          dlg = JoinDialog(self, _("Join Tables"), self.application.session)          dlg = JoinDialog(self, _("Join Tables"), self.application.session)
738          if dlg.ShowModal() == wxID_OK:          dlg.ShowModal()
739              print "OK"  
740        def ShowTableView(self, table):
741            """Open a table view for the table and optionally"""
742            name = "table_view%d" % id(table)
743            dialog = self.get_open_dialog(name)
744            if dialog is None:
745                dialog = tableview.QueryTableFrame(self, name,
746                                                   _("Table: %s") % table.Title(),
747                                                   table)
748                self.add_dialog(name, dialog)
749                dialog.Show(True)
750            # FIXME: else bring dialog to front
751    
752      def ZoomInTool(self):      def ZoomInTool(self):
753          self.canvas.ZoomInTool()          self.canvas.ZoomInTool()
# Line 837  _method_command("map_rename", _("&Rename Line 936  _method_command("map_rename", _("&Rename
936                  helptext = _("Rename the map"))                  helptext = _("Rename the map"))
937  _method_command("layer_add", _("&Add Layer..."), "AddLayer",  _method_command("layer_add", _("&Add Layer..."), "AddLayer",
938                  helptext = _("Add a new layer to active map"))                  helptext = _("Add a new layer to active map"))
939    _method_command("rasterlayer_add", _("&Add Image Layer..."), "AddRasterLayer",
940                    helptext = _("Add a new image layer to active map"))
941  _method_command("layer_remove", _("&Remove Layer"), "RemoveLayer",  _method_command("layer_remove", _("&Remove Layer"), "RemoveLayer",
942                  helptext = _("Remove selected layer(s)"),                  helptext = _("Remove selected layer(s)"),
943                  sensitive = _can_remove_layer)                  sensitive = _can_remove_layer)
# Line 844  _method_command("layer_remove", _("&Remo Line 945  _method_command("layer_remove", _("&Remo
945  # Layer menu  # Layer menu
946  _method_command("layer_projection", _("Pro&jection..."), "LayerProjection",  _method_command("layer_projection", _("Pro&jection..."), "LayerProjection",
947                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
948    _method_command("layer_duplicate", _("&Duplicate"), "DuplicateLayer",
949                    helptext = _("Duplicate selected layer(s)"),
950              sensitive = lambda context: context.mainwindow.CanDuplicateLayer())
951  _method_command("layer_raise", _("&Raise"), "RaiseLayer",  _method_command("layer_raise", _("&Raise"), "RaiseLayer",
952                  helptext = _("Raise selected layer(s)"),                  helptext = _("Raise selected layer(s)"),
953                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
# Line 863  _method_command("layer_properties", _("& Line 967  _method_command("layer_properties", _("&
967                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
968  _method_command("layer_jointable", _("&Join Table..."), "LayerJoinTable",  _method_command("layer_jointable", _("&Join Table..."), "LayerJoinTable",
969                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
970    
971    def _can_unjoin(context):
972        """Return whether the Layer/Unjoin command can be executed.
973    
974        This is the case if a layer is selected and that layer has a
975        shapestore that has an original shapestore.
976        """
977        layer = context.mainwindow.SelectedLayer()
978        if layer is None:
979            return 0
980        getstore = getattr(layer, "ShapeStore", None)
981        if getstore is not None:
982            return getstore().OrigShapeStore() is not None
983        else:
984            return 0
985  _method_command("layer_unjointable", _("&Unjoin Table..."), "LayerUnjoinTable",  _method_command("layer_unjointable", _("&Unjoin Table..."), "LayerUnjoinTable",
986                  sensitive = _has_selected_layer)                  sensitive = _can_unjoin)
987    
988  # Table menu  # Table menu
989  _method_command("table_open", _("&Open..."), "TableOpen")  _method_command("table_open", _("&Open..."), "TableOpen")
990  _method_command("table_close", _("&Close"), "TableClose")  _method_command("table_close", _("&Close"), "TableClose",
991           sensitive = lambda context: bool(context.session.UnreferencedTables()))
992  _method_command("table_show", _("&Show"), "TableShow")  _method_command("table_show", _("&Show"), "TableShow")
 _method_command("table_hide", _("&Hide"), "TableHide")  
993  _method_command("table_join", _("&Join..."), "TableJoin")  _method_command("table_join", _("&Join..."), "TableJoin")
994    
995  #  Export only under Windows ...  #  Export only under Windows ...
996  map_menu = ["layer_add", "layer_remove", "map_rename",  map_menu = ["layer_add", "rasterlayer_add", "layer_remove", "map_rename",
997                          None,                          None,
998                          "map_projection",                          "map_projection",
999                          None,                          None,
# Line 905  main_menu = Menu("<main>", "<main>", Line 1024  main_menu = Menu("<main>", "<main>",
1024                          None,                          None,
1025                          "layer_show", "layer_hide",                          "layer_show", "layer_hide",
1026                          None,                          None,
1027                            "layer_duplicate",
1028                            None,
1029                          "layer_projection",                          "layer_projection",
1030                          None,                          None,
1031                          "layer_show_table",                          "layer_show_table",
# Line 915  main_menu = Menu("<main>", "<main>", Line 1036  main_menu = Menu("<main>", "<main>",
1036                    Menu("table", _("&Table"),                    Menu("table", _("&Table"),
1037                         ["table_open", "table_close",                         ["table_open", "table_close",
1038                         None,                         None,
1039                         "table_show", "table_hide",                         "table_show",
1040                         None,                         None,
1041                         "table_join"]),                         "table_join"]),
1042                    Menu("help", _("&Help"),                    Menu("help", _("&Help"),
Legend:
Removed from v.923  
changed lines
  Added in v.1108
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26