DBot/overdrive_that_matters
DBot/overdrive_that_matters
3
1
Fork 0
Code Issues 63 Pull Requests Packages Releases Wiki Activity

Generalize slot iterator

Browse Source
This commit is contained in:
DBotThePony 2022-09-02 14:04:20 +07:00
parent 96eb3c3491
commit 732039b321
Signed by: DBot
GPG Key ID: DCC23B5715498507
2 changed files with 26 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
src/main/kotlin/ru/dbotthepony/mc/otm/core/MenuIterator.kt
View File

@ -1,30 +0,0 @@
package ru.dbotthepony.mc.otm.core
import net.minecraft.world.inventory.AbstractContainerMenu
import net.minecraft.world.item.ItemStack
class MenuIterator(private val menu: AbstractContainerMenu) : MutableIterator<ItemStack> {
private var index = 0
override fun hasNext(): Boolean {
return index < menu.slots.size
}
override fun next(): ItemStack {
if (index >= menu.slots.size) {
throw IllegalStateException("Already finished iterating")
}
return menu.slots[index++].item
}
override fun remove() {
if (index == 0) {
throw IllegalStateException("Never called next()")
}
menu.slots[index - 1].set(ItemStack.EMPTY)
}
}
fun AbstractContainerMenu.itemStackIterator() : MutableIterator<ItemStack> = MenuIterator(this)

26
src/main/kotlin/ru/dbotthepony/mc/otm/core/SlotIterator.kt Normal file
View File

@ -0,0 +1,26 @@
package ru.dbotthepony.mc.otm.core
import net.minecraft.world.inventory.AbstractContainerMenu
import net.minecraft.world.inventory.Slot
import net.minecraft.world.item.ItemStack
private class SlotIterator(private val parent: Iterator<Slot>) : MutableIterator<ItemStack> {
private var last: Slot? = null
override fun hasNext(): Boolean {
return parent.hasNext()
}
override fun next(): ItemStack {
return parent.next().also { last = it }.item
}
override fun remove() {
val last = last ?: throw IllegalStateException("Never called next()")
last.set(ItemStack.EMPTY)
}
}
fun AbstractContainerMenu.itemStackIterator() : MutableIterator<ItemStack> = SlotIterator(slots.iterator())
fun List<Slot>.itemStackIterator() : MutableIterator<ItemStack> = SlotIterator(iterator())
fun Iterator<Slot>.asItemStackIterator() : MutableIterator<ItemStack> = SlotIterator(this)