Module:FunList/Iterators: Difference between revisions
From Melvor Idle
No edit summary |
No edit summary |
||
Line 16: | Line 16: | ||
function Enumerator:moveNext() | function Enumerator:moveNext() | ||
mw.log('calling broken func') | |||
error('Not implemented in base class.') | error('Not implemented in base class.') | ||
end | end | ||
Line 46: | Line 47: | ||
function TableEnumerator:moveNext() | function TableEnumerator:moveNext() | ||
mw.log('calling override with') | |||
mw.log('key: ' .. self.index) | |||
mw.log('val: ' .. self.current) | |||
-- Grab the next index for the internal table. | -- Grab the next index for the internal table. | ||
local key = self.index | local key = self.index | ||
Line 52: | Line 56: | ||
-- If the index exist, we have succesfuly moved to the next. | -- If the index exist, we have succesfuly moved to the next. | ||
if key then | if key ~= nil then | ||
self.current = self.data[key] | self.current = self.data[key] | ||
return true | return true | ||
end | end |
Revision as of 15:53, 21 July 2024
Documentation for this module may be created at Module:FunList/Iterators/doc
-- BASE ENUMERATOR CLASS --
--enumerable = {}
local Enumerator = {}
local Enumerator_mt = {
__index = Enumerator,
__pairs = function(t) return t:overridePairs() end,
--__ipairs = function(t) return t:overrideiPairs() end
}
function Enumerator.new()
local self = setmetatable({}, Enumerator_mt)
self.current = nil
self.index = nil
return self
end
function Enumerator:moveNext()
mw.log('calling broken func')
error('Not implemented in base class.')
end
-- Hooks the moveNext function into the Lua 'pairs' function
function Enumerator:overridePairs()
local function iterator(t, k)
if self:moveNext() == true then
return self.index, self.current
end
return nil, nil
end
self.index = nil
self.current = nil
return iterator, self, nil
end
-- TABLE ENUMERATOR CLASS --
-- This is essentially a wrapper for the table object,
-- since it provides no state machine esque iteration out of the box
local TableEnumerator = setmetatable({}, { __index = Enumerator })
TableEnumerator.__index = TableEnumerator
function TableEnumerator.new(tbl)
local self = setmetatable(Enumerator.new(), TableEnumerator)
self.data = tbl
return self
end
function TableEnumerator:moveNext()
mw.log('calling override with')
mw.log('key: ' .. self.index)
mw.log('val: ' .. self.current)
-- Grab the next index for the internal table.
local key = self.index
key = next(self.data, key)
self.index = key
-- If the index exist, we have succesfuly moved to the next.
if key ~= nil then
self.current = self.data[key]
return true
end
return false
end
return {
Enumerator = Enumerator,
TableEnumerator = TableEnumerator
}