/*
* Copyright 2013 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.partition.stack;
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.list.MutableList;
import com.gs.collections.api.partition.stack.PartitionImmutableStack;
import com.gs.collections.api.partition.stack.PartitionMutableStack;
import com.gs.collections.api.stack.MutableStack;
import com.gs.collections.impl.list.mutable.FastList;
import com.gs.collections.impl.stack.mutable.ArrayStack;
public class PartitionArrayStack<T> implements PartitionMutableStack<T>
{
private final MutableList<T> selected = FastList.newList();
private final MutableList<T> rejected = FastList.newList();
public MutableStack<T> getSelected()
{
return ArrayStack.newStackFromTopToBottom(this.selected);
}
public MutableStack<T> getRejected()
{
return ArrayStack.newStackFromTopToBottom(this.rejected);
}
public PartitionImmutableStack<T> toImmutable()
{
return new PartitionImmutableStackImpl<T>(this);
}
public void add(T t)
{
throw new UnsupportedOperationException("add is no longer supported for PartitionArrayStack");
}
public static final class PartitionProcedure<T> implements Procedure<T>
{
private static final long serialVersionUID = 1L;
private final Predicate<? super T> predicate;
private final PartitionArrayStack<T> partitionMutableStack;
public PartitionProcedure(Predicate<? super T> predicate, PartitionArrayStack<T> partitionMutableStack)
{
this.predicate = predicate;
this.partitionMutableStack = partitionMutableStack;
}
public void value(T each)
{
MutableList<T> bucket = this.predicate.accept(each)
? this.partitionMutableStack.selected
: this.partitionMutableStack.rejected;
bucket.add(each);
}
}
public static final class PartitionPredicate2Procedure<T, P> implements Procedure<T>
{
private static final long serialVersionUID = 1L;
private final Predicate2<? super T, ? super P> predicate;
private final P parameter;
private final PartitionArrayStack<T> partitionMutableStack;
public PartitionPredicate2Procedure(Predicate2<? super T, ? super P> predicate, P parameter, PartitionArrayStack<T> partitionMutableStack)
{
this.predicate = predicate;
this.parameter = parameter;
this.partitionMutableStack = partitionMutableStack;
}
public void value(T each)
{
MutableList<T> bucket = this.predicate.accept(each, this.parameter)
? this.partitionMutableStack.selected
: this.partitionMutableStack.rejected;
bucket.add(each);
}
}
}