minecraft-src/net/minecraft/client/gui/screens/inventory/EnchantmentScreen.java
2025-07-04 03:45:38 +03:00

257 lines
10 KiB
Java

package net.minecraft.client.gui.screens.inventory;
import com.google.common.collect.Lists;
import com.mojang.blaze3d.platform.Lighting;
import com.mojang.blaze3d.vertex.VertexConsumer;
import com.mojang.math.Axis;
import java.util.List;
import java.util.Optional;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.ChatFormatting;
import net.minecraft.client.gui.GuiGraphics;
import net.minecraft.client.model.BookModel;
import net.minecraft.client.model.geom.ModelLayers;
import net.minecraft.client.renderer.RenderType;
import net.minecraft.client.renderer.texture.OverlayTexture;
import net.minecraft.core.Holder;
import net.minecraft.core.Holder.Reference;
import net.minecraft.core.registries.Registries;
import net.minecraft.network.chat.CommonComponents;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.FormattedText;
import net.minecraft.network.chat.MutableComponent;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.Mth;
import net.minecraft.util.RandomSource;
import net.minecraft.world.entity.player.Inventory;
import net.minecraft.world.inventory.EnchantmentMenu;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.enchantment.Enchantment;
@Environment(EnvType.CLIENT)
public class EnchantmentScreen extends AbstractContainerScreen<EnchantmentMenu> {
private static final ResourceLocation[] ENABLED_LEVEL_SPRITES = new ResourceLocation[]{
ResourceLocation.withDefaultNamespace("container/enchanting_table/level_1"),
ResourceLocation.withDefaultNamespace("container/enchanting_table/level_2"),
ResourceLocation.withDefaultNamespace("container/enchanting_table/level_3")
};
private static final ResourceLocation[] DISABLED_LEVEL_SPRITES = new ResourceLocation[]{
ResourceLocation.withDefaultNamespace("container/enchanting_table/level_1_disabled"),
ResourceLocation.withDefaultNamespace("container/enchanting_table/level_2_disabled"),
ResourceLocation.withDefaultNamespace("container/enchanting_table/level_3_disabled")
};
private static final ResourceLocation ENCHANTMENT_SLOT_DISABLED_SPRITE = ResourceLocation.withDefaultNamespace(
"container/enchanting_table/enchantment_slot_disabled"
);
private static final ResourceLocation ENCHANTMENT_SLOT_HIGHLIGHTED_SPRITE = ResourceLocation.withDefaultNamespace(
"container/enchanting_table/enchantment_slot_highlighted"
);
private static final ResourceLocation ENCHANTMENT_SLOT_SPRITE = ResourceLocation.withDefaultNamespace("container/enchanting_table/enchantment_slot");
/**
* The ResourceLocation containing the Enchantment GUI texture location
*/
private static final ResourceLocation ENCHANTING_TABLE_LOCATION = ResourceLocation.withDefaultNamespace("textures/gui/container/enchanting_table.png");
/**
* The ResourceLocation containing the texture for the Book rendered above the enchantment table
*/
private static final ResourceLocation ENCHANTING_BOOK_LOCATION = ResourceLocation.withDefaultNamespace("textures/entity/enchanting_table_book.png");
/**
* A Random instance for use with the enchantment gui
*/
private final RandomSource random = RandomSource.create();
private BookModel bookModel;
public float flip;
public float oFlip;
public float flipT;
public float flipA;
public float open;
public float oOpen;
private ItemStack last = ItemStack.EMPTY;
public EnchantmentScreen(EnchantmentMenu menu, Inventory playerInventory, Component title) {
super(menu, playerInventory, title);
}
@Override
protected void init() {
super.init();
this.bookModel = new BookModel(this.minecraft.getEntityModels().bakeLayer(ModelLayers.BOOK));
}
@Override
public void containerTick() {
super.containerTick();
this.tickBook();
}
@Override
public boolean mouseClicked(double mouseX, double mouseY, int button) {
int i = (this.width - this.imageWidth) / 2;
int j = (this.height - this.imageHeight) / 2;
for (int k = 0; k < 3; k++) {
double d = mouseX - (i + 60);
double e = mouseY - (j + 14 + 19 * k);
if (d >= 0.0 && e >= 0.0 && d < 108.0 && e < 19.0 && this.menu.clickMenuButton(this.minecraft.player, k)) {
this.minecraft.gameMode.handleInventoryButtonClick(this.menu.containerId, k);
return true;
}
}
return super.mouseClicked(mouseX, mouseY, button);
}
@Override
protected void renderBg(GuiGraphics guiGraphics, float partialTick, int mouseX, int mouseY) {
int i = (this.width - this.imageWidth) / 2;
int j = (this.height - this.imageHeight) / 2;
guiGraphics.blit(RenderType::guiTextured, ENCHANTING_TABLE_LOCATION, i, j, 0.0F, 0.0F, this.imageWidth, this.imageHeight, 256, 256);
this.renderBook(guiGraphics, i, j, partialTick);
EnchantmentNames.getInstance().initSeed(this.menu.getEnchantmentSeed());
int k = this.menu.getGoldCount();
for (int l = 0; l < 3; l++) {
int m = i + 60;
int n = m + 20;
int o = this.menu.costs[l];
if (o == 0) {
guiGraphics.blitSprite(RenderType::guiTextured, ENCHANTMENT_SLOT_DISABLED_SPRITE, m, j + 14 + 19 * l, 108, 19);
} else {
String string = o + "";
int p = 86 - this.font.width(string);
FormattedText formattedText = EnchantmentNames.getInstance().getRandomName(this.font, p);
int q = 6839882;
if ((k < l + 1 || this.minecraft.player.experienceLevel < o) && !this.minecraft.player.hasInfiniteMaterials()) {
guiGraphics.blitSprite(RenderType::guiTextured, ENCHANTMENT_SLOT_DISABLED_SPRITE, m, j + 14 + 19 * l, 108, 19);
guiGraphics.blitSprite(RenderType::guiTextured, DISABLED_LEVEL_SPRITES[l], m + 1, j + 15 + 19 * l, 16, 16);
guiGraphics.drawWordWrap(this.font, formattedText, n, j + 16 + 19 * l, p, (q & 16711422) >> 1, false);
q = 4226832;
} else {
int r = mouseX - (i + 60);
int s = mouseY - (j + 14 + 19 * l);
if (r >= 0 && s >= 0 && r < 108 && s < 19) {
guiGraphics.blitSprite(RenderType::guiTextured, ENCHANTMENT_SLOT_HIGHLIGHTED_SPRITE, m, j + 14 + 19 * l, 108, 19);
q = 16777088;
} else {
guiGraphics.blitSprite(RenderType::guiTextured, ENCHANTMENT_SLOT_SPRITE, m, j + 14 + 19 * l, 108, 19);
}
guiGraphics.blitSprite(RenderType::guiTextured, ENABLED_LEVEL_SPRITES[l], m + 1, j + 15 + 19 * l, 16, 16);
guiGraphics.drawWordWrap(this.font, formattedText, n, j + 16 + 19 * l, p, q, false);
q = 8453920;
}
guiGraphics.drawString(this.font, string, n + 86 - this.font.width(string), j + 16 + 19 * l + 7, q);
}
}
}
private void renderBook(GuiGraphics guiGraphics, int x, int y, float partialTick) {
float f = Mth.lerp(partialTick, this.oOpen, this.open);
float g = Mth.lerp(partialTick, this.oFlip, this.flip);
guiGraphics.flush();
Lighting.setupForEntityInInventory();
guiGraphics.pose().pushPose();
guiGraphics.pose().translate(x + 33.0F, y + 31.0F, 100.0F);
float h = 40.0F;
guiGraphics.pose().scale(-40.0F, 40.0F, 40.0F);
guiGraphics.pose().mulPose(Axis.XP.rotationDegrees(25.0F));
guiGraphics.pose().translate((1.0F - f) * 0.2F, (1.0F - f) * 0.1F, (1.0F - f) * 0.25F);
float i = -(1.0F - f) * 90.0F - 90.0F;
guiGraphics.pose().mulPose(Axis.YP.rotationDegrees(i));
guiGraphics.pose().mulPose(Axis.XP.rotationDegrees(180.0F));
float j = Mth.clamp(Mth.frac(g + 0.25F) * 1.6F - 0.3F, 0.0F, 1.0F);
float k = Mth.clamp(Mth.frac(g + 0.75F) * 1.6F - 0.3F, 0.0F, 1.0F);
this.bookModel.setupAnim(0.0F, j, k, f);
guiGraphics.drawSpecial(multiBufferSource -> {
VertexConsumer vertexConsumer = multiBufferSource.getBuffer(this.bookModel.renderType(ENCHANTING_BOOK_LOCATION));
this.bookModel.renderToBuffer(guiGraphics.pose(), vertexConsumer, 15728880, OverlayTexture.NO_OVERLAY);
});
guiGraphics.flush();
guiGraphics.pose().popPose();
Lighting.setupFor3DItems();
}
@Override
public void render(GuiGraphics guiGraphics, int mouseX, int mouseY, float partialTick) {
float f = this.minecraft.getDeltaTracker().getGameTimeDeltaPartialTick(false);
super.render(guiGraphics, mouseX, mouseY, f);
this.renderTooltip(guiGraphics, mouseX, mouseY);
boolean bl = this.minecraft.player.hasInfiniteMaterials();
int i = this.menu.getGoldCount();
for (int j = 0; j < 3; j++) {
int k = this.menu.costs[j];
Optional<Reference<Enchantment>> optional = this.minecraft.level.registryAccess().lookupOrThrow(Registries.ENCHANTMENT).get(this.menu.enchantClue[j]);
if (!optional.isEmpty()) {
int l = this.menu.levelClue[j];
int m = j + 1;
if (this.isHovering(60, 14 + 19 * j, 108, 17, mouseX, mouseY) && k > 0 && l >= 0 && optional != null) {
List<Component> list = Lists.<Component>newArrayList();
list.add(Component.translatable("container.enchant.clue", Enchantment.getFullname((Holder<Enchantment>)optional.get(), l)).withStyle(ChatFormatting.WHITE));
if (!bl) {
list.add(CommonComponents.EMPTY);
if (this.minecraft.player.experienceLevel < k) {
list.add(Component.translatable("container.enchant.level.requirement", this.menu.costs[j]).withStyle(ChatFormatting.RED));
} else {
MutableComponent mutableComponent;
if (m == 1) {
mutableComponent = Component.translatable("container.enchant.lapis.one");
} else {
mutableComponent = Component.translatable("container.enchant.lapis.many", m);
}
list.add(mutableComponent.withStyle(i >= m ? ChatFormatting.GRAY : ChatFormatting.RED));
MutableComponent mutableComponent2;
if (m == 1) {
mutableComponent2 = Component.translatable("container.enchant.level.one");
} else {
mutableComponent2 = Component.translatable("container.enchant.level.many", m);
}
list.add(mutableComponent2.withStyle(ChatFormatting.GRAY));
}
}
guiGraphics.renderComponentTooltip(this.font, list, mouseX, mouseY);
break;
}
}
}
}
public void tickBook() {
ItemStack itemStack = this.menu.getSlot(0).getItem();
if (!ItemStack.matches(itemStack, this.last)) {
this.last = itemStack;
do {
this.flipT = this.flipT + (this.random.nextInt(4) - this.random.nextInt(4));
} while (this.flip <= this.flipT + 1.0F && this.flip >= this.flipT - 1.0F);
}
this.oFlip = this.flip;
this.oOpen = this.open;
boolean bl = false;
for (int i = 0; i < 3; i++) {
if (this.menu.costs[i] != 0) {
bl = true;
}
}
if (bl) {
this.open += 0.2F;
} else {
this.open -= 0.2F;
}
this.open = Mth.clamp(this.open, 0.0F, 1.0F);
float f = (this.flipT - this.flip) * 0.4F;
float g = 0.2F;
f = Mth.clamp(f, -0.2F, 0.2F);
this.flipA = this.flipA + (f - this.flipA) * 0.9F;
this.flip = this.flip + this.flipA;
}
}