1/*
2 * Copyright (C) 2007 The Guava Authors
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.google.common.collect;
18
19import static com.google.common.base.Preconditions.checkArgument;
20import static com.google.common.base.Preconditions.checkNotNull;
21
22import com.google.common.annotations.GwtCompatible;
23import com.google.common.base.Predicate;
24import com.google.common.base.Predicates;
25import com.google.common.collect.Collections2.FilteredCollection;
26
27import java.util.AbstractSet;
28import java.util.Arrays;
29import java.util.Collection;
30import java.util.Collections;
31import java.util.Comparator;
32import java.util.EnumSet;
33import java.util.HashSet;
34import java.util.Iterator;
35import java.util.LinkedHashSet;
36import java.util.List;
37import java.util.Map;
38import java.util.NoSuchElementException;
39import java.util.Set;
40import java.util.SortedSet;
41import java.util.TreeSet;
42import java.util.concurrent.ConcurrentHashMap;
43
44import javax.annotation.Nullable;
45
46/**
47 * Static utility methods pertaining to {@link Set} instances. Also see this
48 * class's counterparts {@link Lists}, {@link Maps} and {@link Queues}.
49 *
50 * <p>See the Guava User Guide article on <a href=
51 * "http://code.google.com/p/guava-libraries/wiki/CollectionUtilitiesExplained#Sets">
52 * {@code Sets}</a>.
53 *
54 * @author Kevin Bourrillion
55 * @author Jared Levy
56 * @author Chris Povirk
57 * @since 2.0 (imported from Google Collections Library)
58 */
59@GwtCompatible(emulated = true)
60public final class Sets {
61  private Sets() {}
62
63  /**
64   * {@link AbstractSet} substitute without the potentially-quadratic
65   * {@code removeAll} implementation.
66   */
67  abstract static class ImprovedAbstractSet<E> extends AbstractSet<E> {
68    @Override
69    public boolean removeAll(Collection<?> c) {
70      return removeAllImpl(this, c);
71    }
72
73    @Override
74    public boolean retainAll(Collection<?> c) {
75      return super.retainAll(checkNotNull(c)); // GWT compatibility
76    }
77  }
78
79  /**
80   * Returns an immutable set instance containing the given enum elements.
81   * Internally, the returned set will be backed by an {@link EnumSet}.
82   *
83   * <p>The iteration order of the returned set follows the enum's iteration
84   * order, not the order in which the elements are provided to the method.
85   *
86   * @param anElement one of the elements the set should contain
87   * @param otherElements the rest of the elements the set should contain
88   * @return an immutable set containing those elements, minus duplicates
89   */
90  // http://code.google.com/p/google-web-toolkit/issues/detail?id=3028
91  @GwtCompatible(serializable = true)
92  public static <E extends Enum<E>> ImmutableSet<E> immutableEnumSet(
93      E anElement, E... otherElements) {
94    return ImmutableEnumSet.asImmutable(EnumSet.of(anElement, otherElements));
95  }
96
97  /**
98   * Returns an immutable set instance containing the given enum elements.
99   * Internally, the returned set will be backed by an {@link EnumSet}.
100   *
101   * <p>The iteration order of the returned set follows the enum's iteration
102   * order, not the order in which the elements appear in the given collection.
103   *
104   * @param elements the elements, all of the same {@code enum} type, that the
105   *     set should contain
106   * @return an immutable set containing those elements, minus duplicates
107   */
108  // http://code.google.com/p/google-web-toolkit/issues/detail?id=3028
109  @GwtCompatible(serializable = true)
110  public static <E extends Enum<E>> ImmutableSet<E> immutableEnumSet(
111      Iterable<E> elements) {
112    if (elements instanceof ImmutableEnumSet) {
113      return (ImmutableEnumSet<E>) elements;
114    } else if (elements instanceof Collection) {
115      Collection<E> collection = (Collection<E>) elements;
116      if (collection.isEmpty()) {
117        return ImmutableSet.of();
118      } else {
119        return ImmutableEnumSet.asImmutable(EnumSet.copyOf(collection));
120      }
121    } else {
122      Iterator<E> itr = elements.iterator();
123      if (itr.hasNext()) {
124        EnumSet<E> enumSet = EnumSet.of(itr.next());
125        Iterators.addAll(enumSet, itr);
126        return ImmutableEnumSet.asImmutable(enumSet);
127      } else {
128        return ImmutableSet.of();
129      }
130    }
131  }
132
133  /**
134   * Returns a new {@code EnumSet} instance containing the given elements.
135   * Unlike {@link EnumSet#copyOf(Collection)}, this method does not produce an
136   * exception on an empty collection, and it may be called on any iterable, not
137   * just a {@code Collection}.
138   */
139  public static <E extends Enum<E>> EnumSet<E> newEnumSet(Iterable<E> iterable,
140      Class<E> elementType) {
141    EnumSet<E> set = EnumSet.noneOf(elementType);
142    Iterables.addAll(set, iterable);
143    return set;
144  }
145
146  // HashSet
147
148  /**
149   * Creates a <i>mutable</i>, empty {@code HashSet} instance.
150   *
151   * <p><b>Note:</b> if mutability is not required, use {@link
152   * ImmutableSet#of()} instead.
153   *
154   * <p><b>Note:</b> if {@code E} is an {@link Enum} type, use {@link
155   * EnumSet#noneOf} instead.
156   *
157   * @return a new, empty {@code HashSet}
158   */
159  public static <E> HashSet<E> newHashSet() {
160    return new HashSet<E>();
161  }
162
163  /**
164   * Creates a <i>mutable</i> {@code HashSet} instance containing the given
165   * elements in unspecified order.
166   *
167   * <p><b>Note:</b> if mutability is not required and the elements are
168   * non-null, use an overload of {@link ImmutableSet#of()} (for varargs) or
169   * {@link ImmutableSet#copyOf(Object[])} (for an array) instead.
170   *
171   * <p><b>Note:</b> if {@code E} is an {@link Enum} type, use {@link
172   * EnumSet#of(Enum, Enum[])} instead.
173   *
174   * @param elements the elements that the set should contain
175   * @return a new {@code HashSet} containing those elements (minus duplicates)
176   */
177  public static <E> HashSet<E> newHashSet(E... elements) {
178    HashSet<E> set = newHashSetWithExpectedSize(elements.length);
179    Collections.addAll(set, elements);
180    return set;
181  }
182
183  /**
184   * Creates a {@code HashSet} instance, with a high enough "initial capacity"
185   * that it <i>should</i> hold {@code expectedSize} elements without growth.
186   * This behavior cannot be broadly guaranteed, but it is observed to be true
187   * for OpenJDK 1.6. It also can't be guaranteed that the method isn't
188   * inadvertently <i>oversizing</i> the returned set.
189   *
190   * @param expectedSize the number of elements you expect to add to the
191   *        returned set
192   * @return a new, empty {@code HashSet} with enough capacity to hold {@code
193   *         expectedSize} elements without resizing
194   * @throws IllegalArgumentException if {@code expectedSize} is negative
195   */
196  public static <E> HashSet<E> newHashSetWithExpectedSize(int expectedSize) {
197    return new HashSet<E>(Maps.capacity(expectedSize));
198  }
199
200  /**
201   * Creates a <i>mutable</i> {@code HashSet} instance containing the given
202   * elements in unspecified order.
203   *
204   * <p><b>Note:</b> if mutability is not required and the elements are
205   * non-null, use {@link ImmutableSet#copyOf(Iterable)} instead.
206   *
207   * <p><b>Note:</b> if {@code E} is an {@link Enum} type, use
208   * {@link #newEnumSet(Iterable, Class)} instead.
209   *
210   * @param elements the elements that the set should contain
211   * @return a new {@code HashSet} containing those elements (minus duplicates)
212   */
213  public static <E> HashSet<E> newHashSet(Iterable<? extends E> elements) {
214    return (elements instanceof Collection)
215        ? new HashSet<E>(Collections2.cast(elements))
216        : newHashSet(elements.iterator());
217  }
218
219  /**
220   * Creates a <i>mutable</i> {@code HashSet} instance containing the given
221   * elements in unspecified order.
222   *
223   * <p><b>Note:</b> if mutability is not required and the elements are
224   * non-null, use {@link ImmutableSet#copyOf(Iterable)} instead.
225   *
226   * <p><b>Note:</b> if {@code E} is an {@link Enum} type, you should create an
227   * {@link EnumSet} instead.
228   *
229   * @param elements the elements that the set should contain
230   * @return a new {@code HashSet} containing those elements (minus duplicates)
231   */
232  public static <E> HashSet<E> newHashSet(Iterator<? extends E> elements) {
233    HashSet<E> set = newHashSet();
234    Iterators.addAll(set, elements);
235    return set;
236  }
237
238  /**
239   * Creates a thread-safe set backed by a hash map. The set is backed by a
240   * {@link ConcurrentHashMap} instance, and thus carries the same concurrency
241   * guarantees.
242   *
243   * <p>Unlike {@code HashSet}, this class does NOT allow {@code null} to be
244   * used as an element. The set is serializable.
245   *
246   * @return a new, empty thread-safe {@code Set}
247   * @since 15.0
248   */
249  public static <E> Set<E> newConcurrentHashSet() {
250    return newSetFromMap(new ConcurrentHashMap<E, Boolean>());
251  }
252
253  /**
254   * Creates a thread-safe set backed by a hash map and containing the given
255   * elements. The set is backed by a {@link ConcurrentHashMap} instance, and
256   * thus carries the same concurrency guarantees.
257   *
258   * <p>Unlike {@code HashSet}, this class does NOT allow {@code null} to be
259   * used as an element. The set is serializable.
260   *
261   * @param elements the elements that the set should contain
262   * @return a new thread-safe set containing those elements (minus duplicates)
263   * @throws NullPointerException if {@code elements} or any of its contents is
264   *      null
265   * @since 15.0
266   */
267  public static <E> Set<E> newConcurrentHashSet(
268      Iterable<? extends E> elements) {
269    Set<E> set = newConcurrentHashSet();
270    Iterables.addAll(set, elements);
271    return set;
272  }
273
274  // LinkedHashSet
275
276  /**
277   * Creates a <i>mutable</i>, empty {@code LinkedHashSet} instance.
278   *
279   * <p><b>Note:</b> if mutability is not required, use {@link
280   * ImmutableSet#of()} instead.
281   *
282   * @return a new, empty {@code LinkedHashSet}
283   */
284  public static <E> LinkedHashSet<E> newLinkedHashSet() {
285    return new LinkedHashSet<E>();
286  }
287
288  /**
289   * Creates a {@code LinkedHashSet} instance, with a high enough "initial
290   * capacity" that it <i>should</i> hold {@code expectedSize} elements without
291   * growth. This behavior cannot be broadly guaranteed, but it is observed to
292   * be true for OpenJDK 1.6. It also can't be guaranteed that the method isn't
293   * inadvertently <i>oversizing</i> the returned set.
294   *
295   * @param expectedSize the number of elements you expect to add to the
296   *        returned set
297   * @return a new, empty {@code LinkedHashSet} with enough capacity to hold
298   *         {@code expectedSize} elements without resizing
299   * @throws IllegalArgumentException if {@code expectedSize} is negative
300   * @since 11.0
301   */
302  public static <E> LinkedHashSet<E> newLinkedHashSetWithExpectedSize(
303      int expectedSize) {
304    return new LinkedHashSet<E>(Maps.capacity(expectedSize));
305  }
306
307  /**
308   * Creates a <i>mutable</i> {@code LinkedHashSet} instance containing the
309   * given elements in order.
310   *
311   * <p><b>Note:</b> if mutability is not required and the elements are
312   * non-null, use {@link ImmutableSet#copyOf(Iterable)} instead.
313   *
314   * @param elements the elements that the set should contain, in order
315   * @return a new {@code LinkedHashSet} containing those elements (minus
316   *     duplicates)
317   */
318  public static <E> LinkedHashSet<E> newLinkedHashSet(
319      Iterable<? extends E> elements) {
320    if (elements instanceof Collection) {
321      return new LinkedHashSet<E>(Collections2.cast(elements));
322    }
323    LinkedHashSet<E> set = newLinkedHashSet();
324    Iterables.addAll(set, elements);
325    return set;
326  }
327
328  // TreeSet
329
330  /**
331   * Creates a <i>mutable</i>, empty {@code TreeSet} instance sorted by the
332   * natural sort ordering of its elements.
333   *
334   * <p><b>Note:</b> if mutability is not required, use {@link
335   * ImmutableSortedSet#of()} instead.
336   *
337   * @return a new, empty {@code TreeSet}
338   */
339  public static <E extends Comparable> TreeSet<E> newTreeSet() {
340    return new TreeSet<E>();
341  }
342
343  /**
344   * Creates a <i>mutable</i> {@code TreeSet} instance containing the given
345   * elements sorted by their natural ordering.
346   *
347   * <p><b>Note:</b> if mutability is not required, use {@link
348   * ImmutableSortedSet#copyOf(Iterable)} instead.
349   *
350   * <p><b>Note:</b> If {@code elements} is a {@code SortedSet} with an explicit
351   * comparator, this method has different behavior than
352   * {@link TreeSet#TreeSet(SortedSet)}, which returns a {@code TreeSet} with
353   * that comparator.
354   *
355   * @param elements the elements that the set should contain
356   * @return a new {@code TreeSet} containing those elements (minus duplicates)
357   */
358  public static <E extends Comparable> TreeSet<E> newTreeSet(
359      Iterable<? extends E> elements) {
360    TreeSet<E> set = newTreeSet();
361    Iterables.addAll(set, elements);
362    return set;
363  }
364
365  /**
366   * Creates a <i>mutable</i>, empty {@code TreeSet} instance with the given
367   * comparator.
368   *
369   * <p><b>Note:</b> if mutability is not required, use {@code
370   * ImmutableSortedSet.orderedBy(comparator).build()} instead.
371   *
372   * @param comparator the comparator to use to sort the set
373   * @return a new, empty {@code TreeSet}
374   * @throws NullPointerException if {@code comparator} is null
375   */
376  public static <E> TreeSet<E> newTreeSet(Comparator<? super E> comparator) {
377    return new TreeSet<E>(checkNotNull(comparator));
378  }
379
380  /**
381   * Creates an empty {@code Set} that uses identity to determine equality. It
382   * compares object references, instead of calling {@code equals}, to
383   * determine whether a provided object matches an element in the set. For
384   * example, {@code contains} returns {@code false} when passed an object that
385   * equals a set member, but isn't the same instance. This behavior is similar
386   * to the way {@code IdentityHashMap} handles key lookups.
387   *
388   * @since 8.0
389   */
390  public static <E> Set<E> newIdentityHashSet() {
391    return Sets.newSetFromMap(Maps.<E, Boolean>newIdentityHashMap());
392  }
393
394  /**
395   * Creates an {@code EnumSet} consisting of all enum values that are not in
396   * the specified collection. If the collection is an {@link EnumSet}, this
397   * method has the same behavior as {@link EnumSet#complementOf}. Otherwise,
398   * the specified collection must contain at least one element, in order to
399   * determine the element type. If the collection could be empty, use
400   * {@link #complementOf(Collection, Class)} instead of this method.
401   *
402   * @param collection the collection whose complement should be stored in the
403   *     enum set
404   * @return a new, modifiable {@code EnumSet} containing all values of the enum
405   *     that aren't present in the given collection
406   * @throws IllegalArgumentException if {@code collection} is not an
407   *     {@code EnumSet} instance and contains no elements
408   */
409  public static <E extends Enum<E>> EnumSet<E> complementOf(
410      Collection<E> collection) {
411    if (collection instanceof EnumSet) {
412      return EnumSet.complementOf((EnumSet<E>) collection);
413    }
414    checkArgument(!collection.isEmpty(),
415        "collection is empty; use the other version of this method");
416    Class<E> type = collection.iterator().next().getDeclaringClass();
417    return makeComplementByHand(collection, type);
418  }
419
420  /**
421   * Creates an {@code EnumSet} consisting of all enum values that are not in
422   * the specified collection. This is equivalent to
423   * {@link EnumSet#complementOf}, but can act on any input collection, as long
424   * as the elements are of enum type.
425   *
426   * @param collection the collection whose complement should be stored in the
427   *     {@code EnumSet}
428   * @param type the type of the elements in the set
429   * @return a new, modifiable {@code EnumSet} initially containing all the
430   *     values of the enum not present in the given collection
431   */
432  public static <E extends Enum<E>> EnumSet<E> complementOf(
433      Collection<E> collection, Class<E> type) {
434    checkNotNull(collection);
435    return (collection instanceof EnumSet)
436        ? EnumSet.complementOf((EnumSet<E>) collection)
437        : makeComplementByHand(collection, type);
438  }
439
440  private static <E extends Enum<E>> EnumSet<E> makeComplementByHand(
441      Collection<E> collection, Class<E> type) {
442    EnumSet<E> result = EnumSet.allOf(type);
443    result.removeAll(collection);
444    return result;
445  }
446
447  /**
448   * Returns a set backed by the specified map. The resulting set displays
449   * the same ordering, concurrency, and performance characteristics as the
450   * backing map. In essence, this factory method provides a {@link Set}
451   * implementation corresponding to any {@link Map} implementation. There is no
452   * need to use this method on a {@link Map} implementation that already has a
453   * corresponding {@link Set} implementation (such as {@link java.util.HashMap}
454   * or {@link java.util.TreeMap}).
455   *
456   * <p>Each method invocation on the set returned by this method results in
457   * exactly one method invocation on the backing map or its {@code keySet}
458   * view, with one exception. The {@code addAll} method is implemented as a
459   * sequence of {@code put} invocations on the backing map.
460   *
461   * <p>The specified map must be empty at the time this method is invoked,
462   * and should not be accessed directly after this method returns. These
463   * conditions are ensured if the map is created empty, passed directly
464   * to this method, and no reference to the map is retained, as illustrated
465   * in the following code fragment: <pre>  {@code
466   *
467   *   Set<Object> identityHashSet = Sets.newSetFromMap(
468   *       new IdentityHashMap<Object, Boolean>());}</pre>
469   *
470   * <p>This method has the same behavior as the JDK 6 method
471   * {@code Collections.newSetFromMap()}. The returned set is serializable if
472   * the backing map is.
473   *
474   * @param map the backing map
475   * @return the set backed by the map
476   * @throws IllegalArgumentException if {@code map} is not empty
477   */
478  public static <E> Set<E> newSetFromMap(Map<E, Boolean> map) {
479    return Platform.newSetFromMap(map);
480  }
481
482  /**
483   * An unmodifiable view of a set which may be backed by other sets; this view
484   * will change as the backing sets do. Contains methods to copy the data into
485   * a new set which will then remain stable. There is usually no reason to
486   * retain a reference of type {@code SetView}; typically, you either use it
487   * as a plain {@link Set}, or immediately invoke {@link #immutableCopy} or
488   * {@link #copyInto} and forget the {@code SetView} itself.
489   *
490   * @since 2.0 (imported from Google Collections Library)
491   */
492  public abstract static class SetView<E> extends AbstractSet<E> {
493    private SetView() {} // no subclasses but our own
494
495    /**
496     * Returns an immutable copy of the current contents of this set view.
497     * Does not support null elements.
498     *
499     * <p><b>Warning:</b> this may have unexpected results if a backing set of
500     * this view uses a nonstandard notion of equivalence, for example if it is
501     * a {@link TreeSet} using a comparator that is inconsistent with {@link
502     * Object#equals(Object)}.
503     */
504    public ImmutableSet<E> immutableCopy() {
505      return ImmutableSet.copyOf(this);
506    }
507
508    /**
509     * Copies the current contents of this set view into an existing set. This
510     * method has equivalent behavior to {@code set.addAll(this)}, assuming that
511     * all the sets involved are based on the same notion of equivalence.
512     *
513     * @return a reference to {@code set}, for convenience
514     */
515    // Note: S should logically extend Set<? super E> but can't due to either
516    // some javac bug or some weirdness in the spec, not sure which.
517    public <S extends Set<E>> S copyInto(S set) {
518      set.addAll(this);
519      return set;
520    }
521  }
522
523  /**
524   * Returns an unmodifiable <b>view</b> of the union of two sets. The returned
525   * set contains all elements that are contained in either backing set.
526   * Iterating over the returned set iterates first over all the elements of
527   * {@code set1}, then over each element of {@code set2}, in order, that is not
528   * contained in {@code set1}.
529   *
530   * <p>Results are undefined if {@code set1} and {@code set2} are sets based on
531   * different equivalence relations (as {@link HashSet}, {@link TreeSet}, and
532   * the {@link Map#keySet} of an {@code IdentityHashMap} all are).
533   *
534   * <p><b>Note:</b> The returned view performs better when {@code set1} is the
535   * smaller of the two sets. If you have reason to believe one of your sets
536   * will generally be smaller than the other, pass it first.
537   *
538   * <p>Further, note that the current implementation is not suitable for nested
539   * {@code union} views, i.e. the following should be avoided when in a loop:
540   * {@code union = Sets.union(union, anotherSet);}, since iterating over the resulting
541   * set has a cubic complexity to the depth of the nesting.
542   */
543  public static <E> SetView<E> union(
544      final Set<? extends E> set1, final Set<? extends E> set2) {
545    checkNotNull(set1, "set1");
546    checkNotNull(set2, "set2");
547
548    final Set<? extends E> set2minus1 = difference(set2, set1);
549
550    return new SetView<E>() {
551      @Override public int size() {
552        return set1.size() + set2minus1.size();
553      }
554      @Override public boolean isEmpty() {
555        return set1.isEmpty() && set2.isEmpty();
556      }
557      @Override public Iterator<E> iterator() {
558        return Iterators.unmodifiableIterator(
559            Iterators.concat(set1.iterator(), set2minus1.iterator()));
560      }
561      @Override public boolean contains(Object object) {
562        return set1.contains(object) || set2.contains(object);
563      }
564      @Override public <S extends Set<E>> S copyInto(S set) {
565        set.addAll(set1);
566        set.addAll(set2);
567        return set;
568      }
569      @Override public ImmutableSet<E> immutableCopy() {
570        return new ImmutableSet.Builder<E>()
571            .addAll(set1).addAll(set2).build();
572      }
573    };
574  }
575
576  /**
577   * Returns an unmodifiable <b>view</b> of the intersection of two sets. The
578   * returned set contains all elements that are contained by both backing sets.
579   * The iteration order of the returned set matches that of {@code set1}.
580   *
581   * <p>Results are undefined if {@code set1} and {@code set2} are sets based
582   * on different equivalence relations (as {@code HashSet}, {@code TreeSet},
583   * and the keySet of an {@code IdentityHashMap} all are).
584   *
585   * <p><b>Note:</b> The returned view performs slightly better when {@code
586   * set1} is the smaller of the two sets. If you have reason to believe one of
587   * your sets will generally be smaller than the other, pass it first.
588   * Unfortunately, since this method sets the generic type of the returned set
589   * based on the type of the first set passed, this could in rare cases force
590   * you to make a cast, for example: <pre>   {@code
591   *
592   *   Set<Object> aFewBadObjects = ...
593   *   Set<String> manyBadStrings = ...
594   *
595   *   // impossible for a non-String to be in the intersection
596   *   SuppressWarnings("unchecked")
597   *   Set<String> badStrings = (Set) Sets.intersection(
598   *       aFewBadObjects, manyBadStrings);}</pre>
599   *
600   * <p>This is unfortunate, but should come up only very rarely.
601   */
602  public static <E> SetView<E> intersection(
603      final Set<E> set1, final Set<?> set2) {
604    checkNotNull(set1, "set1");
605    checkNotNull(set2, "set2");
606
607    final Predicate<Object> inSet2 = Predicates.in(set2);
608    return new SetView<E>() {
609      @Override public Iterator<E> iterator() {
610        return Iterators.filter(set1.iterator(), inSet2);
611      }
612      @Override public int size() {
613        return Iterators.size(iterator());
614      }
615      @Override public boolean isEmpty() {
616        return !iterator().hasNext();
617      }
618      @Override public boolean contains(Object object) {
619        return set1.contains(object) && set2.contains(object);
620      }
621      @Override public boolean containsAll(Collection<?> collection) {
622        return set1.containsAll(collection)
623            && set2.containsAll(collection);
624      }
625    };
626  }
627
628  /**
629   * Returns an unmodifiable <b>view</b> of the difference of two sets. The
630   * returned set contains all elements that are contained by {@code set1} and
631   * not contained by {@code set2}. {@code set2} may also contain elements not
632   * present in {@code set1}; these are simply ignored. The iteration order of
633   * the returned set matches that of {@code set1}.
634   *
635   * <p>Results are undefined if {@code set1} and {@code set2} are sets based
636   * on different equivalence relations (as {@code HashSet}, {@code TreeSet},
637   * and the keySet of an {@code IdentityHashMap} all are).
638   */
639  public static <E> SetView<E> difference(
640      final Set<E> set1, final Set<?> set2) {
641    checkNotNull(set1, "set1");
642    checkNotNull(set2, "set2");
643
644    final Predicate<Object> notInSet2 = Predicates.not(Predicates.in(set2));
645    return new SetView<E>() {
646      @Override public Iterator<E> iterator() {
647        return Iterators.filter(set1.iterator(), notInSet2);
648      }
649      @Override public int size() {
650        return Iterators.size(iterator());
651      }
652      @Override public boolean isEmpty() {
653        return set2.containsAll(set1);
654      }
655      @Override public boolean contains(Object element) {
656        return set1.contains(element) && !set2.contains(element);
657      }
658    };
659  }
660
661  /**
662   * Returns an unmodifiable <b>view</b> of the symmetric difference of two
663   * sets. The returned set contains all elements that are contained in either
664   * {@code set1} or {@code set2} but not in both. The iteration order of the
665   * returned set is undefined.
666   *
667   * <p>Results are undefined if {@code set1} and {@code set2} are sets based
668   * on different equivalence relations (as {@code HashSet}, {@code TreeSet},
669   * and the keySet of an {@code IdentityHashMap} all are).
670   *
671   * @since 3.0
672   */
673  public static <E> SetView<E> symmetricDifference(
674      Set<? extends E> set1, Set<? extends E> set2) {
675    checkNotNull(set1, "set1");
676    checkNotNull(set2, "set2");
677
678    // TODO(kevinb): Replace this with a more efficient implementation
679    return difference(union(set1, set2), intersection(set1, set2));
680  }
681
682  /**
683   * Returns the elements of {@code unfiltered} that satisfy a predicate. The
684   * returned set is a live view of {@code unfiltered}; changes to one affect
685   * the other.
686   *
687   * <p>The resulting set's iterator does not support {@code remove()}, but all
688   * other set methods are supported. When given an element that doesn't satisfy
689   * the predicate, the set's {@code add()} and {@code addAll()} methods throw
690   * an {@link IllegalArgumentException}. When methods such as {@code
691   * removeAll()} and {@code clear()} are called on the filtered set, only
692   * elements that satisfy the filter will be removed from the underlying set.
693   *
694   * <p>The returned set isn't threadsafe or serializable, even if
695   * {@code unfiltered} is.
696   *
697   * <p>Many of the filtered set's methods, such as {@code size()}, iterate
698   * across every element in the underlying set and determine which elements
699   * satisfy the filter. When a live view is <i>not</i> needed, it may be faster
700   * to copy {@code Iterables.filter(unfiltered, predicate)} and use the copy.
701   *
702   * <p><b>Warning:</b> {@code predicate} must be <i>consistent with equals</i>,
703   * as documented at {@link Predicate#apply}. Do not provide a predicate such
704   * as {@code Predicates.instanceOf(ArrayList.class)}, which is inconsistent
705   * with equals. (See {@link Iterables#filter(Iterable, Class)} for related
706   * functionality.)
707   */
708  // TODO(kevinb): how to omit that last sentence when building GWT javadoc?
709  public static <E> Set<E> filter(
710      Set<E> unfiltered, Predicate<? super E> predicate) {
711    if (unfiltered instanceof SortedSet) {
712      return filter((SortedSet<E>) unfiltered, predicate);
713    }
714    if (unfiltered instanceof FilteredSet) {
715      // Support clear(), removeAll(), and retainAll() when filtering a filtered
716      // collection.
717      FilteredSet<E> filtered = (FilteredSet<E>) unfiltered;
718      Predicate<E> combinedPredicate
719          = Predicates.<E>and(filtered.predicate, predicate);
720      return new FilteredSet<E>(
721          (Set<E>) filtered.unfiltered, combinedPredicate);
722    }
723
724    return new FilteredSet<E>(
725        checkNotNull(unfiltered), checkNotNull(predicate));
726  }
727
728  private static class FilteredSet<E> extends FilteredCollection<E>
729      implements Set<E> {
730    FilteredSet(Set<E> unfiltered, Predicate<? super E> predicate) {
731      super(unfiltered, predicate);
732    }
733
734    @Override public boolean equals(@Nullable Object object) {
735      return equalsImpl(this, object);
736    }
737
738    @Override public int hashCode() {
739      return hashCodeImpl(this);
740    }
741  }
742
743  /**
744   * Returns the elements of a {@code SortedSet}, {@code unfiltered}, that
745   * satisfy a predicate. The returned set is a live view of {@code unfiltered};
746   * changes to one affect the other.
747   *
748   * <p>The resulting set's iterator does not support {@code remove()}, but all
749   * other set methods are supported. When given an element that doesn't satisfy
750   * the predicate, the set's {@code add()} and {@code addAll()} methods throw
751   * an {@link IllegalArgumentException}. When methods such as
752   * {@code removeAll()} and {@code clear()} are called on the filtered set,
753   * only elements that satisfy the filter will be removed from the underlying
754   * set.
755   *
756   * <p>The returned set isn't threadsafe or serializable, even if
757   * {@code unfiltered} is.
758   *
759   * <p>Many of the filtered set's methods, such as {@code size()}, iterate across
760   * every element in the underlying set and determine which elements satisfy
761   * the filter. When a live view is <i>not</i> needed, it may be faster to copy
762   * {@code Iterables.filter(unfiltered, predicate)} and use the copy.
763   *
764   * <p><b>Warning:</b> {@code predicate} must be <i>consistent with equals</i>,
765   * as documented at {@link Predicate#apply}. Do not provide a predicate such as
766   * {@code Predicates.instanceOf(ArrayList.class)}, which is inconsistent with
767   * equals. (See {@link Iterables#filter(Iterable, Class)} for related
768   * functionality.)
769   *
770   * @since 11.0
771   */
772  public static <E> SortedSet<E> filter(
773      SortedSet<E> unfiltered, Predicate<? super E> predicate) {
774    return Platform.setsFilterSortedSet(unfiltered, predicate);
775  }
776
777  static <E> SortedSet<E> filterSortedIgnoreNavigable(
778      SortedSet<E> unfiltered, Predicate<? super E> predicate) {
779    if (unfiltered instanceof FilteredSet) {
780      // Support clear(), removeAll(), and retainAll() when filtering a filtered
781      // collection.
782      FilteredSet<E> filtered = (FilteredSet<E>) unfiltered;
783      Predicate<E> combinedPredicate
784          = Predicates.<E>and(filtered.predicate, predicate);
785      return new FilteredSortedSet<E>(
786          (SortedSet<E>) filtered.unfiltered, combinedPredicate);
787    }
788
789    return new FilteredSortedSet<E>(
790        checkNotNull(unfiltered), checkNotNull(predicate));
791  }
792
793  private static class FilteredSortedSet<E> extends FilteredSet<E>
794      implements SortedSet<E> {
795
796    FilteredSortedSet(SortedSet<E> unfiltered, Predicate<? super E> predicate) {
797      super(unfiltered, predicate);
798    }
799
800    @Override
801    public Comparator<? super E> comparator() {
802      return ((SortedSet<E>) unfiltered).comparator();
803    }
804
805    @Override
806    public SortedSet<E> subSet(E fromElement, E toElement) {
807      return new FilteredSortedSet<E>(((SortedSet<E>) unfiltered).subSet(fromElement, toElement),
808          predicate);
809    }
810
811    @Override
812    public SortedSet<E> headSet(E toElement) {
813      return new FilteredSortedSet<E>(((SortedSet<E>) unfiltered).headSet(toElement), predicate);
814    }
815
816    @Override
817    public SortedSet<E> tailSet(E fromElement) {
818      return new FilteredSortedSet<E>(((SortedSet<E>) unfiltered).tailSet(fromElement), predicate);
819    }
820
821    @Override
822    public E first() {
823      return iterator().next();
824    }
825
826    @Override
827    public E last() {
828      SortedSet<E> sortedUnfiltered = (SortedSet<E>) unfiltered;
829      while (true) {
830        E element = sortedUnfiltered.last();
831        if (predicate.apply(element)) {
832          return element;
833        }
834        sortedUnfiltered = sortedUnfiltered.headSet(element);
835      }
836    }
837  }
838
839  /**
840   * Returns every possible list that can be formed by choosing one element
841   * from each of the given sets in order; the "n-ary
842   * <a href="http://en.wikipedia.org/wiki/Cartesian_product">Cartesian
843   * product</a>" of the sets. For example: <pre>   {@code
844   *
845   *   Sets.cartesianProduct(ImmutableList.of(
846   *       ImmutableSet.of(1, 2),
847   *       ImmutableSet.of("A", "B", "C")))}</pre>
848   *
849   * <p>returns a set containing six lists:
850   *
851   * <ul>
852   * <li>{@code ImmutableList.of(1, "A")}
853   * <li>{@code ImmutableList.of(1, "B")}
854   * <li>{@code ImmutableList.of(1, "C")}
855   * <li>{@code ImmutableList.of(2, "A")}
856   * <li>{@code ImmutableList.of(2, "B")}
857   * <li>{@code ImmutableList.of(2, "C")}
858   * </ul>
859   *
860   * <p>The result is guaranteed to be in the "traditional", lexicographical
861   * order for Cartesian products that you would get from nesting for loops:
862   * <pre>   {@code
863   *
864   *   for (B b0 : sets.get(0)) {
865   *     for (B b1 : sets.get(1)) {
866   *       ...
867   *       ImmutableList<B> tuple = ImmutableList.of(b0, b1, ...);
868   *       // operate on tuple
869   *     }
870   *   }}</pre>
871   *
872   * <p>Note that if any input set is empty, the Cartesian product will also be
873   * empty. If no sets at all are provided (an empty list), the resulting
874   * Cartesian product has one element, an empty list (counter-intuitive, but
875   * mathematically consistent).
876   *
877   * <p><i>Performance notes:</i> while the cartesian product of sets of size
878   * {@code m, n, p} is a set of size {@code m x n x p}, its actual memory
879   * consumption is much smaller. When the cartesian set is constructed, the
880   * input sets are merely copied. Only as the resulting set is iterated are the
881   * individual lists created, and these are not retained after iteration.
882   *
883   * @param sets the sets to choose elements from, in the order that
884   *     the elements chosen from those sets should appear in the resulting
885   *     lists
886   * @param <B> any common base class shared by all axes (often just {@link
887   *     Object})
888   * @return the Cartesian product, as an immutable set containing immutable
889   *     lists
890   * @throws NullPointerException if {@code sets}, any one of the {@code sets},
891   *     or any element of a provided set is null
892   * @since 2.0
893   */
894  public static <B> Set<List<B>> cartesianProduct(
895      List<? extends Set<? extends B>> sets) {
896    return CartesianSet.create(sets);
897  }
898
899  /**
900   * Returns every possible list that can be formed by choosing one element
901   * from each of the given sets in order; the "n-ary
902   * <a href="http://en.wikipedia.org/wiki/Cartesian_product">Cartesian
903   * product</a>" of the sets. For example: <pre>   {@code
904   *
905   *   Sets.cartesianProduct(
906   *       ImmutableSet.of(1, 2),
907   *       ImmutableSet.of("A", "B", "C"))}</pre>
908   *
909   * <p>returns a set containing six lists:
910   *
911   * <ul>
912   * <li>{@code ImmutableList.of(1, "A")}
913   * <li>{@code ImmutableList.of(1, "B")}
914   * <li>{@code ImmutableList.of(1, "C")}
915   * <li>{@code ImmutableList.of(2, "A")}
916   * <li>{@code ImmutableList.of(2, "B")}
917   * <li>{@code ImmutableList.of(2, "C")}
918   * </ul>
919   *
920   * <p>The result is guaranteed to be in the "traditional", lexicographical
921   * order for Cartesian products that you would get from nesting for loops:
922   * <pre>   {@code
923   *
924   *   for (B b0 : sets.get(0)) {
925   *     for (B b1 : sets.get(1)) {
926   *       ...
927   *       ImmutableList<B> tuple = ImmutableList.of(b0, b1, ...);
928   *       // operate on tuple
929   *     }
930   *   }}</pre>
931   *
932   * <p>Note that if any input set is empty, the Cartesian product will also be
933   * empty. If no sets at all are provided (an empty list), the resulting
934   * Cartesian product has one element, an empty list (counter-intuitive, but
935   * mathematically consistent).
936   *
937   * <p><i>Performance notes:</i> while the cartesian product of sets of size
938   * {@code m, n, p} is a set of size {@code m x n x p}, its actual memory
939   * consumption is much smaller. When the cartesian set is constructed, the
940   * input sets are merely copied. Only as the resulting set is iterated are the
941   * individual lists created, and these are not retained after iteration.
942   *
943   * @param sets the sets to choose elements from, in the order that
944   *     the elements chosen from those sets should appear in the resulting
945   *     lists
946   * @param <B> any common base class shared by all axes (often just {@link
947   *     Object})
948   * @return the Cartesian product, as an immutable set containing immutable
949   *     lists
950   * @throws NullPointerException if {@code sets}, any one of the {@code sets},
951   *     or any element of a provided set is null
952   * @since 2.0
953   */
954  public static <B> Set<List<B>> cartesianProduct(
955      Set<? extends B>... sets) {
956    return cartesianProduct(Arrays.asList(sets));
957  }
958
959  private static final class CartesianSet<E>
960      extends ForwardingCollection<List<E>> implements Set<List<E>> {
961    private transient final ImmutableList<ImmutableSet<E>> axes;
962    private transient final CartesianList<E> delegate;
963
964    static <E> Set<List<E>> create(List<? extends Set<? extends E>> sets) {
965      ImmutableList.Builder<ImmutableSet<E>> axesBuilder =
966          new ImmutableList.Builder<ImmutableSet<E>>(sets.size());
967      for (Set<? extends E> set : sets) {
968        ImmutableSet<E> copy = ImmutableSet.copyOf(set);
969        if (copy.isEmpty()) {
970          return ImmutableSet.of();
971        }
972        axesBuilder.add(copy);
973      }
974      final ImmutableList<ImmutableSet<E>> axes = axesBuilder.build();
975      ImmutableList<List<E>> listAxes = new ImmutableList<List<E>>() {
976
977        @Override
978        public int size() {
979          return axes.size();
980        }
981
982        @Override
983        public List<E> get(int index) {
984          return axes.get(index).asList();
985        }
986
987        @Override
988        boolean isPartialView() {
989          return true;
990        }
991      };
992      return new CartesianSet<E>(axes, new CartesianList<E>(listAxes));
993    }
994
995    private CartesianSet(
996        ImmutableList<ImmutableSet<E>> axes, CartesianList<E> delegate) {
997      this.axes = axes;
998      this.delegate = delegate;
999    }
1000
1001    @Override
1002    protected Collection<List<E>> delegate() {
1003      return delegate;
1004    }
1005
1006    @Override public boolean equals(@Nullable Object object) {
1007      // Warning: this is broken if size() == 0, so it is critical that we
1008      // substitute an empty ImmutableSet to the user in place of this
1009      if (object instanceof CartesianSet) {
1010        CartesianSet<?> that = (CartesianSet<?>) object;
1011        return this.axes.equals(that.axes);
1012      }
1013      return super.equals(object);
1014    }
1015
1016    @Override
1017    public int hashCode() {
1018      // Warning: this is broken if size() == 0, so it is critical that we
1019      // substitute an empty ImmutableSet to the user in place of this
1020
1021      // It's a weird formula, but tests prove it works.
1022      int adjust = size() - 1;
1023      for (int i = 0; i < axes.size(); i++) {
1024        adjust *= 31;
1025        adjust = ~~adjust;
1026        // in GWT, we have to deal with integer overflow carefully
1027      }
1028      int hash = 1;
1029      for (Set<E> axis : axes) {
1030        hash = 31 * hash + (size() / axis.size() * axis.hashCode());
1031
1032        hash = ~~hash;
1033      }
1034      hash += adjust;
1035      return ~~hash;
1036    }
1037  }
1038
1039  /**
1040   * Returns the set of all possible subsets of {@code set}. For example,
1041   * {@code powerSet(ImmutableSet.of(1, 2))} returns the set {@code {{},
1042   * {1}, {2}, {1, 2}}}.
1043   *
1044   * <p>Elements appear in these subsets in the same iteration order as they
1045   * appeared in the input set. The order in which these subsets appear in the
1046   * outer set is undefined. Note that the power set of the empty set is not the
1047   * empty set, but a one-element set containing the empty set.
1048   *
1049   * <p>The returned set and its constituent sets use {@code equals} to decide
1050   * whether two elements are identical, even if the input set uses a different
1051   * concept of equivalence.
1052   *
1053   * <p><i>Performance notes:</i> while the power set of a set with size {@code
1054   * n} is of size {@code 2^n}, its memory usage is only {@code O(n)}. When the
1055   * power set is constructed, the input set is merely copied. Only as the
1056   * power set is iterated are the individual subsets created, and these subsets
1057   * themselves occupy only a small constant amount of memory.
1058   *
1059   * @param set the set of elements to construct a power set from
1060   * @return the power set, as an immutable set of immutable sets
1061   * @throws IllegalArgumentException if {@code set} has more than 30 unique
1062   *     elements (causing the power set size to exceed the {@code int} range)
1063   * @throws NullPointerException if {@code set} is or contains {@code null}
1064   * @see <a href="http://en.wikipedia.org/wiki/Power_set">Power set article at
1065   *      Wikipedia</a>
1066   * @since 4.0
1067   */
1068  @GwtCompatible(serializable = false)
1069  public static <E> Set<Set<E>> powerSet(Set<E> set) {
1070    return new PowerSet<E>(set);
1071  }
1072
1073  private static final class SubSet<E> extends AbstractSet<E> {
1074    private final ImmutableMap<E, Integer> inputSet;
1075    private final int mask;
1076
1077    SubSet(ImmutableMap<E, Integer> inputSet, int mask) {
1078      this.inputSet = inputSet;
1079      this.mask = mask;
1080    }
1081
1082    @Override
1083    public Iterator<E> iterator() {
1084      return new UnmodifiableIterator<E>() {
1085        final ImmutableList<E> elements = inputSet.keySet().asList();
1086        int remainingSetBits = mask;
1087
1088        @Override
1089        public boolean hasNext() {
1090          return remainingSetBits != 0;
1091        }
1092
1093        @Override
1094        public E next() {
1095          int index = Integer.numberOfTrailingZeros(remainingSetBits);
1096          if (index == 32) {
1097            throw new NoSuchElementException();
1098          }
1099          remainingSetBits &= ~(1 << index);
1100          return elements.get(index);
1101        }
1102      };
1103    }
1104
1105    @Override
1106    public int size() {
1107      return Integer.bitCount(mask);
1108    }
1109
1110    @Override
1111    public boolean contains(@Nullable Object o) {
1112      Integer index = inputSet.get(o);
1113      return index != null && (mask & (1 << index)) != 0;
1114    }
1115  }
1116
1117  private static final class PowerSet<E> extends AbstractSet<Set<E>> {
1118    final ImmutableMap<E, Integer> inputSet;
1119
1120    PowerSet(Set<E> input) {
1121      ImmutableMap.Builder<E, Integer> builder = ImmutableMap.builder();
1122      int i = 0;
1123      for (E e : checkNotNull(input)) {
1124        builder.put(e, i++);
1125      }
1126      this.inputSet = builder.build();
1127      checkArgument(inputSet.size() <= 30,
1128          "Too many elements to create power set: %s > 30", inputSet.size());
1129    }
1130
1131    @Override public int size() {
1132      return 1 << inputSet.size();
1133    }
1134
1135    @Override public boolean isEmpty() {
1136      return false;
1137    }
1138
1139    @Override public Iterator<Set<E>> iterator() {
1140      return new AbstractIndexedListIterator<Set<E>>(size()) {
1141        @Override protected Set<E> get(final int setBits) {
1142          return new SubSet<E>(inputSet, setBits);
1143        }
1144      };
1145    }
1146
1147    @Override public boolean contains(@Nullable Object obj) {
1148      if (obj instanceof Set) {
1149        Set<?> set = (Set<?>) obj;
1150        return inputSet.keySet().containsAll(set);
1151      }
1152      return false;
1153    }
1154
1155    @Override public boolean equals(@Nullable Object obj) {
1156      if (obj instanceof PowerSet) {
1157        PowerSet<?> that = (PowerSet<?>) obj;
1158        return inputSet.equals(that.inputSet);
1159      }
1160      return super.equals(obj);
1161    }
1162
1163    @Override public int hashCode() {
1164      /*
1165       * The sum of the sums of the hash codes in each subset is just the sum of
1166       * each input element's hash code times the number of sets that element
1167       * appears in. Each element appears in exactly half of the 2^n sets, so:
1168       */
1169      return inputSet.keySet().hashCode() << (inputSet.size() - 1);
1170    }
1171
1172    @Override public String toString() {
1173      return "powerSet(" + inputSet + ")";
1174    }
1175  }
1176
1177  /**
1178   * An implementation for {@link Set#hashCode()}.
1179   */
1180  static int hashCodeImpl(Set<?> s) {
1181    int hashCode = 0;
1182    for (Object o : s) {
1183      hashCode += o != null ? o.hashCode() : 0;
1184
1185      hashCode = ~~hashCode;
1186      // Needed to deal with unusual integer overflow in GWT.
1187    }
1188    return hashCode;
1189  }
1190
1191  /**
1192   * An implementation for {@link Set#equals(Object)}.
1193   */
1194  static boolean equalsImpl(Set<?> s, @Nullable Object object) {
1195    if (s == object) {
1196      return true;
1197    }
1198    if (object instanceof Set) {
1199      Set<?> o = (Set<?>) object;
1200
1201      try {
1202        return s.size() == o.size() && s.containsAll(o);
1203      } catch (NullPointerException ignored) {
1204        return false;
1205      } catch (ClassCastException ignored) {
1206        return false;
1207      }
1208    }
1209    return false;
1210  }
1211
1212  /**
1213   * Remove each element in an iterable from a set.
1214   */
1215  static boolean removeAllImpl(Set<?> set, Iterator<?> iterator) {
1216    boolean changed = false;
1217    while (iterator.hasNext()) {
1218      changed |= set.remove(iterator.next());
1219    }
1220    return changed;
1221  }
1222
1223  static boolean removeAllImpl(Set<?> set, Collection<?> collection) {
1224    checkNotNull(collection); // for GWT
1225    if (collection instanceof Multiset) {
1226      collection = ((Multiset<?>) collection).elementSet();
1227    }
1228    /*
1229     * AbstractSet.removeAll(List) has quadratic behavior if the list size
1230     * is just less than the set's size.  We augment the test by
1231     * assuming that sets have fast contains() performance, and other
1232     * collections don't.  See
1233     * http://code.google.com/p/guava-libraries/issues/detail?id=1013
1234     */
1235    if (collection instanceof Set && collection.size() > set.size()) {
1236      return Iterators.removeAll(set.iterator(), collection);
1237    } else {
1238      return removeAllImpl(set, collection.iterator());
1239    }
1240  }
1241}
1242