Quarry version 2.3.3

--[[TO DO:
Awesome code that will automatically dig 3 for every layer until last, when specs matter
Make auto-empty smarter: e.g. it will empty every few layers instead of every layer.

Version 2.3.3
Recent changes: added an argument for which side the chest is on.
    REDNET SUPPORT: Add the argument "true" after chest to send to
    rednet!!!!!!!!!!!
Directions: If you want, place a chest right behind where you place the turtle.
It will place things here when its done

Also, if auto-empty is on, it will always eject the things it has
into the place where a chest is supposed to be,
it just dosen't check for a chest
]]


--Check Fuel Disabling for those on softcore
doCheckFuel = true
--For those that find this annoying
forceAutoEmpty = true
--If set to false, will ignore all safety restraints
careAboutResources = true

doDebug = false

--This is how many (theoretical) slots it will fill with cobble. e.g. 13 allows 13 slots for cobble (or other stack) and 2 slots for other things.
--Maximum here is 16
maxslots = 13
--If your job is bigger than it can carry (as defined by max slots) it will default to auto-empty
--Note: For counting to work properly, cobblestone must be the first block it mines, or put a cobble in slot 1

--Defining things
tArgs = {...}
if tArgs[1] == nil then
dropSide = "front"
else
dropSide = tArgs[1]
end
if tArgs[2] == "true" then
rednetEnabled = true
else
rednetEnabled = false
end

if careAboutResources == false then
forceAutoEmpty = false
end

function biometrics()
if pass == true then
rednet.open("right")
local toSend = {length = x, height = y, width = z,
    percent = percent, blocksMined = mined,
    fuel = checkFuel(), blocksMoved = moved,
    blocksLeft = (volume-moved), ID = os.getComputerID(),
    finishedLayers = doneHeight, finishedWidth = doneWidth,
    volume = volume}
rednet.send(host, textutils.serialize(toSend))
hostCheck, confirm, distX = 0,0,0
hostCheck, confirm, distX = rednet.receive(1)
if hostCheck == host and confirm == "Quarry Confirm" then
displayRedCheck = true --This section displays to the screen that
else                   --the host computer is still receiving.
displayRedCheck = false
end
rednet.close("right")
end
end

function screen(xPos,yPos)
term.clear()
term.setCursorPos(xPos,yPos)
end
function mine() -- Basic Move Forward
while turtle.forward() == false do
if turtle.dig() == false then
turtle.attack()
mined = mined - 1
end
mined = mined + 1
end
if Cdouble == 1 then
if turtle.digDown() == true then
mined = mined + 1
end
end
if Cdouble == 0 then
moved = moved + 1
else
moved = moved + 2
end
display()
end
function mineRow() -- Move a row then turn
if cRows ~= 0 then -- cRows is "Rows Completed"
for length=1, x do --It does this check because on the first row of a layer,
mine()             --   it already occupies the first position
end
else
for length=1, x-1 do
mine()
end
end
cRows = cRows + 1
biometrics()
end
function display() --Runs in Mine(), display information to the screen in a certain place
percent = math.ceil((moved/volume)*100)
term.setCursorPos(1,2)
term.clearLine()
print(mined)
term.setCursorPos(1,4)
term.clearLine()
print(percent.."%")
if rednetEnabled then
if displayRedCheck then
display1 = "Good"
else
display1 = "Lost"
end
term.setCursorPos(1,6)
print("Connection: "..display1)
end
end

totals = {cobble = 0, sand = 0, other = 0} -- Total for display (cannot go inside function)
function count() --Done after every row of auto-empty and at end
slot = {}        --Detection of blocks code is 1: Cobble 2: Other Raw 3:Other
for i=1, 16 do   --[1] is type, [2] is number
slot[i] = {}
slot[i][2] = turtle.getItemCount(i)
end
doCheckSand = false
if area > 100 then
doCheckSand = true
end

slot[1][1] = 1   -- = Assumes Cobble/Main
for i=2, 16 do   --Cobble Check
turtle.select(i)
if turtle.compareTo(1) == true then
slot[i][1] = 1
end
end
for i=2, 16 do   --Sand/Raw Check
tempItemCount = turtle.getItemCount(i)
if slot[i][1] ~= 1 then
if tempItemCount > 32 and doCheckSand then
slot[i][1] = 2
elseif tempItemCount > 0 and slot[i][1] then
slot[i][1] = 3
end
end
end
for i=1, 16 do
if slot[i][1] == 1 then
totals.cobble = slot[i][2] + totals.cobble
elseif slot[i][1] == 2 then
totals.sand = slot[i][2] + totals.sand
else
totals.other = slot[i][2] + totals.other
end
end

if doDebug == true then
for i=1, 16 do
print(slot[i][2])
end
end
end

function endDrop(final)
if final ~= false then
final = true
end
if dropSide == "left" then --Turning
turtle.turnRight()
elseif dropSide == "right" then
turtle.turnLeft()
end
turtle.select(1)
if dropSide == "top" then -- Drops to top
if final == false then
avar = turtle.getItemCount(1) --Dosen't drop 1 cobblestone for future counting purposes if autofill
turtle.dropUp(avar-1)
for i=2, 16 do
turtle.select(i)
turtle.dropUp()
end
elseif turtle.detectUp() then
for i=1, 16 do
turtle.select(i)
turtle.dropUp()
end
end
elseif dropSide == "bottom" then --Drops to bottom
if final == false then
avar = turtle.getItemCount(1)
turtle.dropDown(avar-1)
for i=2, 16 do
turtle.select(i)
turtle.dropDown()
end
elseif turtle.detectDown() then
for i=1, 16 do
turtle.select(i)
turtle.dropDown()
end
end
else --Drops to front
if final == false then
avar = turtle.getItemCount(1)
turtle.drop(avar-1)
for i=2, 16 do
turtle.select(i)
turtle.drop()
end
elseif turtle.detect() then
for i=1, 16 do
turtle.select(i)
turtle.drop()
end
end

end
turtle.select(1)

if dropSide == "right" then --Turning, inverted
turtle.turnRight()
elseif dropSide == "left" then
turtle.turnLeft()
end

end
function checkFuel()
fuelLevel = turtle.getFuelLevel()
return fuelLevel
end


 -----------------------------------------------------------------
--Input Phase

screen(1,1)
print("----- Welcome to Quarry! -----")
print("")

--Dimensions of Hole
print("What dimensions?")
print("")
term.write("Length: ")
x = io.read()

term.write("Width: ")
z = io.read()

term.write("Height: ")
y = io.read()

if not tonumber(x) or x == 0 then
x = 3
end
if not tonumber(y) or y == 0 then
y = 2
end
if not tonumber(z) or z == 0 then
z = 3
end

x = math.abs(tonumber(x))
y = math.abs(tonumber(y))
z = math.abs(tonumber(z))
if doDebug == true then
print(x," ",z," ",y)
end

Cdouble = 0
--Check for if it can do doubles
if math.ceil(y/2) == y/2 then
Cdouble = 1
end



volume = x*y*z
area = x*z

if maxslots > 16 then
maxslots = 16
end
maxsize = maxslots*64
if Cdouble == 1 then
maxsize = maxsize/2
end

empty = 0


--Checks if bigger than inventory: 832 blocks theoretical Assuming at least 2 are non-cobble

if forceAutoEmpty == true then
if volume > maxsize then
print("Job very big")
print("Auto-emptying on")
empty = 1
sleep(2)
end
end

if empty ~= 1 and careAboutResources == true then
print("Turn on auto-emptying? (Y/N)")
ans = string.sub(string.lower(io.read()),1,1)
if ans == "y" or ans == "1" then
empty = 1
elseif ans == "s" then
print("Stop? (type \"yes\")")
if io.read() == "yes" then
os.shutdown()
end
end
end

if area > maxsize and careAboutResources then
print("Area too Large, please restart program")
sleep(5)
os.reboot()
end

--Getting Fuel
if doCheckFuel == true then --Calculating Needed Fuel
subHeight = 0
if check == 1 then
for i=0, y, 2 do --Usage is y = -2i+y, but I don't know how to do that
subHeight = (2*y-i) + subHeight
end
else
subHeight = y*2
end
if Cdouble == 1 then
neededFuel = math.ceil((volume + x*y*z)/2 + subHeight)
else
neededFuel = math.ceil(volume + x*y*z + subHeight)
end
if checkFuel() < neededFuel then
term.clear()
term.setCursorPos(1,1)

print("More Fuel Needed")
print("Current Fuel: ",checkFuel()," Needed: ",neededFuel)
print("Place in Bottom Right, press any key")
os.pullEvent("char")
print("I'll take a bit more, so we don't have to do this as often")
turtle.select(16)
while checkFuel() < neededFuel+100 do
if turtle.refuel(1) == false then
term.clearLine()
print("Still too little fuel")
term.clearLine()
print("Press a key to resume fueling")
os.pullEvent("char")
end
xx,xy = term.getCursorPos()
print(fuelLevel.." Fuel")
term.setCursorPos(xx,xy)
end
print(fuelLevel.." Units of Fuel")
sleep(3)
turtle.select(1)
end

end

--Initial Rednet Handshake
if rednetEnabled then
screen(1,1)
print("----- Welcome to Quarry! -----")
print("")
rednet.open("right")
pass = false
print("Host Computer ID: ")
host = tonumber(io.read())
initRedCheck = 0
repeat
initRedCheck = initRedCheck + 1
print("Sending: "..initRedCheck)
rednet.send(host, "Initial Confirm")
hostCheck, confirmCheck, distX = rednet.receive(3)
if hostCheck == host and confirmCheck == "Host Confirm" then
pass = true
print("Established Connection")
print("with Computer "..hostCheck.."!")
sleep(1)
displayRedCheck = true
end
if initRedCheck == 10 then
error("Connection Failed to establish, try again...",0)
end
until pass == true
rednet.close("right")
end

--Check Double changing the Depth y (times it goes down)
if Cdouble == 1 then
Dy = y/2
else
Dy = y
end

--Height Calculation
if Cdouble == 0 then
b = 1
else
b = 2
end


--Mining Phase

turn = 1 --Turn right is 0, turn left is 1 opposite to start
mined = 0 -- Total Blocks Mined
moved = 0 -- Total Spaces Moved

turtle.select(1)
screen(1,1)
print("Blocks Mined")
print("")
print("Percent Complete")
mine()

doneHeight = 0
doneWidth = 0 --Initialize for first biometrics
biometrics()
for depth=1, Dy do
doneHeight = doneHeight+b
doneWidth = 0
for width = 1, z do
doneWidth = doneWidth + 1
cRows = 0 -- Rows completed (z)
mineRow()
if turn == 1 then
turn = 0
else
turn = 1
end
if width ~= z then --Check if last row of set, will not turn if last
if turn == 0 then
turtle.turnRight()
mine()
turtle.turnRight()
else
turtle.turnLeft()
mine()
turtle.turnLeft()
end
end
end
if turn == 0 then --This If statement gets it back to the start of this layer
turtle.turnRight()
if Cdouble == 1 then
if turtle.digDown() == true then
mined = mined + 1
end
end
turtle.turnRight()
for i=1, x-1 do
while not turtle.forward() do
end
end
turtle.turnRight()
for i=1, z-1 do
while not turtle.forward() do
end
end
else
turtle.turnRight()
for i=1, z-1 do
while not turtle.forward() do
end
end
end
if empty == 1 and depth ~= Dy then --AutoEmptying at the end of every layer
if Cdouble == 1 then
AEup = depth*2
else
AEup = depth
end
for i=1, AEup-b do
while not turtle.up() do
end
end
turtle.turnLeft()
while not turtle.forward() do
end
count()   --Counting
endDrop(false) -- Drop into specified chest

turtle.turnLeft()
turtle.turnLeft()
while not turtle.forward() do
end
for i=1, AEup-b do
turtle.down()
end
turtle.turnLeft()
end
if depth ~= Dy then --This checks if it is the last layer of the run
for a=1, b do
while not turtle.down() do
turtle.digDown()
mined = mined+1
end
moved = moved + 1
end
turtle.turnRight()
if Cdouble == 1 then
if turtle.digDown() == true then
mined = mined + 1
end
end
turn = 1
end
end

for i=1, y-b do --This gets the turtle back to the top layer
while not turtle.up() do
end
end
turtle.turnLeft()
while not turtle.forward() do
end

--Counts all blocks at the end also
count()

--Output to a chest or sit there
endDrop()
turtle.turnRight()
turtle.turnRight()
turtle.select(1)
--Display
screen(1,1)
print("Total Blocks Mined: "..mined)
print("Current Fuel Level: "..turtle.getFuelLevel())
print("Cobble: "..totals.cobble)
print("Other Raw: "..totals.sand)
print("Other: "..totals.other)
if rednetEnabled then
print("")
print("Sent Stop Message")
rednet.open("right")
rednet.send(host, "Stop")
finalTable = {cobble = totals.cobble, sand = totals.sand,
    other = totals.other, fuel = checkFuel(), mined = mined}
rednet.send(host, textutils.serialize(finalTable))

rednet.close("right")
end