Button Count

-As Lua starts counting from 1, and there may be a slot 0, increment the slot before saving / loading, storing data for slot 0 in states[1]
-Prevented the script from loading non-existent data.
-Removed the space from the file name so that it is more consistent with the FCEUX Lua repository, which this script is now a member of. :)
-Added messages when you save or (fail to) load data to / from a state.
-Made it so the script displays the button counts when you save / load a state.
This commit is contained in:
brandonevans 2011-07-17 20:15:41 +00:00
parent d3c2998c8e
commit a1b6392832
1 changed files with 152 additions and 0 deletions

View File

@ -0,0 +1,152 @@
--Written by Brandon Evans
--You can change the position of the text here.
local x = 0
local y = 8
local holds = 0
local players = {}
local presses = 0
local states = {}
function table.copy(t)
local t2 = {}
for k, v in pairs(t) do
t2[k] = v
end
return t2
end
function counts()
--Display the counts of the holds and the presses.
gui.text(x, y, 'Holds: ' .. holds)
gui.text(x, y + 8, 'Presses: ' .. presses)
end
function load(slot)
--As Lua starts counting from 1, and there may be a slot 0, increment.
slot = slot + 1
if not states[slot] or not states[slot].holds then
gui.text(x, y + 16, 'No data loaded from slot ' .. tostring(slot - 1))
counts()
return
end
--Load the data if there is any available for this slot.
holds = states[slot].holds
players = table.copy(states[slot].players)
presses = states[slot].presses
gui.text(x, y + 16, 'Data loaded from slot ' .. tostring(slot - 1))
counts()
end
function parse()
--If there is an open, read-only FM2 file, parse it for the initial data.
if not movie.active() then
return false
end
local fh = io.open(movie.name())
if not fh or movie.mode() == 'record' or movie.name():match(
'.%.(%w+)$'
) ~= 'fm2' then
return false
end
local frame = -1
local last = {}
local match = {
'right', 'left', 'down', 'up', 'start', 'select', 'B', 'A'
}
--Parse up until two frames before the current one.
while frame ~= emu.framecount() - 2 do
line = fh:read()
if not line then
break
end
--This is only a frame if it starts with a vertical bar.
if string.sub(line, 0, 1) == '|' then
frame = frame + 1
players = {}
local player = -1
--Split up the sections by a vertical bar.
for section in string.gmatch(line, '[^|]+') do
player = player + 1
--Only deal with actual players.
if player ~= 0 then
local button = 0
--Run through all the buttons.
for text in string.gmatch(section, '.') do
button = button + 1
--Check if this button is pressed.
if text ~= ' ' and text ~= '.' then
holds = holds + 1
--If the button was not previously pressed,
--increment.
if table.maxn(last) < player or not last[player][
match[button]
] then
presses = presses + 1
end
if table.maxn(players) < player then
table.insert(players, {})
end
--Mark this button as pressed.
players[player][match[button]] = true
end
end
end
end
--Save the players to compare with the next frame in the file.
last = players
end
end
return true
end
function save(slot)
--As Lua starts counting from 1, and there may be a slot 0, increment.
slot = slot + 1
while table.maxn(states) < slot do
table.insert(states, {})
end
--Mark the current data as the data for this slot.
states[slot].holds = holds
states[slot].players = table.copy(players)
states[slot].presses = presses
gui.text(x, y + 16, 'Data saved to slot ' .. tostring(slot - 1))
counts()
end
if parse() then
gui.text(x, y + 16, 'Movie parsed for data')
else
gui.text(x, y + 16, 'No movie parsed for data')
end
savestate.registerload(load)
savestate.registersave(save)
while true do
--If this is the first frame, reset the data.
if emu.framecount() == 0 then
holds = 0
players = {}
presses = 0
end
--Check players one and two.
for player = 1, 2 do
local buttons = joypad.getdown(player)
--Run through all of the pressed buttons.
for i, v in pairs(buttons) do
holds = holds + 1
--If in the previous frame the button was not pressed, increment.
if table.maxn(players) >= player and not players[player][i] then
presses = presses + 1
end
end
if table.maxn(players) < player then
table.insert(players, true)
end
--Mark these buttons as pressed.
players[player] = buttons
end
counts()
emu.frameadvance()
end