dynamic buffer source
This commit is contained in:
parent
cf12971861
commit
279e28095b
GPG Key ID:
DCC23B5715498507
@ -0,0 +1,324 @@
|
||||
package ru.dbotthepony.mc.otm.client.render
|
||||
|
||||
import com.google.common.collect.ImmutableList
|
||||
import com.mojang.blaze3d.vertex.BufferBuilder
|
||||
import com.mojang.blaze3d.vertex.VertexConsumer
|
||||
import it.unimi.dsi.fastutil.objects.Reference2ObjectOpenHashMap
|
||||
import it.unimi.dsi.fastutil.objects.ReferenceArraySet
|
||||
import net.minecraft.client.renderer.MultiBufferSource
|
||||
import net.minecraft.client.renderer.RenderType
|
||||
import net.minecraft.client.renderer.Sheets
|
||||
|
||||
private fun equals(existing: ImmutableList<RenderType>?, types: Array<out RenderType>): Boolean {
|
||||
if (types.isEmpty()) {
|
||||
return existing == null
|
||||
}
|
||||
|
||||
if (existing == null) {
|
||||
return false
|
||||
}
|
||||
|
||||
if (existing.size != types.size) {
|
||||
return false
|
||||
}
|
||||
|
||||
for (i in types.indices) {
|
||||
if (types[i] !== existing[i]) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
private fun equals(existing: ImmutableList<RenderType>?, types: ImmutableList<RenderType>): Boolean {
|
||||
if (existing === types) {
|
||||
return true
|
||||
}
|
||||
|
||||
if (types.isEmpty()) {
|
||||
return existing == null
|
||||
}
|
||||
|
||||
if (existing == null) {
|
||||
return false
|
||||
}
|
||||
|
||||
if (existing.size != types.size) {
|
||||
return false
|
||||
}
|
||||
|
||||
for (i in types.indices) {
|
||||
if (types[i] !== existing[i]) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
class DynamicBufferSource : MultiBufferSource {
|
||||
private class State(
|
||||
val type: RenderType,
|
||||
var after: ImmutableList<RenderType>? = null,
|
||||
val immutableAfter: Boolean = false
|
||||
) {
|
||||
var priority = -1
|
||||
var dirty: Boolean = false
|
||||
val builder by lazy {
|
||||
BufferBuilder(type.bufferSize())
|
||||
}
|
||||
}
|
||||
|
||||
private val bufferList = ArrayList<State>().also {
|
||||
var next = State(Sheets.solidBlockSheet(), null, true)
|
||||
it.add(next)
|
||||
next = State(Sheets.cutoutBlockSheet(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(Sheets.bannerSheet(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(Sheets.translucentCullBlockSheet(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(Sheets.shieldSheet(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(Sheets.bedSheet(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(Sheets.shulkerBoxSheet(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(Sheets.signSheet(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(Sheets.chestSheet(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.translucentNoCrumbling(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.armorGlint(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.armorEntityGlint(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.glint(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.glintDirect(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.glintTranslucent(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.entityGlint(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.entityGlintDirect(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
next = State(RenderType.waterMask(), ImmutableList.of(next.type), true)
|
||||
it.add(next)
|
||||
}
|
||||
|
||||
private fun determineHeight(input: State, seen: MutableSet<State>) {
|
||||
if (!seen.add(input)) {
|
||||
throw IllegalStateException("Recursive dependencies detected when going through ${input.type}: ${seen.joinToString(", ") { it.type.toString() + "[${it.after?.joinToString(", ")}]" }}")
|
||||
}
|
||||
|
||||
if (input.after == null) {
|
||||
input.priority = 0
|
||||
} else {
|
||||
for (type in input.after!!) {
|
||||
for (input2 in bufferList) {
|
||||
if (input2.type == type) {
|
||||
if (input2.priority == -1) {
|
||||
determineHeight(input2, seen)
|
||||
}
|
||||
|
||||
input.priority = input.priority.coerceAtLeast(input2.priority + 1)
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (input.priority == -1) {
|
||||
input.priority = 0
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private fun sortBufferList() {
|
||||
for (state in bufferList) {
|
||||
state.priority = -1
|
||||
}
|
||||
|
||||
for (state in bufferList) {
|
||||
val seen = ReferenceArraySet<State>()
|
||||
determineHeight(state, seen)
|
||||
}
|
||||
|
||||
bufferList.sortWith { a, b ->
|
||||
// most referenced buffers go first
|
||||
b.priority.compareTo(a.priority)
|
||||
}
|
||||
}
|
||||
|
||||
private val buffers = Reference2ObjectOpenHashMap<RenderType, State>().also {
|
||||
for (v in bufferList) it[v.type] = v
|
||||
}
|
||||
|
||||
override fun getBuffer(renderType: RenderType): VertexConsumer {
|
||||
var getState = buffers[renderType]
|
||||
|
||||
if (getState == null) {
|
||||
getState = State(renderType)
|
||||
buffers[renderType] = getState
|
||||
bufferList.add(getState)
|
||||
getState.dirty = true
|
||||
return getState.builder.also {
|
||||
it.begin(renderType.mode(), renderType.format())
|
||||
}
|
||||
}
|
||||
|
||||
if (!getState.dirty) {
|
||||
getState.dirty = true
|
||||
getState.builder.begin(renderType.mode(), renderType.format())
|
||||
}
|
||||
|
||||
return getState.builder
|
||||
}
|
||||
|
||||
fun getBuffer(renderType: RenderType, after: RenderType): VertexConsumer {
|
||||
var getState = buffers[renderType]
|
||||
|
||||
if (getState == null) {
|
||||
getState = State(renderType, ImmutableList.of(after))
|
||||
buffers[renderType] = getState
|
||||
bufferList.add(getState)
|
||||
getState.dirty = true
|
||||
sortBufferList()
|
||||
return getState.builder.also {
|
||||
it.begin(renderType.mode(), renderType.format())
|
||||
}
|
||||
}
|
||||
|
||||
if (!getState.immutableAfter && getState.after?.getOrNull(0) !== after) {
|
||||
getState.after = ImmutableList.of(after)
|
||||
sortBufferList()
|
||||
}
|
||||
|
||||
if (!getState.dirty) {
|
||||
getState.dirty = true
|
||||
getState.builder.begin(renderType.mode(), renderType.format())
|
||||
}
|
||||
|
||||
return getState.builder
|
||||
}
|
||||
|
||||
fun getBuffer(renderType: RenderType, vararg after: RenderType): VertexConsumer {
|
||||
var getState = buffers[renderType]
|
||||
|
||||
if (getState == null) {
|
||||
getState = State(renderType, ImmutableList.copyOf(after))
|
||||
buffers[renderType] = getState
|
||||
bufferList.add(getState)
|
||||
getState.dirty = true
|
||||
sortBufferList()
|
||||
return getState.builder.also {
|
||||
it.begin(renderType.mode(), renderType.format())
|
||||
}
|
||||
}
|
||||
|
||||
if (!getState.immutableAfter && !equals(getState.after, after)) {
|
||||
getState.after = ImmutableList.copyOf(after)
|
||||
sortBufferList()
|
||||
}
|
||||
|
||||
if (!getState.dirty) {
|
||||
getState.dirty = true
|
||||
getState.builder.begin(renderType.mode(), renderType.format())
|
||||
}
|
||||
|
||||
return getState.builder
|
||||
}
|
||||
|
||||
fun getBuffer(renderType: RenderType, after: ImmutableList<RenderType>): VertexConsumer {
|
||||
var getState = buffers[renderType]
|
||||
|
||||
if (getState == null) {
|
||||
getState = State(renderType, after)
|
||||
buffers[renderType] = getState
|
||||
bufferList.add(getState)
|
||||
getState.dirty = true
|
||||
sortBufferList()
|
||||
return getState.builder.also {
|
||||
it.begin(renderType.mode(), renderType.format())
|
||||
}
|
||||
}
|
||||
|
||||
if (!getState.immutableAfter && !equals(getState.after, after)) {
|
||||
getState.after = after
|
||||
sortBufferList()
|
||||
}
|
||||
|
||||
if (!getState.dirty) {
|
||||
getState.dirty = true
|
||||
getState.builder.begin(renderType.mode(), renderType.format())
|
||||
}
|
||||
|
||||
return getState.builder
|
||||
}
|
||||
|
||||
fun registerType(renderType: RenderType, after: RenderType) {
|
||||
var getState = buffers[renderType]
|
||||
|
||||
if (getState == null) {
|
||||
getState = State(renderType, ImmutableList.of(after))
|
||||
buffers[renderType] = getState
|
||||
bufferList.add(getState)
|
||||
sortBufferList()
|
||||
} else if (!getState.immutableAfter && getState.after?.getOrNull(0) !== after) {
|
||||
getState.after = ImmutableList.of(after)
|
||||
sortBufferList()
|
||||
}
|
||||
}
|
||||
|
||||
fun registerType(renderType: RenderType, vararg after: RenderType) {
|
||||
var getState = buffers[renderType]
|
||||
|
||||
if (getState == null) {
|
||||
getState = State(renderType, ImmutableList.copyOf(after))
|
||||
buffers[renderType] = getState
|
||||
bufferList.add(getState)
|
||||
sortBufferList()
|
||||
} else if (!getState.immutableAfter && !equals(getState.after, after)) {
|
||||
getState.after = ImmutableList.copyOf(after)
|
||||
sortBufferList()
|
||||
}
|
||||
}
|
||||
|
||||
fun registerType(renderType: RenderType, after: ImmutableList<RenderType>) {
|
||||
var getState = buffers[renderType]
|
||||
|
||||
if (getState == null) {
|
||||
getState = State(renderType, after)
|
||||
buffers[renderType] = getState
|
||||
bufferList.add(getState)
|
||||
sortBufferList()
|
||||
} else if (!getState.immutableAfter && !equals(getState.after, after)) {
|
||||
getState.after = after
|
||||
sortBufferList()
|
||||
}
|
||||
}
|
||||
|
||||
fun endBatch() {
|
||||
for (state in bufferList) {
|
||||
if (state.dirty) {
|
||||
state.dirty = false
|
||||
state.type.end(state.builder, 0, 0, 0)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fun endBatch(type: RenderType) {
|
||||
val state = buffers[type] ?: return
|
||||
|
||||
if (state.dirty) {
|
||||
state.dirty = false
|
||||
type.end(state.builder, 0, 0, 0)
|
||||
}
|
||||
}
|
||||
|
||||
companion object {
|
||||
val GUI = DynamicBufferSource()
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user