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 |
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, 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 |
|
|
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 |
|
|
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. |
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 |
|
|
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 |
|
|
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 |
|
|
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() |
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 |
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): |
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: |
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: |
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 = "", |
def _method_command(name, title, method, helptext = "", |
508 |
icon = "", sensitive = None): |
icon = "", sensitive = None): |
509 |
"""Add a command implemented by a method of the context object""" |
"""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, icon = icon, |
helptext = helptext, icon = icon, |
512 |
sensitive = sensitive)) |
sensitive = sensitive)) |
513 |
|
|
514 |
|
def make_check_current_tool(toolname): |
515 |
|
"""Return a function that tests if the currently active tool is toolname |
516 |
|
|
517 |
|
The returned function can be called with the context and returns |
518 |
|
true iff the currently active tool's name is toolname. It's directly |
519 |
|
usable as the 'checked' callback of a command. |
520 |
|
""" |
521 |
|
def check_current_tool(context, name=toolname): |
522 |
|
return context.mainwindow.canvas.CurrentTool() == name |
523 |
|
return check_current_tool |
524 |
|
|
525 |
def _tool_command(name, title, method, toolname, helptext = "", |
def _tool_command(name, title, method, toolname, helptext = "", |
526 |
icon = ""): |
icon = ""): |
527 |
"""Add a tool command""" |
"""Add a tool command""" |
|
def check_current_tool(context, name=toolname): |
|
|
return context.canvas.CurrentTool() == name |
|
528 |
registry.Add(Command(name, title, call_method, args=(method,), |
registry.Add(Command(name, title, call_method, args=(method,), |
529 |
helptext = helptext, icon = icon, |
helptext = helptext, icon = icon, |
530 |
checked = check_current_tool)) |
checked = make_check_current_tool(toolname))) |
531 |
|
|
532 |
def _has_selected_layer(context): |
def _has_selected_layer(context): |
533 |
"""Return true if a layer is selected in the context""" |
"""Return true if a layer is selected in the context""" |
534 |
return context.has_selected_layer() |
return context.mainwindow.has_selected_layer() |
535 |
|
|
536 |
|
def _has_tree_window_shown(context): |
537 |
|
"""Return true if the tree window is shown""" |
538 |
|
return context.mainwindow.get_open_dialog("session_tree") is None |
539 |
|
|
540 |
# File menu |
# File menu |
541 |
_method_command("new_session", "&New Session", "NewSession") |
_method_command("new_session", "&New Session", "NewSession") |
542 |
_method_command("open_session", "&Open Session", "OpenSession") |
_method_command("open_session", "&Open Session", "OpenSession") |
543 |
_method_command("save_session", "&Save Session", "SaveSession") |
_method_command("save_session", "&Save Session", "SaveSession") |
544 |
_method_command("save_session_as", "Save Session &As", "SaveSessionAs") |
_method_command("save_session_as", "Save Session &As", "SaveSessionAs") |
545 |
|
_method_command("show_session_tree", "Show Session &Tree", "ShowSessionTree", |
546 |
|
sensitive = _has_tree_window_shown) |
547 |
_method_command("exit", "&Exit", "Exit") |
_method_command("exit", "&Exit", "Exit") |
548 |
|
|
549 |
# Help menu |
# Help menu |
580 |
"LayerTransparentFill", |
"LayerTransparentFill", |
581 |
helptext = "Do not fill the selected layer(s)", |
helptext = "Do not fill the selected layer(s)", |
582 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_layer) |
583 |
_method_command("layer_ourline_color", "&Outline Color", "LayerOutlineColor", |
_method_command("layer_outline_color", "&Outline Color", "LayerOutlineColor", |
584 |
helptext = "Set the outline color of selected layer(s)", |
helptext = "Set the outline color of selected layer(s)", |
585 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_layer) |
586 |
_method_command("layer_no_outline", "&No Outline", "LayerNoOutline", |
_method_command("layer_no_outline", "&No Outline", "LayerNoOutline", |
601 |
_method_command("layer_show_table", "Show Ta&ble", "LayerShowTable", |
_method_command("layer_show_table", "Show Ta&ble", "LayerShowTable", |
602 |
helptext = "Show the selected layer's table", |
helptext = "Show the selected layer's table", |
603 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_layer) |
604 |
|
|
605 |
|
|
606 |
|
# the menu structure |
607 |
|
main_menu = Menu("<main>", "<main>", |
608 |
|
[Menu("file", "&File", |
609 |
|
["new_session", "open_session", None, |
610 |
|
"save_session", "save_session_as", None, |
611 |
|
"show_session_tree", None, |
612 |
|
"exit"]), |
613 |
|
Menu("map", "&Map", |
614 |
|
["layer_add", "layer_remove", |
615 |
|
None, |
616 |
|
"map_projection", |
617 |
|
None, |
618 |
|
"map_zoom_in_tool", "map_zoom_out_tool", |
619 |
|
"map_pan_tool", "map_identify_tool", "map_label_tool", |
620 |
|
None, |
621 |
|
"map_full_extent", |
622 |
|
None, |
623 |
|
"map_print"]), |
624 |
|
Menu("layer", "&Layer", |
625 |
|
["layer_fill_color", "layer_transparent_fill", |
626 |
|
"layer_outline_color", "layer_no_outline", |
627 |
|
None, |
628 |
|
"layer_raise", "layer_lower", |
629 |
|
None, |
630 |
|
"layer_show", "layer_hide", |
631 |
|
None, |
632 |
|
"layer_show_table"]), |
633 |
|
Menu("help", "&Help", |
634 |
|
["help_about"])]) |
635 |
|
|
636 |
|
# the main toolbar |
637 |
|
|
638 |
|
main_toolbar = Menu("<toolbar>", "<toolbar>", |
639 |
|
["map_zoom_in_tool", "map_zoom_out_tool", "map_pan_tool", |
640 |
|
"map_identify_tool", "map_label_tool", "map_full_extent"]) |