package org.eclipse.collections.impl.bag;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.StringJoiner;
import org.eclipse.collections.api.bag.Bag;
import org.eclipse.collections.api.bag.MutableBag;
import org.eclipse.collections.api.bag.MutableBagIterable;
import org.eclipse.collections.api.bag.primitive.MutableBooleanBag;
import org.eclipse.collections.api.bag.primitive.MutableByteBag;
import org.eclipse.collections.api.bag.primitive.MutableCharBag;
import org.eclipse.collections.api.bag.primitive.MutableDoubleBag;
import org.eclipse.collections.api.bag.primitive.MutableFloatBag;
import org.eclipse.collections.api.bag.primitive.MutableIntBag;
import org.eclipse.collections.api.bag.primitive.MutableLongBag;
import org.eclipse.collections.api.bag.primitive.MutableShortBag;
import org.eclipse.collections.api.bag.sorted.MutableSortedBag;
import org.eclipse.collections.api.block.function.Function;
import org.eclipse.collections.api.block.function.Function2;
import org.eclipse.collections.api.block.function.Function3;
import org.eclipse.collections.api.block.function.primitive.BooleanFunction;
import org.eclipse.collections.api.block.function.primitive.ByteFunction;
import org.eclipse.collections.api.block.function.primitive.CharFunction;
import org.eclipse.collections.api.block.function.primitive.DoubleFunction;
import org.eclipse.collections.api.block.function.primitive.DoubleObjectToDoubleFunction;
import org.eclipse.collections.api.block.function.primitive.FloatFunction;
import org.eclipse.collections.api.block.function.primitive.FloatObjectToFloatFunction;
import org.eclipse.collections.api.block.function.primitive.IntFunction;
import org.eclipse.collections.api.block.function.primitive.IntObjectToIntFunction;
import org.eclipse.collections.api.block.function.primitive.LongFunction;
import org.eclipse.collections.api.block.function.primitive.LongObjectToLongFunction;
import org.eclipse.collections.api.block.function.primitive.ShortFunction;
import org.eclipse.collections.api.block.predicate.Predicate;
import org.eclipse.collections.api.block.predicate.Predicate2;
import org.eclipse.collections.api.collection.primitive.MutableBooleanCollection;
import org.eclipse.collections.api.collection.primitive.MutableByteCollection;
import org.eclipse.collections.api.collection.primitive.MutableCharCollection;
import org.eclipse.collections.api.collection.primitive.MutableDoubleCollection;
import org.eclipse.collections.api.collection.primitive.MutableFloatCollection;
import org.eclipse.collections.api.collection.primitive.MutableIntCollection;
import org.eclipse.collections.api.collection.primitive.MutableLongCollection;
import org.eclipse.collections.api.collection.primitive.MutableShortCollection;
import org.eclipse.collections.api.list.MutableList;
import org.eclipse.collections.api.multimap.MutableMultimap;
import org.eclipse.collections.api.set.MutableSet;
import org.eclipse.collections.api.set.SetIterable;
import org.eclipse.collections.api.set.sorted.MutableSortedSet;
import org.eclipse.collections.api.tuple.primitive.ObjectIntPair;
import org.eclipse.collections.impl.AbstractRichIterable;
import org.eclipse.collections.impl.Counter;
import org.eclipse.collections.impl.bag.mutable.HashBag;
import org.eclipse.collections.impl.bag.sorted.mutable.TreeBag;
import org.eclipse.collections.impl.factory.Sets;
import org.eclipse.collections.impl.factory.SortedSets;
import org.eclipse.collections.impl.list.mutable.FastList;
import org.eclipse.collections.impl.set.mutable.UnifiedSet;
import org.eclipse.collections.impl.tuple.primitive.PrimitiveTuples;
import org.eclipse.collections.impl.utility.Iterate;
public abstract class AbstractBag<T>
extends AbstractRichIterable<T>
implements Collection<T>, Bag<T>
{
@Override
public <R extends Collection<T>> R select(Predicate<? super T> predicate, R target)
{
if (target instanceof MutableBagIterable<?>)
{
MutableBagIterable<T> targetBag = (MutableBagIterable<T>) target;
this.forEachWithOccurrences((each, occurrences) -> {
if (predicate.accept(each))
{
targetBag.addOccurrences(each, occurrences);
}
});
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
if (predicate.accept(each))
{
for (int i = 0; i < occurrences; i++)
{
target.add(each);
}
}
});
}
return target;
}
@Override
public <P, R extends Collection<T>> R selectWith(
Predicate2<? super T, ? super P> predicate,
P parameter,
R target)
{
if (target instanceof MutableBagIterable<?>)
{
MutableBagIterable<T> targetBag = (MutableBagIterable<T>) target;
this.forEachWithOccurrences((each, occurrences) -> {
if (predicate.accept(each, parameter))
{
targetBag.addOccurrences(each, occurrences);
}
});
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
if (predicate.accept(each, parameter))
{
for (int i = 0; i < occurrences; i++)
{
target.add(each);
}
}
});
}
return target;
}
@Override
public <R extends Collection<T>> R reject(Predicate<? super T> predicate, R target)
{
if (target instanceof MutableBagIterable<?>)
{
MutableBagIterable<T> targetBag = (MutableBagIterable<T>) target;
this.forEachWithOccurrences((each, occurrences) -> {
if (!predicate.accept(each))
{
targetBag.addOccurrences(each, occurrences);
}
});
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
if (!predicate.accept(each))
{
for (int i = 0; i < occurrences; i++)
{
target.add(each);
}
}
});
}
return target;
}
@Override
public <P, R extends Collection<T>> R rejectWith(
Predicate2<? super T, ? super P> predicate,
P parameter,
R target)
{
if (target instanceof MutableBagIterable<?>)
{
MutableBagIterable<T> targetBag = (MutableBagIterable<T>) target;
this.forEachWithOccurrences((each, occurrences) -> {
if (!predicate.accept(each, parameter))
{
targetBag.addOccurrences(each, occurrences);
}
});
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
if (!predicate.accept(each, parameter))
{
for (int i = 0; i < occurrences; i++)
{
target.add(each);
}
}
});
}
return target;
}
@Override
public int count(Predicate<? super T> predicate)
{
Counter result = new Counter();
this.forEachWithOccurrences((each, occurrences) -> {
if (predicate.accept(each))
{
result.add(occurrences);
}
});
return result.getCount();
}
@Override
public <V, R extends Collection<V>> R collect(Function<? super T, ? extends V> function, R target)
{
if (target instanceof MutableBagIterable<?>)
{
MutableBagIterable<V> targetBag = (MutableBagIterable<V>) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(function.valueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
V value = function.valueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <P, V, R extends Collection<V>> R collectWith(
Function2<? super T, ? super P, ? extends V> function,
P parameter,
R target)
{
if (target instanceof MutableBagIterable<?>)
{
MutableBagIterable<V> targetBag = (MutableBagIterable<V>) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(function.value(each, parameter), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
V value = function.value(each, parameter);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <V, R extends Collection<V>> R collectIf(
Predicate<? super T> predicate,
Function<? super T, ? extends V> function,
R target)
{
if (target instanceof MutableBagIterable<?>)
{
MutableBagIterable<V> targetBag = (MutableBagIterable<V>) target;
this.forEachWithOccurrences((each, occurrences) -> {
if (predicate.accept(each))
{
targetBag.addOccurrences(function.valueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
if (predicate.accept(each))
{
V value = function.valueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public <V, R extends Collection<V>> R flatCollect(Function<? super T, ? extends Iterable<V>> function, R target)
{
if (target instanceof MutableBagIterable<?>)
{
MutableBagIterable<V> targetBag = (MutableBagIterable<V>) target;
this.forEachWithOccurrences((each, occurrences) -> {
Iterable<V> values = function.valueOf(each);
Iterate.forEach(values, eachValue -> targetBag.addOccurrences(eachValue, occurrences));
});
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
Iterable<V> values = function.valueOf(each);
for (int i = 0; i < occurrences; i++)
{
Iterate.forEach(values, target::add);
}
});
}
return target;
}
@Override
public <R extends MutableBooleanCollection> R collectBoolean(BooleanFunction<? super T> booleanFunction, R target)
{
if (target instanceof MutableBooleanBag)
{
MutableBooleanBag targetBag = (MutableBooleanBag) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(booleanFunction.booleanValueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
boolean value = booleanFunction.booleanValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <R extends MutableByteCollection> R collectByte(ByteFunction<? super T> byteFunction, R target)
{
if (target instanceof MutableByteBag)
{
MutableByteBag targetBag = (MutableByteBag) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(byteFunction.byteValueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
byte value = byteFunction.byteValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <R extends MutableCharCollection> R collectChar(CharFunction<? super T> charFunction, R target)
{
if (target instanceof MutableCharBag)
{
MutableCharBag targetBag = (MutableCharBag) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(charFunction.charValueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
char value = charFunction.charValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <R extends MutableDoubleCollection> R collectDouble(DoubleFunction<? super T> doubleFunction, R target)
{
if (target instanceof MutableDoubleBag)
{
MutableDoubleBag targetBag = (MutableDoubleBag) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(doubleFunction.doubleValueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
double value = doubleFunction.doubleValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <R extends MutableFloatCollection> R collectFloat(FloatFunction<? super T> floatFunction, R target)
{
if (target instanceof MutableFloatBag)
{
MutableFloatBag targetBag = (MutableFloatBag) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(floatFunction.floatValueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
float value = floatFunction.floatValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <R extends MutableIntCollection> R collectInt(IntFunction<? super T> intFunction, R target)
{
if (target instanceof MutableIntBag)
{
MutableIntBag targetBag = (MutableIntBag) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(intFunction.intValueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
int value = intFunction.intValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <R extends MutableLongCollection> R collectLong(LongFunction<? super T> longFunction, R target)
{
if (target instanceof MutableLongBag)
{
MutableLongBag targetBag = (MutableLongBag) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(longFunction.longValueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
long value = longFunction.longValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <R extends MutableShortCollection> R collectShort(ShortFunction<? super T> shortFunction, R target)
{
if (target instanceof MutableShortBag)
{
MutableShortBag targetBag = (MutableShortBag) target;
this.forEachWithOccurrences((each, occurrences) -> targetBag.addOccurrences(shortFunction.shortValueOf(each), occurrences));
}
else
{
this.forEachWithOccurrences((each, occurrences) -> {
short value = shortFunction.shortValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
});
}
return target;
}
@Override
public <V, R extends MutableMultimap<V, T>> R groupBy(
Function<? super T, ? extends V> function,
R target)
{
this.forEachWithOccurrences((each, occurrences) -> {
V value = function.valueOf(each);
target.putAll(value, Collections.nCopies(occurrences, each));
});
return target;
}
@Override
public <V, R extends MutableMultimap<V, T>> R groupByEach(
Function<? super T, ? extends Iterable<V>> function,
R target)
{
this.forEachWithOccurrences((each, occurrences) -> {
Iterable<V> values = function.valueOf(each);
Iterate.forEach(values, value -> target.putAll(value, Collections.nCopies(occurrences, each)));
});
return target;
}
@Override
public long sumOfInt(IntFunction<? super T> function)
{
long[] sum = {0L};
this.forEachWithOccurrences((each, occurrences) -> {
int intValue = function.intValueOf(each);
sum[0] += (long) intValue * (long) occurrences;
});
return sum[0];
}
@Override
public double sumOfFloat(FloatFunction<? super T> function)
{
double[] sum = {0.0d};
double[] compensation = {0.0d};
this.forEachWithOccurrences((each, occurrences) -> {
float f = function.floatValueOf(each);
for (int i = 0; i < occurrences; i++)
{
double adjustedValue = f - compensation[0];
double nextSum = sum[0] + adjustedValue;
compensation[0] = nextSum - sum[0] - adjustedValue;
sum[0] = nextSum;
}
});
return sum[0];
}
@Override
public long sumOfLong(LongFunction<? super T> function)
{
long[] sum = {0L};
this.forEachWithOccurrences((each, occurrences) -> {
long longValue = function.longValueOf(each);
sum[0] += longValue * (long) occurrences;
});
return sum[0];
}
@Override
public double sumOfDouble(DoubleFunction<? super T> function)
{
double[] sum = {0.0d};
double[] compensation = {0.0d};
this.forEachWithOccurrences((each, occurrences) -> {
double d = function.doubleValueOf(each);
for (int i = 0; i < occurrences; i++)
{
double y = d - compensation[0];
double t = sum[0] + y;
compensation[0] = t - sum[0] - y;
sum[0] = t;
}
});
return sum[0];
}
@Override
public <IV> IV injectInto(IV injectedValue, Function2<? super IV, ? super T, ? extends IV> function)
{
IV[] result = (IV[]) new Object[]{injectedValue};
this.forEachWithOccurrences((each, occurrences) -> {
for (int i = 0; i < occurrences; i++)
{
result[0] = function.value(result[0], each);
}
});
return result[0];
}
@Override
public int injectInto(int injectedValue, IntObjectToIntFunction<? super T> function)
{
int[] result = {injectedValue};
this.forEachWithOccurrences((each, occurrences) -> {
for (int i = 0; i < occurrences; i++)
{
result[0] = function.intValueOf(result[0], each);
}
});
return result[0];
}
@Override
public long injectInto(long injectedValue, LongObjectToLongFunction<? super T> function)
{
long[] result = {injectedValue};
this.forEachWithOccurrences((each, occurrences) -> {
for (int i = 0; i < occurrences; i++)
{
result[0] = function.longValueOf(result[0], each);
}
});
return result[0];
}
@Override
public double injectInto(double injectedValue, DoubleObjectToDoubleFunction<? super T> function)
{
double[] result = {injectedValue};
this.forEachWithOccurrences((each, occurrences) -> {
for (int i = 0; i < occurrences; i++)
{
result[0] = function.doubleValueOf(result[0], each);
}
});
return result[0];
}
@Override
public float injectInto(float injectedValue, FloatObjectToFloatFunction<? super T> function)
{
float[] result = {injectedValue};
this.forEachWithOccurrences((each, occurrences) -> {
for (int i = 0; i < occurrences; i++)
{
result[0] = function.floatValueOf(result[0], each);
}
});
return result[0];
}
public <IV, P> IV injectIntoWith(IV injectedValue, Function3<? super IV, ? super T, ? super P, ? extends IV> function, P parameter)
{
IV[] result = (IV[]) new Object[]{injectedValue};
this.forEachWithOccurrences((each, occurrences) -> {
for (int i = 0; i < occurrences; i++)
{
result[0] = function.value(result[0], each, parameter);
}
});
return result[0];
}
@Override
public String toStringOfItemToCount()
{
StringJoiner joiner = new StringJoiner(", ", "{", "}");
this.forEachWithOccurrences((each, occurrences) -> joiner.add(each + "=" + occurrences));
return joiner.toString();
}
protected MutableList<ObjectIntPair<T>> toListWithOccurrences()
{
MutableList<ObjectIntPair<T>> result = FastList.newList(this.sizeDistinct());
this.forEachWithOccurrences((each, count) -> result.add(PrimitiveTuples.pair(each, count)));
return result;
}
@Override
public MutableList<T> toList()
{
MutableList<T> result = FastList.newList(this.size());
this.forEachWithOccurrences((each, occurrences) -> {
for (int i = 0; i < occurrences; i++)
{
result.add(each);
}
});
return result;
}
@Override
public MutableList<T> toSortedList(Comparator<? super T> comparator)
{
MutableList<ObjectIntPair<T>> sorted = this.toListWithOccurrences().sortThis((o1, o2) -> comparator.compare(o1.getOne(), o2.getOne()));
MutableList<T> result = FastList.newList(this.size());
sorted.each(each -> {
T object = each.getOne();
int occurrences = each.getTwo();
for (int i = 0; i < occurrences; i++)
{
result.add(object);
}
});
return result;
}
@Override
public MutableSet<T> toSet()
{
MutableSet<T> result = UnifiedSet.newSet(this.sizeDistinct());
this.forEachWithOccurrences((each, occurrences) -> result.add(each));
return result;
}
@Override
public MutableSortedSet<T> toSortedSet()
{
MutableSortedSet<T> result = SortedSets.mutable.empty();
this.forEachWithOccurrences((each, occurrences) -> result.add(each));
return result;
}
@Override
public MutableSortedSet<T> toSortedSet(Comparator<? super T> comparator)
{
MutableSortedSet<T> result = SortedSets.mutable.with(comparator);
this.forEachWithOccurrences((each, occurrences) -> result.add(each));
return result;
}
@Override
public MutableBag<T> toBag()
{
MutableBag<T> result = HashBag.newBag(this.sizeDistinct());
this.forEachWithOccurrences(result::addOccurrences);
return result;
}
@Override
public MutableSortedBag<T> toSortedBag()
{
MutableSortedBag<T> result = TreeBag.newBag();
this.forEachWithOccurrences(result::addOccurrences);
return result;
}
@Override
public MutableSortedBag<T> toSortedBag(Comparator<? super T> comparator)
{
MutableSortedBag<T> result = TreeBag.newBag(comparator);
this.forEachWithOccurrences(result::addOccurrences);
return result;
}
protected MutableList<ObjectIntPair<T>> occurrencesSortingBy(int n, IntFunction<ObjectIntPair<T>> function, MutableList<ObjectIntPair<T>> returnWhenEmpty)
{
if (n < 0)
{
throw new IllegalArgumentException("Cannot use a value of n < 0");
}
if (n == 0)
{
return returnWhenEmpty;
}
int keySize = Math.min(n, this.sizeDistinct());
MutableList<ObjectIntPair<T>> sorted = this.toListWithOccurrences().sortThisByInt(function);
MutableList<ObjectIntPair<T>> results = sorted.subList(0, keySize).toList();
while (keySize < sorted.size() && results.getLast().getTwo() == sorted.get(keySize).getTwo())
{
results.add(sorted.get(keySize));
keySize++;
}
return results;
}
@Override
public SetIterable<T> selectUnique()
{
MutableSet<T> result = Sets.mutable.empty();
this.forEachWithOccurrences((each, occurrences) ->
{
if (occurrences == 1)
{
result.add(each);
}
});
return result;
}
}
AbstractBag
https://github.com/eclipse/eclipse-collections/eclipse-collections: Eclipse Collections is a collections framework for Java. It has JDK-compatible List, Set and Map
implementations with a rich API and set of utility classes that work with any JDK compatible Collections,
Arrays, Maps or Strings. The iteration protocol was inspired by the Smalltalk collection framework.