/[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 33 by bh, Thu Sep 6 15:31:31 2001 UTC revision 193 by bh, Wed May 29 10:33:41 2002 UTC
# Line 1  Line 1 
1  # Copyright (C) 2001 by Intevation GmbH  # Copyright (C) 2001, 2002 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  __version__ = "$Revision$"  __version__ = "$Revision$"
14    
15  import sys, os  import os
16    
17  from wxPython.wx import *  from wxPython.wx import *
18    
19  import Thuban  import Thuban
20  from Thuban.Model.session import Session  from Thuban.Model.session import create_empty_session
 from Thuban.Model.map import Map  
21  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer
22  from Thuban.Model.color import Color  from Thuban.Model.color import Color
23  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
# Line 27  import view Line 26  import view
26  import tree  import tree
27  import proj4dialog  import proj4dialog
28  import tableview, identifyview  import tableview, identifyview
29    from menu import Menu
30    
31  import main  import main
32  from command import registry, Command  from command import registry, Command
33  from messages import SELECTED_SHAPE  from messages import SELECTED_SHAPE, VIEW_POSITION
34    
35    
36  # the directory where the toolbar icons are stored  # the directory where the toolbar icons are stored
# Line 44  class MainWindow(wxFrame): Line 44  class MainWindow(wxFrame):
44          wxFrame.__init__(self, parent, ID, 'Thuban',          wxFrame.__init__(self, parent, ID, 'Thuban',
45                           wxDefaultPosition, wxSize(400, 300))                           wxDefaultPosition, wxSize(400, 300))
46    
47            self.interactor = interactor
48    
49          self.CreateStatusBar()          self.CreateStatusBar()
50          self.SetStatusText("This is the wxPython-based "          self.SetStatusText("This is the wxPython-based "
51                        "Graphical User Interface for exploring geographic data")                        "Graphical User Interface for exploring geographic data")
# Line 52  class MainWindow(wxFrame): Line 54  class MainWindow(wxFrame):
54    
55          self.init_ids()          self.init_ids()
56    
57          menuBar = wxMenuBar()          # creat the menubar from the main_menu description
58            self.SetMenuBar(self.build_menu_bar(main_menu))
         menu = wxMenu()  
         menuBar.Append(menu, "&File");  
         for name in ["new_session", "open_session", None,  
                      "save_session", "save_session_as", None,  
                      "exit"]:  
             self.add_menu_command(menu, name)  
   
         menu = wxMenu()  
         menuBar.Append(menu, "&Map");  
         for name in ["map_projection",  
                      None,  
                      "map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool",  
                      "map_identify_tool", "map_label_tool",  
                      None,  
                      "map_full_extent",  
                      None,  
                      "map_print"]:  
             self.add_menu_command(menu, name)  
   
         menu = wxMenu()  
         menuBar.Append(menu, "&Layer");  
         for name in ["layer_add", "layer_remove",  
                      None,  
                      "layer_fill_color", "layer_transparent_fill",  
                      "layer_ourline_color", "layer_no_outline",  
                      None,  
                      "layer_raise", "layer_lower",  
                      None,  
                      "layer_show", "layer_hide",  
                      None,  
                      "layer_show_table"]:  
             self.add_menu_command(menu, name)  
   
         menu = wxMenu()  
         menuBar.Append(menu, "&Help");  
         self.add_menu_command(menu, "help_about")  
   
         self.SetMenuBar(menuBar)  
59    
60          # toolbar          # Similarly, create the toolbar from main_toolbar
61          toolbar = self.CreateToolBar(wxTB_3DBUTTONS)          toolbar = self.build_toolbar(main_toolbar)
         for name in ["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool",  
                      "map_identify_tool", "map_label_tool"]:  
             self.add_toolbar_command(toolbar, name)  
62          # call Realize to make sure that the tools appear.          # call Realize to make sure that the tools appear.
63          toolbar.Realize()          toolbar.Realize()
64    
65          # Create the map canvas          # Create the map canvas
66          canvas = view.MapCanvas(self, -1, interactor)          canvas = view.MapCanvas(self, -1, interactor)
67            canvas.Subscribe(VIEW_POSITION, self.view_position_changed)
68          self.canvas = canvas          self.canvas = canvas
69    
70          self.init_dialogs()          self.init_dialogs()
# Line 116  class MainWindow(wxFrame): Line 78  class MainWindow(wxFrame):
78          self.current_id = 6000          self.current_id = 6000
79          self.id_to_name = {}          self.id_to_name = {}
80          self.name_to_id = {}          self.name_to_id = {}
81            self.events_bound = {}
82    
83      def get_id(self, name):      def get_id(self, name):
84          """Return the wxWindows id for the command named name.          """Return the wxWindows id for the command named name.
# Line 128  class MainWindow(wxFrame): Line 91  class MainWindow(wxFrame):
91              self.name_to_id[name] = ID              self.name_to_id[name] = ID
92              self.id_to_name[ID] = name              self.id_to_name[ID] = name
93          return ID          return ID
94            
95        def bind_command_events(self, command, ID):
96            """Bind the necessary events for the given command and ID"""
97            if not self.events_bound.has_key(ID):
98                # the events haven't been bound yet
99                EVT_MENU(self, ID, self.invoke_command)
100                if command.IsDynamic():
101                    EVT_UPDATE_UI(self, ID, self.update_command_ui)
102    
103        def build_menu_bar(self, menudesc):
104            """Build and return the menu bar from the menu description"""
105            menu_bar = wxMenuBar()
106    
107            for item in menudesc.items:
108                # here the items must all be Menu instances themselves
109                menu_bar.Append(self.build_menu(item), item.title)
110    
111            return menu_bar
112    
113        def build_menu(self, menudesc):
114            """Build and return a wxMenu from a menudescription"""
115            wxmenu = wxMenu()
116            last = None
117            for item in menudesc.items:
118                # here the items must all be Menu instances themselves
119                if item is None:
120                    # a separator. Only add one if the last item was not a
121                    # separator
122                    if last is not None:
123                        wxmenu.AppendSeparator()
124                elif isinstance(item, Menu):
125                    # a submenu
126                    wxmenu.AppendMenu(wxNewId(), item.title, self.build_menu(item))
127                else:
128                    # must the name the name of a command
129                    self.add_menu_command(wxmenu, item)
130                last = item
131            return wxmenu
132    
133        def build_toolbar(self, toolbardesc):
134            """Build and return the main toolbar window from a toolbar description
135    
136            The parameter should be an instance of the Menu class but it
137            should not contain submenus.
138            """
139            toolbar = self.CreateToolBar(wxTB_3DBUTTONS)
140    
141            # set the size of the tools' bitmaps. Not needed on wxGTK, but
142            # on Windows, although it doesn't work very well there. It seems
143            # that only 16x16 icons are really supported on windows.
144            # We probably shouldn't hardwire the bitmap size here.
145            toolbar.SetToolBitmapSize(wxSize(24, 24))
146    
147            for item in toolbardesc.items:
148                if item is None:
149                    toolbar.AddSeparator()
150                else:
151                    # assume it's a string.
152                    self.add_toolbar_command(toolbar, item)
153    
154            return toolbar
155    
156      def add_menu_command(self, menu, name):      def add_menu_command(self, menu, name):
157          """Add the command with name name to the menu menu.          """Add the command with name name to the menu menu.
158    
# Line 142  class MainWindow(wxFrame): Line 166  class MainWindow(wxFrame):
166                  ID = self.get_id(name)                  ID = self.get_id(name)
167                  menu.Append(ID, command.Title(), command.HelpText(),                  menu.Append(ID, command.Title(), command.HelpText(),
168                              command.IsCheckCommand())                              command.IsCheckCommand())
169                  EVT_MENU(self, ID, self.invoke_command)                  self.bind_command_events(command, ID)
                 if command.IsDynamic():  
                     EVT_UPDATE_UI(self, ID, self.update_command_ui)  
170              else:              else:
171                  print "Unknown command %s" % name                  print "Unknown command %s" % name
172    
# Line 166  class MainWindow(wxFrame): Line 188  class MainWindow(wxFrame):
188                  toolbar.AddTool(ID, bitmap,                  toolbar.AddTool(ID, bitmap,
189                                  shortHelpString = command.HelpText(),                                  shortHelpString = command.HelpText(),
190                                  isToggle = command.IsCheckCommand())                                  isToggle = command.IsCheckCommand())
191                    self.bind_command_events(command, ID)
192              else:              else:
193                  print "Unknown command %s" % name                  print "Unknown command %s" % name
194    
# Line 187  class MainWindow(wxFrame): Line 210  class MainWindow(wxFrame):
210                  event.Check(command.Checked(self))                  event.Check(command.Checked(self))
211    
212      def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION):      def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION):
213          """Run a modla message box with the given text, title and flags          """Run a modal message box with the given text, title and flags
214          and return the result"""          and return the result"""
215          dlg = wxMessageDialog(self, text, title, flags)          dlg = wxMessageDialog(self, text, title, flags)
216          result = dlg.ShowModal()          result = dlg.ShowModal()
# Line 215  class MainWindow(wxFrame): Line 238  class MainWindow(wxFrame):
238      def get_open_dialog(self, name):      def get_open_dialog(self, name):
239          return self.dialogs.get(name)          return self.dialogs.get(name)
240    
241        def view_position_changed(self):
242            pos = self.canvas.CurrentPosition()
243            if pos is not None:
244                text = "(%10.10g, %10.10g)" % pos
245            else:
246                text = ""
247            self.SetStatusText(text)
248    
249        def save_modified_session(self, can_veto = 1):
250            """If the current session has been modified, ask the user
251            whether to save it and do so if requested. Return the outcome of
252            the dialog (either wxID_OK, wxID_CANCEL or wxID_NO). If the
253            dialog wasn't run return wxID_NO.
254    
255            If the can_veto parameter is true (default) the dialog includes
256            a cancel button, otherwise not.
257            """
258            if main.app.session.WasModified():
259                flags = wxYES_NO | wxICON_QUESTION
260                if can_veto:
261                    flags = flags | wxCANCEL
262                result = self.RunMessageBox("Exit",
263                                            ("The session has been modified."
264                                             " Do you want to save it?"),
265                                            flags)
266                if result == wxID_YES:
267                    self.SaveSession()
268            else:
269                result = wxID_NO
270            return result
271    
272      def NewSession(self):      def NewSession(self):
273          session = Session("")          self.save_modified_session()
274          session.AddMap(Map(""))          main.app.SetSession(create_empty_session())
         main.app.SetSession(session)  
275    
276      def OpenSession(self):      def OpenSession(self):
277            self.save_modified_session()
278          dlg = wxFileDialog(self, "Select a session file", ".", "",          dlg = wxFileDialog(self, "Select a session file", ".", "",
279                             "*.session", wxOPEN)                             "*.thuban", wxOPEN)
280          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
281              main.app.OpenSession(dlg.GetPath())              main.app.OpenSession(dlg.GetPath())
282          dlg.Destroy()          dlg.Destroy()
283    
284      def SaveSession(self):      def SaveSession(self):
285            if main.app.session.filename == None:
286                self.SaveSessionAs()
287          main.app.SaveSession()          main.app.SaveSession()
288    
289      def SaveSessionAs(self):      def SaveSessionAs(self):
290          dlg = wxFileDialog(self, "Enter a filename for session", ".", "",          dlg = wxFileDialog(self, "Enter a filename for session", ".", "",
291                             "*.session", wxOPEN)                             "*.thuban", wxOPEN)
292          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
293              main.app.session.SetFilename(dlg.GetPath())              main.app.session.SetFilename(dlg.GetPath())
294              main.app.SaveSession()              main.app.SaveSession()
# Line 242  class MainWindow(wxFrame): Line 298  class MainWindow(wxFrame):
298          self.Close(false)          self.Close(false)
299    
300      def OnClose(self, event):      def OnClose(self, event):
301          veto = 0          result = self.save_modified_session(can_veto = event.CanVeto())
302          if main.app.session.WasModified():          if result == wxID_CANCEL:
             flags = wxYES_NO | wxICON_QUESTION  
             if event.CanVeto():  
                 flags = flags | wxCANCEL  
             result = self.RunMessageBox("Exit",  
                                         ("The session has been modified."  
                                          " Do you want to save it?"),  
                                         flags)  
             if result == wxID_YES:  
                 self.SaveSession()  
             elif result == wxID_CANCEL:  
                 veto = 1  
   
         if veto:  
303              event.Veto()              event.Veto()
304          else:          else:
305              self.Destroy()              self.Destroy()
# Line 264  class MainWindow(wxFrame): Line 307  class MainWindow(wxFrame):
307      def SetMap(self, map):      def SetMap(self, map):
308          self.canvas.SetMap(map)          self.canvas.SetMap(map)
309    
310        def ShowSessionTree(self):
311            name = "session_tree"
312            dialog = self.get_open_dialog(name)
313            if dialog is None:
314                dialog = tree.SessionTreeView(self, main.app, name)
315                self.add_dialog(name, dialog)
316                dialog.Show(true)
317            else:
318                # FIXME: bring dialog to front here
319                pass
320    
321      def About(self):      def About(self):
322          self.RunMessageBox("About",          self.RunMessageBox("About",
323                             ("Thuban is a program for\n"                             ("Thuban is a program for\n"
# Line 273  class MainWindow(wxFrame): Line 327  class MainWindow(wxFrame):
327                             wxOK | wxICON_INFORMATION)                             wxOK | wxICON_INFORMATION)
328    
329      def AddLayer(self):      def AddLayer(self):
330          dlg = wxFileDialog(self, "Select a session file", ".", "", "*.*",          dlg = wxFileDialog(self, "Select a data file", ".", "", "*.*",
331                             wxOPEN)                             wxOPEN)
332          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
333              filename = dlg.GetPath()              filename = dlg.GetPath()
# Line 314  class MainWindow(wxFrame): Line 368  class MainWindow(wxFrame):
368    
369          If no layer is selected, return None          If no layer is selected, return None
370          """          """
371          tree = main.app.tree.tree          return self.interactor.SelectedLayer()
         layer = tree.GetPyData(tree.GetSelection())  
         if isinstance(layer, Layer):  
             return layer  
         return None  
372    
373      def has_selected_layer(self):      def has_selected_layer(self):
374          """Return true if a layer is currently selected"""          """Return true if a layer is currently selected"""
375          tree = main.app.tree.tree          return self.interactor.HasSelectedLayer()
         layer = tree.GetPyData(tree.GetSelection())  
         return isinstance(layer, Layer)  
376    
377      def choose_color(self):      def choose_color(self):
378          """Run the color selection dialog and return the selected color.          """Run the color selection dialog and return the selected color.
# Line 383  class MainWindow(wxFrame): Line 431  class MainWindow(wxFrame):
431              name = "table_view" + str(id(table))              name = "table_view" + str(id(table))
432              dialog = self.get_open_dialog(name)              dialog = self.get_open_dialog(name)
433              if dialog is None:              if dialog is None:
434                  dialog = tableview.TableFrame(self, main.app.interactor, name,                  dialog = tableview.TableFrame(self, self.interactor, name,
435                                                "Table: %s" % layer.Title(),                                                "Table: %s" % layer.Title(),
436                                                layer, table)                                                layer, table)
437                  self.add_dialog(name, dialog)                  self.add_dialog(name, dialog)
# Line 396  class MainWindow(wxFrame): Line 444  class MainWindow(wxFrame):
444          map = self.canvas.Map()          map = self.canvas.Map()
445          proj = map.projection          proj = map.projection
446          if proj is None:          if proj is None:
447              proj4Dlg = proj4dialog.Proj4Dialog(NULL, None)              proj4Dlg = proj4dialog.Proj4Dialog(NULL, None, map.BoundingBox())
448          else:          else:
449              proj4Dlg = proj4dialog.Proj4Dialog(NULL, map.projection.params)              proj4Dlg = proj4dialog.Proj4Dialog(NULL, map.projection.params,
450                                                   map.BoundingBox())
451          if proj4Dlg.ShowModal() == wxID_OK:          if proj4Dlg.ShowModal() == wxID_OK:
452              params = proj4Dlg.GetParams()              params = proj4Dlg.GetParams()
453              if params is not None:              if params is not None:
# Line 419  class MainWindow(wxFrame): Line 468  class MainWindow(wxFrame):
468    
469      def IdentifyTool(self):      def IdentifyTool(self):
470          self.canvas.IdentifyTool()          self.canvas.IdentifyTool()
471            self.identify_view_on_demand(None, None)
472    
473      def LabelTool(self):      def LabelTool(self):
474          self.canvas.LabelTool()          self.canvas.LabelTool()
# Line 433  class MainWindow(wxFrame): Line 483  class MainWindow(wxFrame):
483          name = "identify_view"          name = "identify_view"
484          if self.canvas.CurrentTool() == "IdentifyTool":          if self.canvas.CurrentTool() == "IdentifyTool":
485              if not self.dialog_open(name):              if not self.dialog_open(name):
486                  dialog = identifyview.IdentifyView(self, main.app.interactor,                  dialog = identifyview.IdentifyView(self, self.interactor, name)
                                                    name)  
487                  self.add_dialog(name, dialog)                  self.add_dialog(name, dialog)
488                  dialog.Show(true)                  dialog.Show(true)
489              else:              else:
# Line 451  def call_method(context, methodname, *ar Line 500  def call_method(context, methodname, *ar
500      """Call the context's method methodname with args *args"""      """Call the context's method methodname with args *args"""
501      apply(getattr(context, methodname), args)      apply(getattr(context, methodname), args)
502    
503  def _method_command(name, title, method, helptext = "", sensitive = None):  def _method_command(name, title, method, helptext = "",
504                        icon = "", sensitive = None):
505      """Add a command implemented by a method of the context object"""      """Add a command implemented by a method of the context object"""
506      registry.Add(Command(name, title, call_method, args=(method,),      registry.Add(Command(name, title, call_method, args=(method,),
507                           helptext = helptext, sensitive = sensitive))                           helptext = helptext, icon = icon,
508                             sensitive = sensitive))
509    
510  def _tool_command(name, title, method, toolname, helptext = "",  def _tool_command(name, title, method, toolname, helptext = "",
511                    icon = ""):                    icon = ""):
512      """Add a tool command"""      """Add a tool command"""
# Line 473  _method_command("new_session", "&New Ses Line 525  _method_command("new_session", "&New Ses
525  _method_command("open_session", "&Open Session", "OpenSession")  _method_command("open_session", "&Open Session", "OpenSession")
526  _method_command("save_session", "&Save Session", "SaveSession")  _method_command("save_session", "&Save Session", "SaveSession")
527  _method_command("save_session_as", "Save Session &As", "SaveSessionAs")  _method_command("save_session_as", "Save Session &As", "SaveSessionAs")
528    _method_command("show_session_tree", "Show Session &Tree", "ShowSessionTree")
529  _method_command("exit", "&Exit", "Exit")  _method_command("exit", "&Exit", "Exit")
530    
531  # Help menu  # Help menu
# Line 492  _tool_command("map_identify_tool", "&Ide Line 545  _tool_command("map_identify_tool", "&Ide
545                helptext = "Switch to map-mode 'identify'", icon = "identify")                helptext = "Switch to map-mode 'identify'", icon = "identify")
546  _tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool",  _tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool",
547                helptext = "Add/Remove labels", icon = "label")                helptext = "Add/Remove labels", icon = "label")
548  _method_command("map_full_extent", "&Full extent", "FullExtent")  _method_command("map_full_extent", "&Full extent", "FullExtent",
549                   helptext = "Full Extent", icon = "fullextent")
550  _method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map")  _method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map")
551    
552  # Layer menu  # Layer menu
553  _method_command("layer_add", "&Add", "AddLayer",  _method_command("layer_add", "&Add Layer", "AddLayer",
554                  helptext = "Add a new layer to active map")                  helptext = "Add a new layer to active map")
555  _method_command("layer_remove", "&Remove", "RemoveLayer",  _method_command("layer_remove", "&Remove Layer", "RemoveLayer",
556                  helptext = "Remove selected layer(s)",                  helptext = "Remove selected layer(s)",
557                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
558  _method_command("layer_fill_color", "&Fill Color", "LayerFillColor",  _method_command("layer_fill_color", "&Fill Color", "LayerFillColor",
# Line 508  _method_command("layer_transparent_fill" Line 562  _method_command("layer_transparent_fill"
562                  "LayerTransparentFill",                  "LayerTransparentFill",
563                  helptext = "Do not fill the selected layer(s)",                  helptext = "Do not fill the selected layer(s)",
564                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
565  _method_command("layer_ourline_color", "&Outline Color", "LayerOutlineColor",  _method_command("layer_outline_color", "&Outline Color", "LayerOutlineColor",
566                  helptext = "Set the outline color of selected layer(s)",                  helptext = "Set the outline color of selected layer(s)",
567                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
568  _method_command("layer_no_outline", "&No Outline", "LayerNoOutline",  _method_command("layer_no_outline", "&No Outline", "LayerNoOutline",
# Line 529  _method_command("layer_hide", "&Hide", " Line 583  _method_command("layer_hide", "&Hide", "
583  _method_command("layer_show_table", "Show Ta&ble", "LayerShowTable",  _method_command("layer_show_table", "Show Ta&ble", "LayerShowTable",
584                  helptext = "Show the selected layer's table",                  helptext = "Show the selected layer's table",
585                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
586    
587    
588    # the menu structure
589    main_menu = Menu("<main>", "<main>",
590                     [Menu("file", "&File",
591                           ["new_session", "open_session", None,
592                            "save_session", "save_session_as", None,
593                            "show_session_tree", None,
594                            "exit"]),
595                      Menu("map", "&Map",
596                           ["layer_add", "layer_remove",
597                            None,
598                            "map_projection",
599                            None,
600                            "map_zoom_in_tool", "map_zoom_out_tool",
601                            "map_pan_tool", "map_identify_tool", "map_label_tool",
602                            None,
603                            "map_full_extent",
604                            None,
605                            "map_print"]),
606                      Menu("layer", "&Layer",
607                           ["layer_fill_color", "layer_transparent_fill",
608                            "layer_outline_color", "layer_no_outline",
609                            None,
610                            "layer_raise", "layer_lower",
611                            None,
612                            "layer_show", "layer_hide",
613                            None,
614                            "layer_show_table"]),
615                      Menu("help", "&Help",
616                           ["help_about"])])
617    
618    # the main toolbar
619    
620    main_toolbar = Menu("<toolbar>", "<toolbar>",
621                        ["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool",
622                         "map_identify_tool", "map_label_tool", "map_full_extent"])
Legend:
Removed from v.33  
changed lines
  Added in v.193
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26