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(range.GetRange())
        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, value):
        """convert string or number to float"""
        if value == 'oo':
            return _inf
        elif value == '-oo':
            return -_inf
        else:
            return float(value)

    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	tkoester	993	self._SetRange(range)
33	jonathan	887
34	tkoester	993	def _SetRange(self, range):
35	jonathan	887	if isinstance(range, Range):
36	tkoester	993	self._SetRange(range.GetRange())
37	jonathan	887	elif range in [None, '']:
38	tkoester	993	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	tkoester	993	self._SetRange((match.group('left'),
57	jonathan	962	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	tkoester	993	def float(self, value):
72			"""convert string or number to float"""
73			if value == 'oo':
74	jonathan	887	return _inf
75	tkoester	993	elif value == '-oo':
76	jonathan	887	return -_inf
77			else:
78	tkoester	993	return float(value)
79	jonathan	887
80			def _float2string(self, value):
81			"""convert float value to string
82
83			(minus) infinity will be converted to (-)oo,
84			scientific notation will be used if necessary.
85
86			"""
87			if value == _inf:
88			return 'oo'
89			elif value == -_inf:
90			return '-oo'
91			else:
92	jonathan	962	return "%g" % (value,)
93	jonathan	887
94			def string(self, range):
95			"""convert internal representation to string"""
96			left, begin, end, right = range
97			return "%s%s;%s%s" % (left, self._float2string(begin),
98			self._float2string(end), right)
99
100			def __contains__(self, value):
101			if self._left == ']':
102			contains = value > self._begin
103			else:
104			contains = value >= self._begin
105			if self._right == '[':
106			contains = contains and (value < self._end)
107			else:
108			contains = contains and (value <= self._end)
109			return contains
110
111			def __eq__(self, other):
112			return (self.GetRange() == other.GetRange())
113
114			def __ne__(self, other):
115			return not self.__eq__(other)
116
117			def __str__(self):
118			return self.string(self.GetRange())
119
120
121			def _test():
122			range1 = Range(']0;99]')
123			print 'range1 =', range1, range1.GetRange()
124			for i in [-0.1, 0, 0.1, 9.9, 99, 99.9]:
125			print '%4.1f in range1 =' % i, i in range1
126			range2 = Range(']-oo;10[')
127			print 'range2 =', range2, range2.GetRange()
128			for i in [-0.1, 0, 0.1, 9.9, 10, 10.1]:
129			print '%4.1f not in range2 =' % i, i not in range2
130			range3 = Range(']1e-1;1E2]')
131			print 'range3 =', range3, range3.GetRange()
132			for i in [0, 0.1, 0.11, 10, 100, 101]:
133			print '%4.1f not in range3 =' % i, i not in range3
134			print 'range3 != range2 =', range3 != range2
135			print 'range3 != Range("]1e-1;1E2]") =', range3 != Range("]1e-1;1E2]")
136
137			range4 = Range('')
138			print 'range4 =', range4, range4.GetRange()
139
140			range5 = Range(']0;99E+00]')
141			print 'range5 =', range5, range5.GetRange()
142			range6 = Range(']0;99E+01]')
143			print 'range6 =', range6, range6.GetRange()
144			range7 = Range(']0;99E-01]')
145			print 'range7 =', range7, range7.GetRange()
146
147			if __name__ == "__main__":
148			_test()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision