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

Stream'y

Browse Source
This commit is contained in:
DBotThePony 2022-11-19 16:03:00 +07:00
parent 471ddb49e5
commit 3ca8f11c48
Signed by: DBot
GPG Key ID: DCC23B5715498507
1 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/main/kotlin/ru/dbotthepony/mc/otm/matter/MatterManager.kt
View File

@ -967,6 +967,18 @@ object MatterManager {
return get(value).hasMatterValue return get(value).hasMatterValue
} }
private fun reduce(a: IMatterValue, b: IMatterValue): IMatterValue {
if (a.hasMatterValue && b.hasMatterValue) {
return MatterValue(a.matter + b.matter, a.complexity + b.complexity)
} else if (a.hasMatterValue) {
return a
} else if (b.hasMatterValue) {
return b
} else {
return IMatterValue.Companion
}
}
private fun get(value: ItemStack, level: Int, accountForStackSize: Boolean): IMatterValue { private fun get(value: ItemStack, level: Int, accountForStackSize: Boolean): IMatterValue {
if (value.isEmpty) { if (value.isEmpty) {
return IMatterValue.Companion return IMatterValue.Companion
@ -997,29 +1009,20 @@ object MatterManager {
val drive = value.getCapability(MatteryCapability.DRIVE).orNull() val drive = value.getCapability(MatteryCapability.DRIVE).orNull()
if (drive != null && drive.storageType === OverdriveThatMatters.INSTANCE.ITEM_STORAGE()) { if (drive != null && drive.storageType == OverdriveThatMatters.INSTANCE.ITEM_STORAGE()) {
for (item in (drive as IMatteryDrive<ItemStackWrapper>).stacks) { (drive as IMatteryDrive<ItemStackWrapper>).stacks
val tuple = get(item.stack.stack, level + 1, true) .map { it.stack.stack }
.filter { !it.isEmpty }
if (tuple.hasMatterValue) { .map { get(it, level + 1, true) }
matter += tuple .reduce(::reduce)
undamagedMatterValue += tuple .ifPresent {
} matter += it
undamagedMatterValue += it
} }
} }
value.getCapability(ForgeCapabilities.ITEM_HANDLER).ifPresentK { value.getCapability(ForgeCapabilities.ITEM_HANDLER).ifPresentK {
it.stream().filter { !it.isEmpty }.map { get(it, level + 1, true) }.reduce { a, b -> it.stream().filter { !it.isEmpty }.map { get(it, level + 1, true) }.reduce(::reduce).ifPresent {
if (a.hasMatterValue && b.hasMatterValue) {
return@reduce MatterValue(a.matter + b.matter, a.complexity + b.complexity)
} else if (a.hasMatterValue) {
return@reduce a
} else if (b.hasMatterValue) {
return@reduce b
} else {
return@reduce IMatterValue.Companion
}
}.ifPresent {
matter += it matter += it
undamagedMatterValue += it undamagedMatterValue += it
} }