src/main/scala/li/cil/oc/common/entity/Drone.scala

package li.cil.oc.common.entity

import cpw.mods.fml.relauncher.Side
import cpw.mods.fml.relauncher.SideOnly
import li.cil.oc.OpenComputers
import li.cil.oc.Settings
import li.cil.oc.api
import li.cil.oc.api.Driver
import li.cil.oc.api.Machine
import li.cil.oc.api.driver.item.Memory
import li.cil.oc.api.driver.item.Processor
import li.cil.oc.api.internal
import li.cil.oc.api.machine.MachineHost
import li.cil.oc.api.network._
import li.cil.oc.common.GuiType
import li.cil.oc.common.Slot
import li.cil.oc.common.inventory.ComponentInventory
import li.cil.oc.common.inventory.Inventory
import li.cil.oc.server.component
import li.cil.oc.util.ExtendedNBT._
import li.cil.oc.util.ItemUtils
import net.minecraft.entity.Entity
import net.minecraft.entity.item.EntityItem
import net.minecraft.entity.player.EntityPlayer
import net.minecraft.item.ItemStack
import net.minecraft.nbt.NBTTagCompound
import net.minecraft.util.AxisAlignedBB
import net.minecraft.util.MovingObjectPosition
import net.minecraft.util.Vec3
import net.minecraft.world.World

class Drone(val world: World) extends Entity(world) with MachineHost with internal.Drone {
  // Some basic constants.
  val gravity = 0.05f // low for slow fall (float down)
  val drag = 0.8f
  val bounds = AxisAlignedBB.getBoundingBox(-8, -3, -8, 8, 3, 8)
  val maxAcceleration = 0.1f
  val maxVelocity = 0.4f

  // Rendering stuff, purely eyecandy.
  val targetFlapAngles = Array.fill(4, 2)(0f)
  val flapAngles = Array.fill(4, 2)(0f)
  var nextFlapChange = 0
  var bodyAngle = math.random.toFloat * 90
  var angularVelocity = 0f
  var nextAngularVelocityChange = 0

  // Logic stuff, components, machine and such.
  val info = new ItemUtils.MicrocontrollerData()
  val machine = if (!world.isRemote) Machine.create(this) else null
  val control = if (!world.isRemote) new component.Drone(this) else null
  val components = new ComponentInventory {
    override def host = Drone.this

    override def items = info.components.map(Option(_))

    override def getSizeInventory = info.components.length

    override def markDirty() {}

    override def isItemValidForSlot(slot: Int, stack: ItemStack) = true

    override def isUseableByPlayer(player: EntityPlayer) = true

    override def node = Option(machine).map(_.node).orNull

    override def onConnect(node: Node) {}

    override def onDisconnect(node: Node) {}

    override def onMessage(message: Message) {}
  }
  val inventory = new Inventory {
    var items = Array.fill[Option[ItemStack]](8)(None)

    override def getSizeInventory = items.length

    override def getInventoryStackLimit = 64

    override def markDirty() {} // TODO update client GUI?

    override def isItemValidForSlot(slot: Int, stack: ItemStack) = slot >= 0 && slot < getSizeInventory

    override def isUseableByPlayer(player: EntityPlayer) = player.getDistanceSqToEntity(Drone.this) < 64
  }

  // ----------------------------------------------------------------------- //

  override def getBoundingBox = bounds.copy()

  override def getCollisionBox(entity: Entity) = bounds.copy()

  override def canBeCollidedWith = true

  override def canBePushed = true

  override def isEntityInvulnerable = super.isEntityInvulnerable

  // ----------------------------------------------------------------------- //

  override def xPosition = posX

  override def yPosition = posY

  override def zPosition = posZ

  override def markChanged() {}

  // ----------------------------------------------------------------------- //

  override def cpuArchitecture = info.components.map(stack => (stack, Driver.driverFor(stack, getClass))).collectFirst {
    case (stack, driver: Processor) if driver.slot(stack) == Slot.CPU => driver.architecture(stack)
  }.orNull

  override def callBudget = info.components.foldLeft(0.0)((sum, item) => sum + (Option(item) match {
    case Some(stack) => Option(Driver.driverFor(stack, getClass)) match {
      case Some(driver: Processor) if driver.slot(stack) == Slot.CPU => Settings.get.callBudgets(driver.tier(stack))
      case _ => 0
    }
    case _ => 0
  }))

  override def installedMemory = info.components.foldLeft(0)((sum, item) => sum + (Option(item) match {
    case Some(stack) => Option(Driver.driverFor(stack, getClass)) match {
      case Some(driver: Memory) => driver.amount(stack)
      case _ => 0
    }
    case _ => 0
  }))

  override def maxComponents = 32

  override def componentSlot(address: String) = -1 // TODO

  override def markForSaving() {}

  override def onMachineConnect(node: Node) {}

  override def onMachineDisconnect(node: Node) {}

  // ----------------------------------------------------------------------- //

  override def entityInit() {
    dataWatcher.addObject(2, byte2Byte(0: Byte))
    dataWatcher.addObject(3, float2Float(0f))
    dataWatcher.addObject(4, float2Float(0f))
    dataWatcher.addObject(5, float2Float(0f))
    dataWatcher.addObject(6, float2Float(0f))
    dataWatcher.addObject(7, byte2Byte(0: Byte))
  }

  def isRunning = dataWatcher.getWatchableObjectByte(2) != 0
  def targetX = dataWatcher.getWatchableObjectFloat(3)
  def targetY = dataWatcher.getWatchableObjectFloat(4)
  def targetZ = dataWatcher.getWatchableObjectFloat(5)
  def targetAcceleration = dataWatcher.getWatchableObjectFloat(6)
  def selectedSlot = dataWatcher.getWatchableObjectByte(7) & 0xFF

  private def setRunning(value: Boolean) = dataWatcher.updateObject(2, byte2Byte(if (value) 1: Byte else 0: Byte))
  // Round target values to low accuracy to avoid floating point errors accumulating.
  def targetX_=(value: Float): Unit = dataWatcher.updateObject(3, float2Float(math.round(value * 5) / 5f))
  def targetY_=(value: Float): Unit = dataWatcher.updateObject(4, float2Float(math.round(value * 5) / 5f))
  def targetZ_=(value: Float): Unit = dataWatcher.updateObject(5, float2Float(math.round(value * 5) / 5f))
  def targetAcceleration_=(value: Float): Unit = dataWatcher.updateObject(6, float2Float(math.max(0, math.min(maxAcceleration, value))))
  def selectedSlot_=(value: Int) = dataWatcher.updateObject(7, byte2Byte(value.toByte))

  @SideOnly(Side.CLIENT)
  override def setPositionAndRotation2(x: Double, y: Double, z: Double, yaw: Float, pitch: Float, data: Int) {
    // Only set exact position if we're too far away from the server's
    // position, otherwise keep interpolating. This removes jitter and
    // is good enough for drones.
    if (!isRunning || getDistanceSq(x, y, z) > 1) {
      super.setPositionAndRotation(x, y, z, yaw, pitch)
    }
    else {
      targetX = x.toFloat
      targetY = y.toFloat
      targetZ = z.toFloat
    }
  }

  override def setDead() {
    super.setDead()
    if (!world.isRemote) {
      machine.stop()
      machine.node.remove()
      components.saveComponents()
      val stack = api.Items.get("drone").createItemStack(1)
      info.save(stack)
      val entity = new EntityItem(world, posX, posY, posZ, stack)
      entity.delayBeforeCanPickup = 15
      world.spawnEntityInWorld(entity)
    }
  }

  override def onEntityUpdate() {
    super.onEntityUpdate()

    if (!world.isRemote) {
      if (isInWater) { // We're not water-proof!
        machine.stop()
      }
      machine.node.asInstanceOf[Connector].changeBuffer(100)
      machine.update()
      components.updateComponents()
      setRunning(machine.isRunning)
    }
    else if (isRunning) {
      // Client side update; occasionally update wing pitch and rotation to
      // make the drones look a bit more dynamic.
      val rng = world.rand
      nextFlapChange -= 1
      nextAngularVelocityChange -= 1

      if (nextFlapChange < 0) {
        nextFlapChange = 5 + rng.nextInt(10)
        for (i <- 0 until 2) {
          val flap = rng.nextInt(targetFlapAngles.length)
          targetFlapAngles(flap)(0) = math.toRadians(rng.nextFloat() * 4 - 2).toFloat
          targetFlapAngles(flap)(1) = math.toRadians(rng.nextFloat() * 4 - 2).toFloat
        }
      }

      if (nextAngularVelocityChange < 0) {
        if (angularVelocity != 0) {
          angularVelocity = 0
          nextAngularVelocityChange = 20
        }
        else {
          angularVelocity = if (rng.nextBoolean()) 0.1f else -0.1f
          nextAngularVelocityChange = 100
        }
      }

      // Interpolate wing rotations.
      (flapAngles, targetFlapAngles).zipped.foreach((f, t) => {
        f(0) = f(0) * 0.7f + t(0) * 0.3f
        f(1) = f(1) * 0.7f + t(1) * 0.3f
      })

      // Update body rotation.
      bodyAngle += angularVelocity
    }

    if (isRunning) {
      val delta = Vec3.createVectorHelper(targetX - posX, targetY - posY, targetZ - posZ)
      val acceleration = math.min(targetAcceleration, delta.lengthVector())
      val velocity = delta.normalize()
      velocity.xCoord = motionX + velocity.xCoord * acceleration
      velocity.yCoord = motionY + velocity.yCoord * acceleration
      velocity.zCoord = motionZ + velocity.zCoord * acceleration
      motionX = math.max(-maxVelocity, math.min(maxVelocity, velocity.xCoord))
      motionY = math.max(-maxVelocity, math.min(maxVelocity, velocity.yCoord))
      motionZ = math.max(-maxVelocity, math.min(maxVelocity, velocity.zCoord))
    }
    else {
      // No power, free fall: engage!
      motionY -= gravity
    }
    moveEntity(motionX, motionY, motionZ)

    // Make sure we don't get infinitely faster.
    motionX *= drag
    motionY *= drag
    motionZ *= drag
  }

  override def hitByEntity(entity: Entity) = {
    if (isRunning) {
      val direction = Vec3.createVectorHelper(entity.posX - posX, entity.posY - posY, entity.posZ - posZ).normalize()
      if (!world.isRemote) {
        if (Settings.get.inputUsername)
          machine.signal("hit", double2Double(direction.xCoord), double2Double(direction.zCoord), double2Double(direction.yCoord), entity.getCommandSenderName)
        else
          machine.signal("hit", double2Double(direction.xCoord), double2Double(direction.zCoord), double2Double(direction.yCoord))
      }
      motionX -= direction.xCoord
      motionY -= direction.yCoord
      motionZ -= direction.zCoord
    }
    super.hitByEntity(entity)
  }

  override def getPickedResult(target: MovingObjectPosition) = {
    val stack = api.Items.get("drone").createItemStack(1)
    info.save(stack)
    stack
  }

  def preparePowerUp() {
    targetX = math.floor(posX).toFloat + 0.5f
    targetY = math.floor(posY).toFloat + 0.5f
    targetZ = math.floor(posZ).toFloat + 0.5f
    targetAcceleration = maxAcceleration

    api.Network.joinNewNetwork(machine.node)
    components.connectComponents()
    machine.node.connect(control.node)
  }

  override def interactFirst(player: EntityPlayer) = {
    if (player.isSneaking) {
      kill()
    }
    else if (!world.isRemote) {
      player.openGui(OpenComputers, GuiType.Drone.id, world, getEntityId, 0, 0)
    }
    true
  }

  // ----------------------------------------------------------------------- //

  override def readEntityFromNBT(nbt: NBTTagCompound): Unit = {
    info.load(nbt.getCompoundTag("info"))
    if (!world.isRemote) {
      machine.load(nbt.getCompoundTag("machine"))
      control.load(nbt.getCompoundTag("control"))
      components.load(nbt.getCompoundTag("components"))
      inventory.load(nbt.getCompoundTag("inventory"))

      api.Network.joinNewNetwork(machine.node)
      components.connectComponents()
      machine.node.connect(control.node)
    }
    targetX = nbt.getFloat("targetX")
    targetY = nbt.getFloat("targetY")
    targetZ = nbt.getFloat("targetZ")
    targetAcceleration = nbt.getFloat("targetAcceleration")
  }

  override def writeEntityToNBT(nbt: NBTTagCompound): Unit = {
    components.saveComponents()
    nbt.setNewCompoundTag("info", info.save)
    if (!world.isRemote) {
      nbt.setNewCompoundTag("machine", machine.save)
      nbt.setNewCompoundTag("control", control.save)
      nbt.setNewCompoundTag("components", components.save)
      nbt.setNewCompoundTag("inventory", inventory.save)
    }
    nbt.setFloat("targetX", targetX)
    nbt.setFloat("targetY", targetY)
    nbt.setFloat("targetZ", targetZ)
    nbt.setFloat("targetAcceleration", targetAcceleration)
  }
}