Implement profiler

halyde/apps/help.lua

@@ -128,7 +128,7 @@ handle:close()
 
 --print(require("serialize")(data, "\t"))
 
--- Halyde terminal doesn't support bold (CSI 1 m) but who cares
+-- Halyde io doesn't support bold (CSI 1 m) but who cares
 
 if data.command then
   terminal.write("\27[1mUsage: \27[0m\n")

halyde/apps/helpdb/echo

@@ -1,9 +1,9 @@
 COMMAND echo
 USAGE [TEXT]...
-DESCRIPTION Concatenates and prints text to the terminal.
+DESCRIPTION Concatenates and prints text to the standard output.
 ARG1 TEXT
 ARG1DESCRIPTION Text to print.
 EXAMPLE1 echo test
 EXAMPLE2 echo Hello World!
-EXAMPLE1DESCRIPTION Prints "test" to the terminal.
-EXAMPLE2DESCRIPTION Prints "Hello World!" to the terminal.
+EXAMPLE1DESCRIPTION Prints "test" to the standard output.
+EXAMPLE2DESCRIPTION Prints "Hello World!" to the standard output.

halyde/apps/profile.lua

@@ -0,0 +1,222 @@
+-- TODO: make it somehow not depend on ESC s and ESC u
+local profiler = require("profiler")
+local PAL = {
+  { fg = "\27[31m", bg = "\27[41m" },
+  { fg = "\27[32m", bg = "\27[42m" },
+  { fg = "\27[33m", bg = "\27[43m" },
+  { fg = "\27[34m", bg = "\27[44m" },
+  { fg = "\27[35m", bg = "\27[45m" },
+  { fg = "\27[36m", bg = "\27[46m" },
+  { fg = "\27[37m", bg = "\27[47m" }
+}
+local function pal(i) return PAL[((i - 1) % #PAL) + 1] end
+
+local results = profiler.results()
+if not results then
+  print("No profiling data")
+  return
+end
+
+local radius
+local W = terminal.getResolution()
+if (W == 160) then radius = 20
+elseif (W == 80) then radius = 6
+else radius = 0; PAL = {{fg = "", bg = ""}} end
+
+local MIN_PC = math.min(3, 100 / (radius * 2))
+
+local total = 0
+for _, r in ipairs(results) do total = total + r.time end
+
+local main = {}
+local other = {}
+local ot = 0
+local on_ = 0
+for _, r in ipairs(results) do
+  if r.time / total * 100 >= MIN_PC then
+    table.insert(main, r)
+  else
+    table.insert(other, r)
+    ot = ot + r.time;
+    on_ = on_ + 1
+  end
+end
+if on_ then
+  table.insert(main, { label = "other (" .. on_ .. ")", time = ot })
+end
+
+local START = -math.pi / 2
+local slices = {}
+local cur = START
+for i, m in ipairs(main) do
+  local sw = m.time / total * 2 * math.pi
+  slices[i] = {
+    label = m.label,
+    time = m.time,
+    pc = ("%.1f%%"):format(m.time / total * 100),
+    color = pal(i),
+    sa = cur,
+    sw = sw
+  }
+  cur = cur + sw
+end
+if on_ then slices[#slices].color={fg = "\27[30m", bg = "\27[40m"} end
+
+local ASP = 2
+local cx = radius * ASP
+local cy = radius
+
+local function slice_at(sx, sy)
+  local dy = sy - cy
+  local dx = (sx - cx) / ASP
+  if dx * dx + dy * dy > radius * radius then return nil end
+  local a = math.atan2(dy, dx)
+  if a < START then a = a + 2 * math.pi end
+  for i, sl in ipairs(slices) do
+    if a >= sl.sa and a < sl.sa + sl.sw then return i end
+  end
+  return #slices
+end
+
+local SUB = {
+  { dx = -0.25, dy = -0.375 },
+  { dx = -0.25, dy = -0.125 },
+  { dx = -0.25, dy =  0.125 },
+  { dx =  0.25, dy = -0.375 },
+  { dx =  0.25, dy = -0.125 },
+  { dx =  0.25, dy =  0.125 },
+  { dx = -0.25, dy =  0.375 },
+  { dx =  0.25, dy =  0.375 },
+}
+
+for y = 0, radius * 2 do
+  local lx, rx = math.ceil(cx - radius * ASP), math.floor(cx + radius * ASP)
+
+  terminal.write(("\27[%dC"):format(lx))
+  for x = lx, rx do
+    local c = {}
+    local n = 0
+    for k = 1, 8 do
+      local s = slice_at(x + SUB[k].dx, y + SUB[k].dy)
+      c[k] = s
+      if s then n = n + 1 end
+    end
+
+    if n == 0 then
+      terminal.write("\27[0m ")
+    else
+      local counts = {}
+      local order = {}
+      for k = 1, 8 do
+        if c[k] then
+          local key = c[k]
+          if not counts[key] then
+            counts[key] = 0
+            table.insert(order, key)
+          end
+          counts[key] = counts[key] + 1
+        end
+      end
+      table.sort(order, function(a, b) return counts[a] > counts[b] end)
+      local dom = order[1]
+      local sub = order[2]
+
+      if n == 8 then
+        terminal.write(slices[dom].color.bg)
+        if sub == nil then
+          terminal.write(" ")
+        else
+          local mask = 0
+          for k = 1, 8 do
+            if c[k] == sub then mask = mask + (1 << (k - 1)) end
+          end
+          terminal.write(slices[sub].color.fg)
+          terminal.write(utf8.char(0x2800 + mask))
+        end
+      else
+        local mask = 0
+        for k = 1, 8 do
+          if c[k] == dom then mask = mask + (1 << (k - 1)) end
+        end
+        terminal.write("\27[0m")
+        terminal.write(slices[dom].color.fg)
+        terminal.write(utf8.char(0x2800 + mask))
+      end
+    end
+  end
+  terminal.write('\n')
+end
+
+terminal.write("\27[" .. radius * 2 + 1 .. "A\27[s")
+local function draw_text(col, row, s)
+  terminal.write("\27[u\27[" .. col .. "C\27[" .. row - 1 .. "B" .. s)
+end
+
+terminal.write("\27[0m")
+if radius == 0 then goto tier1gpu end
+for _, sl in ipairs(slices) do
+  local y = math.floor(cy + radius * 0.62 * math.sin(sl.sa + sl.sw / 2) + 0.5)
+
+  local function in_slice(x, row)
+    local a = math.atan2(row - cy, (x - cx) / ASP)
+    if a < START then
+      a = a + 2 * math.pi
+    end
+    return a >= sl.sa and a < sl.sa + sl.sw
+  end
+  local function centered_draw(row, text)
+    local dy = row - cy
+    local half_w = math.sqrt(radius * radius - dy * dy) * ASP
+    local lx, rx = math.ceil(cx - half_w), math.floor(cx + half_w)
+    while lx <= rx and not in_slice(lx, row) do
+      lx = lx + 1
+    end
+    while rx >= lx and not in_slice(rx, row) do
+      rx = rx - 1
+    end
+    local avail = rx - lx + 1
+    if avail < 3 then
+      return
+    end
+    local t = #text > avail and text:sub(1, avail - 3) .. "..." or text
+    local start_x = lx + math.floor((avail - #t) / 2)
+    draw_text(start_x, row, sl.color.bg .. t)
+  end
+  centered_draw(y, sl.label)
+  centered_draw(y + 1, sl.pc)
+end
+::tier1gpu::
+
+local max_w = 0
+for _, sl in pairs(slices) do
+  max_w = math.max(#sl.label + 1, max_w)
+end
+for _, sl in pairs(other) do
+  max_w = math.max(#sl.label + 3, max_w)
+end
+local x = radius * 2 * ASP + 2
+if radius == 0 then x = 0 end
+local cw = W - x - 1
+local suffix_w = 18
+local max_lbl = math.max(4, cw - suffix_w)
+terminal.write("\27[u\27[" .. x .. "C" .. ("\27[0m%" .. max_w .. "s  %9s   %6s"):format("label", "time", "share"))
+terminal.write("\n")
+for _, sl in ipairs(slices) do
+  local lbl = sl.label
+  if #lbl > max_lbl then
+    lbl = sl.label:sub(1, max_lbl - 1) .. "…"
+  end
+  terminal.write("\n\27[" .. x .. "C" .. ("%s%" .. max_w .. "s\27[0m  %9.4fs  %6s"):format(sl.color.bg, lbl, sl.time, sl.pc))
+end
+for _, sl in ipairs(other) do
+  local lbl = sl.label
+  if #lbl > max_lbl - 2 then
+    lbl = sl.label:sub(1, max_lbl - 3) .. "…"
+  end
+  terminal.write("\n\27[" .. x .. "C" .. ("%s%" .. max_w .. "s\27[0m  %9.4fs  %6s"):format(slices[#slices].color.bg, lbl, sl.time, ("%.1f%%"):format(sl.time / total * 100)))
+end
+
+if radius * 2 + 2 > #slices + #other then
+  terminal.write("\27[" .. radius * 2 + 2 .. "B")
+end
+terminal.write("\n")

halyde/apps/test.lua

@@ -0,0 +1,15 @@
+for i = 1, 100 do
+   profiler.start("math_work")
+   local x = 0
+   for j = 1, 200000 do x = x + math.sqrt(j) end
+   profiler.stop("math_work")
+
+   profiler.start("string_work")
+   local s = ""
+   for j = 1, 2000 do s = s .. tostring(j) end
+   profiler.stop("string_work")
+ end
+
+for _, r in ipairs(profiler.results()) do
+  print(r.label, r.time)
+end