Home | History | Annotate | Download | only in collect
      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 
     17 package com.google.common.collect;
     18 
     19 import com.google.common.annotations.Beta;
     20 import com.google.common.annotations.GwtCompatible;
     21 import com.google.common.base.Objects;
     22 
     23 import java.util.Collection;
     24 import java.util.Iterator;
     25 import java.util.Set;
     26 
     27 import javax.annotation.Nullable;
     28 
     29 /**
     30  * A multiset which forwards all its method calls to another multiset.
     31  * Subclasses should override one or more methods to modify the behavior of the
     32  * backing multiset as desired per the <a
     33  * href="http://en.wikipedia.org/wiki/Decorator_pattern">decorator pattern</a>.
     34  *
     35  * <p><b>Warning:</b> The methods of {@code ForwardingMultiset} forward
     36  * <b>indiscriminately</b> to the methods of the delegate. For example,
     37  * overriding {@link #add(Object, int)} alone <b>will not</b> change the
     38  * behavior of {@link #add(Object)}, which can lead to unexpected behavior. In
     39  * this case, you should override {@code add(Object)} as well, either providing
     40  * your own implementation, or delegating to the provided {@code standardAdd}
     41  * method.
     42  *
     43  * <p>The {@code standard} methods and any collection views they return are not
     44  * guaranteed to be thread-safe, even when all of the methods that they depend
     45  * on are thread-safe.
     46  *
     47  * @author Kevin Bourrillion
     48  * @author Louis Wasserman
     49  * @since 2.0 (imported from Google Collections Library)
     50  */
     51 @GwtCompatible
     52 public abstract class ForwardingMultiset<E> extends ForwardingCollection<E>
     53     implements Multiset<E> {
     54 
     55   /** Constructor for use by subclasses. */
     56   protected ForwardingMultiset() {}
     57 
     58   @Override protected abstract Multiset<E> delegate();
     59 
     60   @Override
     61   public int count(Object element) {
     62     return delegate().count(element);
     63   }
     64 
     65   @Override
     66   public int add(E element, int occurrences) {
     67     return delegate().add(element, occurrences);
     68   }
     69 
     70   @Override
     71   public int remove(Object element, int occurrences) {
     72     return delegate().remove(element, occurrences);
     73   }
     74 
     75   @Override
     76   public Set<E> elementSet() {
     77     return delegate().elementSet();
     78   }
     79 
     80   @Override
     81   public Set<Entry<E>> entrySet() {
     82     return delegate().entrySet();
     83   }
     84 
     85   @Override public boolean equals(@Nullable Object object) {
     86     return object == this || delegate().equals(object);
     87   }
     88 
     89   @Override public int hashCode() {
     90     return delegate().hashCode();
     91   }
     92 
     93   @Override
     94   public int setCount(E element, int count) {
     95     return delegate().setCount(element, count);
     96   }
     97 
     98   @Override
     99   public boolean setCount(E element, int oldCount, int newCount) {
    100     return delegate().setCount(element, oldCount, newCount);
    101   }
    102 
    103   /**
    104    * A sensible definition of {@link #contains} in terms of {@link #count}. If
    105    * you override {@link #count}, you may wish to override {@link #contains} to
    106    * forward to this implementation.
    107    *
    108    * @since 7.0
    109    */
    110   @Override @Beta protected boolean standardContains(@Nullable Object object) {
    111     return count(object) > 0;
    112   }
    113 
    114   /**
    115    * A sensible definition of {@link #clear} in terms of the {@code iterator}
    116    * method of {@link #entrySet}. If you override {@link #entrySet}, you may
    117    * wish to override {@link #clear} to forward to this implementation.
    118    *
    119    * @since 7.0
    120    */
    121   @Override @Beta protected void standardClear() {
    122     Iterator<Entry<E>> entryIterator = entrySet().iterator();
    123     while (entryIterator.hasNext()) {
    124       entryIterator.next();
    125       entryIterator.remove();
    126     }
    127   }
    128 
    129   /**
    130    * A sensible, albeit inefficient, definition of {@link #count} in terms of
    131    * {@link #entrySet}. If you override {@link #entrySet}, you may wish to
    132    * override {@link #count} to forward to this implementation.
    133    *
    134    * @since 7.0
    135    */
    136   @Beta protected int standardCount(@Nullable Object object) {
    137     for (Entry<?> entry : this.entrySet()) {
    138       if (Objects.equal(entry.getElement(), object)) {
    139         return entry.getCount();
    140       }
    141     }
    142     return 0;
    143   }
    144 
    145   /**
    146    * A sensible definition of {@link #add(Object)} in terms of {@link
    147    * #add(Object, int)}. If you override {@link #add(Object, int)}, you may
    148    * wish to override {@link #add(Object)} to forward to this implementation.
    149    *
    150    * @since 7.0
    151    */
    152   @Beta protected boolean standardAdd(E element) {
    153     add(element, 1);
    154     return true;
    155   }
    156 
    157   /**
    158    * A sensible definition of {@link #addAll(Collection)} in terms of {@link
    159    * #add(Object)} and {@link #add(Object, int)}. If you override either of
    160    * these methods, you may wish to override {@link #addAll(Collection)} to
    161    * forward to this implementation.
    162    *
    163    * @since 7.0
    164    */
    165   @Beta @Override protected boolean standardAddAll(
    166       Collection<? extends E> elementsToAdd) {
    167     return Multisets.addAllImpl(this, elementsToAdd);
    168   }
    169 
    170   /**
    171    * A sensible definition of {@link #remove(Object)} in terms of {@link
    172    * #remove(Object, int)}. If you override {@link #remove(Object, int)}, you
    173    * may wish to override {@link #remove(Object)} to forward to this
    174    * implementation.
    175    *
    176    * @since 7.0
    177    */
    178   @Beta @Override protected boolean standardRemove(Object element) {
    179     return remove(element, 1) > 0;
    180   }
    181 
    182   /**
    183    * A sensible definition of {@link #removeAll} in terms of the {@code
    184    * removeAll} method of {@link #elementSet}. If you override {@link
    185    * #elementSet}, you may wish to override {@link #removeAll} to forward to
    186    * this implementation.
    187    *
    188    * @since 7.0
    189    */
    190   @Beta @Override protected boolean standardRemoveAll(
    191       Collection<?> elementsToRemove) {
    192     return Multisets.removeAllImpl(this, elementsToRemove);
    193   }
    194 
    195   /**
    196    * A sensible definition of {@link #retainAll} in terms of the {@code
    197    * retainAll} method of {@link #elementSet}. If you override {@link
    198    * #elementSet}, you may wish to override {@link #retainAll} to forward to
    199    * this implementation.
    200    *
    201    * @since 7.0
    202    */
    203   @Beta @Override protected boolean standardRetainAll(
    204       Collection<?> elementsToRetain) {
    205     return Multisets.retainAllImpl(this, elementsToRetain);
    206   }
    207 
    208   /**
    209    * A sensible definition of {@link #setCount(Object, int)} in terms of {@link
    210    * #count(Object)}, {@link #add(Object, int)}, and {@link #remove(Object,
    211    * int)}. {@link #entrySet()}. If you override any of these methods, you may
    212    * wish to override {@link #setCount(Object, int)} to forward to this
    213    * implementation.
    214    *
    215    * @since 7.0
    216    */
    217   @Beta protected int standardSetCount(E element, int count) {
    218     return Multisets.setCountImpl(this, element, count);
    219   }
    220 
    221   /**
    222    * A sensible definition of {@link #setCount(Object, int, int)} in terms of
    223    * {@link #count(Object)} and {@link #setCount(Object, int)}. If you override
    224    * either of these methods, you may wish to override {@link #setCount(Object,
    225    * int, int)} to forward to this implementation.
    226    *
    227    * @since 7.0
    228    */
    229   @Beta protected boolean standardSetCount(
    230       E element, int oldCount, int newCount) {
    231     return Multisets.setCountImpl(this, element, oldCount, newCount);
    232   }
    233 
    234   /**
    235    * A sensible implementation of {@link Multiset#elementSet} in terms of the
    236    * following methods: {@link ForwardingMultiset#clear}, {@link
    237    * ForwardingMultiset#contains}, {@link ForwardingMultiset#containsAll},
    238    * {@link ForwardingMultiset#count}, {@link ForwardingMultiset#isEmpty}, the
    239    * {@link Set#size} and {@link Set#iterator} methods of {@link
    240    * ForwardingMultiset#entrySet}, and {@link ForwardingMultiset#remove(Object,
    241    * int)}.  In many situations, you may wish to override {@link
    242    * ForwardingMultiset#elementSet} to forward to this implementation or a
    243    * subclass thereof.
    244    *
    245    * @since 10.0
    246    */
    247   @Beta
    248   protected class StandardElementSet extends Multisets.ElementSet<E> {
    249     public StandardElementSet() {}
    250 
    251     @Override
    252     Multiset<E> multiset() {
    253       return ForwardingMultiset.this;
    254     }
    255   }
    256 
    257   /**
    258    * A sensible definition of {@link #iterator} in terms of {@link #entrySet}
    259    * and {@link #remove(Object)}. If you override either of these methods, you
    260    * may wish to override {@link #iterator} to forward to this implementation.
    261    *
    262    * @since 7.0
    263    */
    264   @Beta protected Iterator<E> standardIterator() {
    265     return Multisets.iteratorImpl(this);
    266   }
    267 
    268   /**
    269    * A sensible, albeit inefficient, definition of {@link #size} in terms of
    270    * {@link #entrySet}. If you override {@link #entrySet}, you may wish to
    271    * override {@link #size} to forward to this implementation.
    272    *
    273    * @since 7.0
    274    */
    275   @Beta protected int standardSize() {
    276     return Multisets.sizeImpl(this);
    277   }
    278 
    279   /**
    280    * A sensible, albeit inefficient, definition of {@link #size} in terms of
    281    * {@code entrySet().size()} and {@link #count}. If you override either of
    282    * these methods, you may wish to override {@link #size} to forward to this
    283    * implementation.
    284    *
    285    * @since 7.0
    286    */
    287   @Beta protected boolean standardEquals(@Nullable Object object) {
    288     return Multisets.equalsImpl(this, object);
    289   }
    290 
    291   /**
    292    * A sensible definition of {@link #hashCode} as {@code entrySet().hashCode()}
    293    * . If you override {@link #entrySet}, you may wish to override {@link
    294    * #hashCode} to forward to this implementation.
    295    *
    296    * @since 7.0
    297    */
    298   @Beta protected int standardHashCode() {
    299     return entrySet().hashCode();
    300   }
    301 
    302   /**
    303    * A sensible definition of {@link #toString} as {@code entrySet().toString()}
    304    * . If you override {@link #entrySet}, you may wish to override {@link
    305    * #toString} to forward to this implementation.
    306    *
    307    * @since 7.0
    308    */
    309   @Beta @Override protected String standardToString() {
    310     return entrySet().toString();
    311   }
    312 }
    313