bdlib/src/net/bdew/lib/data/DataSlotTank.scala

/*
 * Copyright (c) bdew, 2013
 * https://github.com/bdew/bdlib
 *
 * This mod is distributed under the terms of the Minecraft Mod Public
 * License 1.0, or MMPL. Please check the contents of the license located in
 * https://raw.github.com/bdew/bdlib/master/MMPL-1.0.txt
 */

package net.bdew.lib.data

import net.minecraftforge.fluids.{IFluidTank, FluidTank, FluidStack}
import net.minecraft.nbt.NBTTagCompound
import net.bdew.lib.data.base.{UpdateKind, TileDataSlots, DataSlot}

abstract class DataSlotTankBase(sz: Int) extends FluidTank(sz) with IFluidTank with DataSlot {
  var oldStack: FluidStack = null
  var size: Int

  parent.serverTick.listen(checkUpdate)

  updateKind = Set(UpdateKind.GUI, UpdateKind.SAVE)

  def checkUpdate() {
    if (!isSame(oldStack, fluid)) {
      oldStack = if (fluid == null) null else fluid.copy()
      parent.dataSlotChanged(this)
    }
  }

  def isSame(v1: FluidStack, v2: FluidStack): Boolean = {
    if (v1 == null && v2 == null) return true
    if (v1 == null || v2 == null) return false
    return v1.isFluidStackIdentical(v2)
  }

  def save(t: NBTTagCompound, k: UpdateKind.Value) {
    val tag = new NBTTagCompound()
    writeToNBT(tag)
    if (k == UpdateKind.GUI)
      tag.setInteger("size", size)
    t.setCompoundTag(name, tag)
  }

  def load(t: NBTTagCompound, k: UpdateKind.Value) {
    val tag = t.getCompoundTag(name)
    readFromNBT(tag)
    if (k == UpdateKind.GUI)
      size = tag.getInteger("size")
  }
}

case class DataSlotTank(name: String, parent: TileDataSlots, var size: Int) extends DataSlotTankBase(size)

case class DataSlotTankRestricted(name: String, parent: TileDataSlots, var size: Int, fluidID: Int) extends DataSlotTankBase(size) {
  def fill(amount: Int, doFill: Boolean) = super.fill(new FluidStack(fluidID, amount), doFill)

  override def fill(resource: FluidStack, doFill: Boolean): Int = {
    if ((resource != null) && (resource.fluidID != fluidID)) return 0
    return super.fill(resource, doFill)
  }
}