28 |
import tableview, identifyview |
import tableview, identifyview |
29 |
from menu import Menu |
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, VIEW_POSITION |
from messages import SELECTED_SHAPE, VIEW_POSITION |
34 |
|
|
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 |
|
|
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. |
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): |
def build_menu_bar(self, menudesc): |
105 |
"""Build and return the menu bar from the menu description""" |
"""Build and return the menu bar from the menu description""" |
106 |
menu_bar = wxMenuBar() |
menu_bar = wxMenuBar() |
112 |
return menu_bar |
return menu_bar |
113 |
|
|
114 |
def build_menu(self, menudesc): |
def build_menu(self, menudesc): |
115 |
"""Build and return a wxMenu from a menudescription""" |
"""Return a wxMenu built from the menu description menudesc""" |
116 |
wxmenu = wxMenu() |
wxmenu = wxMenu() |
117 |
last = None |
last = None |
118 |
for item in menudesc.items: |
for item in menudesc.items: |
|
# here the items must all be Menu instances themselves |
|
119 |
if item is None: |
if item is None: |
120 |
# a separator. Only add one if the last item was not a |
# a separator. Only add one if the last item was not a |
121 |
# separator |
# separator |
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 |
|
|
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 |
|
|
195 |
|
def Context(self): |
196 |
|
"""Return the context object for a command invoked from this window |
197 |
|
""" |
198 |
|
return Context(self.application, self.application.Session(), self) |
199 |
|
|
200 |
def invoke_command(self, event): |
def invoke_command(self, event): |
201 |
name = self.id_to_name.get(event.GetId()) |
name = self.id_to_name.get(event.GetId()) |
202 |
if name is not None: |
if name is not None: |
203 |
command = registry.Command(name) |
command = registry.Command(name) |
204 |
command.Execute(self) |
command.Execute(self.Context()) |
205 |
else: |
else: |
206 |
print "Unknown command ID %d" % event.GetId() |
print "Unknown command ID %d" % event.GetId() |
207 |
|
|
208 |
def update_command_ui(self, event): |
def update_command_ui(self, event): |
209 |
#print "update_command_ui", self.id_to_name[event.GetId()] |
#print "update_command_ui", self.id_to_name[event.GetId()] |
210 |
|
context = self.Context() |
211 |
command = registry.Command(self.id_to_name[event.GetId()]) |
command = registry.Command(self.id_to_name[event.GetId()]) |
212 |
if command is not None: |
if command is not None: |
213 |
event.Enable(command.Sensitive(self)) |
event.Enable(command.Sensitive(context)) |
214 |
event.SetText(command.DynText(self)) |
event.SetText(command.DynText(context)) |
215 |
if command.IsCheckCommand(): |
if command.IsCheckCommand(): |
216 |
event.Check(command.Checked(self)) |
event.Check(command.Checked(context)) |
217 |
|
|
218 |
def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION): |
def RunMessageBox(self, title, text, flags = wxOK | wxICON_INFORMATION): |
219 |
"""Run a modal message box with the given text, title and flags |
"""Run a modal message box with the given text, title and flags |
220 |
and return the result""" |
and return the result""" |
221 |
dlg = wxMessageDialog(self, text, title, flags) |
dlg = wxMessageDialog(self, text, title, flags) |
222 |
|
dlg.CenterOnParent() |
223 |
result = dlg.ShowModal() |
result = dlg.ShowModal() |
224 |
dlg.Destroy() |
dlg.Destroy() |
225 |
return result |
return result |
251 |
text = "(%10.10g, %10.10g)" % pos |
text = "(%10.10g, %10.10g)" % pos |
252 |
else: |
else: |
253 |
text = "" |
text = "" |
254 |
|
self.set_position_text(text) |
255 |
|
|
256 |
|
def set_position_text(self, text): |
257 |
|
"""Set the statusbar text showing the current position. |
258 |
|
|
259 |
|
By default the text is shown in field 0 of the status bar. |
260 |
|
Override this method in derived classes to put it into a |
261 |
|
different field of the statusbar. |
262 |
|
""" |
263 |
self.SetStatusText(text) |
self.SetStatusText(text) |
264 |
|
|
265 |
def save_modified_session(self, can_veto = 1): |
def save_modified_session(self, can_veto = 1): |
271 |
If the can_veto parameter is true (default) the dialog includes |
If the can_veto parameter is true (default) the dialog includes |
272 |
a cancel button, otherwise not. |
a cancel button, otherwise not. |
273 |
""" |
""" |
274 |
if main.app.session.WasModified(): |
if self.application.session.WasModified(): |
275 |
flags = wxYES_NO | wxICON_QUESTION |
flags = wxYES_NO | wxICON_QUESTION |
276 |
if can_veto: |
if can_veto: |
277 |
flags = flags | wxCANCEL |
flags = flags | wxCANCEL |
287 |
|
|
288 |
def NewSession(self): |
def NewSession(self): |
289 |
self.save_modified_session() |
self.save_modified_session() |
290 |
main.app.SetSession(create_empty_session()) |
self.application.SetSession(create_empty_session()) |
291 |
|
|
292 |
def OpenSession(self): |
def OpenSession(self): |
293 |
self.save_modified_session() |
self.save_modified_session() |
294 |
dlg = wxFileDialog(self, "Select a session file", ".", "", |
dlg = wxFileDialog(self, "Select a session file", ".", "", |
295 |
"*.thuban", wxOPEN) |
"*.thuban", wxOPEN) |
296 |
if dlg.ShowModal() == wxID_OK: |
if dlg.ShowModal() == wxID_OK: |
297 |
main.app.OpenSession(dlg.GetPath()) |
self.application.OpenSession(dlg.GetPath()) |
298 |
dlg.Destroy() |
dlg.Destroy() |
299 |
|
|
300 |
def SaveSession(self): |
def SaveSession(self): |
301 |
if main.app.session.filename == None: |
if self.application.session.filename == None: |
302 |
self.SaveSessionAs() |
self.SaveSessionAs() |
303 |
main.app.SaveSession() |
self.application.SaveSession() |
304 |
|
|
305 |
def SaveSessionAs(self): |
def SaveSessionAs(self): |
306 |
dlg = wxFileDialog(self, "Enter a filename for session", ".", "", |
dlg = wxFileDialog(self, "Enter a filename for session", ".", "", |
307 |
"*.thuban", wxOPEN) |
"*.thuban", wxOPEN) |
308 |
if dlg.ShowModal() == wxID_OK: |
if dlg.ShowModal() == wxID_OK: |
309 |
main.app.session.SetFilename(dlg.GetPath()) |
self.application.session.SetFilename(dlg.GetPath()) |
310 |
main.app.SaveSession() |
self.application.SaveSession() |
311 |
dlg.Destroy() |
dlg.Destroy() |
312 |
|
|
313 |
def Exit(self): |
def Exit(self): |
318 |
if result == wxID_CANCEL: |
if result == wxID_CANCEL: |
319 |
event.Veto() |
event.Veto() |
320 |
else: |
else: |
321 |
|
# FIXME: it would be better to tie the unsubscription to |
322 |
|
# wx's destroy event, but that isn't implemented for wxGTK |
323 |
|
# yet. |
324 |
|
self.canvas.Unsubscribe(VIEW_POSITION, self.view_position_changed) |
325 |
self.Destroy() |
self.Destroy() |
326 |
|
|
327 |
def SetMap(self, map): |
def SetMap(self, map): |
328 |
self.canvas.SetMap(map) |
self.canvas.SetMap(map) |
329 |
|
|
330 |
|
def Map(self): |
331 |
|
"""Return the map displayed by this mainwindow""" |
332 |
|
return self.canvas.Map() |
333 |
|
|
334 |
def ShowSessionTree(self): |
def ShowSessionTree(self): |
335 |
name = "session_tree" |
name = "session_tree" |
336 |
dialog = self.get_open_dialog(name) |
dialog = self.get_open_dialog(name) |
337 |
if dialog is None: |
if dialog is None: |
338 |
dialog = tree.SessionTreeView(self, main.app, name) |
dialog = tree.SessionTreeView(self, self.application, name) |
339 |
self.add_dialog(name, dialog) |
self.add_dialog(name, dialog) |
340 |
dialog.Show(true) |
dialog.Show(true) |
341 |
else: |
else: |
346 |
self.RunMessageBox("About", |
self.RunMessageBox("About", |
347 |
("Thuban is a program for\n" |
("Thuban is a program for\n" |
348 |
"exploring geographic data.\n" |
"exploring geographic data.\n" |
349 |
"Copyright (C) 2001 Intevation GmbH.\n" |
"Copyright (C) 2001, 2002 Intevation GmbH.\n" |
350 |
"Thuban is licensed under the GPL"), |
"Thuban is licensed under the GPL"), |
351 |
wxOK | wxICON_INFORMATION) |
wxOK | wxICON_INFORMATION) |
352 |
|
|
377 |
if layer is not None: |
if layer is not None: |
378 |
self.canvas.Map().RemoveLayer(layer) |
self.canvas.Map().RemoveLayer(layer) |
379 |
|
|
380 |
|
def CanRemoveLayer(self): |
381 |
|
"""Return true if the currently selected layer can be deleted. |
382 |
|
|
383 |
|
If no layer is selected return false. |
384 |
|
|
385 |
|
The return value of this method determines whether the remove |
386 |
|
layer command is sensitive in menu. |
387 |
|
""" |
388 |
|
layer = self.current_layer() |
389 |
|
if layer is not None: |
390 |
|
return self.canvas.Map().CanRemoveLayer(layer) |
391 |
|
return 0 |
392 |
|
|
393 |
def RaiseLayer(self): |
def RaiseLayer(self): |
394 |
layer = self.current_layer() |
layer = self.current_layer() |
395 |
if layer is not None: |
if layer is not None: |
396 |
self.canvas.Map().RaiseLayer(layer) |
self.canvas.Map().RaiseLayer(layer) |
397 |
|
|
398 |
def LowerLayer(self): |
def LowerLayer(self): |
399 |
layer = self.current_layer() |
layer = self.current_layer() |
400 |
if layer is not None: |
if layer is not None: |
468 |
name = "table_view" + str(id(table)) |
name = "table_view" + str(id(table)) |
469 |
dialog = self.get_open_dialog(name) |
dialog = self.get_open_dialog(name) |
470 |
if dialog is None: |
if dialog is None: |
471 |
dialog = tableview.TableFrame(self, self.interactor, name, |
dialog = tableview.LayerTableFrame(self, self.interactor, name, |
472 |
"Table: %s" % layer.Title(), |
"Table: %s" % layer.Title(), |
473 |
layer, table) |
layer, table) |
474 |
self.add_dialog(name, dialog) |
self.add_dialog(name, dialog) |
475 |
dialog.Show(true) |
dialog.Show(true) |
476 |
else: |
else: |
534 |
|
|
535 |
# Helper functions to define common command implementations |
# Helper functions to define common command implementations |
536 |
def call_method(context, methodname, *args): |
def call_method(context, methodname, *args): |
537 |
"""Call the context's method methodname with args *args""" |
"""Call the mainwindow's method methodname with args *args""" |
538 |
apply(getattr(context, methodname), args) |
apply(getattr(context.mainwindow, methodname), args) |
539 |
|
|
540 |
def _method_command(name, title, method, helptext = "", |
def _method_command(name, title, method, helptext = "", |
541 |
icon = "", sensitive = None): |
icon = "", sensitive = None): |
542 |
"""Add a command implemented by a method of the context object""" |
"""Add a command implemented by a method of the mainwindow object""" |
543 |
registry.Add(Command(name, title, call_method, args=(method,), |
registry.Add(Command(name, title, call_method, args=(method,), |
544 |
helptext = helptext, icon = icon, |
helptext = helptext, icon = icon, |
545 |
sensitive = sensitive)) |
sensitive = sensitive)) |
546 |
|
|
547 |
|
def make_check_current_tool(toolname): |
548 |
|
"""Return a function that tests if the currently active tool is toolname |
549 |
|
|
550 |
|
The returned function can be called with the context and returns |
551 |
|
true iff the currently active tool's name is toolname. It's directly |
552 |
|
usable as the 'checked' callback of a command. |
553 |
|
""" |
554 |
|
def check_current_tool(context, name=toolname): |
555 |
|
return context.mainwindow.canvas.CurrentTool() == name |
556 |
|
return check_current_tool |
557 |
|
|
558 |
def _tool_command(name, title, method, toolname, helptext = "", |
def _tool_command(name, title, method, toolname, helptext = "", |
559 |
icon = ""): |
icon = "", sensitive = None): |
560 |
"""Add a tool command""" |
"""Add a tool command""" |
|
def check_current_tool(context, name=toolname): |
|
|
return context.canvas.CurrentTool() == name |
|
561 |
registry.Add(Command(name, title, call_method, args=(method,), |
registry.Add(Command(name, title, call_method, args=(method,), |
562 |
helptext = helptext, icon = icon, |
helptext = helptext, icon = icon, |
563 |
checked = check_current_tool)) |
checked = make_check_current_tool(toolname), |
564 |
|
sensitive = sensitive)) |
565 |
|
|
566 |
def _has_selected_layer(context): |
def _has_selected_layer(context): |
567 |
"""Return true if a layer is selected in the context""" |
"""Return true if a layer is selected in the context""" |
568 |
return context.has_selected_layer() |
return context.mainwindow.has_selected_layer() |
569 |
|
|
570 |
|
def _can_remove_layer(context): |
571 |
|
return context.mainwindow.CanRemoveLayer() |
572 |
|
|
573 |
|
def _has_tree_window_shown(context): |
574 |
|
"""Return true if the tree window is shown""" |
575 |
|
return context.mainwindow.get_open_dialog("session_tree") is None |
576 |
|
|
577 |
|
def _has_visible_map(context): |
578 |
|
"""Return true iff theres a visible map in the mainwindow. |
579 |
|
|
580 |
|
A visible map is a map with at least one visible layer.""" |
581 |
|
map = context.mainwindow.Map() |
582 |
|
if map is not None: |
583 |
|
for layer in map.Layers(): |
584 |
|
if layer.Visible(): |
585 |
|
return 1 |
586 |
|
return 0 |
587 |
|
|
588 |
|
|
589 |
# File menu |
# File menu |
590 |
_method_command("new_session", "&New Session", "NewSession") |
_method_command("new_session", "&New Session", "NewSession") |
591 |
_method_command("open_session", "&Open Session", "OpenSession") |
_method_command("open_session", "&Open Session", "OpenSession") |
592 |
_method_command("save_session", "&Save Session", "SaveSession") |
_method_command("save_session", "&Save Session", "SaveSession") |
593 |
_method_command("save_session_as", "Save Session &As", "SaveSessionAs") |
_method_command("save_session_as", "Save Session &As", "SaveSessionAs") |
594 |
_method_command("show_session_tree", "Show Session &Tree", "ShowSessionTree") |
_method_command("show_session_tree", "Show Session &Tree", "ShowSessionTree", |
595 |
_method_command("exit", "&Exit", "Exit") |
sensitive = _has_tree_window_shown) |
596 |
|
_method_command("exit", "E&xit", "Exit") |
597 |
|
|
598 |
# Help menu |
# Help menu |
599 |
_method_command("help_about", "&About", "About") |
_method_command("help_about", "&About", "About") |
603 |
_method_command("map_projection", "Pro&jection", "Projection") |
_method_command("map_projection", "Pro&jection", "Projection") |
604 |
|
|
605 |
_tool_command("map_zoom_in_tool", "&Zoom in", "ZoomInTool", "ZoomInTool", |
_tool_command("map_zoom_in_tool", "&Zoom in", "ZoomInTool", "ZoomInTool", |
606 |
helptext = "Switch to map-mode 'zoom-in'", icon = "zoom_in") |
helptext = "Switch to map-mode 'zoom-in'", icon = "zoom_in", |
607 |
|
sensitive = _has_visible_map) |
608 |
_tool_command("map_zoom_out_tool", "Zoom &out", "ZoomOutTool", "ZoomOutTool", |
_tool_command("map_zoom_out_tool", "Zoom &out", "ZoomOutTool", "ZoomOutTool", |
609 |
helptext = "Switch to map-mode 'zoom-out'", icon = "zoom_out") |
helptext = "Switch to map-mode 'zoom-out'", icon = "zoom_out", |
610 |
|
sensitive = _has_visible_map) |
611 |
_tool_command("map_pan_tool", "&Pan", "PanTool", "PanTool", |
_tool_command("map_pan_tool", "&Pan", "PanTool", "PanTool", |
612 |
helptext = "Switch to map-mode 'pan'", icon = "pan") |
helptext = "Switch to map-mode 'pan'", icon = "pan", |
613 |
|
sensitive = _has_visible_map) |
614 |
_tool_command("map_identify_tool", "&Identify", "IdentifyTool", "IdentifyTool", |
_tool_command("map_identify_tool", "&Identify", "IdentifyTool", "IdentifyTool", |
615 |
helptext = "Switch to map-mode 'identify'", icon = "identify") |
helptext = "Switch to map-mode 'identify'", icon = "identify", |
616 |
|
sensitive = _has_visible_map) |
617 |
_tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool", |
_tool_command("map_label_tool", "&Label", "LabelTool", "LabelTool", |
618 |
helptext = "Add/Remove labels", icon = "label") |
helptext = "Add/Remove labels", icon = "label", |
619 |
|
sensitive = _has_visible_map) |
620 |
_method_command("map_full_extent", "&Full extent", "FullExtent", |
_method_command("map_full_extent", "&Full extent", "FullExtent", |
621 |
helptext = "Full Extent", icon = "fullextent") |
helptext = "Full Extent", icon = "fullextent", |
622 |
|
sensitive = _has_visible_map) |
623 |
_method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map") |
_method_command("map_print", "Prin&t", "PrintMap", helptext = "Print the map") |
624 |
|
|
625 |
# Layer menu |
# Layer menu |
627 |
helptext = "Add a new layer to active map") |
helptext = "Add a new layer to active map") |
628 |
_method_command("layer_remove", "&Remove Layer", "RemoveLayer", |
_method_command("layer_remove", "&Remove Layer", "RemoveLayer", |
629 |
helptext = "Remove selected layer(s)", |
helptext = "Remove selected layer(s)", |
630 |
sensitive = _has_selected_layer) |
sensitive = _can_remove_layer) |
631 |
_method_command("layer_fill_color", "&Fill Color", "LayerFillColor", |
_method_command("layer_fill_color", "&Fill Color", "LayerFillColor", |
632 |
helptext = "Set the fill color of selected layer(s)", |
helptext = "Set the fill color of selected layer(s)", |
633 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_layer) |
691 |
# the main toolbar |
# the main toolbar |
692 |
|
|
693 |
main_toolbar = Menu("<toolbar>", "<toolbar>", |
main_toolbar = Menu("<toolbar>", "<toolbar>", |
694 |
["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool", |
["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool", |
695 |
"map_identify_tool", "map_label_tool", "map_full_extent"]) |
"map_full_extent", None, |
696 |
|
"map_identify_tool", "map_label_tool"]) |