From 616bd6af839373c95d9191cf808da0446963d3eb Mon Sep 17 00:00:00 2001 From: shedaniel Date: Sun, 12 Dec 2021 23:20:43 +0800 Subject: Add entry settings adapter --- .../me/shedaniel/rei/RoughlyEnoughItemsCore.java | 2 + .../search/argument/type/ModArgumentType.java | 6 +-- .../rei/impl/common/entry/AbstractEntryStack.java | 24 ++++++---- .../settings/EntrySettingsAdapterRegistryImpl.java | 51 ++++++++++++++++++++++ 4 files changed, 72 insertions(+), 11 deletions(-) create mode 100644 runtime/src/main/java/me/shedaniel/rei/impl/common/entry/settings/EntrySettingsAdapterRegistryImpl.java (limited to 'runtime') diff --git a/runtime/src/main/java/me/shedaniel/rei/RoughlyEnoughItemsCore.java b/runtime/src/main/java/me/shedaniel/rei/RoughlyEnoughItemsCore.java index 654552028..9fc7a2b96 100644 --- a/runtime/src/main/java/me/shedaniel/rei/RoughlyEnoughItemsCore.java +++ b/runtime/src/main/java/me/shedaniel/rei/RoughlyEnoughItemsCore.java @@ -49,6 +49,7 @@ import me.shedaniel.rei.impl.common.entry.EntryStackProviderImpl; import me.shedaniel.rei.impl.common.entry.comparison.FluidComparatorRegistryImpl; import me.shedaniel.rei.impl.common.entry.comparison.ItemComparatorRegistryImpl; import me.shedaniel.rei.impl.common.entry.comparison.NbtHasherProviderImpl; +import me.shedaniel.rei.impl.common.entry.settings.EntrySettingsAdapterRegistryImpl; import me.shedaniel.rei.impl.common.entry.type.EntryTypeRegistryImpl; import me.shedaniel.rei.impl.common.fluid.FluidSupportProviderImpl; import me.shedaniel.rei.impl.common.logging.FileLogger; @@ -106,6 +107,7 @@ public class RoughlyEnoughItemsCore { ); }, new EntryTypeRegistryImpl(), + new EntrySettingsAdapterRegistryImpl(), new RecipeManagerContextImpl<>(RecipeManagerContextImpl.supplier()), new ItemComparatorRegistryImpl(), new FluidComparatorRegistryImpl(), diff --git a/runtime/src/main/java/me/shedaniel/rei/impl/client/search/argument/type/ModArgumentType.java b/runtime/src/main/java/me/shedaniel/rei/impl/client/search/argument/type/ModArgumentType.java index efea0ffac..cbcf6fd4b 100644 --- a/runtime/src/main/java/me/shedaniel/rei/impl/client/search/argument/type/ModArgumentType.java +++ b/runtime/src/main/java/me/shedaniel/rei/impl/client/search/argument/type/ModArgumentType.java @@ -64,9 +64,9 @@ public final class ModArgumentType extends ArgumentType data, EntryStack stack, String searchText, Unit filterData) { if (data.getValue() == null) { - ResourceLocation id = stack.getIdentifier(); - data.setValue(id != null ? new ModInfoPair( - id.getNamespace(), + String containingNs = stack.getContainingNamespace(); + data.setValue(containingNs != null ? new ModInfoPair( + containingNs, null ) : ModInfoPair.EMPTY); } diff --git a/runtime/src/main/java/me/shedaniel/rei/impl/common/entry/AbstractEntryStack.java b/runtime/src/main/java/me/shedaniel/rei/impl/common/entry/AbstractEntryStack.java index 1d3486971..c31b54cf9 100644 --- a/runtime/src/main/java/me/shedaniel/rei/impl/common/entry/AbstractEntryStack.java +++ b/runtime/src/main/java/me/shedaniel/rei/impl/common/entry/AbstractEntryStack.java @@ -35,6 +35,7 @@ import me.shedaniel.rei.api.client.ClientHelper; import me.shedaniel.rei.api.client.gui.Renderer; import me.shedaniel.rei.api.client.gui.widgets.Tooltip; import me.shedaniel.rei.api.common.entry.EntryStack; +import me.shedaniel.rei.api.common.entry.settings.EntrySettingsAdapterRegistry; import me.shedaniel.rei.api.common.util.EntryStacks; import me.shedaniel.rei.api.common.util.FormattingUtils; import me.shedaniel.rei.impl.client.util.CrashReportUtils; @@ -126,6 +127,12 @@ public abstract class AbstractEntryStack implements EntryStack, Renderer { return getDefinition().getIdentifier(this, getValue()); } + @Override + @Nullable + public String getContainingNamespace() { + return get(Settings.CONTAINING_NS).apply(this, getDefinition().getContainingNamespace(this, getValue())); + } + @Override public boolean isEmpty() { return getDefinition().isEmpty(this, getValue()); @@ -163,11 +170,12 @@ public abstract class AbstractEntryStack implements EntryStack, Renderer { @Override public T get(Settings settings) { - Object o = this.settings == null ? null : this.settings.get(settings.getId()); + T o = this.settings == null ? null : (T) this.settings.get(settings.getId()); + o = EntrySettingsAdapterRegistry.getInstance().adapt(this, settings, o); if (o == null) { - return settings.getDefaultValue(); + o = settings.getDefaultValue(); } - return (T) o; + return o; } @Override @@ -191,13 +199,13 @@ public abstract class AbstractEntryStack implements EntryStack, Renderer { tooltip.getValue().addAllTexts(get(Settings.TOOLTIP_APPEND_EXTRA).apply(this)); tooltip.setValue(get(Settings.TOOLTIP_PROCESSOR).apply(this, tooltip.getValue())); if (tooltip.getValue() == null) return null; - ResourceLocation location = getIdentifier(); + String containingNs = getContainingNamespace(); if (appendModName) { - if (location != null) { - ClientHelper.getInstance().appendModIdToTooltips(tooltip.getValue(), location.getNamespace()); + if (containingNs != null) { + ClientHelper.getInstance().appendModIdToTooltips(tooltip.getValue(), containingNs); } - } else { - final String modName = ClientHelper.getInstance().getModFromModId(location.getNamespace()); + } else if (containingNs != null) { + final String modName = ClientHelper.getInstance().getModFromModId(containingNs); Iterator iterator = tooltip.getValue().entries().iterator(); while (iterator.hasNext()) { Tooltip.Entry s = iterator.next(); diff --git a/runtime/src/main/java/me/shedaniel/rei/impl/common/entry/settings/EntrySettingsAdapterRegistryImpl.java b/runtime/src/main/java/me/shedaniel/rei/impl/common/entry/settings/EntrySettingsAdapterRegistryImpl.java new file mode 100644 index 000000000..11385492c --- /dev/null +++ b/runtime/src/main/java/me/shedaniel/rei/impl/common/entry/settings/EntrySettingsAdapterRegistryImpl.java @@ -0,0 +1,51 @@ +package me.shedaniel.rei.impl.common.entry.settings; + +import com.google.common.collect.Multimap; +import com.google.common.collect.Multimaps; +import it.unimi.dsi.fastutil.objects.Reference2ObjectOpenHashMap; +import me.shedaniel.rei.api.client.entry.renderer.EntryRenderer; +import me.shedaniel.rei.api.client.entry.renderer.EntryRendererProvider; +import me.shedaniel.rei.api.client.plugins.REIClientPlugin; +import me.shedaniel.rei.api.common.entry.EntryStack; +import me.shedaniel.rei.api.common.entry.settings.EntrySettingsAdapter; +import me.shedaniel.rei.api.common.entry.settings.EntrySettingsAdapterRegistry; +import me.shedaniel.rei.api.common.entry.type.EntryType; +import me.shedaniel.rei.api.common.plugins.REIPlugin; +import org.jetbrains.annotations.Nullable; + +import java.util.ArrayList; +import java.util.Collection; +import java.util.HashMap; +import java.util.Map; + +public class EntrySettingsAdapterRegistryImpl implements EntrySettingsAdapterRegistry { + private final Map, Multimap, EntrySettingsAdapter>> providers = new HashMap<>(); + + @Override + public void register(EntryType type, EntryStack.Settings settings, EntrySettingsAdapter provider) { + Multimap, EntrySettingsAdapter> multimap = this.providers.computeIfAbsent(settings, $ -> Multimaps.newMultimap(new Reference2ObjectOpenHashMap<>(), ArrayList::new)); + multimap.put(type, provider); + } + + @Override + @Nullable + public S adapt(EntryStack stack, EntryStack.Settings settings, @Nullable S value) { + Multimap, EntrySettingsAdapter> multimap = providers.get(settings); + if (multimap != null) { + for (EntrySettingsAdapter adapter : (Collection>) (Collection>) multimap.get(stack.getType())) { + value = adapter.provide(stack, settings, value); + } + } + return value; + } + + @Override + public void startReload() { + providers.clear(); + } + + @Override + public void acceptPlugin(REIPlugin plugin) { + plugin.registerEntrySettingsAdapters(this); + } +} -- cgit