20 |
from Thuban import _ |
from Thuban import _ |
21 |
from Thuban.UI.common import Color2wxColour, wxColour2Color |
from Thuban.UI.common import Color2wxColour, wxColour2Color |
22 |
|
|
23 |
|
from Thuban.Model.messages import MAP_LAYERS_REMOVED, LAYER_SHAPESTORE_REPLACED |
24 |
from Thuban.Model.range import Range |
from Thuban.Model.range import Range |
25 |
from Thuban.Model.classification import \ |
from Thuban.Model.classification import \ |
26 |
Classification, ClassGroupDefault, \ |
Classification, ClassGroupDefault, \ |
27 |
ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \ |
ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \ |
28 |
ClassGroupProperties |
ClassGroupProperties |
29 |
|
|
30 |
from Thuban.Model.color import Color |
from Thuban.Model.color import Transparent |
31 |
|
|
32 |
from Thuban.Model.layer import Layer, RasterLayer, \ |
from Thuban.Model.layer import Layer, RasterLayer, \ |
33 |
SHAPETYPE_ARC, SHAPETYPE_POLYGON, SHAPETYPE_POINT |
SHAPETYPE_ARC, SHAPETYPE_POLYGON, SHAPETYPE_POINT |
34 |
|
|
35 |
from Thuban.UI.classgen import ClassGenDialog, ClassGenerator |
from Thuban.UI.classgen import ClassGenDialog |
36 |
|
|
37 |
from dialogs import NonModalDialog |
from dialogs import NonModalNonParentDialog |
38 |
|
|
39 |
ID_CLASS_TABLE = 40011 |
ID_CLASS_TABLE = 40011 |
40 |
|
|
671 |
EB_SELECT_FIELD = 1 |
EB_SELECT_FIELD = 1 |
672 |
EB_GEN_CLASS = 2 |
EB_GEN_CLASS = 2 |
673 |
|
|
674 |
class Classifier(NonModalDialog): |
class Classifier(NonModalNonParentDialog): |
675 |
|
|
676 |
type2string = {None: _("None"), |
type2string = {None: _("None"), |
677 |
FIELDTYPE_STRING: _("Text"), |
FIELDTYPE_STRING: _("Text"), |
678 |
FIELDTYPE_INT: _("Integer"), |
FIELDTYPE_INT: _("Integer"), |
679 |
FIELDTYPE_DOUBLE: _("Decimal")} |
FIELDTYPE_DOUBLE: _("Decimal")} |
680 |
|
|
681 |
def __init__(self, parent, name, layer, group = None): |
def __init__(self, parent, name, map, layer, group = None): |
682 |
NonModalDialog.__init__(self, parent, name, "") |
NonModalNonParentDialog.__init__(self, parent, name, "") |
683 |
|
|
684 |
self.__SetTitle(layer.Title()) |
self.__SetTitle(layer.Title()) |
685 |
|
|
686 |
self.layer = layer |
self.layer = layer |
687 |
|
self.map = map |
688 |
|
|
689 |
|
self.map.Subscribe(MAP_LAYERS_REMOVED, self.map_layers_removed) |
690 |
|
self.layer.Subscribe(LAYER_SHAPESTORE_REPLACED, |
691 |
|
self.layer_shapestore_replaced) |
692 |
|
|
693 |
self.genDlg = None |
self.genDlg = None |
694 |
|
|
706 |
field = self.originalClass.GetField() |
field = self.originalClass.GetField() |
707 |
fieldType = self.originalClass.GetFieldType() |
fieldType = self.originalClass.GetFieldType() |
708 |
|
|
709 |
|
table = layer.ShapeStore().Table() |
710 |
# |
# |
711 |
# make field choice box |
# make field choice box |
712 |
# |
# |
713 |
self.fields = wxChoice(panel, ID_PROPERTY_SELECT,) |
self.fields = wxChoice(panel, ID_PROPERTY_SELECT,) |
714 |
|
|
715 |
self.num_cols = layer.table.NumColumns() |
self.num_cols = table.NumColumns() |
716 |
# just assume the first field in case one hasn't been |
# just assume the first field in case one hasn't been |
717 |
# specified in the file. |
# specified in the file. |
718 |
self.__cur_field = 0 |
self.__cur_field = 0 |
725 |
self.fields.SetClientData(0, None) |
self.fields.SetClientData(0, None) |
726 |
|
|
727 |
for i in range(self.num_cols): |
for i in range(self.num_cols): |
728 |
name = layer.table.Column(i).name |
name = table.Column(i).name |
729 |
self.fields.Append(name) |
self.fields.Append(name) |
730 |
|
|
731 |
if name == field: |
if name == field: |
755 |
self.fields.SetSelection(self.__cur_field) |
self.fields.SetSelection(self.__cur_field) |
756 |
self.__SelectField(self.__cur_field, group = group) |
self.__SelectField(self.__cur_field, group = group) |
757 |
|
|
758 |
button_try = wxButton(panel, ID_PROPERTY_TRY, _("Try")) |
button_try = wxButton(self, ID_PROPERTY_TRY, _("Try")) |
759 |
button_revert = wxButton(panel, ID_PROPERTY_REVERT, _("Revert")) |
button_revert = wxButton(self, ID_PROPERTY_REVERT, _("Revert")) |
760 |
button_ok = wxButton(panel, wxID_OK, _("OK")) |
button_ok = wxButton(self, wxID_OK, _("OK")) |
761 |
|
button_close = wxButton(self, wxID_CANCEL, _("Close")) |
762 |
button_ok.SetDefault() |
button_ok.SetDefault() |
|
button_close = wxButton(panel, wxID_CANCEL, _("Close")) |
|
763 |
|
|
764 |
############################ |
############################ |
765 |
# Layout the controls |
# Layout the controls |
819 |
|
|
820 |
|
|
821 |
buttonBox = wxBoxSizer(wxHORIZONTAL) |
buttonBox = wxBoxSizer(wxHORIZONTAL) |
822 |
buttonBox.Add(button_try, 0, wxALL, 4) |
buttonBox.Add(button_try, 0, wxRIGHT|wxEXPAND, 10) |
823 |
buttonBox.Add(60, 20, 0, wxALL, 4) |
buttonBox.Add(button_revert, 0, wxRIGHT|wxEXPAND, 10) |
824 |
buttonBox.Add(button_revert, 0, wxALL, 4) |
buttonBox.Add(button_ok, 0, wxRIGHT|wxEXPAND, 10) |
825 |
buttonBox.Add(60, 20, 0, wxALL, 4) |
buttonBox.Add(button_close, 0, wxRIGHT|wxEXPAND, 10) |
|
buttonBox.Add(button_ok, 0, wxALL, 4) |
|
|
buttonBox.Add(60, 20, 0, wxALL, 4) |
|
|
buttonBox.Add(button_close, 0, wxALL, 4) |
|
|
panelBox.Add(buttonBox, 0, |
|
|
wxALL|wxALIGN_CENTER_HORIZONTAL|wxALIGN_BOTTOM, 0) |
|
826 |
|
|
827 |
panel.SetAutoLayout(True) |
panel.SetAutoLayout(True) |
828 |
panel.SetSizer(panelBox) |
panel.SetSizer(panelBox) |
830 |
panelBox.SetSizeHints(panel) |
panelBox.SetSizeHints(panel) |
831 |
|
|
832 |
topBox.Add(panel, 1, wxGROW | wxALL, 4) |
topBox.Add(panel, 1, wxGROW | wxALL, 4) |
833 |
|
topBox.Add(buttonBox, 0, wxALIGN_RIGHT|wxBOTTOM|wxTOP, 10) |
834 |
|
|
835 |
self.SetAutoLayout(True) |
self.SetAutoLayout(True) |
836 |
self.SetSizer(topBox) |
self.SetSizer(topBox) |
859 |
text_title.SetFocus() |
text_title.SetFocus() |
860 |
self.haveApplied = False |
self.haveApplied = False |
861 |
|
|
862 |
|
def unsubscribe_messages(self): |
863 |
|
self.map.Unsubscribe(MAP_LAYERS_REMOVED, self.map_layers_removed) |
864 |
|
self.layer.Unsubscribe(LAYER_SHAPESTORE_REPLACED, |
865 |
|
self.layer_shapestore_replaced) |
866 |
|
|
867 |
|
def map_layers_removed(self, map): |
868 |
|
if self.layer not in self.map.Layers(): |
869 |
|
self.Close() |
870 |
|
|
871 |
|
def layer_shapestore_replaced(self, *args): |
872 |
|
self.Close() |
873 |
|
|
874 |
def EditSymbol(self, row): |
def EditSymbol(self, row): |
875 |
table = self.classGrid.GetTable() |
table = self.classGrid.GetTable() |
876 |
prop = table.GetValueAsCustom(row, COL_SYMBOL, None) |
prop = table.GetValueAsCustom(row, COL_SYMBOL, None) |
877 |
|
|
878 |
# get a new ClassGroupProperties object and copy the |
# get a new ClassGroupProperties object and copy the |
879 |
# values over to our current object |
# values over to our current object |
880 |
propDlg = SelectPropertiesDialog(NULL, prop, self.layer.ShapeType()) |
propDlg = SelectPropertiesDialog(self, prop, self.layer.ShapeType()) |
881 |
|
|
882 |
self.Enable(False) |
self.Enable(False) |
883 |
if propDlg.ShowModal() == wxID_OK: |
if propDlg.ShowModal() == wxID_OK: |
909 |
if copyClass: |
if copyClass: |
910 |
clazz = copy.deepcopy(clazz) |
clazz = copy.deepcopy(clazz) |
911 |
|
|
912 |
clazz.SetField(fieldName) |
clazz.SetFieldInfo(fieldName, fieldType) |
|
clazz.SetFieldType(fieldType) |
|
913 |
|
|
914 |
|
|
915 |
# table = self.classGrid.GetTable() |
# table = self.classGrid.GetTable() |
933 |
|
|
934 |
fieldName = self.fields.GetString(fieldIndex) |
fieldName = self.fields.GetString(fieldIndex) |
935 |
fieldType = self.layer.GetFieldType(fieldName) |
fieldType = self.layer.GetFieldType(fieldName) |
936 |
clazz.SetFieldType(fieldType) |
clazz.SetFieldInfo(fieldName, fieldType) |
937 |
|
|
938 |
self.classGrid.CreateTable(clazz, self.layer.ShapeType(), group) |
self.classGrid.CreateTable(clazz, self.layer.ShapeType(), group) |
939 |
|
|
960 |
|
|
961 |
self.__SetGridTable(newIndex, group) |
self.__SetGridTable(newIndex, group) |
962 |
|
|
963 |
self.__EnableButtons(EB_SELECT_FIELD, newIndex != 0) |
self.__EnableButtons(EB_SELECT_FIELD) |
964 |
|
|
965 |
self.__SetFieldTypeText(newIndex) |
self.__SetFieldTypeText(newIndex) |
966 |
|
|
1006 |
self.Close() |
self.Close() |
1007 |
|
|
1008 |
def OnClose(self, event): |
def OnClose(self, event): |
1009 |
NonModalDialog.OnClose(self, event) |
self.unsubscribe_messages() |
1010 |
|
NonModalNonParentDialog.OnClose(self, event) |
1011 |
|
|
1012 |
def _OnCloseBtn(self, event): |
def _OnCloseBtn(self, event): |
1013 |
"""Close is similar to Cancel except that any changes that were |
"""Close is similar to Cancel except that any changes that were |
1037 |
|
|
1038 |
EVT_CLOSE(self.genDlg, self._OnGenDialogClose) |
EVT_CLOSE(self.genDlg, self._OnGenDialogClose) |
1039 |
|
|
1040 |
self.__EnableButtons(EB_GEN_CLASS, False) |
self.__EnableButtons(EB_GEN_CLASS) |
1041 |
|
|
1042 |
self.genDlg.Show() |
self.genDlg.Show() |
1043 |
|
|
1044 |
def _OnGenDialogClose(self, event): |
def _OnGenDialogClose(self, event): |
1045 |
self.genDlg.Destroy() |
self.genDlg.Destroy() |
1046 |
self.__EnableButtons(EB_GEN_CLASS, True) |
self.genDlg = None |
1047 |
|
self.__EnableButtons(EB_GEN_CLASS) |
1048 |
|
|
1049 |
def _OnMoveUp(self, event): |
def _OnMoveUp(self, event): |
1050 |
sel = self.classGrid.GetCurrentSelection() |
sel = self.classGrid.GetCurrentSelection() |
1082 |
self.layer.SetTitle(obj.GetValue()) |
self.layer.SetTitle(obj.GetValue()) |
1083 |
self.__SetTitle(self.layer.Title()) |
self.__SetTitle(self.layer.Title()) |
1084 |
|
|
1085 |
self.__EnableButtons(EB_LAYER_TITLE, self.layer.Title() != "") |
self.__EnableButtons(EB_LAYER_TITLE) |
1086 |
|
|
1087 |
|
def __EnableButtons(self, case): |
1088 |
|
|
1089 |
def __EnableButtons(self, case, enable): |
list = {wxID_OK : True, |
1090 |
|
wxID_CANCEL : True, |
1091 |
|
ID_PROPERTY_ADD : True, |
1092 |
|
ID_PROPERTY_MOVEUP : True, |
1093 |
|
ID_PROPERTY_MOVEDOWN : True, |
1094 |
|
ID_PROPERTY_REMOVE : True, |
1095 |
|
ID_PROPERTY_SELECT : True, |
1096 |
|
ID_PROPERTY_FIELDTEXT : True, |
1097 |
|
ID_PROPERTY_GENCLASS : True, |
1098 |
|
ID_PROPERTY_EDITSYM : True} |
1099 |
|
|
1100 |
if case == EB_LAYER_TITLE: |
if case == EB_LAYER_TITLE: |
1101 |
list = (wxID_OK, |
if self.layer.Title() == "": |
1102 |
wxID_CANCEL) |
list[wxID_OK] = False |
1103 |
|
list[wxID_CANCEL] = False |
1104 |
|
|
1105 |
elif case == EB_SELECT_FIELD: |
elif case == EB_SELECT_FIELD: |
1106 |
list = (ID_PROPERTY_GENCLASS, |
if self.fields.GetSelection() == 0: |
1107 |
ID_PROPERTY_ADD, |
list[ID_PROPERTY_GENCLASS] = False |
1108 |
ID_PROPERTY_MOVEUP, |
list[ID_PROPERTY_ADD] = False |
1109 |
ID_PROPERTY_MOVEDOWN, |
list[ID_PROPERTY_MOVEUP] = False |
1110 |
ID_PROPERTY_EDITSYM, |
list[ID_PROPERTY_MOVEDOWN] = False |
1111 |
ID_PROPERTY_REMOVE) |
list[ID_PROPERTY_REMOVE] = False |
1112 |
|
|
1113 |
elif case == EB_GEN_CLASS: |
elif case == EB_GEN_CLASS: |
1114 |
list = (ID_PROPERTY_SELECT, |
if self.genDlg is not None: |
1115 |
ID_PROPERTY_FIELDTEXT, |
list[ID_PROPERTY_SELECT] = False |
1116 |
ID_PROPERTY_GENCLASS, |
list[ID_PROPERTY_FIELDTEXT] = False |
1117 |
ID_PROPERTY_EDITSYM) |
list[ID_PROPERTY_GENCLASS] = False |
1118 |
|
|
1119 |
for id in list: |
for id, enable in list.items(): |
1120 |
self.FindWindowById(id).Enable(enable) |
win = self.FindWindowById(id) |
1121 |
|
if win: |
1122 |
|
win.Enable(enable) |
1123 |
|
|
1124 |
ID_SELPROP_SPINCTRL = 4002 |
ID_SELPROP_SPINCTRL = 4002 |
1125 |
ID_SELPROP_PREVIEW = 4003 |
ID_SELPROP_PREVIEW = 4003 |
1208 |
# |
# |
1209 |
buttonBox = wxBoxSizer(wxHORIZONTAL) |
buttonBox = wxBoxSizer(wxHORIZONTAL) |
1210 |
button_ok = wxButton(self, wxID_OK, _("OK")) |
button_ok = wxButton(self, wxID_OK, _("OK")) |
1211 |
button_ok.SetDefault() |
buttonBox.Add(button_ok, 0, wxRIGHT|wxEXPAND, 10) |
|
buttonBox.Add(button_ok, 0, wxALL, 4) |
|
1212 |
buttonBox.Add(wxButton(self, wxID_CANCEL, _("Cancel")), |
buttonBox.Add(wxButton(self, wxID_CANCEL, _("Cancel")), |
1213 |
0, wxALL, 4) |
0, wxRIGHT|wxEXPAND, 10) |
1214 |
topBox.Add(buttonBox, 0, wxALIGN_CENTER_HORIZONTAL|wxALIGN_BOTTOM, 10) |
topBox.Add(buttonBox, 0, wxALIGN_RIGHT|wxBOTTOM|wxTOP, 10) |
1215 |
|
|
1216 |
|
button_ok.SetDefault() |
1217 |
|
|
1218 |
#EVT_BUTTON(self, wxID_OK, self._OnOK) |
#EVT_BUTTON(self, wxID_OK, self._OnOK) |
1219 |
#EVT_BUTTON(self, ID_SELPROP_CANCEL, self._OnCancel) |
#EVT_BUTTON(self, ID_SELPROP_CANCEL, self._OnCancel) |
1235 |
|
|
1236 |
def __GetColor(self, cur): |
def __GetColor(self, cur): |
1237 |
dialog = wxColourDialog(self) |
dialog = wxColourDialog(self) |
1238 |
if cur is not Color.Transparent: |
if cur is not Transparent: |
1239 |
dialog.GetColourData().SetColour(Color2wxColour(cur)) |
dialog.GetColourData().SetColour(Color2wxColour(cur)) |
1240 |
|
|
1241 |
ret = None |
ret = None |
1253 |
self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer |
self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer |
1254 |
|
|
1255 |
def _OnChangeLineColorTrans(self, event): |
def _OnChangeLineColorTrans(self, event): |
1256 |
self.prop.SetLineColor(Color.Transparent) |
self.prop.SetLineColor(Transparent) |
1257 |
self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer |
self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer |
1258 |
|
|
1259 |
def _OnChangeFillColor(self, event): |
def _OnChangeFillColor(self, event): |
1263 |
self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer |
self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer |
1264 |
|
|
1265 |
def _OnChangeFillColorTrans(self, event): |
def _OnChangeFillColorTrans(self, event): |
1266 |
self.prop.SetFill(Color.Transparent) |
self.prop.SetFill(Transparent) |
1267 |
self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer |
self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer |
1268 |
|
|
1269 |
def GetClassGroupProperties(self): |
def GetClassGroupProperties(self): |
1319 |
h = rect.GetHeight() |
h = rect.GetHeight() |
1320 |
|
|
1321 |
stroke = prop.GetLineColor() |
stroke = prop.GetLineColor() |
1322 |
if stroke is Color.Transparent: |
if stroke is Transparent: |
1323 |
pen = wxTRANSPARENT_PEN |
pen = wxTRANSPARENT_PEN |
1324 |
else: |
else: |
1325 |
pen = wxPen(Color2wxColour(stroke), |
pen = wxPen(Color2wxColour(stroke), |
1327 |
wxSOLID) |
wxSOLID) |
1328 |
|
|
1329 |
stroke = prop.GetFill() |
stroke = prop.GetFill() |
1330 |
if stroke is Color.Transparent: |
if stroke is Transparent: |
1331 |
brush = wxTRANSPARENT_BRUSH |
brush = wxTRANSPARENT_BRUSH |
1332 |
else: |
else: |
1333 |
brush = wxBrush(Color2wxColour(stroke), wxSOLID) |
brush = wxBrush(Color2wxColour(stroke), wxSOLID) |
1386 |
|
|
1387 |
wxWindow.__init__(self, parent, id, size = size, style = style) |
wxWindow.__init__(self, parent, id, size = size, style = style) |
1388 |
|
|
1389 |
|
self.parent = parent |
1390 |
|
|
1391 |
self.SetProperties(props) |
self.SetProperties(props) |
1392 |
self.SetShapeType(shapeType) |
self.SetShapeType(shapeType) |
1393 |
self.AllowEdit(True) |
self.AllowEdit(True) |
1431 |
def DoEdit(self): |
def DoEdit(self): |
1432 |
if not self.allowEdit: return |
if not self.allowEdit: return |
1433 |
|
|
1434 |
propDlg = SelectPropertiesDialog(NULL, |
propDlg = SelectPropertiesDialog(self.parent, |
1435 |
self.GetProperties(), |
self.GetProperties(), |
1436 |
self.GetShapeType()) |
self.GetShapeType()) |
1437 |
|
|