From 91585cebff914e178e75b93f168e45884b458133 Mon Sep 17 00:00:00 2001
From: Vasco Lavos <vazkii@hotmail.com>
Date: Tue, 16 Jan 2024 07:28:48 +0000
Subject: [PATCH] Fix #4585

---
 .../be/MatrixEnchantingTableBlockEntity.java  | 14 +++---
 .../client/screen/MatrixEnchantingScreen.java | 46 ++++++++++---------
 2 files changed, 31 insertions(+), 29 deletions(-)

diff --git a/src/main/java/org/violetmoon/quark/addons/oddities/block/be/MatrixEnchantingTableBlockEntity.java b/src/main/java/org/violetmoon/quark/addons/oddities/block/be/MatrixEnchantingTableBlockEntity.java
index b3d9ad8739..34b31cba2f 100644
--- a/src/main/java/org/violetmoon/quark/addons/oddities/block/be/MatrixEnchantingTableBlockEntity.java
+++ b/src/main/java/org/violetmoon/quark/addons/oddities/block/be/MatrixEnchantingTableBlockEntity.java
@@ -235,14 +235,7 @@ private void makeUUID() {
 			matrixId = UUID.randomUUID();
 	}
 
-	private void updateEnchantPower() {
-		ItemStack item = getItem(0);
-		influences.clear();
-		if(item.isEmpty())
-			return;
-
-		enchantability = Quark.ZETA.itemExtensions.get(item).getEnchantmentValueZeta(item);
-
+	public void updateEnchantPower() {
 		boolean allowWater = MatrixEnchantingModule.allowUnderwaterEnchanting;
 		boolean allowShort = MatrixEnchantingModule.allowShortBlockEnchanting;
 
@@ -263,6 +256,11 @@ private void updateEnchantPower() {
 		}
 
 		bookshelfPower = Math.min((int) power, MatrixEnchantingModule.maxBookshelves);
+		
+		ItemStack item = getItem(0);
+		influences.clear();
+		if(!item.isEmpty())
+			enchantability = Quark.ZETA.itemExtensions.get(item).getEnchantmentValueZeta(item);
 	}
 
 	private boolean isAirGap(int j, int k, boolean allowWater, boolean allowShortBlock) {
diff --git a/src/main/java/org/violetmoon/quark/addons/oddities/client/screen/MatrixEnchantingScreen.java b/src/main/java/org/violetmoon/quark/addons/oddities/client/screen/MatrixEnchantingScreen.java
index 697f5c20f7..9c49d011f2 100644
--- a/src/main/java/org/violetmoon/quark/addons/oddities/client/screen/MatrixEnchantingScreen.java
+++ b/src/main/java/org/violetmoon/quark/addons/oddities/client/screen/MatrixEnchantingScreen.java
@@ -65,6 +65,7 @@ public void init() {
 		updateButtonStatus();
 
 		pieceList.refresh();
+		enchanter.updateEnchantPower();
 	}
 
 	@Override
@@ -103,17 +104,35 @@ protected void renderBg(@NotNull GuiGraphics guiGraphics, float partialTicks, in
 
 		pieceList.render(guiGraphics, mouseX, mouseY, partialTicks);
 
-		if(enchanter.matrix != null
+		boolean showCost = enchanter.matrix != null
 				&& enchanter.matrix.canGeneratePiece(enchanter.influences, enchanter.bookshelfPower, enchanter.enchantability)
-				&& !mc.player.getAbilities().instabuild) {
-			int x = i + 71;
-			int y = j + 56;
+				&& !mc.player.getAbilities().instabuild;
+
+		String text = ""+enchanter.bookshelfPower;
+		int x = i + 50;
+		int y = j + 55;
+		
+		if(enchanter.bookshelfPower > 0) {
+			pose.pushPose();
+			guiGraphics.renderItem(new ItemStack(Items.BOOK), x, y);
+			pose.translate(0, 0, 1000);
+	
+			x -= font.width(text) / 2;
+	
+			drawBorderedText(guiGraphics, text, x + 3, y + 6, 0xc8ff8f);
+			pose.popPose();
+		}
+
+		if(showCost) {
 			int xpCost = enchanter.matrix.getNewPiecePrice();
 			int xpMin = enchanter.matrix.getMinXpLevel(enchanter.bookshelfPower);
 			boolean has = enchanter.matrix.validateXp(mc.player, enchanter.bookshelfPower);
 			
+			x = i + 71;
+			y = j + 56;
+			text = String.valueOf(xpCost);
+			
 			guiGraphics.blit(BACKGROUND, x, y, 0, imageHeight, 10, 10);
-			String text = String.valueOf(xpCost);
 
 			if(!has && mc.player.experienceLevel < xpMin) {
 				text = I18n.get("quark.gui.enchanting.min", xpMin);
@@ -122,24 +141,9 @@ protected void renderBg(@NotNull GuiGraphics guiGraphics, float partialTicks, in
 
 			x -= font.width(text) / 2;
 			drawBorderedText(guiGraphics, text, x + 2, y + 5, has ? 0xc8ff8f : 0xff8f8f);
-			
-			text = ""+enchanter.bookshelfPower;
-			x = i + 50;
-			y = j + 55;
-			
-			pose.pushPose();
-			guiGraphics.renderItem(new ItemStack(Items.BOOK), x, y);
-			pose.translate(0, 0, 1000);
-			
-			x -= font.width(text) / 2;
-			
-			drawBorderedText(guiGraphics, text, x + 3, y + 6, 0xc8ff8f);
-			pose.popPose();
 		}
-		
-
 	}
-	
+
 	private void drawBorderedText(GuiGraphics guiGraphics, String text, int x, int y, int color) {
 		guiGraphics.drawString(font, text, x - 1, y, 0, false);
 		guiGraphics.drawString(font, text, x + 1, y, 0, false);