com.gs.collections.impl.bag.AbstractBag Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of gs-collections Show documentation
Show all versions of gs-collections Show documentation
GS 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.
/*
* Copyright 2015 Goldman Sachs.
*
* 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.gs.collections.impl.bag;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import com.gs.collections.api.bag.Bag;
import com.gs.collections.api.bag.MutableBag;
import com.gs.collections.api.bag.primitive.MutableBooleanBag;
import com.gs.collections.api.bag.primitive.MutableByteBag;
import com.gs.collections.api.bag.primitive.MutableCharBag;
import com.gs.collections.api.bag.primitive.MutableDoubleBag;
import com.gs.collections.api.bag.primitive.MutableFloatBag;
import com.gs.collections.api.bag.primitive.MutableIntBag;
import com.gs.collections.api.bag.primitive.MutableLongBag;
import com.gs.collections.api.bag.primitive.MutableShortBag;
import com.gs.collections.api.bag.sorted.MutableSortedBag;
import com.gs.collections.api.block.function.Function;
import com.gs.collections.api.block.function.Function2;
import com.gs.collections.api.block.function.Function3;
import com.gs.collections.api.block.function.primitive.BooleanFunction;
import com.gs.collections.api.block.function.primitive.ByteFunction;
import com.gs.collections.api.block.function.primitive.CharFunction;
import com.gs.collections.api.block.function.primitive.DoubleFunction;
import com.gs.collections.api.block.function.primitive.DoubleObjectToDoubleFunction;
import com.gs.collections.api.block.function.primitive.FloatFunction;
import com.gs.collections.api.block.function.primitive.FloatObjectToFloatFunction;
import com.gs.collections.api.block.function.primitive.IntFunction;
import com.gs.collections.api.block.function.primitive.IntObjectToIntFunction;
import com.gs.collections.api.block.function.primitive.LongFunction;
import com.gs.collections.api.block.function.primitive.LongObjectToLongFunction;
import com.gs.collections.api.block.function.primitive.ShortFunction;
import com.gs.collections.api.block.predicate.Predicate;
import com.gs.collections.api.block.predicate.Predicate2;
import com.gs.collections.api.block.procedure.Procedure;
import com.gs.collections.api.block.procedure.primitive.ObjectIntProcedure;
import com.gs.collections.api.collection.primitive.MutableBooleanCollection;
import com.gs.collections.api.collection.primitive.MutableByteCollection;
import com.gs.collections.api.collection.primitive.MutableCharCollection;
import com.gs.collections.api.collection.primitive.MutableDoubleCollection;
import com.gs.collections.api.collection.primitive.MutableFloatCollection;
import com.gs.collections.api.collection.primitive.MutableIntCollection;
import com.gs.collections.api.collection.primitive.MutableLongCollection;
import com.gs.collections.api.collection.primitive.MutableShortCollection;
import com.gs.collections.api.list.MutableList;
import com.gs.collections.api.multimap.MutableMultimap;
import com.gs.collections.api.set.MutableSet;
import com.gs.collections.api.set.sorted.MutableSortedSet;
import com.gs.collections.api.tuple.primitive.ObjectIntPair;
import com.gs.collections.impl.AbstractRichIterable;
import com.gs.collections.impl.Counter;
import com.gs.collections.impl.bag.mutable.HashBag;
import com.gs.collections.impl.bag.sorted.mutable.TreeBag;
import com.gs.collections.impl.factory.SortedSets;
import com.gs.collections.impl.list.mutable.FastList;
import com.gs.collections.impl.set.mutable.UnifiedSet;
import com.gs.collections.impl.tuple.primitive.PrimitiveTuples;
import com.gs.collections.impl.utility.Iterate;
/**
* @since 7.0
*/
public abstract class AbstractBag
extends AbstractRichIterable
implements Collection, Bag
{
@Override
public > R select(final Predicate predicate, final R target)
{
if (target instanceof MutableBag)
{
final MutableBag targetBag = (MutableBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (predicate.accept(each))
{
targetBag.addOccurrences(each, occurrences);
}
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (predicate.accept(each))
{
for (int i = 0; i < occurrences; i++)
{
target.add(each);
}
}
}
});
}
return target;
}
@Override
public > R selectWith(
final Predicate2 predicate,
final P parameter,
final R target)
{
if (target instanceof MutableBag)
{
final MutableBag targetBag = (MutableBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (predicate.accept(each, parameter))
{
targetBag.addOccurrences(each, occurrences);
}
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (predicate.accept(each, parameter))
{
for (int i = 0; i < occurrences; i++)
{
target.add(each);
}
}
}
});
}
return target;
}
@Override
public > R reject(final Predicate predicate, final R target)
{
if (target instanceof MutableBag)
{
final MutableBag targetBag = (MutableBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (!predicate.accept(each))
{
targetBag.addOccurrences(each, occurrences);
}
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (!predicate.accept(each))
{
for (int i = 0; i < occurrences; i++)
{
target.add(each);
}
}
}
});
}
return target;
}
@Override
public > R rejectWith(
final Predicate2 predicate,
final P parameter,
final R target)
{
if (target instanceof MutableBag)
{
final MutableBag targetBag = (MutableBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (!predicate.accept(each, parameter))
{
targetBag.addOccurrences(each, occurrences);
}
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (!predicate.accept(each, parameter))
{
for (int i = 0; i < occurrences; i++)
{
target.add(each);
}
}
}
});
}
return target;
}
@Override
public int count(final Predicate predicate)
{
final Counter result = new Counter();
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (predicate.accept(each))
{
result.add(occurrences);
}
}
});
return result.getCount();
}
@Override
public > R collect(final Function function, final R target)
{
if (target instanceof MutableBag)
{
final MutableBag targetBag = (MutableBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(function.valueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
V value = function.valueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public > R collectWith(
final Function2 function,
final P parameter,
final R target)
{
if (target instanceof MutableBag)
{
final MutableBag targetBag = (MutableBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(function.value(each, parameter), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
V value = function.value(each, parameter);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public > R collectIf(
final Predicate predicate,
final Function function,
final R target)
{
if (target instanceof MutableBag)
{
final MutableBag targetBag = (MutableBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (predicate.accept(each))
{
targetBag.addOccurrences(function.valueOf(each), occurrences);
}
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
if (predicate.accept(each))
{
V value = function.valueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
}
});
}
return target;
}
@Override
public > R flatCollect(final Function> function, final R target)
{
if (target instanceof MutableBag)
{
final MutableBag targetBag = (MutableBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, final int occurrences)
{
Iterable values = function.valueOf(each);
Iterate.forEach(values, new Procedure()
{
public void value(V each)
{
targetBag.addOccurrences(each, occurrences);
}
});
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
Iterable values = function.valueOf(each);
for (int i = 0; i < occurrences; i++)
{
Iterate.forEach(values, new Procedure()
{
public void value(V each)
{
target.add(each);
}
});
}
}
});
}
return target;
}
@Override
public R collectBoolean(final BooleanFunction booleanFunction, final R target)
{
if (target instanceof MutableBooleanBag)
{
final MutableBooleanBag targetBag = (MutableBooleanBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(booleanFunction.booleanValueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
boolean value = booleanFunction.booleanValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public R collectByte(final ByteFunction byteFunction, final R target)
{
if (target instanceof MutableByteBag)
{
final MutableByteBag targetBag = (MutableByteBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(byteFunction.byteValueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
byte value = byteFunction.byteValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public R collectChar(final CharFunction charFunction, final R target)
{
if (target instanceof MutableCharBag)
{
final MutableCharBag targetBag = (MutableCharBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(charFunction.charValueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
char value = charFunction.charValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public R collectDouble(final DoubleFunction doubleFunction, final R target)
{
if (target instanceof MutableDoubleBag)
{
final MutableDoubleBag targetBag = (MutableDoubleBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(doubleFunction.doubleValueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
double value = doubleFunction.doubleValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public R collectFloat(final FloatFunction floatFunction, final R target)
{
if (target instanceof MutableFloatBag)
{
final MutableFloatBag targetBag = (MutableFloatBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(floatFunction.floatValueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
float value = floatFunction.floatValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public R collectInt(final IntFunction intFunction, final R target)
{
if (target instanceof MutableIntBag)
{
final MutableIntBag targetBag = (MutableIntBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(intFunction.intValueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
int value = intFunction.intValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public R collectLong(final LongFunction longFunction, final R target)
{
if (target instanceof MutableLongBag)
{
final MutableLongBag targetBag = (MutableLongBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(longFunction.longValueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
long value = longFunction.longValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public R collectShort(final ShortFunction shortFunction, final R target)
{
if (target instanceof MutableShortBag)
{
final MutableShortBag targetBag = (MutableShortBag) target;
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
targetBag.addOccurrences(shortFunction.shortValueOf(each), occurrences);
}
});
}
else
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
short value = shortFunction.shortValueOf(each);
for (int i = 0; i < occurrences; i++)
{
target.add(value);
}
}
});
}
return target;
}
@Override
public > R groupBy(
final Function function,
final R target)
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
V value = function.valueOf(each);
target.putAll(value, Collections.nCopies(occurrences, each));
}
});
return target;
}
@Override
public > R groupByEach(
final Function> function,
final R target)
{
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(final T each, final int occurrences)
{
Iterable values = function.valueOf(each);
Iterate.forEach(values, new Procedure()
{
public void value(V value)
{
target.putAll(value, Collections.nCopies(occurrences, each));
}
});
}
});
return target;
}
@Override
public long sumOfInt(final IntFunction function)
{
final long[] sum = {0L};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
int intValue = function.intValueOf(each);
sum[0] += (long) intValue * (long) occurrences;
}
});
return sum[0];
}
@Override
public double sumOfFloat(final FloatFunction function)
{
final double[] sum = {0.0d};
final double[] compensation = {0.0d};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int 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(final LongFunction function)
{
final long[] sum = {0L};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
long longValue = function.longValueOf(each);
sum[0] += longValue * (long) occurrences;
}
});
return sum[0];
}
@Override
public double sumOfDouble(final DoubleFunction function)
{
final double[] sum = {0.0d};
final double[] compensation = {0.0d};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int 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 injectInto(IV injectedValue, final Function2 function)
{
final IV[] result = (IV[]) new Object[]{injectedValue};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
for (int i = 0; i < occurrences; i++)
{
result[0] = function.value(result[0], each);
}
}
});
return result[0];
}
@Override
public int injectInto(int injectedValue, final IntObjectToIntFunction function)
{
final int[] result = {injectedValue};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
for (int i = 0; i < occurrences; i++)
{
result[0] = function.intValueOf(result[0], each);
}
}
});
return result[0];
}
@Override
public long injectInto(long injectedValue, final LongObjectToLongFunction function)
{
final long[] result = {injectedValue};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
for (int i = 0; i < occurrences; i++)
{
result[0] = function.longValueOf(result[0], each);
}
}
});
return result[0];
}
@Override
public double injectInto(double injectedValue, final DoubleObjectToDoubleFunction function)
{
final double[] result = {injectedValue};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
for (int i = 0; i < occurrences; i++)
{
result[0] = function.doubleValueOf(result[0], each);
}
}
});
return result[0];
}
@Override
public float injectInto(float injectedValue, final FloatObjectToFloatFunction function)
{
final float[] result = {injectedValue};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
for (int i = 0; i < occurrences; i++)
{
result[0] = function.floatValueOf(result[0], each);
}
}
});
return result[0];
}
public IV injectIntoWith(IV injectedValue, final Function3 function, final P parameter)
{
final IV[] result = (IV[]) new Object[]{injectedValue};
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
for (int i = 0; i < occurrences; i++)
{
result[0] = function.value(result[0], each, parameter);
}
}
});
return result[0];
}
public String toStringOfItemToCount()
{
if (this.isEmpty())
{
return "{}";
}
final StringBuilder builder = new StringBuilder().append('{');
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
builder.append(each);
builder.append('=');
builder.append(occurrences);
builder.append(", ");
}
});
builder.deleteCharAt(builder.length() - 1);
builder.deleteCharAt(builder.length() - 1);
return builder.append('}').toString();
}
protected MutableList> toListWithOccurrences()
{
final MutableList> result = FastList.newList(this.sizeDistinct());
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int count)
{
result.add(PrimitiveTuples.pair(each, count));
}
});
return result;
}
@Override
public MutableList toList()
{
final MutableList result = FastList.newList(this.size());
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
for (int i = 0; i < occurrences; i++)
{
result.add(each);
}
}
});
return result;
}
@Override
public MutableList toSortedList(final Comparator comparator)
{
MutableList> sorted = this.toListWithOccurrences().sortThis(new Comparator>()
{
public int compare(ObjectIntPair o1, ObjectIntPair o2)
{
return comparator.compare(o1.getOne(), o2.getOne());
}
});
final MutableList result = FastList.newList(this.size());
sorted.forEach(new Procedure>()
{
public void value(ObjectIntPair each)
{
T object = each.getOne();
int occurrences = each.getTwo();
for (int i = 0; i < occurrences; i++)
{
result.add(object);
}
}
});
return result;
}
@Override
public MutableSet toSet()
{
final MutableSet result = UnifiedSet.newSet(this.sizeDistinct());
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
result.add(each);
}
});
return result;
}
@Override
public MutableSortedSet toSortedSet()
{
final MutableSortedSet result = SortedSets.mutable.empty();
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
result.add(each);
}
});
return result;
}
@Override
public MutableSortedSet toSortedSet(Comparator comparator)
{
final MutableSortedSet result = SortedSets.mutable.with(comparator);
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
result.add(each);
}
});
return result;
}
@Override
public MutableBag toBag()
{
final MutableBag result = HashBag.newBag(this.sizeDistinct());
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
result.addOccurrences(each, occurrences);
}
});
return result;
}
@Override
public MutableSortedBag toSortedBag()
{
final MutableSortedBag result = TreeBag.newBag();
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
result.addOccurrences(each, occurrences);
}
});
return result;
}
@Override
public MutableSortedBag toSortedBag(Comparator comparator)
{
final MutableSortedBag result = TreeBag.newBag(comparator);
this.forEachWithOccurrences(new ObjectIntProcedure()
{
public void value(T each, int occurrences)
{
result.addOccurrences(each, occurrences);
}
});
return result;
}
protected MutableList> occurrencesSortingBy(int n, IntFunction> function, MutableList> 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> sorted = this.toListWithOccurrences().sortThisByInt(function);
MutableList> 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;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy