@yassimba/pi-loom-mermaid 0.2.0

package/src/loom-mermaid/width.ts

@@ -0,0 +1,69 @@
+import { WIDTHS } from './width-data.ts'
+
+/**
+ * Display width, measured in grapheme clusters.
+ *
+ * A cluster is the unit both of measuring and of painting, so a box is always
+ * sized for exactly what gets drawn into it. Splitting those two — sizing by
+ * cluster but painting by code point — is what makes `👨‍👩‍👧` overflow its
+ * border in the Rust original.
+ *
+ * Clustering comes from `Intl.Segmenter` (UAX #29), which already handles ZWJ
+ * sequences, skin-tone modifiers, variation selectors, keycaps, flags and
+ * Hangul. Per-code-point widths are generated from the `unicode-width` crate.
+ */
+
+const segmenter = new Intl.Segmenter('en', { granularity: 'grapheme' })
+
+const VS16 = 0xfe0f
+const isRegionalIndicator = (cp: number): boolean => cp >= 0x1f1e6 && cp <= 0x1f1ff
+
+/** Width of one code point; the table covers the whole code point space. */
+function codePointWidth(cp: number): number {
+  let lo = 0
+  let hi = WIDTHS.length - 1
+  while (lo <= hi) {
+    const mid = (lo + hi) >> 1
+    const run = WIDTHS[mid]
+    if (cp < run[0]) hi = mid - 1
+    else if (cp > run[1]) lo = mid + 1
+    else return run[2]
+  }
+  return 1
+}
+
+/**
+ * Columns occupied by one grapheme cluster.
+ *
+ * The widest code point wins, so a base plus its combining marks measures as
+ * the base. Two adjustments: a variation selector requesting emoji
+ * presentation forces two columns, as does a regional indicator pair (a flag).
+ *
+ * Zero is a real answer — a soft hyphen or zero-width space occupies nothing,
+ * and callers skip painting such a cluster rather than reserving a cell.
+ */
+function clusterWidth(cluster: string): number {
+  let w = 0
+  let vs16 = false
+  let regional = 0
+  for (const ch of cluster) {
+    const cp = ch.codePointAt(0) as number
+    if (cp === VS16) vs16 = true
+    if (isRegionalIndicator(cp)) regional++
+    const cw = codePointWidth(cp)
+    if (cw > w) w = cw
+  }
+  return vs16 || regional >= 2 ? 2 : w
+}
+
+/** Iterate clusters paired with their display width. */
+export function* measured(s: string): Generator<[string, number]> {
+  for (const { segment } of segmenter.segment(s)) yield [segment, clusterWidth(segment)]
+}
+
+/** Display columns of a string. */
+export function stringWidth(s: string): number {
+  let w = 0
+  for (const { segment } of segmenter.segment(s)) w += clusterWidth(segment)
+  return w
+}