5.x - 20w30a

Signed-off-by: shedaniel <daniel@shedaniel.me>

1 files changed, 408 insertions, 0 deletions

diff --git a/RoughlyEnoughItems-runtime/src/main/java/me/shedaniel/rei/impl/ItemEntryStack.java b/RoughlyEnoughItems-runtime/src/main/java/me/shedaniel/rei/impl/ItemEntryStack.java
new file mode 100644
index 000000000..b5b684093
--- /dev/null
+++ b/RoughlyEnoughItems-runtime/src/main/java/me/shedaniel/rei/impl/ItemEntryStack.java
@@ -0,0 +1,408 @@
+/*
+ * This file is licensed under the MIT License, part of Roughly Enough Items.
+ * Copyright (c) 2018, 2019, 2020 shedaniel
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+package me.shedaniel.rei.impl;
+
+import com.google.common.collect.Lists;
+import com.mojang.blaze3d.platform.GlStateManager;
+import com.mojang.blaze3d.systems.RenderSystem;
+import me.shedaniel.math.Point;
+import me.shedaniel.math.Rectangle;
+import me.shedaniel.rei.api.ClientHelper;
+import me.shedaniel.rei.api.ConfigObject;
+import me.shedaniel.rei.api.EntryStack;
+import me.shedaniel.rei.api.fractions.Fraction;
+import me.shedaniel.rei.api.widgets.Tooltip;
+import net.minecraft.client.MinecraftClient;
+import net.minecraft.client.item.TooltipContext;
+import net.minecraft.client.render.DiffuseLighting;
+import net.minecraft.client.render.OverlayTexture;
+import net.minecraft.client.render.VertexConsumerProvider;
+import net.minecraft.client.render.model.BakedModel;
+import net.minecraft.client.render.model.json.ModelTransformation;
+import net.minecraft.client.texture.SpriteAtlasTexture;
+import net.minecraft.client.util.math.MatrixStack;
+import net.minecraft.item.Item;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.CompoundTag;
+import net.minecraft.nbt.ListTag;
+import net.minecraft.nbt.Tag;
+import net.minecraft.text.LiteralText;
+import net.minecraft.text.Text;
+import net.minecraft.text.TranslatableText;
+import net.minecraft.util.Identifier;
+import net.minecraft.util.math.MathHelper;
+import net.minecraft.util.registry.Registry;
+import org.jetbrains.annotations.ApiStatus;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+
+@ApiStatus.Internal
+public class ItemEntryStack extends AbstractEntryStack implements OptimalEntryStack {
+    private ItemStack itemStack;
+    
+    public ItemEntryStack(ItemStack itemStack) {
+        this.itemStack = itemStack;
+    }
+    
+    @Override
+    public Optional<Identifier> getIdentifier() {
+        return Optional.ofNullable(Registry.ITEM.getId(getItem()));
+    }
+    
+    @Override
+    public Type getType() {
+        return Type.ITEM;
+    }
+    
+    @Override
+    public int getAmount() {
+        return itemStack.getCount();
+    }
+    
+    @Override
+    public Fraction getAccurateAmount() {
+        return Fraction.ofWhole(itemStack.getCount());
+    }
+    
+    @Override
+    public void setAmount(Fraction amount) {
+        itemStack.setCount(amount.intValue());
+    }
+    
+    @Override
+    public void setFloatingAmount(double amount) {
+        itemStack.setCount(MathHelper.floor(amount));
+    }
+    
+    @Override
+    public boolean isEmpty() {
+        return itemStack.isEmpty();
+    }
+    
+    @Override
+    public EntryStack copy() {
+        EntryStack stack = EntryStack.create(getItemStack().copy());
+        for (Map.Entry<Settings<?>, Object> entry : getSettings().entrySet()) {
+            stack.setting((Settings<? super Object>) entry.getKey(), entry.getValue());
+        }
+        return stack;
+    }
+    
+    @Override
+    public Object getObject() {
+        return itemStack;
+    }
+    
+    /**
+     * type:
+     * 0: ignore tags and amount
+     * 1: ignore tags
+     * 2: ignore amount
+     * 3: all
+     */
+    private Boolean compareIfFluid(EntryStack stack, int type) {
+        EntryStack fluid = EntryStack.copyItemToFluid(this);
+        if (fluid.isEmpty()) return null;
+        if (stack.getType() == Type.ITEM)
+            stack = EntryStack.copyItemToFluid(stack);
+        if (stack.isEmpty()) return null;
+        switch (type) {
+            case 0:
+                return fluid.equalsIgnoreTagsAndAmount(stack);
+            case 1:
+                return fluid.equalsIgnoreTags(stack);
+            case 2:
+                return fluid.equalsIgnoreAmount(stack);
+            case 3:
+                return fluid.equalsAll(stack);
+        }
+        return null;
+    }
+    
+    @Override
+    public boolean equalsIgnoreTagsAndAmount(EntryStack stack) {
+        Boolean ifFluid = compareIfFluid(stack, 0);
+        if (ifFluid != null) return ifFluid;
+        if (stack.getType() != Type.ITEM)
+            return false;
+        return itemStack.getItem() == stack.getItem();
+    }
+    
+    @Override
+    public boolean equalsAll(EntryStack stack) {
+        if (stack.getType() != Type.ITEM)
+            return false;
+        return itemStack.getItem() == stack.getItem() && getAmount() != stack.getAmount() && ItemStack.areTagsEqual(itemStack, stack.getItemStack());
+    }
+    
+    @Override
+    public boolean equalsIgnoreAmount(EntryStack stack) {
+        Boolean ifFluid = compareIfFluid(stack, 2);
+        if (ifFluid != null) return ifFluid;
+        if (stack.getType() != Type.ITEM)
+            return false;
+        if (itemStack.getItem() != stack.getItem())
+            return false;
+        ItemStack otherStack = stack.getItemStack();
+        CompoundTag o1 = itemStack.getTag();
+        CompoundTag o2 = otherStack.getTag();
+        return o1 == o2 || ((o1 != null && o2 != null) && equalsTagWithoutCount(o1, o2));
+    }
+    
+    private boolean equalsTagWithoutCount(CompoundTag o1, CompoundTag o2) {
+        int o1Size = 0;
+        int o2Size = 0;
+        for (String key : o1.getKeys()) {
+            if (key.equals("Count"))
+                continue;
+            o1Size++;
+        }
+        for (String key : o2.getKeys()) {
+            if (key.equals("Count"))
+                continue;
+            o2Size++;
+            if (o2Size > o1Size)
+                return false;
+        }
+        if (o1Size != o2Size)
+            return false;
+        
+        try {
+            for (String key : o1.getKeys()) {
+                if (key.equals("Count"))
+                    continue;
+                Tag value = o1.get(key);
+                Tag otherValue = o2.get(key);
+                if (!equalsTag(value, otherValue))
+                    return false;
+            }
+        } catch (ClassCastException | NullPointerException unused) {
+            return false;
+        }
+        
+        return true;
+    }
+    
+    private boolean equalsTag(Tag tag, Tag otherTag) {
+        if (tag == null || otherTag == null) {
+            return tag == otherTag;
+        }
+        if (tag instanceof ListTag && otherTag instanceof ListTag)
+            return equalsList((ListTag) tag, (ListTag) otherTag);
+        return tag.equals(otherTag);
+    }
+    
+    private boolean equalsList(ListTag listTag, ListTag otherListTag) {
+        if (listTag.size() != otherListTag.size())
+            return false;
+        for (int i = 0; i < listTag.size(); i++) {
+            Tag value = listTag.get(i);
+            Tag otherValue = otherListTag.get(i);
+            if (!equalsTag(value, otherValue))
+                return false;
+        }
+        return true;
+    }
+    
+    @Override
+    public boolean equalsIgnoreTags(EntryStack stack) {
+        Boolean ifFluid = compareIfFluid(stack, 1);
+        if (ifFluid != null) return ifFluid;
+        if (stack.getType() != Type.ITEM)
+            return false;
+        if (itemStack.getItem() != stack.getItem())
+            return false;
+        return getAmount() == stack.getAmount();
+    }
+    
+    @Override
+    public int hashOfAll() {
+        int result = hashIgnoreAmount();
+        result = 31 * result + itemStack.getCount();
+        return result;
+    }
+    
+    @Override
+    public int hashIgnoreTags() {
+        int result = hashIgnoreAmountAndTags();
+        result = 31 * result + itemStack.getCount();
+        return result;
+    }
+    
+    @Override
+    public int hashIgnoreAmount() {
+        int result = hashIgnoreAmountAndTags();
+        if (itemStack.hasTag()) {
+            result = 31 * result + itemStack.getTag().toString().hashCode();
+        } else {
+            result = 31 * result;
+        }
+        return result;
+    }
+    
+    @Override
+    public int hashIgnoreAmountAndTags() {
+        int result = 1;
+        result = 31 * result + itemStack.getItem().hashCode();
+        return result;
+    }
+    
+    @Nullable
+    @Override
+    public Tooltip getTooltip(Point point) {
+        if (isEmpty() || !get(Settings.TOOLTIP_ENABLED).get())
+            return null;
+        List<Text> toolTip = tryGetItemStackToolTip(true);
+        toolTip.addAll(get(Settings.TOOLTIP_APPEND_EXTRA).apply(this));
+        if (get(Settings.TOOLTIP_APPEND_MOD).get() && ConfigObject.getInstance().shouldAppendModNames()) {
+            final String modId = ClientHelper.getInstance().getModFromItem(getItem());
+            boolean alreadyHasMod = false;
+            for (Text s : toolTip)
+                if (s.getString().equalsIgnoreCase(modId)) {
+                    alreadyHasMod = true;
+                    break;
+                }
+            if (!alreadyHasMod)
+                toolTip.add(ClientHelper.getInstance().getFormattedModFromItem(getItem()));
+        }
+        return Tooltip.create(toolTip);
+    }
+    
+    @Override
+    public void render(MatrixStack matrices, Rectangle bounds, int mouseX, int mouseY, float delta) {
+        optimisedRenderStart(matrices, delta);
+        VertexConsumerProvider.Immediate immediate = MinecraftClient.getInstance().getBufferBuilders().getEntityVertexConsumers();
+        optimisedRenderBase(matrices, immediate, bounds, mouseX, mouseY, delta);
+        immediate.draw();
+        optimisedRenderOverlay(matrices, bounds, mouseX, mouseY, delta);
+        optimisedRenderEnd(matrices, delta);
+    }
+    
+    @Override
+    public void optimisedRenderStart(MatrixStack matrices, float delta) {
+        optimisedRenderStart(matrices, delta, true);
+    }
+    
+    @SuppressWarnings("deprecation")
+    public void optimisedRenderStart(MatrixStack matrices, float delta, boolean isOptimised) {
+        MinecraftClient.getInstance().getTextureManager().bindTexture(SpriteAtlasTexture.BLOCK_ATLAS_TEX);
+        MinecraftClient.getInstance().getTextureManager().getTexture(SpriteAtlasTexture.BLOCK_ATLAS_TEX).setFilter(false, false);
+        RenderSystem.pushMatrix();
+        RenderSystem.enableRescaleNormal();
+        RenderSystem.enableAlphaTest();
+        RenderSystem.defaultAlphaFunc();
+        RenderSystem.enableBlend();
+        RenderSystem.blendFunc(GlStateManager.SrcFactor.SRC_ALPHA, GlStateManager.DstFactor.ONE_MINUS_SRC_ALPHA);
+        RenderSystem.color4f(1.0F, 1.0F, 1.0F, 1.0F);
+        if (isOptimised) {
+            boolean sideLit = getModelFromStack(itemStack).isSideLit();
+            if (!sideLit)
+                DiffuseLighting.disableGuiDepthLighting();
+        }
+    }
+    
+    @Override
+    public void optimisedRenderEnd(MatrixStack matrices, float delta) {
+        optimisedRenderEnd(matrices, delta, true);
+    }
+    
+    @SuppressWarnings("deprecation")
+    public void optimisedRenderEnd(MatrixStack matrices, float delta, boolean isOptimised) {
+        RenderSystem.enableDepthTest();
+        RenderSystem.disableAlphaTest();
+        RenderSystem.disableRescaleNormal();
+        if (isOptimised) {
+            boolean sideLit = getModelFromStack(itemStack).isSideLit();
+            if (!sideLit)
+                DiffuseLighting.enableGuiDepthLighting();
+        }
+        RenderSystem.popMatrix();
+    }
+    
+    private BakedModel getModelFromStack(ItemStack stack) {
+        return MinecraftClient.getInstance().getItemRenderer().getHeldItemModel(stack, null, null);
+    }
+    
+    @Override
+    public int groupingHash() {
+        return 1738923 + (getModelFromStack(itemStack).isSideLit() ? 1 : 0);
+    }
+    
+    @Override
+    public void optimisedRenderBase(MatrixStack matrices, VertexConsumerProvider.Immediate immediate, Rectangle bounds, int mouseX, int mouseY, float delta) {
+        if (!isEmpty() && get(Settings.RENDER).get()) {
+            ItemStack stack = getItemStack();
+            ((ItemStackHook) (Object) stack).rei_setRenderEnchantmentGlint(get(Settings.Item.RENDER_ENCHANTMENT_GLINT).get());
+            matrices.push();
+            matrices.translate(bounds.getCenterX(), bounds.getCenterY(), 100.0F + getZ());
+            matrices.scale(bounds.getWidth(), (bounds.getWidth() + bounds.getHeight()) / -2f, bounds.getHeight());
+            MinecraftClient.getInstance().getItemRenderer().renderItem(stack, ModelTransformation.Mode.GUI, false, matrices, immediate, 15728880, OverlayTexture.DEFAULT_UV, getModelFromStack(stack));
+            matrices.pop();
+            ((ItemStackHook) (Object) stack).rei_setRenderEnchantmentGlint(false);
+        }
+    }
+    
+    @Override
+    public void optimisedRenderOverlay(MatrixStack matrices, Rectangle bounds, int mouseX, int mouseY, float delta) {
+        if (!isEmpty() && get(Settings.RENDER).get()) {
+            MinecraftClient.getInstance().getItemRenderer().zOffset = getZ();
+            MinecraftClient.getInstance().getItemRenderer().renderGuiItemOverlay(MinecraftClient.getInstance().textRenderer, getItemStack(), bounds.x, bounds.y, get(Settings.RENDER_COUNTS).get() ? get(Settings.COUNTS).apply(this) : "");
+            MinecraftClient.getInstance().getItemRenderer().zOffset = 0.0F;
+        }
+    }
+    
+    private static final List<Item> SEARCH_BLACKLISTED = Lists.newArrayList();
+    
+    @Override
+    public @NotNull Text asFormattedText() {
+        if (!SEARCH_BLACKLISTED.contains(getItem()))
+            try {
+                return getItemStack().getName();
+            } catch (Throwable e) {
+                e.printStackTrace();
+                SEARCH_BLACKLISTED.add(getItem());
+            }
+        try {
+            return new TranslatableText("item." + Registry.ITEM.getId(getItem()).toString().replace(":", "."));
+        } catch (Throwable e) {
+            e.printStackTrace();
+        }
+        return new LiteralText("ERROR");
+    }
+    
+    private List<Text> tryGetItemStackToolTip(boolean careAboutAdvanced) {
+        if (!SEARCH_BLACKLISTED.contains(getItem()))
+            try {
+                return itemStack.getTooltip(MinecraftClient.getInstance().player, MinecraftClient.getInstance().options.advancedItemTooltips && careAboutAdvanced ? TooltipContext.Default.ADVANCED : TooltipContext.Default.NORMAL);
+            } catch (Throwable e) {
+                e.printStackTrace();
+                SEARCH_BLACKLISTED.add(getItem());
+            }
+        return Lists.newArrayList(asFormattedText());
+    }
+}