@ggterm/core 0.2.10 → 0.2.11

package/dist/cli-plot.js

@@ -2309,6 +2309,58 @@ function parseColor(color) {
   }
   return { r: 128, g: 128, b: 128, a: 1 };
 }
+function renderGeomBeeswarm(data, geom, aes, scales, canvas) {
+  const alpha = geom.params.alpha ?? 1;
+  const fixedColor = geom.params.color;
+  const shape = getPointShape(geom.params.shape);
+  const plotLeft = Math.round(scales.x.range[0]);
+  const plotRight = Math.round(scales.x.range[1]);
+  const plotTop = Math.round(Math.min(scales.y.range[0], scales.y.range[1]));
+  const plotBottom = Math.round(Math.max(scales.y.range[0], scales.y.range[1]));
+  const defaultColors = [
+    { r: 79, g: 169, b: 238, a: 1 },
+    { r: 238, g: 136, b: 102, a: 1 },
+    { r: 102, g: 204, b: 153, a: 1 },
+    { r: 204, g: 102, b: 204, a: 1 },
+    { r: 255, g: 200, b: 87, a: 1 },
+    { r: 138, g: 201, b: 222, a: 1 },
+    { r: 255, g: 153, b: 153, a: 1 },
+    { r: 170, g: 170, b: 170, a: 1 }
+  ];
+  const categories = new Set;
+  for (const row of data) {
+    categories.add(String(row.xOriginal ?? row[aes.x] ?? "default"));
+  }
+  const categoryList = [...categories];
+  for (const row of data) {
+    const xVal = row.x;
+    const yVal = row.y;
+    if (xVal === null || xVal === undefined || yVal === null || yVal === undefined) {
+      continue;
+    }
+    const numGroups = categoryList.length;
+    const xRange = plotRight - plotLeft;
+    const xNormalized = (Number(xVal) + 0.5) / numGroups;
+    const cx = Math.round(plotLeft + xNormalized * xRange);
+    const cy = Math.round(scales.y.map(yVal));
+    let color;
+    if (fixedColor) {
+      color = parseColorToRgba(fixedColor);
+    } else if (scales.color && aes.color) {
+      color = getPointColor(row, aes, scales.color);
+    } else {
+      const category = String(row.xOriginal ?? row[aes.x] ?? "default");
+      const categoryIdx = categoryList.indexOf(category);
+      color = defaultColors[categoryIdx % defaultColors.length];
+    }
+    if (alpha < 1) {
+      color = { ...color, a: alpha };
+    }
+    if (cx >= plotLeft && cx <= plotRight && cy >= plotTop && cy <= plotBottom) {
+      canvas.drawChar(cx, cy, shape, color);
+    }
+  }
+}
 function renderGeom(data, geom, aes, scales, canvas, coordType) {
   switch (geom.type) {
     case "point":
@@ -2390,6 +2442,10 @@ function renderGeom(data, geom, aes, scales, canvas, coordType) {
     case "smooth":
       renderGeomSmooth(data, geom, aes, scales, canvas);
       break;
+    case "beeswarm":
+    case "quasirandom":
+      renderGeomBeeswarm(data, geom, aes, scales, canvas);
+      break;
     default:
       break;
   }
@@ -3517,6 +3573,224 @@ function stat_xdensity(params = {}) {
   };
 }
 
+// src/stats/beeswarm.ts
+function swarmArrange(yValues, params) {
+  const n = yValues.length;
+  if (n === 0)
+    return { offsets: [], indices: [] };
+  const cex = params.cex ?? 1;
+  const spacing = params.spacing ?? 1;
+  const side = params.side ?? 0;
+  const yRange = Math.max(...yValues) - Math.min(...yValues);
+  const pointSize = yRange / Math.max(n, 10) * cex * spacing;
+  let indices = yValues.map((_, i) => i);
+  const priority = params.priority ?? "ascending";
+  switch (priority) {
+    case "ascending":
+      indices.sort((a, b) => yValues[a] - yValues[b]);
+      break;
+    case "descending":
+      indices.sort((a, b) => yValues[b] - yValues[a]);
+      break;
+    case "random":
+      for (let i = indices.length - 1;i > 0; i--) {
+        const j = Math.floor(Math.random() * (i + 1));
+        [indices[i], indices[j]] = [indices[j], indices[i]];
+      }
+      break;
+    case "density":
+      const median = yValues.slice().sort((a, b) => a - b)[Math.floor(n / 2)];
+      indices.sort((a, b) => Math.abs(yValues[a] - median) - Math.abs(yValues[b] - median));
+      break;
+  }
+  const placed = [];
+  const offsets = new Array(n).fill(0);
+  for (const idx of indices) {
+    const y = yValues[idx];
+    let bestOffset = 0;
+    if (placed.length > 0) {
+      const nearby = placed.filter((p) => Math.abs(p.y - y) < pointSize * 2);
+      if (nearby.length > 0) {
+        const maxOffset = nearby.length * pointSize;
+        let foundSpot = false;
+        for (let tryOffset = 0;tryOffset <= maxOffset && !foundSpot; tryOffset += pointSize * 0.5) {
+          if (side >= 0) {
+            let collision = false;
+            for (const p of nearby) {
+              const dx = tryOffset - p.x;
+              const dy = y - p.y;
+              const dist = Math.sqrt(dx * dx + dy * dy);
+              if (dist < pointSize) {
+                collision = true;
+                break;
+              }
+            }
+            if (!collision) {
+              bestOffset = tryOffset;
+              foundSpot = true;
+              break;
+            }
+          }
+          if (side <= 0 && tryOffset > 0 && !foundSpot) {
+            let collision = false;
+            for (const p of nearby) {
+              const dx = -tryOffset - p.x;
+              const dy = y - p.y;
+              const dist = Math.sqrt(dx * dx + dy * dy);
+              if (dist < pointSize) {
+                collision = true;
+                break;
+              }
+            }
+            if (!collision) {
+              bestOffset = -tryOffset;
+              foundSpot = true;
+              break;
+            }
+          }
+        }
+      }
+    }
+    offsets[idx] = bestOffset;
+    placed.push({ y, x: bestOffset });
+  }
+  const maxAbsOffset = Math.max(...offsets.map(Math.abs), 0.001);
+  const dodge = params.dodge ?? 0.8;
+  const scale = dodge * 0.5 / maxAbsOffset;
+  for (let i = 0;i < offsets.length; i++) {
+    offsets[i] *= scale;
+  }
+  return { offsets, indices };
+}
+function centerArrange(yValues, params) {
+  const n = yValues.length;
+  if (n === 0)
+    return { offsets: [], indices: [] };
+  const dodge = params.dodge ?? 0.8;
+  const side = params.side ?? 0;
+  const indices = yValues.map((_, i) => i);
+  indices.sort((a, b) => yValues[a] - yValues[b]);
+  const offsets = new Array(n).fill(0);
+  const maxOffset = dodge * 0.4;
+  for (let i = 0;i < indices.length; i++) {
+    const idx = indices[i];
+    const layer = Math.floor(i / 2) + 1;
+    const offset = layer / Math.ceil(n / 2) * maxOffset;
+    if (side === 0) {
+      offsets[idx] = i % 2 === 0 ? offset : -offset;
+    } else {
+      offsets[idx] = side * offset;
+    }
+  }
+  return { offsets, indices };
+}
+function squareArrange(yValues, params) {
+  const n = yValues.length;
+  if (n === 0)
+    return { offsets: [], indices: [] };
+  const dodge = params.dodge ?? 0.8;
+  const side = params.side ?? 0;
+  const indices = yValues.map((_, i) => i);
+  indices.sort((a, b) => yValues[a] - yValues[b]);
+  const offsets = new Array(n).fill(0);
+  const yRange = Math.max(...yValues) - Math.min(...yValues);
+  const binSize = yRange / Math.max(Math.sqrt(n), 3);
+  const bins = new Map;
+  for (let i = 0;i < indices.length; i++) {
+    const idx = indices[i];
+    const y = yValues[idx];
+    const binKey = Math.floor(y / binSize);
+    if (!bins.has(binKey)) {
+      bins.set(binKey, []);
+    }
+    bins.get(binKey).push(idx);
+  }
+  for (const binIndices of bins.values()) {
+    const binN = binIndices.length;
+    const maxOffset = dodge * 0.4;
+    for (let i = 0;i < binN; i++) {
+      const idx = binIndices[i];
+      const offset = (i - (binN - 1) / 2) * (maxOffset * 2 / Math.max(binN - 1, 1));
+      if (side === 0) {
+        offsets[idx] = offset;
+      } else if (side > 0) {
+        offsets[idx] = Math.abs(offset);
+      } else {
+        offsets[idx] = -Math.abs(offset);
+      }
+    }
+  }
+  return { offsets, indices };
+}
+function computeBeeswarm(data, _xField, yField, groupKey, groupIndex, params = {}) {
+  const yValues = [];
+  const originalRows = [];
+  for (const row of data) {
+    const yVal = row[yField];
+    if (yVal === null || yVal === undefined)
+      continue;
+    const numY = Number(yVal);
+    if (isNaN(numY))
+      continue;
+    yValues.push(numY);
+    originalRows.push(row);
+  }
+  if (yValues.length === 0)
+    return [];
+  const method = params.method ?? "swarm";
+  let result;
+  switch (method) {
+    case "center":
+      result = centerArrange(yValues, params);
+      break;
+    case "square":
+      result = squareArrange(yValues, params);
+      break;
+    case "swarm":
+    default:
+      result = swarmArrange(yValues, params);
+  }
+  const output = [];
+  for (let i = 0;i < yValues.length; i++) {
+    const originalRow = originalRows[i];
+    output.push({
+      x: groupIndex + result.offsets[i],
+      y: yValues[i],
+      xOriginal: groupKey,
+      yOriginal: yValues[i],
+      xOffset: result.offsets[i],
+      ...originalRow
+    });
+  }
+  return output;
+}
+function stat_beeswarm(params = {}) {
+  return {
+    type: "beeswarm",
+    compute(data, aes) {
+      const groups = new Map;
+      const groupOrder = [];
+      for (const row of data) {
+        const groupKey = String(row[aes.x] ?? "default");
+        if (!groups.has(groupKey)) {
+          groups.set(groupKey, []);
+          groupOrder.push(groupKey);
+        }
+        groups.get(groupKey).push(row);
+      }
+      const result = [];
+      let groupIndex = 0;
+      for (const groupKey of groupOrder) {
+        const groupData = groups.get(groupKey);
+        const swarmResult = computeBeeswarm(groupData, aes.x, aes.y, groupKey, groupIndex, params);
+        result.push(...swarmResult);
+        groupIndex++;
+      }
+      return result;
+    }
+  };
+}
+
 // src/stats/smooth.ts
 function linearRegression(xs, ys) {
   const n = xs.length;
@@ -4396,6 +4670,15 @@ function applyStatTransform(data, geom, aes) {
       adjust: geom.params.adjust
     });
     return xdensityStat.compute(data, aes);
+  } else if (geom.stat === "beeswarm") {
+    const beeswarmStat = stat_beeswarm({
+      method: geom.params.method,
+      cex: geom.params.cex,
+      side: geom.params.side,
+      priority: geom.params.priority,
+      dodge: geom.params.dodge
+    });
+    return beeswarmStat.compute(data, aes);
   } else if (geom.stat === "smooth") {
     const smoothStat = stat_smooth({
       method: geom.params.method,
@@ -4522,6 +4805,10 @@ function renderToCanvas(spec, options) {
       scaleData = applyStatTransform(spec.data, geom, spec.aes);
       scaleAes = { ...spec.aes, x: "x", y: "y" };
       break;
+    } else if (geom.stat === "beeswarm") {
+      scaleData = spec.data;
+      scaleAes = spec.aes;
+      break;
     }
   }
   scaleData = applyCoordTransform(scaleData, scaleAes, spec.coord);
@@ -4562,6 +4849,8 @@ function renderToCanvas(spec, options) {
         geomAes = { ...spec.aes, x: "x", y: "y" };
       } else if (geom.stat === "xdensity") {
         geomAes = { ...spec.aes, x: "x", y: "y" };
+      } else if (geom.stat === "beeswarm") {
+        geomAes = { ...spec.aes, x: "x", y: "y" };
       }
       geomData = applyCoordTransform(geomData, geomAes, spec.coord);
     }
@@ -5506,6 +5795,29 @@ var init_ridgeline = __esm(() => {
   geom_joy = geom_ridgeline;
 });
 
+// src/geoms/beeswarm.ts
+function geom_beeswarm(options = {}) {
+  return {
+    type: "beeswarm",
+    stat: "beeswarm",
+    position: "identity",
+    params: {
+      method: options.method ?? "swarm",
+      size: options.size ?? 1,
+      cex: options.cex ?? 1,
+      alpha: options.alpha ?? 1,
+      color: options.color,
+      shape: options.shape ?? "circle",
+      side: options.side ?? 0,
+      priority: options.priority ?? "ascending",
+      dodge: options.dodge ?? 0.8
+    }
+  };
+}
+function geom_quasirandom(options = {}) {
+  return geom_beeswarm({ ...options, method: "center" });
+}
+
 // src/geoms/index.ts
 var init_geoms = __esm(() => {
   init_ridgeline();
@@ -9914,6 +10226,7 @@ __export(exports_src, {
   stat_boxplot: () => stat_boxplot,
   stat_bin2d: () => stat_bin2d,
   stat_bin: () => stat_bin,
+  stat_beeswarm: () => stat_beeswarm,
   startREPL: () => startREPL,
   selectRenderer: () => selectRenderer,
   selectColorMode: () => selectColorMode,
@@ -10022,6 +10335,7 @@ __export(exports_src, {
   geom_ribbon: () => geom_ribbon,
   geom_rect: () => geom_rect,
   geom_raster: () => geom_raster,
+  geom_quasirandom: () => geom_quasirandom,
   geom_qq_line: () => geom_qq_line,
   geom_qq: () => geom_qq,
   geom_pointrange: () => geom_pointrange,
@@ -10044,6 +10358,7 @@ __export(exports_src, {
   geom_col: () => geom_col,
   geom_boxplot: () => geom_boxplot,
   geom_bin2d: () => geom_bin2d,
+  geom_beeswarm: () => geom_beeswarm,
   geom_bar: () => geom_bar,
   geom_area: () => geom_area,
   geom_abline: () => geom_abline,
@@ -10090,6 +10405,7 @@ __export(exports_src, {
   computeBoxplotStats: () => computeBoxplotStats,
   computeBins2d: () => computeBins2d,
   computeBins: () => computeBins,
+  computeBeeswarm: () => computeBeeswarm,
   colorDistance: () => colorDistance,
   clearCapabilityCache: () => clearCapabilityCache,
   calculatePanelLayouts: () => calculatePanelLayouts,
@@ -10941,6 +11257,8 @@ var GEOM_TYPES = [
   "violin",
   "ridgeline",
   "joy",
+  "beeswarm",
+  "quasirandom",
   "area",
   "ribbon",
   "rug",
@@ -11335,7 +11653,7 @@ Error: Unknown geometry type "${geomType}"`);
 Available geom types:`);
     console.error(`  Points/Lines: point, line, path, step, smooth, segment`);
     console.error(`  Bars/Areas:   bar, col, histogram, freqpoly, area, ribbon`);
-    console.error(`  Distributions: boxplot, violin, ridgeline, joy, qq, density_2d`);
+    console.error(`  Distributions: boxplot, violin, ridgeline, joy, beeswarm, quasirandom, qq, density_2d`);
     console.error(`  Uncertainty:  errorbar, errorbarh, crossbar, linerange, pointrange`);
     console.error(`  2D:           tile, rect, raster, bin2d, contour, contour_filled`);
     console.error(`  Text:         text, label`);
@@ -11484,6 +11802,10 @@ If you want a univariate plot, try: histogram, bar, qq, or freqpoly`);
     case "joy":
       plot = plot.geom(geom_ridgeline());
       break;
+    case "beeswarm":
+    case "quasirandom":
+      plot = plot.geom(geom_beeswarm());
+      break;
     case "bar":
       plot = plot.geom(geom_bar());
       break;