Spaces:
Build error
Build error
File size: 5,536 Bytes
d46f4a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 |
package net.minecraft.client;
import com.mojang.blaze3d.pipeline.RenderTarget;
import com.mojang.blaze3d.platform.NativeImage;
import com.mojang.blaze3d.systems.RenderSystem;
import com.mojang.logging.LogUtils;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.function.Consumer;
import javax.annotation.Nullable;
import net.minecraft.ChatFormatting;
import net.minecraft.Util;
import net.minecraft.network.chat.ClickEvent;
import net.minecraft.network.chat.Component;
import net.minecraft.network.chat.Style;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.api.distmarker.OnlyIn;
import org.slf4j.Logger;
@OnlyIn(Dist.CLIENT)
public class Screenshot {
private static final Logger LOGGER = LogUtils.getLogger();
public static final String SCREENSHOT_DIR = "screenshots";
private int rowHeight;
private final DataOutputStream outputStream;
private final byte[] bytes;
private final int width;
private final int height;
private File file;
public static void grab(File p_92290_, RenderTarget p_92293_, Consumer<Component> p_92294_) {
grab(p_92290_, null, p_92293_, p_92294_);
}
public static void grab(File p_92296_, @Nullable String p_92297_, RenderTarget p_92300_, Consumer<Component> p_92301_) {
if (!RenderSystem.isOnRenderThread()) {
RenderSystem.recordRenderCall(() -> _grab(p_92296_, p_92297_, p_92300_, p_92301_));
} else {
_grab(p_92296_, p_92297_, p_92300_, p_92301_);
}
}
private static void _grab(File p_92306_, @Nullable String p_92307_, RenderTarget p_92310_, Consumer<Component> p_92311_) {
NativeImage nativeimage = takeScreenshot(p_92310_);
File file1 = new File(p_92306_, "screenshots");
file1.mkdir();
File file2;
if (p_92307_ == null) {
file2 = getFile(file1);
} else {
file2 = new File(file1, p_92307_);
}
Util.ioPool()
.execute(
() -> {
try {
nativeimage.writeToFile(file2);
Component component = Component.literal(file2.getName())
.withStyle(ChatFormatting.UNDERLINE)
.withStyle(p_168608_ -> p_168608_.withClickEvent(new ClickEvent(ClickEvent.Action.OPEN_FILE, file2.getAbsolutePath())));
p_92311_.accept(Component.translatable("screenshot.success", component));
} catch (Exception exception) {
LOGGER.warn("Couldn't save screenshot", (Throwable)exception);
p_92311_.accept(Component.translatable("screenshot.failure", exception.getMessage()));
} finally {
nativeimage.close();
}
}
);
}
public static NativeImage takeScreenshot(RenderTarget p_92282_) {
int i = p_92282_.width;
int j = p_92282_.height;
NativeImage nativeimage = new NativeImage(i, j, false);
RenderSystem.bindTexture(p_92282_.getColorTextureId());
nativeimage.downloadTexture(0, true);
nativeimage.flipY();
return nativeimage;
}
private static File getFile(File p_92288_) {
String s = Util.getFilenameFormattedDateTime();
int i = 1;
while (true) {
File file1 = new File(p_92288_, s + (i == 1 ? "" : "_" + i) + ".png");
if (!file1.exists()) {
return file1;
}
i++;
}
}
public Screenshot(File p_168601_, int p_168602_, int p_168603_, int p_168604_) throws IOException {
this.width = p_168602_;
this.height = p_168603_;
this.rowHeight = p_168604_;
File file1 = new File(p_168601_, "screenshots");
file1.mkdir();
String s = "huge_" + Util.getFilenameFormattedDateTime();
int i = 1;
while ((this.file = new File(file1, s + (i == 1 ? "" : "_" + i) + ".tga")).exists()) {
i++;
}
byte[] abyte = new byte[18];
abyte[2] = 2;
abyte[12] = (byte)(p_168602_ % 256);
abyte[13] = (byte)(p_168602_ / 256);
abyte[14] = (byte)(p_168603_ % 256);
abyte[15] = (byte)(p_168603_ / 256);
abyte[16] = 24;
this.bytes = new byte[p_168602_ * p_168604_ * 3];
this.outputStream = new DataOutputStream(new FileOutputStream(this.file));
this.outputStream.write(abyte);
}
public void addRegion(ByteBuffer p_168610_, int p_168611_, int p_168612_, int p_168613_, int p_168614_) {
int i = p_168613_;
int j = p_168614_;
if (p_168613_ > this.width - p_168611_) {
i = this.width - p_168611_;
}
if (p_168614_ > this.height - p_168612_) {
j = this.height - p_168612_;
}
this.rowHeight = j;
for (int k = 0; k < j; k++) {
p_168610_.position((p_168614_ - j) * p_168613_ * 3 + k * p_168613_ * 3);
int l = (p_168611_ + k * this.width) * 3;
p_168610_.get(this.bytes, l, i * 3);
}
}
public void saveRow() throws IOException {
this.outputStream.write(this.bytes, 0, this.width * 3 * this.rowHeight);
}
public File close() throws IOException {
this.outputStream.close();
return this.file;
}
} |