16 |
from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \ |
from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \ |
17 |
FIELDTYPE_STRING |
FIELDTYPE_STRING |
18 |
|
|
19 |
|
from Thuban.Model.layer import SHAPETYPE_ARC |
20 |
from Thuban.Model.range import Range |
from Thuban.Model.range import Range |
21 |
|
from Thuban.UI.common import ThubanBeginBusyCursor, ThubanEndBusyCursor |
22 |
|
|
23 |
import classifier, resource |
import classifier, resource |
24 |
|
|
27 |
CustomRamp, GreyRamp, RedRamp, GreenRamp, BlueRamp, GreenToRedRamp, \ |
CustomRamp, GreyRamp, RedRamp, GreenRamp, BlueRamp, GreenToRedRamp, \ |
28 |
HotToColdRamp |
HotToColdRamp |
29 |
|
|
30 |
|
|
31 |
USEALL_BMP = "group_use_all" |
USEALL_BMP = "group_use_all" |
32 |
USE_BMP = "group_use" |
USE_BMP = "group_use" |
33 |
USENOT_BMP = "group_use_not" |
USENOT_BMP = "group_use_not" |
48 |
ID_CLASSGEN_GENCOMBO = 4007 |
ID_CLASSGEN_GENCOMBO = 4007 |
49 |
ID_CLASSGEN_PROPCOMBO = 4008 |
ID_CLASSGEN_PROPCOMBO = 4008 |
50 |
|
|
51 |
|
ID_BORDER_COLOR = 4009 |
52 |
|
ID_BORDER_COLOR_CHANGE = 4010 |
53 |
|
|
54 |
class ClassGenDialog(wxDialog): |
class ClassGenDialog(wxDialog): |
55 |
|
|
56 |
def __init__(self, parent, layer, fieldName): |
def __init__(self, parent, layer, fieldName): |
81 |
# panels they will call AllowGenerate() which uses genButton. |
# panels they will call AllowGenerate() which uses genButton. |
82 |
# |
# |
83 |
self.genButton = wxButton(self, wxID_OK, _("Generate")) |
self.genButton = wxButton(self, wxID_OK, _("Generate")) |
|
self.genButton.SetDefault() |
|
84 |
self.cancelButton = wxButton(self, wxID_CANCEL, _("Close")) |
self.cancelButton = wxButton(self, wxID_CANCEL, _("Close")) |
85 |
|
self.genButton.SetDefault() |
86 |
|
|
87 |
self.genChoice = wxChoice(self, ID_CLASSGEN_GENCOMBO) |
self.genChoice = wxChoice(self, ID_CLASSGEN_GENCOMBO) |
88 |
|
|
136 |
self.propPanel = None |
self.propPanel = None |
137 |
custom_ramp_panel = CustomRampPanel(self, layer.ShapeType()) |
custom_ramp_panel = CustomRampPanel(self, layer.ShapeType()) |
138 |
|
|
139 |
self.propCombo.Append(PROPCOMBOSTR_GREY, GreyRamp()) |
self.propCombo.Append(PROPCOMBOSTR_GREY, GreyRamp) |
140 |
self.propCombo.Append(PROPCOMBOSTR_RED, RedRamp()) |
self.propCombo.Append(PROPCOMBOSTR_RED, RedRamp) |
141 |
self.propCombo.Append(PROPCOMBOSTR_GREEN, GreenRamp()) |
self.propCombo.Append(PROPCOMBOSTR_GREEN, GreenRamp) |
142 |
self.propCombo.Append(PROPCOMBOSTR_BLUE, BlueRamp()) |
self.propCombo.Append(PROPCOMBOSTR_BLUE, BlueRamp) |
143 |
self.propCombo.Append(PROPCOMBOSTR_GREEN2RED, GreenToRedRamp()) |
self.propCombo.Append(PROPCOMBOSTR_GREEN2RED, GreenToRedRamp) |
144 |
self.propCombo.Append(PROPCOMBOSTR_HOT2COLD, HotToColdRamp()) |
self.propCombo.Append(PROPCOMBOSTR_HOT2COLD, HotToColdRamp()) |
145 |
self.propCombo.Append(PROPCOMBOSTR_CUSTOM, custom_ramp_panel) |
self.propCombo.Append(PROPCOMBOSTR_CUSTOM, custom_ramp_panel) |
146 |
|
|
149 |
psizer.Add(self.propCombo, 1, wxALL | wxGROW, 4) |
psizer.Add(self.propCombo, 1, wxALL | wxGROW, 4) |
150 |
sizer.Add(psizer, 0, wxALL | wxGROW, 4) |
sizer.Add(psizer, 0, wxALL | wxGROW, 4) |
151 |
|
|
152 |
|
if layer.ShapeType() != SHAPETYPE_ARC: |
153 |
|
psizer = wxBoxSizer(wxHORIZONTAL) |
154 |
|
self.fix_border_check = wxCheckBox(self, -1, _("Fix Border Color")) |
155 |
|
psizer.Add(self.fix_border_check, 0, wxALL | wxGROW, 4) |
156 |
|
self.border_color = classifier.ClassGroupPropertiesCtrl( |
157 |
|
self, ID_BORDER_COLOR, |
158 |
|
ClassGroupProperties(), SHAPETYPE_ARC, |
159 |
|
style=wxSIMPLE_BORDER, size=(40, 20)) |
160 |
|
psizer.Add(self.border_color, 0, wxALL | wxGROW, 4) |
161 |
|
psizer.Add(wxButton(self, ID_BORDER_COLOR_CHANGE, _("Change")), |
162 |
|
0, wxALL, 4) |
163 |
|
sizer.Add(psizer, 0, wxALL | wxGROW, 4) |
164 |
|
EVT_BUTTON(self, ID_BORDER_COLOR_CHANGE, self.OnBorderColorChange) |
165 |
|
else: |
166 |
|
self.border_color = None |
167 |
|
|
168 |
sizer.Add(custom_ramp_panel, 1, wxGROW | wxALL, 4) |
sizer.Add(custom_ramp_panel, 1, wxGROW | wxALL, 4) |
169 |
sizer.Show(custom_ramp_panel, False) |
sizer.Show(custom_ramp_panel, False) |
170 |
|
|
223 |
propPanel = self.propCombo.GetClientData(index) |
propPanel = self.propCombo.GetClientData(index) |
224 |
|
|
225 |
ramp = propPanel.GetRamp() |
ramp = propPanel.GetRamp() |
226 |
|
if self.border_color and self.fix_border_check.IsChecked(): |
227 |
|
props = self.border_color.GetProperties() |
228 |
|
lineColor = props.GetLineColor() |
229 |
|
lineWidth = props.GetLineWidth() |
230 |
|
else: |
231 |
|
lineColor = None |
232 |
|
lineWidth = None |
233 |
|
|
234 |
if genSel == GENCOMBOSTR_UNIFORM: |
if genSel == GENCOMBOSTR_UNIFORM: |
235 |
|
|
242 |
|
|
243 |
self.clazz = generate_uniform_distribution( |
self.clazz = generate_uniform_distribution( |
244 |
min, max, numGroups, ramp, |
min, max, numGroups, ramp, |
245 |
self.type == FIELDTYPE_INT) |
self.type == FIELDTYPE_INT, |
246 |
|
fixes = (lineColor, lineWidth, None)) |
247 |
|
|
248 |
self.parent._SetClassification(self.clazz) |
self.parent._SetClassification(self.clazz) |
249 |
|
|
251 |
|
|
252 |
list = genPanel.GetValueList() |
list = genPanel.GetValueList() |
253 |
|
|
254 |
if len(list) > 0 \ |
if len(list) > 0: |
255 |
and numGroups is not None: |
self.clazz = generate_singletons(list, ramp, |
256 |
|
(lineColor, lineWidth, None)) |
|
self.clazz = generate_singletons( |
|
|
list, numGroups, ramp) |
|
|
|
|
257 |
self.parent._SetClassification(self.clazz) |
self.parent._SetClassification(self.clazz) |
258 |
|
|
259 |
elif genSel == GENCOMBOSTR_QUANTILES: |
elif genSel == GENCOMBOSTR_QUANTILES: |
265 |
delta = 1 / float(numGroups) |
delta = 1 / float(numGroups) |
266 |
percents = [delta * i for i in range(1, numGroups + 1)] |
percents = [delta * i for i in range(1, numGroups + 1)] |
267 |
adjusted, self.clazz = \ |
adjusted, self.clazz = \ |
268 |
generate_quantiles(_list, percents, ramp, _range) |
generate_quantiles(_list, percents, ramp, _range, |
269 |
|
(lineColor, lineWidth, None)) |
270 |
|
|
271 |
if adjusted: |
if adjusted: |
272 |
dlg = wxMessageDialog(self, |
dlg = wxMessageDialog(self, |
284 |
def OnCancel(self, event): |
def OnCancel(self, event): |
285 |
self.Close() |
self.Close() |
286 |
|
|
287 |
|
def OnBorderColorChange(self, event): |
288 |
|
self.border_color.DoEdit() |
289 |
|
|
290 |
def _OnGenTypeSelect(self, event): |
def _OnGenTypeSelect(self, event): |
291 |
self.__DoOnGenTypeSelect() |
self.__DoOnGenTypeSelect() |
292 |
return |
return |
537 |
def _OnRetrieve(self, event): |
def _OnRetrieve(self, event): |
538 |
table = self.layer.ShapeStore().Table() |
table = self.layer.ShapeStore().Table() |
539 |
if table is not None: |
if table is not None: |
540 |
wxBeginBusyCursor() |
ThubanBeginBusyCursor() |
541 |
try: |
try: |
542 |
min, max = table.ValueRange(self.fieldName) |
min, max = table.ValueRange(self.fieldName) |
543 |
self.minCtrl.SetValue(str(min)) |
self.minCtrl.SetValue(str(min)) |
544 |
self.maxCtrl.SetValue(str(max)) |
self.maxCtrl.SetValue(str(max)) |
545 |
finally: |
finally: |
546 |
wxEndBusyCursor() |
ThubanEndBusyCursor() |
547 |
|
|
548 |
def __GetValidatedTypeEntry(self, win, value, type, badValue = None): |
def __GetValidatedTypeEntry(self, win, value, type, badValue = None): |
549 |
|
|
751 |
self.list_avail.DeleteAllItems() |
self.list_avail.DeleteAllItems() |
752 |
self.list_avail_data = [] |
self.list_avail_data = [] |
753 |
|
|
754 |
list = self.layer.ShapeStore().Table().UniqueValues(self.fieldName) |
ThubanBeginBusyCursor() |
755 |
index = 0 |
try: |
756 |
for v in list: |
list = self.layer.ShapeStore().Table().UniqueValues(self.fieldName) |
757 |
self.dataList.append(v) |
index = 0 |
758 |
i = self.list_avail.InsertStringItem(index, str(v)) |
for v in list: |
759 |
self.list_avail.SetItemData(index, i) |
self.dataList.append(v) |
760 |
|
i = self.list_avail.InsertStringItem(index, str(v)) |
761 |
self.list_avail_data.append(v) |
self.list_avail.SetItemData(index, i) |
762 |
index += 1 |
|
763 |
|
self.list_avail_data.append(v) |
764 |
|
index += 1 |
765 |
|
finally: |
766 |
|
ThubanEndBusyCursor() |
767 |
|
|
768 |
def _OnUseAll(self, event): |
def _OnUseAll(self, event): |
769 |
for i in range(self.list_avail.GetItemCount()): |
for i in range(self.list_avail.GetItemCount()): |
870 |
_list = [] |
_list = [] |
871 |
table = self.layer.ShapeStore().Table() |
table = self.layer.ShapeStore().Table() |
872 |
if table is not None: |
if table is not None: |
873 |
wxBeginBusyCursor() |
ThubanBeginBusyCursor() |
874 |
try: |
try: |
875 |
# |
# |
876 |
# FIXME: Replace with a call to table when the method |
# FIXME: Replace with a call to table when the method |
879 |
for i in range(table.NumRows()): |
for i in range(table.NumRows()): |
880 |
_list.append(table.ReadValue(i, self.fieldName)) |
_list.append(table.ReadValue(i, self.fieldName)) |
881 |
finally: |
finally: |
882 |
wxEndBusyCursor() |
ThubanEndBusyCursor() |
883 |
|
|
884 |
return _list |
return _list |
885 |
|
|
898 |
def OnRetrieve(self, event): |
def OnRetrieve(self, event): |
899 |
table = self.layer.ShapeStore().Table() |
table = self.layer.ShapeStore().Table() |
900 |
if table is not None: |
if table is not None: |
901 |
wxBeginBusyCursor() |
ThubanBeginBusyCursor() |
902 |
try: |
try: |
903 |
min, max = table.ValueRange(self.fieldName) |
min, max = table.ValueRange(self.fieldName) |
904 |
self.text_range.SetValue("[" + str(min) + ";" + str(max) + "]") |
self.text_range.SetValue("[" + str(min) + ";" + str(max) + "]") |
905 |
finally: |
finally: |
906 |
wxEndBusyCursor() |
ThubanEndBusyCursor() |
907 |
|
|
908 |
ID_CUSTOMRAMP_COPYSTART = 4001 |
ID_CUSTOMRAMP_COPYSTART = 4001 |
909 |
ID_CUSTOMRAMP_COPYEND = 4002 |
ID_CUSTOMRAMP_COPYEND = 4002 |