/[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 102 by jan, Fri Apr 19 14:22:25 2002 UTC revision 227 by bh, Thu Jul 18 16:27:11 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, application, interactor):
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.application = application
48          self.interactor = interactor          self.interactor = interactor
49    
50          self.CreateStatusBar()          self.CreateStatusBar()
# Line 54  class MainWindow(wxFrame): Line 55  class MainWindow(wxFrame):
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"]:  
             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    
# Line 181  class MainWindow(wxFrame): Line 197  class MainWindow(wxFrame):
197          name = self.id_to_name.get(event.GetId())          name = self.id_to_name.get(event.GetId())
198          if name is not None:          if name is not None:
199              command = registry.Command(name)              command = registry.Command(name)
200              command.Execute(self)              context = Context(self.application, self.application.Session(),
201                                  self)
202                command.Execute(context)
203          else:          else:
204              print "Unknown command ID %d" % event.GetId()              print "Unknown command ID %d" % event.GetId()
205    
206      def update_command_ui(self, event):      def update_command_ui(self, event):
207          #print "update_command_ui", self.id_to_name[event.GetId()]          #print "update_command_ui", self.id_to_name[event.GetId()]
208            context = Context(self.application, self.application.Session(), self)
209          command = registry.Command(self.id_to_name[event.GetId()])          command = registry.Command(self.id_to_name[event.GetId()])
210          if command is not None:          if command is not None:
211              event.Enable(command.Sensitive(self))              event.Enable(command.Sensitive(context))
212              event.SetText(command.DynText(self))              event.SetText(command.DynText(context))
213              if command.IsCheckCommand():              if command.IsCheckCommand():
214                  event.Check(command.Checked(self))                  event.Check(command.Checked(context))
215    
216      def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION):      def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION):
217          """Run a modla message box with the given text, title and flags          """Run a modal message box with the given text, title and flags
218          and return the result"""          and return the result"""
219          dlg = wxMessageDialog(self, text, title, flags)          dlg = wxMessageDialog(self, text, title, flags)
220          result = dlg.ShowModal()          result = dlg.ShowModal()
# Line 223  class MainWindow(wxFrame): Line 242  class MainWindow(wxFrame):
242      def get_open_dialog(self, name):      def get_open_dialog(self, name):
243          return self.dialogs.get(name)          return self.dialogs.get(name)
244    
245        def view_position_changed(self):
246            pos = self.canvas.CurrentPosition()
247            if pos is not None:
248                text = "(%10.10g, %10.10g)" % pos
249            else:
250                text = ""
251            self.SetStatusText(text)
252    
253      def save_modified_session(self, can_veto = 1):      def save_modified_session(self, can_veto = 1):
254          """If the current session has been modified, ask the user          """If the current session has been modified, ask the user
255          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 259  class MainWindow(wxFrame):
259          If the can_veto parameter is true (default) the dialog includes          If the can_veto parameter is true (default) the dialog includes
260          a cancel button, otherwise not.          a cancel button, otherwise not.
261          """          """
262          if main.app.session.WasModified():          if self.application.session.WasModified():
263              flags = wxYES_NO | wxICON_QUESTION              flags = wxYES_NO | wxICON_QUESTION
264              if can_veto:              if can_veto:
265                  flags = flags | wxCANCEL                  flags = flags | wxCANCEL
# Line 248  class MainWindow(wxFrame): Line 275  class MainWindow(wxFrame):
275    
276      def NewSession(self):      def NewSession(self):
277          self.save_modified_session()          self.save_modified_session()
278          main.app.SetSession(create_empty_session())          self.application.SetSession(create_empty_session())
279    
280      def OpenSession(self):      def OpenSession(self):
281          self.save_modified_session()          self.save_modified_session()
282          dlg = wxFileDialog(self, "Select a session file", ".", "",          dlg = wxFileDialog(self, "Select a session file", ".", "",
283                             "*.session", wxOPEN)                             "*.thuban", wxOPEN)
284          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
285              main.app.OpenSession(dlg.GetPath())              self.application.OpenSession(dlg.GetPath())
286          dlg.Destroy()          dlg.Destroy()
287    
288      def SaveSession(self):      def SaveSession(self):
289          if main.app.session.filename == None:          if self.application.session.filename == None:
290              self.SaveSessionAs()              self.SaveSessionAs()
291          main.app.SaveSession()          self.application.SaveSession()
292    
293      def SaveSessionAs(self):      def SaveSessionAs(self):
294          dlg = wxFileDialog(self, "Enter a filename for session", ".", "",          dlg = wxFileDialog(self, "Enter a filename for session", ".", "",
295                             "*.session", wxOPEN)                             "*.thuban", wxOPEN)
296          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
297              main.app.session.SetFilename(dlg.GetPath())              self.application.session.SetFilename(dlg.GetPath())
298              main.app.SaveSession()              self.application.SaveSession()
299          dlg.Destroy()          dlg.Destroy()
300    
301      def Exit(self):      def Exit(self):
# Line 288  class MainWindow(wxFrame): Line 315  class MainWindow(wxFrame):
315          name = "session_tree"          name = "session_tree"
316          dialog = self.get_open_dialog(name)          dialog = self.get_open_dialog(name)
317          if dialog is None:          if dialog is None:
318              dialog = tree.SessionTreeView(self, main.app, name)              dialog = tree.SessionTreeView(self, self.application, name)
319              self.add_dialog(name, dialog)              self.add_dialog(name, dialog)
320              dialog.Show(true)              dialog.Show(true)
321          else:          else:
# Line 304  class MainWindow(wxFrame): Line 331  class MainWindow(wxFrame):
331                             wxOK | wxICON_INFORMATION)                             wxOK | wxICON_INFORMATION)
332    
333      def AddLayer(self):      def AddLayer(self):
334          dlg = wxFileDialog(self, "Select a session file", ".", "", "*.*",          dlg = wxFileDialog(self, "Select a data file", ".", "", "*.*",
335                             wxOPEN)                             wxOPEN)
336          if dlg.ShowModal() == wxID_OK:          if dlg.ShowModal() == wxID_OK:
337              filename = dlg.GetPath()              filename = dlg.GetPath()
# Line 334  class MainWindow(wxFrame): Line 361  class MainWindow(wxFrame):
361          layer = self.current_layer()          layer = self.current_layer()
362          if layer is not None:          if layer is not None:
363              self.canvas.Map().RaiseLayer(layer)              self.canvas.Map().RaiseLayer(layer)
364            
365      def LowerLayer(self):      def LowerLayer(self):
366          layer = self.current_layer()          layer = self.current_layer()
367          if layer is not None:          if layer is not None:
# Line 421  class MainWindow(wxFrame): Line 448  class MainWindow(wxFrame):
448          map = self.canvas.Map()          map = self.canvas.Map()
449          proj = map.projection          proj = map.projection
450          if proj is None:          if proj is None:
451              proj4Dlg = proj4dialog.Proj4Dialog(NULL, None)              proj4Dlg = proj4dialog.Proj4Dialog(NULL, None, map.BoundingBox())
452          else:          else:
453              proj4Dlg = proj4dialog.Proj4Dialog(NULL, map.projection.params)              proj4Dlg = proj4dialog.Proj4Dialog(NULL, map.projection.params,
454                                                   map.BoundingBox())
455          if proj4Dlg.ShowModal() == wxID_OK:          if proj4Dlg.ShowModal() == wxID_OK:
456              params = proj4Dlg.GetParams()              params = proj4Dlg.GetParams()
457              if params is not None:              if params is not None:
# Line 473  class MainWindow(wxFrame): Line 501  class MainWindow(wxFrame):
501    
502  # Helper functions to define common command implementations  # Helper functions to define common command implementations
503  def call_method(context, methodname, *args):  def call_method(context, methodname, *args):
504      """Call the context's method methodname with args *args"""      """Call the mainwindow's method methodname with args *args"""
505      apply(getattr(context, methodname), args)      apply(getattr(context.mainwindow, methodname), args)
506    
507  def _method_command(name, title, method, helptext = "", sensitive = None):  def _method_command(name, title, method, helptext = "",
508      """Add a command implemented by a method of the context object"""                      icon = "", sensitive = None):
509        """Add a command implemented by a method of the mainwindow object"""
510      registry.Add(Command(name, title, call_method, args=(method,),      registry.Add(Command(name, title, call_method, args=(method,),
511                           helptext = helptext, sensitive = sensitive))                           helptext = helptext, icon = icon,
512                             sensitive = sensitive))
513    
514  def _tool_command(name, title, method, toolname, helptext = "",  def _tool_command(name, title, method, toolname, helptext = "",
515                    icon = ""):                    icon = ""):
516      """Add a tool command"""      """Add a tool command"""
517      def check_current_tool(context, name=toolname):      def check_current_tool(context, name=toolname):
518          return context.canvas.CurrentTool() == name          return context.mainwindow.canvas.CurrentTool() == name
519      registry.Add(Command(name, title, call_method, args=(method,),      registry.Add(Command(name, title, call_method, args=(method,),
520                           helptext = helptext, icon = icon,                           helptext = helptext, icon = icon,
521                           checked = check_current_tool))                           checked = check_current_tool))
522    
523  def _has_selected_layer(context):  def _has_selected_layer(context):
524      """Return true if a layer is selected in the context"""      """Return true if a layer is selected in the context"""
525      return context.has_selected_layer()      return context.mainwindow.has_selected_layer()
526    
527  # File menu  # File menu
528  _method_command("new_session", "&New Session", "NewSession")  _method_command("new_session", "&New Session", "NewSession")
529  _method_command("open_session", "&Open Session", "OpenSession")  _method_command("open_session", "&Open Session", "OpenSession")
530  _method_command("save_session", "&Save Session", "SaveSession")  _method_command("save_session", "&Save Session", "SaveSession")
531  _method_command("save_session_as", "Save Session &As", "SaveSessionAs")  _method_command("save_session_as", "Save Session &As", "SaveSessionAs")
532    _method_command("show_session_tree", "Show Session &Tree", "ShowSessionTree")
533  _method_command("exit", "&Exit", "Exit")  _method_command("exit", "&Exit", "Exit")
534    
535  # Help menu  # Help menu
# Line 517  _tool_command("map_identify_tool", "&Ide Line 549  _tool_command("map_identify_tool", "&Ide
549                helptext = "Switch to map-mode 'identify'", icon = "identify")                helptext = "Switch to map-mode 'identify'", icon = "identify")
550  _tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool",  _tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool",
551                helptext = "Add/Remove labels", icon = "label")                helptext = "Add/Remove labels", icon = "label")
552  _method_command("map_full_extent", "&Full extent", "FullExtent")  _method_command("map_full_extent", "&Full extent", "FullExtent",
553                   helptext = "Full Extent", icon = "fullextent")
554  _method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map")  _method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map")
555    
556  # Layer menu  # Layer menu
# Line 533  _method_command("layer_transparent_fill" Line 566  _method_command("layer_transparent_fill"
566                  "LayerTransparentFill",                  "LayerTransparentFill",
567                  helptext = "Do not fill the selected layer(s)",                  helptext = "Do not fill the selected layer(s)",
568                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
569  _method_command("layer_ourline_color", "&Outline Color", "LayerOutlineColor",  _method_command("layer_outline_color", "&Outline Color", "LayerOutlineColor",
570                  helptext = "Set the outline color of selected layer(s)",                  helptext = "Set the outline color of selected layer(s)",
571                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
572  _method_command("layer_no_outline", "&No Outline", "LayerNoOutline",  _method_command("layer_no_outline", "&No Outline", "LayerNoOutline",
# Line 554  _method_command("layer_hide", "&Hide", " Line 587  _method_command("layer_hide", "&Hide", "
587  _method_command("layer_show_table", "Show Ta&ble", "LayerShowTable",  _method_command("layer_show_table", "Show Ta&ble", "LayerShowTable",
588                  helptext = "Show the selected layer's table",                  helptext = "Show the selected layer's table",
589                  sensitive = _has_selected_layer)                  sensitive = _has_selected_layer)
590    
591    
592    # the menu structure
593    main_menu = Menu("<main>", "<main>",
594                     [Menu("file", "&File",
595                           ["new_session", "open_session", None,
596                            "save_session", "save_session_as", None,
597                            "show_session_tree", None,
598                            "exit"]),
599                      Menu("map", "&Map",
600                           ["layer_add", "layer_remove",
601                            None,
602                            "map_projection",
603                            None,
604                            "map_zoom_in_tool", "map_zoom_out_tool",
605                            "map_pan_tool", "map_identify_tool", "map_label_tool",
606                            None,
607                            "map_full_extent",
608                            None,
609                            "map_print"]),
610                      Menu("layer", "&Layer",
611                           ["layer_fill_color", "layer_transparent_fill",
612                            "layer_outline_color", "layer_no_outline",
613                            None,
614                            "layer_raise", "layer_lower",
615                            None,
616                            "layer_show", "layer_hide",
617                            None,
618                            "layer_show_table"]),
619                      Menu("help", "&Help",
620                           ["help_about"])])
621    
622    # the main toolbar
623    
624    main_toolbar = Menu("<toolbar>", "<toolbar>",
625                        ["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool",
626                         "map_identify_tool", "map_label_tool", "map_full_extent"])
Legend:
Removed from v.102  
changed lines
  Added in v.227
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26