107 lines
4.1 KiB
Java
107 lines
4.1 KiB
Java
package net.minecraft.client.renderer;
|
|
|
|
import com.mojang.blaze3d.systems.RenderSystem;
|
|
import com.mojang.blaze3d.vertex.BufferBuilder;
|
|
import com.mojang.blaze3d.vertex.ByteBufferBuilder;
|
|
import com.mojang.blaze3d.vertex.MeshData;
|
|
import com.mojang.blaze3d.vertex.VertexConsumer;
|
|
import it.unimi.dsi.fastutil.objects.Object2ObjectSortedMaps;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.SequencedMap;
|
|
import javax.annotation.Nullable;
|
|
import net.minecraftforge.api.distmarker.Dist;
|
|
import net.minecraftforge.api.distmarker.OnlyIn;
|
|
|
|
@OnlyIn(Dist.CLIENT)
|
|
public interface MultiBufferSource {
|
|
static MultiBufferSource.BufferSource immediate(ByteBufferBuilder p_344614_) {
|
|
return immediateWithBuffers(Object2ObjectSortedMaps.emptyMap(), p_344614_);
|
|
}
|
|
|
|
static MultiBufferSource.BufferSource immediateWithBuffers(SequencedMap<RenderType, ByteBufferBuilder> p_342750_, ByteBufferBuilder p_344601_) {
|
|
return new MultiBufferSource.BufferSource(p_344601_, p_342750_);
|
|
}
|
|
|
|
VertexConsumer getBuffer(RenderType p_109903_);
|
|
|
|
@OnlyIn(Dist.CLIENT)
|
|
public static class BufferSource implements MultiBufferSource {
|
|
protected final ByteBufferBuilder sharedBuffer;
|
|
protected final SequencedMap<RenderType, ByteBufferBuilder> fixedBuffers;
|
|
protected final Map<RenderType, BufferBuilder> startedBuilders = new HashMap<>();
|
|
@Nullable
|
|
protected RenderType lastSharedType;
|
|
|
|
protected BufferSource(ByteBufferBuilder p_344223_, SequencedMap<RenderType, ByteBufferBuilder> p_344104_) {
|
|
this.sharedBuffer = p_344223_;
|
|
this.fixedBuffers = p_344104_;
|
|
}
|
|
|
|
@Override
|
|
public VertexConsumer getBuffer(RenderType p_109919_) {
|
|
BufferBuilder bufferbuilder = this.startedBuilders.get(p_109919_);
|
|
if (bufferbuilder != null && !p_109919_.canConsolidateConsecutiveGeometry()) {
|
|
this.endBatch(p_109919_, bufferbuilder);
|
|
bufferbuilder = null;
|
|
}
|
|
|
|
if (bufferbuilder != null) {
|
|
return bufferbuilder;
|
|
} else {
|
|
ByteBufferBuilder bytebufferbuilder = this.fixedBuffers.get(p_109919_);
|
|
if (bytebufferbuilder != null) {
|
|
bufferbuilder = new BufferBuilder(bytebufferbuilder, p_109919_.mode(), p_109919_.format());
|
|
} else {
|
|
if (this.lastSharedType != null) {
|
|
this.endBatch(this.lastSharedType);
|
|
}
|
|
|
|
bufferbuilder = new BufferBuilder(this.sharedBuffer, p_109919_.mode(), p_109919_.format());
|
|
this.lastSharedType = p_109919_;
|
|
}
|
|
|
|
this.startedBuilders.put(p_109919_, bufferbuilder);
|
|
return bufferbuilder;
|
|
}
|
|
}
|
|
|
|
public void endLastBatch() {
|
|
if (this.lastSharedType != null) {
|
|
this.endBatch(this.lastSharedType);
|
|
this.lastSharedType = null;
|
|
}
|
|
}
|
|
|
|
public void endBatch() {
|
|
this.endLastBatch();
|
|
|
|
for (RenderType rendertype : this.fixedBuffers.keySet()) {
|
|
this.endBatch(rendertype);
|
|
}
|
|
}
|
|
|
|
public void endBatch(RenderType p_109913_) {
|
|
BufferBuilder bufferbuilder = this.startedBuilders.remove(p_109913_);
|
|
if (bufferbuilder != null) {
|
|
this.endBatch(p_109913_, bufferbuilder);
|
|
}
|
|
}
|
|
|
|
private void endBatch(RenderType p_345497_, BufferBuilder p_344480_) {
|
|
MeshData meshdata = p_344480_.build();
|
|
if (meshdata != null) {
|
|
if (p_345497_.sortOnUpload()) {
|
|
ByteBufferBuilder bytebufferbuilder = this.fixedBuffers.getOrDefault(p_345497_, this.sharedBuffer);
|
|
meshdata.sortQuads(bytebufferbuilder, RenderSystem.getProjectionType().vertexSorting());
|
|
}
|
|
|
|
p_345497_.draw(meshdata);
|
|
}
|
|
|
|
if (p_345497_.equals(this.lastSharedType)) {
|
|
this.lastSharedType = null;
|
|
}
|
|
}
|
|
}
|
|
} |