-- chui: a set of VR UI push-to-operate components (no laser pointers)

local m = {}

local function vibrate(device, strength, duration, frequency)
  if device ~= 'mouse' and lovr.headset then
    lovr.headset.vibrate(device, strength, duration, frequency)
  end
end

local Q = 0.02  -- quant; all paddings and margins are its multiples
local S = 0.05  -- size of widget actuators
local button_roundness = 0.3
local slider_roundness = 0.1

m.palettes = { -- a built-in collection of UI color palettes
  --widget body     color for OFF        color for ON       highlight color   them letters     back-panel color
  { cap = 0xf0f0fb, inactive = 0xa2a6c1, active = 0xa479c7, hover = 0xb9aecd, text = 0x41486c, panel = 0xf6f7fe },
  { cap = 0x291d22, inactive = 0x3b3235, active = 0xf9b18e, hover = 0x9d5550, text = 0xfae8bc, panel = 0x374549 },
  { cap = 0x3b3149, inactive = 0x5c6181, active = 0xd47563, hover = 0xecc197, text = 0xecece0, panel = 0x191822 },
  { cap = 0x313131, inactive = 0x6f564c, active = 0xff9300, hover = 0xfdc484, text = 0x827d6d, panel = 0xf6b511 },
  { cap = 0xffeecc, inactive = 0x00b9be, active = 0xf57d7d, hover = 0xffb0a3, text = 0x15788c, panel = 0x264452 },
  { cap = 0x413a42, inactive = 0x1f1f29, active = 0xe68056, hover = 0x596070, text = 0xeaf0d8, panel = 0x16181b },
  { cap = 0x392b35, inactive = 0x7a9c96, active = 0xffab53, hover = 0x486b7f, text = 0xdac1c1, panel = 0x5e747e },
  { cap = 0x100f13, inactive = 0x372437, active = 0xa05642, hover = 0x693540, text = 0xc7955c, panel = 0x1a0d1e },
  { cap = 0x2a2a2b, inactive = 0x454a4d, active = 0x5a9470, hover = 0x2f7571, text = 0x81b071, panel = 0x202020 },
  { cap = 0x212124, inactive = 0x464c54, active = 0x76add8, hover = 0x5b8087, text = 0xa3e7f0, panel = 0x2b3a49 },
  { cap = 0x2e3b43, inactive = 0x619094, active = 0xdcfdcb, hover = 0x5a9e89, text = 0x5fa6ac, panel = 0x9ac0ba },
  { cap = 0xdddddd, inactive = 0x566063, active = 0x8caab5, hover = 0xfdfaf8, text = 0x073336, panel = 0xf4f4f3 },
  { cap = 0xa5c09d, inactive = 0x82a775, active = 0x7fe2e5, hover = 0xedf4f2, text = 0x165c44, panel = 0x308e7f },
  { cap = 0xd7417f, inactive = 0x3b3235, active = 0x785ea0, hover = 0x74509d, text = 0xfcd8d8, panel = 0xfd6193 },
  { cap = 0xecf1e6, inactive = 0xa8a9b9, active = 0x67bdc8, hover = 0x7fdd8e, text = 0x2d2614, panel = 0xf4fefe },
  { cap = 0xf5fffc, inactive = 0xa6a6a6, active = 0x5dd276, hover = 0x78cfd0, text = 0x173b4e, panel = 0xf0f4f3 },
  { cap = 0x46425e, inactive = 0xb28e7c, active = 0xdd9e43, hover = 0x72677b, text = 0xddc2bd, panel = 0x81828e },
}

m.mouse_available = (not lovr.headset) or (not lovr.headset.getName())
m.segments = 7  -- amount of geometry for roundrects and cylinders

m.panels = {}
m.widget_types = {}

-- SPACER ---------------------------------------------------------------------
m.spacer = {}
m.spacer.defaults = {}
table.insert(m.widget_types, 'spacer')

function m.spacer:init()
end


function m.spacer:draw(pass, pose)
end


function m.spacer:update(dt, pointer, pointer_name)
end


-- LABEL ----------------------------------------------------------------------
m.label = {}
m.label.defaults = { text = '', text_scale = 1 }
table.insert(m.widget_types, 'label')

function m.label:init(options)
  self.text = options.text
  self.text_scale = options.text_scale
end


function m.label:draw(pass, pose)
  -- text
  pass:setColor(self.parent.palette.text)
  pass:text(self.text, 0, 0, Q,  0.2 * self.text_scale)
end


function m.label:update(dt, pointer, pointer_name)
end


-- BUTTON ---------------------------------------------------------------------
m.button = {}
m.button.defaults = { text = '', thickness = 0.3, callback = nil, held = nil, text_scale = 0.3 }
table.insert(m.widget_types, 'button')

function m.button:init(options)
  self.interactive = true
  self.hovered = false
  self.text = options.text
  self.callback = options.callback
  self.held = options.held
  self.thickness = options.thickness
  self.text_scale = options.text_scale
  self.depth = self.thickness
end


function m.button:draw(pass)
  -- body
  pass:setColor(
    (self.depth < self.thickness / 2 and self.parent.palette.active) or
    (self.hovered and self.parent.palette.hover) or
    self.parent.palette.cap)
  pass:roundrect(0, 0, self.depth / 2,
    self.span[1] - 2 * Q, self.span[2] - 2 * Q, self.depth - Q,
    0, 0,1,0,
    button_roundness * 0.75, m.segments)
  -- frame
  pass:setColor(self.parent.palette.inactive)
  pass:roundrect(0, 0, Q / 2,
    self.span[1], self.span[2], Q,
    0, 0,1,0,
    button_roundness * 0.75, m.segments)
  -- text
  pass:setColor(self.parent.palette.text)
  pass:text(self.text, 0, 0, self.depth + Q, self.text_scale * self.span[2])
end


function m.button:update(dt, pointer, pointer_name)
  local new_depth = self.depth
  if pointer_name then -- pressing the button inward
    new_depth = math.min(self.thickness, math.max(2 * Q, pointer.z))
  end
  if pointer_name and self.hovered and -- button passed the threshold
      new_depth < self.thickness / 2 then
    if self.held then
      self.held(self)
    end
    if self.depth > self.thickness / 2 then
      vibrate(pointer_name, 0.2, 0.1)
      if self.callback then
        self.callback(self)
      end
    end
  end
  self.depth = new_depth
  self.hovered = pointer_name and true or false
  if not pointer_name then -- slowly rebound to above-hover depth when pointer leaves the widget
    self.depth = math.min(self.thickness, self.depth + 4 * dt)
  end
end


function m.button:get()
  return self.depth < self.thickness / 2
end


-- TOGGLE ---------------------------------------------------------------------
m.toggle = {}
m.toggle.defaults = { text = '', thickness = 0.3, state = false, callback = nil, text_scale = 0.3 }
table.insert(m.widget_types, 'toggle')

function m.toggle:init(options)
  self.interactive = true
  self.state = options.state
  self.hovered = false
  self.text = options.text
  self.callback = options.callback
  self.thickness = options.thickness
  self.text_scale = options.text_scale
  self.depth = self.thickness
end


function m.toggle:draw(pass)
  -- body
  pass:setColor(
    (self.state and self.parent.palette.active) or
    (self.hovered and self.parent.palette.hover) or
    self.parent.palette.cap)
  pass:roundrect(0, 0, self.depth / 2,
    self.span[1] - 2 * Q, self.span[2] - 2 * Q, self.depth - Q,
    0, 0,1,0,
    button_roundness, m.segments)
  -- frame
  pass:setColor(self.parent.palette.inactive)
  pass:roundrect(0, 0, Q / 2,
    self.span[1], self.span[2], Q,
    0, 0,1,0,
    button_roundness, m.segments)
  -- text
  pass:setColor(self.parent.palette.text)
  pass:text(self.text, 0, 0, self.depth + Q, self.text_scale * self.span[2])
end


function m.toggle:update(dt, pointer, pointer_name)
  local new_depth = self.depth
  if pointer_name then -- pressing the toggle inward
    new_depth = math.min(self.thickness, math.max(2 * Q, pointer.z))
  end
  if pointer_name and self.hovered and -- toggle button passed the threshold
      new_depth < self.thickness / 2 and
      self.depth > self.thickness / 2 then
    vibrate(pointer_name, 0.2, 0.1)
    self.state = not self.state
    if self.callback then
      self.callback(self, self.state)
    end
  end
  self.depth = new_depth
  self.hovered = pointer_name and true or false
  if not pointer_name then -- rebound
    self.depth = math.min(self.thickness, self.depth + 4 * dt)
  end
end


function m.toggle:get()
  return self.state
end


function m.toggle:set(state)
  self.state = state and true or false
  if self.callback then
    self.callback(self, self.state)
  end
end


-- GLOW -------------------------------------------------------------------------
m.glow = {}
m.glow.defaults = { text = '', thickness = 0.1, state = false, text_scale = 0.3 }
table.insert(m.widget_types, 'glow')

function m.glow:init(options)
  self.state = options.state
  self.text = options.text
  self.thickness = options.thickness
  self.text_scale = options.text_scale
end


function m.glow:draw(pass)
  -- body
  pass:setColor(
    (self.state and self.parent.palette.active) or
    self.parent.palette.inactive)
  pass:cylinder(0, 0, self.thickness / 2,
    0.5, self.thickness,
    0, 0,1,0, true, nil, nil, m.segments * 6)
  -- frame
  pass:setColor(self.parent.palette.inactive)
  pass:cylinder(0, 0, Q / 2,
    0.5 + Q, Q,
    0, 0,1,0, true, nil, nil, m.segments * 6)
  -- text
  pass:setColor(self.parent.palette.text)
  pass:text(self.text, 0, 0, self.thickness + Q,  self.text_scale)
end


function m.glow:update(dt, pointer, pointer_name)
end


function m.glow:get()
  return self.state
end


function m.glow:set(state)
  self.state = state and true or false
end


-- PROGRESS ---------------------------------------------------------------------
m.progress = {}
m.progress.defaults = { text = '', value = 0, text_scale = 0.3 }
table.insert(m.widget_types, 'progress')

function m.progress:init(options)
  self.text = options.text
  self.text_scale = options.text_scale
  self:set(options.value)
end


function m.progress:draw(pass)
  -- value as horizontal bar
  local y = -0.15
  local aw = self.span[1] - S - 2 * Q -- available width
  local w = self.value * aw
  pass:setColor(self.parent.palette.text)
  pass:box(0, y, 2 * Q,  aw - 2 * Q, 2 * S, S / 2)
  pass:setColor(self.parent.palette.active)
  pass:roundrect(-aw / 2 + w / 2, y, 4 * Q,
    w, 4 * S, 2 * S,
    0, 0,1,0,
    2 * Q, m.segments)
  -- text
  pass:setColor(self.parent.palette.text)
  pass:text(self.text, 0, 0.2, 2 * Q,  self.text_scale)
end


function m.progress:get()
  return self.value
end


function m.progress:set(value)
  self.value = math.max(0, math.min(1, value))
end


function m.progress:update(dt, pointer, pointer_name)
end


-- SLIDER ---------------------------------------------------------------------
m.slider = {}
m.slider.__index = m.slider
m.slider.defaults = { text = '',  format = '%s %.2f',
  min = 0, max = 1, value = 0, step = nil,
  text_scale = 0.3, thickness = 0.15,
  callback = nil, live_update = true }
table.insert(m.widget_types, 'slider')

local function roundBy(value, step)
  local quant, frac = math.modf(value / step)
  return step * (quant + (frac > 0.5 and 1 or 0))
end


function m.slider:init(options)
  self.interactive = true
  self.text = options.text
  self.min = options.min
  self.max = options.max
  self.thickness = options.thickness
  self.text_scale = options.text_scale
  self.callback = options.callback
  self.step = options.step
  self.format = options.format
  self.live_update = options.live_update
  self.altered = false

  if not options.format and self.step then
    local digits = math.max(0, math.ceil(-math.log(self.step, 10)))
    self.format = string.format('%%s %%.%df', digits)
  end
  local value = options.value
  if self.step then
    value = roundBy(value, self.step)
  end
  self.value = math.max(self.min, math.min(self.max, value))
end


function m.slider:draw(pass)
  -- value knob
  local y = -0.15
  local aw = self.span[1] - S - 2 * Q -- available width
  local pos = (self.value - self.min) / (self.max - self.min) * aw
  pass:setColor(self.parent.palette.text)
  pass:box(0, y, 2 * Q,  aw, 2 * S, S / 2)
  pass:setColor(self.parent.palette.active)
  pass:roundrect(-aw / 2 + pos, y, 2 * Q + self.thickness / 2,
    2 * S, 6 * S, self.thickness,
    0, 0,1,0,
    S, m.segments)
  -- frame
  pass:setColor(
    (self.altered and self.parent.palette.hover) or
    self.parent.palette.cap)
  pass:roundrect(0, 0, Q / 2,
    self.span[1], 1, Q,
    0, 0,1,0,
    slider_roundness, m.segments)
  -- text
  pass:setColor(self.parent.palette.text)
  pass:text(string.format(self.format, self.text, self.value),
    0, 0.2, 2 * Q,  self.text_scale)
end


function m.slider:update(dt, pointer, pointer_name)
  local hovered = pointer_name and true or false
  local altered_next = pointer.z < self.thickness
  if hovered and altered_next then
    local aw = self.span[1] - 16 * Q -- available width
    local value = self.min + (aw / 2 + pointer.x) / aw * (self.max - self.min)
    self:set(value)
    vibrate(pointer_name, 0.2, dt)
  end
  if not altered_next and self.altered and self.callback then
    self.callback(self, self.value)
  end
  self.altered = altered_next
end


function m.slider:get()
  return self.value
end


function m.slider:set(value)
  if self.step then
    value = roundBy(value, self.step)
  end
  self.value = math.max(self.min, math.min(self.max, value))
  if self.callback and self.live_update then
    self.callback(self, self.value)
  end
end


-- PANEL ----------------------------------------------------------------------
local panel = {}
panel.__index = panel

local panel_defaults = {
  frame = 'backpanel',
  palette = m.palettes[1],
}

function m.panel(options)
  options = options or {}
  local self = setmetatable({}, panel)
  self.is_panel = true
  self.frame = options.frame == nil and panel_defaults.frame or options.frame
  self.pose = Mat4(options.pose) -- the options.pose is allowed to be nil
  self.world_from_screen = Mat4()
  self.widgets = {}
  self.rows = {{}}
  self.span = {1, 1}
  self.palette = options.palette or panel_defaults.palette
  self.visible = true
  self.align_offset = Vec3()
  self.layout_options = {'center', 'center'}
  table.insert(m.panels, self)
  return self
end


function panel:reset()
  self.widgets = {}
  self.rows = {{}}
  self.align_offset:set(0, 0)
end


function panel:row()
  table.insert(self.rows, {})
end


function panel:nest(child_panel)
  assert(child_panel and type(child_panel) == 'table' and getmetatable(child_panel) == panel,
    '`child_panel` is not panel table', tostring(child_panel))
  child_panel.parent = self
  child_panel.widget_type = 'nested panel'
  self:appendWidget(child_panel)
end


local function scaledSpan(widget)
  local scale = 1
  if widget.is_panel then
    if widget.visible then
      scale = select(4, widget.pose:unpack())
    else
      scale = 0
    end
  end
  return widget.span[1] * scale, widget.span[2] * scale
end


-- set poses of contained widgets and calculate own span
function panel:layout(horizontal_alignment, vertical_alignment)
  horizontal_alignment = horizontal_alignment or self.layout_options[1]
  vertical_alignment = vertical_alignment or self.layout_options[2]
  self.layout_options = {horizontal_alignment, vertical_alignment}
  local margin = 8 * Q -- margin between rows and widgets in row
  self.span[1] = 0
  self.span[2] = 0
  -- calculate total dimensions
  local row_heights = {}
  local row_widths = {}
  for r, row in ipairs(self.rows) do
    local max_height = 0
    local row_width = 0
    for c, widget in ipairs(row) do
      local hspan, vspan = scaledSpan(widget)
      max_height = math.max(max_height, vspan)
      row_width = row_width + hspan + (c < #row and margin or 0)
    end
    row_widths[r] = row_width
    self.span[1] = math.max(self.span[1], row_width)
    table.insert(row_heights, max_height)
    self.span[2] = self.span[2] + max_height + (r < #self.rows and margin or 0)
  end
  -- lay out all widgets across all rows
  local x_row, y_row
  y_row = self.span[2] / 2

  for r, row in ipairs(self.rows) do
    local max_height = row_heights[r]
    local row_width = row_widths[r]
    if horizontal_alignment == 'left' then
      x_row = -self.span[1] / 2
    elseif horizontal_alignment == 'right' then
      x_row = self.span[1] / 2 - row_width
    else
      x_row = -row_width / 2
    end
    local x = x_row
    for _, widget in ipairs(row) do
      local hspan, vspan = scaledSpan(widget)
      local y = y_row
      if vertical_alignment == 'top' then
        y = y - vspan / 2
      elseif vertical_alignment == 'bottom' then
        y = y - max_height + vspan / 2
      else
        y = y - max_height / 2
      end
      local scale = widget.is_panel and select(4, widget.pose:unpack()) or 1
      widget.pose = Mat4(x + hspan / 2, y, 0):scale(scale)
      x = x + hspan + margin
    end
    y_row = y_row - max_height - margin
  end
  -- include panel border in the span
  if self.frame == 'backpanel' then
    self.span[1] = self.span[1] + 0.5
    self.span[2] = self.span[2] + 0.5
  end
  -- calculate offset from the panel's pose to align to edge or corner of the panel
  self.align_offset:set(0, 0)
  if horizontal_alignment == 'left' then
    self.align_offset:add(self.span[1] / 2, 0, 0)
  elseif horizontal_alignment == 'right' then
    self.align_offset:add(-self.span[1] / 2, 0, 0)
  end
  if vertical_alignment == 'top' then
    self.align_offset:add(0, -self.span[2] / 2, 0)
  elseif vertical_alignment == "bottom" then
    self.align_offset:add(0, self.span[2] / 2, 0)
  end
end


function panel:updateWidgets(dt, pointers)
  if not self.visible then return end
  local z_front, z_back = 1.5, -0.3 -- z boundaries of widget AABB
  local panel_pose_inv = self:getWorldPose():invert()
  for _, widget in ipairs(self.widgets) do
    local closest_pos
    local closest_name
    if widget.interactive then
      closest_pos = vec3(math.huge)
      for _, pointer in ipairs(pointers) do -- process each pointer
        local pos = vec3()
        local is_hovered = false
        -- reproject pointer onto panel coordinate system and check widget's AABB
        local pos_panel = panel_pose_inv:mul(vec3(pointer[2]))
        pos = mat4(widget.pose):invert():mul(pos_panel) -- in panel's coordinate system
        is_hovered = pos.x > -widget.span[1] / 2 and pos.x < widget.span[1] / 2 and
                     pos.y > -widget.span[2] / 2 and pos.y < widget.span[2] / 2 and
                     pos.z < z_front and pos.z > z_back
        if is_hovered and math.abs(pos.z) < math.abs(closest_pos.z) then
          closest_pos:set(pos)
          closest_name = pointer[1]
        end
      end
    end
    widget:update(dt, closest_pos, closest_name)
  end
end


function panel:getHeadsetPointers(pointers)
  for _, hand in ipairs(lovr.headset.getHands()) do
    local skeleton = lovr.headset.getSkeleton(hand)
    if skeleton then
      table.insert(pointers, {hand, vec3(unpack(skeleton[11]))})
    else
      table.insert(pointers, {hand, vec3(lovr.headset.getPosition(hand .. '/point'))})
    end
  end
end


function panel:getMousePointer(pointers, click_offset)
  -- flatten pose with parent poses
  local pose = self:getWorldPose()
  local scale = select(4, pose:unpack())
  -- overwrite hand/left in desktop VR sim, or make a new pointer for 3d desktop
  local mouse_pointer = pointers[1] or {'mouse', vec3()}
  -- make a ray in 3D space extending from underneath the mouse cursor to -Z
  local mx, my
  if lovr.system.isMouseGrabbed() then
    mx, my = lovr.system.getWindowDimensions()
    mx, my = mx / 2, my / 2
  else
    mx, my = lovr.system.getMousePosition()
  end
  local ray_origin = vec3(self.world_from_screen:mul(mx, my, 1))
  local ray_target = vec3(self.world_from_screen:mul(mx, my, 0.001))
  local ray_direction = (ray_target - ray_origin):normalize()
  -- intersect the ray onto panel plane and see if it lands within panel
  local plane_direction = quat(pose):direction()
  local dot = ray_direction:dot(plane_direction)
  if math.abs(dot) > 1e-5 then
    local plane_pos = vec3(pose)
    local ray_length = (plane_pos - ray_origin):dot(plane_direction) / dot
    local hit_spot = ray_origin + ray_direction * ray_length
    if click_offset then
      if lovr.system.isMouseDown(2) then
        mouse_pointer[2]:set(hit_spot)
      else -- back off the mouse pointer away from panel to emulate the hovering
        mouse_pointer[2]:set(hit_spot + plane_direction * -(0.25 * scale))
      end
    else
      mouse_pointer[2]:set(hit_spot)
    end
  end
  pointers[1] = mouse_pointer
end


function panel:getPointers(click_offset)
  local pointers = {}
  if lovr.headset then
    self:getHeadsetPointers(pointers)
  end
  if m.mouse_available then
    self:getMousePointer(pointers, click_offset)
  end
  return pointers
end


function panel:getScreenToWorldTransform(pass)
  local w, h = pass:getDimensions()
  local clip_from_screen = mat4(-1, -1, 0):scale(2 / w, 2 / h, 1)
  local view_pose = mat4(pass:getViewPose(1))
  local view_proj = pass:getProjection(1, mat4())
  -- m.is_orthographic = view_proj[16] == 1
  local world_from_screen = view_pose:mul(view_proj:invert()):mul(clip_from_screen)
  self.world_from_screen:set(world_from_screen)
end


function panel:getWorldPose()
  -- for nested panels, collect all transforms up to parentless root
  local stacked_pose = mat4()
  local parent = self.parent
  local child = self
  while parent do
    stacked_pose = child.pose * stacked_pose
    child = parent
    parent = parent.parent
  end
  -- for root apply both alignment translation and its world pose
  stacked_pose = mat4(child.pose):translate(child.align_offset) * stacked_pose
  return stacked_pose
end


function panel:update(dt)
  if not self.visible then return end
  local pointers = self:getPointers(true)
  -- TODO: skip update if outside the panel's AABB
  self:updateWidgets(dt, pointers)
end


function panel:draw(pass, draw_pointers)
  if not self.visible then return end
  if m.mouse_available then
    self:getScreenToWorldTransform(pass)
  end
  pass:push()
  if self.parent then
    pass:transform(0, 0, Q)
  else
    pass:transform(self.pose)
    pass:transform(vec3(self.align_offset))
  end
  pass:setColor(0.820, 0.816, 0.808)
  if self.frame == 'backpanel' then
    pass:setColor(self.palette.panel)
    pass:roundrect(0, 0, -Q / 2,
      self.span[1], self.span[2], Q ,
      0, 0,1,0, 0.4)
  end
  pass:setFont(m.font)
  for _, w in ipairs(self.widgets) do
    pass:push()
    pass:transform(w.pose)
    --[[ widget frames for debugging
    pass:setColor(1,0,0)
      pass:box(0, 0, 0.2, w.span[1], w.span[2], 0.01, 0, 0,1,0, 'line')
      pass:text(w.widget_type or 'FOO', 0, w.span[2] / 2 - 0.2, 0.2, 0.3)
    pass:setColor(1,1,1)
    --]]
    w:draw(pass)
    pass:pop()
  end
  pass:pop()
  if draw_pointers then
    local pointers = pointers or self:getPointers(false)
    pass:setColor(0x404040)
    local radius = 0.005
    for _, pointer in ipairs(pointers) do
      pass:sphere(mat4(pointer[2]):scale(radius), m.segments, m.segments)
    end
  end
end


function panel:setVisible(is_visible)
  if self.visible and not is_visible then
    -- complete any ongoing interactions (hovered pointers)
    local dt = lovr.timer.getDelta()
    for _, widget in ipairs(self.widgets) do
      widget:update(dt, vec3(math.huge), nil)
    end
  end
  self.visible = is_visible
end


function panel:appendWidget(widget)
  table.insert(self.widgets, widget)
  table.insert(self.rows[#self.rows], widget)
end


-- creates panel methods for constructing widgets with light OOP based on metatables
function m.initWidgetType(widget_name, widget_proto)
  widget_proto.__index = widget_proto
  -- define constructor, for example panel:button{text = 'click'} adds new button to the panel
  panel[widget_name] = function(self, options)
    options = options or {}
    setmetatable(options, widget_proto.defaults)
    widget_proto.defaults.__index = widget_proto.defaults
    local widget = setmetatable({}, widget_proto)
    if type(options.span) == 'number' then
      widget.span = {options.span, 1}
    elseif type(options.span) == 'table' and #options.span == 2 then
      widget.span = {options.span[1], options.span[2]}
    elseif not options.span then
      widget.span = {1, 1}
    else
      assert(false, "unsupported widget span value")
    end
    widget.widget_type = widget_name
    widget.parent = self
    self:appendWidget(widget)
    widget:init(options)
    return widget
  end
end


local function initAllWidgets()
  for _, widget_name in ipairs(m.widget_types) do
    local widget_proto = m[widget_name]
    m.initWidgetType(widget_name, widget_proto)
  end
end

initAllWidgets()


-- CHUI HELPERS ---------------------------------------------------------------

function m.setFont(font) -- accepts path to file or loaded font instance
  if type(font) == 'string' then -- path to font file
    local ok, res = pcall(lovr.graphics.newFont, font, 32, 4)
    if ok then
      m.font = res
    else
      print('could not load \'' .. font .. '\', defaulting to built-in Varela Round')
      m.font = lovr.graphics.getDefaultFont()
    end
  elseif tostring(font):match('Font') then -- a font instance used as-is
    m.font = font
  else
    m.font = lovr.graphics.getDefaultFont()
  end
end


-- convenience functions for multiple panels, user can also just call :update & :draw on the panel

function m.update(dt) -- neccessary for UI interactions
  for _, pnl in ipairs(m.panels) do
    if not pnl.parent then
      pnl:update(dt)
    end
  end
end


function m.draw(pass, draw_pointers)
  for _, pnl in ipairs(m.panels) do
    if not pnl.parent then
      pnl:draw(pass, draw_pointers)
    end
  end
end


function m.reset() -- forget the collected panels
  m.panels = {}
end

return m