Thuban/Model/range.py

#!/usr/bin/env python
#
# Copyright (C) 2002, 2003 by Intevation GmbH
# Authors:
# Thomas Koester <[email protected]>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with the software for details.

"""
Range class for Scientific Parameter
"""

__version__ = "$Revision$"
# $Source$
# $Id$

import re
import types

_inf = float('1e1000')   # FIXME: hack for infinite

class Range:

    number_re = '(?P<%s>-?(\d*\.?\d*([eE][\-+]?\d+)?|oo))'
    brace_re = '(?P<%s>[][])'
    range_re = '^' + brace_re % 'left' + number_re % 'begin' + \
               ';' + number_re % 'end' + brace_re % 'right' + '$'
    parse_range = re.compile(range_re)

    def __init__(self, range=None):
        self.SetRange(range)

    def SetRange(self, range):
        if isinstance(range, Range):
            self.SetRange(str(range))
        elif range in [None, '']:
            self.SetRange(']-oo;oo[')
        elif type(range) == types.TupleType:
            if (len(range) == 4 and range[0] in ['[', ']']
                and range[3] in ['[', ']']):
                self._left = range[0]
                self._begin = self.float(range[1])
                self._end = self.float(range[2])
                self._right = range[3]
                if (self._begin > self._end or
                    (self._begin == self._end and
                     (self._left != '[' or self._right != ']'))):
                    raise ValueError("illegal range: %s" % (range,))
            else:
                raise ValueError("can't parse range: %r" % (range,))
        else:
            self._range = range
            match = self.parse_range.match(self._range)
            if match:
                self.SetRange((match.group('left'),
                              match.group('begin'),
                              match.group('end'),
                              match.group('right')))
            else:
                raise ValueError("can't parse range: %s" % (range,))

    def GetRange(self):
        """return internal representation of range

        4-tuple ('[' or ']', begin(float), end(float), '[' or ']')

        """
        return (self._left, self._begin, self._end, self._right)

    def float(self, string):
        if string == 'oo':
            return _inf
        elif string == '-oo':
            return -_inf
        else:
            return float(string)

    def _float2string(self, value):
        """convert float value to string

        (minus) infinity will be converted to (-)oo,
        scientific notation will be used if necessary.

        """
        if value == _inf:
            return 'oo'
        elif value == -_inf:
            return '-oo'
        else:
            return "%g" % (value,)

    def string(self, range):
        """convert internal representation to string"""
        left, begin, end, right = range
        return "%s%s;%s%s" % (left, self._float2string(begin),
                              self._float2string(end), right)

    def __contains__(self, value):
        if self._left == ']':
            contains = value > self._begin
        else:
            contains = value >= self._begin
        if self._right == '[':
            contains = contains and (value < self._end)
        else:
            contains = contains and (value <= self._end)
        return contains

    def __eq__(self, other):
        return (self.GetRange() == other.GetRange())

    def __ne__(self, other):
        return not self.__eq__(other)

    def __str__(self):
        return self.string(self.GetRange())


def _test():
    range1 = Range(']0;99]')
    print 'range1 =', range1, range1.GetRange()
    for i in [-0.1, 0, 0.1, 9.9, 99, 99.9]:
        print '%4.1f in range1 =' % i, i in range1
    range2 = Range(']-oo;10[')
    print 'range2 =', range2, range2.GetRange()
    for i in [-0.1, 0, 0.1, 9.9, 10, 10.1]:
        print '%4.1f not in range2 =' % i, i not in range2
    range3 = Range(']1e-1;1E2]')
    print 'range3 =', range3, range3.GetRange()
    for i in [0, 0.1, 0.11, 10, 100, 101]:
        print '%4.1f not in range3 =' % i, i not in range3
    print 'range3 != range2 =', range3 != range2
    print 'range3 != Range("]1e-1;1E2]") =', range3 != Range("]1e-1;1E2]")

    range4 = Range('')
    print 'range4 =', range4, range4.GetRange()

    range5 = Range(']0;99E+00]')
    print 'range5 =', range5, range5.GetRange()
    range6 = Range(']0;99E+01]')
    print 'range6 =', range6, range6.GetRange()
    range7 = Range(']0;99E-01]')
    print 'range7 =', range7, range7.GetRange()

if __name__ == "__main__":
    _test()
1	jonathan	887	#!/usr/bin/env python
2			#
3			# Copyright (C) 2002, 2003 by Intevation GmbH
4			# Authors:
5			# Thomas Koester <[email protected]>
6			#
7			# This program is free software under the GPL (>=v2)
8			# Read the file COPYING coming with the software for details.
9
10			"""
11			Range class for Scientific Parameter
12			"""
13
14			__version__ = "$Revision$"
15			# $Source$
16			# $Id$
17
18			import re
19	jonathan	962	import types
20	jonathan	887
21	jonathan	962	_inf = float('1e1000') # FIXME: hack for infinite
22	jonathan	887
23			class Range:
24
25	jonathan	901	number_re = '(?P<%s>-?(\d\.?\d([eE][\-+]?\d+)?\|oo))'
26	jonathan	887	brace_re = '(?P<%s>[][])'
27			range_re = '^' + brace_re % 'left' + number_re % 'begin' + \
28			';' + number_re % 'end' + brace_re % 'right' + '$'
29			parse_range = re.compile(range_re)
30
31			def __init__(self, range=None):
32			self.SetRange(range)
33
34			def SetRange(self, range):
35			if isinstance(range, Range):
36			self.SetRange(str(range))
37			elif range in [None, '']:
38			self.SetRange(']-oo;oo[')
39	jonathan	962	elif type(range) == types.TupleType:
40			if (len(range) == 4 and range[0] in ['[', ']']
41			and range[3] in ['[', ']']):
42			self._left = range[0]
43			self._begin = self.float(range[1])
44			self._end = self.float(range[2])
45			self._right = range[3]
46			if (self._begin > self._end or
47			(self._begin == self._end and
48			(self._left != '[' or self._right != ']'))):
49			raise ValueError("illegal range: %s" % (range,))
50			else:
51			raise ValueError("can't parse range: %r" % (range,))
52	jonathan	887	else:
53			self._range = range
54			match = self.parse_range.match(self._range)
55			if match:
56	jonathan	962	self.SetRange((match.group('left'),
57			match.group('begin'),
58			match.group('end'),
59			match.group('right')))
60	jonathan	887	else:
61	jonathan	962	raise ValueError("can't parse range: %s" % (range,))
62	jonathan	887
63			def GetRange(self):
64			"""return internal representation of range
65
66			4-tuple ('[' or ']', begin(float), end(float), '[' or ']')
67
68			"""
69			return (self._left, self._begin, self._end, self._right)
70
71			def float(self, string):
72			if string == 'oo':
73			return _inf
74			elif string == '-oo':
75			return -_inf
76			else:
77			return float(string)
78
79			def _float2string(self, value):
80			"""convert float value to string
81
82			(minus) infinity will be converted to (-)oo,
83			scientific notation will be used if necessary.
84
85			"""
86			if value == _inf:
87			return 'oo'
88			elif value == -_inf:
89			return '-oo'
90			else:
91	jonathan	962	return "%g" % (value,)
92	jonathan	887
93			def string(self, range):
94			"""convert internal representation to string"""
95			left, begin, end, right = range
96			return "%s%s;%s%s" % (left, self._float2string(begin),
97			self._float2string(end), right)
98
99			def __contains__(self, value):
100			if self._left == ']':
101			contains = value > self._begin
102			else:
103			contains = value >= self._begin
104			if self._right == '[':
105			contains = contains and (value < self._end)
106			else:
107			contains = contains and (value <= self._end)
108			return contains
109
110			def __eq__(self, other):
111			return (self.GetRange() == other.GetRange())
112
113			def __ne__(self, other):
114			return not self.__eq__(other)
115
116			def __str__(self):
117			return self.string(self.GetRange())
118
119
120			def _test():
121			range1 = Range(']0;99]')
122			print 'range1 =', range1, range1.GetRange()
123			for i in [-0.1, 0, 0.1, 9.9, 99, 99.9]:
124			print '%4.1f in range1 =' % i, i in range1
125			range2 = Range(']-oo;10[')
126			print 'range2 =', range2, range2.GetRange()
127			for i in [-0.1, 0, 0.1, 9.9, 10, 10.1]:
128			print '%4.1f not in range2 =' % i, i not in range2
129			range3 = Range(']1e-1;1E2]')
130			print 'range3 =', range3, range3.GetRange()
131			for i in [0, 0.1, 0.11, 10, 100, 101]:
132			print '%4.1f not in range3 =' % i, i not in range3
133			print 'range3 != range2 =', range3 != range2
134			print 'range3 != Range("]1e-1;1E2]") =', range3 != Range("]1e-1;1E2]")
135
136			range4 = Range('')
137			print 'range4 =', range4, range4.GetRange()
138
139			range5 = Range(']0;99E+00]')
140			print 'range5 =', range5, range5.GetRange()
141			range6 = Range(']0;99E+01]')
142			print 'range6 =', range6, range6.GetRange()
143			range7 = Range(']0;99E-01]')
144			print 'range7 =', range7, range7.GetRange()
145
146			if __name__ == "__main__":
147			_test()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision