initial import

tools/_micro_draw.py

@@ -0,0 +1,69 @@
+"""Micro-bench the per-char steps of VtScreen.draw."""
+import os
+import time
+
+import sys
+ST_LIB = os.path.expanduser("~/Library/Application Support/Sublime Text/Lib/python38")
+sys.path.insert(0, ST_LIB)
+sys.path.insert(0, "tests/stubs")
+from pyte.screens import wcwidth as pyte_wcwidth  # lru_cache'd
+from wcwidth import wcwidth as raw_wcwidth
+from functools import lru_cache
+
+data = ("x" * 50 + "\x1b[38;5;100m") * 40000  # ~1.2M printable chars
+
+
+def t(label, fn, n=3):
+    best = 1e9
+    for _ in range(n):
+        t0 = time.perf_counter()
+        fn()
+        best = min(best, time.perf_counter() - t0)
+    print("{:<42} {:8.3f} ms".format(label, best * 1000))
+
+
+def loop_wcwidth_pyte():
+    for c in data:
+        w = pyte_wcwidth(c)
+
+
+def loop_wcwidth_raw():
+    for c in data:
+        w = raw_wcwidth(c)
+
+
+def loop_iter_only():
+    for c in data:
+        pass
+
+
+def loop_min():
+    x = 0
+    for c in data:
+        x = min(x + 1, 80)
+
+
+def loop_dict_get_set():
+    d = {}
+    line = {}
+    for i, c in enumerate(data):
+        v = d.get(c)
+        if v is None:
+            d[c] = c
+        line[i % 80] = c
+
+
+def loop_min_vs_cmp():
+    x = 0
+    cols = 80
+    for c in data:
+        nx = x + 1
+        x = cols if nx > cols else nx
+
+
+t("iter only", loop_iter_only)
+t("wcwidth (pyte lru_cache)", loop_wcwidth_pyte)
+t("wcwidth (raw)", loop_wcwidth_raw)
+t("min(x+w, cols) per char", loop_min)
+t("min -> branch per char", loop_min_vs_cmp)
+t("dict get/miss/set per char", loop_dict_get_set)