local class = require('class')
local DEFLATE = require('deflatelua')
local UI = require('ui')
local Logger = require('logger')
local Profile = require('profile')
local Point = require('point')

--[[
  Loading and manipulating a schematic
--]]

local schematicMagic = 0x0a00
local gzipMagic = 0x1f8b

local Schematic = class()
function Schematic:init(args)
  self.blocks = { }
  self.damages = { }
  self.originalBlocks = { }
  self.placementChains = { }
  self.x, self.y, self.z = 0, 0, 0
  self.height = 0
  self.index = 1
end

--[[
  Credit to Orwell for the schematic file reader code
  http://www.computercraft.info/forums2/index.php?/topic/1949-turtle-schematic-file-builder/
 
  Some parts of the file reader code was modified from the original
--]]
 
function Schematic:discardBytes(h, n, spinner)
  for i = 1,n do
    h:readbyte()
    if (i % 1000) == 0 then
      spinner:spin()
    end
  end
end
 
function Schematic:readname(h)  
  local n1 = h:readbyte(h)
  local n2 = h:readbyte(h)
 
  if(n1 == nil or n2 == nil) then
    return ""
  end
 
  local n = n1*256 + n2
 
  local str = ""
  for i=1,n do
    local c = h:readbyte(h)
    if c == nil then
      return
    end  
    str = str .. string.char(c)
  end
  return str
end
 
function Schematic:parse(a, h, containsName, spinner)
 
  if a==0 then
    return
  end
 
  local name
  if containsName then
    name = self:readname(h)
  end

  if a==1 then
    self:discardBytes(h, 1, spinner)
  elseif a==2 then
    local i1 = h:readbyte(h)
    local i2 = h:readbyte(h)
    local i = i1*256 + i2
    if(name=="Height") then
      --self.height = i
    elseif (name=="Length") then
      self.length = i
    elseif (name=="Width") then
      self.width = i
    end
    return 2
  elseif a==3 then
    self:discardBytes(h, 4, spinner)
    return 4
  elseif a==4 then
    self:discardBytes(h,8, spinner)
    return 8
  elseif a==5 then
    self:discardBytes(h,4, spinner)
    return 4
  elseif a==6 then
    self:discardBytes(h,8, spinner)
  elseif a==7 then
    local i1 = h:readbyte(h)
    local i2 = h:readbyte(h)
    local i3 = h:readbyte(h)
    local i4 = h:readbyte(h)
    local i = bit.blshift(i1, 24) + bit.blshift(i2, 16) + bit.blshift(i3, 8) + i4

    if not self.length or not self.width then

      self:discardBytes(h,i, spinner)
      self.twopass = true

    elseif name == "Blocks" then
      for i = 1, i do
        local id = h:readbyte(h)
        self:assignCoord(i, id)
        if (i % 1000) == 0 then
          spinner:spin()
        end
      end
    elseif name == "Data" then
      for i = 1, i do
        local dmg = h:readbyte(h)
        if dmg > 0 then
          self.damages[i] = dmg
        end
        if (i % 1000) == 0 then
          spinner:spin()
        end
      end
    else
      self:discardBytes(h,i, spinner)
    end
  elseif a==8 then
    local i1 = h:readbyte(h)
    local i2 = h:readbyte(h)
    local i = i1*256 + i2
    self:discardBytes(h,i, spinner)
  elseif a==9 then
    local type = h:readbyte(h)
    local i1 = h:readbyte(h)
    local i2 = h:readbyte(h)
    local i3 = h:readbyte(h)
    local i4 = h:readbyte(h)
    local i = bit.blshift(i1, 24) + bit.blshift(i2, 16) + bit.blshift(i3, 8) + i4

    for j=1,i do
      self:parse(type, h, false, spinner)
    end
  elseif a > 11 then
    error('invalid tag')
  end
end
-- end http://www.computercraft.info/forums2/index.php?/topic/1949-turtle-schematic-file-builder/
 
function Schematic:copyBlocks(iblocks, oblocks, spinner)
  Profile.start('copyBlocks')
  for k,b in ipairs(iblocks) do
    oblocks[k] = Util.shallowCopy(b)
    if spinner then
      if (k % 1000) == 0 then
        spinner:spin()
      end
    end
  end
  Profile.stop('copyBlocks')
end
 
function Schematic:reload()
  self.placementChains = {}
  self.blocks = { }
  self:copyBlocks(self.originalBlocks, self.blocks)
  --[[
  self.planes = { }
  for i = 0, self.height - 1 do
    self.planes[i] = { }
  end
  for k,b in ipairs(self.blocks) do
    if not self.planes[b.y].start then
      self.planes[b.y].start = k
    end
  end
  --]]
end

function Schematic:getMagic(fh)
  fh:open()
 
  local magic = fh:readbyte() * 256 +  fh:readbyte()

  fh:close()

  return magic
end

function Schematic:isCompressed(filename)
 local h = fs.open(filename, "rb")
 
  if not h then
    error('unable to open: ' .. filename)
  end
 
  local magic = h.read() * 256 +  h.read()

  h.close()

  return magic == gzipMagic
end

function Schematic:checkFileType(fh)
 
  local magic = self:getMagic(fh)
  if magic ~= schematicMagic then
    error('Unknown file type')
  end
end

local DiskFile = class()
function DiskFile:init(args)
  Util.merge(self, args)
end

function DiskFile:open()
  self.h = fs.open(self.filename, "rb")
  if not self.h then
    error('unable to open: ' .. self.filename)
  end
end
function DiskFile:readbyte()
  return self.h.read()
end
function DiskFile:close()
  self.h.close()
end

local MemoryFile = class()
function MemoryFile:init(args)
  self.s = { }
  self.i = 1
end
function MemoryFile:open(filename)
  self.i = 1
end
function MemoryFile:close() end

function MemoryFile:readbyte()
  local b = self.s[self.i]
  self.i = self.i + 1
  return b
end

function MemoryFile:write(b)
  self.s[#self.s + 1] = b
end

function Schematic:decompress(ifname, spinner)

  Profile.start('decompress')

  local ifh = fs.open(ifname, "rb")
  if not ifh then
    error('Unable to open ' .. ifname)
  end

  local mh = MemoryFile()

  DEFLATE.gunzip({
    input=function(...) spinner:spin() return ifh.read() end,
    output=function(b) mh:write(b) end,
    disable_crc=true
  })

  ifh.close()

  spinner:stop()
  Profile.stop('decompress')

  return mh
end
 
function Schematic:loadpass(fh, spinner)
 
  Profile.start('load')

  fh:open()
 
  while true do
    local a = fh:readbyte()
 
    if not a then
      break
    end
    self:parse(a, fh, true, spinner)
    if self.twopass and self.width and self.length then
      break
    end

    spinner:spin()
  end

  fh:close()
  Profile.stop('load')
 
  self:assignDamages(spinner)
  self.damages = nil
 
  self:copyBlocks(self.blocks, self.originalBlocks, spinner)
  Profile.display()

  spinner:stop()
end

function Schematic:load(filename)

  local cursorX, cursorY = term.getCursorPos()
  local spinner = UI.Spinner({
    x = UI.term.width,
    y = cursorY - 1
  })
  local f
 
  if self:isCompressed(filename) then
    local originalFile = filename
    filename = originalFile .. '.uncompressed'

    if not fs.exists(filename) then
      print('Decompressing')
      f = self:decompress(originalFile, spinner)
    end
  end

  self.filename = string.match(filename, '([^/]+)$')

  if not f then
    f = DiskFile({ filename = filename })
  end

  self:checkFileType(f)

  --[[
  local size = fs.getSize(filename)

  local buffer = {
    h = h,
    i = 1,
    s = { },
    l = size,
  }

  for i = 1,size do
    buffer.s[i] = h.read()
  end

  function buffer:readbyte()
    --return self.h.read()
    local b = self.s[self.i]
    self.i = self.i + 1
    return b
  end
  ]]--

  print('Initial pass     ')
  self:loadpass(f, spinner)

  if self.twopass then
    self.twopass = nil
    self.blocks = { }
    self.damages = { }
    self.originalBlocks = { }
    self.placementChains = { }
    self.x, self.y, self.z = 0, 0, 0
    self.height = 0
    self.index = 1

    print('Second pass      ')
    self:loadpass(f, spinner)
  end 
end

function Schematic:assignCoord(i, id)

  if id > 0 then
    table.insert(self.blocks, {
      id = id,
      index = i,
      x = self.x,
      z = self.z,
      y = self.y,
    })
  end

  self.x = self.x + 1
  if self.x >= self.width then
    self.x = 0
    self.z = self.z + 1
  end
  if self.z >= self.length then
    self.z = 0
    self.y = self.y + 1
  end
  if self.y >= self.height then
    self.height = self.y + 1
  end
end
 
function Schematic:assignDamages(spinner)
  local i = 0
 
  Profile.start('assignDamages')
  print('Assigning damages')

  for _,b in pairs(self.blocks) do
    b.dmg = self.damages[b.index] or 0
    i = i + 1
    if (i % 1000) == 0 then
      spinner:spin()
    end
  end
  Profile.stop('assignDamages')
end

function Schematic:findIndexAt(x, z, y)
  if y < 0 then
    return
  end

  local ri = self.rowIndex[y]
  if ri then
    for i = ri.s, ri.e do
      local b = self.blocks[i]
      if b.x == x and b.z == z and b.y == y then
        if b.id == 'minecraft:air' then 
          -- this will definitely screw up placement order if a substition is made with air after starting
          -- as blocks will be placed differently and could have a different heading
          break
        end
        return i, b
      end
    end
  end
end

function Schematic:findBlockAtSide(b, side)
  local hi = turtle.getHeadingInfo(side)
  local index = self:findIndexAt(b.x + hi.xd, b.z + hi.zd, b.y + hi.yd)
  if index then
    return self.blocks[index] -- could be better
  end
end
 
function Schematic:addPlacementChain(chain)
  local t = { }
  for _,v in ipairs(chain) do
    local k = self:findIndexAt(v.x, v.z, v.y)
    if k then
      local b = self.blocks[k] -- could be better
      b.index = v.y * self.width * self.length + v.z * self.width + v.x + 1
      table.insert(t, b)
    end
  end
  if #t > 1 then
    local keys = { }
    for _,b in pairs(t) do
      keys[b.index] = true
    end
    table.insert(self.placementChains, {
      blocks = t,
      keys = keys
    })
  end
end
 
function Schematic:bestSide(b, ...)
  local directions = { ... }
  local blocks = { }
 
  for k,d in pairs(directions) do
    local hi = turtle.getHeadingInfo(d)
    local sb = self:findIndexAt(b.x - hi.xd, b.z - hi.zd, b.y)
    if not sb then
      b.heading = turtle.getHeadingInfo(d).heading
      b.direction = d .. '-block'
      return
    end
    blocks[k] = {
      b = self.blocks[sb],
      hi = hi,
      d = d
    }
  end
 
  local bestBlock
  for _,sb in ipairs(blocks) do
    if not sb.b.direction then -- could be better
      bestBlock = sb
      break
    end
  end
 
  if not bestBlock then
    local sideDirections = {
      [ 'east-block' ] = 'east',
      [ 'south-block' ] = 'south',
      [ 'west-block'  ] = 'west',
      [ 'north-block'  ] = 'north'
    }
    for _,sb in ipairs(blocks) do
      if not bestBlock then
        bestBlock = sb
      end
      if not sideDirections[sb.b.direction] then
        bestBlock = sb
        break
      end
    end
  end
 
  local hi = bestBlock.hi
  b.heading = hi.heading     -- ?????????????????????????????????
  b.direction = bestBlock.d .. '-block'
  self:addPlacementChain({
    { x = b.x,         z = b.z,         y = b.y },
    { x = b.x - hi.xd, z = b.z - hi.zd, y = b.y }
  })
end
 
function Schematic:bestOfTwoSides(b, side1, side2) -- could be better

  local sb
  local fb = b  -- first block
  local lb = b  -- last block
  local od = b.direction -- original direction

  -- find the last block in the row with the same two-sided direction
  while true do
    sb = self:findBlockAtSide(lb, side2)
    if not sb or sb.direction ~= b.direction then
      break
    end
    lb = sb
  end

  -- find the first block
  while true do
    sb = self:findBlockAtSide(fb, side1)
    if not sb or sb.direction ~= b.direction then
      break
    end
    fb = sb
  end

  -- set the placement order to side1 -> side2
  if fb ~= lb then -- only 1 block

    local pc = { }  -- placementChain
    b = fb

    while true do

      table.insert(pc, { x = b.x,   z = b.z,   y = b.y })

      b.direction = side1 .. '-block'
      b.heading = turtle.getHeadingInfo(side1).heading

      if b == lb then
        break
      end

      b = self:findBlockAtSide(b, side2)
    end

    self:addPlacementChain(pc)
  end

  -- can we place the first block from the side (instead of using piston) ?
  sb = self:findBlockAtSide(fb, side1)
  if not sb then
    local ub = self:findBlockAtSide(fb, 'down')
    if not ub then
      fb.direction = side1 .. '-block'
      fb.heading = turtle.getHeadingInfo(side1).heading
    else
      fb.direction = od
    end
  else  -- really should use placement chain
    fb.direction = od
  end

  -- can we place the last block from the side (instead of using piston) ?
  sb = self:findBlockAtSide(lb, side2)
  if not sb then
    local ub = self:findBlockAtSide(lb, 'down')
    if not ub then
      lb.direction = side1 .. '-block'
      lb.heading = turtle.getHeadingInfo(side1).heading
    else
      fb.direction = od
    end
  else
    lb.direction = od
  end

end
 
-- Determine the best way to place each block
function Schematic:determineBlockPlacement(row)

  -- NOTE: blocks are evaluated top to bottom

  local spinner = UI.Spinner({
    x = 1,
    spinSymbols = { 'o.....', '.o....', '..o...', '...o..', '....o.', '.....o' }
  })
  local stairDownDirections = {
    [ 'north-down' ] = 'north',
    [ 'south-down' ] = 'south',
    [ 'east-down'  ] = 'east',
    [ 'west-down'  ] = 'west'
  }
  local stairUpDirections = {
    [ 'east-up' ] = { 'east', 'east-block', 1, 0, 'west-block' },
    [ 'west-up' ] = { 'west', 'west-block', -1, 0, 'east-block' },
    [ 'north-up' ] = { 'north', 'north-block', 0, -1, 'south-block' },
    [ 'south-up' ] = { 'south', 'south-block', 0, 1, 'north-block' }
  }
  local twoSideDirections = {
    [ 'east-west-block'   ] = true,
    [ 'north-south-block' ] = true,
  }
  local directions = {
    [ 'north' ] = 'north',
    [ 'south' ] = 'south',
    [ 'east'  ] = 'east',
    [ 'west'  ] = 'west',
  }
  local blockDirections = {
    [ 'east-block' ] = 'east',
    [ 'south-block' ] = 'south',
    [ 'west-block'  ] = 'west',
    [ 'north-block'  ] = 'north',
  }
  local doorDirections = {
    [ 'east-door' ] = 'east',
    [ 'south-door' ] = 'south',
    [ 'west-door'  ] = 'west',
    [ 'north-door'  ] = 'north',
  }
  local vineDirections = {
    [ 'east-block-vine' ] = 'east-block',
    [ 'south-block-vine' ] = 'south-block',
    [ 'west-block-vine'  ] = 'west-block',
    [ 'north-block-vine'  ] = 'north-block'
  }
 
  local dirtyBlocks = {}
  local dirtyBlocks2 = {}

  self.rowIndex = { }
  for k,b in ipairs(self.blocks) do
    local ri = self.rowIndex[b.y]
    if not ri then
      self.rowIndex[b.y] = { s = k, e = k }
    else
      ri.e = k
    end
  end

  for k,b in ipairs(self.blocks) do
    local d = b.direction

    if d then
      if vineDirections[d] then
        local _, aboveBlock = self:findIndexAt(b.x, b.z, b.y+1)

        if aboveBlock and aboveBlock.id == b.id and aboveBlock.dmg == b.dmg and aboveBlock.direction == d then
          -- only need to place top vine
          b.id = 'minecraft:air'
          b.dmg = 0
          b.direction = nil
        else
          b.direction = vineDirections[d]
          table.insert(dirtyBlocks, b)
        end
      elseif twoSideDirections[d] then
        table.insert(dirtyBlocks2, b)
      else
        table.insert(dirtyBlocks, b)
      end
      spinner:spin(#dirtyBlocks + #dirtyBlocks2 .. ' blocks remaining ')
    end
  end

--  Util.filterInplace(dirtyBlocks, function(b) return b.id ~= 'minecraft:air' end)

  -- remove directional info from slabs where possible
  -- iterate backwards to process top planes first
  for k = #dirtyBlocks, 1, -1 do
    local b = dirtyBlocks[k]
    local d = b.direction

    if d == 'top' then
      -- slab occupying top of voxel
      -- can be placed from the top if there is no block below
      local belowBlock = self:findIndexAt(b.x, b.z, b.y-1)
      if not belowBlock then
        b.direction = nil
        table.remove(dirtyBlocks, k)
      end
    elseif d == 'bottom' then
      local _,db = self:findIndexAt(b.x, b.z, b.y-1)
      if db then
        if not db.direction or db.direction ~= 'bottom' then
          -- not a slab below, ok to place from above
          b.direction = nil
        end
        -- it is a slab below - must be pistoned
        table.remove(dirtyBlocks, k)
      end
    end
    spinner:spin(#dirtyBlocks + #dirtyBlocks2 .. ' blocks remaining ')
  end

  -- iterate through the directional blocks setting the placement strategy
  while #dirtyBlocks > 0 do
    local b = table.remove(dirtyBlocks)
    local d = b.direction or ''

    spinner:spin(#dirtyBlocks + #dirtyBlocks2 .. ' blocks remaining ')
 
    if directions[d] then
      b.heading = turtle.getHeadingInfo(directions[d]).heading
    end
  
    if doorDirections[d] then
 
      local hi = turtle.getHeadingInfo(doorDirections[d])
      b.heading = hi.heading
      b.twoHigh = true
 
      self:addPlacementChain({
        { x = b.x, z = b.z, y = b.y },
        { x = b.x - hi.xd, z = b.z - hi.zd, y = b.y },
      })
    end
 
    if stairDownDirections[d] then
      if not self:findIndexAt(b.x, b.z, b.y-1) then
        b.direction = stairDownDirections[b.direction]
        b.heading = turtle.getHeadingInfo(b.direction).heading
      else
        b.heading = turtle.getHeadingInfo(stairDownDirections[b.direction]).heading
      end
    end
 
    if d == 'bottom' then
      -- slab occupying bottom of voxel
      -- can be placed from top if a block is below
      -- otherwise, needs to be placed from side

      -- except... if the block below is a slab :(
        --local _,db = self:findIndexAt(b.x, b.z, b.y-1)
        --if not db then
          -- no block below, place from side

          -- took care of all other cases above
          self:bestSide(b, 'east', 'south', 'west', 'north')

        -- elseif not db.direction or db.direction ~= 'bottom' then
        -- not a slab below, ok to place from above
        --  b.direction = nil
        --end
        -- otherwise, builder will piston it in from above

    elseif stairUpDirections[d] then
      -- a directional stair
      -- turtle can place correctly from above if there is a block below
      -- otherwise, the turtle must place the block from the same plane
      -- against another block
      -- if no block to place against (from side) then the turtle must place from
      -- the other side 
      --
      -- Stair bug in 1.7 - placing a stair southward doesn't respect the turtle's direction
      -- all other directions are fine
      -- any stair southwards that can't be placed against another block must be pistoned
      local sd = stairUpDirections[d]
 
      if self:findIndexAt(b.x, b.z, b.y-1) then
        -- there's a block below
        b.direction = sd[1]
        b.heading = turtle.getHeadingInfo(b.direction).heading
      else
        local _,pb = self:findIndexAt(b.x + sd[3], b.z + sd[4], b.y)
        if pb and pb.direction ~= sd[5] then
          -- place stair against another block (that's not relying on this block to be down first)
          d = sd[2]  -- fall through to the blockDirections code below
          b.direction = sd[2]
        else
          b.heading = (turtle.getHeadingInfo(sd[1]).heading + 2) % 4
        end
      end
    end

    if blockDirections[d] then
      -- placing a block from the side
      local hi = turtle.getHeadingInfo(blockDirections[d])
      b.heading = hi.heading
      self:addPlacementChain({
        { x = b.x + hi.xd, z = b.z + hi.zd, y = b.y },  -- block we are placing against
        { x = b.x,         z = b.z,         y = b.y },  -- the block (or torch, etc)
        { x = b.x - hi.xd, z = b.z - hi.zd, y = b.y }   -- room for the turtle
      })
    end
  end
 
  -- pass 3
  while #dirtyBlocks2 > 0 do
    local b = table.remove(dirtyBlocks2)
    local d = b.direction

    spinner:spin(#dirtyBlocks2 .. ' blocks remaining ')

    if d == 'east-west-block' then
      self:bestOfTwoSides(b, 'east', 'west')
    elseif d == 'north-south-block' then
      self:bestOfTwoSides(b, 'north', 'south')
    end
  end
 
  spinner:stop()
  term.clearLine()
end
 
-- set the order for block dependencies
function Schematic:setPlacementOrder()
  local cursorX, cursorY = term.getCursorPos()
  local spinner = UI.Spinner({
    x = 1
  })
 
  Profile.start('overlapping')
  -- optimize for overlapping check
  for _,chain in pairs(self.placementChains) do
    for index,_ in pairs(chain.keys) do
      if not chain.startRow or (index < chain.startRow) then
        chain.startRow = index
      end
      if not chain.endRow or (index > chain.endRow) then
        chain.endRow = index
      end
    end
  end
 
  local function groupOverlappingChains(t, groupedChain, chain, spinner)
    local found = true
 
    local function overlaps(chain1, chain2)
      if chain1.startRow > chain2.endRow or
         chain2.startRow > chain1.endRow then
        return false
      end
      for k,_ in pairs(chain1.keys) do
        if chain2.keys[k] then
          return true
        end
      end
    end
 
    while found do
      found = false
      for k, v in pairs(t) do
        local o = overlaps(chain, v)
        if o then
          table.remove(t, k)
          table.insert(groupedChain, v)
          groupOverlappingChains(t, groupedChain, v, spinner)
          spinner:spin()
          found = true
          break
        end
      end
    end
  end

  -- group together any placement chains with overlapping blocks
  local groupedChains = {}
  while #self.placementChains > 0 do
    local groupedChain = {}
    local chain = table.remove(self.placementChains)
    table.insert(groupedChain, chain)
    table.insert(groupedChains, groupedChain)
    groupOverlappingChains(self.placementChains, groupedChain, chain, spinner)
    spinner:spin('chains: ' .. #groupedChains .. '  ' .. #self.placementChains .. '  ')
  end
  Profile.stop('overlapping')
 
  --Logger.log('schematic', 'groups: ' .. #groupedChains)
  --Logger.setFileLogging('chains')

  local function mergeChains(chains)

    --[[
    Logger.debug('---------------')
    Logger.log('schematic', 'mergeChains: ' .. #chains)
    for _,chain in ipairs(chains) do
      Logger.log('schematic', chain)
      for _,e in ipairs(chain) do
        Logger.log('schematic', string.format('%d:%d:%d %s %d:%d',
          e.block.x, e.block.z, e.block.y, tostring(e.block.direction), e.block.id, e.block.dmg))
      end
    end
    ]]--

    local masterChain = table.remove(chains)
 
    --[[ it's something like this:
 
       A chain     B chain           result
          1                            1
          2 --------  2                2
                      3                3
          4                            4
                      5                5
          6 --------  6                6
                      7                7
    --]]
    local function splice(chain1, chain2)
      for k,v in ipairs(chain1) do
        for k2,v2 in ipairs(chain2) do
          if v == v2 then
            local index = k
            local dupe
            for i = k2-1, 1, -1 do
              dupe = false
              -- traverse back through the first chain aligning on matches
              for j = index-1, 1, -1 do
                if chain1[j] == chain2[i] then
                 index = j
                 dupe = true
                 break
                end
              end
              if not dupe then
                table.insert(chain1, index, chain2[i])
              end
            end
            index = k+1
            for i = k2+1, #chain2, 1 do
              dupe = false
              for j = index, #chain1, 1 do
                if chain1[j] == chain2[i] then
                 index = j
                 dupe = true
                 break
                end
              end
              if not dupe then
                table.insert(chain1, index, chain2[i])
              end
              index = index + 1
            end
            return true
          end
        end
      end
    end
 
    while #chains > 0 do
      for k,chain in pairs(chains) do
        if splice(masterChain.blocks, chain.blocks) then
          table.remove(chains, k)
          break
        end
      end
    end

    --[[
    Logger.log('schematic', 'master chain: ')
    Logger.log('schematic', masterChain)
    Logger.log('schematic', '---------------')
    for _,e in ipairs(masterChain.blocks) do
      Logger.log('schematic', string.format('%d:%d:%d %s %s:%d',
        e.x, e.z, e.y, tostring(e.direction), e.id, e.dmg))
    end
    --]]

    return masterChain
  end
 
  -- combine the individual overlapping placement chains into 1 long master chain
  Profile.start('masterchains')
  local masterChains = {}
  for _,group in pairs(groupedChains) do
    spinner:spin('chains: ' .. #masterChains)
    table.insert(masterChains, mergeChains(group))
  end
  Profile.stop('masterchains')
 
  local function removeDuplicates(chain)
    for k,v in ipairs(chain) do
      for i = #chain, k+1, -1 do
        if v == chain[i] then
v.info = 'Unplaceable'
          table.remove(chain, i)
        end
      end
    end
  end
 
  -- any chains with duplicates cannot be placed correctly
  -- there are some cases where a turtle cannot place blocks the same as a player
  Profile.start('duplicates')
  for _,chain in pairs(masterChains) do
    removeDuplicates(chain.blocks)
    spinner:spin('chains: ' .. #masterChains)

    --[[
    Logger.log('schematic', "MASTER CHAIN")
    for _,e in ipairs(chain) do
      Logger.log('schematic', string.format('%d:%d:%d %s %d:%d',
        e.block.x, e.block.z, e.block.y, tostring(e.block.direction), e.block.id, e.block.dmg))
    end
    --]]

  end
  Profile.stop('duplicates')
  term.clearLine()
 
  -- adjust row indices as blocks are being moved
  Profile.start('reordering')
  for k,chain in pairs(masterChains) do
    spinner:spin('chains: ' .. #masterChains - k)
 
    local startBlock = table.remove(chain.blocks, 1)
    startBlock.movedBlocks = chain.blocks
 
    for _,b in pairs(chain.blocks) do
      b.moved = true
    end
  end
 
  local t = { }
  for k,b in ipairs(self.blocks) do

    -- adjust y so the turtle travels above the two high blocks
    if b.twoHigh then
      b.y = b.y + 1
    end

    spinner:spin('blocks: ' .. #self.blocks - k .. '  ')
    if not b.moved then
      table.insert(t, b)
    end
    if b.movedBlocks then
      for _,mb in ipairs(b.movedBlocks) do
        table.insert(t, mb)
      end
    end
  end
 
  self.blocks = t

  Profile.stop('reordering')

  --Logger.setWirelessLogging()

  term.clearLine()
  spinner:stop()
end
 
function Schematic:optimizeRoute()
 
  local function getNearestNeighbor(p, pt, maxDistance)
    local key, block, heading
    local moves = maxDistance

    local function getMoves(b, k)
      local distance = math.abs(pt.x - b.x) + math.abs(pt.z - b.z)

      if distance < moves then
        -- this operation is expensive - only run if distance is close
        local c, h = Point.calculateMoves(pt, b, distance)
        if c < moves then
          block = b
          key = k
          moves = c
          heading = h
        end
      end
    end
 
    local mid = pt.index
    local forward = mid + 1
    local backward = mid - 1
    while forward <= #p or backward > 0 do
      if forward <= #p then
        local b = p[forward]
        if not b.u then
          getMoves(b, forward)
          if moves <= 1 then
            break
          end
          if moves < maxDistance and math.abs(b.z - pt.z) > moves and pt.index > 0 then
            forward = #p
          end
        end
        forward = forward + 1
      end
      if backward > 0 then
        local b = p[backward]
        if not b.u then
          getMoves(b, backward)
          if moves <= 1 then
            break
          end
          if moves < maxDistance and math.abs(pt.z - b.z) > moves then
            backward = 0
          end
        end
        backward = backward - 1
      end
    end
    pt.x = block.x
    pt.z = block.z
    pt.y = block.y
    pt.heading = heading
    pt.index = key
    block.u = true
    return block
  end
 
  local pt = { x = -1, z = -1, y = 0, heading = 0 }
  local t = {}
  local cursorX, cursorY = term.getCursorPos()
  local spinner = UI.Spinner({
    x = 0,
    y = cursorY
  })
 
  local function extractPlane(y)
    local t = {}
    local dt = {}
    for _, b in pairs(self.blocks) do
      if b.y == y then
        if b.twoHigh then
          table.insert(dt, b)
        else
          table.insert(t, b)
        end
      end
    end
    return t, dt
  end
 
  local maxDistance = self.width*self.length
  Profile.start('optimize')
  for y = 0, self.height do
    local percent = math.floor(#t * 100 / #self.blocks) .. '%'
    spinner:spin(percent)
    local plane, doors = extractPlane(y)
    pt.index = 0
    for i = 1, #plane do
      local b = getNearestNeighbor(plane, pt, maxDistance)
      table.insert(t, b)
      spinner:spin(percent .. ' ' .. #plane - i .. '    ')
    end
    -- all two high blocks are placed last on each plane
    pt.index = 0
    for i = 1, #doors do
      local b = getNearestNeighbor(doors, pt, maxDistance)
      table.insert(t, b)
      spinner:spin(percent .. ' ' .. #doors - i .. '    ')
    end
  end

  Profile.stop('optimize')
  self.blocks = t
  spinner:stop('      ')
end

return Schematic