20 |
on the mapping algorithm. |
on the mapping algorithm. |
21 |
""" |
""" |
22 |
|
|
23 |
# fix for people using python2.1 |
import copy, operator, types |
|
from __future__ import nested_scopes |
|
|
|
|
|
import copy |
|
24 |
|
|
25 |
from Thuban import _ |
from Thuban import _ |
26 |
|
|
|
from types import * |
|
|
|
|
27 |
from messages import \ |
from messages import \ |
28 |
LAYER_PROJECTION_CHANGED, \ |
LAYER_PROJECTION_CHANGED, \ |
29 |
LAYER_LEGEND_CHANGED, \ |
LAYER_LEGEND_CHANGED, \ |
30 |
LAYER_VISIBILITY_CHANGED |
LAYER_VISIBILITY_CHANGED |
31 |
|
|
32 |
from Thuban.Model.color import Color |
from Thuban.Model.color import Color |
33 |
|
from Thuban.Model.range import Range |
34 |
|
|
35 |
import Thuban.Model.layer |
import Thuban.Model.layer |
36 |
|
|
192 |
|
|
193 |
fill -- a Color object. |
fill -- a Color object. |
194 |
""" |
""" |
|
assert isinstance(fill, Color) |
|
195 |
self.GetDefaultGroup().GetProperties().SetFill(fill) |
self.GetDefaultGroup().GetProperties().SetFill(fill) |
196 |
self.__SendNotification() |
self.__SendNotification() |
197 |
|
|
204 |
|
|
205 |
color -- a Color object. |
color -- a Color object. |
206 |
""" |
""" |
|
assert isinstance(color, Color) |
|
207 |
self.GetDefaultGroup().GetProperties().SetLineColor(color) |
self.GetDefaultGroup().GetProperties().SetLineColor(color) |
208 |
self.__SendNotification() |
self.__SendNotification() |
209 |
|
|
216 |
|
|
217 |
lineWidth -- an integer > 0. |
lineWidth -- an integer > 0. |
218 |
""" |
""" |
219 |
assert isinstance(lineWidth, IntType) |
assert isinstance(lineWidth, types.IntType) |
220 |
self.GetDefaultGroup().GetProperties().SetLineWidth(lineWidth) |
self.GetDefaultGroup().GetProperties().SetLineWidth(lineWidth) |
221 |
self.__SendNotification() |
self.__SendNotification() |
222 |
|
|
443 |
|
|
444 |
def GetLineColor(self): |
def GetLineColor(self): |
445 |
"""Return the line color as a Color object.""" |
"""Return the line color as a Color object.""" |
446 |
return self.stroke |
return self.__stroke |
447 |
|
|
448 |
def SetLineColor(self, color): |
def SetLineColor(self, color): |
449 |
"""Set the line color. |
"""Set the line color. |
451 |
color -- the color of the line. This must be a Color object. |
color -- the color of the line. This must be a Color object. |
452 |
""" |
""" |
453 |
|
|
454 |
assert isinstance(color, Color) |
self.__stroke = color |
|
self.stroke = color |
|
455 |
|
|
456 |
def GetLineWidth(self): |
def GetLineWidth(self): |
457 |
"""Return the line width.""" |
"""Return the line width.""" |
458 |
return self.strokeWidth |
return self.__strokeWidth |
459 |
|
|
460 |
def SetLineWidth(self, lineWidth): |
def SetLineWidth(self, lineWidth): |
461 |
"""Set the line width. |
"""Set the line width. |
462 |
|
|
463 |
lineWidth -- the new line width. This must be > 0. |
lineWidth -- the new line width. This must be > 0. |
464 |
""" |
""" |
465 |
assert isinstance(lineWidth, IntType) |
assert isinstance(lineWidth, types.IntType) |
466 |
if (lineWidth < 1): |
if (lineWidth < 1): |
467 |
raise ValueError(_("lineWidth < 1")) |
raise ValueError(_("lineWidth < 1")) |
468 |
|
|
469 |
self.strokeWidth = lineWidth |
self.__strokeWidth = lineWidth |
470 |
|
|
471 |
def GetFill(self): |
def GetFill(self): |
472 |
"""Return the fill color as a Color object.""" |
"""Return the fill color as a Color object.""" |
473 |
return self.fill |
return self.__fill |
474 |
|
|
475 |
def SetFill(self, fill): |
def SetFill(self, fill): |
476 |
"""Set the fill color. |
"""Set the fill color. |
478 |
fill -- the color of the fill. This must be a Color object. |
fill -- the color of the fill. This must be a Color object. |
479 |
""" |
""" |
480 |
|
|
481 |
assert isinstance(fill, Color) |
self.__fill = fill |
|
self.fill = fill |
|
482 |
|
|
483 |
def __eq__(self, other): |
def __eq__(self, other): |
484 |
"""Return true if 'props' has the same attributes as this class""" |
"""Return true if 'props' has the same attributes as this class""" |
485 |
|
|
486 |
|
# |
487 |
|
# using 'is' over '==' results in a huge performance gain |
488 |
|
# in the renderer |
489 |
|
# |
490 |
return isinstance(other, ClassGroupProperties) \ |
return isinstance(other, ClassGroupProperties) \ |
491 |
and self.stroke == other.GetLineColor() \ |
and (self.__stroke is other.__stroke or \ |
492 |
and self.strokeWidth == other.GetLineWidth() \ |
self.__stroke == other.__stroke) \ |
493 |
and self.fill == other.GetFill() |
and (self.__fill is other.__fill or \ |
494 |
|
self.__fill == other.__fill) \ |
495 |
|
and self.__strokeWidth == other.__strokeWidth |
496 |
|
|
497 |
def __ne__(self, other): |
def __ne__(self, other): |
498 |
return not self.__eq__(other) |
return not self.__eq__(other) |
503 |
def __deepcopy__(self): |
def __deepcopy__(self): |
504 |
return ClassGroupProperties(self) |
return ClassGroupProperties(self) |
505 |
|
|
506 |
|
def __repr__(self): |
507 |
|
return repr((self.__stroke, self.__strokeWidth, self.__fill)) |
508 |
|
|
509 |
class ClassGroup: |
class ClassGroup: |
510 |
"""A base class for all Groups within a Classification""" |
"""A base class for all Groups within a Classification""" |
511 |
|
|
534 |
label -- a string representing the Group's label. This must |
label -- a string representing the Group's label. This must |
535 |
not be None. |
not be None. |
536 |
""" |
""" |
537 |
assert isinstance(label, StringType) |
assert isinstance(label, types.StringTypes) |
538 |
self.label = label |
self.label = label |
539 |
|
|
540 |
def GetDisplayText(self): |
def GetDisplayText(self): |
573 |
|
|
574 |
def __eq__(self, other): |
def __eq__(self, other): |
575 |
return isinstance(other, ClassGroup) \ |
return isinstance(other, ClassGroup) \ |
576 |
|
and self.label == other.label \ |
577 |
and self.GetProperties() == other.GetProperties() |
and self.GetProperties() == other.GetProperties() |
578 |
|
|
579 |
def __ne__(self, other): |
def __ne__(self, other): |
580 |
return not self.__eq__(other) |
return not self.__eq__(other) |
581 |
|
|
582 |
|
def __repr__(self): |
583 |
|
return repr(self.label) + ", " + repr(self.GetProperties()) |
584 |
|
|
585 |
class ClassGroupSingleton(ClassGroup): |
class ClassGroupSingleton(ClassGroup): |
586 |
"""A Group that is associated with a single value.""" |
"""A Group that is associated with a single value.""" |
609 |
|
|
610 |
def GetValue(self): |
def GetValue(self): |
611 |
"""Return the associated value.""" |
"""Return the associated value.""" |
612 |
return self.value |
return self.__value |
613 |
|
|
614 |
def SetValue(self, value): |
def SetValue(self, value): |
615 |
"""Associate this Group with the given value.""" |
"""Associate this Group with the given value.""" |
616 |
self.value = value |
self.__value = value |
617 |
|
|
618 |
def Matches(self, value): |
def Matches(self, value): |
619 |
"""Determine if the given value matches the associated Group value.""" |
"""Determine if the given value matches the associated Group value.""" |
620 |
|
|
621 |
"""Returns True if the value matches, False otherwise.""" |
"""Returns True if the value matches, False otherwise.""" |
622 |
|
|
623 |
return self.value == value |
return self.__value == value |
624 |
|
|
625 |
def GetDisplayText(self): |
def GetDisplayText(self): |
626 |
label = self.GetLabel() |
label = self.GetLabel() |
632 |
def __eq__(self, other): |
def __eq__(self, other): |
633 |
return ClassGroup.__eq__(self, other) \ |
return ClassGroup.__eq__(self, other) \ |
634 |
and isinstance(other, ClassGroupSingleton) \ |
and isinstance(other, ClassGroupSingleton) \ |
635 |
and self.GetValue() == other.GetValue() |
and self.__value == other.__value |
636 |
|
|
637 |
|
def __repr__(self): |
638 |
|
return "(" + repr(self.__value) + ", " + ClassGroup.__repr__(self) + ")" |
639 |
|
|
640 |
class ClassGroupDefault(ClassGroup): |
class ClassGroupDefault(ClassGroup): |
641 |
"""The default Group. When values do not match any other |
"""The default Group. When values do not match any other |
674 |
and isinstance(other, ClassGroupDefault) \ |
and isinstance(other, ClassGroupDefault) \ |
675 |
and self.GetProperties() == other.GetProperties() |
and self.GetProperties() == other.GetProperties() |
676 |
|
|
677 |
|
def __repr__(self): |
678 |
|
return "(" + ClassGroup.__repr__(self) + ")" |
679 |
|
|
680 |
class ClassGroupRange(ClassGroup): |
class ClassGroupRange(ClassGroup): |
681 |
"""A Group that represents a range of values that map to the same |
"""A Group that represents a range of values that map to the same |
682 |
set of properties.""" |
set of properties.""" |
698 |
|
|
699 |
ClassGroup.__init__(self, label, props, group) |
ClassGroup.__init__(self, label, props, group) |
700 |
|
|
701 |
self.min = self.max = 0 |
#self.__min = self.__max = 0 |
702 |
|
#self.__range = Range("[" + repr(float(min)) + ";" + |
703 |
|
#repr(float(max)) + "[") |
704 |
self.SetRange(min, max) |
self.SetRange(min, max) |
705 |
|
|
706 |
def __copy__(self): |
def __copy__(self): |
707 |
return ClassGroupRange(self.GetMin(), |
return ClassGroupRange(min = self.__range, |
708 |
self.GetMax(), |
max = None, |
709 |
self.GetProperties(), |
props = self.GetProperties(), |
710 |
self.GetLabel()) |
label = self.GetLabel()) |
711 |
|
|
712 |
def __deepcopy__(self, memo): |
def __deepcopy__(self, memo): |
713 |
return ClassGroupRange(copy.copy(self.GetMin()), |
return ClassGroupRange(min = copy.copy(self.__range), |
714 |
copy.copy(self.GetMax()), |
max = copy.copy(self.GetMax()), |
715 |
group = self) |
group = self) |
716 |
|
|
717 |
def GetMin(self): |
def GetMin(self): |
718 |
"""Return the range's minimum value.""" |
"""Return the range's minimum value.""" |
719 |
return self.min |
return self.__range.GetRange()[1] |
720 |
|
|
721 |
def SetMin(self, min): |
def SetMin(self, min): |
722 |
"""Set the range's minimum value. |
"""Set the range's minimum value. |
725 |
maximum value. Use SetRange() to change both min and max values. |
maximum value. Use SetRange() to change both min and max values. |
726 |
""" |
""" |
727 |
|
|
728 |
self.SetRange(min, self.max) |
self.SetRange(min, self.__range.GetRange()[2]) |
729 |
|
|
730 |
def GetMax(self): |
def GetMax(self): |
731 |
"""Return the range's maximum value.""" |
"""Return the range's maximum value.""" |
732 |
return self.max |
return self.__range.GetRange()[2] |
733 |
|
|
734 |
def SetMax(self, max): |
def SetMax(self, max): |
735 |
"""Set the range's maximum value. |
"""Set the range's maximum value. |
737 |
max -- the new maximum. Note that this must be greater than the current |
max -- the new maximum. Note that this must be greater than the current |
738 |
minimum value. Use SetRange() to change both min and max values. |
minimum value. Use SetRange() to change both min and max values. |
739 |
""" |
""" |
740 |
self.SetRange(self.min, max) |
self.SetRange(self.__range.GetRange()[1], max) |
741 |
|
|
742 |
def SetRange(self, min, max): |
def SetRange(self, min, max = None): |
743 |
"""Set a new range. |
"""Set a new range. |
744 |
|
|
745 |
Note that min must be strictly less than max. |
Note that min must be strictly less than max. |
748 |
min -- the new maximum value |
min -- the new maximum value |
749 |
""" |
""" |
750 |
|
|
751 |
if min >= max: |
if isinstance(min, Range): |
752 |
raise ValueError(_("ClassGroupRange: %i(min) >= %i(max)!") % |
self.__range = min |
753 |
(min, max)) |
else: |
754 |
self.min = min |
if max is None: |
755 |
self.max = max |
raise ValueError() |
756 |
|
|
757 |
|
self.__range = Range(("[", min, max, "[")) |
758 |
|
|
759 |
def GetRange(self): |
def GetRange(self): |
760 |
"""Return the range as a tuple (min, max)""" |
"""Return the range as a string""" |
761 |
return (self.min, self.max) |
#return (self.__min, self.__max) |
762 |
|
return self.__range.string(self.__range.GetRange()) |
763 |
|
|
764 |
def Matches(self, value): |
def Matches(self, value): |
765 |
"""Determine if the given value lies with the current range. |
"""Determine if the given value lies with the current range. |
767 |
The following check is used: min <= value < max. |
The following check is used: min <= value < max. |
768 |
""" |
""" |
769 |
|
|
770 |
return self.min <= value < self.max |
return operator.contains(self.__range, value) |
771 |
|
#return self.__min <= value < self.__max |
772 |
|
|
773 |
def GetDisplayText(self): |
def GetDisplayText(self): |
774 |
label = self.GetLabel() |
label = self.GetLabel() |
775 |
|
|
776 |
if label != "": return label |
if label != "": return label |
777 |
|
|
778 |
return _("%s - %s") % (self.GetMin(), self.GetMax()) |
#return _("%s - %s") % (self.GetMin(), self.GetMax()) |
779 |
|
#return repr(self.__range) |
780 |
|
return self.__range.string(self.__range.GetRange()) |
781 |
|
|
782 |
def __eq__(self, other): |
def __eq__(self, other): |
783 |
return ClassGroup.__eq__(self, other) \ |
return ClassGroup.__eq__(self, other) \ |
784 |
and isinstance(other, ClassGroupRange) \ |
and isinstance(other, ClassGroupRange) \ |
785 |
and self.GetRange() == other.GetRange() |
and self.__range == other.__range |
786 |
|
#and self.__min == other.__min \ |
787 |
|
#and self.__max == other.__max |
788 |
|
|
789 |
|
def __repr__(self): |
790 |
|
return "(" + str(self.__range) + ClassGroup.__repr__(self) + ")" |
791 |
|
#return "(" + repr(self.__min) + ", " + repr(self.__max) + ", " + \ |
792 |
|
#ClassGroup.__repr__(self) + ")" |
793 |
|
|
794 |
class ClassGroupMap(ClassGroup): |
class ClassGroupMap(ClassGroup): |
795 |
"""Currently, this class is not used.""" |
"""Currently, this class is not used.""" |