115 lines
3.6 KiB
Java
115 lines
3.6 KiB
Java
package com.replaymod.render.capturer;
|
|
|
|
import com.mojang.blaze3d.platform.GlStateManager;
|
|
import com.replaymod.core.versions.MCVer;
|
|
import com.replaymod.render.frame.OpenGlFrame;
|
|
import com.replaymod.render.rendering.Frame;
|
|
import com.replaymod.render.rendering.FrameCapturer;
|
|
import com.replaymod.render.utils.ByteBufferPool;
|
|
import de.johni0702.minecraft.gui.utils.lwjgl.Dimension;
|
|
import de.johni0702.minecraft.gui.utils.lwjgl.ReadableDimension;
|
|
import de.johni0702.minecraft.gui.utils.lwjgl.WritableDimension;
|
|
import net.minecraft.client.MinecraftClient;
|
|
import net.minecraft.client.gl.Framebuffer;
|
|
import org.lwjgl.opengl.GL11;
|
|
import org.lwjgl.opengl.GL12;
|
|
|
|
import java.io.IOException;
|
|
import java.nio.ByteBuffer;
|
|
|
|
import static com.replaymod.core.versions.MCVer.popMatrix;
|
|
import static com.replaymod.core.versions.MCVer.pushMatrix;
|
|
import static com.replaymod.core.versions.MCVer.resizeMainWindow;
|
|
import static org.lwjgl.opengl.GL11.GL_COLOR_BUFFER_BIT;
|
|
import static org.lwjgl.opengl.GL11.GL_DEPTH_BUFFER_BIT;
|
|
|
|
public abstract class OpenGlFrameCapturer<F extends Frame, D extends CaptureData> implements FrameCapturer<F> {
|
|
protected final WorldRenderer worldRenderer;
|
|
protected final RenderInfo renderInfo;
|
|
protected int framesDone;
|
|
private Framebuffer frameBuffer;
|
|
|
|
protected final MinecraftClient mc = MCVer.getMinecraft();
|
|
|
|
public OpenGlFrameCapturer(WorldRenderer worldRenderer, RenderInfo renderInfo) {
|
|
this.worldRenderer = worldRenderer;
|
|
this.renderInfo = renderInfo;
|
|
}
|
|
|
|
protected final ReadableDimension frameSize = new ReadableDimension() {
|
|
@Override
|
|
public int getWidth() {
|
|
return getFrameWidth();
|
|
}
|
|
|
|
@Override
|
|
public int getHeight() {
|
|
return getFrameHeight();
|
|
}
|
|
|
|
@Override
|
|
public void getSize(WritableDimension dest) {
|
|
dest.setSize(getWidth(), getHeight());
|
|
}
|
|
};
|
|
|
|
protected int getFrameWidth() {
|
|
return renderInfo.getFrameSize().getWidth();
|
|
}
|
|
|
|
protected int getFrameHeight() {
|
|
return renderInfo.getFrameSize().getHeight();
|
|
}
|
|
|
|
protected Framebuffer frameBuffer() {
|
|
if (frameBuffer == null) {
|
|
frameBuffer = mc.getFramebuffer();
|
|
}
|
|
return frameBuffer;
|
|
}
|
|
|
|
@Override
|
|
public boolean isDone() {
|
|
return framesDone >= renderInfo.getTotalFrames();
|
|
}
|
|
|
|
protected OpenGlFrame renderFrame(int frameId, float partialTicks) {
|
|
return renderFrame(frameId, partialTicks, null);
|
|
}
|
|
|
|
protected OpenGlFrame renderFrame(int frameId, float partialTicks, D captureData) {
|
|
resizeMainWindow(mc, getFrameWidth(), getFrameHeight());
|
|
|
|
pushMatrix();
|
|
frameBuffer().beginWrite(true);
|
|
|
|
GlStateManager.clear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT
|
|
//#if MC>=11400
|
|
, false
|
|
//#endif
|
|
);
|
|
GlStateManager.enableTexture();
|
|
|
|
worldRenderer.renderWorld(partialTicks, captureData);
|
|
|
|
frameBuffer().endWrite();
|
|
popMatrix();
|
|
|
|
return captureFrame(frameId, captureData);
|
|
}
|
|
|
|
protected OpenGlFrame captureFrame(int frameId, D captureData) {
|
|
ByteBuffer buffer = ByteBufferPool.allocate(getFrameWidth() * getFrameHeight() * 4);
|
|
frameBuffer().beginWrite(true);
|
|
GL11.glReadPixels(0, 0, getFrameWidth(), getFrameHeight(), GL12.GL_BGRA, GL11.GL_UNSIGNED_BYTE, buffer);
|
|
frameBuffer().endWrite();
|
|
buffer.rewind();
|
|
|
|
return new OpenGlFrame(frameId, new Dimension(getFrameWidth(), getFrameHeight()), 4, buffer);
|
|
}
|
|
|
|
@Override
|
|
public void close() throws IOException {
|
|
}
|
|
}
|