neat-donk/runner-wrapper.lua

95 lines
2.5 KiB
Lua
Raw Normal View History

2021-04-09 15:06:55 -04:00
local base = string.gsub(@@LUA_SCRIPT_FILENAME@@, "(.*[/\\])(.*)", "%1")
local json = dofile(base.."/dkjson.lua")
local tmpFileName = "/tmp/donk_runner_"..tostring(math.floor(random.integer(0, 0xffffffffffffffff))):hex()
local function message(_M, msg, color)
if color == nil then
color = 0x00009900
end
for i=#_M.onMessageHandler,1,-1 do
_M.onMessageHandler[i](msg, color)
end
end
local function save(_M)
for i=#_M.onSaveHandler,1,-1 do
_M.onSaveHandler[i](_M.saveLoadFile)
end
end
local function onSave(_M, handler)
table.insert(_M.onSaveHandler, handler)
end
local function load(_M)
for i=#_M.onLoadHandler,1,-1 do
_M.onLoadHandler[i](_M.saveLoadFile)
end
end
local function onLoad(_M, handler)
table.insert(_M.onLoadHandler, handler)
end
return function()
local _M = {
onMessageHandler = {},
}
_M.onRenderForm = function(handler)
end
_M.onMessage = function(handler)
onMessage(_M, handler)
end
_M.run = function(species, generationIdx, speciesIdx, genomeCallback, finishCallback)
local trunc = io.open(tmpFileName, 'w')
trunc:close()
local poppets = {}
for i=1,#species,1 do
local poppet = io.popen("RUNNER_DATA='"..json.encode({species[i], generationIdx, speciesIdx + i - 1, tmpFileName}).."' lsnes --rom="..base.."/rom.sfc --unpause --lua="..base.."/runner-process.lua", 'r')
table.insert(poppets, poppet)
end
for i=1,#poppets,1 do
local poppet = poppets[i]
poppet:read('*a')
poppet:close()
end
local tmpFile = io.open(tmpFileName, "r")
local line = ""
repeat
local obj, pos, err = json.decode(line)
if err ~= nil then
goto continue
end
if obj.type == 'onMessage' then
message(_M, obj.msg, obj.color)
elseif obj.type == 'onGenome' then
species[obj.speciesIndex - speciesIdx + 1].genomes[obj.genomeIndex] = obj.genome
genomeCallback(obj.genome, obj.index)
elseif obj.type == 'onFinish' then
finishCallback()
end
::continue::
line = tmpFile:read()
until(line == "")
tmpFile:close()
local ok, err = os.remove(tmpFileName)
if err ~= nil then
message(_M, err)
elseif ok ~= true then
message(_M, 'UNSPECIFIED ERROR ON REMOVAL')
end
end
return _M
end