копалка от серафима

--[[ opencomputers карьер by serafim update 05.11.20

Без замены инструмента для неломаемой кирки из тинкера !!!

Копает змейкой квадрат проходя три слоя блоков за один подход,
пока не выкопает весь обьём до бэдрока или не упрётся в неразрушаемый блок,
при этом вернётся к старту и в случае проблем сообщит о ней,
также предложит продолжить копать с последней  позиции.

требования:
инвентарь, контроллер инвентаря, генератор, улучшение парение.

пример сборки:
https://i.imgur.com/T1X13Gk.png

использование:
на старте, сзади робота поставить сундук,уголь больше 20 штук.
Робот может работать без сундука и всего прочего, нужна только кирка.

пример запуска: (по умолчанию один чанк)
quarry 16 16 2  или  quarry
-- 2 опустится в низ на 2 блока от старта.

получить список предметов:
https://pastebin.com/au9etcfF
]]--

local a = {...}
local width = tonumber(a[1]) or 16 --ширина
local length = tonumber(a[2]) or 16 --длина
local down = tonumber(a[3]) or 2 --опустится в низ

--список мусора
local scrap = {
  "minecraft:cobblestone",
  "minecraft:dirt",
  "minecraft:gravel",
  "minecraft:flint",
  "minecraft:sand",
  "minecraft:sandstone"
}

--список инструмента
local toollist = {
  "ickaxe",
  "rill",
  "pick",
  "vajra"
}

local com = require('component')
local computer = require("computer")
local cheststart,fuelchest,fuelinv,refuel = true,true,true,true
local depth,xPos,zPos,xDir,zDir = 0,0,0,0,1
local ridht,bedrock,unloaded = true,false,0
local x,y,z,xd,zd

if not com.isAvailable("robot") then
  print("только роботы могут использовать эту программу")
  os.exit()
end
local r = require("robot")

local invsize = r.inventorySize()
local tools = invsize
local coal = invsize-1

if not com.isAvailable("inventory_controller") then
  print("для работы нужен контроллер инвентаря")
  os.exit()
end
local i_c = com.inventory_controller

if not com.isAvailable("generator") then
  print("для работы нужен генератор")
  os.exit()
end
local gen = com.generator

--чанк лоадер
local function chunkload(state)
  if com.isAvailable("chunkloader") then
    cloader = com.chunkloader
    if state then
      cloader.setActive(true)
      print("чанк лоадер активен")
    else
      cloader.setActive(false)
      print("чанк лоадер деактивирован")
    end
    print("")
  end
end

--поиск предмета в инвентаре робота
local function slotitem(slot, list)
  for j, name in pairs(list) do
    local item = i_c.getStackInInternalSlot(slot)
    if item and string.find(item.name,name) then
      if item.charge == nil or item.charge > 10000 then
        return true
      end
    end
  end
  for i = 1, invsize do
    if r.count(i) > 0 then
      for j, name in pairs(list) do
        local item = i_c.getStackInInternalSlot(i)
        if item and string.find(item.name,name) then
          if item.charge == nil or item.charge > 10000 then
            r.select(i)
            r.transferTo(slot)
            return true
          end
        end
      end
    end
  end
  return false
end

--взять из сундука расходники
local function servise(slotinv, list)
  local inv = i_c.getInventorySize(3)
  if inv then
    if slotitem(slotinv, list) then
      return true
    end
    r.select(slotinv)
    if r.count(slotinv) > 0 and not r.drop() then
      returnHome("в сундуке нет места")
      return false
    end
    for slot = 1, inv do
      for j, name in pairs(list) do
        local item = i_c.getStackInSlot(3,slot)
        if item and string.find(item.name,name) then
          if item.charge == nil or item.charge > 10000 then
            i_c.suckFromSlot(3, slot)
            return true
          end
        end
      end
    end
    return false
  else
    returnHome("нет сундука")
  end
end

--сортируем лут
local function sortlut(droplut)
  print("сортирую лут...")
  r.swingDown()
  for i = 1, invsize do
    local slot = i_c.getStackInInternalSlot(i)
    if slot ~= nil then
      for k, name in pairs(scrap) do
        if slot.name == name then
          r.select(i)
          r.dropDown()
        end
      end
    end
  end
  for i = 1, invsize-2 do
    if r.count(i) == 0 then
      for j = invsize-2, 1, -1 do
        if r.count(j) > 0 then
          if j < i then
            break
          end
          r.select(j)
          r.transferTo(i)
          break
        end
      end
    end
  end
  local unload = function()
    local inv = i_c.getInventorySize(3)
    if inv then
      cheststart = true
      for i = 1,invsize-2 do
        if r.count(i) > 0 then
          r.select(i)
          unloaded = unloaded + r.count(i)
          if not r.drop() then
            returnHome("в сундуке нет места")
            return
          end
        end
      end
      servise(tools, toollist)
      r.select(coal)
      gen.insert()
      servise(coal, {"coal"})
    elseif not cheststart then
      returnHome("инвентарь заполнен")
    else
      returnHome("нет сундука")
    end
    fuelchest = true
    print("всего добыто руды: "..unloaded)
  end
  if r.count(invsize-6) > 0 then
    print("возвращаюсь к сундуку сложить лут")
    x,y,z,xd,zd = xPos,depth,zPos,xDir,zDir
    goTo( 0,0,0,0,-1 )
    unload()
    goTo( x,y,z,xd,zd )
  end
  if droplut and cheststart then
    unload()
  end
  r.select(1)
end

--стоп
local function stop(message)
  print("\n".."!!! стоп !!!".."\n"..message.."\n")
  r.setLightColor(0xFF0000)
  r.turnAround()
  chunkload(false)
  computer.beep(1000, 1)
  print("программа завершена")
  os.exit()
end

--возврат к старту
function returnHome(message, save)
  r.setLightColor(0xFF0000)
  if save then
    x,y,z,xd,zd = xPos,depth,zPos,xDir,zDir
  end
  if y ~= nil then
    print("ширина "..width.."  длина "..length.."  глубина "..y)
  end
  goTo( 0,0,0,0,-1 )
  if save and cheststart and r.count(1) > 0 then
    sortlut(true)
  end
  goTo( 0,0,0,0,1 )
  print("\n".."!!! стоп !!!".."\n"..message.."\n")
  chunkload(false)
  computer.beep(1000, 1)
  io.write("продолжить копать ? [y/n] > ")
  oq = io.read()
  if oq == "n" or oq == "N" then
    print("программа завершена")
    os.exit()
  else
    r.setLightColor(0xFFFFFF)
    print("продолжаю копать...")
    chunkload(true)
    goTo( x,y,z,xd,zd )
  end
end

--дозаправка углём
local function checkfuel()
  if computer.energy() < 6000 then
    refuel = true
  end
  if gen.count() <= 10 and refuel then
    for i = 1,invsize do
      local item = i_c.getStackInInternalSlot(i)
      if item and string.find(item.name,"coal") then
        r.select(i)
        r.transferTo(coal)
      end
    end
    r.select(coal)
    if gen.insert() then
      print("заправился, угля в генераторе = "..gen.count())
      fuelinv = true
    elseif fuelchest and cheststart then
      x,y,z,xd,zd = xPos,depth,zPos,xDir,zDir
      print("возвращаюсь к сундуку за углем")
      goTo( 0,0,0,0,-1 )
      if servise(coal, {"coal"}) then
        r.select(coal)
        gen.insert()
        print("заправился, угля в генераторе = "..gen.count())
      else
        fuelchest = false
        print("в сундуке нет угля")
      end
      goTo( x,y,z,xd,zd )
    else
      refuel = false
      if fuelinv then
        print("нет угля в инвентаре робота !!!")
        fuelinv = false
      end
    end
    r.select(1)
  end
  if computer.energy() < 5000 then
    if gen.count() >= 10 then
      print("мало энергии, заряжаюсь до 10000")
      r.swingUp()
      while true do
        os.sleep(10)
        if computer.energy() >= 10000 then
          print("зарядился :)")
          break
        end
        if gen.count() == 0 then
          returnHome("закончился уголь",true)
          break
        end
      end
    else
      returnHome("закончился уголь",true)
    end
  end
end

--идём по Х в 0 координат
local function x_forward()
  for xx = 1,width-1 do
    while true do
      if r.forward() then
        xPos = xPos - 1
        break
      else
        r.swing()
      end
    end
  end
end

--движемся в верх
local function tryUp()
  local tru = 0
  while not r.up() do
    r.swingUp()
    tru = tru + 1
    if tru >= 5 then
      tru = 0
      returnHome("непреодолимое препятствие",true)
    end
  end
  depth = depth - 1
end

--копаем и движемся вперед
local function digForward()
  if r.detectUp() then
    r.swingUp()
  end
  if r.detect() then
    r.swing()
  end
  if r.detectDown() and not r.swingDown() then
    if bedrock then
      r.swingDown()
    else
      print("низ достигнут !")
      bedrock = true
    end
  end
  if r.count(invsize-2) > 0 then
    sortlut()
  end
  local tru = 0
  while true do
    if r.forward() then
      xPos = xPos + xDir
      zPos = zPos + zDir
      break
    end
    r.swing()
    tru = tru + 1
    if tru == 5 and bedrock then
      tryUp()
    end
    if tru >= 10 then
      tru = 0
      returnHome("непреодолимое препятствие",true)
    end
  end
end

--спускаемся отлавливая бедрок
local function go_down()
  for zz = 1,down do
    while true do
      if r.detectDown() and not r.swingDown() then
        print("низ достигнут !")
        bedrock = true
      end
      if r.down() then
        depth = depth + 1
        break
      elseif bedrock then
        while true do
          r.swing()
          if r.detect() then
            tryUp()
          else
            return
          end
        end
      end
    end
  end
end

--поворот в лево
local function turnLeft()
  r.turnLeft()
  xDir, zDir = -zDir, xDir
end

--поворот в право
local function turnRight()
  r.turnRight()
  xDir, zDir = zDir, -xDir
end

--навигация к сундуку
function goTo( x, y, z, xd, zd )
  local Xposition = function()
    if xPos > x then
      while xDir ~= -1 do
        turnLeft()
      end
      while xPos > x do
        if r.forward() then
          xPos = xPos - 1
        else
          r.swing()
        end
      end
    elseif xPos < x then
      while xDir ~= 1 do
        turnRight()
      end
      while xPos < x do
        if r.forward() then
          xPos = xPos + 1
        else
          r.swing()
        end
      end
    end
  end
  local Zposition = function()
    if zPos > z then
      while zDir ~= -1 do
        turnLeft()
      end
      while zPos > z do
        if r.forward() then
          zPos = zPos - 1
        else
          r.swing()
        end
      end
    elseif zPos < z then
      while zDir ~= 1 do
        turnRight()
      end
      while zPos < z do
        if r.forward() then
          zPos = zPos + 1
        else
          r.swing()
        end
      end
    end
  end
  r.select(1)
  while depth < y do
    if r.down() then
      depth = depth + 1
    else
      r.swingDown()
    end
  end
  if y > 0 then
    Zposition()
    Xposition()
  else
    Xposition()
    Zposition()
  end
  while depth > y do
    if r.up() then
      depth = depth - 1
    else
      r.swingUp()
    end
  end
  while zDir ~= zd or xDir ~= xd do
    turnRight()
  end
end

--копаем
local function dig()
  os.execute("cls")
  print("размер карьера:  ширина "..width.."  длина "..length)
  print("угля в генераторе = "..gen.count())
  print("энергии = "..math.floor(100*computer.energy()/computer.maxEnergy()).." %")
  print("подготовка...")
  goTo( 0,0,0,0,-1 )
  for slot = 1, invsize do
    if r.count(slot) == 0 then
      r.select(slot)
      i_c.equip()
      break
    end
    if slot == invsize then
      stop("в инвентаре робота нет пустых слотов")
    end
  end
  local inv = i_c.getInventorySize(3)
  if inv then
    cheststart = true
    print("доступен сундук")
    if servise(coal, {"coal"}) then
      r.select(coal)
      if gen.insert() then
        print("заправился, угля в генераторе = "..gen.count())
        servise(coal, {"coal"})
      end
    else
      fuelchest = false
    end
    if servise(tools, toollist) then
      r.select(tools)
      i_c.equip()
      servise(tools, toollist)
    else
      stop("нет инструмента")
    end
  else
    cheststart = false
    print("сундук сзади робота не найден")
    checkfuel()
    if slotitem(tools, toollist) then
      r.select(tools)
      i_c.equip()
      slotitem(tools, toollist)
    else
      stop("нет инструмента")
    end
  end
  goTo( 0,0,0,0,1 )
  r.setLightColor(0xFFFFFF)
  chunkload(true)
  print("опускаюсь на "..down.." блока(ов)")
  for height = 1,256 do
    refuel = true
    checkfuel()
    go_down()
    down = 3
    for xx = 1,width do
      for yy = 1,length-1 do
        digForward()
      end
      checkfuel()
      if xx ~= width then
        if ridht then
          turnRight()
          digForward()
          turnRight()
          ridht = false
        else
          turnLeft()
          digForward()
          turnLeft()
          ridht = true
        end
      end
    end
    r.swingUp()
    r.swingDown()
    if ridht then
      turnLeft()
      x_forward()
      turnLeft()
      ridht = false
    else
      turnRight()
      x_forward()
      turnRight()
      ridht = true
    end
    if bedrock then
      goTo( 0,0,0,0,-1 )
      sortlut(true)
      stop("завершил копать :)")
    end
  end
end

dig()