diff options
Diffstat (limited to 'RoughlyEnoughItems-default-plugin/src/main/java/me/shedaniel/rei/plugin/DefaultPlugin.java')
| -rw-r--r-- | RoughlyEnoughItems-default-plugin/src/main/java/me/shedaniel/rei/plugin/DefaultPlugin.java | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/RoughlyEnoughItems-default-plugin/src/main/java/me/shedaniel/rei/plugin/DefaultPlugin.java b/RoughlyEnoughItems-default-plugin/src/main/java/me/shedaniel/rei/plugin/DefaultPlugin.java index 010b71f9a..b8c3b7a23 100644 --- a/RoughlyEnoughItems-default-plugin/src/main/java/me/shedaniel/rei/plugin/DefaultPlugin.java +++ b/RoughlyEnoughItems-default-plugin/src/main/java/me/shedaniel/rei/plugin/DefaultPlugin.java @@ -100,6 +100,8 @@ import org.apache.logging.log4j.Logger; import org.jetbrains.annotations.ApiStatus; import java.util.*; +import java.util.function.Consumer; +import java.util.function.IntConsumer; import java.util.function.UnaryOperator; import java.util.stream.Stream; @@ -183,12 +185,18 @@ public class DefaultPlugin implements REIPluginV0, BuiltinPlugin { EntryStack stack = EntryStack.create(Items.ENCHANTED_BOOK); List<EntryStack> enchantments = new ArrayList<>(); for (Enchantment enchantment : Registry.ENCHANTMENT) { - for (int i = enchantment.getMinLevel(); i <= enchantment.getMaxLevel(); i++) { + IntConsumer consumer = level -> { Map<Enchantment, Integer> map = new HashMap<>(); - map.put(enchantment, i); + map.put(enchantment, level); ItemStack itemStack = new ItemStack(Items.ENCHANTED_BOOK); EnchantmentHelper.setEnchantments(map, itemStack); enchantments.add(EntryStack.create(itemStack).setting(EntryStack.Settings.CHECK_TAGS, EntryStack.Settings.TRUE)); + }; + if (enchantment.getMaxLevel() - enchantment.getMinLevel() >= 10) { + consumer.accept(enchantment.getMinLevel()); + consumer.accept(enchantment.getMaxLevel()); + } else { + for (int i = enchantment.getMinLevel(); i <= enchantment.getMaxLevel(); i++) consumer.accept(i); } } entryRegistry.registerEntriesAfter(stack, enchantments); |
