|
4 | 4 | import io.netty.buffer.ByteBuf; |
5 | 5 | import java.util.List; |
6 | 6 | import java.util.Locale; |
7 | | -import java.util.Optional; |
8 | 7 | import java.util.function.Consumer; |
9 | 8 | import java.util.function.IntFunction; |
10 | 9 | import mekanism.api.IIncrementalEnum; |
|
24 | 23 | import mekanism.common.registries.MekanismDataComponents; |
25 | 24 | import mekanism.common.util.StorageUtils; |
26 | 25 | import net.minecraft.core.Holder; |
27 | | -import net.minecraft.core.Holder.Reference; |
28 | | -import net.minecraft.core.HolderLookup.RegistryLookup; |
29 | 26 | import net.minecraft.core.component.DataComponentType; |
30 | 27 | import net.minecraft.network.chat.Component; |
31 | 28 | import net.minecraft.network.codec.ByteBufCodecs; |
|
45 | 42 | import net.minecraft.world.item.ItemStack; |
46 | 43 | import net.minecraft.world.item.Rarity; |
47 | 44 | import net.minecraft.world.item.TooltipFlag; |
48 | | -import net.minecraft.world.item.enchantment.Enchantment; |
49 | | -import net.minecraft.world.item.enchantment.Enchantments; |
50 | | -import net.minecraft.world.item.enchantment.ItemEnchantments; |
51 | 45 | import net.neoforged.neoforge.event.ItemAttributeModifierEvent; |
52 | 46 | import org.jetbrains.annotations.NotNull; |
53 | 47 |
|
@@ -97,35 +91,6 @@ public int getBarColor(@NotNull ItemStack stack) { |
97 | 91 | return MekanismConfig.client.energyColor.get(); |
98 | 92 | } |
99 | 93 |
|
100 | | - private int getFakeEnchantmentLevel(@NotNull ItemStack stack, @NotNull Holder<Enchantment> enchantment) { |
101 | | - if (enchantment.is(Enchantments.SOUL_SPEED) && getMode(stack) == FreeRunnerMode.NORMAL) { |
102 | | - return 1; |
103 | | - } |
104 | | - return 0; |
105 | | - } |
106 | | - |
107 | | - @Override |
108 | | - public int getEnchantmentLevel(@NotNull ItemStack stack, @NotNull Holder<Enchantment> enchantment) { |
109 | | - return Math.max(getFakeEnchantmentLevel(stack, enchantment), super.getEnchantmentLevel(stack, enchantment)); |
110 | | - } |
111 | | - |
112 | | - @NotNull |
113 | | - @Override |
114 | | - public ItemEnchantments getAllEnchantments(@NotNull ItemStack stack, @NotNull RegistryLookup<Enchantment> lookup) { |
115 | | - ItemEnchantments enchantments = super.getAllEnchantments(stack, lookup); |
116 | | - Optional<Reference<Enchantment>> enchantment = lookup.get(Enchantments.SOUL_SPEED); |
117 | | - if (enchantment.isPresent()) { |
118 | | - Holder<Enchantment> soulSpeed = enchantment.get(); |
119 | | - int fakeLevel = getFakeEnchantmentLevel(stack, soulSpeed); |
120 | | - if (enchantments.getLevel(soulSpeed) < fakeLevel) { |
121 | | - ItemEnchantments.Mutable mutable = new ItemEnchantments.Mutable(enchantments); |
122 | | - mutable.set(soulSpeed, fakeLevel); |
123 | | - return mutable.toImmutable(); |
124 | | - } |
125 | | - } |
126 | | - return enchantments; |
127 | | - } |
128 | | - |
129 | 94 | @Override |
130 | 95 | public DataComponentType<FreeRunnerMode> getModeDataType() { |
131 | 96 | return MekanismDataComponents.FREE_RUNNER_MODE.get(); |
|
0 commit comments