30 |
import tree |
import tree |
31 |
import proj4dialog |
import proj4dialog |
32 |
import tableview, identifyview |
import tableview, identifyview |
33 |
import classifier |
from Thuban.UI.classifier import Classifier |
34 |
import legend |
import legend |
35 |
from menu import Menu |
from menu import Menu |
36 |
|
|
37 |
from context import Context |
from context import Context |
38 |
from command import registry, Command, ToolCommand |
from command import registry, Command, ToolCommand |
39 |
from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION, DOCKABLE_DOCKED, DOCKABLE_UNDOCKED, DOCKABLE_CLOSED |
from messages import LAYER_SELECTED, SHAPES_SELECTED, VIEW_POSITION |
40 |
|
|
41 |
from Thuban.UI.dock import DockableWindow, DockFrame, DockPanel |
from Thuban.UI.dock import DockFrame |
42 |
|
|
43 |
|
import resource |
44 |
|
|
45 |
# the directory where the toolbar icons are stored |
import projdialog |
|
bitmapdir = os.path.join(Thuban.__path__[0], os.pardir, "Resources", "Bitmaps") |
|
|
bitmapext = ".xpm" |
|
46 |
|
|
|
ID_WINDOW_LEGEND = 4001 |
|
|
ID_WINDOW_CANVAS = 4002 |
|
47 |
|
|
48 |
|
|
49 |
class MainWindow(DockFrame): |
class MainWindow(DockFrame): |
240 |
command = registry.Command(name) |
command = registry.Command(name) |
241 |
if command is not None: |
if command is not None: |
242 |
ID = self.get_id(name) |
ID = self.get_id(name) |
243 |
filename = os.path.join(bitmapdir, command.Icon()) + bitmapext |
bitmap = resource.GetBitmapResource(command.Icon(), |
244 |
bitmap = wxBitmap(filename, wxBITMAP_TYPE_XPM) |
wxBITMAP_TYPE_XPM) |
245 |
toolbar.AddTool(ID, bitmap, |
toolbar.AddTool(ID, bitmap, |
246 |
shortHelpString = command.HelpText(), |
shortHelpString = command.HelpText(), |
247 |
isToggle = command.IsCheckCommand()) |
isToggle = command.IsCheckCommand()) |
377 |
|
|
378 |
def SaveSessionAs(self): |
def SaveSessionAs(self): |
379 |
dlg = wxFileDialog(self, _("Save Session As"), ".", "", |
dlg = wxFileDialog(self, _("Save Session As"), ".", "", |
380 |
"*.thuban", wxOPEN) |
"*.thuban", wxSAVE|wxOVERWRITE_PROMPT) |
381 |
if dlg.ShowModal() == wxID_OK: |
if dlg.ShowModal() == wxID_OK: |
382 |
self.application.session.SetFilename(dlg.GetPath()) |
self.application.session.SetFilename(dlg.GetPath()) |
383 |
self.application.SaveSession() |
self.application.SaveSession() |
384 |
dlg.Destroy() |
dlg.Destroy() |
385 |
|
|
386 |
def Exit(self): |
def Exit(self): |
387 |
self.Close(false) |
self.Close(False) |
388 |
|
|
389 |
def _OnClose(self, event): |
def _OnClose(self, event): |
390 |
result = self.save_modified_session(can_veto = event.CanVeto()) |
result = self.save_modified_session(can_veto = event.CanVeto()) |
400 |
|
|
401 |
def SetMap(self, map): |
def SetMap(self, map): |
402 |
self.canvas.SetMap(map) |
self.canvas.SetMap(map) |
403 |
#self.legendPanel.SetMap(map) |
self.__SetTitle(map.Title()) |
404 |
|
|
405 |
|
dialog = self.FindRegisteredDock("legend") |
406 |
|
if dialog is not None: |
407 |
|
dialog.GetPanel().SetMap(self.Map()) |
408 |
|
|
409 |
def Map(self): |
def Map(self): |
410 |
"""Return the map displayed by this mainwindow""" |
"""Return the map displayed by this mainwindow""" |
411 |
|
|
|
# sanity check |
|
|
#assert(self.canvas.Map() is self.legendPanel.GetMap()) |
|
|
|
|
412 |
return self.canvas.Map() |
return self.canvas.Map() |
413 |
|
|
414 |
def ShowSessionTree(self): |
def ToggleSessionTree(self): |
415 |
|
"""If the session tree is shown close it otherwise create a new tree""" |
416 |
name = "session_tree" |
name = "session_tree" |
417 |
dialog = self.get_open_dialog(name) |
dialog = self.get_open_dialog(name) |
418 |
if dialog is None: |
if dialog is None: |
420 |
self.add_dialog(name, dialog) |
self.add_dialog(name, dialog) |
421 |
dialog.Show(True) |
dialog.Show(True) |
422 |
else: |
else: |
423 |
# FIXME: bring dialog to front here |
dialog.Close() |
|
pass |
|
424 |
|
|
425 |
|
def SessionTreeShown(self): |
426 |
|
"""Return true iff the session tree is currently shown""" |
427 |
|
return self.get_open_dialog("session_tree") is not None |
428 |
|
|
429 |
def About(self): |
def About(self): |
430 |
self.RunMessageBox(_("About"), |
self.RunMessageBox(_("About"), |
444 |
if dlg.ShowModal() == wxID_OK: |
if dlg.ShowModal() == wxID_OK: |
445 |
filename = dlg.GetPath() |
filename = dlg.GetPath() |
446 |
title = os.path.splitext(os.path.basename(filename))[0] |
title = os.path.splitext(os.path.basename(filename))[0] |
447 |
layer = Layer(title, filename) |
store = self.application.Session().OpenShapefile(filename) |
448 |
|
layer = Layer(title, store) |
449 |
map = self.canvas.Map() |
map = self.canvas.Map() |
450 |
has_layers = map.HasLayers() |
has_layers = map.HasLayers() |
451 |
try: |
try: |
469 |
def CanRemoveLayer(self): |
def CanRemoveLayer(self): |
470 |
"""Return true if the currently selected layer can be deleted. |
"""Return true if the currently selected layer can be deleted. |
471 |
|
|
472 |
If no layer is selected return false. |
If no layer is selected return False. |
473 |
|
|
474 |
The return value of this method determines whether the remove |
The return value of this method determines whether the remove |
475 |
layer command is sensitive in menu. |
layer command is sensitive in menu. |
477 |
layer = self.current_layer() |
layer = self.current_layer() |
478 |
if layer is not None: |
if layer is not None: |
479 |
return self.canvas.Map().CanRemoveLayer(layer) |
return self.canvas.Map().CanRemoveLayer(layer) |
480 |
return 0 |
return False |
481 |
|
|
482 |
def RaiseLayer(self): |
def RaiseLayer(self): |
483 |
layer = self.current_layer() |
layer = self.current_layer() |
516 |
dlg.Destroy() |
dlg.Destroy() |
517 |
return color |
return color |
518 |
|
|
|
def LayerFillColor(self): |
|
|
layer = self.current_layer() |
|
|
if layer is not None: |
|
|
color = self.choose_color() |
|
|
if color is not None: |
|
|
layer.GetClassification().SetDefaultFill(color) |
|
|
|
|
|
def LayerTransparentFill(self): |
|
|
layer = self.current_layer() |
|
|
if layer is not None: |
|
|
layer.GetClassification().SetDefaultFill(Color.None) |
|
|
|
|
|
def LayerOutlineColor(self): |
|
|
layer = self.current_layer() |
|
|
if layer is not None: |
|
|
color = self.choose_color() |
|
|
if color is not None: |
|
|
layer.GetClassification().SetDefaultLineColor(color) |
|
|
|
|
|
def LayerNoOutline(self): |
|
|
layer = self.current_layer() |
|
|
if layer is not None: |
|
|
layer.GetClassification().SetDefaultLineColor(Color.None) |
|
|
|
|
519 |
def HideLayer(self): |
def HideLayer(self): |
520 |
layer = self.current_layer() |
layer = self.current_layer() |
521 |
if layer is not None: |
if layer is not None: |
542 |
# FIXME: bring dialog to front here |
# FIXME: bring dialog to front here |
543 |
pass |
pass |
544 |
|
|
545 |
def Projection(self): |
def MapProjection(self): |
546 |
map = self.canvas.Map() |
|
547 |
proj = map.projection |
name = "map_projection" |
548 |
if proj is None: |
dialog = self.get_open_dialog(name) |
549 |
proj4Dlg = proj4dialog.Proj4Dialog(NULL, None, map.BoundingBox()) |
|
550 |
else: |
if dialog is None: |
551 |
proj4Dlg = proj4dialog.Proj4Dialog(NULL, map.projection.params, |
map = self.canvas.Map() |
552 |
map.BoundingBox()) |
dialog = projdialog.ProjFrame(self, name, |
553 |
if proj4Dlg.ShowModal() == wxID_OK: |
_("Map Projection: %s") % map.Title(), map) |
554 |
params = proj4Dlg.GetParams() |
self.add_dialog(name, dialog) |
555 |
if params is not None: |
dialog.Show() |
556 |
proj = Projection(params) |
dialog.Raise() |
557 |
else: |
|
558 |
proj = None |
def LayerProjection(self): |
559 |
map.SetProjection(proj) |
|
560 |
proj4Dlg.Destroy() |
layer = self.current_layer() |
561 |
|
|
562 |
|
name = "layer_projection" + str(id(layer)) |
563 |
|
dialog = self.get_open_dialog(name) |
564 |
|
|
565 |
|
if dialog is None: |
566 |
|
map = self.canvas.Map() |
567 |
|
dialog = projdialog.ProjFrame(self, name, |
568 |
|
_("Layer Projection: %s") % layer.Title(), layer) |
569 |
|
self.add_dialog(name, dialog) |
570 |
|
dialog.Show() |
571 |
|
dialog.Raise() |
572 |
|
|
573 |
def Classify(self): |
def LayerEditProperties(self): |
574 |
|
|
575 |
# |
# |
576 |
# the menu option for this should only be available if there |
# the menu option for this should only be available if there |
579 |
# |
# |
580 |
|
|
581 |
layer = self.current_layer() |
layer = self.current_layer() |
582 |
self.OpenClassifier(layer) |
self.OpenLayerProperties(layer) |
583 |
|
|
584 |
def OpenClassifier(self, layer, group = None): |
def OpenLayerProperties(self, layer, group = None): |
585 |
name = "classifier" + str(id(layer)) |
name = "layer_properties" + str(id(layer)) |
586 |
dialog = self.get_open_dialog(name) |
dialog = self.get_open_dialog(name) |
587 |
|
|
588 |
if dialog is None: |
if dialog is None: |
589 |
dialog = classifier.Classifier(self, name, layer, group) |
dialog = Classifier(self, name, layer, group) |
590 |
self.add_dialog(name, dialog) |
self.add_dialog(name, dialog) |
591 |
dialog.Show() |
dialog.Show() |
592 |
dialog.Raise() |
dialog.Raise() |
593 |
|
|
594 |
|
|
595 |
def ShowLegend(self, switch = False): |
def ShowLegend(self): |
596 |
|
if not self.LegendShown(): |
597 |
|
self.ToggleLegend() |
598 |
|
|
599 |
|
def ToggleLegend(self): |
600 |
|
"""Show the legend if it's not shown otherwise hide it again""" |
601 |
name = "legend" |
name = "legend" |
602 |
dialog = self.FindRegisteredDock(name) |
dialog = self.FindRegisteredDock(name) |
603 |
|
|
604 |
if dialog is None: |
if dialog is None: |
605 |
title = "Legend: %s" % self.Map().Title() |
dialog = self.CreateDock(name, -1, _("Legend"), wxLAYOUT_LEFT) |
|
dialog = self.CreateDock(name, -1, title, wxLAYOUT_LEFT) |
|
606 |
legend.LegendPanel(dialog, None, self) |
legend.LegendPanel(dialog, None, self) |
607 |
dialog.Dock() |
dialog.Dock() |
608 |
|
dialog.GetPanel().SetMap(self.Map()) |
609 |
|
dialog.Show() |
610 |
|
else: |
611 |
|
dialog.Show(not dialog.IsShown()) |
612 |
|
|
613 |
dialog.GetPanel().SetMap(self.Map()) |
def LegendShown(self): |
614 |
dialog.Show() |
"""Return true iff the legend is currently open""" |
615 |
|
dialog = self.FindRegisteredDock("legend") |
616 |
|
return dialog is not None and dialog.IsShown() |
617 |
|
|
618 |
def ZoomInTool(self): |
def ZoomInTool(self): |
619 |
self.canvas.ZoomInTool() |
self.canvas.ZoomInTool() |
634 |
def FullExtent(self): |
def FullExtent(self): |
635 |
self.canvas.FitMapToWindow() |
self.canvas.FitMapToWindow() |
636 |
|
|
637 |
|
def FullLayerExtent(self): |
638 |
|
self.canvas.FitLayerToWindow(self.current_layer()) |
639 |
|
|
640 |
def PrintMap(self): |
def PrintMap(self): |
641 |
self.canvas.Print() |
self.canvas.Print() |
642 |
|
|
643 |
|
def RenameMap(self): |
644 |
|
dlg = wxTextEntryDialog(self, "Map Title: ", "Rename Map", |
645 |
|
self.Map().Title()) |
646 |
|
if dlg.ShowModal() == wxID_OK: |
647 |
|
title = dlg.GetValue() |
648 |
|
if title != "": |
649 |
|
self.Map().SetTitle(title) |
650 |
|
self.__SetTitle(title) |
651 |
|
|
652 |
|
dlg.Destroy() |
653 |
|
|
654 |
def identify_view_on_demand(self, layer, shapes): |
def identify_view_on_demand(self, layer, shapes): |
655 |
|
"""Subscribed to the canvas' SHAPES_SELECTED message |
656 |
|
|
657 |
|
If the current tool is the identify tool, at least one shape is |
658 |
|
selected and the identify dialog is not shown, show the dialog. |
659 |
|
""" |
660 |
|
# If the selection has become empty we don't need to do |
661 |
|
# anything. Otherwise it could happen that the dialog was popped |
662 |
|
# up when the selection became empty, e.g. when a new selection |
663 |
|
# is opened while the identify tool is active and dialog had |
664 |
|
# been closed |
665 |
|
if not shapes: |
666 |
|
return |
667 |
|
|
668 |
name = "identify_view" |
name = "identify_view" |
669 |
if self.canvas.CurrentTool() == "IdentifyTool": |
if self.canvas.CurrentTool() == "IdentifyTool": |
670 |
if not self.dialog_open(name): |
if not self.dialog_open(name): |
675 |
# FIXME: bring dialog to front? |
# FIXME: bring dialog to front? |
676 |
pass |
pass |
677 |
|
|
678 |
|
def __SetTitle(self, title): |
679 |
|
self.SetTitle("Thuban - " + title) |
680 |
|
|
681 |
# |
# |
682 |
# Define all the commands available in the main window |
# Define all the commands available in the main window |
683 |
# |
# |
689 |
apply(getattr(context.mainwindow, methodname), args) |
apply(getattr(context.mainwindow, methodname), args) |
690 |
|
|
691 |
def _method_command(name, title, method, helptext = "", |
def _method_command(name, title, method, helptext = "", |
692 |
icon = "", sensitive = None): |
icon = "", sensitive = None, checked = None): |
693 |
"""Add a command implemented by a method of the mainwindow object""" |
"""Add a command implemented by a method of the mainwindow object""" |
694 |
registry.Add(Command(name, title, call_method, args=(method,), |
registry.Add(Command(name, title, call_method, args=(method,), |
695 |
helptext = helptext, icon = icon, |
helptext = helptext, icon = icon, |
696 |
sensitive = sensitive)) |
sensitive = sensitive, checked = checked)) |
697 |
|
|
698 |
def make_check_current_tool(toolname): |
def make_check_current_tool(toolname): |
699 |
"""Return a function that tests if the currently active tool is toolname |
"""Return a function that tests if the currently active tool is toolname |
723 |
|
|
724 |
def _has_tree_window_shown(context): |
def _has_tree_window_shown(context): |
725 |
"""Return true if the tree window is shown""" |
"""Return true if the tree window is shown""" |
726 |
return context.mainwindow.get_open_dialog("session_tree") is None |
return context.mainwindow.SessionTreeShown() |
727 |
|
|
728 |
def _has_visible_map(context): |
def _has_visible_map(context): |
729 |
"""Return true iff theres a visible map in the mainwindow. |
"""Return true iff theres a visible map in the mainwindow. |
738 |
|
|
739 |
def _has_legend_shown(context): |
def _has_legend_shown(context): |
740 |
"""Return true if the legend window is shown""" |
"""Return true if the legend window is shown""" |
741 |
return context.mainwindow.FindRegisteredDock("legend") is None |
return context.mainwindow.LegendShown() |
742 |
|
|
743 |
|
|
744 |
# File menu |
# File menu |
745 |
_method_command("new_session", _("&New Session"), "NewSession") |
_method_command("new_session", _("&New Session"), "NewSession") |
746 |
_method_command("open_session", _("&Open Session"), "OpenSession") |
_method_command("open_session", _("&Open Session..."), "OpenSession") |
747 |
_method_command("save_session", _("&Save Session"), "SaveSession") |
_method_command("save_session", _("&Save Session"), "SaveSession") |
748 |
_method_command("save_session_as", _("Save Session &As"), "SaveSessionAs") |
_method_command("save_session_as", _("Save Session &As..."), "SaveSessionAs") |
749 |
_method_command("show_session_tree", _("Show Session &Tree"), "ShowSessionTree", |
_method_command("toggle_session_tree", _("Session &Tree"), "ToggleSessionTree", |
750 |
sensitive = _has_tree_window_shown) |
checked = _has_tree_window_shown) |
751 |
|
_method_command("toggle_legend", _("Legend"), "ToggleLegend", |
752 |
|
checked = _has_legend_shown) |
753 |
_method_command("exit", _("E&xit"), "Exit") |
_method_command("exit", _("E&xit"), "Exit") |
754 |
|
|
755 |
# Help menu |
# Help menu |
756 |
_method_command("help_about", _("&About"), "About") |
_method_command("help_about", _("&About..."), "About") |
757 |
|
|
758 |
|
|
759 |
# Map menu |
# Map menu |
760 |
_method_command("map_projection", _("Pro&jection"), "Projection") |
_method_command("map_projection", _("Pro&jection..."), "MapProjection") |
761 |
|
|
762 |
_tool_command("map_zoom_in_tool", _("&Zoom in"), "ZoomInTool", "ZoomInTool", |
_tool_command("map_zoom_in_tool", _("&Zoom in"), "ZoomInTool", "ZoomInTool", |
763 |
helptext = _("Switch to map-mode 'zoom-in'"), icon = "zoom_in", |
helptext = _("Switch to map-mode 'zoom-in'"), icon = "zoom_in", |
778 |
_method_command("map_full_extent", _("&Full extent"), "FullExtent", |
_method_command("map_full_extent", _("&Full extent"), "FullExtent", |
779 |
helptext = _("Full Extent"), icon = "fullextent", |
helptext = _("Full Extent"), icon = "fullextent", |
780 |
sensitive = _has_visible_map) |
sensitive = _has_visible_map) |
781 |
|
_method_command("layer_full_extent", _("&Full layer extent"), "FullLayerExtent", |
782 |
|
helptext = _("Full Layer Extent"), icon = "fullextent", |
783 |
|
sensitive = _has_selected_layer) |
784 |
_method_command("map_print", _("Prin&t"), "PrintMap", |
_method_command("map_print", _("Prin&t"), "PrintMap", |
785 |
helptext = _("Print the map")) |
helptext = _("Print the map")) |
786 |
|
_method_command("map_rename", _("&Rename..."), "RenameMap", |
787 |
# Layer menu |
helptext = _("Rename the map")) |
788 |
_method_command("layer_add", _("&Add Layer"), "AddLayer", |
_method_command("layer_add", _("&Add Layer..."), "AddLayer", |
789 |
helptext = _("Add a new layer to active map")) |
helptext = _("Add a new layer to active map")) |
790 |
_method_command("layer_remove", _("&Remove Layer"), "RemoveLayer", |
_method_command("layer_remove", _("&Remove Layer"), "RemoveLayer", |
791 |
helptext = _("Remove selected layer(s)"), |
helptext = _("Remove selected layer(s)"), |
792 |
sensitive = _can_remove_layer) |
sensitive = _can_remove_layer) |
793 |
_method_command("layer_fill_color", _("&Fill Color"), "LayerFillColor", |
|
794 |
helptext = _("Set the fill color of selected layer(s)"), |
# Layer menu |
795 |
sensitive = _has_selected_layer) |
_method_command("layer_projection", _("Pro&jection..."), "LayerProjection", |
|
_method_command("layer_transparent_fill", _("&Transparent Fill"), |
|
|
"LayerTransparentFill", |
|
|
helptext = _("Do not fill the selected layer(s)"), |
|
|
sensitive = _has_selected_layer) |
|
|
_method_command("layer_outline_color", _("&Outline Color"), "LayerOutlineColor", |
|
|
helptext = _("Set the outline color of selected layer(s)"), |
|
|
sensitive = _has_selected_layer) |
|
|
_method_command("layer_no_outline", _("&No Outline"), "LayerNoOutline", |
|
|
helptext= _("Do not draw the outline of the selected layer(s)"), |
|
796 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_layer) |
797 |
_method_command("layer_raise", _("&Raise"), "RaiseLayer", |
_method_command("layer_raise", _("&Raise"), "RaiseLayer", |
798 |
helptext = _("Raise selected layer(s)"), |
helptext = _("Raise selected layer(s)"), |
809 |
_method_command("layer_show_table", _("Show Ta&ble"), "LayerShowTable", |
_method_command("layer_show_table", _("Show Ta&ble"), "LayerShowTable", |
810 |
helptext = _("Show the selected layer's table"), |
helptext = _("Show the selected layer's table"), |
811 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_layer) |
812 |
_method_command("layer_classifier", _("Classify"), "Classify", |
_method_command("layer_properties", _("&Properties..."), "LayerEditProperties", |
813 |
sensitive = _has_selected_layer) |
sensitive = _has_selected_layer) |
|
_method_command("show_legend", _("Show Legend"), "ShowLegend", |
|
|
sensitive = _has_legend_shown) |
|
814 |
|
|
815 |
# the menu structure |
# the menu structure |
816 |
main_menu = Menu("<main>", "<main>", |
main_menu = Menu("<main>", "<main>", |
817 |
[Menu("file", _("&File"), |
[Menu("file", _("&File"), |
818 |
["new_session", "open_session", None, |
["new_session", "open_session", None, |
819 |
"save_session", "save_session_as", None, |
"save_session", "save_session_as", None, |
820 |
"show_session_tree", None, |
"toggle_session_tree", None, |
|
"show_legend", None, |
|
821 |
"exit"]), |
"exit"]), |
822 |
Menu("map", _("&Map"), |
Menu("map", _("&Map"), |
823 |
["layer_add", "layer_remove", |
["layer_add", "layer_remove", "map_rename", |
824 |
None, |
None, |
825 |
"map_projection", |
"map_projection", |
826 |
None, |
None, |
827 |
"map_zoom_in_tool", "map_zoom_out_tool", |
"map_zoom_in_tool", "map_zoom_out_tool", |
828 |
"map_pan_tool", "map_identify_tool", "map_label_tool", |
"map_pan_tool", "map_full_extent", "layer_full_extent", |
829 |
|
None, |
830 |
|
"map_identify_tool", "map_label_tool", |
831 |
None, |
None, |
832 |
"map_full_extent", |
"toggle_legend", |
833 |
None, |
None, |
834 |
"map_print"]), |
"map_print"]), |
835 |
Menu("layer", _("&Layer"), |
Menu("layer", _("&Layer"), |
836 |
["layer_fill_color", "layer_transparent_fill", |
["layer_raise", "layer_lower", |
|
"layer_outline_color", "layer_no_outline", |
|
|
None, |
|
|
"layer_raise", "layer_lower", |
|
837 |
None, |
None, |
838 |
"layer_show", "layer_hide", |
"layer_show", "layer_hide", |
839 |
None, |
None, |
840 |
"layer_show_table", |
"layer_show_table", |
841 |
None, |
None, |
842 |
"layer_classifier"]), |
"layer_projection", |
843 |
|
"layer_properties"]), |
844 |
Menu("help", _("&Help"), |
Menu("help", _("&Help"), |
845 |
["help_about"])]) |
["help_about"])]) |
846 |
|
|