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,
}
