| /* |
| * Copyright 2000-2013 JetBrains s.r.o. |
| * |
| * 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.intellij.util; |
| |
| import com.intellij.util.containers.ContainerUtil; |
| import gnu.trove.THashSet; |
| import gnu.trove.TObjectHashingStrategy; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.HashSet; |
| import java.util.Set; |
| |
| /** |
| * @author max |
| */ |
| public class CommonProcessors { |
| public static class CollectProcessor<T> implements Processor<T> { |
| private final Collection<T> myCollection; |
| |
| public CollectProcessor(@NotNull Collection<T> collection) { |
| myCollection = collection; |
| } |
| |
| public CollectProcessor() { |
| myCollection = new ArrayList<T>(); |
| } |
| |
| @Override |
| public boolean process(T t) { |
| if (accept(t)) { |
| myCollection.add(t); |
| } |
| return true; |
| } |
| |
| protected boolean accept(T t) { |
| return true; |
| } |
| |
| @NotNull |
| public T[] toArray(@NotNull T[] a) { |
| return myCollection.toArray(a); |
| } |
| |
| @NotNull |
| public Collection<T> getResults() { |
| return myCollection; |
| } |
| } |
| |
| @NotNull |
| public static <T> Processor<T> notNullProcessor(@NotNull final Processor<T> processor) { |
| return new Processor<T>() { |
| @Override |
| public boolean process(@NotNull T t) { |
| return processor.process(t); |
| } |
| }; |
| } |
| |
| |
| public static class CollectUniquesProcessor<T> implements Processor<T> { |
| private final Set<T> myCollection; |
| |
| public CollectUniquesProcessor() { |
| myCollection = new HashSet<T>(); |
| } |
| |
| @Override |
| public boolean process(T t) { |
| myCollection.add(t); |
| return true; |
| } |
| |
| @NotNull |
| public T[] toArray(@NotNull T[] a) { |
| return myCollection.toArray(a); |
| } |
| |
| @NotNull |
| public Collection<T> getResults() { |
| return myCollection; |
| } |
| } |
| public static class UniqueProcessor<T> implements Processor<T> { |
| private final Set<T> processed; |
| private final Processor<T> myDelegate; |
| |
| public UniqueProcessor(@NotNull Processor<T> delegate) { |
| this(delegate, ContainerUtil.<T>canonicalStrategy()); |
| } |
| public UniqueProcessor(@NotNull Processor<T> delegate, @NotNull TObjectHashingStrategy<T> strategy) { |
| myDelegate = delegate; |
| processed = new THashSet<T>(strategy); |
| } |
| |
| @Override |
| public boolean process(T t) { |
| synchronized (processed) { |
| if (!processed.add(t)) { |
| return true; |
| } |
| } |
| return myDelegate.process(t); |
| } |
| } |
| |
| public abstract static class FindProcessor<T> implements Processor<T> { |
| private T myValue = null; |
| |
| public boolean isFound() { |
| return myValue != null; |
| } |
| |
| @Nullable |
| public T getFoundValue() { |
| return myValue; |
| } |
| |
| @Nullable |
| public T reset() { |
| T prev = myValue; |
| myValue = null; |
| return prev; |
| } |
| |
| @Override |
| public boolean process(T t) { |
| if (accept(t)) { |
| myValue = t; |
| return false; |
| } |
| else return true; |
| } |
| |
| protected abstract boolean accept(T t); |
| } |
| |
| public static class FindFirstProcessor<T> extends FindProcessor<T> { |
| @Override |
| protected boolean accept(T t) { |
| return true; |
| } |
| } |
| |
| public static class FindFirstAndOnlyProcessor<T> extends FindFirstProcessor<T> { |
| |
| @Override |
| public boolean process(T t) { |
| boolean firstFound = getFoundValue() != null; |
| boolean result = super.process(t); |
| if (!result) { |
| if (firstFound) reset(); |
| return !firstFound; |
| } |
| return true; |
| } |
| } |
| |
| private static final Processor FALSE = new Processor<Object>() { |
| @Override |
| public boolean process(Object t) { |
| return false; |
| } |
| }; |
| private static final Processor TRUE = new Processor<Object>() { |
| @Override |
| public boolean process(Object t) { |
| return true; |
| } |
| }; |
| |
| @NotNull |
| @SuppressWarnings({"unchecked"}) |
| public static <T> Processor<T> alwaysFalse() { |
| return FALSE; |
| } |
| |
| @NotNull |
| @SuppressWarnings({"unchecked"}) |
| public static <T> Processor<T> alwaysTrue() { |
| return TRUE; |
| } |
| } |