update caching & benchmark

Author: frostice482

bench.lua

@@ -19,8 +19,6 @@ end
 require"talisman.globals"
 require"big-num.omeganum"
 
-local B = to_big
-
 --- @param desc string
 --- @param n number
 --- @param fn fun()
@@ -29,21 +27,29 @@ local function bench(desc, n, fn)
 
 	local t = os.clock()
 	local m = collectgarbage("count")
+	local c = Big.created_instances
+
 	for i=1, n do fn() end
+
 	local td = os.clock() - t
 	local md = collectgarbage("count") - m
+	local cd = Big.created_instances - c
 
-	print(string.format('%10s - %6.2f ms; %6.2fMB; %.1f ops', desc, td * 1000, md / 1024, n/td))
+	print(string.format(
+		'%10s - %7.2f ms; %6.2fMB; %6.0f inst/n; %11.2f bytes/n; %.1f ops',
+		desc, td * 1000, md / 1024, cd / n, md / n * 1024, n/td
+	))
 
 	collectgarbage("collect")
 	collectgarbage("restart")
 end
 
-local y = B(111)
+local y = Big:create(111)
 local n
 
 bench("raw", 100000000, function() end)
 
+n = y
 bench("add", 100000, function() n = n + y end)
 
 n = y
@@ -59,4 +65,7 @@ n = y
 bench("tetrate", 100000, function() n = n:tetrate(4) end)
 
 n = y
-bench("arrow", 100000, function() n = n:tetrate(4, 4) end)
+bench("arrow", 100000, function() n = n:arrow(3, 3) end)
+
+n = Big:create(2.001)
+bench("arrow2", 10, function() n = n:arrow(200, 2.001) end)

big-num/omeganum.lua

@@ -31,7 +31,9 @@ local TalismanOmega_sizeof = assert(ffi.sizeof("struct TalismanOmega"))
 --- @operator mod(t.Omega|number): t.Omega
 --- @operator pow(t.Omega|number): t.Omega
 --- @operator unm(): t.Omega
-local Big = {}
+local Big = {
+    created_instances = 0
+}
 ;(Big).array = {} -- lsp hack, assign without recognized by lsp
 
 OmegaMeta = {
@@ -50,13 +52,18 @@ _G.Big = Big
 local B = {}
 
 -- prevent multiple allocation of same number at a frame
+--- @class _t.OmegaCache
+--- @field list table<number, t.Omega> persisting list
+--- @field current table<number, t.Omega> created omegas in this frame, move to old on next frame
+--- @field old table<number, t.Omega> created omegas in previous frame, cleared on next frame
 local caches = {
-    --- @type table<number, t.Omega>
     list = {},
-    --- @type table<number, number>
-    frames = {}
+    current = {},
+    currentcount = {},
 }
 
+Big.caches = caches
+
 -- prevent overrides
 local type = type
 local _math = math
@@ -100,6 +107,8 @@ end
 --- @param noNormalize? boolean
 --- @return t.Omega
 function Big:new(arr, sign, noNormalize)
+    Big.created_instances = Big.created_instances + 1
+
     --- @type t.Omega
     local obj = TalismanOmega() --- @diagnostic disable-line
     obj.asize = 1
@@ -116,12 +125,12 @@ function Big:create(input, sign)
         if obj then return obj end
         if input ~= input then return B.NaN end
 
-        local obj = Big:new({input}, input < 0 and -1 or 1)
-        caches.frames[input] = (caches.frames[input] or 0) + 1
-        if caches.frames[input] > 100 then
-            caches.frames[input] = nil
+        obj = caches.current[input] or Big:new({input}, input < 0 and -1 or 1)
+        caches.currentcount[input] = (caches.currentcount[input] or 0) + 1
+        if caches.currentcount[input] > 50 then
             caches.list[input] = obj
         end
+
         return obj
     elseif type(input) == "string" then
         return Big:parse(input)
@@ -1618,9 +1627,15 @@ end
 
 if love then
 
+local nextclear = 0
 local update = love.update
 function love.update(...)
-    caches.frames = {}
+    if G.TIMERS.REAL > nextclear or G.TIMERS.REAL < nextclear - 2 then
+        caches.current = {}
+        caches.currentcount = {}
+        nextclear = G.TIMERS.REAL + 1
+    end
+
     return update(...)
 end