Compare commits
3 Commits
test
...
gpu-accumu
| Author | SHA1 | Date | |
|---|---|---|---|
| 4fce67a560 | |||
| 088fb3bb30 | |||
| e2e04fa86f |
@@ -87,6 +87,11 @@ public abstract class OpenGlFrameCapturer<F extends Frame, D extends CaptureData
|
|||||||
}
|
}
|
||||||
|
|
||||||
protected OpenGlFrame renderFrame(int frameId, float partialTicks, D captureData) {
|
protected OpenGlFrame renderFrame(int frameId, float partialTicks, D captureData) {
|
||||||
|
renderWorldOnly(partialTicks, captureData);
|
||||||
|
return captureFrame(frameId, captureData);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected void renderWorldOnly(float partialTicks, D captureData) {
|
||||||
resizeMainWindow(mc, getFrameWidth(), getFrameHeight());
|
resizeMainWindow(mc, getFrameWidth(), getFrameHeight());
|
||||||
|
|
||||||
pushMatrix();
|
pushMatrix();
|
||||||
@@ -115,8 +120,6 @@ public abstract class OpenGlFrameCapturer<F extends Frame, D extends CaptureData
|
|||||||
frameBuffer().endWrite();
|
frameBuffer().endWrite();
|
||||||
//#endif
|
//#endif
|
||||||
popMatrix();
|
popMatrix();
|
||||||
|
|
||||||
return captureFrame(frameId, captureData);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
protected OpenGlFrame captureFrame(int frameId, D captureData) {
|
protected OpenGlFrame captureFrame(int frameId, D captureData) {
|
||||||
|
|||||||
@@ -10,11 +10,15 @@ import com.replaymod.pathing.properties.LensProperties;
|
|||||||
|
|
||||||
import com.replaymod.render.RenderSettings;
|
import com.replaymod.render.RenderSettings;
|
||||||
import com.replaymod.render.utils.ByteBufferPool;
|
import com.replaymod.render.utils.ByteBufferPool;
|
||||||
|
import com.replaymod.render.utils.AccumulationBuffer;
|
||||||
import com.replaymod.replay.ReplayHandler;
|
import com.replaymod.replay.ReplayHandler;
|
||||||
import com.replaymod.replay.ReplayModReplay;
|
import com.replaymod.replay.ReplayModReplay;
|
||||||
import com.replaymod.replaystudio.pathing.path.Timeline;
|
import com.replaymod.replaystudio.pathing.path.Timeline;
|
||||||
import com.replaymod.simplepathing.SPTimeline;
|
import com.replaymod.simplepathing.SPTimeline;
|
||||||
|
|
||||||
|
import com.replaymod.core.mixin.MinecraftAccessor;
|
||||||
|
import com.replaymod.pathing.player.ReplayTimer;
|
||||||
|
|
||||||
import org.apache.commons.lang3.tuple.Triple;
|
import org.apache.commons.lang3.tuple.Triple;
|
||||||
|
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
@@ -109,6 +113,8 @@ public class RealLensOpenGlFrameCapturer
|
|||||||
private final Path positionPath;
|
private final Path positionPath;
|
||||||
private final Timeline timeline;
|
private final Timeline timeline;
|
||||||
private final ReplayHandler replayHandler;
|
private final ReplayHandler replayHandler;
|
||||||
|
private final AccumulationBuffer accum = new AccumulationBuffer();
|
||||||
|
private long lastConsumedTime = -1;
|
||||||
|
|
||||||
public RealLensOpenGlFrameCapturer(
|
public RealLensOpenGlFrameCapturer(
|
||||||
WorldRenderer worldRenderer,
|
WorldRenderer worldRenderer,
|
||||||
@@ -123,72 +129,71 @@ public class RealLensOpenGlFrameCapturer
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
public Map<Channel, RealLensOpenGlFrame> process() {
|
public Map<Channel, RealLensOpenGlFrame> process() {
|
||||||
// Продвигает состояние игры к времени ТЕКУЩЕГО выходного кадра
|
|
||||||
// (через onTick -> timeline.applyToGame(videoTime)).
|
|
||||||
renderInfo.updateForNextFrame();
|
renderInfo.updateForNextFrame();
|
||||||
int frameId = framesDone++;
|
int frameId = framesDone++;
|
||||||
|
|
||||||
RenderSettings settings = renderInfo.getRenderSettings();
|
RenderSettings settings = renderInfo.getRenderSettings();
|
||||||
int fps = settings.getFramesPerSecond();
|
int fps = settings.getFramesPerSecond();
|
||||||
int N = Math.max(1, settings.getLensBlurSamples()); // единый бюджет сэмплов
|
int N = Math.max(1, settings.getLensBlurSamples());
|
||||||
float shutter = settings.getShutterAngle(); // градусы
|
float shutter = settings.getShutterAngle();
|
||||||
|
|
||||||
boolean motionBlur = shutter > 0f;
|
boolean motionBlur = shutter > 0f;
|
||||||
double dt = 1000.0 / fps; // длительность выходного кадра, мс
|
double dt = 1000.0 / fps;
|
||||||
double shutterFrac = motionBlur ? (shutter / 360.0) : 0.0;
|
double shutterFrac = motionBlur ? (shutter / 360.0) : 0.0;
|
||||||
|
|
||||||
int width = getFrameWidth();
|
int width = getFrameWidth();
|
||||||
int height = getFrameHeight();
|
int height = getFrameHeight();
|
||||||
int bpp = 4;
|
|
||||||
int totalBytes = width * height * bpp;
|
|
||||||
|
|
||||||
long[] accumulator = new long[totalBytes];
|
// перемешанный апертурный индекс (декорреляция DOF <-> время)
|
||||||
|
int[] apertureOrder = new int[N];
|
||||||
// перемешанные слои времени, чтобы время не коррелировало с индексом апертурной спирали
|
for (int i = 0; i < N; i++) apertureOrder[i] = i;
|
||||||
int[] order = new int[N];
|
for (int i = N - 1; i > 0; i--) {
|
||||||
for (int i = 0; i < N; i++) order[i] = i;
|
|
||||||
for (int i = N - 1; i > 0; i--) { // Fisher-Yates
|
|
||||||
int j = ThreadLocalRandom.current().nextInt(i + 1);
|
int j = ThreadLocalRandom.current().nextInt(i + 1);
|
||||||
int tmp = order[i];
|
int tmp = apertureOrder[i]; apertureOrder[i] = apertureOrder[j]; apertureOrder[j] = tmp;
|
||||||
order[i] = order[j];
|
|
||||||
order[j] = tmp;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
accum.begin(width, height); // float-FBO нужного размера + очистка в 0
|
||||||
|
|
||||||
for (int i = 0; i < N; i++) {
|
for (int i = 0; i < N; i++) {
|
||||||
// --- время: 1 стратифицированный джиттернутый сэмпл на слой ---
|
|
||||||
long tSub;
|
long tSub;
|
||||||
if (motionBlur) {
|
if (motionBlur) {
|
||||||
double u = (i + ThreadLocalRandom.current().nextDouble()) / N;
|
double u = (i + ThreadLocalRandom.current().nextDouble()) / N; // монотонно вперёд
|
||||||
tSub = (long) (frameId * dt + u * shutterFrac * dt);
|
tSub = (long) (frameId * dt + u * shutterFrac * dt);
|
||||||
timeline.applyToGame(tSub, replayHandler);
|
timeline.applyToGame(tSub, replayHandler);
|
||||||
} else {
|
} else {
|
||||||
tSub = (long) (frameId * dt);
|
tSub = (long) (frameId * dt);
|
||||||
}
|
}
|
||||||
// partialTicks от РЕАЛЬНОГО времени реплея (frac(replayTime/50)),
|
|
||||||
// как в AbstractTimelinePlayer.onTick. (tSub % 50) даёт рассинхрон -> раздвоение.
|
|
||||||
long replayTime = replayHandler.getReplaySender().currentTimeStamp();
|
long replayTime = replayHandler.getReplaySender().currentTimeStamp();
|
||||||
|
if (lastConsumedTime < 0) lastConsumedTime = replayTime;
|
||||||
|
int pendingTicks = (int) (replayTime / 50L - lastConsumedTime / 50L);
|
||||||
|
if (pendingTicks > 0) {
|
||||||
|
ReplayTimer timer = (ReplayTimer) ((MinecraftAccessor) mc).getTimer();
|
||||||
|
while (pendingTicks-- > 0) {
|
||||||
|
mc.tick(); // двигает prev/cur частиц через границу тика
|
||||||
|
timer.tickDelta -= 1f; // компенсация: VideoRenderer не должен тикнуть повторно
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (replayTime > lastConsumedTime) lastConsumedTime = replayTime;
|
||||||
float subPartialTicks = (float) ((replayTime % 50L) / 50.0);
|
float subPartialTicks = (float) ((replayTime % 50L) / 50.0);
|
||||||
|
|
||||||
float focal = positionPath.getValue(LensProperties.FOCAL_DISTANCE, tSub).map(Triple::getLeft).orElse(5.0f);
|
float focal = positionPath.getValue(LensProperties.FOCAL_DISTANCE, tSub).map(Triple::getLeft).orElse(5.0f);
|
||||||
float apert = positionPath.getValue(LensProperties.APERTURE_RADIUS, tSub).map(Triple::getLeft).orElse(0.05f);
|
float apert = positionPath.getValue(LensProperties.APERTURE_RADIUS, tSub).map(Triple::getLeft).orElse(0.05f);
|
||||||
Data cam = singleDiskSample(order[i], N, focal, apert);
|
Data cam = singleDiskSample(apertureOrder[i], N, focal, apert);
|
||||||
|
|
||||||
OpenGlFrame sub = renderFrame(frameId, subPartialTicks, cam);
|
renderWorldOnly(subPartialTicks, cam); // мир -> главный FBO, без readback
|
||||||
ByteBuffer buf = sub.getByteBuffer();
|
accum.add(frameBuffer().getColorAttachment()); // прибавить на GPU
|
||||||
buf.rewind();
|
|
||||||
for (int k = 0; k < totalBytes; k++) {
|
|
||||||
accumulator[k] += (buf.get(k) & 0xFF);
|
|
||||||
}
|
|
||||||
ByteBufferPool.release(buf);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
ByteBuffer out = ByteBufferPool.allocate(totalBytes);
|
ByteBuffer out = accum.finishAveraged(N); // ОДИН readback + деление на N
|
||||||
for (int k = 0; k < totalBytes; k++) {
|
accum.unbind(); // вернуться в главный FBO
|
||||||
out.put((byte) (accumulator[k] / N));
|
|
||||||
}
|
|
||||||
out.rewind();
|
|
||||||
|
|
||||||
OpenGlFrame averaged = new OpenGlFrame(frameId, new Dimension(width, height), bpp, out);
|
OpenGlFrame averaged = new OpenGlFrame(frameId, new Dimension(width, height), 4, out);
|
||||||
return Collections.singletonMap(Channel.BRGA, new RealLensOpenGlFrame(new OpenGlFrame[]{ averaged }));
|
return Collections.singletonMap(Channel.BRGA, new RealLensOpenGlFrame(new OpenGlFrame[]{ averaged }));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void close() throws java.io.IOException {
|
||||||
|
super.close();
|
||||||
|
accum.close();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
140
src/main/java/com/replaymod/render/utils/AccumulationBuffer.java
Normal file
140
src/main/java/com/replaymod/render/utils/AccumulationBuffer.java
Normal file
@@ -0,0 +1,140 @@
|
|||||||
|
package com.replaymod.render.utils;
|
||||||
|
|
||||||
|
//#if MC<12105
|
||||||
|
import com.mojang.blaze3d.systems.RenderSystem;
|
||||||
|
import com.mojang.blaze3d.systems.VertexSorter;
|
||||||
|
import net.minecraft.client.render.BufferBuilder;
|
||||||
|
import net.minecraft.client.render.Tessellator;
|
||||||
|
import net.minecraft.client.render.VertexFormats;
|
||||||
|
import net.minecraft.client.render.GameRenderer;
|
||||||
|
import org.joml.Matrix4f;
|
||||||
|
import org.lwjgl.opengl.GL11;
|
||||||
|
import org.lwjgl.opengl.GL12;
|
||||||
|
import org.lwjgl.opengl.GL30;
|
||||||
|
import org.lwjgl.opengl.GL33;
|
||||||
|
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.nio.FloatBuffer;
|
||||||
|
|
||||||
|
public class AccumulationBuffer {
|
||||||
|
private int fbo = 0;
|
||||||
|
private int tex = 0;
|
||||||
|
private int width = 0;
|
||||||
|
private int height = 0;
|
||||||
|
|
||||||
|
public void begin(int width, int height) {
|
||||||
|
if (fbo == 0 || width != this.width || height != this.height) {
|
||||||
|
allocate(width, height);
|
||||||
|
}
|
||||||
|
GL30.glBindFramebuffer(GL30.GL_FRAMEBUFFER, fbo);
|
||||||
|
GL11.glViewport(0, 0, width, height);
|
||||||
|
GL11.glClearColor(0f, 0f, 0f, 0f);
|
||||||
|
GL11.glClear(GL11.GL_COLOR_BUFFER_BIT);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void allocate(int width, int height) {
|
||||||
|
close();
|
||||||
|
this.width = width;
|
||||||
|
this.height = height;
|
||||||
|
|
||||||
|
// Сырые GL-вызовы не обновляют кэш GlStateManager -> сохраняем и
|
||||||
|
// возвращаем привязки, иначе MC пропустит ребинд (текстура неба и т.п.).
|
||||||
|
int prevTex = GL11.glGetInteger(GL11.GL_TEXTURE_BINDING_2D);
|
||||||
|
int prevFbo = GL11.glGetInteger(GL30.GL_FRAMEBUFFER_BINDING);
|
||||||
|
|
||||||
|
tex = GL11.glGenTextures();
|
||||||
|
GL11.glBindTexture(GL11.GL_TEXTURE_2D, tex);
|
||||||
|
GL11.glTexParameteri(GL11.GL_TEXTURE_2D, GL11.GL_TEXTURE_MIN_FILTER, GL11.GL_NEAREST);
|
||||||
|
GL11.glTexParameteri(GL11.GL_TEXTURE_2D, GL11.GL_TEXTURE_MAG_FILTER, GL11.GL_NEAREST);
|
||||||
|
GL11.glTexImage2D(GL11.GL_TEXTURE_2D, 0, GL30.GL_RGBA32F, width, height, 0,
|
||||||
|
GL11.GL_RGBA, GL11.GL_FLOAT, (ByteBuffer) null);
|
||||||
|
|
||||||
|
fbo = GL30.glGenFramebuffers();
|
||||||
|
GL30.glBindFramebuffer(GL30.GL_FRAMEBUFFER, fbo);
|
||||||
|
GL30.glFramebufferTexture2D(GL30.GL_FRAMEBUFFER, GL30.GL_COLOR_ATTACHMENT0,
|
||||||
|
GL11.GL_TEXTURE_2D, tex, 0);
|
||||||
|
|
||||||
|
GL11.glBindTexture(GL11.GL_TEXTURE_2D, prevTex);
|
||||||
|
GL30.glBindFramebuffer(GL30.GL_FRAMEBUFFER, prevFbo);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void add(int srcColorTex) {
|
||||||
|
GL30.glBindFramebuffer(GL30.GL_FRAMEBUFFER, fbo);
|
||||||
|
GL11.glViewport(0, 0, width, height);
|
||||||
|
|
||||||
|
RenderSystem.disableDepthTest();
|
||||||
|
RenderSystem.depthMask(false);
|
||||||
|
RenderSystem.enableBlend();
|
||||||
|
RenderSystem.blendFunc(GL11.GL_ONE, GL11.GL_ONE);
|
||||||
|
RenderSystem.setShaderColor(1f, 1f, 1f, 1f);
|
||||||
|
|
||||||
|
Matrix4f savedProj = RenderSystem.getProjectionMatrix();
|
||||||
|
VertexSorter savedSorter = RenderSystem.getVertexSorting();
|
||||||
|
RenderSystem.setProjectionMatrix(new Matrix4f(), VertexSorter.BY_Z);
|
||||||
|
RenderSystem.getModelViewStack().push();
|
||||||
|
RenderSystem.getModelViewStack().loadIdentity();
|
||||||
|
RenderSystem.applyModelViewMatrix();
|
||||||
|
|
||||||
|
RenderSystem.setShader(GameRenderer::getPositionTexProgram);
|
||||||
|
RenderSystem.setShaderTexture(0, srcColorTex);
|
||||||
|
|
||||||
|
int prevTex = GL11.glGetInteger(GL11.GL_TEXTURE_BINDING_2D);
|
||||||
|
GL11.glBindTexture(GL11.GL_TEXTURE_2D, srcColorTex);
|
||||||
|
GL11.glTexParameteri(GL11.GL_TEXTURE_2D, GL33.GL_TEXTURE_SWIZZLE_A, GL11.GL_ONE);
|
||||||
|
GL11.glBindTexture(GL11.GL_TEXTURE_2D, prevTex);
|
||||||
|
|
||||||
|
Tessellator tessellator = Tessellator.getInstance();
|
||||||
|
BufferBuilder bb = tessellator.getBuffer();
|
||||||
|
bb.begin(GL11.GL_QUADS, VertexFormats.POSITION_TEXTURE);
|
||||||
|
bb.vertex(-1f, -1f, 0f).texture(0f, 0f).next();
|
||||||
|
bb.vertex( 1f, -1f, 0f).texture(1f, 0f).next();
|
||||||
|
bb.vertex( 1f, 1f, 0f).texture(1f, 1f).next();
|
||||||
|
bb.vertex(-1f, 1f, 0f).texture(0f, 1f).next();
|
||||||
|
tessellator.draw();
|
||||||
|
|
||||||
|
prevTex = GL11.glGetInteger(GL11.GL_TEXTURE_BINDING_2D);
|
||||||
|
GL11.glBindTexture(GL11.GL_TEXTURE_2D, srcColorTex);
|
||||||
|
GL11.glTexParameteri(GL11.GL_TEXTURE_2D, GL33.GL_TEXTURE_SWIZZLE_A, GL11.GL_ALPHA);
|
||||||
|
GL11.glBindTexture(GL11.GL_TEXTURE_2D, prevTex);
|
||||||
|
|
||||||
|
RenderSystem.getModelViewStack().pop();
|
||||||
|
RenderSystem.applyModelViewMatrix();
|
||||||
|
RenderSystem.setProjectionMatrix(savedProj, savedSorter);
|
||||||
|
RenderSystem.disableBlend();
|
||||||
|
RenderSystem.depthMask(true);
|
||||||
|
RenderSystem.enableDepthTest();
|
||||||
|
|
||||||
|
RenderSystem.defaultBlendFunc();
|
||||||
|
}
|
||||||
|
|
||||||
|
public ByteBuffer finishAveraged(int samples) {
|
||||||
|
GL30.glBindFramebuffer(GL30.GL_FRAMEBUFFER, fbo);
|
||||||
|
|
||||||
|
int pixels = width * height;
|
||||||
|
FloatBuffer fb = org.lwjgl.BufferUtils.createFloatBuffer(pixels * 4);
|
||||||
|
GL11.glReadPixels(0, 0, width, height, GL12.GL_BGRA, GL11.GL_FLOAT, fb);
|
||||||
|
|
||||||
|
ByteBuffer out = ByteBufferPool.allocate(pixels * 4);
|
||||||
|
float inv = 1f / samples;
|
||||||
|
for (int i = 0; i < pixels * 4; i++) {
|
||||||
|
float v = fb.get(i) * inv;
|
||||||
|
int b = (int) (v * 255f + 0.5f);
|
||||||
|
if (b < 0) b = 0; else if (b > 255) b = 255;
|
||||||
|
out.put((byte) b);
|
||||||
|
}
|
||||||
|
out.rewind();
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void unbind() {
|
||||||
|
net.minecraft.client.MinecraftClient.getInstance().getFramebuffer().beginWrite(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void close() {
|
||||||
|
if (fbo != 0) { GL30.glDeleteFramebuffers(fbo); fbo = 0; }
|
||||||
|
if (tex != 0) { GL11.glDeleteTextures(tex); tex = 0; }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//#else
|
||||||
|
//$$ // TODO: путь под MC>=12105 (GpuDevice/CommandEncoder) — заполнить позже.
|
||||||
|
//#endif
|
||||||
Reference in New Issue
Block a user