baseAtlas.lua

-- Copyright (c) 2021 EngineerSmith
-- Under the MIT license, see license suppiled with this file

local lg = love.graphics

local baseAtlas = {
  _canvasSettings = {
    dpiscale = 1,
  },
  _maxCanvasSize =  lg.getSystemLimits().texturesize - 1,
  _arrayTextureSupport = lg.getTextureTypes()["array"],
}
baseAtlas.__index = baseAtlas

baseAtlas.new = function(padding, extrude, spacing)
  return setmetatable({
    padding = padding or 1,
    extrude = extrude or 0,
    spacing = spacing or 0,
    image,
    images = {},
    imagesSize = 0,
    ids = {},
    quads = {},
    filterMin = "linear", 
    filterMag = "linear",
    bakeAsPow2 = false,
    _dirty = false, -- Marked dirty if image is added or removed,
    _hardBake = false, -- Marked true if hardBake has been called, cannot use add, remove or bake once true
  }, baseAtlas)
end

-- TA:add(img, "foo")
-- TA:add(img, 68513, true)
baseAtlas.add = function(self, image, id, bake, ...)
  if self._hardBake then
    error("Cannot add images to a texture atlas that has been hard baked")
  end
  self.imagesSize = self.imagesSize + 1
  local index = self.imagesSize
  assert(type(id) ~= "nil", "Must give an id")
  self:remove(id)
  self.images[index] = {
    image = image,
    id = id,
    index = index,
  }
  self.ids[id] = index
  
  self._dirty = true
  if bake then
    self:bake(...)
  end
  
  return self
end

-- TA:remove("foo", true)
-- TA:remove(68513)
baseAtlas.remove = function(self, id, bake, ...)
  if self._hardBake then
    error("Cannot remove images from a texture atlas that has been hard baked")
  end
  local index = self.ids[id]
  if index then
    self.images[index] = nil
    self.quads[id] = nil
    self.ids[id] = nil
    self._dirty = true
    if bake == true then
      self:bake(...)
    end
  end
  
  return self
end

baseAtlas.bake = function(self)
  error("Warning! Created atlas hasn't overriden bake function!")
  return self
end

baseAtlas.hardBake = function(self, ...)
  local _, data = self:bake(...)
  self.images = nil
  self.ids = nil
  self._hardBake = true
  return self, data
end

-- returns position on texture atlas, x,y, w,h
baseAtlas.getViewport = function(self, id)
  local quad = self.quads[id]
  if quad then
    return quad:getViewport()
  end
  error("Warning! Quad hasn't been baked for id: " .. tostring(id))
end

baseAtlas.setFilter = function(self, min, mag)
  self.filterMin = min or "linear"
  self.filterMag = mag or self.filterMin
  if self.image then
    self.image:setFilter(self.filterMin, self.filterMag)
  end
  return self
end

baseAtlas.setBakeAsPow2 = function(self, bakeAsPow2)
  self.bakeAsPow2 = bakeAsPow2 or false
end

baseAtlas.setPadding = function(self, padding)
  self.padding = padding or 1
end

baseAtlas.setExtrude = function(self, extrude)
  self.extrude = extrude or 0
end

baseAtlas.setSpacing = function(self, spacing)
  self.spacing = spacing or 0
end

baseAtlas.draw = function(self, id, ...)
  lg.draw(self.image, self.quads[id], ...)
end

baseAtlas.getDrawFuncForID = function(self, id)
  return function(...)
    lg.draw(self.image, self.quads[id], ...)
  end
end

return baseAtlas