2 |
# Authors: |
# Authors: |
3 |
# Jan-Oliver Wagner <[email protected]> (2003-2004) |
# Jan-Oliver Wagner <[email protected]> (2003-2004) |
4 |
# Martin Schulze <[email protected]> (2004) |
# Martin Schulze <[email protected]> (2004) |
5 |
# Frank Koormann <[email protected]> (2003) |
# Frank Koormann <[email protected]> (2003, 2006) |
6 |
# Bernhard Herzog <[email protected]> (2003) |
# Bernhard Herzog <[email protected]> (2003) |
7 |
# Jonathan Coles <[email protected]> (2003) |
# Jonathan Coles <[email protected]> (2003) |
8 |
# |
# |
16 |
# $Id$ |
# $Id$ |
17 |
|
|
18 |
import copy |
import copy |
19 |
|
import re |
20 |
|
|
21 |
from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \ |
from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \ |
22 |
FIELDTYPE_STRING |
FIELDTYPE_STRING |
31 |
from Thuban.Model.range import Range |
from Thuban.Model.range import Range |
32 |
from Thuban.Model.classification import \ |
from Thuban.Model.classification import \ |
33 |
Classification, ClassGroupDefault, \ |
Classification, ClassGroupDefault, \ |
34 |
ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \ |
ClassGroupSingleton, ClassGroupPattern, ClassGroupRange, ClassGroupMap, \ |
35 |
ClassGroupProperties |
ClassGroupProperties |
36 |
|
|
37 |
from Thuban.Model.color import Transparent |
from Thuban.Model.color import Transparent |
85 |
EVT_GRID_SELECT_CELL(self, self._OnSelectedCell) |
EVT_GRID_SELECT_CELL(self, self._OnSelectedCell) |
86 |
EVT_GRID_COL_SIZE(self, self._OnCellResize) |
EVT_GRID_COL_SIZE(self, self._OnCellResize) |
87 |
EVT_GRID_ROW_SIZE(self, self._OnCellResize) |
EVT_GRID_ROW_SIZE(self, self._OnCellResize) |
88 |
|
EVT_GRID_LABEL_RIGHT_CLICK(self, self._OnLabelRightClicked) |
89 |
|
|
90 |
|
|
91 |
#def GetCellAttr(self, row, col): |
#def GetCellAttr(self, row, col): |
92 |
#print "GetCellAttr ", row, col |
#print "GetCellAttr ", row, col |
259 |
self.FitInside() |
self.FitInside() |
260 |
event.Skip() |
event.Skip() |
261 |
|
|
262 |
|
def _OnLabelRightClicked(self, event): |
263 |
|
"""Process right click on label, raise popup for row labels.""" |
264 |
|
row, col = event.GetRow(), event.GetCol() |
265 |
|
if col == -1: |
266 |
|
self.labelPopup(event, row) |
267 |
|
|
268 |
|
def labelPopup(self, event, row): |
269 |
|
"""Raise grid label popup.""" |
270 |
|
# check if row label is Pattern or Singleton |
271 |
|
label = self.GetRowLabelValue(row) |
272 |
|
if (label == _("Pattern") or label == _("Singleton")): |
273 |
|
xe,ye = event.GetPosition() |
274 |
|
x=self.GetRowSize(row)/2 |
275 |
|
menu = wxMenu() |
276 |
|
patternID = wxNewId() |
277 |
|
singletonID = wxNewId() |
278 |
|
|
279 |
|
def _SetSingleton(event, self=self, row=row): |
280 |
|
table = self.GetTable() |
281 |
|
group = table.clazz.GetGroup(row - 1) |
282 |
|
if not isinstance(group, ClassGroupSingleton): |
283 |
|
ngroup = ClassGroupSingleton( |
284 |
|
group.GetPattern(), |
285 |
|
group.GetProperties(), |
286 |
|
group.GetLabel() |
287 |
|
) |
288 |
|
table.SetClassGroup(row, ngroup) |
289 |
|
|
290 |
|
def _SetPattern(event, self=self, row=row): |
291 |
|
table = self.GetTable() |
292 |
|
group = table.clazz.GetGroup(row - 1) |
293 |
|
if not isinstance(group, ClassGroupPattern): |
294 |
|
try: |
295 |
|
re.compile(group.GetValue()) |
296 |
|
except: |
297 |
|
pass |
298 |
|
else: |
299 |
|
ngroup = ClassGroupPattern( |
300 |
|
group.GetValue(), |
301 |
|
group.GetProperties(), |
302 |
|
group.GetLabel() |
303 |
|
) |
304 |
|
table.SetClassGroup(row, ngroup) |
305 |
|
|
306 |
|
menu.Append(singletonID, _("Singleton")) |
307 |
|
EVT_MENU(self, singletonID, _SetSingleton) |
308 |
|
if self.GetTable().fieldType == FIELDTYPE_STRING: |
309 |
|
menu.Append(patternID, _("Pattern")) |
310 |
|
EVT_MENU(self, patternID, _SetPattern) |
311 |
|
self.PopupMenu(menu, wxPoint(x,ye)) |
312 |
|
menu.Destroy() |
313 |
|
|
314 |
class ClassTable(wxPyGridTableBase): |
class ClassTable(wxPyGridTableBase): |
315 |
"""Represents the underlying data structure for the grid.""" |
"""Represents the underlying data structure for the grid.""" |
316 |
|
|
464 |
group = self.clazz.GetGroup(row - 1) |
group = self.clazz.GetGroup(row - 1) |
465 |
if isinstance(group, ClassGroupDefault): return _("Default") |
if isinstance(group, ClassGroupDefault): return _("Default") |
466 |
if isinstance(group, ClassGroupSingleton): return _("Singleton") |
if isinstance(group, ClassGroupSingleton): return _("Singleton") |
467 |
|
if isinstance(group, ClassGroupPattern): return _("Pattern") |
468 |
if isinstance(group, ClassGroupRange): return _("Range") |
if isinstance(group, ClassGroupRange): return _("Range") |
469 |
if isinstance(group, ClassGroupMap): return _("Map") |
if isinstance(group, ClassGroupMap): return _("Map") |
470 |
|
|
528 |
return _("DEFAULT") |
return _("DEFAULT") |
529 |
elif isinstance(group, ClassGroupSingleton): |
elif isinstance(group, ClassGroupSingleton): |
530 |
return group.GetValue() |
return group.GetValue() |
531 |
|
elif isinstance(group, ClassGroupPattern): |
532 |
|
return group.GetPattern() |
533 |
elif isinstance(group, ClassGroupRange): |
elif isinstance(group, ClassGroupRange): |
534 |
return group.GetRange() |
return group.GetRange() |
535 |
|
|
541 |
(string, number, or range) |
(string, number, or range) |
542 |
|
|
543 |
Returns a tuple (type, data) where type is 0 if data is |
Returns a tuple (type, data) where type is 0 if data is |
544 |
a singleton value, or 1 if is a range |
a singleton value, 1 if is a range or 2 if it is a pattern. |
545 |
""" |
""" |
546 |
|
|
547 |
type = self.fieldType |
type = self.fieldType |
548 |
|
|
549 |
if type == FIELDTYPE_STRING: |
if type == FIELDTYPE_STRING: |
550 |
return (0, value) |
# Approach: if we can compile the value as an expression, |
551 |
|
# make it a pattern, else a singleton. |
552 |
|
# This is quite crude, however I don't have a better idea: |
553 |
|
# How to distinct the singleton "Thuban" from the pattern "Thuban"? |
554 |
|
try: |
555 |
|
re.compile(value) |
556 |
|
except: |
557 |
|
return (0, value) |
558 |
|
else: |
559 |
|
return (2, value) |
560 |
elif type in (FIELDTYPE_INT, FIELDTYPE_DOUBLE): |
elif type in (FIELDTYPE_INT, FIELDTYPE_DOUBLE): |
561 |
if type == FIELDTYPE_INT: |
if type == FIELDTYPE_INT: |
562 |
# the float call allows the user to enter 1.0 for 1 |
# the float call allows the user to enter 1.0 for 1 |
640 |
ngroup = ClassGroupRange(props = props) |
ngroup = ClassGroupRange(props = props) |
641 |
changed = True |
changed = True |
642 |
ngroup.SetRange(dataInfo[1]) |
ngroup.SetRange(dataInfo[1]) |
643 |
|
elif dataInfo[0] == 2: |
644 |
|
if not isinstance(group, ClassGroupPattern): |
645 |
|
ngroup = ClassGroupPattern(props = props) |
646 |
|
changed = True |
647 |
|
ngroup.SetPattern(dataInfo[1]) |
648 |
else: |
else: |
649 |
assert False |
assert False |
650 |
pass |
pass |