RoughlyEnoughItems/src/main/java/me/shedaniel/gui/RecipeGui.java

package me.shedaniel.gui;

import com.mojang.blaze3d.platform.GlStateManager;
import me.shedaniel.api.IDisplayCategory;
import me.shedaniel.api.IRecipe;
import me.shedaniel.gui.widget.Button;
import me.shedaniel.gui.widget.Control;
import me.shedaniel.gui.widget.REISlot;
import me.shedaniel.gui.widget.Tab;
import me.shedaniel.impl.REIRecipeManager;
import net.minecraft.client.MinecraftClient;
import net.minecraft.client.gui.ContainerGui;
import net.minecraft.client.gui.Gui;
import net.minecraft.client.render.GuiLighting;
import net.minecraft.client.util.Window;
import net.minecraft.container.Container;
import net.minecraft.util.Identifier;
import net.minecraft.util.math.MathHelper;

import java.util.*;

public class RecipeGui extends ContainerGui {
    
    private static final Identifier CREATIVE_INVENTORY_TABS = new Identifier("textures/gui/container/creative_inventory/tabs.png");
    private static final Identifier CHEST_GUI_TEXTURE = new Identifier("roughlyenoughitems", "textures/gui/recipecontainer.png");
    private final Window mainWindow;
    private final Container container;
    private final Gui prevScreen;
    public final Map<IDisplayCategory, List<IRecipe>> recipes;
    public final int guiWidth = 176;
    public final int guiHeight = 222;
    ArrayList<IDisplayCategory> categories = new ArrayList<>();
    private int categoryTabPage = 0;
    public IDisplayCategory selectedCategory;
    private int recipePointer = 0;
    public List<REISlot> slots;
    List<Control> controls = new LinkedList<>();
    private List<Tab> tabs;
    private boolean tabsEnabled = false;
    private Button btnCategoryPageLeft, btnCategoryPageRight;
    public Button btnRecipeLeft, btnRecipeRight;
    
    public RecipeGui(Container p_i1072_1_, Gui prevScreen, Map<IDisplayCategory, List<IRecipe>> recipes) {
        super(new RecipeContainer());
        this.container = p_i1072_1_;
        this.prevScreen = prevScreen;
        this.recipes = recipes;
        this.client = MinecraftClient.getInstance();
        this.itemRenderer = client.getItemRenderer();
        this.fontRenderer = client.fontRenderer;
        this.mainWindow = client.window;
        
        setupCategories();
    }
    
    private void setupCategories() {
        for(IDisplayCategory adapter : REIRecipeManager.instance().getDisplayAdapters())
            if (recipes.containsKey(adapter))
                categories.add(adapter);
        selectedCategory = categories.get(0);
        categoryTabPage = 0;
        tabs = new ArrayList<>();
        for(int i = 0; i < 6; i++)
            tabs.add(new Tab(i, 0, 0, 0, 28, 32));
        tabs.forEach(tab -> tab.setOnClick(i -> {
            if (tab.getId() + categoryTabPage * 6 == categories.indexOf(selectedCategory))
                return false;
            selectedCategory = categories.get(tab.getId() + categoryTabPage * 6);
            recipePointer = 0;
            updateRecipe();
            return false;
        }));
        updateRecipe();
    }
    
    @Override
    public void draw(int mouseX, int mouseY, float partialTicks) {
        drawBackground();
        super.draw(mouseX, mouseY, partialTicks);
        int y = (int) ((mainWindow.getScaledHeight() / 2 - this.guiHeight / 2));
        drawStringCentered(this.fontRenderer, selectedCategory.getDisplayName(), left + guiWidth / 2, y + 11, -1);
        drawStringCentered(this.fontRenderer, String.format("%d/%d", 1 + getCurrentPage(), getTotalPages()), left + guiWidth / 2, y + 34, -1);
        controls.forEach(Control::draw);
    }
    
    private int getCurrentPage() {
        return recipePointer / 2;
    }
    
    @Override
    public void update() {
        super.update();
        slots.forEach(REISlot::tick);
        controls.forEach(Control::tick);
    }
    
    @Override
    public void onScaleChanged(MinecraftClient p_onResize_1_, int p_onResize_2_, int p_onResize_3_) {
        super.onScaleChanged(p_onResize_1_, p_onResize_2_, p_onResize_3_);
        updateRecipe();
    }
    
    private void updateRecipe() {
        int categoryPointer = categories.indexOf(selectedCategory);
        
        IRecipe recipe = recipes.get(categories.get(categoryPointer)).get(recipePointer);
        selectedCategory.resetRecipes();
        selectedCategory.addRecipe(recipe);
        slots = selectedCategory.setupDisplay(0);
        if (recipes.get(selectedCategory).size() >= recipePointer + 2) {
            IRecipe recipe2 = recipes.get(selectedCategory).get(recipePointer + 1);
            selectedCategory.addRecipe(recipe2);
            slots.addAll(selectedCategory.setupDisplay(1));
        }
        
        left = (int) ((mainWindow.getScaledWidth() / 2 - this.guiWidth / 2));
        top = (int) ((mainWindow.getScaledHeight() / 2 - this.guiHeight / 2));
        
        slots.forEach(reiSlot -> reiSlot.move(left, top));
        
        Button btnCategoryLeft = new Button(left + 10, top + 5, 15, 20, "<");
        Button btnCategoryRight = new Button(left + guiWidth - 25, top + 5, 15, 20, ">");
        btnCategoryRight.onClick = this::btnCategoryRight;
        btnCategoryLeft.onClick = this::btnCategoryLeft;
        
        btnRecipeLeft = new Button(left + 10, top + 28, 15, 20, "<");
        btnRecipeRight = new Button(left + guiWidth - 25, top + 28, 15, 20, ">");
        btnRecipeLeft.setEnabled(recipes.get(selectedCategory).size() > 2);
        btnRecipeRight.setEnabled(recipes.get(selectedCategory).size() > 2);
        btnRecipeRight.onClick = this::btnRecipeRight;
        btnRecipeLeft.onClick = this::btnRecipeLeft;
        
        controls.clear();
        controls.add(btnCategoryLeft);
        controls.add(btnCategoryRight);
        if (categories.size() <= 1) {
            btnCategoryLeft.setEnabled(false);
            btnCategoryRight.setEnabled(false);
        } else if (categories.size() > 6) {
            btnCategoryPageLeft = new Button(left, top - 52, 20, 20, "<");
            btnCategoryPageRight = new Button(left + guiWidth - 20, top - 52, 20, 20, ">");
            btnCategoryPageLeft.setOnClick(i -> {
                categoryTabPage--;
                if (categoryTabPage <= 0)
                    categoryTabPage = MathHelper.ceil(categories.size() / 6d);
                updateRecipe();
                return true;
            });
            btnCategoryPageRight.setOnClick(i -> {
                categoryTabPage++;
                if (categoryTabPage >= MathHelper.ceil(categories.size() / 6d))
                    categoryTabPage = 0;
                updateRecipe();
                return true;
            });
            if (top - 52 >= 2)
                controls.addAll(Arrays.asList(btnCategoryPageLeft, btnCategoryPageRight));
        }
        
        controls.add(btnRecipeLeft);
        controls.add(btnRecipeRight);
        
        List<Control> newControls = new LinkedList<>();
        categories.get(categoryPointer).addWidget(newControls, 0);
        if (recipes.get(categories.get(categoryPointer)).size() >= recipePointer + 2)
            categories.get(categoryPointer).addWidget(newControls, 1);
        newControls.forEach(f -> f.move(left, top));
        controls.addAll(newControls);
        
        updateTabs();
    }
    
    private void updateTabs() {
        tabsEnabled = top - 28 > 4;
        if (tabsEnabled) {
            tabs.forEach(tab -> tab.moveTo(left + 4, left + 2 + tabs.indexOf(tab) * 28, top - 28));
            for(int i = 0; i < tabs.size(); i++) {
                int ref = i + categoryTabPage * 6;
                if (categories.size() > ref) {
                    tabs.get(i).setItem(categories.get(ref).getCategoryIcon(), categories.get(ref).getDisplayName(), categories.get(ref).equals(selectedCategory));
                } else tabs.get(i).setItem(null, null, false);
            }
            controls.addAll(tabs);
        }
    }
    
    @Override
    protected void drawBackground(float v, int i, int i1) {
        //Tabs
        if (tabsEnabled) {
            GlStateManager.color4f(1.0F, 1.0F, 1.0F, 1.0F);
            GuiLighting.enableForItems();
            this.client.getTextureManager().bindTexture(CREATIVE_INVENTORY_TABS);
            tabs.stream().filter(tab -> tab.getId() + categoryTabPage * 6 == categories.indexOf(selectedCategory)).forEach(Tab::drawTab);
        }
        
        drawBackground();
        GlStateManager.color4f(1.0F, 1.0F, 1.0F, 1.0F);
        GuiLighting.disable();
        this.client.getTextureManager().bindTexture(CHEST_GUI_TEXTURE);
        
        int lvt_4_1_ = (int) ((mainWindow.getScaledWidth() / 2 - this.guiWidth / 2));
        int lvt_5_1_ = (int) ((mainWindow.getScaledHeight() / 2 - this.guiHeight / 2));
        
        this.drawTexturedRect(lvt_4_1_, lvt_5_1_, 0, 0, this.guiWidth, this.guiHeight);
        slots.forEach(reiSlot -> {
            GlStateManager.color4f(1.0F, 1.0F, 1.0F, 1.0F);
            GuiLighting.disable();
            reiSlot.draw();
        });
        
        if (tabsEnabled)
            tabs.stream().filter(tab -> tab.getId() + categoryTabPage * 6 != categories.indexOf(selectedCategory)).forEach(tab -> {
                GlStateManager.color4f(1.0F, 1.0F, 1.0F, 1.0F);
                GuiLighting.enableForItems();
                this.client.getTextureManager().bindTexture(CREATIVE_INVENTORY_TABS);
                tab.drawTab();
            });
    }
    
    @Override
    public boolean keyPressed(int p_keyPressed_1_, int p_keyPressed_2_, int p_keyPressed_3_) {
        if (p_keyPressed_1_ == 256 && prevScreen != null) {
            this.client.openGui(prevScreen);
            return true;
        }
        return super.keyPressed(p_keyPressed_1_, p_keyPressed_2_, p_keyPressed_3_);
    }
    
    private boolean btnCategoryLeft(int button) {
        recipePointer = 0;
        int categoryPointer = categories.indexOf(selectedCategory);
        categoryPointer--;
        if (categoryPointer <= 0)
            categoryPointer = categories.size() - 1;
        selectedCategory = categories.get(categoryPointer);
        categoryTabPage = categoryPointer / 6;
        updateRecipe();
        return true;
    }
    
    private boolean btnCategoryRight(int button) {
        recipePointer = 0;
        int categoryPointer = categories.indexOf(selectedCategory);
        categoryPointer++;
        if (categoryPointer >= categories.size())
            categoryPointer = 0;
        selectedCategory = categories.get(categoryPointer);
        categoryTabPage = categoryPointer / 6;
        updateRecipe();
        return true;
    }
    
    private boolean btnRecipeLeft(int button) {
        recipePointer -= 2;
        if (recipePointer <= 0)
            recipePointer = MathHelper.floor((recipes.get(selectedCategory).size() - 1) / 2d) * 2;
        updateRecipe();
        return true;
    }
    
    private boolean btnRecipeRight(int button) {
        recipePointer += 2;
        if (recipePointer >= recipes.get(selectedCategory).size())
            recipePointer = 0;
        updateRecipe();
        return true;
    }
    
    private int riseDoublesToInt(double i) {
        return MathHelper.ceil(i);
    }
    
    private int getTotalPages() {
        return MathHelper.clamp(riseDoublesToInt(recipes.get(selectedCategory).size() / 2d), 1, Integer.MAX_VALUE);
    }
}