34 lines
1.3 KiB
Java
34 lines
1.3 KiB
Java
package net.minecraft.client.searchtree;
|
|
|
|
import com.google.common.collect.ImmutableList;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.function.Function;
|
|
import java.util.stream.Stream;
|
|
import net.fabricmc.api.EnvType;
|
|
import net.fabricmc.api.Environment;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
|
|
@Environment(EnvType.CLIENT)
|
|
public class FullTextSearchTree<T> extends IdSearchTree<T> {
|
|
private final SearchTree<T> plainTextSearchTree;
|
|
|
|
public FullTextSearchTree(Function<T, Stream<String>> filter, Function<T, Stream<ResourceLocation>> idGetter, List<T> contents) {
|
|
super(idGetter, contents);
|
|
this.plainTextSearchTree = SearchTree.plainText(contents, filter);
|
|
}
|
|
|
|
@Override
|
|
protected List<T> searchPlainText(String query) {
|
|
return this.plainTextSearchTree.search(query);
|
|
}
|
|
|
|
@Override
|
|
protected List<T> searchResourceLocation(String namespace, String path) {
|
|
List<T> list = this.resourceLocationSearchTree.searchNamespace(namespace);
|
|
List<T> list2 = this.resourceLocationSearchTree.searchPath(path);
|
|
List<T> list3 = this.plainTextSearchTree.search(path);
|
|
Iterator<T> iterator = new MergingUniqueIterator<>(list2.iterator(), list3.iterator(), this.additionOrder);
|
|
return ImmutableList.copyOf(new IntersectionIterator<>(list.iterator(), iterator, this.additionOrder));
|
|
}
|
|
}
|