benchmark.lua

local vec2 = require("math.vec2")
local vec3 = require("math.vec3")
local framebuffer = require("render.framebuffer")
local draw = require("render.draw")
local colorwheel_shader = require("shaders.colorwheel")
local median_split = require("quantization.median_split")
local color_distance = require("quantization.color_distance")

-- Get term size
local tw, th = term.getSize()
local term_area = tw * th
local term_size = vec2(tw, th)

-- Create and shade the framebuffer
local shade_start = os.epoch("utc")
local buf = framebuffer.new(term_size)
buf:shade(colorwheel_shader)
local shade_end = os.epoch("utc")

-- Quantize the image
local quant_start = os.epoch("utc")
local image = buf:quantize(median_split, color_distance.rgb)
local quant_end = os.epoch("utc")

-- Draw the image
local draw_start = os.epoch("utc")
term.clear()
term.setCursorPos(1,1)
draw.draw_image(image)
local draw_end = os.epoch("utc")

-- Clear screen
draw.default_palette()
term.clear()
term.setCursorPos(1,1)

-- Print calculations
local shade_time = shade_end - shade_start
local quant_time = quant_end - quant_start
local draw_time = draw_end - draw_start
local total_time = shade_time + quant_time + draw_time
print("shade time: " .. shade_time .. " ms")
print("shade p/s: " .. ("%.f"):format(term_area / shade_time * 1000))
print("quant time: " .. quant_time .. " ms")
print("quant p/s: " .. ("%.f"):format(term_area / quant_time * 1000)) -- 32.3 * 1000 is cursed, it prints as 323 so we must format it for safety
print("draw time: " .. draw_time .. " ms")
print("draw p/s: " ..  ("%.f"):format(term_area / draw_time * 1000))
print("total: " .. total_time .. " ms")
print("fps: " .. 1000 / total_time)