001    /*
002     * Copyright (C) 2007 The Guava Authors
003     *
004     * Licensed under the Apache License, Version 2.0 (the "License");
005     * you may not use this file except in compliance with the License.
006     * You may obtain a copy of the License at
007     *
008     * http://www.apache.org/licenses/LICENSE-2.0
009     *
010     * Unless required by applicable law or agreed to in writing, software
011     * distributed under the License is distributed on an "AS IS" BASIS,
012     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013     * See the License for the specific language governing permissions and
014     * limitations under the License.
015     */
016    
017    package com.google.common.collect;
018    
019    import com.google.common.annotations.GwtCompatible;
020    
021    import java.util.Collection;
022    import java.util.Collections;
023    import java.util.Iterator;
024    import java.util.List;
025    import java.util.Set;
026    
027    import javax.annotation.Nullable;
028    
029    /**
030     * A collection that supports order-independent equality, like {@link Set}, but
031     * may have duplicate elements. A multiset is also sometimes called a
032     * <i>bag</i>.
033     *
034     * <p>Elements of a multiset that are equal to one another (see "Note on
035     * element equivalence", below) are referred to as <i>occurrences</i> of the
036     * same single element. The total number of occurrences of an element in a
037     * multiset is called the <i>count</i> of that element (the terms "frequency"
038     * and "multiplicity" are equivalent, but not used in this API). Since the count
039     * of an element is represented as an {@code int}, a multiset may never contain
040     * more than {@link Integer#MAX_VALUE} occurrences of any one element.
041     *
042     * <p>{@code Multiset} refines the specifications of several methods from
043     * {@code Collection}. It also defines an additional query operation, {@link
044     * #count}, which returns the count of an element. There are five new
045     * bulk-modification operations, for example {@link #add(Object, int)}, to add
046     * or remove multiple occurrences of an element at once, or to set the count of
047     * an element to a specific value. These modification operations are optional,
048     * but implementations which support the standard collection operations {@link
049     * #add(Object)} or {@link #remove(Object)} are encouraged to implement the
050     * related methods as well. Finally, two collection views are provided: {@link
051     * #elementSet} contains the distinct elements of the multiset "with duplicates
052     * collapsed", and {@link #entrySet} is similar but contains {@link Entry
053     * Multiset.Entry} instances, each providing both a distinct element and the
054     * count of that element.
055     *
056     * <p>In addition to these required methods, implementations of {@code
057     * Multiset} are expected to provide two {@code static} creation methods:
058     * {@code create()}, returning an empty multiset, and {@code
059     * create(Iterable<? extends E>)}, returning a multiset containing the
060     * given initial elements. This is simply a refinement of {@code Collection}'s
061     * constructor recommendations, reflecting the new developments of Java 5.
062     *
063     * <p>As with other collection types, the modification operations are optional,
064     * and should throw {@link UnsupportedOperationException} when they are not
065     * implemented. Most implementations should support either all add operations
066     * or none of them, all removal operations or none of them, and if and only if
067     * all of these are supported, the {@code setCount} methods as well.
068     *
069     * <p>A multiset uses {@link Object#equals} to determine whether two instances
070     * should be considered "the same," <i>unless specified otherwise</i> by the
071     * implementation.
072     *
073     * <p>Common implementations include {@link ImmutableMultiset}, {@link
074     * HashMultiset}, and {@link ConcurrentHashMultiset}.
075     *
076     * <p>If your values may be zero, negative, or outside the range of an int, you
077     * may wish to use {@link com.google.common.util.concurrent.AtomicLongMap}
078     * instead. Note, however, that unlike {@code Multiset}, {@code AtomicLongMap}
079     * does not automatically remove zeros.
080     *
081     * @author Kevin Bourrillion
082     * @since 2.0 (imported from Google Collections Library)
083     */
084    @GwtCompatible
085    public interface Multiset<E> extends Collection<E> {
086      // Query Operations
087    
088      /**
089       * Returns the number of occurrences of an element in this multiset (the
090       * <i>count</i> of the element). Note that for an {@link Object#equals}-based
091       * multiset, this gives the same result as {@link Collections#frequency}
092       * (which would presumably perform more poorly).
093       *
094       * <p><b>Note:</b> the utility method {@link Iterables#frequency} generalizes
095       * this operation; it correctly delegates to this method when dealing with a
096       * multiset, but it can also accept any other iterable type.
097       *
098       * @param element the element to count occurrences of
099       * @return the number of occurrences of the element in this multiset; possibly
100       *     zero but never negative
101       */
102      int count(@Nullable Object element);
103    
104      // Bulk Operations
105    
106      /**
107       * Adds a number of occurrences of an element to this multiset. Note that if
108       * {@code occurrences == 1}, this method has the identical effect to {@link
109       * #add(Object)}. This method is functionally equivalent (except in the case
110       * of overflow) to the call {@code addAll(Collections.nCopies(element,
111       * occurrences))}, which would presumably perform much more poorly.
112       *
113       * @param element the element to add occurrences of; may be null only if
114       *     explicitly allowed by the implementation
115       * @param occurrences the number of occurrences of the element to add. May be
116       *     zero, in which case no change will be made.
117       * @return the count of the element before the operation; possibly zero
118       * @throws IllegalArgumentException if {@code occurrences} is negative, or if
119       *     this operation would result in more than {@link Integer#MAX_VALUE}
120       *     occurrences of the element
121       * @throws NullPointerException if {@code element} is null and this
122       *     implementation does not permit null elements. Note that if {@code
123       *     occurrences} is zero, the implementation may opt to return normally.
124       */
125      int add(@Nullable E element, int occurrences);
126    
127      /**
128       * Removes a number of occurrences of the specified element from this
129       * multiset. If the multiset contains fewer than this number of occurrences to
130       * begin with, all occurrences will be removed.  Note that if
131       * {@code occurrences == 1}, this is functionally equivalent to the call
132       * {@code remove(element)}.
133       *
134       * @param element the element to conditionally remove occurrences of
135       * @param occurrences the number of occurrences of the element to remove. May
136       *     be zero, in which case no change will be made.
137       * @return the count of the element before the operation; possibly zero
138       * @throws IllegalArgumentException if {@code occurrences} is negative
139       */
140      int remove(@Nullable Object element, int occurrences);
141    
142      /**
143       * Adds or removes the necessary occurrences of an element such that the
144       * element attains the desired count.
145       *
146       * @param element the element to add or remove occurrences of; may be null
147       *     only if explicitly allowed by the implementation
148       * @param count the desired count of the element in this multiset
149       * @return the count of the element before the operation; possibly zero
150       * @throws IllegalArgumentException if {@code count} is negative
151       * @throws NullPointerException if {@code element} is null and this
152       *     implementation does not permit null elements. Note that if {@code
153       *     count} is zero, the implementor may optionally return zero instead.
154       */
155      int setCount(E element, int count);
156    
157      /**
158       * Conditionally sets the count of an element to a new value, as described in
159       * {@link #setCount(Object, int)}, provided that the element has the expected
160       * current count. If the current count is not {@code oldCount}, no change is
161       * made.
162       *
163       * @param element the element to conditionally set the count of; may be null
164       *     only if explicitly allowed by the implementation
165       * @param oldCount the expected present count of the element in this multiset
166       * @param newCount the desired count of the element in this multiset
167       * @return {@code true} if the condition for modification was met. This
168       *     implies that the multiset was indeed modified, unless
169       *     {@code oldCount == newCount}.
170       * @throws IllegalArgumentException if {@code oldCount} or {@code newCount} is
171       *     negative
172       * @throws NullPointerException if {@code element} is null and the
173       *     implementation does not permit null elements. Note that if {@code
174       *     oldCount} and {@code newCount} are both zero, the implementor may
175       *     optionally return {@code true} instead.
176       */
177      boolean setCount(E element, int oldCount, int newCount);
178    
179      // Views
180    
181      /**
182       * Returns the set of distinct elements contained in this multiset. The
183       * element set is backed by the same data as the multiset, so any change to
184       * either is immediately reflected in the other. The order of the elements in
185       * the element set is unspecified.
186       *
187       * <p>If the element set supports any removal operations, these necessarily
188       * cause <b>all</b> occurrences of the removed element(s) to be removed from
189       * the multiset. Implementations are not expected to support the add
190       * operations, although this is possible.
191       *
192       * <p>A common use for the element set is to find the number of distinct
193       * elements in the multiset: {@code elementSet().size()}.
194       *
195       * @return a view of the set of distinct elements in this multiset
196       */
197      Set<E> elementSet();
198    
199      /**
200       * Returns a view of the contents of this multiset, grouped into {@code
201       * Multiset.Entry} instances, each providing an element of the multiset and
202       * the count of that element. This set contains exactly one entry for each
203       * distinct element in the multiset (thus it always has the same size as the
204       * {@link #elementSet}). The order of the elements in the element set is
205       * unspecified.
206       *
207       * <p>The entry set is backed by the same data as the multiset, so any change
208       * to either is immediately reflected in the other. However, multiset changes
209       * may or may not be reflected in any {@code Entry} instances already
210       * retrieved from the entry set (this is implementation-dependent).
211       * Furthermore, implementations are not required to support modifications to
212       * the entry set at all, and the {@code Entry} instances themselves don't
213       * even have methods for modification. See the specific implementation class
214       * for more details on how its entry set handles modifications.
215       *
216       * @return a set of entries representing the data of this multiset
217       */
218      Set<Entry<E>> entrySet();
219    
220      /**
221       * An unmodifiable element-count pair for a multiset. The {@link
222       * Multiset#entrySet} method returns a view of the multiset whose elements
223       * are of this class. A multiset implementation may return Entry instances
224       * that are either live "read-through" views to the Multiset, or immutable
225       * snapshots. Note that this type is unrelated to the similarly-named type
226       * {@code Map.Entry}.
227       *
228       * @since 2.0 (imported from Google Collections Library)
229       */
230      interface Entry<E> {
231    
232        /**
233         * Returns the multiset element corresponding to this entry. Multiple calls
234         * to this method always return the same instance.
235         *
236         * @return the element corresponding to this entry
237         */
238        E getElement();
239    
240        /**
241         * Returns the count of the associated element in the underlying multiset.
242         * This count may either be an unchanging snapshot of the count at the time
243         * the entry was retrieved, or a live view of the current count of the
244         * element in the multiset, depending on the implementation. Note that in
245         * the former case, this method can never return zero, while in the latter,
246         * it will return zero if all occurrences of the element were since removed
247         * from the multiset.
248         *
249         * @return the count of the element; never negative
250         */
251        int getCount();
252    
253        /**
254         * {@inheritDoc}
255         *
256         * <p>Returns {@code true} if the given object is also a multiset entry and
257         * the two entries represent the same element and count. That is, two
258         * entries {@code a} and {@code b} are equal if: <pre>   {@code
259         *
260         *   Objects.equal(a.getElement(), b.getElement())
261         *       && a.getCount() == b.getCount()}</pre>
262         */
263        @Override
264        // TODO(kevinb): check this wrt TreeMultiset?
265        boolean equals(Object o);
266    
267        /**
268         * {@inheritDoc}
269         *
270         * <p>The hash code of a multiset entry for element {@code element} and
271         * count {@code count} is defined as: <pre>   {@code
272         *
273         *   ((element == null) ? 0 : element.hashCode()) ^ count}</pre>
274         */
275        @Override
276        int hashCode();
277    
278        /**
279         * Returns the canonical string representation of this entry, defined as
280         * follows. If the count for this entry is one, this is simply the string
281         * representation of the corresponding element. Otherwise, it is the string
282         * representation of the element, followed by the three characters {@code
283         * " x "} (space, letter x, space), followed by the count.
284         */
285        @Override
286        String toString();
287      }
288    
289      // Comparison and hashing
290    
291      /**
292       * Compares the specified object with this multiset for equality. Returns
293       * {@code true} if the given object is also a multiset and contains equal
294       * elements with equal counts, regardless of order.
295       */
296      @Override
297      // TODO(kevinb): caveats about equivalence-relation?
298      boolean equals(@Nullable Object object);
299    
300      /**
301       * Returns the hash code for this multiset. This is defined as the sum of
302       * <pre>   {@code
303       *
304       *   ((element == null) ? 0 : element.hashCode()) ^ count(element)}</pre>
305       *
306       * over all distinct elements in the multiset. It follows that a multiset and
307       * its entry set always have the same hash code.
308       */
309      @Override
310      int hashCode();
311    
312      /**
313       * {@inheritDoc}
314       *
315       * <p>It is recommended, though not mandatory, that this method return the
316       * result of invoking {@link #toString} on the {@link #entrySet}, yielding a
317       * result such as {@code [a x 3, c, d x 2, e]}.
318       */
319      @Override
320      String toString();
321    
322      // Refined Collection Methods
323    
324      /**
325       * {@inheritDoc}
326       *
327       * <p>Elements that occur multiple times in the multiset will appear
328       * multiple times in this iterator, though not necessarily sequentially.
329       */
330      @Override
331      Iterator<E> iterator();
332    
333      /**
334       * Determines whether this multiset contains the specified element.
335       *
336       * <p>This method refines {@link Collection#contains} to further specify that
337       * it <b>may not</b> throw an exception in response to {@code element} being
338       * null or of the wrong type.
339       *
340       * @param element the element to check for
341       * @return {@code true} if this multiset contains at least one occurrence of
342       *     the element
343       */
344      @Override
345      boolean contains(@Nullable Object element);
346    
347      /**
348       * Returns {@code true} if this multiset contains at least one occurrence of
349       * each element in the specified collection.
350       *
351       * <p>This method refines {@link Collection#containsAll} to further specify
352       * that it <b>may not</b> throw an exception in response to any of {@code
353       * elements} being null or of the wrong type.
354       *
355       * <p><b>Note:</b> this method does not take into account the occurrence
356       * count of an element in the two collections; it may still return {@code
357       * true} even if {@code elements} contains several occurrences of an element
358       * and this multiset contains only one. This is no different than any other
359       * collection type like {@link List}, but it may be unexpected to the user of
360       * a multiset.
361       *
362       * @param elements the collection of elements to be checked for containment in
363       *     this multiset
364       * @return {@code true} if this multiset contains at least one occurrence of
365       *     each element contained in {@code elements}
366       * @throws NullPointerException if {@code elements} is null
367       */
368      @Override
369      boolean containsAll(Collection<?> elements);
370    
371      /**
372       * Adds a single occurrence of the specified element to this multiset.
373       *
374       * <p>This method refines {@link Collection#add}, which only <i>ensures</i>
375       * the presence of the element, to further specify that a successful call must
376       * always increment the count of the element, and the overall size of the
377       * collection, by one.
378       *
379       * @param element the element to add one occurrence of; may be null only if
380       *     explicitly allowed by the implementation
381       * @return {@code true} always, since this call is required to modify the
382       *     multiset, unlike other {@link Collection} types
383       * @throws NullPointerException if {@code element} is null and this
384       *     implementation does not permit null elements
385       * @throws IllegalArgumentException if {@link Integer#MAX_VALUE} occurrences
386       *     of {@code element} are already contained in this multiset
387       */
388      @Override
389      boolean add(E element);
390    
391      /**
392       * Removes a <i>single</i> occurrence of the specified element from this
393       * multiset, if present.
394       *
395       * <p>This method refines {@link Collection#remove} to further specify that it
396       * <b>may not</b> throw an exception in response to {@code element} being null
397       * or of the wrong type.
398       *
399       * @param element the element to remove one occurrence of
400       * @return {@code true} if an occurrence was found and removed
401       */
402      @Override
403      boolean remove(@Nullable Object element);
404    
405      /**
406       * {@inheritDoc}
407       *
408       * <p><b>Note:</b> This method ignores how often any element might appear in
409       * {@code c}, and only cares whether or not an element appears at all.
410       * If you wish to remove one occurrence in this multiset for every occurrence
411       * in {@code c}, see {@link Multisets#removeOccurrences(Multiset, Multiset)}.
412       * 
413       * <p>This method refines {@link Collection#removeAll} to further specify that
414       * it <b>may not</b> throw an exception in response to any of {@code elements}
415       * being null or of the wrong type. 
416       */
417      @Override
418      boolean removeAll(Collection<?> c);
419    
420      /**
421       * {@inheritDoc}
422       *
423       * <p><b>Note:</b> This method ignores how often any element might appear in
424       * {@code c}, and only cares whether or not an element appears at all.
425       * If you wish to remove one occurrence in this multiset for every occurrence
426       * in {@code c}, see {@link Multisets#retainOccurrences(Multiset, Multiset)}.
427       * 
428       * <p>This method refines {@link Collection#retainAll} to further specify that
429       * it <b>may not</b> throw an exception in response to any of {@code elements}
430       * being null or of the wrong type.
431       * 
432       * @see Multisets#retainOccurrences(Multiset, Multiset)
433       */
434      @Override
435      boolean retainAll(Collection<?> c);
436    }