package com.example.screenshotmanager; import com.mojang.blaze3d.systems.RenderSystem; import net.minecraft.client.MinecraftClient; import net.minecraft.client.gui.DrawContext; import net.minecraft.client.gui.screen.Screen; import net.minecraft.client.gui.widget.ButtonWidget; import net.minecraft.client.render.RenderLayer; import net.minecraft.client.texture.NativeImage; import net.minecraft.client.texture.NativeImageBackedTexture; import net.minecraft.text.Text; import net.minecraft.util.Identifier; import java.awt.*; import java.io.File; import java.nio.file.Files; import java.util.HashMap; import java.util.List; import java.util.Map; public class ScreenshotGalleryScreen extends Screen { private final Screen parent; private static final int THUMBNAIL_SIZE = 128; private static final int PADDING = 8; private static final int COLUMNS = 5; private final Map thumbnails = new HashMap<>(); private int scrollOffset = 0; private int maxScroll = 0; public ScreenshotGalleryScreen(Screen parent) { super(Text.translatable("gui.screenshotmanager.gallery.title")); this.parent = parent; } @Override protected void init() { super.init(); int buttonWidth = 80; int buttonHeight = 20; int buttonY = this.height - 30; int spacing = 5; this.addDrawableChild(ButtonWidget.builder( Text.translatable("gui.screenshotmanager.gallery.open_folder"), button -> openScreenshotFolder() ).dimensions(10, buttonY, buttonWidth, buttonHeight).build()); this.addDrawableChild(ButtonWidget.builder( Text.translatable("gui.screenshotmanager.gallery.refresh"), button -> refresh() ).dimensions(10 + buttonWidth + spacing, buttonY, buttonWidth, buttonHeight).build()); this.addDrawableChild(ButtonWidget.builder( Text.literal("Close"), button -> this.close() ).dimensions(this.width - buttonWidth - 10, buttonY, buttonWidth, buttonHeight).build()); loadThumbnails(); calculateMaxScroll(); } private void loadThumbnails() { List screenshots = ScreenshotManager.getScreenshots(); for (ScreenshotData data : screenshots) { if (!thumbnails.containsKey(data)) { try { NativeImage image = NativeImage.read(Files.newInputStream(data.imagePath)); int originalWidth = image.getWidth(); int originalHeight = image.getHeight(); float scale = Math.min( (float) THUMBNAIL_SIZE / originalWidth, (float) THUMBNAIL_SIZE / originalHeight ); int thumbWidth = (int) (originalWidth * scale); int thumbHeight = (int) (originalHeight * scale); NativeImage thumbnail = new NativeImage(thumbWidth, thumbHeight, false); thumbnail.resizeSubRectTo(0, 0, thumbWidth, thumbHeight, image); NativeImageBackedTexture texture = new NativeImageBackedTexture(thumbnail); Identifier textureId = Identifier.of("screenshotmanager", "screenshot_" + data.filename); MinecraftClient.getInstance().getTextureManager().registerTexture(textureId, texture); thumbnails.put(data, textureId); image.close(); } catch (Exception e) { ScreenshotManagerMod.LOGGER.error("Failed to load thumbnail for " + data.filename, e); } } } } private void calculateMaxScroll() { List screenshots = ScreenshotManager.getScreenshots(); int rows = (int) Math.ceil((double) screenshots.size() / COLUMNS); int totalHeight = rows * (THUMBNAIL_SIZE + PADDING); maxScroll = Math.max(0, totalHeight - (this.height - 80)); } @Override public void render(DrawContext context, int mouseX, int mouseY, float delta) { this.renderBackground(context, mouseX, mouseY, delta); context.drawCenteredTextWithShadow(this.textRenderer, this.title, this.width / 2, 10, 0xFFFFFF); List screenshots = ScreenshotManager.getScreenshots(); if (screenshots.isEmpty()) { String emptyMessage = Text.translatable( "gui.screenshotmanager.gallery.empty", ModKeybindings.TAKE_SCREENSHOT.getBoundKeyLocalizedText().getString() ).getString(); context.drawCenteredTextWithShadow(this.textRenderer, emptyMessage, this.width / 2, this.height / 2, 0x808080); } else { int startX = (this.width - (COLUMNS * (THUMBNAIL_SIZE + PADDING) - PADDING)) / 2; int startY = 30 - scrollOffset; for (int i = 0; i < screenshots.size(); i++) { ScreenshotData data = screenshots.get(i); int col = i % COLUMNS; int row = i / COLUMNS; int x = startX + col * (THUMBNAIL_SIZE + PADDING); int y = startY + row * (THUMBNAIL_SIZE + PADDING); if (y + THUMBNAIL_SIZE < 30 || y > this.height - 50) continue; Identifier texture = thumbnails.get(data); if (texture != null) { context.fill(x - 2, y - 2, x + THUMBNAIL_SIZE + 2, y + THUMBNAIL_SIZE + 2, 0xFF404040); RenderSystem.setShaderColor(1.0F, 1.0F, 1.0F, 1.0F); context.drawTexture(RenderLayer::getGuiTextured, texture, x, y, 0, 0, THUMBNAIL_SIZE, THUMBNAIL_SIZE, THUMBNAIL_SIZE, THUMBNAIL_SIZE); if (mouseX >= x && mouseX <= x + THUMBNAIL_SIZE && mouseY >= y && mouseY <= y + THUMBNAIL_SIZE) { context.fill(x, y, x + THUMBNAIL_SIZE, y + THUMBNAIL_SIZE, 0x80FFFFFF); } } } } super.render(context, mouseX, mouseY, delta); } @Override public boolean mouseClicked(double mouseX, double mouseY, int button) { List screenshots = ScreenshotManager.getScreenshots(); int startX = (this.width - (COLUMNS * (THUMBNAIL_SIZE + PADDING) - PADDING)) / 2; int startY = 30 - scrollOffset; for (int i = 0; i < screenshots.size(); i++) { ScreenshotData data = screenshots.get(i); int col = i % COLUMNS; int row = i / COLUMNS; int x = startX + col * (THUMBNAIL_SIZE + PADDING); int y = startY + row * (THUMBNAIL_SIZE + PADDING); if (mouseX >= x && mouseX <= x + THUMBNAIL_SIZE && mouseY >= y && mouseY <= y + THUMBNAIL_SIZE) { if (button == 0) { this.client.setScreen(new ScreenshotViewerScreen(this, data, i)); return true; } else if (button == 1) { handleRightClick(data); return true; } } } return super.mouseClicked(mouseX, mouseY, button); } private void handleRightClick(ScreenshotData data) { if (ClipboardHelper.copyImageToClipboard(data.imagePath)) { if (this.client.player != null) { this.client.player.sendMessage( Text.translatable("message.screenshotmanager.copied_to_clipboard"), false ); } } } @Override public boolean mouseScrolled(double mouseX, double mouseY, double horizontalAmount, double verticalAmount) { scrollOffset -= (int) (verticalAmount * 20); scrollOffset = Math.max(0, Math.min(maxScroll, scrollOffset)); return true; } private void openScreenshotFolder() { try { Desktop.getDesktop().open(ScreenshotManager.getScreenshotDirectory().toFile()); } catch (Exception e) { ScreenshotManagerMod.LOGGER.error("Failed to open screenshot folder", e); } } private void refresh() { thumbnails.clear(); ScreenshotManager.loadScreenshots(); loadThumbnails(); calculateMaxScroll(); scrollOffset = 0; } @Override public void close() { this.client.setScreen(parent); } }