/[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 110 by jan, Sun Apr 21 17:28:16 2002 UTC revision 281 by bh, Mon Aug 26 18:16:36 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, create_empty_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, size = wxSize(-1, -1)):
45                           wxDefaultPosition, wxSize(400, 300))          wxFrame.__init__(self, parent, ID, title, wxDefaultPosition, size)
46    
47            self.application = application
48          self.interactor = interactor          self.interactor = interactor
49    
50          self.CreateStatusBar()          self.CreateStatusBar()
51          self.SetStatusText("This is the wxPython-based "          if initial_message:
52                        "Graphical User Interface for exploring geographic data")              self.SetStatusText(initial_message)
53    
54          self.identify_view = None          self.identify_view = None
55    
56          self.init_ids()          self.init_ids()
57    
58          menuBar = wxMenuBar()          # creat the menubar from the main_menu description
59            self.SetMenuBar(self.build_menu_bar(main_menu))
60    
61          menu = wxMenu()          # Similarly, create the toolbar from main_toolbar
62          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 ["layer_add", "layer_remove",  
                      None,  
                      "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_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)  
   
         # set the size of the tools' bitmaps. Not needed on wxGTK, but  
         # on Windows. We probably shouldn't hardwire the bitmap size  
         # here  
         toolbar.SetToolBitmapSize(wxSize(24, 24))  
   
         for name in ["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool",  
                      "map_identify_tool", "map_label_tool", "map_full_extent"]:  
             self.add_toolbar_command(toolbar, name)  
63          # call Realize to make sure that the tools appear.          # call Realize to make sure that the tools appear.
64          toolbar.Realize()          toolbar.Realize()
65    
66          # Create the map canvas          # Create the map canvas
67          canvas = view.MapCanvas(self, -1, interactor)          canvas = view.MapCanvas(self, -1, interactor)
68            canvas.Subscribe(VIEW_POSITION, self.view_position_changed)
69          self.canvas = canvas          self.canvas = canvas
70    
71          self.init_dialogs()          self.init_dialogs()
# Line 124  class MainWindow(wxFrame): Line 79  class MainWindow(wxFrame):
79          self.current_id = 6000          self.current_id = 6000
80          self.id_to_name = {}          self.id_to_name = {}
81          self.name_to_id = {}          self.name_to_id = {}
82            self.events_bound = {}
83    
84      def get_id(self, name):      def get_id(self, name):
85          """Return the wxWindows id for the command named name.          """Return the wxWindows id for the command named name.
# Line 136  class MainWindow(wxFrame): Line 92  class MainWindow(wxFrame):
92              self.name_to_id[name] = ID              self.name_to_id[name] = ID
93              self.id_to_name[ID] = name              self.id_to_name[ID] = name
94          return ID          return ID
95            
96        def bind_command_events(self, command, ID):
97            """Bind the necessary events for the given command and ID"""
98            if not self.events_bound.has_key(ID):
99                # the events haven't been bound yet
100                EVT_MENU(self, ID, self.invoke_command)
101                if command.IsDynamic():
102                    EVT_UPDATE_UI(self, ID, self.update_command_ui)
103    
104        def build_menu_bar(self, menudesc):
105            """Build and return the menu bar from the menu description"""
106            menu_bar = wxMenuBar()
107    
108            for item in menudesc.items:
109                # here the items must all be Menu instances themselves
110                menu_bar.Append(self.build_menu(item), item.title)
111    
112            return menu_bar
113    
114        def build_menu(self, menudesc):
115            """Build and return a wxMenu from a menudescription"""
116            wxmenu = wxMenu()
117            last = None
118            for item in menudesc.items:
119                # here the items must all be Menu instances themselves
120                if item is None:
121                    # a separator. Only add one if the last item was not a
122                    # separator
123                    if last is not None:
124                        wxmenu.AppendSeparator()
125                elif isinstance(item, Menu):
126                    # a submenu
127                    wxmenu.AppendMenu(wxNewId(), item.title, self.build_menu(item))
128                else:
129                    # must the name the name of a command
130                    self.add_menu_command(wxmenu, item)
131                last = item
132            return wxmenu
133    
134        def build_toolbar(self, toolbardesc):
135            """Build and return the main toolbar window from a toolbar description
136    
137            The parameter should be an instance of the Menu class but it
138            should not contain submenus.
139            """
140            toolbar = self.CreateToolBar(wxTB_3DBUTTONS)
141    
142            # set the size of the tools' bitmaps. Not needed on wxGTK, but
143            # on Windows, although it doesn't work very well there. It seems
144            # that only 16x16 icons are really supported on windows.
145            # We probably shouldn't hardwire the bitmap size here.
146            toolbar.SetToolBitmapSize(wxSize(24, 24))
147    
148            for item in toolbardesc.items:
149                if item is None:
150                    toolbar.AddSeparator()
151                else:
152                    # assume it's a string.
153                    self.add_toolbar_command(toolbar, item)
154    
155            return toolbar
156    
157      def add_menu_command(self, menu, name):      def add_menu_command(self, menu, name):
158          """Add the command with name name to the menu menu.          """Add the command with name name to the menu menu.
159    
# Line 150  class MainWindow(wxFrame): Line 167  class MainWindow(wxFrame):
167                  ID = self.get_id(name)                  ID = self.get_id(name)
168                  menu.Append(ID, command.Title(), command.HelpText(),                  menu.Append(ID, command.Title(), command.HelpText(),
169                              command.IsCheckCommand())                              command.IsCheckCommand())
170                  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)  
171              else:              else:
172                  print "Unknown command %s" % name                  print "Unknown command %s" % name
173    
# Line 174  class MainWindow(wxFrame): Line 189  class MainWindow(wxFrame):
189                  toolbar.AddTool(ID, bitmap,                  toolbar.AddTool(ID, bitmap,
190                                  shortHelpString = command.HelpText(),                                  shortHelpString = command.HelpText(),
191                                  isToggle = command.IsCheckCommand())                                  isToggle = command.IsCheckCommand())
192                    self.bind_command_events(command, ID)
193              else:              else:
194                  print "Unknown command %s" % name                  print "Unknown command %s" % name
195    
196        def Context(self):
197            """Return the context object for a command invoked from this window
198            """
199            return Context(self.application, self.application.Session(), self)
200    
201      def invoke_command(self, event):      def invoke_command(self, event):
202          name = self.id_to_name.get(event.GetId())          name = self.id_to_name.get(event.GetId())
203          if name is not None:          if name is not None:
204              command = registry.Command(name)              command = registry.Command(name)
205              command.Execute(self)              command.Execute(self.Context())
206          else:          else:
207              print "Unknown command ID %d" % event.GetId()              print "Unknown command ID %d" % event.GetId()
208    
209      def update_command_ui(self, event):      def update_command_ui(self, event):
210          #print "update_command_ui", self.id_to_name[event.GetId()]          #print "update_command_ui", self.id_to_name[event.GetId()]
211            context = self.Context()
212          command = registry.Command(self.id_to_name[event.GetId()])          command = registry.Command(self.id_to_name[event.GetId()])
213          if command is not None:          if command is not None:
214              event.Enable(command.Sensitive(self))              event.Enable(command.Sensitive(context))
215              event.SetText(command.DynText(self))              event.SetText(command.DynText(context))
216              if command.IsCheckCommand():              if command.IsCheckCommand():
217                  event.Check(command.Checked(self))                  event.Check(command.Checked(context))
218    
219      def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION):      def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION):
220          """Run a modla message box with the given text, title and flags          """Run a modal message box with the given text, title and flags
221          and return the result"""          and return the result"""
222          dlg = wxMessageDialog(self, text, title, flags)          dlg = wxMessageDialog(self, text, title, flags)
223          result = dlg.ShowModal()          result = dlg.ShowModal()
# Line 223  class MainWindow(wxFrame): Line 245  class MainWindow(wxFrame):
245      def get_open_dialog(self, name):      def get_open_dialog(self, name):
246          return self.dialogs.get(name)          return self.dialogs.get(name)
247    
248        def view_position_changed(self):
249            pos = self.canvas.CurrentPosition()
250            if pos is not None:
251                text = "(%10.10g, %10.10g)" % pos
252            else:
253                text = ""
254            self.SetStatusText(text)
255    
256      def save_modified_session(self, can_veto = 1):      def save_modified_session(self, can_veto = 1):
257          """If the current session has been modified, ask the user          """If the current session has been modified, ask the user
258          whether to save it and do so if requested. Return the outcome of          whether to save it and do so if requested. Return the outcome of
# Line 232  class MainWindow(wxFrame): Line 262  class MainWindow(wxFrame):
262          If the can_veto parameter is true (default) the dialog includes          If the can_veto parameter is true (default) the dialog includes
263          a cancel button, otherwise not.          a cancel button, otherwise not.
264          """          """
265          if main.app.session.WasModified():          if self.application.session.WasModified():
266              flags = wxYES_NO | wxICON_QUESTION              flags = wxYES_NO | wxICON_QUESTION
267              if can_veto:              if can_veto:
268                  flags = flags | wxCANCEL                  flags = flags | wxCANCEL
# Line 248  class MainWindow(wxFrame): Line 278  class MainWindow(wxFrame):
278    
279      def NewSession(self):      def NewSession(self):
280          self.save_modified_session()          self.save_modified_session()
281          main.app.SetSession(create_empty_session())          self.application.SetSession(create_empty_session())
282    
283      def OpenSession(self):      def OpenSession(self):
284          self.save_modified_session()          self.save_modified_session()
285          dlg = wxFileDialog(self, "Select a session file", ".", "",          dlg = wxFileDialog(self, "Select a session file", ".", "",
286                             "*.session", wxOPEN)                             "*.thuban", wxOPEN)
287          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
288              main.app.OpenSession(dlg.GetPath())              self.application.OpenSession(dlg.GetPath())
289          dlg.Destroy()          dlg.Destroy()
290    
291      def SaveSession(self):      def SaveSession(self):
292          if main.app.session.filename == None:          if self.application.session.filename == None:
293              self.SaveSessionAs()              self.SaveSessionAs()
294          main.app.SaveSession()          self.application.SaveSession()
295    
296      def SaveSessionAs(self):      def SaveSessionAs(self):
297          dlg = wxFileDialog(self, "Enter a filename for session", ".", "",          dlg = wxFileDialog(self, "Enter a filename for session", ".", "",
298                             "*.session", wxOPEN)                             "*.thuban", wxOPEN)
299          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
300              main.app.session.SetFilename(dlg.GetPath())              self.application.session.SetFilename(dlg.GetPath())
301              main.app.SaveSession()              self.application.SaveSession()
302          dlg.Destroy()          dlg.Destroy()
303    
304      def Exit(self):      def Exit(self):
# Line 288  class MainWindow(wxFrame): Line 318  class MainWindow(wxFrame):
318          name = "session_tree"          name = "session_tree"
319          dialog = self.get_open_dialog(name)          dialog = self.get_open_dialog(name)
320          if dialog is None:          if dialog is None:
321              dialog = tree.SessionTreeView(self, main.app, name)              dialog = tree.SessionTreeView(self, self.application, name)
322              self.add_dialog(name, dialog)              self.add_dialog(name, dialog)
323              dialog.Show(true)              dialog.Show(true)
324          else:          else:
# Line 304  class MainWindow(wxFrame): Line 334  class MainWindow(wxFrame):
334                             wxOK | wxICON_INFORMATION)                             wxOK | wxICON_INFORMATION)
335    
336      def AddLayer(self):      def AddLayer(self):
337          dlg = wxFileDialog(self, "Select a session file", ".", "", "*.*",          dlg = wxFileDialog(self, "Select a data file", ".", "", "*.*",
338                             wxOPEN)                             wxOPEN)
339          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
340              filename = dlg.GetPath()              filename = dlg.GetPath()
# Line 334  class MainWindow(wxFrame): Line 364  class MainWindow(wxFrame):
364          layer = self.current_layer()          layer = self.current_layer()
365          if layer is not None:          if layer is not None:
366              self.canvas.Map().RaiseLayer(layer)              self.canvas.Map().RaiseLayer(layer)
367            
368      def LowerLayer(self):      def LowerLayer(self):
369          layer = self.current_layer()          layer = self.current_layer()
370          if layer is not None:          if layer is not None:
# Line 408  class MainWindow(wxFrame): Line 438  class MainWindow(wxFrame):
438              name = "table_view" + str(id(table))              name = "table_view" + str(id(table))
439              dialog = self.get_open_dialog(name)              dialog = self.get_open_dialog(name)
440              if dialog is None:              if dialog is None:
441                  dialog = tableview.TableFrame(self, self.interactor, name,                  dialog = tableview.LayerTableFrame(self, self.interactor, name,
442                                                "Table: %s" % layer.Title(),                                                     "Table: %s" % layer.Title(),
443                                                layer, table)                                                     layer, table)
444                  self.add_dialog(name, dialog)                  self.add_dialog(name, dialog)
445                  dialog.Show(true)                  dialog.Show(true)
446              else:              else:
# Line 474  class MainWindow(wxFrame): Line 504  class MainWindow(wxFrame):
504    
505  # Helper functions to define common command implementations  # Helper functions to define common command implementations
506  def call_method(context, methodname, *args):  def call_method(context, methodname, *args):
507      """Call the context's method methodname with args *args"""      """Call the mainwindow's method methodname with args *args"""
508      apply(getattr(context, methodname), args)      apply(getattr(context.mainwindow, methodname), args)
509    
510  def _method_command(name, title, method, helptext = "",  def _method_command(name, title, method, helptext = "",
511                      icon = "", sensitive = None):                      icon = "", sensitive = None):
512      """Add a command implemented by a method of the context object"""      """Add a command implemented by a method of the mainwindow object"""
513      registry.Add(Command(name, title, call_method, args=(method,),      registry.Add(Command(name, title, call_method, args=(method,),
514                           helptext = helptext, icon = icon,                           helptext = helptext, icon = icon,
515                           sensitive = sensitive))                           sensitive = sensitive))
516    
517    def make_check_current_tool(toolname):
518        """Return a function that tests if the currently active tool is toolname
519    
520        The returned function can be called with the context and returns
521        true iff the currently active tool's name is toolname. It's directly
522        usable as the 'checked' callback of a command.
523        """
524        def check_current_tool(context, name=toolname):
525            return context.mainwindow.canvas.CurrentTool() == name
526        return check_current_tool
527    
528  def _tool_command(name, title, method, toolname, helptext = "",  def _tool_command(name, title, method, toolname, helptext = "",
529                    icon = ""):                    icon = ""):
530      """Add a tool command"""      """Add a tool command"""
     def check_current_tool(context, name=toolname):  
         return context.canvas.CurrentTool() == name  
531      registry.Add(Command(name, title, call_method, args=(method,),      registry.Add(Command(name, title, call_method, args=(method,),
532                           helptext = helptext, icon = icon,                           helptext = helptext, icon = icon,
533                           checked = check_current_tool))                           checked = make_check_current_tool(toolname)))
534    
535  def _has_selected_layer(context):  def _has_selected_layer(context):
536      """Return true if a layer is selected in the context"""      """Return true if a layer is selected in the context"""
537      return context.has_selected_layer()      return context.mainwindow.has_selected_layer()
538    
539    def _has_tree_window_shown(context):
540        """Return true if the tree window is shown"""
541        return context.mainwindow.get_open_dialog("session_tree") is None
542    
543  # File menu  # File menu
544  _method_command("new_session", "&New Session", "NewSession")  _method_command("new_session", "&New Session", "NewSession")
545  _method_command("open_session", "&Open Session", "OpenSession")  _method_command("open_session", "&Open Session", "OpenSession")
546  _method_command("save_session", "&Save Session", "SaveSession")  _method_command("save_session", "&Save Session", "SaveSession")
547  _method_command("save_session_as", "Save Session &As", "SaveSessionAs")  _method_command("save_session_as", "Save Session &As", "SaveSessionAs")
548    _method_command("show_session_tree", "Show Session &Tree", "ShowSessionTree",
549                    sensitive = _has_tree_window_shown)
550  _method_command("exit", "&Exit", "Exit")  _method_command("exit", "&Exit", "Exit")
551    
552  # Help menu  # Help menu
# Line 521  _tool_command("map_identify_tool", "&Ide Line 566  _tool_command("map_identify_tool", "&Ide
566                helptext = "Switch to map-mode 'identify'", icon = "identify")                helptext = "Switch to map-mode 'identify'", icon = "identify")
567  _tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool",  _tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool",
568                helptext = "Add/Remove labels", icon = "label")                helptext = "Add/Remove labels", icon = "label")
 #_tool_command("map_full_extent", "&Full extent", "FullExtent", "FullExtent",  
 #              helptext = "Full Extent", icon = "fullextent")  
569  _method_command("map_full_extent", "&Full extent", "FullExtent",  _method_command("map_full_extent", "&Full extent", "FullExtent",
570                 helptext = "Full Extent", icon = "fullextent")                 helptext = "Full Extent", icon = "fullextent")
571  _method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map")  _method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map")
# Line 540  _method_command("layer_transparent_fill" Line 583  _method_command("layer_transparent_fill"
583                  "LayerTransparentFill",                  "LayerTransparentFill",
584                  helptext = "Do not fill the selected layer(s)",                  helptext = "Do not fill the selected layer(s)",
585                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
586  _method_command("layer_ourline_color", "&Outline Color", "LayerOutlineColor",  _method_command("layer_outline_color", "&Outline Color", "LayerOutlineColor",
587                  helptext = "Set the outline color of selected layer(s)",                  helptext = "Set the outline color of selected layer(s)",
588                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
589  _method_command("layer_no_outline", "&No Outline", "LayerNoOutline",  _method_command("layer_no_outline", "&No Outline", "LayerNoOutline",
# Line 561  _method_command("layer_hide", "&Hide", " Line 604  _method_command("layer_hide", "&Hide", "
604  _method_command("layer_show_table", "Show Ta&ble", "LayerShowTable",  _method_command("layer_show_table", "Show Ta&ble", "LayerShowTable",
605                  helptext = "Show the selected layer's table",                  helptext = "Show the selected layer's table",
606                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
607    
608    
609    # the menu structure
610    main_menu = Menu("<main>", "<main>",
611                     [Menu("file", "&File",
612                           ["new_session", "open_session", None,
613                            "save_session", "save_session_as", None,
614                            "show_session_tree", None,
615                            "exit"]),
616                      Menu("map", "&Map",
617                           ["layer_add", "layer_remove",
618                            None,
619                            "map_projection",
620                            None,
621                            "map_zoom_in_tool", "map_zoom_out_tool",
622                            "map_pan_tool", "map_identify_tool", "map_label_tool",
623                            None,
624                            "map_full_extent",
625                            None,
626                            "map_print"]),
627                      Menu("layer", "&Layer",
628                           ["layer_fill_color", "layer_transparent_fill",
629                            "layer_outline_color", "layer_no_outline",
630                            None,
631                            "layer_raise", "layer_lower",
632                            None,
633                            "layer_show", "layer_hide",
634                            None,
635                            "layer_show_table"]),
636                      Menu("help", "&Help",
637                           ["help_about"])])
638    
639    # the main toolbar
640    
641    main_toolbar = Menu("<toolbar>", "<toolbar>",
642                        ["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool",
643                         "map_identify_tool", "map_label_tool", "map_full_extent"])
Legend:
Removed from v.110  
changed lines
  Added in v.281
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26