aboutsummaryrefslogtreecommitdiff
path: root/RoughlyEnoughItems-default-plugin/src/main/java
diff options
context:
space:
mode:
Diffstat (limited to 'RoughlyEnoughItems-default-plugin/src/main/java')
-rw-r--r--RoughlyEnoughItems-default-plugin/src/main/java/me/shedaniel/rei/plugin/DefaultPlugin.java12
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);