| 1 |
#! /usr/bin/python |
| 2 |
# Copyright (c) 2001 by Intevation GmbH |
| 3 |
# Authors: |
| 4 |
# Jan-Oliver Wagner <[email protected]> |
| 5 |
# Bernhard Herzog <[email protected]> |
| 6 |
# |
| 7 |
# This program is free software under the GPL (>=v2) |
| 8 |
# Read the file COPYING coming with Thuban for details. |
| 9 |
|
| 10 |
__version__ = "$Revision$" |
| 11 |
|
| 12 |
from wxPython.wx import * |
| 13 |
|
| 14 |
from Thuban.Model.messages import MAPS_CHANGED, MAP_PROJECTION_CHANGED, \ |
| 15 |
LAYERS_CHANGED, LAYER_LEGEND_CHANGED, LAYER_VISIBILITY_CHANGED |
| 16 |
from Thuban.Model.layer import Layer, shapetype_names |
| 17 |
from Thuban.Model.map import Map |
| 18 |
|
| 19 |
from dialogs import NonModalDialog |
| 20 |
from messages import SESSION_CHANGED, SELECTED_LAYER |
| 21 |
|
| 22 |
def color_string(color): |
| 23 |
if color is None: |
| 24 |
return "None" |
| 25 |
return "(%.3f, %.3f, %.3f)" % (color.red, color.green, color.blue) |
| 26 |
|
| 27 |
|
| 28 |
class SessionTreeCtrl(wxTreeCtrl): |
| 29 |
|
| 30 |
def __init__(self, parent, ID, app): |
| 31 |
# Use the WANTS_CHARS style so the panel doesn't eat the Return key. |
| 32 |
wxTreeCtrl.__init__(self, parent, ID) |
| 33 |
|
| 34 |
self.app = app |
| 35 |
# boolean to indicate that we manipulate the selection ourselves |
| 36 |
# so that we can ignore the selection events generated |
| 37 |
self.changing_selection = 0 |
| 38 |
|
| 39 |
# Dictionary mapping layer id's to tree items |
| 40 |
self.layer_to_item = {} |
| 41 |
|
| 42 |
self.app.Subscribe(SESSION_CHANGED, self.session_changed) |
| 43 |
self.app.interactor.Subscribe(SELECTED_LAYER, self.layer_selected) |
| 44 |
# pretend the session has changed to build the initial tree |
| 45 |
self.session_changed() |
| 46 |
|
| 47 |
EVT_TREE_SEL_CHANGED(self, self.GetId(), self.OnSelChanged) |
| 48 |
|
| 49 |
def update_tree(self, *args): |
| 50 |
"""Clear and rebuild the tree""" |
| 51 |
self.DeleteAllItems() |
| 52 |
session = self.app.session |
| 53 |
root = self.AddRoot("Session: %s" % session.title) |
| 54 |
self.layer_to_item.clear() |
| 55 |
for map in session.Maps(): |
| 56 |
mapitem = self.AppendItem(root, "Map: %s" % map.title) |
| 57 |
self.SetPyData(mapitem, map) |
| 58 |
if map.projection and len(map.projection.params) > 0: |
| 59 |
projectionitem = self.AppendItem(mapitem, "Projection") |
| 60 |
for param in map.projection.params: |
| 61 |
parameteritem = self.AppendItem(projectionitem, str(param)) |
| 62 |
self.Expand(projectionitem) |
| 63 |
|
| 64 |
layers = map.Layers() |
| 65 |
for layer_index in range(len(layers) - 1, -1, -1): |
| 66 |
layer = layers[layer_index] |
| 67 |
idata = wxTreeItemData() |
| 68 |
idata.SetData(layer) |
| 69 |
layeritem = self.AppendItem(mapitem, |
| 70 |
"Layer '%s'" % layer.Title(), |
| 71 |
data = idata) |
| 72 |
self.layer_to_item[id(layer)] = layeritem |
| 73 |
if layer is self.app.interactor.selected_layer: |
| 74 |
self.SelectItem(layeritem) |
| 75 |
if isinstance(layer, Layer): |
| 76 |
if layer.Visible(): |
| 77 |
text = "Shown" |
| 78 |
else: |
| 79 |
text = "Hidden" |
| 80 |
self.AppendItem(layeritem, text) |
| 81 |
self.AppendItem(layeritem, "Shapes: %d" %layer.NumShapes()) |
| 82 |
self.AppendItem(layeritem, ("Extents: (%g, %g, %g, %g)" |
| 83 |
% layer.LatLongBoundingBox())) |
| 84 |
self.AppendItem(layeritem, |
| 85 |
"Shapetype: %s" |
| 86 |
% shapetype_names[layer.ShapeType()]) |
| 87 |
self.AppendItem(layeritem, |
| 88 |
"Fill: " + color_string(layer.fill)) |
| 89 |
self.AppendItem(layeritem, |
| 90 |
"Outline: " + color_string(layer.stroke)) |
| 91 |
self.Expand(layeritem) |
| 92 |
self.Expand(mapitem) |
| 93 |
self.Expand(root) |
| 94 |
|
| 95 |
def session_changed(self, *args): |
| 96 |
for channel in (MAPS_CHANGED, |
| 97 |
MAP_PROJECTION_CHANGED, |
| 98 |
LAYERS_CHANGED, |
| 99 |
LAYER_LEGEND_CHANGED, |
| 100 |
LAYER_VISIBILITY_CHANGED): |
| 101 |
self.app.session.Subscribe(channel, self.update_tree) |
| 102 |
self.update_tree() |
| 103 |
|
| 104 |
def normalize_selection(self): |
| 105 |
"""Select the layer or map containing currently selected item""" |
| 106 |
item = self.GetSelection() |
| 107 |
while item.IsOk(): |
| 108 |
object = self.GetPyData(item) |
| 109 |
if isinstance(object, Layer) or isinstance(object, Map): |
| 110 |
break |
| 111 |
item = self.GetItemParent(item) |
| 112 |
|
| 113 |
self.changing_selection = 1 |
| 114 |
try: |
| 115 |
self.SelectItem(item) |
| 116 |
finally: |
| 117 |
self.changing_selection = 0 |
| 118 |
|
| 119 |
def SelectedLayer(self): |
| 120 |
"""Return the layer object currently selected in the tree. |
| 121 |
Return None if no layer is selected""" |
| 122 |
layer = self.GetPyData(self.GetSelection()) |
| 123 |
if isinstance(layer, Layer): |
| 124 |
return layer |
| 125 |
return None |
| 126 |
|
| 127 |
def OnSelChanged(self, event): |
| 128 |
if self.changing_selection: |
| 129 |
# we're changing the selection ourselves (probably through |
| 130 |
# self.normalize_selection(). ignore the event. |
| 131 |
return |
| 132 |
self.normalize_selection() |
| 133 |
# SelectedLayer returns None if no layer is selected. Since |
| 134 |
# passing None to interactor.SelectLayer deselects the layer we |
| 135 |
# can simply pass the result of SelectedLayer on in all cases |
| 136 |
self.app.interactor.SelectLayer(self.SelectedLayer()) |
| 137 |
|
| 138 |
def layer_selected(self, layer): |
| 139 |
item = self.layer_to_item.get(id(layer)) |
| 140 |
if item is not None and item != self.GetSelection(): |
| 141 |
self.SelectItem(item) |
| 142 |
|
| 143 |
|
| 144 |
class SessionTreeView(NonModalDialog): |
| 145 |
|
| 146 |
"""Non modal dialog showing the session as a tree""" |
| 147 |
|
| 148 |
def __init__(self, parent, app, name): |
| 149 |
NonModalDialog.__init__(self, parent, app.interactor, name, "Session") |
| 150 |
self.tree = SessionTreeCtrl(self, -1, app) |
| 151 |
|
| 152 |
def OnClose(self, event): |
| 153 |
#self.interactor.Unsubscribe(SELECTED_SHAPE, self.select_shape) |
| 154 |
NonModalDialog.OnClose(self, event) |
| 155 |
|
Parent Directory
| 
Revision Log