li/cil/oc/common/tileentity/Screen.scala - minecraft/opencomputers - Rivoreo Source Code Repositories

 package li.cil.oc.common.tileentity

 import li.cil.oc.Config
 import li.cil.oc.api.Network
 import li.cil.oc.api.network.Visibility
 import li.cil.oc.client.gui
 import li.cil.oc.client.{PacketSender => ClientPacketSender}
 import li.cil.oc.common.component.Screen.{Environment => ScreenEnvironment}
 import li.cil.oc.server.{PacketSender => ServerPacketSender}
 import li.cil.oc.util.PackedColor
 import net.minecraft.client.Minecraft
 import net.minecraft.nbt.NBTTagCompound
 import net.minecraft.util.AxisAlignedBB
 import net.minecraftforge.common.ForgeDirection
 import scala.collection.mutable

 class ScreenTier1 extends Screen {
   protected def tier = 0
 }

 class ScreenTier2 extends Screen {
   protected def tier = 1
 }

 class ScreenTier3 extends Screen {
   protected def tier = 2
 }

 abstract class Screen extends Rotatable with ScreenEnvironment {
   var currentGui: Option[gui.Screen] = None

   /**
    * Read and reset to false from the tile entity renderer. This is used to
    * keep rendering a little more efficient by compiling the displayed text
    * into an OpenGL display list, and only re-compiling that list when the
    * text/display has actually changed.
    */
   var hasChanged = true

   /**
    * Check for multi-block screen option in next update. We do this in the
    * update to avoid unnecessary checks on chunk unload.
    */
   private var shouldCheckForMultiBlock = true

   private val ordering = new Ordering[Screen] {
     def compare(a: Screen, b: Screen) =
       if (a.xCoord != b.xCoord) a.xCoord - b.xCoord
       else if (a.yCoord != b.yCoord) a.yCoord - b.yCoord
       else a.zCoord - b.zCoord
   }

   var width, height = 1

   var origin = this

   val screens = mutable.Set(this)

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

   def isOrigin = origin == this

   def localPosition = {
     val (x, y, _) = project(this)
     val (ox, oy, _) = project(origin)
     ((ox - x).abs, (oy - y).abs)
   }

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

   override def readFromNBT(nbt: NBTTagCompound) {
     super.readFromNBT(nbt)
     super.load(nbt)
   }

   override def writeToNBT(nbt: NBTTagCompound) {
     super.writeToNBT(nbt)
     super.save(nbt)
   }

   override def validate() {
     super.validate()
     if (worldObj.isRemote) ClientPacketSender.sendScreenBufferRequest(this)
   }

   override def invalidate() {
     super.invalidate()
     if (currentGui.isDefined) Minecraft.getMinecraft.displayGuiScreen(null)
     screens.clone().foreach(_.checkMultiBlock())
   }

   override def onRotationChanged() = screens.clone().foreach(_.checkMultiBlock())

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

   override def updateEntity() {
     super.updateEntity()
     if (node != null && node.network == null) {
       Network.joinOrCreateNetwork(worldObj, xCoord, yCoord, zCoord)
     }
     if (shouldCheckForMultiBlock) {
       // Make sure we merge in a deterministic order, to avoid getting
       // different results on server and client due to the update order
       // differing between the two. This also saves us from having to save
       // any multi-block specific state information.
       // We use a very primitive hash for the coordinates, which should be
       // good enough for... "normal" screen sizes.
       val pending = mutable.SortedSet(this)(ordering)
       val queue = mutable.Queue(this)
       while (queue.nonEmpty) {
         val current = queue.dequeue()
         val (x, y, z) = project(current)
         def tryQueue(dx: Int, dy: Int) {
           val (nx, ny, nz) = unproject(x + dx, y + dy, z)
           worldObj.getBlockTileEntity(nx, ny, nz) match {
             case s: Screen if s.pitch == pitch && s.yaw == yaw && pending.add(s) => queue += s
             case _ => // Ignore.
           }
         }
         tryQueue(-1, 0)
         tryQueue(1, 0)
         tryQueue(0, -1)
         tryQueue(0, 1)
       }
       // Perform actual merges.
       while (pending.nonEmpty) {
         val current = pending.firstKey
         while (current.tryMerge()) {}
         current.screens.foreach {
           screen =>
             screen.shouldCheckForMultiBlock = false
             screen.hasChanged = true
             pending.remove(screen)
             queue += screen
         }
         val bounds = current.origin.getRenderBoundingBox
         worldObj.markBlockRangeForRenderUpdate(bounds.minX.toInt, bounds.minY.toInt, bounds.minZ.toInt,
           bounds.maxX.toInt, bounds.maxY.toInt, bounds.maxZ.toInt)
       }
       // Update visibility after everything is done, to avoid noise.
       queue.foreach(screen =>
         if (screen.isOrigin) {
           if (!worldObj.isRemote) screen.node.setVisibility(Visibility.Network)
         }
         else {
           if (!worldObj.isRemote) screen.node.setVisibility(Visibility.None)
           val s = screen.instance
           val (w, h) = s.resolution
           s.fill(0, 0, w, h, ' ')
         }
       )
     }
   }

   def checkMultiBlock() {
     shouldCheckForMultiBlock = true
     width = 1
     height = 1
     origin = this
     screens.clear()
     screens += this
   }

   private def tryMerge() = {
     val (x, y, z) = project(origin)
     def tryMergeTowards(dx: Int, dy: Int) = {
       val (nx, ny, nz) = unproject(x + dx, y + dy, z)
       worldObj.getBlockTileEntity(nx, ny, nz) match {
         case s: Screen if s.tier == tier && s.pitch == pitch && s.yaw == yaw && !screens.contains(s) =>
           val (sx, sy, _) = project(s.origin)
           val canMergeAlongX = sy == y && s.height == height && s.width + width <= Config.maxScreenWidth
           val canMergeAlongY = sx == x && s.width == width && s.height + height <= Config.maxScreenHeight
           if (canMergeAlongX || canMergeAlongY) {
             val (newOrigin) =
               if (canMergeAlongX) {
                 if (sx < x) s.origin else origin
               }
               else {
                 if (sy < y) s.origin else origin
               }
             val (newWidth, newHeight) =
               if (canMergeAlongX) (width + s.width, height)
               else (width, height + s.height)
             val newScreens = screens ++ s.screens
             for (screen <- newScreens) {
               screen.width = newWidth
               screen.height = newHeight
               screen.origin = newOrigin
               screen.screens ++= newScreens // It's a set, so there won't be duplicates.
             }
             true
           }
           else false // Cannot merge.
         case _ => false
       }
     }
     tryMergeTowards(width, 0) || tryMergeTowards(0, height) || tryMergeTowards(-1, 0) || tryMergeTowards(0, -1)
   }

   private def project(t: Screen) = {
     def dot(f: ForgeDirection, s: Screen) = f.offsetX * s.xCoord + f.offsetY * s.yCoord + f.offsetZ * s.zCoord
     (dot(toGlobal(ForgeDirection.EAST), t), dot(toGlobal(ForgeDirection.UP), t), dot(toGlobal(ForgeDirection.SOUTH), t))
   }

   private def unproject(x: Int, y: Int, z: Int) = {
     def dot(f: ForgeDirection) = f.offsetX * x + f.offsetY * y + f.offsetZ * z
     (dot(toLocal(ForgeDirection.EAST)), dot(toLocal(ForgeDirection.UP)), dot(toLocal(ForgeDirection.SOUTH)))
   }

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

   override def getRenderBoundingBox =
     if ((width == 1 && height == 1) || !isOrigin) super.getRenderBoundingBox
     else {
       val (sx, sy, sz) = unproject(width, height, 1)
       val ox = xCoord + (if (sx < 0) 1 else 0)
       val oy = yCoord + (if (sy < 0) 1 else 0)
       val oz = zCoord + (if (sz < 0) 1 else 0)
       val b = AxisAlignedBB.getAABBPool.getAABB(ox, oy, oz, ox + sx, oy + sy, oz + sz)
       b.setBounds(b.minX min b.maxX, b.minY min b.maxY, b.minZ min b.maxZ,
         b.minX max b.maxX, b.minY max b.maxY, b.minZ max b.maxZ)
       b
     }

   override def getMaxRenderDistanceSquared = if (isOrigin) super.getMaxRenderDistanceSquared else 0

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

   override def onScreenColorChange(foreground: Int, background: Int) {
     super.onScreenColorChange(foreground, background)
     if (!worldObj.isRemote) {
       worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
       ServerPacketSender.sendScreenColorChange(this, foreground, background)
     }
   }

   override def onScreenCopy(col: Int, row: Int, w: Int, h: Int, tx: Int, ty: Int) = {
     super.onScreenCopy(col, row, w, h, tx, ty)
     if (worldObj.isRemote) {
       currentGui.foreach(_.updateText())
       hasChanged = true
     }
     else {
       worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
       ServerPacketSender.sendScreenCopy(this, col, row, w, h, tx, ty)
     }
   }

   override def onScreenDepthChange(depth: PackedColor.Depth.Value) {
     super.onScreenDepthChange(depth)
     if (!worldObj.isRemote) {
       worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
       ServerPacketSender.sendScreenDepthChange(this, depth)
     }
   }

   override def onScreenFill(col: Int, row: Int, w: Int, h: Int, c: Char) = {
     super.onScreenFill(col, row, w, h, c)
     if (worldObj.isRemote) {
       currentGui.foreach(_.updateText())
       hasChanged = true
     }
     else {
       worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
       ServerPacketSender.sendScreenFill(this, col, row, w, h, c)
     }
   }

   override def onScreenResolutionChange(w: Int, h: Int) = {
     super.onScreenResolutionChange(w, h)
     if (worldObj.isRemote) {
       currentGui.foreach(_.changeSize(w, h))
       hasChanged = true
     }
     else {
       worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
       ServerPacketSender.sendScreenResolutionChange(this, w, h)
     }
   }

   override def onScreenSet(col: Int, row: Int, s: String) = {
     super.onScreenSet(col, row, s)
     if (worldObj.isRemote) {
       currentGui.foreach(_.updateText())
       hasChanged = true
     }
     else {
       worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
       ServerPacketSender.sendScreenSet(this, col, row, s)
     }
   }
 }
	package li.cil.oc.common.tileentity

	import li.cil.oc.Config
	import li.cil.oc.api.Network
	import li.cil.oc.api.network.Visibility
	import li.cil.oc.client.gui
	import li.cil.oc.client.{PacketSender => ClientPacketSender}
	import li.cil.oc.common.component.Screen.{Environment => ScreenEnvironment}
	import li.cil.oc.server.{PacketSender => ServerPacketSender}
	import li.cil.oc.util.PackedColor
	import net.minecraft.client.Minecraft
	import net.minecraft.nbt.NBTTagCompound
	import net.minecraft.util.AxisAlignedBB
	import net.minecraftforge.common.ForgeDirection
	import scala.collection.mutable

	class ScreenTier1 extends Screen {
	protected def tier = 0
	}

	class ScreenTier2 extends Screen {
	protected def tier = 1
	}

	class ScreenTier3 extends Screen {
	protected def tier = 2
	}

	abstract class Screen extends Rotatable with ScreenEnvironment {
	var currentGui: Option[gui.Screen] = None

	/**
	* Read and reset to false from the tile entity renderer. This is used to
	* keep rendering a little more efficient by compiling the displayed text
	* into an OpenGL display list, and only re-compiling that list when the
	* text/display has actually changed.
	*/
	var hasChanged = true

	/**
	* Check for multi-block screen option in next update. We do this in the
	* update to avoid unnecessary checks on chunk unload.
	*/
	private var shouldCheckForMultiBlock = true

	private val ordering = new Ordering[Screen] {
	def compare(a: Screen, b: Screen) =
	if (a.xCoord != b.xCoord) a.xCoord - b.xCoord
	else if (a.yCoord != b.yCoord) a.yCoord - b.yCoord
	else a.zCoord - b.zCoord
	}

	var width, height = 1

	var origin = this

	val screens = mutable.Set(this)

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

	def isOrigin = origin == this

	def localPosition = {
	val (x, y, _) = project(this)
	val (ox, oy, _) = project(origin)
	((ox - x).abs, (oy - y).abs)
	}

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

	override def readFromNBT(nbt: NBTTagCompound) {
	super.readFromNBT(nbt)
	super.load(nbt)
	}

	override def writeToNBT(nbt: NBTTagCompound) {
	super.writeToNBT(nbt)
	super.save(nbt)
	}

	override def validate() {
	super.validate()
	if (worldObj.isRemote) ClientPacketSender.sendScreenBufferRequest(this)
	}

	override def invalidate() {
	super.invalidate()
	if (currentGui.isDefined) Minecraft.getMinecraft.displayGuiScreen(null)
	screens.clone().foreach(_.checkMultiBlock())
	}

	override def onRotationChanged() = screens.clone().foreach(_.checkMultiBlock())

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

	override def updateEntity() {
	super.updateEntity()
	if (node != null && node.network == null) {
	Network.joinOrCreateNetwork(worldObj, xCoord, yCoord, zCoord)
	}
	if (shouldCheckForMultiBlock) {
	// Make sure we merge in a deterministic order, to avoid getting
	// different results on server and client due to the update order
	// differing between the two. This also saves us from having to save
	// any multi-block specific state information.
	// We use a very primitive hash for the coordinates, which should be
	// good enough for... "normal" screen sizes.
	val pending = mutable.SortedSet(this)(ordering)
	val queue = mutable.Queue(this)
	while (queue.nonEmpty) {
	val current = queue.dequeue()
	val (x, y, z) = project(current)
	def tryQueue(dx: Int, dy: Int) {
	val (nx, ny, nz) = unproject(x + dx, y + dy, z)
	worldObj.getBlockTileEntity(nx, ny, nz) match {
	case s: Screen if s.pitch == pitch && s.yaw == yaw && pending.add(s) => queue += s
	case _ => // Ignore.
	}
	}
	tryQueue(-1, 0)
	tryQueue(1, 0)
	tryQueue(0, -1)
	tryQueue(0, 1)
	}
	// Perform actual merges.
	while (pending.nonEmpty) {
	val current = pending.firstKey
	while (current.tryMerge()) {}
	current.screens.foreach {
	screen =>
	screen.shouldCheckForMultiBlock = false
	screen.hasChanged = true
	pending.remove(screen)
	queue += screen
	}
	val bounds = current.origin.getRenderBoundingBox
	worldObj.markBlockRangeForRenderUpdate(bounds.minX.toInt, bounds.minY.toInt, bounds.minZ.toInt,
	bounds.maxX.toInt, bounds.maxY.toInt, bounds.maxZ.toInt)
	}
	// Update visibility after everything is done, to avoid noise.
	queue.foreach(screen =>
	if (screen.isOrigin) {
	if (!worldObj.isRemote) screen.node.setVisibility(Visibility.Network)
	}
	else {
	if (!worldObj.isRemote) screen.node.setVisibility(Visibility.None)
	val s = screen.instance
	val (w, h) = s.resolution
	s.fill(0, 0, w, h, ' ')
	}
	)
	}
	}

	def checkMultiBlock() {
	shouldCheckForMultiBlock = true
	width = 1
	height = 1
	origin = this
	screens.clear()
	screens += this
	}

	private def tryMerge() = {
	val (x, y, z) = project(origin)
	def tryMergeTowards(dx: Int, dy: Int) = {
	val (nx, ny, nz) = unproject(x + dx, y + dy, z)
	worldObj.getBlockTileEntity(nx, ny, nz) match {
	case s: Screen if s.tier == tier && s.pitch == pitch && s.yaw == yaw && !screens.contains(s) =>
	val (sx, sy, _) = project(s.origin)
	val canMergeAlongX = sy == y && s.height == height && s.width + width <= Config.maxScreenWidth
	val canMergeAlongY = sx == x && s.width == width && s.height + height <= Config.maxScreenHeight
	if (canMergeAlongX \|\| canMergeAlongY) {
	val (newOrigin) =
	if (canMergeAlongX) {
	if (sx < x) s.origin else origin
	}
	else {
	if (sy < y) s.origin else origin
	}
	val (newWidth, newHeight) =
	if (canMergeAlongX) (width + s.width, height)
	else (width, height + s.height)
	val newScreens = screens ++ s.screens
	for (screen <- newScreens) {
	screen.width = newWidth
	screen.height = newHeight
	screen.origin = newOrigin
	screen.screens ++= newScreens // It's a set, so there won't be duplicates.
	}
	true
	}
	else false // Cannot merge.
	case _ => false
	}
	}
	tryMergeTowards(width, 0) \|\| tryMergeTowards(0, height) \|\| tryMergeTowards(-1, 0) \|\| tryMergeTowards(0, -1)
	}

	private def project(t: Screen) = {
	def dot(f: ForgeDirection, s: Screen) = f.offsetX * s.xCoord + f.offsetY * s.yCoord + f.offsetZ * s.zCoord
	(dot(toGlobal(ForgeDirection.EAST), t), dot(toGlobal(ForgeDirection.UP), t), dot(toGlobal(ForgeDirection.SOUTH), t))
	}

	private def unproject(x: Int, y: Int, z: Int) = {
	def dot(f: ForgeDirection) = f.offsetX * x + f.offsetY * y + f.offsetZ * z
	(dot(toLocal(ForgeDirection.EAST)), dot(toLocal(ForgeDirection.UP)), dot(toLocal(ForgeDirection.SOUTH)))
	}

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

	override def getRenderBoundingBox =
	if ((width == 1 && height == 1) \|\| !isOrigin) super.getRenderBoundingBox
	else {
	val (sx, sy, sz) = unproject(width, height, 1)
	val ox = xCoord + (if (sx < 0) 1 else 0)
	val oy = yCoord + (if (sy < 0) 1 else 0)
	val oz = zCoord + (if (sz < 0) 1 else 0)
	val b = AxisAlignedBB.getAABBPool.getAABB(ox, oy, oz, ox + sx, oy + sy, oz + sz)
	b.setBounds(b.minX min b.maxX, b.minY min b.maxY, b.minZ min b.maxZ,
	b.minX max b.maxX, b.minY max b.maxY, b.minZ max b.maxZ)
	b
	}

	override def getMaxRenderDistanceSquared = if (isOrigin) super.getMaxRenderDistanceSquared else 0

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

	override def onScreenColorChange(foreground: Int, background: Int) {
	super.onScreenColorChange(foreground, background)
	if (!worldObj.isRemote) {
	worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
	ServerPacketSender.sendScreenColorChange(this, foreground, background)
	}
	}

	override def onScreenCopy(col: Int, row: Int, w: Int, h: Int, tx: Int, ty: Int) = {
	super.onScreenCopy(col, row, w, h, tx, ty)
	if (worldObj.isRemote) {
	currentGui.foreach(_.updateText())
	hasChanged = true
	}
	else {
	worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
	ServerPacketSender.sendScreenCopy(this, col, row, w, h, tx, ty)
	}
	}

	override def onScreenDepthChange(depth: PackedColor.Depth.Value) {
	super.onScreenDepthChange(depth)
	if (!worldObj.isRemote) {
	worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
	ServerPacketSender.sendScreenDepthChange(this, depth)
	}
	}

	override def onScreenFill(col: Int, row: Int, w: Int, h: Int, c: Char) = {
	super.onScreenFill(col, row, w, h, c)
	if (worldObj.isRemote) {
	currentGui.foreach(_.updateText())
	hasChanged = true
	}
	else {
	worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
	ServerPacketSender.sendScreenFill(this, col, row, w, h, c)
	}
	}

	override def onScreenResolutionChange(w: Int, h: Int) = {
	super.onScreenResolutionChange(w, h)
	if (worldObj.isRemote) {
	currentGui.foreach(_.changeSize(w, h))
	hasChanged = true
	}
	else {
	worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
	ServerPacketSender.sendScreenResolutionChange(this, w, h)
	}
	}

	override def onScreenSet(col: Int, row: Int, s: String) = {
	super.onScreenSet(col, row, s)
	if (worldObj.isRemote) {
	currentGui.foreach(_.updateText())
	hasChanged = true
	}
	else {
	worldObj.markTileEntityChunkModified(xCoord, yCoord, zCoord, this)
	ServerPacketSender.sendScreenSet(this, col, row, s)
	}
	}
	}