local c = require("component")
local event = require("event")
local geo = c.geolyzer
local holo = c.hologram
local gpu = c.gpu
local ecs = require("ECSAPI")

local args = {...}


if not c.isAvailable("geolyzer") or not c.isAvailable("hologram") then
  error("Подключите геоанализатор и голографический проектор 2-ого уровня")
end
-------------------------

local xScanFrom, xScanTo = -24, 23
local zScanFrom, zScanTo = xScanFrom, xScanTo

local massiv = {}

local yModifyer = -20
local scales = {0.33, 0.75, 1, 1.5, 2, 2.5, 3}
local currentScale = 1
local countOfScales = #scales

local command = args[1] or "scan"
local argument1 = args[2] or -24
local argument2 = args[3] or 23

local xSize, ySize = gpu.getResolution()
local yCenter = math.floor(ySize / 2)

---------------------------------------

local function clear()
  holo.clear()
end

local function changeScale()
  if currentScale < countOfScales then
    currentScale = currentScale + 1
  else
    currentScale = 1
  end

  holo.setScale(scales[currentScale])
end

local function displayRow(x, yModifyer, z,  tablica)
  local color = 1
  for i = 1, #tablica do
    if tablica[i] > 0.5 then
      
      color = 1

      if tablica[i] < 2 then
        color = 2
      elseif tablica[i] >= 2 and tablica[i] < 3 then
        color = 1
      elseif tablica[i] >= 3 and tablica[i] then
        color = 3
      end

      if tablica[i + yModifyer] then
        holo.set(xScanTo - x + 1, i + yModifyer, zScanTo - z + 1, color)
      end
    end
  end
end

local function displayAllRows()
  clear()
  for x, val in pairs(massiv) do
    for z, val2 in pairs(massiv[x]) do
      displayRow(x, yModifyer, z, val2)
    end
  end
end

local function scan()
  clear()
  ecs.clearScreen(0xffffff)
  local barWidth = math.floor(xSize / 3 * 2)
  local percent = 0
  local xBar, yBar = math.floor(xSize/2 - barWidth / 2), yCenter
  local countOfAll = (math.abs(xScanFrom) + math.abs(xScanTo) + 1) ^ 2

  local counter = 0
  for x = xScanFrom, xScanTo do
    massiv[x] = {}
    for z = zScanFrom, zScanTo do

      massiv[x][z] = geo.scan(x, z, true)
      for i = 1, #massiv[x][z] do
        displayRow(x, yModifyer, z, massiv[x][z])
      end
      percent = counter / countOfAll * 100
      ecs.progressBar(xBar, yBar, barWidth, 1, 0xcccccc, ecs.colors.blue, percent)
      gpu.setForeground(0x444444)
      gpu.setBackground(0xffffff)
      ecs.centerText("x", yBar + 1, "   Сканирование стека на x = "..x..", z = "..z.."   ")
      counter = counter + 1

    end
  end
end

local obj = {}
local function newObj(class, name, ...)
  obj[class] = obj[class] or {}
  obj[class][name] = {...}
end

local function main()
  ecs.clearScreen(0xffffff)
  local yPos = yCenter - 10
  newObj("buttons", "Сканировать местность", ecs.drawAdaptiveButton("auto", yPos, 3, 1, "Сканировать местность", 0x444444, 0xffffff)); yPos = yPos + 4
  newObj("buttons", "Перерисовать голограмму", ecs.drawAdaptiveButton("auto", yPos, 3, 1, "Перерисовать голограмму", 0x444444, 0xffffff)); yPos = yPos + 4
  newObj("buttons", "Масштаб", ecs.drawAdaptiveButton("auto", yPos, 3, 1, "Масштаб", 0x444444, 0xffffff)); yPos = yPos + 4
  newObj("buttons", "+ 10 блоков", ecs.drawAdaptiveButton("auto", yPos, 3, 1, "+ 10 блоков", 0x444444, 0xffffff)); yPos = yPos + 4
  newObj("buttons", "- 10 блоков", ecs.drawAdaptiveButton("auto", yPos, 3, 1, "- 10 блоков", 0x444444, 0xffffff)); yPos = yPos + 4
  newObj("buttons", "Выйти", ecs.drawAdaptiveButton("auto", yPos, 3, 1, "Выйти", 0x666666, 0xffffff)); yPos = yPos + 4
  gpu.setBackground(0xffffff)
  gpu.setForeground(0x444444)
  ecs.centerText("x", yPos, "Модификатор высоты: "..yModifyer)
end

----------------------------

changeScale()
main()

while true do
  local e = {event.pull()}
  if e[1] == "touch" then
    for key, val in pairs(obj["buttons"]) do
      if ecs.clickedAtArea(e[3], e[4], obj["buttons"][key][1], obj["buttons"][key][2], obj["buttons"][key][3], obj["buttons"][key][4]) then
        ecs.drawAdaptiveButton(obj["buttons"][key][1], obj["buttons"][key][2], 3, 1, key, ecs.colors.green, 0xffffff)
        os.sleep(0.3)
        if key == "Сканировать местность" then
          scan()
        elseif key == "Масштаб" then
          changeScale()
        elseif key == "Перерисовать голограмму" then
          displayAllRows()
        elseif key == "+ 10 блоков" then
          yModifyer = yModifyer + 10
        elseif key == "- 10 блоков" then
          yModifyer = yModifyer - 10
        elseif key == "Выйти" then
          ecs.prepareToExit()
          return 0
        end
        main()
        break
      end
    end
  end
end