/[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 49 by bh, Mon Sep 10 16:03:59 2001 UTC revision 235 by bh, Tue Jul 23 10:56:29 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  from context import Context
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 40  bitmapext = ".xpm" Line 40  bitmapext = ".xpm"
40    
41  class MainWindow(wxFrame):  class MainWindow(wxFrame):
42    
43      def __init__(self, parent, ID, interactor):      def __init__(self, parent, ID, title, application, interactor,
44          wxFrame.__init__(self, parent, ID, 'Thuban',                   initial_message = None):
45                           wxDefaultPosition, wxSize(400, 300))          wxFrame.__init__(self, parent, ID, title, wxDefaultPosition,
46                             wxSize(400, 300))
47    
48            self.application = application
49          self.interactor = interactor          self.interactor = interactor
50    
51          self.CreateStatusBar()          self.CreateStatusBar()
52          self.SetStatusText("This is the wxPython-based "          if initial_message:
53                        "Graphical User Interface for exploring geographic data")              self.SetStatusText(initial_message)
54    
55          self.identify_view = None          self.identify_view = None
56    
57          self.init_ids()          self.init_ids()
58    
59          menuBar = wxMenuBar()          # creat the menubar from the main_menu description
60            self.SetMenuBar(self.build_menu_bar(main_menu))
61    
62          menu = wxMenu()          # Similarly, create the toolbar from main_toolbar
63          menuBar.Append(menu, "&File");          toolbar = self.build_toolbar(main_toolbar)
         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)  
   
         # toolbar  
         toolbar = self.CreateToolBar(wxTB_3DBUTTONS)  
         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)  
64          # call Realize to make sure that the tools appear.          # call Realize to make sure that the tools appear.
65          toolbar.Realize()          toolbar.Realize()
66    
67          # Create the map canvas          # Create the map canvas
68          canvas = view.MapCanvas(self, -1, interactor)          canvas = view.MapCanvas(self, -1, interactor)
69            canvas.Subscribe(VIEW_POSITION, self.view_position_changed)
70          self.canvas = canvas          self.canvas = canvas
71    
72          self.init_dialogs()          self.init_dialogs()
# Line 118  class MainWindow(wxFrame): Line 80  class MainWindow(wxFrame):
80          self.current_id = 6000          self.current_id = 6000
81          self.id_to_name = {}          self.id_to_name = {}
82          self.name_to_id = {}          self.name_to_id = {}
83            self.events_bound = {}
84    
85      def get_id(self, name):      def get_id(self, name):
86          """Return the wxWindows id for the command named name.          """Return the wxWindows id for the command named name.
# Line 130  class MainWindow(wxFrame): Line 93  class MainWindow(wxFrame):
93              self.name_to_id[name] = ID              self.name_to_id[name] = ID
94              self.id_to_name[ID] = name              self.id_to_name[ID] = name
95          return ID          return ID
96            
97        def bind_command_events(self, command, ID):
98            """Bind the necessary events for the given command and ID"""
99            if not self.events_bound.has_key(ID):
100                # the events haven't been bound yet
101                EVT_MENU(self, ID, self.invoke_command)
102                if command.IsDynamic():
103                    EVT_UPDATE_UI(self, ID, self.update_command_ui)
104    
105        def build_menu_bar(self, menudesc):
106            """Build and return the menu bar from the menu description"""
107            menu_bar = wxMenuBar()
108    
109            for item in menudesc.items:
110                # here the items must all be Menu instances themselves
111                menu_bar.Append(self.build_menu(item), item.title)
112    
113            return menu_bar
114    
115        def build_menu(self, menudesc):
116            """Build and return a wxMenu from a menudescription"""
117            wxmenu = wxMenu()
118            last = None
119            for item in menudesc.items:
120                # here the items must all be Menu instances themselves
121                if item is None:
122                    # a separator. Only add one if the last item was not a
123                    # separator
124                    if last is not None:
125                        wxmenu.AppendSeparator()
126                elif isinstance(item, Menu):
127                    # a submenu
128                    wxmenu.AppendMenu(wxNewId(), item.title, self.build_menu(item))
129                else:
130                    # must the name the name of a command
131                    self.add_menu_command(wxmenu, item)
132                last = item
133            return wxmenu
134    
135        def build_toolbar(self, toolbardesc):
136            """Build and return the main toolbar window from a toolbar description
137    
138            The parameter should be an instance of the Menu class but it
139            should not contain submenus.
140            """
141            toolbar = self.CreateToolBar(wxTB_3DBUTTONS)
142    
143            # set the size of the tools' bitmaps. Not needed on wxGTK, but
144            # on Windows, although it doesn't work very well there. It seems
145            # that only 16x16 icons are really supported on windows.
146            # We probably shouldn't hardwire the bitmap size here.
147            toolbar.SetToolBitmapSize(wxSize(24, 24))
148    
149            for item in toolbardesc.items:
150                if item is None:
151                    toolbar.AddSeparator()
152                else:
153                    # assume it's a string.
154                    self.add_toolbar_command(toolbar, item)
155    
156            return toolbar
157    
158      def add_menu_command(self, menu, name):      def add_menu_command(self, menu, name):
159          """Add the command with name name to the menu menu.          """Add the command with name name to the menu menu.
160    
# Line 144  class MainWindow(wxFrame): Line 168  class MainWindow(wxFrame):
168                  ID = self.get_id(name)                  ID = self.get_id(name)
169                  menu.Append(ID, command.Title(), command.HelpText(),                  menu.Append(ID, command.Title(), command.HelpText(),
170                              command.IsCheckCommand())                              command.IsCheckCommand())
171                  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)  
172              else:              else:
173                  print "Unknown command %s" % name                  print "Unknown command %s" % name
174    
# Line 168  class MainWindow(wxFrame): Line 190  class MainWindow(wxFrame):
190                  toolbar.AddTool(ID, bitmap,                  toolbar.AddTool(ID, bitmap,
191                                  shortHelpString = command.HelpText(),                                  shortHelpString = command.HelpText(),
192                                  isToggle = command.IsCheckCommand())                                  isToggle = command.IsCheckCommand())
193                    self.bind_command_events(command, ID)
194              else:              else:
195                  print "Unknown command %s" % name                  print "Unknown command %s" % name
196    
# Line 175  class MainWindow(wxFrame): Line 198  class MainWindow(wxFrame):
198          name = self.id_to_name.get(event.GetId())          name = self.id_to_name.get(event.GetId())
199          if name is not None:          if name is not None:
200              command = registry.Command(name)              command = registry.Command(name)
201              command.Execute(self)              context = Context(self.application, self.application.Session(),
202                                  self)
203                command.Execute(context)
204          else:          else:
205              print "Unknown command ID %d" % event.GetId()              print "Unknown command ID %d" % event.GetId()
206    
207      def update_command_ui(self, event):      def update_command_ui(self, event):
208          #print "update_command_ui", self.id_to_name[event.GetId()]          #print "update_command_ui", self.id_to_name[event.GetId()]
209            context = Context(self.application, self.application.Session(), self)
210          command = registry.Command(self.id_to_name[event.GetId()])          command = registry.Command(self.id_to_name[event.GetId()])
211          if command is not None:          if command is not None:
212              event.Enable(command.Sensitive(self))              event.Enable(command.Sensitive(context))
213              event.SetText(command.DynText(self))              event.SetText(command.DynText(context))
214              if command.IsCheckCommand():              if command.IsCheckCommand():
215                  event.Check(command.Checked(self))                  event.Check(command.Checked(context))
216    
217      def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION):      def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION):
218          """Run a modla message box with the given text, title and flags          """Run a modal message box with the given text, title and flags
219          and return the result"""          and return the result"""
220          dlg = wxMessageDialog(self, text, title, flags)          dlg = wxMessageDialog(self, text, title, flags)
221          result = dlg.ShowModal()          result = dlg.ShowModal()
# Line 217  class MainWindow(wxFrame): Line 243  class MainWindow(wxFrame):
243      def get_open_dialog(self, name):      def get_open_dialog(self, name):
244          return self.dialogs.get(name)          return self.dialogs.get(name)
245    
246        def view_position_changed(self):
247            pos = self.canvas.CurrentPosition()
248            if pos is not None:
249                text = "(%10.10g, %10.10g)" % pos
250            else:
251                text = ""
252            self.SetStatusText(text)
253    
254        def save_modified_session(self, can_veto = 1):
255            """If the current session has been modified, ask the user
256            whether to save it and do so if requested. Return the outcome of
257            the dialog (either wxID_OK, wxID_CANCEL or wxID_NO). If the
258            dialog wasn't run return wxID_NO.
259    
260            If the can_veto parameter is true (default) the dialog includes
261            a cancel button, otherwise not.
262            """
263            if self.application.session.WasModified():
264                flags = wxYES_NO | wxICON_QUESTION
265                if can_veto:
266                    flags = flags | wxCANCEL
267                result = self.RunMessageBox("Exit",
268                                            ("The session has been modified."
269                                             " Do you want to save it?"),
270                                            flags)
271                if result == wxID_YES:
272                    self.SaveSession()
273            else:
274                result = wxID_NO
275            return result
276    
277      def NewSession(self):      def NewSession(self):
278          session = Session("")          self.save_modified_session()
279          session.AddMap(Map(""))          self.application.SetSession(create_empty_session())
         main.app.SetSession(session)  
280    
281      def OpenSession(self):      def OpenSession(self):
282            self.save_modified_session()
283          dlg = wxFileDialog(self, "Select a session file", ".", "",          dlg = wxFileDialog(self, "Select a session file", ".", "",
284                             "*.session", wxOPEN)                             "*.thuban", wxOPEN)
285          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
286              main.app.OpenSession(dlg.GetPath())              self.application.OpenSession(dlg.GetPath())
287          dlg.Destroy()          dlg.Destroy()
288    
289      def SaveSession(self):      def SaveSession(self):
290          main.app.SaveSession()          if self.application.session.filename == None:
291                self.SaveSessionAs()
292            self.application.SaveSession()
293    
294      def SaveSessionAs(self):      def SaveSessionAs(self):
295          dlg = wxFileDialog(self, "Enter a filename for session", ".", "",          dlg = wxFileDialog(self, "Enter a filename for session", ".", "",
296                             "*.session", wxOPEN)                             "*.thuban", wxOPEN)
297          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
298              main.app.session.SetFilename(dlg.GetPath())              self.application.session.SetFilename(dlg.GetPath())
299              main.app.SaveSession()              self.application.SaveSession()
300          dlg.Destroy()          dlg.Destroy()
301    
302      def Exit(self):      def Exit(self):
303          self.Close(false)          self.Close(false)
304    
305      def OnClose(self, event):      def OnClose(self, event):
306          veto = 0          result = self.save_modified_session(can_veto = event.CanVeto())
307          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:  
308              event.Veto()              event.Veto()
309          else:          else:
310              self.Destroy()              self.Destroy()
# Line 270  class MainWindow(wxFrame): Line 316  class MainWindow(wxFrame):
316          name = "session_tree"          name = "session_tree"
317          dialog = self.get_open_dialog(name)          dialog = self.get_open_dialog(name)
318          if dialog is None:          if dialog is None:
319              dialog = tree.SessionTreeView(self, main.app, name)              dialog = tree.SessionTreeView(self, self.application, name)
320              self.add_dialog(name, dialog)              self.add_dialog(name, dialog)
321              dialog.Show(true)              dialog.Show(true)
322          else:          else:
# Line 286  class MainWindow(wxFrame): Line 332  class MainWindow(wxFrame):
332                             wxOK | wxICON_INFORMATION)                             wxOK | wxICON_INFORMATION)
333    
334      def AddLayer(self):      def AddLayer(self):
335          dlg = wxFileDialog(self, "Select a session file", ".", "", "*.*",          dlg = wxFileDialog(self, "Select a data file", ".", "", "*.*",
336                             wxOPEN)                             wxOPEN)
337          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
338              filename = dlg.GetPath()              filename = dlg.GetPath()
# Line 316  class MainWindow(wxFrame): Line 362  class MainWindow(wxFrame):
362          layer = self.current_layer()          layer = self.current_layer()
363          if layer is not None:          if layer is not None:
364              self.canvas.Map().RaiseLayer(layer)              self.canvas.Map().RaiseLayer(layer)
365            
366      def LowerLayer(self):      def LowerLayer(self):
367          layer = self.current_layer()          layer = self.current_layer()
368          if layer is not None:          if layer is not None:
# Line 403  class MainWindow(wxFrame): Line 449  class MainWindow(wxFrame):
449          map = self.canvas.Map()          map = self.canvas.Map()
450          proj = map.projection          proj = map.projection
451          if proj is None:          if proj is None:
452              proj4Dlg = proj4dialog.Proj4Dialog(NULL, None)              proj4Dlg = proj4dialog.Proj4Dialog(NULL, None, map.BoundingBox())
453          else:          else:
454              proj4Dlg = proj4dialog.Proj4Dialog(NULL, map.projection.params)              proj4Dlg = proj4dialog.Proj4Dialog(NULL, map.projection.params,
455                                                   map.BoundingBox())
456          if proj4Dlg.ShowModal() == wxID_OK:          if proj4Dlg.ShowModal() == wxID_OK:
457              params = proj4Dlg.GetParams()              params = proj4Dlg.GetParams()
458              if params is not None:              if params is not None:
# Line 455  class MainWindow(wxFrame): Line 502  class MainWindow(wxFrame):
502    
503  # Helper functions to define common command implementations  # Helper functions to define common command implementations
504  def call_method(context, methodname, *args):  def call_method(context, methodname, *args):
505      """Call the context's method methodname with args *args"""      """Call the mainwindow's method methodname with args *args"""
506      apply(getattr(context, methodname), args)      apply(getattr(context.mainwindow, methodname), args)
507    
508  def _method_command(name, title, method, helptext = "", sensitive = None):  def _method_command(name, title, method, helptext = "",
509      """Add a command implemented by a method of the context object"""                      icon = "", sensitive = None):
510        """Add a command implemented by a method of the mainwindow object"""
511      registry.Add(Command(name, title, call_method, args=(method,),      registry.Add(Command(name, title, call_method, args=(method,),
512                           helptext = helptext, sensitive = sensitive))                           helptext = helptext, icon = icon,
513                             sensitive = sensitive))
514    
515  def _tool_command(name, title, method, toolname, helptext = "",  def _tool_command(name, title, method, toolname, helptext = "",
516                    icon = ""):                    icon = ""):
517      """Add a tool command"""      """Add a tool command"""
518      def check_current_tool(context, name=toolname):      def check_current_tool(context, name=toolname):
519          return context.canvas.CurrentTool() == name          return context.mainwindow.canvas.CurrentTool() == name
520      registry.Add(Command(name, title, call_method, args=(method,),      registry.Add(Command(name, title, call_method, args=(method,),
521                           helptext = helptext, icon = icon,                           helptext = helptext, icon = icon,
522                           checked = check_current_tool))                           checked = check_current_tool))
523    
524  def _has_selected_layer(context):  def _has_selected_layer(context):
525      """Return true if a layer is selected in the context"""      """Return true if a layer is selected in the context"""
526      return context.has_selected_layer()      return context.mainwindow.has_selected_layer()
527    
528  # File menu  # File menu
529  _method_command("new_session", "&New Session", "NewSession")  _method_command("new_session", "&New Session", "NewSession")
530  _method_command("open_session", "&Open Session", "OpenSession")  _method_command("open_session", "&Open Session", "OpenSession")
531  _method_command("save_session", "&Save Session", "SaveSession")  _method_command("save_session", "&Save Session", "SaveSession")
532  _method_command("save_session_as", "Save Session &As", "SaveSessionAs")  _method_command("save_session_as", "Save Session &As", "SaveSessionAs")
533    _method_command("show_session_tree", "Show Session &Tree", "ShowSessionTree")
534  _method_command("exit", "&Exit", "Exit")  _method_command("exit", "&Exit", "Exit")
535    
536  # Help menu  # Help menu
# Line 499  _tool_command("map_identify_tool", "&Ide Line 550  _tool_command("map_identify_tool", "&Ide
550                helptext = "Switch to map-mode 'identify'", icon = "identify")                helptext = "Switch to map-mode 'identify'", icon = "identify")
551  _tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool",  _tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool",
552                helptext = "Add/Remove labels", icon = "label")                helptext = "Add/Remove labels", icon = "label")
553  _method_command("map_full_extent", "&Full extent", "FullExtent")  _method_command("map_full_extent", "&Full extent", "FullExtent",
554                   helptext = "Full Extent", icon = "fullextent")
555  _method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map")  _method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map")
556    
557  # Layer menu  # Layer menu
558  _method_command("layer_add", "&Add", "AddLayer",  _method_command("layer_add", "&Add Layer", "AddLayer",
559                  helptext = "Add a new layer to active map")                  helptext = "Add a new layer to active map")
560  _method_command("layer_remove", "&Remove", "RemoveLayer",  _method_command("layer_remove", "&Remove Layer", "RemoveLayer",
561                  helptext = "Remove selected layer(s)",                  helptext = "Remove selected layer(s)",
562                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
563  _method_command("layer_fill_color", "&Fill Color", "LayerFillColor",  _method_command("layer_fill_color", "&Fill Color", "LayerFillColor",
# Line 515  _method_command("layer_transparent_fill" Line 567  _method_command("layer_transparent_fill"
567                  "LayerTransparentFill",                  "LayerTransparentFill",
568                  helptext = "Do not fill the selected layer(s)",                  helptext = "Do not fill the selected layer(s)",
569                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
570  _method_command("layer_ourline_color", "&Outline Color", "LayerOutlineColor",  _method_command("layer_outline_color", "&Outline Color", "LayerOutlineColor",
571                  helptext = "Set the outline color of selected layer(s)",                  helptext = "Set the outline color of selected layer(s)",
572                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
573  _method_command("layer_no_outline", "&No Outline", "LayerNoOutline",  _method_command("layer_no_outline", "&No Outline", "LayerNoOutline",
# Line 536  _method_command("layer_hide", "&Hide", " Line 588  _method_command("layer_hide", "&Hide", "
588  _method_command("layer_show_table", "Show Ta&ble", "LayerShowTable",  _method_command("layer_show_table", "Show Ta&ble", "LayerShowTable",
589                  helptext = "Show the selected layer's table",                  helptext = "Show the selected layer's table",
590                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
591    
592    
593    # the menu structure
594    main_menu = Menu("<main>", "<main>",
595                     [Menu("file", "&File",
596                           ["new_session", "open_session", None,
597                            "save_session", "save_session_as", None,
598                            "show_session_tree", None,
599                            "exit"]),
600                      Menu("map", "&Map",
601                           ["layer_add", "layer_remove",
602                            None,
603                            "map_projection",
604                            None,
605                            "map_zoom_in_tool", "map_zoom_out_tool",
606                            "map_pan_tool", "map_identify_tool", "map_label_tool",
607                            None,
608                            "map_full_extent",
609                            None,
610                            "map_print"]),
611                      Menu("layer", "&Layer",
612                           ["layer_fill_color", "layer_transparent_fill",
613                            "layer_outline_color", "layer_no_outline",
614                            None,
615                            "layer_raise", "layer_lower",
616                            None,
617                            "layer_show", "layer_hide",
618                            None,
619                            "layer_show_table"]),
620                      Menu("help", "&Help",
621                           ["help_about"])])
622    
623    # the main toolbar
624    
625    main_toolbar = Menu("<toolbar>", "<toolbar>",
626                        ["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool",
627                         "map_identify_tool", "map_label_tool", "map_full_extent"])
Legend:
Removed from v.49  
changed lines
  Added in v.235
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26