blob: 6f03a4a14b13f6c9f31cf1a3afaf36393efd4321 [file] [log] [blame] [raw]
package li.cil.oc.server.component
import li.cil.oc.Settings
import li.cil.oc.api.Network
import li.cil.oc.api.component.TextBuffer.ColorDepth
import li.cil.oc.api.network._
import li.cil.oc.common.component.ManagedComponent
import li.cil.oc.common.tileentity
import li.cil.oc.util.PackedColor
import net.minecraft.nbt.NBTTagCompound
import net.minecraft.util.StatCollector
import li.cil.oc.api.component.TextBuffer
abstract class GraphicsCard extends ManagedComponent {
val node = Network.newNode(this, Visibility.Neighbors).
withComponent("gpu").
withConnector().
create()
protected val maxResolution: (Int, Int)
protected val maxDepth: ColorDepth
private var screenAddress: Option[String] = None
private var screenInstance: Option[TextBuffer] = None
private def screen(f: (TextBuffer) => Array[AnyRef]) = screenInstance match {
case Some(screen) => screen.synchronized(f(screen))
case _ => Array(Unit, "no screen")
}
// ----------------------------------------------------------------------- //
override val canUpdate = true
override def update() {
super.update()
if (node.network != null && screenInstance.isEmpty && screenAddress.isDefined) {
Option(node.network.node(screenAddress.get)) match {
case Some(node: Node) if node.host.isInstanceOf[TextBuffer] =>
screenInstance = Some(node.host.asInstanceOf[TextBuffer])
case _ =>
// This could theoretically happen after loading an old address, but
// if the screen either disappeared between saving and now or changed
// type. The first scenario is more likely, and could happen if the
// chunk the screen is in isn't loaded when the chunk the GPU is in
// gets loaded.
screenAddress = None
}
}
}
@Callback
def bind(context: Context, args: Arguments): Array[AnyRef] = {
val address = args.checkString(0)
node.network.node(address) match {
case null => result(Unit, "invalid address")
case node: Node if node.host.isInstanceOf[TextBuffer] =>
screenAddress = Option(address)
screenInstance = Some(node.host.asInstanceOf[TextBuffer])
screen(s => {
val (gmw, gmh) = maxResolution
val smw = s.getMaximumWidth
val smh = s.getMaximumHeight
s.setResolution(math.min(gmw, smw), math.min(gmh, smh))
s.setColorDepth(ColorDepth.values.apply(math.min(maxDepth.ordinal, s.getMaximumColorDepth.ordinal)))
s.setForegroundColor(0xFFFFFF)
s.setBackgroundColor(0x000000)
result(true)
})
case _ => result(Unit, "not a screen")
}
}
@Callback(direct = true)
def getBackground(context: Context, args: Arguments): Array[AnyRef] =
screen(s => result(s.getBackgroundColor, s.isBackgroundFromPalette))
def setBackground(context: Context, args: Arguments): Array[AnyRef] = {
val color = args.checkInteger(0)
screen(s => {
val oldColor = s.getBackgroundColor
val oldIsPalette = s.isBackgroundFromPalette
s.setBackgroundColor(color, args.count > 1 && args.checkBoolean(1))
result(oldColor, oldIsPalette)
})
}
@Callback(direct = true)
def getForeground(context: Context, args: Arguments): Array[AnyRef] =
screen(s => result(s.getForegroundColor, s.isForegroundFromPalette))
def setForeground(context: Context, args: Arguments): Array[AnyRef] = {
val color = args.checkInteger(0)
screen(s => {
val oldColor = s.getForegroundColor
val oldIsPalette = s.isForegroundFromPalette
s.setForegroundColor(color, args.count > 1 && args.checkBoolean(1))
result(oldColor, oldIsPalette)
})
}
@Callback(direct = true)
def getPaletteColor(context: Context, args: Arguments): Array[AnyRef] = {
val index = args.checkInteger(0)
screen(s => result(s.getPaletteColor(index)))
}
@Callback
def setPaletteColor(context: Context, args: Arguments): Array[AnyRef] = {
val index = args.checkInteger(0)
val color = args.checkInteger(1)
context.pause(0.1)
screen(s => {
val oldColor = s.getPaletteColor(index)
s.setPaletteColor(index, color)
result(oldColor)
})
}
@Callback(direct = true)
def getDepth(context: Context, args: Arguments): Array[AnyRef] =
screen(s => result(PackedColor.Depth.bits(s.getColorDepth)))
@Callback
def setDepth(context: Context, args: Arguments): Array[AnyRef] = {
val depth = args.checkInteger(0)
screen(s => {
val oldDepth = s.getColorDepth
depth match {
case 1 => s.setColorDepth(ColorDepth.OneBit)
case 4 if maxDepth.ordinal >= ColorDepth.FourBit.ordinal => s.setColorDepth(ColorDepth.FourBit)
case 8 if maxDepth.ordinal >= ColorDepth.EightBit.ordinal => s.setColorDepth(ColorDepth.EightBit)
case _ => throw new IllegalArgumentException("unsupported depth")
}
result(oldDepth)
})
}
@Callback(direct = true)
def maxDepth(context: Context, args: Arguments): Array[AnyRef] =
screen(s => result(PackedColor.Depth.bits(ColorDepth.values.apply(math.min(maxDepth.ordinal, s.getMaximumColorDepth.ordinal)))))
@Callback(direct = true)
def getResolution(context: Context, args: Arguments): Array[AnyRef] =
screen(s => result(s.getWidth, s.getHeight))
@Callback
def setResolution(context: Context, args: Arguments): Array[AnyRef] = {
val w = args.checkInteger(0)
val h = args.checkInteger(1)
val (mw, mh) = maxResolution
// Even though the buffer itself checks this again, we need this here for
// the minimum of screen and GPU resolution.
if (w < 1 || h < 1 || w > mw || h > mw || h * w > mw * mh)
throw new IllegalArgumentException("unsupported resolution")
screen(s => result(s.setResolution(w, h)))
}
@Callback(direct = true)
def maxResolution(context: Context, args: Arguments): Array[AnyRef] =
screen(s => {
val (gmw, gmh) = maxResolution
val smw = s.getMaximumWidth
val smh = s.getMaximumHeight
result(math.min(gmw, smw), math.min(gmh, smh))
})
@Callback(direct = true)
def get(context: Context, args: Arguments): Array[AnyRef] = {
val x = args.checkInteger(0) - 1
val y = args.checkInteger(1) - 1
screen(s => {
result(s.get(x, y), s.getForegroundColor(x, y), s.getBackgroundColor(x, y), s.isForegroundFromPalette(x, y), s.isBackgroundFromPalette(x, y))
})
}
def set(context: Context, args: Arguments): Array[AnyRef] = {
val x = args.checkInteger(0) - 1
val y = args.checkInteger(1) - 1
val value = args.checkString(2)
val vertical = args.count > 3 && args.checkBoolean(3)
screen(s => {
if (consumePower(value.length, Settings.get.gpuSetCost)) {
s.set(x, y, value, vertical)
result(true)
}
else result(Unit, "not enough energy")
})
}
def copy(context: Context, args: Arguments): Array[AnyRef] = {
val x = args.checkInteger(0) - 1
val y = args.checkInteger(1) - 1
val w = args.checkInteger(2)
val h = args.checkInteger(3)
val tx = args.checkInteger(4)
val ty = args.checkInteger(5)
screen(s => {
if (consumePower(w * h, Settings.get.gpuCopyCost)) {
s.copy(x, y, w, h, tx, ty)
result(true)
}
else result(Unit, "not enough energy")
})
}
def fill(context: Context, args: Arguments): Array[AnyRef] = {
val x = args.checkInteger(0) - 1
val y = args.checkInteger(1) - 1
val w = args.checkInteger(2)
val h = args.checkInteger(3)
val value = args.checkString(4)
if (value.length == 1) screen(s => {
val c = value.charAt(0)
val cost = if (c == ' ') Settings.get.gpuClearCost else Settings.get.gpuFillCost
if (consumePower(w * h, cost)) {
s.fill(x, y, w, h, value.charAt(0))
result(true)
}
else {
result(Unit, "not enough energy")
}
})
else throw new Exception("invalid fill value")
}
private def consumePower(n: Double, cost: Double) = node.tryChangeBuffer(-n * cost)
// ----------------------------------------------------------------------- //
override def onMessage(message: Message) {
super.onMessage(message)
if (message.name == "computer.stopped" && node.isNeighborOf(message.source)) {
screenInstance match {
case Some(buffer) => buffer.synchronized {
val w = buffer.getWidth
val h = buffer.getHeight
buffer.setForegroundColor(0xFFFFFF)
message.source.host match {
case machine: machine.Machine if machine.lastError != null =>
if (buffer.getColorDepth.ordinal > ColorDepth.OneBit.ordinal) buffer.setBackgroundColor(0x0000FF)
else buffer.setBackgroundColor(0x000000)
buffer.fill(0, 0, w, h, ' ')
try {
val message = "Unrecoverable error:\n" + StatCollector.translateToLocal(machine.lastError) + "\n"
val wrapRegEx = s"(.{1,${math.max(1, w - 2)}})\\s".r
val lines = wrapRegEx.replaceAllIn(message, m => m.group(1) + "\n").lines.toArray
for ((line, idx) <- lines.zipWithIndex) {
val col = (w - line.length) / 2
val row = (h - lines.length) / 2 + idx
buffer.set(col, row, line, false)
}
}
catch {
case t: Throwable => t.printStackTrace()
}
case _ =>
buffer.setBackgroundColor(0x000000)
buffer.fill(0, 0, w, h, ' ')
}
}
case _ =>
}
}
}
override def onDisconnect(node: Node) {
super.onDisconnect(node)
if (node == this.node || screenAddress.exists(_ == node.address)) {
screenAddress = None
screenInstance = None
}
}
// ----------------------------------------------------------------------- //
override def load(nbt: NBTTagCompound) {
super.load(nbt)
if (nbt.hasKey("screen")) {
nbt.getString("screen") match {
case screen: String if !screen.isEmpty => screenAddress = Some(screen)
case _ => screenAddress = None
}
screenInstance = None
}
}
override def save(nbt: NBTTagCompound) {
super.save(nbt)
if (screenAddress.isDefined) {
nbt.setString("screen", screenAddress.get)
}
}
}
object GraphicsCard {
// IMPORTANT: usually methods with side effects should *not* be direct
// callbacks to avoid the massive headache synchronizing them ensues, in
// particular when it comes to world saving. I'm making an exception for
// screens, though since they'd be painfully sluggish otherwise. This also
// means we have to use a somewhat nasty trick in common.component.Buffer's
// save function: we wait for all computers in the same network to finish
// their current execution and then pause them, to ensure the state of the
// buffer is "clean", meaning the computer has the correct state when it is
// saved in turn. If we didn't, a computer might change a screen after it was
// saved, but before the computer was saved, leading to mismatching states in
// the save file - a Bad Thing (TM).
class Tier1 extends GraphicsCard {
protected val maxDepth = Settings.screenDepthsByTier(0)
protected val maxResolution = Settings.screenResolutionsByTier(0)
@Callback(direct = true, limit = 1)
override def copy(context: Context, args: Arguments) = super.copy(context, args)
@Callback(direct = true, limit = 1)
override def fill(context: Context, args: Arguments) = super.fill(context, args)
@Callback(direct = true, limit = 4)
override def set(context: Context, args: Arguments) = super.set(context, args)
@Callback(direct = true, limit = 2)
override def setBackground(context: Context, args: Arguments) = super.setBackground(context, args)
@Callback(direct = true, limit = 2)
override def setForeground(context: Context, args: Arguments) = super.setForeground(context, args)
}
class Tier2 extends GraphicsCard {
protected val maxDepth = Settings.screenDepthsByTier(1)
protected val maxResolution = Settings.screenResolutionsByTier(1)
@Callback(direct = true, limit = 2)
override def copy(context: Context, args: Arguments) = super.copy(context, args)
@Callback(direct = true, limit = 4)
override def fill(context: Context, args: Arguments) = super.fill(context, args)
@Callback(direct = true, limit = 8)
override def set(context: Context, args: Arguments) = super.set(context, args)
@Callback(direct = true, limit = 4)
override def setBackground(context: Context, args: Arguments) = super.setBackground(context, args)
@Callback(direct = true, limit = 4)
override def setForeground(context: Context, args: Arguments) = super.setForeground(context, args)
}
class Tier3 extends GraphicsCard {
protected val maxDepth = Settings.screenDepthsByTier(2)
protected val maxResolution = Settings.screenResolutionsByTier(2)
@Callback(direct = true, limit = 4)
override def copy(context: Context, args: Arguments) = super.copy(context, args)
@Callback(direct = true, limit = 8)
override def fill(context: Context, args: Arguments) = super.fill(context, args)
@Callback(direct = true, limit = 16)
override def set(context: Context, args: Arguments) = super.set(context, args)
@Callback(direct = true, limit = 8)
override def setBackground(context: Context, args: Arguments) = super.setBackground(context, args)
@Callback(direct = true, limit = 8)
override def setForeground(context: Context, args: Arguments) = super.setForeground(context, args)
}
}