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.

output/luaScripts/ButtonCount.lua

@@ -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