src/reportlab/lib/set_ops.py
author robin
Thu, 27 Sep 2012 14:39:39 +0000
changeset 3617 ae5744e97c42
parent 3029 eded59f94021
child 3721 0c93dd8ff567
permissions -rw-r--r--
reportlab: copyright date changes

#!/bin/env python
#Copyright ReportLab Europe Ltd. 2000-2012
#see license.txt for license details
#history http://www.reportlab.co.uk/cgi-bin/viewcvs.cgi/public/reportlab/trunk/reportlab/lib/set_ops.py
__version__=''' $Id$ '''
__doc__="""From before Python had a Set class..."""

import types
import string

def __set_coerce(t, S):
    if t is types.ListType:
        return list(S)
    elif t is types.TupleType:
        return tuple(S)
    elif t is types.StringType:
        return string.join(S, '')
    return S

def unique(seq):
    result = []
    for i in seq:
        if i not in result:
            result.append(i)
    return __set_coerce(type(seq), result)

def intersect(seq1, seq2):
    result = []
    if type(seq1) != type(seq2) and type(seq2) == types.StringType: seq2 = list(seq2)
    for i in seq1:
        if i in seq2 and i not in result: result.append(i)
    return __set_coerce(type(seq1), result)

def union(seq1, seq2):
    if type(seq1) == type(seq2):
        return unique(seq1 + seq2)
    if type(seq1) == types.ListType or type(seq2) == types.ListType:
        return unique(list(seq1) + list(seq2))
    if type(seq1) == types.TupleType or type(seq2) == types.TupleType:
        return unique(tuple(seq1) + tuple(seq2))
    return unique(list(seq1) + list(seq2))