efibs / mc-lasertag-mod

import net.minecraft.client.gui.Drawable;

import net.minecraft.client.gui.DrawableHelper;

import net.minecraft.client.gui.Element;

import net.minecraft.client.gui.Selectable;

import net.minecraft.client.gui.screen.Screen;

import net.minecraft.client.gui.widget.ButtonWidget;

import net.minecraft.client.util.math.MatrixStack;

import net.minecraft.entity.player.PlayerEntity;

import net.minecraft.text.Text;

/**

 * Base class for screens in the lasertag game manager gui

 *

 * @author Étienne Muser

 */

public abstract class GameManagerScreen extends Screen {

    private int numberOfListedButtons = 0;

    private final Screen parent;

    protected final PlayerEntity player;

    protected static final int horizontalPadding = 20;

    protected static final int verticalPadding = 15;

    protected static final int buttonPadding = 10;

    protected static final int buttonHeight = 20;

    protected static final int buttonWidth = 200;

    protected static final int textColor = 0xFFFFFFFF;

    public GameManagerScreen(Screen parent, String translatableTitle, PlayerEntity player) {

        super(Text.translatable(translatableTitle));

        this.parent = parent;

        this.player = player;

    }

    public void close() {

        this.client.setScreen(this.parent);

    }

    @Override

    public void render(MatrixStack matrices, int mouseX, int mouseY, float delta) {

        this.drawBackground(matrices);

        this.textRenderer.drawWithShadow(matrices, title, horizontalPadding, verticalPadding, textColor);

        DrawableHelper.fill(matrices, horizontalPadding, verticalPadding + this.textRenderer.fontHeight + 1, this.width - horizontalPadding, verticalPadding + this.textRenderer.fontHeight + 2, 0xFFFFFFFF);

        super.render(matrices, mouseX, mouseY, delta);

    }

    /**

     * The init method of the screen. Override to add more gui elements. Must call super.init() first!

     */

    @Override

    protected void init() {

        numberOfListedButtons = 0;

        addBackButton();

    }

    /**

     * Adds a button to the main button list

     *

     * @param translatableName The button text as the translatable key

     * @param pressAction The button click callback

     * @return The created button widget

     */

    protected ButtonWidget addListedButton(String translatableName, ButtonWidget.PressAction pressAction) {

        var button = this.addDrawableChild(new ButtonWidget(horizontalPadding, verticalPadding + textRenderer.fontHeight + buttonPadding + numberOfListedButtons * (buttonHeight + buttonPadding), buttonWidth, buttonHeight, Text.translatable(translatableName), pressAction));

        ++numberOfListedButtons;

        return button;

    }

    /**

     * Adds a widget to the main button list

     *

     * @param buttonGenerator Method creating the button

     * @param <T> The type of the widget

     * @return The created widget

     */

    protected <T extends Element & Drawable & Selectable> T addListedWidget(IListedButtonGenerator<T> buttonGenerator) {

        var button = this.addDrawableChild(buttonGenerator.generate(horizontalPadding, verticalPadding + textRenderer.fontHeight + buttonPadding + numberOfListedButtons * (buttonHeight + buttonPadding), buttonWidth, buttonHeight));

        ++numberOfListedButtons;

        return button;

    }

    /**

     * Functional interface for generating a widget for <code>addListedWidget</code>

     * @param <T> The type of the widget to be generated

     */

    protected interface IListedButtonGenerator<T extends Element & Drawable & Selectable> {

        T generate(int x, int y, int width, int height);

    }

    private void addBackButton() {

        var buttonTranslatableText = this.parent == null ? "gui.exit" : "gui.back";

        this.addDrawableChild(new ButtonWidget(this.width - horizontalPadding - (buttonWidth / 2), this.height - verticalPadding - buttonHeight, buttonWidth / 2, buttonHeight, Text.translatable(buttonTranslatableText), button -> {

            this.close();

        }));

    }

}