Demo/classes/Range.py

# Example of a generator: re-implement the built-in range function
# without actually constructing the list of values.  (It turns out
# that the built-in function is about 20 times faster -- that's why
# it's built-in. :-)


# Wrapper function to emulate the complicated range() arguments

def range(*a):
    if len(a) == 1:
        start, stop, step = 0, a[0], 1
    elif len(a) == 2:
        start, stop = a
        step = 1
    elif len(a) == 3:
        start, stop, step = a
    else:
        raise TypeError, 'range() needs 1-3 arguments'
    return Range(start, stop, step)


# Class implementing a range object.
# To the user the instances feel like immutable sequences
# (and you can't concatenate or slice them)

class Range:

    # initialization -- should be called only by range() above
    def __init__(self, start, stop, step):
        if step == 0:
            raise ValueError, 'range() called with zero step'
        self.start = start
        self.stop = stop
        self.step = step
        self.len = max(0, int((self.stop - self.start) / self.step))

    # implement `x` and is also used by print x
    def __repr__(self):
        return 'range' + `self.start, self.stop, self.step`

    # implement len(x)
    def __len__(self):
        return self.len

    # implement x[i]
    def __getitem__(self, i):
        if 0 <= i < self.len:
            return self.start + self.step * i
        else:
            raise IndexError, 'range[i] index out of range'


# Small test program

def test():
    import time, __builtin__
    print range(10), range(-10, 10), range(0, 10, 2)
    for i in range(100, -100, -10): print i,
    print
    t1 = time.time()
    for i in range(1000):
        pass
    t2 = time.time()
    for i in __builtin__.range(1000):
        pass
    t3 = time.time()
    print t2-t1, 'sec (class)'
    print t3-t2, 'sec (built-in)'


test()