/*
* Copyright (C) 2011 The Guava Authors
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
* in compliance with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software distributed under the License
* is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
* or implied. See the License for the specific language governing permissions and limitations under
* the License.
*/
package com.google.common.collect;
import com.google.common.annotations.GwtCompatible;
import com.google.common.primitives.Ints;
import com.google.errorprone.annotations.concurrent.LazyInit;
import com.google.j2objc.annotations.WeakOuter;
import javax.annotation.Nullable;
/**
* Implementation of {@link ImmutableMultiset} with zero or more elements.
*
* @author Jared Levy
* @author Louis Wasserman
*/
@GwtCompatible(serializable = true)
@SuppressWarnings("serial") // uses writeReplace(), not default serialization
class RegularImmutableMultiset<E> extends ImmutableMultiset<E> {
static final RegularImmutableMultiset<Object> EMPTY =
new RegularImmutableMultiset<Object>(ObjectCountHashMap.create());
private final transient ObjectCountHashMap<E> contents;
private final transient int size;
@LazyInit
private transient ImmutableSet<E> elementSet;
RegularImmutableMultiset(ObjectCountHashMap<E> contents) {
this.contents = contents;
long size = 0;
for (int i = 0; i < contents.size(); i++) {
size += contents.getValue(i);
}
this.size = Ints.saturatedCast(size);
}
@Override
boolean isPartialView() {
return false;
}
@Override
public int count(@Nullable Object element) {
return contents.get(element);
}
@Override
public int size() {
return size;
}
@Override
public ImmutableSet<E> elementSet() {
ImmutableSet<E> result = elementSet;
return (result == null) ? elementSet = new ElementSet() : result;
}
@WeakOuter
private final class ElementSet extends ImmutableSet.Indexed<E> {
@Override
E get(int index) {
return contents.getKey(index);
}
@Override
public boolean contains(@Nullable Object object) {
return RegularImmutableMultiset.this.contains(object);
}
@Override
boolean isPartialView() {
return true;
}
@Override
public int size() {
return contents.size();
}
}
@Override
Entry<E> getEntry(int index) {
return contents.getEntry(index);
}
}