com.gs.collections.impl.bag.mutable.AbstractMutableBag 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
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.
There is a newer version: 7.0.3
Show newest version
/*
 * 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.mutable;

import java.util.concurrent.ExecutorService;

import com.gs.collections.api.annotation.Beta;
import com.gs.collections.api.bag.ImmutableBag;
import com.gs.collections.api.bag.MutableBag;
import com.gs.collections.api.bag.ParallelUnsortedBag;
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.block.function.Function;
import com.gs.collections.api.block.function.Function2;
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.FloatFunction;
import com.gs.collections.api.block.function.primitive.IntFunction;
import com.gs.collections.api.block.function.primitive.LongFunction;
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.ordered.OrderedIterable;
import com.gs.collections.api.partition.bag.PartitionMutableBag;
import com.gs.collections.api.set.MutableSet;
import com.gs.collections.api.tuple.Pair;
import com.gs.collections.impl.bag.mutable.primitive.BooleanHashBag;
import com.gs.collections.impl.bag.mutable.primitive.ByteHashBag;
import com.gs.collections.impl.bag.mutable.primitive.CharHashBag;
import com.gs.collections.impl.bag.mutable.primitive.DoubleHashBag;
import com.gs.collections.impl.bag.mutable.primitive.FloatHashBag;
import com.gs.collections.impl.bag.mutable.primitive.IntHashBag;
import com.gs.collections.impl.bag.mutable.primitive.LongHashBag;
import com.gs.collections.impl.bag.mutable.primitive.ShortHashBag;
import com.gs.collections.impl.factory.Bags;
import com.gs.collections.impl.lazy.parallel.bag.NonParallelUnsortedBag;
import com.gs.collections.impl.partition.bag.PartitionHashBag;
import com.gs.collections.impl.set.mutable.UnifiedSet;

public abstract class AbstractMutableBag
        extends AbstractMutableBagIterable
        implements MutableBag
{
    public ImmutableBag toImmutable()
    {
        return Bags.immutable.withAll(this);
    }

    public UnmodifiableBag asUnmodifiable()
    {
        return UnmodifiableBag.of(this);
    }

    public SynchronizedBag asSynchronized()
    {
        return new SynchronizedBag(this);
    }

    public MutableBag tap(Procedure procedure)
    {
        this.forEach(procedure);
        return this;
    }

    public  MutableBag selectInstancesOf(final Class clazz)
    {
        final MutableBag result = HashBag.newBag();
        this.forEachWithOccurrences(new ObjectIntProcedure()
        {
            public void value(T each, int occurrences)
            {
                if (clazz.isInstance(each))
                {
                    result.addOccurrences((S) each, occurrences);
                }
            }
        });
        return result;
    }

    public MutableBag select(Predicate predicate)
    {
        return this.select(predicate, this.newEmpty());
    }

    public  MutableBag selectWith(Predicate2 predicate, P parameter)
    {
        return this.selectWith(predicate, parameter, this.newEmpty());
    }

    public MutableBag reject(Predicate predicate)
    {
        return this.reject(predicate, this.newEmpty());
    }

    public 
 MutableBag rejectWith(Predicate2 predicate, P parameter)
    {
        return this.rejectWith(predicate, parameter, this.newEmpty());
    }

    public PartitionMutableBag partition(final Predicate predicate)
    {
        final PartitionMutableBag result = new PartitionHashBag();
        this.forEachWithOccurrences(new ObjectIntProcedure()
        {
            public void value(T each, int index)
            {
                MutableBag bucket = predicate.accept(each) ? result.getSelected() : result.getRejected();
                bucket.addOccurrences(each, index);
            }
        });
        return result;
    }

    public  PartitionMutableBag partitionWith(final Predicate2 predicate, final P parameter)
    {
        final PartitionMutableBag result = new PartitionHashBag();
        this.forEachWithOccurrences(new ObjectIntProcedure()
        {
            public void value(T each, int index)
            {
                MutableBag bucket = predicate.accept(each, parameter) ? result.getSelected() : result.getRejected();
                bucket.addOccurrences(each, index);
            }
        });
        return result;
    }

    public  MutableBag collect(Function function)
    {
        return this.collect(function, HashBag.newBag());
    }

    public  MutableBag collectWith(Function2 function, P parameter)
    {
        return this.collectWith(function, parameter, HashBag.newBag());
    }

    public  MutableBag collectIf(Predicate predicate, Function function)
    {
        return this.collectIf(predicate, function, HashBag.newBag());
    }

    public  MutableBag flatCollect(Function> function)
    {
        return this.flatCollect(function, HashBag.newBag());
    }

    public MutableBooleanBag collectBoolean(BooleanFunction booleanFunction)
    {
        return this.collectBoolean(booleanFunction, new BooleanHashBag());
    }

    public MutableByteBag collectByte(ByteFunction byteFunction)
    {
        return this.collectByte(byteFunction, new ByteHashBag());
    }

    public MutableCharBag collectChar(CharFunction charFunction)
    {
        return this.collectChar(charFunction, new CharHashBag());
    }

    public MutableDoubleBag collectDouble(DoubleFunction doubleFunction)
    {
        return this.collectDouble(doubleFunction, new DoubleHashBag());
    }

    public MutableFloatBag collectFloat(FloatFunction floatFunction)
    {
        return this.collectFloat(floatFunction, new FloatHashBag());
    }

    public MutableIntBag collectInt(IntFunction intFunction)
    {
        return this.collectInt(intFunction, new IntHashBag());
    }

    public MutableLongBag collectLong(LongFunction longFunction)
    {
        return this.collectLong(longFunction, new LongHashBag());
    }

    public MutableShortBag collectShort(ShortFunction shortFunction)
    {
        return this.collectShort(shortFunction, new ShortHashBag());
    }

    /**
     * @deprecated in 6.0. Use {@link OrderedIterable#zip(Iterable)} instead.
     */
    @Deprecated
    public  MutableBag> zip(Iterable that)
    {
        return this.zip(that, HashBag.>newBag());
    }

    /**
     * @deprecated in 6.0. Use {@link OrderedIterable#zipWithIndex()} instead.
     */
    @Deprecated
    public MutableSet> zipWithIndex()
    {
        return this.zipWithIndex(UnifiedSet.>newSet());
    }

    @Beta
    public ParallelUnsortedBag asParallel(ExecutorService executorService, int batchSize)
    {
        if (executorService == null)
        {
            throw new NullPointerException();
        }
        if (batchSize < 1)
        {
            throw new IllegalArgumentException();
        }
        return new NonParallelUnsortedBag(this);
    }
}