AstralTransRocketries/OpenComputers
1
0
Fork 0
mirror of https://github.com/MightyPirates/OpenComputers.git synced 2025-09-25 14:05:39 -04:00
Code Issues Packages Projects Releases Wiki Activity

WorldFluidContainerAnalytics

Browse Source
This commit is contained in:
MuXiu1997 2022-04-26 18:57:40 +08:00
parent ab430d1d92
commit e8266229d4
No known key found for this signature in database
GPG Key ID: 8E77811121A841D4
2 changed files with 69 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/scala/li/cil/oc/server/component/Transposer.scala
View File

@ -22,7 +22,7 @@ import scala.language.existentials
object Transposer {
abstract class Common extends prefab.ManagedEnvironment with traits.WorldInventoryAnalytics with traits.WorldTankAnalytics with traits.InventoryTransfer with DeviceInfo {
abstract class Common extends prefab.ManagedEnvironment with traits.WorldInventoryAnalytics with traits.WorldTankAnalytics with traits.WorldFluidContainerAnalytics with traits.InventoryTransfer with DeviceInfo {
override val node = api.Network.newNode(this, Visibility.Network).
withComponent("transposer").
withConnector().

68
src/main/scala/li/cil/oc/server/component/traits/WorldFluidContainerAnalytics.scala Normal file
View File

@ -0,0 +1,68 @@
package li.cil.oc.server.component.traits
import li.cil.oc.Settings
import li.cil.oc.api.machine.{Arguments, Callback, Context}
import li.cil.oc.server.component.result
import li.cil.oc.util.ExtendedArguments.extendedArguments
import li.cil.oc.util.InventoryUtils
import net.minecraft.inventory.IInventory
import net.minecraft.item.ItemStack
import net.minecraftforge.common.util.ForgeDirection
import net.minecraftforge.fluids.{FluidContainerRegistry, FluidStack, IFluidContainerItem}
trait WorldFluidContainerAnalytics extends WorldAware with SideRestricted with NetworkAware {
@Callback(doc = """function(side:number, slot:number):number -- Get the capacity of the fluid container in the specified slot of the inventory on the specified side of the device.""")
def getContainerCapacityInSlot(context: Context, args: Arguments): Array[AnyRef] = if (Settings.get.allowItemStackInspection) {
val facing = checkSideForAction(args, 0)
withInventory(facing, inventory => {
val stack = inventory.getStackInSlot(args.checkSlot(inventory, 1))
withFluidInfo(stack, (_, capacity) => result(capacity))
})
}
else result(Unit, "not enabled in config")
@Callback(doc = """function(side:number, slot:number):number -- Get the capacity the fluid container in the specified slot of the inventory on the specified side of the device.""")
def getContainerLevelInSlot(context: Context, args: Arguments): Array[AnyRef] = if (Settings.get.allowItemStackInspection) {
val facing = checkSideForAction(args, 0)
withInventory(facing, inventory => {
val stack = inventory.getStackInSlot(args.checkSlot(inventory, 1))
withFluidInfo(stack, (fluid, _) => result(fluid.amount))
})
}
else result(Unit, "not enabled in config")
@Callback(doc = """function(side:number, slot:number):table -- Get a description of the fluid in the fluid container in the specified slot of the inventory on the specified side of the device.""")
def getFluidInContainerInSlot(context: Context, args: Arguments): Array[AnyRef] = if (Settings.get.allowItemStackInspection) {
val facing = checkSideForAction(args, 0)
withInventory(facing, inventory => {
val stack = inventory.getStackInSlot(args.checkSlot(inventory, 1))
withFluidInfo(stack, (fluid, _) => result(fluid))
})
}
else result(Unit, "not enabled in config")
private def withFluidInfo(stack: ItemStack, f: (FluidStack, Int) => Array[AnyRef]) = {
val fluidInfo = if (FluidContainerRegistry.isFilledContainer(stack)) {
Option((FluidContainerRegistry.getFluidForFilledItem(stack), FluidContainerRegistry.getContainerCapacity(stack)))
}
else if (FluidContainerRegistry.isEmptyContainer(stack)) {
Option((new FluidStack(0, 0), FluidContainerRegistry.getContainerCapacity(stack)))
}
else stack.getItem match {
case from: IFluidContainerItem => Option((from.getFluid(stack), from.getCapacity(stack)))
case _ => None
}
fluidInfo match {
case Some((fluid, capacity)) => f(fluid, capacity)
case _ => result(Unit, "item is not a fluid container")
}
}
private def withInventory(side: ForgeDirection, f: IInventory => Array[AnyRef]) =
InventoryUtils.inventoryAt(position.offset(side)) match {
case Some(inventory) if inventory.isUseableByPlayer(fakePlayer) && mayInteract(position.offset(side), side.getOpposite) => f(inventory)
case _ => result(Unit, "no inventory")
}
}