local gears = require("gears")
local gcolor = require("gears.color")
local beautiful = require("beautiful")
local math     = math
local screen   = screen
local mylayout = {}
mylayout.name = "equalarea"

local function divide(p,g,low,high,cls,mwfact,mcount)
  if low == high then
    p.geometries[cls[low]] = g
  else
    local masters = math.max(0,math.min(mcount,high)-low+1)
    local numblock = high-low + 1
    local slaves = numblock - masters
    local smalldiv
    if numblock > 5 and (numblock % 5) == 0 then
      smalldiv = math.floor(numblock/5)
    else
      if (numblock % 3) == 0 then
        smalldiv = math.floor(numblock/3)
      else
        smalldiv = math.floor(numblock/2)
      end
    end
    local bigdiv = numblock - smalldiv
    local smallmasters = math.min(masters,smalldiv)
    local bigmasters = masters-smallmasters
    local smallg = {}
    local bigg = {}
    smallg.x = g.x
    smallg.y = g.y
    if g.width > (g.height*1.3) then
      smallg.height = g.height
      bigg.height = g.height
      bigg.width = math.floor(g.width*(bigmasters*(mwfact-1)+bigdiv)/(slaves+mwfact*masters))
      smallg.width = g.width-bigg.width
      bigg.y = g.y
      bigg.x = g.x + smallg.width
    else
      smallg.width = g.width
      bigg.width = g.width
      bigg.height = math.floor(g.height*(bigmasters*(mwfact-1)+bigdiv)/(slaves+mwfact*masters))
      smallg.height = g.height-bigg.height
      bigg.x = g.x
      bigg.y = g.y + smallg.height
    end
    divide(p,smallg,low,high-bigdiv,cls,mwfact,mcount)
    divide(p,bigg,low+smalldiv,high,cls,mwfact,mcount)
  end
  return
end

function mylayout.arrange(p)
  local t   = p.tag or screen[p.screen].selected_tag
  local wa  = p.workarea
  local cls = p.clients

  if #cls == 0 then return end
  local mwfact          = t.master_width_factor*2
  local mcount = t.master_count
  local g = {}
  g.height = wa.height
  g.width = wa.width
  g.x = wa.x
  g.y = wa.y
  divide(p,g,1,#cls,cls,mwfact,mcount)
end

local icon_raw = gears.filesystem.get_configuration_dir() .. tostring(...):match("^.*bling"):gsub("%.", "/") .. "/icons/layouts/equalarea.png"

local function get_icon()
  if icon_raw ~= nil then
    return gcolor.recolor_image(icon_raw, beautiful.fg_normal)
  else
    return nil
  end
end

return {
  layout = mylayout,
  icon_raw = icon_raw,
  get_icon = get_icon,
}