Wrap around plz (2.7.9.94)

Fix #66

3 files changed, 63 insertions, 23 deletions

diff --git a/src/main/java/me/shedaniel/rei/gui/widget/ItemListOverlay.java b/src/main/java/me/shedaniel/rei/gui/widget/ItemListOverlay.java
index f65e6de56..be8e99d75 100644
--- a/src/main/java/me/shedaniel/rei/gui/widget/ItemListOverlay.java
+++ b/src/main/java/me/shedaniel/rei/gui/widget/ItemListOverlay.java
@@ -12,10 +12,12 @@ import me.shedaniel.rei.client.ScreenHelper;
 import me.shedaniel.rei.client.SearchArgument;
 import net.minecraft.client.MinecraftClient;
 import net.minecraft.client.network.ClientPlayerEntity;
+import net.minecraft.client.render.GuiLighting;
 import net.minecraft.client.resource.language.I18n;
 import net.minecraft.item.Item;
 import net.minecraft.item.ItemGroup;
 import net.minecraft.item.ItemStack;
+import net.minecraft.util.ActionResult;
 import net.minecraft.util.math.MathHelper;
 import net.minecraft.util.registry.Registry;
 import org.apache.commons.lang3.StringUtils;
@@ -24,7 +26,6 @@ import java.awt.*;
 import java.util.*;
 import java.util.List;
 import java.util.stream.Collectors;
-import java.util.stream.Stream;
 
 public class ItemListOverlay extends Widget {
     
@@ -68,12 +69,13 @@ public class ItemListOverlay extends Widget {
         return "ERROR";
     }
     
-    public int getTotalSlotsPerPage() {
+    public int getFullTotalSlotsPerPage() {
         return width * height;
     }
     
     @Override
     public void render(int int_1, int int_2, float float_1) {
+        GuiLighting.disable();
         widgets.forEach(widget -> widget.render(int_1, int_2, float_1));
         ClientPlayerEntity player = minecraft.player;
         if (rectangle.contains(ClientUtils.getMouseLocation()) && ClientHelper.isCheating() && !player.inventory.getCursorStack().isEmpty() && minecraft.isInSingleplayer())
@@ -86,19 +88,24 @@ public class ItemListOverlay extends Widget {
         this.widgets = Lists.newLinkedList();
         calculateListSize(rectangle);
         currentDisplayed = processSearchTerm(searchTerm, RoughlyEnoughItemsCore.getItemRegisterer().getItemList(), ScreenHelper.inventoryStacks);
-        double startX = rectangle.getCenterX() - width * 9;
-        double startY = rectangle.getCenterY() - height * 9;
+        int startX = (int) rectangle.getCenterX() - width * 9;
+        int startY = (int) rectangle.getCenterY() - height * 9;
         this.listArea = new Rectangle((int) startX, (int) startY, width * 18, height * 18);
-        for(int i = 0; i < getTotalSlotsPerPage(); i++) {
-            int j = i + page * getTotalSlotsPerPage();
+        int fitSlotsPerPage = getTotalFitSlotsPerPage(listArea.x, listArea.y, listArea);
+        int j = page * fitSlotsPerPage;
+        for(int i = 0; i < getFullTotalSlotsPerPage(); i++) {
             if (j >= currentDisplayed.size())
                 break;
-            widgets.add(new ItemSlotWidget((int) (startX + (i % width) * 18), (int) (startY + MathHelper.floor(i / width) * 18), Collections.singletonList(currentDisplayed.get(j)), false, true, true) {
+            int x = startX + (i % width) * 18, y = startY + MathHelper.floor(i / width) * 18;
+            if (!canBeFit(x, y, listArea))
+                continue;
+            j++;
+            widgets.add(new ItemSlotWidget(x, y, Collections.singletonList(currentDisplayed.get(j)), false, true, true) {
                 @Override
-                protected void drawToolTip(ItemStack itemStack, float delta) {
+                protected void queueTooltip(ItemStack itemStack, float delta) {
                     ClientPlayerEntity player = minecraft.player;
                     if (!ClientHelper.isCheating() || player.inventory.getCursorStack().isEmpty())
-                        super.drawToolTip(itemStack, delta);
+                        super.queueTooltip(itemStack, delta);
                 }
                 
                 @Override
@@ -126,6 +133,34 @@ public class ItemListOverlay extends Widget {
         }
     }
     
+    public int getTotalPage() {
+        int fitSlotsPerPage = getTotalFitSlotsPerPage(listArea.x, listArea.y, listArea);
+        if (fitSlotsPerPage > 0)
+            return MathHelper.ceil(getCurrentDisplayed().size() / fitSlotsPerPage);
+        return 0;
+    }
+    
+    public int getTotalFitSlotsPerPage(int startX, int startY, Rectangle listArea) {
+        int slots = 0;
+        for(int i = 0; i < getFullTotalSlotsPerPage(); i++)
+            if (canBeFit(startX + (i % width) * 18, startY + MathHelper.floor(i / width) * 18, listArea))
+                slots++;
+        return slots;
+    }
+    
+    public boolean canBeFit(int left, int top, Rectangle listArea) {
+        List<DisplayHelper.DisplayBoundsHandler> sortedBoundsHandlers = RoughlyEnoughItemsCore.getDisplayHelper().getSortedBoundsHandlers(minecraft.currentScreen.getClass());
+        ActionResult result = ActionResult.SUCCESS;
+        for(DisplayHelper.DisplayBoundsHandler sortedBoundsHandler : sortedBoundsHandlers) {
+            ActionResult fit = sortedBoundsHandler.canItemSlotWidgetFit(!RoughlyEnoughItemsCore.getConfigManager().getConfig().mirrorItemPanel, left, top, minecraft.currentScreen, listArea);
+            if (fit != ActionResult.PASS) {
+                result = fit;
+                break;
+            }
+        }
+        return result == ActionResult.SUCCESS;
+    }
+    
     @Override
     public boolean keyPressed(int int_1, int int_2, int int_3) {
         for(Widget widget : widgets)
@@ -196,10 +231,10 @@ public class ItemListOverlay extends Widget {
     }
     
     private boolean filterItem(ItemStack itemStack, List<SearchArgument> arguments) {
-        String mod = ClientHelper.getModFromItemStack(itemStack);
+        String mod = ClientHelper.getModFromItem(itemStack.getItem());
         List<String> toolTipsList = tryGetItemStackToolTip(itemStack);
         String toolTipsMixed = toolTipsList.stream().skip(1).collect(Collectors.joining()).toLowerCase();
-        String allMixed = Stream.of(tryGetItemStackName(itemStack), toolTipsMixed).collect(Collectors.joining()).toLowerCase();
+        String allMixed = toolTipsList.stream().collect(Collectors.joining()).toLowerCase();
         for(SearchArgument searchArgument : arguments.stream().filter(searchArgument -> !searchArgument.isInclude()).collect(Collectors.toList())) {
             if (searchArgument.getArgumentType().equals(SearchArgument.ArgumentType.MOD))
                 if (mod.toLowerCase().contains(searchArgument.getText().toLowerCase()))
diff --git a/src/main/java/me/shedaniel/rei/gui/widget/ItemSlotWidget.java b/src/main/java/me/shedaniel/rei/gui/widget/ItemSlotWidget.java
index 44895525e..58a93cefd 100644
--- a/src/main/java/me/shedaniel/rei/gui/widget/ItemSlotWidget.java
+++ b/src/main/java/me/shedaniel/rei/gui/widget/ItemSlotWidget.java
@@ -65,7 +65,8 @@ public class ItemSlotWidget extends HighlightableWidget {
             minecraft.getTextureManager().bindTexture(RECIPE_GUI);
             blit(this.x - 1, this.y - 1, 0, 222, 18, 18);
         }
-        if (drawHighlightedBackground && isHighlighted(mouseX, mouseY)) {
+        boolean highlighted = isHighlighted(mouseX, mouseY);
+        if (drawHighlightedBackground && highlighted) {
             GlStateManager.disableLighting();
             GlStateManager.disableDepthTest();
             GlStateManager.colorMask(true, true, true, false);
@@ -75,7 +76,7 @@ public class ItemSlotWidget extends HighlightableWidget {
             GlStateManager.enableDepthTest();
         }
         if (!itemStack.isEmpty()) {
-            if (RoughlyEnoughItemsCore.getConfigManager().getConfig().aprilFoolsFish2019 && !isHighlighted(mouseX, mouseY))
+            if (RoughlyEnoughItemsCore.getConfigManager().getConfig().aprilFoolsFish2019 && !highlighted)
                 itemStack = Items.TROPICAL_FISH.getDefaultStack();
             GuiLighting.enableForItems();
             ItemRenderer itemRenderer = minecraft.getItemRenderer();
@@ -84,27 +85,31 @@ public class ItemSlotWidget extends HighlightableWidget {
             itemRenderer.renderGuiItemOverlay(font, itemStack, x, y, getItemCountOverlay(itemStack));
             itemRenderer.zOffset = 0.0F;
         }
-        if (!itemStack.isEmpty() && isHighlighted(mouseX, mouseY) && showToolTips)
-            drawToolTip(itemStack, delta);
+        if (!itemStack.isEmpty() && highlighted && showToolTips)
+            queueTooltip(itemStack, delta);
     }
     
-    protected void drawToolTip(ItemStack itemStack, float delta) {
+    protected void queueTooltip(ItemStack itemStack, float delta) {
         ScreenHelper.getLastOverlay().addTooltip(QueuedTooltip.create(getTooltip(itemStack)));
     }
     
     protected List<String> getTooltip(ItemStack itemStack) {
-        final String modString = "§9§o" + ClientHelper.getModFromItemStack(itemStack);
+        final String modString = ClientHelper.getFormattedModFromItem(itemStack.getItem());
         List<String> toolTip = Lists.newArrayList(ItemListOverlay.tryGetItemStackToolTip(itemStack));
         toolTip.addAll(getExtraToolTips(itemStack));
+        boolean alreadyHasMod = false;
         for(String s : Lists.newArrayList(toolTip))
-            if (s.equalsIgnoreCase(modString))
-                toolTip.remove(s);
-        toolTip.add(modString);
+            if (s.equalsIgnoreCase(modString)) {
+                alreadyHasMod = true;
+                break;
+            }
+        if (!alreadyHasMod)
+            toolTip.add(modString);
         return toolTip;
     }
     
     protected List<String> getExtraToolTips(ItemStack stack) {
-        return Lists.newArrayList();
+        return Collections.emptyList();
     }
     
     protected String getItemCountOverlay(ItemStack currentStack) {
diff --git a/src/main/java/me/shedaniel/rei/gui/widget/QueuedTooltip.java b/src/main/java/me/shedaniel/rei/gui/widget/QueuedTooltip.java
index 8d8300c00..96e805d8d 100644
--- a/src/main/java/me/shedaniel/rei/gui/widget/QueuedTooltip.java
+++ b/src/main/java/me/shedaniel/rei/gui/widget/QueuedTooltip.java
@@ -5,7 +5,7 @@ import com.google.common.collect.Lists;
 import me.shedaniel.cloth.api.ClientUtils;
 
 import java.awt.*;
-import java.util.LinkedList;
+import java.util.Collections;
 import java.util.List;
 
 public class QueuedTooltip {
@@ -15,7 +15,7 @@ public class QueuedTooltip {
     
     private QueuedTooltip(Point location, List<String> text) {
         this.location = location;
-        this.text = new LinkedList<>(text);
+        this.text = Collections.unmodifiableList(text);
     }
     
     public static QueuedTooltip create(Point location, List<String> text) {