|
1 | 1 | package net.irisshaders.batchedentityrendering.impl;
|
2 | 2 |
|
| 3 | +import com.mojang.blaze3d.pipeline.RenderTarget; |
| 4 | +import com.mojang.blaze3d.systems.RenderPass; |
| 5 | +import com.mojang.blaze3d.systems.RenderSystem; |
| 6 | +import com.mojang.blaze3d.textures.GpuTexture; |
3 | 7 | import com.mojang.blaze3d.vertex.ByteBufferBuilder;
|
4 | 8 | import com.mojang.blaze3d.vertex.VertexConsumer;
|
5 | 9 | import it.unimi.dsi.fastutil.objects.Object2ObjectSortedMaps;
|
|
19 | 23 | import java.util.LinkedHashMap;
|
20 | 24 | import java.util.List;
|
21 | 25 | import java.util.Map;
|
| 26 | +import java.util.OptionalDouble; |
| 27 | +import java.util.OptionalInt; |
22 | 28 | import java.util.function.Function;
|
23 | 29 |
|
24 | 30 | public class FullyBufferedMultiBufferSource extends MultiBufferSource.BufferSource implements MemoryTrackingBuffer, Groupable, WrappingMultiBufferSource {
|
@@ -137,18 +143,43 @@ public void endBatch() {
|
137 | 143 | profiler.push("iris draw buffers");
|
138 | 144 |
|
139 | 145 | for (RenderType type : renderOrder) {
|
140 |
| - if (!typeToSegment.containsKey(type)) continue; |
141 |
| - |
142 |
| - type.setupRenderState(); |
143 |
| - |
144 |
| - renderTypes += 1; |
145 |
| - |
146 |
| - for (BufferSegment segment : typeToSegment.getOrDefault(type, Collections.emptyList())) { |
147 |
| - segmentRenderer.drawInner(segment); |
148 |
| - drawCalls += 1; |
| 146 | + try { |
| 147 | + if (!typeToSegment.containsKey(type)) continue; |
| 148 | + |
| 149 | + type.setupRenderState(); |
| 150 | + |
| 151 | + RenderTarget renderTarget = type.getRenderTarget(); |
| 152 | + |
| 153 | + try (RenderPass renderPass = RenderSystem.getDevice() |
| 154 | + .createCommandEncoder() |
| 155 | + .createRenderPass( |
| 156 | + renderTarget.getColorTexture(), OptionalInt.empty(), renderTarget.useDepth ? renderTarget.getDepthTexture() : null, OptionalDouble.empty() |
| 157 | + )) { |
| 158 | + renderPass.setPipeline(type.getRenderPipeline()); |
| 159 | + if (RenderSystem.SCISSOR_STATE.isEnabled()) { |
| 160 | + renderPass.enableScissor(RenderSystem.SCISSOR_STATE); |
| 161 | + } |
| 162 | + |
| 163 | + for (int i = 0; i < 12; i++) { |
| 164 | + GpuTexture gpuTexture = RenderSystem.getShaderTexture(i); |
| 165 | + if (gpuTexture != null) { |
| 166 | + renderPass.bindSampler("Sampler" + i, gpuTexture); |
| 167 | + } |
| 168 | + } |
| 169 | + |
| 170 | + |
| 171 | + renderTypes += 1; |
| 172 | + |
| 173 | + for (BufferSegment segment : typeToSegment.getOrDefault(type, Collections.emptyList())) { |
| 174 | + segmentRenderer.drawInner(renderPass, segment); |
| 175 | + drawCalls += 1; |
| 176 | + } |
| 177 | + } |
| 178 | + |
| 179 | + type.clearRenderState(); |
| 180 | + } catch (Exception e) { |
| 181 | + throw new RuntimeException("Failed to draw " + type, e); |
149 | 182 | }
|
150 |
| - |
151 |
| - type.clearRenderState(); |
152 | 183 | }
|
153 | 184 |
|
154 | 185 | int targetClearTime = getTargetClearTime();
|
@@ -182,11 +213,34 @@ public void endBatchWithType(TransparencyType transparencyType) {
|
182 | 213 |
|
183 | 214 | type.setupRenderState();
|
184 | 215 |
|
| 216 | + RenderTarget renderTarget = type.getRenderTarget(); |
| 217 | + |
185 | 218 | renderTypes += 1;
|
186 | 219 |
|
187 |
| - for (BufferSegment segment : typeToSegment.getOrDefault(type, Collections.emptyList())) { |
188 |
| - segmentRenderer.drawInner(segment); |
189 |
| - drawCalls += 1; |
| 220 | + try (RenderPass renderPass = RenderSystem.getDevice() |
| 221 | + .createCommandEncoder() |
| 222 | + .createRenderPass( |
| 223 | + renderTarget.getColorTexture(), OptionalInt.empty(), renderTarget.useDepth ? renderTarget.getDepthTexture() : null, OptionalDouble.empty() |
| 224 | + )) { |
| 225 | + renderPass.setPipeline(type.getRenderPipeline()); |
| 226 | + if (RenderSystem.SCISSOR_STATE.isEnabled()) { |
| 227 | + renderPass.enableScissor(RenderSystem.SCISSOR_STATE); |
| 228 | + } |
| 229 | + |
| 230 | + for (int i = 0; i < 12; i++) { |
| 231 | + GpuTexture gpuTexture = RenderSystem.getShaderTexture(i); |
| 232 | + if (gpuTexture != null) { |
| 233 | + renderPass.bindSampler("Sampler" + i, gpuTexture); |
| 234 | + } |
| 235 | + } |
| 236 | + |
| 237 | + |
| 238 | + renderTypes += 1; |
| 239 | + |
| 240 | + for (BufferSegment segment : typeToSegment.getOrDefault(type, Collections.emptyList())) { |
| 241 | + segmentRenderer.drawInner(renderPass, segment); |
| 242 | + drawCalls += 1; |
| 243 | + } |
190 | 244 | }
|
191 | 245 |
|
192 | 246 | typeToSegment.remove(type);
|
|
0 commit comments