LuPPC/src/lua/core/textgpu.lua

local textgpu = {}

local write = io.write
local flush = io.flush

local background = 0x000000 -- "0"
local foreground = 0xFFFFFF -- "0"

local tbuffer = {}
local bbuffer = {}
local fbuffer = {}

local palette = {
  [0] = 0x0f0f0f,
  0x1e1e1e,
  0x2d2d2d,
  0x3c3c3c,
  0x4b4b4b,
  0x5a5a5a,
  0x696969,
  0x787878,
  0x878787,
  0x969696,
  0xa5a5a5,
  0xb4b4b4,
  0xc3c3c3,
  0xd2d2d2,
  0xe1e1e1,
  0xf0f0f0
}

local function prepareBuffers(w, h)
  local tbline = (" "):rep(w)
  local bbline = ("\0\0\0"):rep(w) -- ("0"):rep(w)
  local fbline = ("\255\255\255"):rep(w) -- ("7"):rep(w)
  for i=1, h do
    tbuffer[i] = tbline
    bbuffer[i] = bbline
    fbuffer[i] = fbline
  end
end

local function setForeground(c)
  local r = c & 0xFF0000
  local g = c & 0x00FF00
  local b = c & 0x0000FF
  io.write("\27[38;2;", r, ";", g, ";", b, "m")
end

local function setBackground(c)
  local r = c & 0xFF0000
  local g = c & 0x00FF00
  local b = c & 0x0000FF
  io.write("\27[48;2;", r, ";", g, ";", b, "m")
end

local usub
local function insertString(main, sub, at)
  checkArg(1, main, "string")
  checkArg(2, sub, "string")
  checkArg(3, at, "number")

  return (usub(main, 1, at - 1) or "")
    .. sub .. (usub(main, at + (utf8.len(sub) or 0)))
end

function textgpu.start()
  usub = modules.sandbox.unicode.sub
  local _height = 0
  local gpu = {}
  function gpu.bind() return false, "This is static bound gpu" end
  function gpu.setBackground(color, isPaletteIndex)
    checkArg(1, color, "number")
    checkArg(2, isPaletteIndex, "boolean", "nil")
    if isPaletteIndex then
      if color > 15 or color < 0 then
        error("invalid palette index", 2)
      end
      return palette[color]
    end
    local old = background
    background = color -- tostring(math.floor(modules.color.nearest(color, mapping)))
    setBackground(background)
    --write("\x1b[4" .. background .. "m")
    flush()
    return old -- mapping[old]
  end
  function gpu.setForeground(color, isPaletteIndex)
    checkArg(1, color, "number")
    checkArg(2, isPaletteIndex, "boolean", "nil")
    if isPaletteIndex then
      if color < 0 or color > 15 then
        error("invalid palette index", 2)
      end
      return palette[color]
    end
    local old = foreground
    foreground = color -- tostring(math.floor(modules.color.nearest(color, mapping)))
    setForeground(foreground)
    --write("\x1b[3" .. foreground .. "m")
    flush()
    return old -- mapping[old]
  end
  function gpu.getBackground()
    return mapping[background], false
  end
  function gpu.getForeground()
    return mapping[foreground], false
  end
  function gpu.getPaletteColor(index)
    if index < 0 or index > 15 then
      error("invalid palette index", 2)
    end
    return palette[index]
  end
  function gpu.setPaletteColor(index, value)
    if index < 0 or index > 15 then
      error("invalid palette index", 2)
    end
    palette[index] = value
    return true
  end
  function gpu.maxDepth()
    return 4
  end
  function gpu.setDepth()
    return true
  end
  function gpu.getDepth()
    return 4
  end
  function gpu.maxResolution()
    return termutils.getSize()
  end
  function gpu.getResolution()
    return termutils.getSize()
  end
  function gpu.getViewport()
    return termutils.getSize()
  end
  function gpu.setViewport(w, h)
    checkArg(1, w, "number")
    checkArg(2, h, "number")
    return false, "Viewport not supported for this gpu"
  end
  function gpu.setResolution(w, h)
    checkArg(1, w, "number")
    checkArg(2, h, "number")
    return false, "Non resizeable gpu"
  end
  function gpu.get(x, y)
    checkArg(1, x, "number")
    checkArg(2, y, "number")
    
    return tbuffer[y]:sub(x,x),
           fbuffer[y]:sub(x*3-2,x*3):unpack(">I3"),
           bbuffer[y]:sub(x*3-2,x*3):unpack(">I3")
  end
  function gpu.set(x, y, value, vertical)
    checkArg(1, x, "number")
    checkArg(2, y, "number")
    checkArg(3, value, "string")
    checkArg(4, vertical, "boolean", "nil")
    x = math.floor(x)
    y = math.floor(y)
    if not vertical then
      if not tbuffer[y] then
        native.log("GPU Set failed: under buffer")
        return false
      end
      tbuffer[y] = insertString(tbuffer[y], value, x)
      bbuffer[y] = insertString(bbuffer[y],
        string.pack(">I3", background):rep(utf8.len(value)), x*3-2)
      fbuffer[y] = insertString(fbuffer[y],
        string.pack(">I3", foreground):rep(utf8.len(value)), x*3-2)
      write("\x1b[" .. y .. ";" .. x .. "H" .. value)
    else
      --TODO: Buffers!
      write("\x1b[" .. y .. ";" .. x .. "H")
      value:gsub("([%z\1-\127\194-\244][\128-\191]*)", function(c)
        write(c .. "\x1b[D\x1b[B")
      end)
    end
    flush()
    return true
  end
  function gpu.copy(x, y, w, h, tx, ty) --TODO: Check(check X multiple times)
    checkArg(1, x, "number")
    checkArg(2, y, "number")
    checkArg(3, w, "number")
    checkArg(4, h, "number")
    checkArg(5, tx, "number")
    checkArg(6, ty, "number")
    local ttbuf = {}
    local btbuf = {}
    local ftbuf = {}
    for i=1, h do
      if i + y - 2 <= _height and i + y > 1 then
        ttbuf[i] = tbuffer[y + i - 1] and usub(tbuffer[y + i - 1], x, x + w - 1) or (" "):rep(w)
        btbuf[i] = bbuffer[y + i - 1] and
          bbuffer[y + i - 1]:sub(x*3-2, (x + w - 1)*3) or string.pack(">I3", background):rep(w)
        ftbuf[i] = fbuffer[y + i - 1] and
          fbuffer[y + i - 1]:sub(x*3-2, (x + w - 1)*3) or string.pack(">I3", foreground):rep(w)
      else
        ttbuf[i] = (" "):rep(w)
        btbuf[i] = string.pack(">I3", background):rep(w)
        ftbuf[i] = string.pack(">I3", foreground):rep(w)
      end
    end
    local bg = background
    local fg = foreground

    for i=1, h do
      local line, linex
      local lwrite = false
      for j=1, w do
        if btbuf[i]:sub(j*3-2,j*3):unpack(">I3") ~= bg then
          lwrite = true
        end
        if ftbuf[i]:sub(j*3-2,j*3):unpack(">I3") ~= fg then
          lwrite = true
        end
        if not line then linex = j end
        line = (line or "")
        if lwrite then
          local wx = (tx + x + linex - 1)|0
          local wy = (ty + y + i - 1)|0
          if tbuffer[wy] then
            setBackground(bg)
            setForeground(foreground)
            --write("\x1b[4" .. bg .. "m\x1b[3" .. fg .. "m
            write("\x1b[" .. wy .. ";" .. wx .. "H" .. line)
            tbuffer[wy] = insertString(tbuffer[wy], line, wx)
            bbuffer[wy] = insertString(bbuffer[wy],
              string.pack(">I3", bg):rep(utf8.len(line)), wx*3-2)
            fbuffer[wy] = insertString(fbuffer[wy],
              string.pack(">I3", fg):rep(utf8.len(line)), wx*3-2)
          end
        
          bg = btbuf[i]:sub(j*3-2,j*3):unpack(">I3")
          fg = ftbuf[i]:sub(j*3-2,j*3):unpack(">I3")
          line = nil
          linex = nil
          lwrite = false
        end
        if not line then linex = j end
        line = (line or "") .. usub(ttbuf[i], j,j)
      end
      if line then
        local wx = (tx + x + linex - 1)|0
        local wy = (ty + y + i - 1)|0
        if tbuffer[wy] then
          setBackground(bg)
          setForeground(fg)
          --write("\x1b[4" .. bg .. "m\x1b[3" .. fg .. "m
          write("\x1b[" .. wy .. ";" .. wx .. "H" .. line)
          tbuffer[wy] = insertString(tbuffer[wy], line, wx)
          bbuffer[wy] = insertString(bbuffer[wy],
            string.pack(">I3", bg):rep(utf8.len(line)), wx*3-2)
          fbuffer[wy] = insertString(fbuffer[wy],
            string.pack(">I3", fg):rep(utf8.len(line)), wx*3-2)
        end
        line = nil
        linex = nil
        lwrite = false
      end
    end
    setBackground(background)
    setForeground(foreground)
    --write("\x1b[4" .. background .. "m")
    --write("\x1b[3" .. foreground .. "m")
    flush()
    return true
  end
  function gpu.fill(x, y, w, h, ch)
    checkArg(1, x, "number")
    checkArg(2, y, "number")
    checkArg(3, w, "number")
    checkArg(4, h, "number")
    checkArg(5, ch, "string")
    ch = usub(ch, 1, 1):rep(math.floor(w))
    for i=1, h do
      if i + y - 1 <= _height and i + y > 1 then
        gpu.set(x, y + i - 1, ch)
      end
    end
    return true
  end

  local screenAddr

  function gpu.getScreen()
    return screenAddr
  end

  if not termutils.init() then
    return nil, "Cannot initialize terminal based gpu"
  end
  write("\x1b[?25l") --Disable cursor
  local w, h = gpu.getResolution()
  _height = h
  prepareBuffers(w, h)
  gpu.setForeground(0xFFFFFF)
  gpu.setBackground(0x000000)

  local gpuaddr = modules.component.api.register(nil, "gpu", gpu)
  screenAddr = modules.component.api.register(nil, "screen", {getKeyboards = function() return {"TODO:SetThisUuid"} end}) --verry dummy screen, TODO: make it better, kbd uuid also in epoll.c
  modules.component.api.register("TODO:SetThisUuid", "keyboard", {})

  deadhooks[#deadhooks + 1] = function()
    write("\x1b[?25h\x1b[" .. ((h-1)|0) .. ";1H") --Enable cursor on quit
    io.flush()
    termutils.restore()
  end

  return gpuaddr
end

return textgpu