org.javafunk.funk.Sets.java Source code

Introduction

Here is the source code for org.javafunk.funk.Sets.java

Source

/*
 * Copyright (C) 2011 Funk committers.
 * All rights reserved.
 *
 * The software in this package is published under the terms of the BSD
 * style license a copy of which has been included with this distribution in
 * the LICENSE.txt file.
 */
package org.javafunk.funk;

import com.google.common.collect.Multiset;
import org.javafunk.funk.functors.Predicate;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

import static java.util.Arrays.asList;
import static org.javafunk.funk.Eagerly.first;
import static org.javafunk.funk.Eagerly.rest;
import static org.javafunk.funk.Iterables.asSet;
import static org.javafunk.funk.Lazily.filter;
import static org.javafunk.funk.Literals.*;

public class Sets {
    private Sets() {
    }

    public static <T> Set<T> union(Iterable<? extends Iterable<? extends T>> iterables) {
        Set<T> unionSet = new HashSet<T>();
        for (Iterable<? extends T> iterable : iterables) {
            unionSet.addAll(collectionFrom(iterable));
        }
        return unionSet;
    }

    public static <T> Set<T> intersection(Iterable<? extends Iterable<? extends T>> iterables) {
        Set<T> intersectionSet = new HashSet<T>(collectionFrom(first(iterables).get()));
        for (Iterable<? extends T> iterable : rest(iterables)) {
            intersectionSet.retainAll(collectionFrom(iterable));
        }
        return intersectionSet;
    }

    public static <T> Set<T> difference(Iterable<? extends Iterable<? extends T>> iterables) {
        List<Iterable<? extends T>> arguments = listFrom(iterables);
        if (arguments.isEmpty()) {
            return new HashSet<T>();
        } else {
            Set<T> differenceSet = asSet(first(arguments).get());
            differenceSet.removeAll(union(rest(arguments)));
            return differenceSet;
        }
    }

    public static <T> Set<T> symmetricDifference(Iterable<? extends Iterable<? extends T>> iterables) {
        final Multiset<T> unionMultiset = Multisets.concatenate(iterables);
        return asSet(filter(unionMultiset, new Predicate<T>() {
            public boolean evaluate(T element) {
                return isOdd(unionMultiset.count(element));
            }

            private boolean isOdd(Integer value) {
                return value % 2 == 1;
            }
        }));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> union(Iterable<? extends T> i1, Iterable<? extends T> i2) {
        return union(asList(i1, i2));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> union(Iterable<? extends T> i1, Iterable<? extends T> i2, Iterable<? extends T> i3) {
        return union(asList(i1, i2, i3));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> union(Iterable<? extends T> i1, Iterable<? extends T> i2, Iterable<? extends T> i3,
            Iterable<? extends T> i4) {
        return union(asList(i1, i2, i3, i4));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> union(Iterable<? extends T> i1, Iterable<? extends T> i2, Iterable<? extends T> i3,
            Iterable<? extends T> i4, Iterable<? extends T> i5) {
        return union(asList(i1, i2, i3, i4, i5));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> union(Iterable<? extends T> i1, Iterable<? extends T> i2, Iterable<? extends T> i3,
            Iterable<? extends T> i4, Iterable<? extends T> i5, Iterable<? extends T> i6) {
        return union(asList(i1, i2, i3, i4, i5, i6));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> union(Iterable<? extends T> i1, Iterable<? extends T> i2, Iterable<? extends T> i3,
            Iterable<? extends T> i4, Iterable<? extends T> i5, Iterable<? extends T> i6,
            Iterable<? extends T>... i7on) {
        return union(iterableBuilderWith(i1, i2, i3, i4, i5, i6).and(asList(i7on)).build());
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> intersection(Iterable<? extends T> i1, Iterable<? extends T> i2) {
        return intersection(asList(i1, i2));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> intersection(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3) {
        return intersection(asList(i1, i2, i3));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> intersection(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4) {
        return intersection(asList(i1, i2, i3, i4));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> intersection(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5) {
        return intersection(asList(i1, i2, i3, i4, i5));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> intersection(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5,
            Iterable<? extends T> i6) {
        return intersection(asList(i1, i2, i3, i4, i5, i6));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> intersection(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5, Iterable<? extends T> i6,
            Iterable<? extends T>... i7on) {
        return intersection(iterableBuilderWith(i1, i2, i3, i4, i5, i6).and(asList(i7on)).build());
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> difference(Iterable<? extends T> i1, Iterable<? extends T> i2) {
        return difference(asList(i1, i2));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> difference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3) {
        return difference(asList(i1, i2, i3));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> difference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4) {
        return difference(asList(i1, i2, i3, i4));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> difference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5) {
        return difference(asList(i1, i2, i3, i4, i5));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> difference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5,
            Iterable<? extends T> i6) {
        return difference(asList(i1, i2, i3, i4, i5, i6));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> difference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5, Iterable<? extends T> i6,
            Iterable<? extends T>... i7on) {
        return difference(iterableBuilderWith(i1, i2, i3, i4, i5, i6).and(asList(i7on)).build());
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> symmetricDifference(Iterable<? extends T> i1, Iterable<? extends T> i2) {
        return symmetricDifference(asList(i1, i2));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> symmetricDifference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3) {
        return symmetricDifference(asList(i1, i2, i3));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> symmetricDifference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4) {
        return symmetricDifference(asList(i1, i2, i3, i4));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> symmetricDifference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5) {
        return symmetricDifference(asList(i1, i2, i3, i4, i5));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> symmetricDifference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5,
            Iterable<? extends T> i6) {
        return symmetricDifference(asList(i1, i2, i3, i4, i5, i6));
    }

    @SuppressWarnings("unchecked")
    public static <T> Set<T> symmetricDifference(Iterable<? extends T> i1, Iterable<? extends T> i2,
            Iterable<? extends T> i3, Iterable<? extends T> i4, Iterable<? extends T> i5, Iterable<? extends T> i6,
            Iterable<? extends T>... i7on) {
        return symmetricDifference(iterableBuilderWith(i1, i2, i3, i4, i5, i6).and(asList(i7on)).build());
    }
}