40 lines
1.7 KiB
Java
40 lines
1.7 KiB
Java
|
package net.minecraft.client.searchtree;
|
||
|
|
|
||
|
|
import com.google.common.collect.ImmutableList;
|
||
|
|
import java.util.Comparator;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.function.Function;
|
||
|
|
import java.util.function.ToIntFunction;
|
||
|
|
import java.util.stream.Stream;
|
||
|
|
import net.minecraft.Util;
|
||
|
|
import net.minecraft.resources.ResourceLocation;
|
||
|
|
import net.minecraftforge.api.distmarker.Dist;
|
||
|
|
import net.minecraftforge.api.distmarker.OnlyIn;
|
||
|
|
|
||
|
|
@OnlyIn(Dist.CLIENT)
|
||
|
|
public class IdSearchTree<T> implements SearchTree<T> {
|
||
|
|
protected final Comparator<T> additionOrder;
|
||
|
|
protected final ResourceLocationSearchTree<T> resourceLocationSearchTree;
|
||
|
|
|
||
|
|
public IdSearchTree(Function<T, Stream<ResourceLocation>> p_235167_, List<T> p_235168_) {
|
||
|
|
ToIntFunction<T> tointfunction = Util.createIndexLookup(p_235168_);
|
||
|
|
this.additionOrder = Comparator.comparingInt(tointfunction);
|
||
|
|
this.resourceLocationSearchTree = ResourceLocationSearchTree.create(p_235168_, p_235167_);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public List<T> search(String p_235173_) {
|
||
|
|
int i = p_235173_.indexOf(58);
|
||
|
|
return i == -1 ? this.searchPlainText(p_235173_) : this.searchResourceLocation(p_235173_.substring(0, i).trim(), p_235173_.substring(i + 1).trim());
|
||
|
|
}
|
||
|
|
|
||
|
|
protected List<T> searchPlainText(String p_235169_) {
|
||
|
|
return this.resourceLocationSearchTree.searchPath(p_235169_);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected List<T> searchResourceLocation(String p_235170_, String p_235171_) {
|
||
|
|
List<T> list = this.resourceLocationSearchTree.searchNamespace(p_235170_);
|
||
|
|
List<T> list1 = this.resourceLocationSearchTree.searchPath(p_235171_);
|
||
|
|
return ImmutableList.copyOf(new IntersectionIterator<>(list.iterator(), list1.iterator(), this.additionOrder));
|
||
|
|
}
|
||
|
|
}
|