Bees

-- BeeAnalyzer 3.0
-- Original code by Direwolf20
-- Hotfix 1 by Mandydax
-- Hotfix 2 by Mikeyhun/MaaadMike

local currslot = 1
local princess = 0
local bestDrone = 0
local data=""
local countBees = 0
local beeTable = {}
local currScore = 0

s = peripheral.wrap("right")

function regTable()
   beeTable["bees.species.industrious"] = 31
   beeTable["bees.species.imperial"] = 30
   beeTable["bees.species.unweary"] = 21
   beeTable["bees.species.majestic"] = 20
   beeTable["bees.species.diligent"] = 11
   beeTable["bees.species.noble"] = 10
   beeTable["bees.species.cultivated"] = 6
   beeTable["bees.species.common"] = 5
   beeTable["bees.species.forest"] = 1
   beeTable["bees.species.meadows"] = 1
end

function getBees()
   turtle.select(1)
   for i = 1,6 do
     if turtle.suck() then countBees = countBees + 1 end
   end
end

function returnBees()
   turtle.turnRight()
   turtle.turnRight()
   turtle.select(princess)
   s.dropSneaky(1,1)
   turtle.select(bestDrone)
   s.dropSneaky(0,1)
end

function ditchCombs()
   turtle.turnLeft()
   m = peripheral.wrap("front")
   for i = 1,8 do
     turtle.suck()
     while (not m.isBee()) and (turtle.getItemCount(i) > 0) do
     turtle.select(i)
     turtle.drop()
     if not m.isBee() then
        print("Gotta ditch the combs!")
        turtle.suck()
        turtle.dropDown()
        turtle.select(countBees)
        turtle.transferTo(i, 1)
        countBees = countBees - 1
     end
   end
   end
end

function scanBees()
   turtle.turnLeft()
   for i = 1, countBees do
     turtle.select(i)
     turtle.drop()
   end
   print("Sleeping for a minute while the bee scans")
   sleep(26)
end

function determineBest(slot)
   local primScore = (beeTable[data["speciesPrimary"]] ~= nil and beeTable[data["speciesPrimary"]] or 0)
   local secScore = (beeTable[data["speciesSecondary"]] ~= nil and beeTable[data["speciesSecondary"]] or 0)
   score = primScore + secScore
   print("Current: "..currScore)
   if primScore == 0 and secScore == 0 then print("Warning bee species not in registry table! Evalutaing as 0!") end
   print("NewScore: "..score)
   if(bestDrone == 0) then
     bestDrone = slot
     currScore = score
   else
     if (score > currScore) then
       bestDrone = slot
       currScore = score
     end
   end
end

function analyzeBees()
  for i = 1, countBees do
    turtle.select(i)
    s.suckSneaky(0,1)
    turtle.turnRight()
    turtle.drop()
    m = peripheral.wrap("front")
    data = m.analyze()
    if (data["type"] == "princess") then
      princess = i
      print("Princess = "..i)
    else
      determineBest(i)
    end
    print(data["speciesPrimary"]..":"..data["speciesSecondary"])
    print("BestDrone = "..bestDrone)
    turtle.suck()
    turtle.turnLeft()
    if i ~= countBees then sleep(26) end
   end
end


function dropExcess()
  for i = 1, 16 do
    turtle.select(i)
    turtle.dropDown()
   end
end

-- Drop same drones so they won't interfere
function dropDupe()
  for i = 1, countBees do
    turtle.select(i)
    local count = turtle.getItemCount(i)
    if count > 1 then
      print("Dropping excess drones.")
      turtle.dropDown(count - 1)
    end
  end
  turtle.select(1)
end
------=============================

currslot = 1
princess = 0
bestDrone = 0
data=""
countBees = 0
beeTable = {}
currScore = 0

regTable()
dropExcess()
getBees()
if (turtle.getItemCount(2) > 0) then
   ditchCombs()
   dropDupe()
   scanBees()
   analyzeBees()
   returnBees()
   dropExcess()
end