@euphrasiologist/lwphylo 1.4.0 → 1.5.1

package/dist/drawPhylogeny.esm.js

@@ -816,6 +816,20 @@ function unrooted (node) {
   return data;
 }
 
+/**
+ * Reverse the child order at one node, flipping which side of the tree
+ * that clade's descendants are drawn on. Defaults to the root. Mutates
+ * the tree in place.
+ */
+function rotate(tree, nodeId = tree.id) {
+  const target = preorder(tree).find((n) => n.id === nodeId);
+  if (!target) {
+    throw new Error(`No node with id ${nodeId} found in tree`);
+  }
+  target.children.reverse();
+  return tree;
+}
+
 /**
  * Parse a Newick tree string into a doubly-linked list of JS Objects.
  * Assigns labels, branch lengths, and node IDs (tips before internals if input emits them that way).
@@ -925,9 +939,12 @@ function addScaleBar(svg, { scale, basis, defaultX, defaultY, scaleBar, fontSize
     .text(opts.label ?? String(length));
 }
 
-function drawPhylogeny(
-  treeText,
-  {
+// input: a Newick string, or an already-parsed tree (the node shape returned
+// by readTree()/randomTree()). Passing the same parsed tree object back in
+// across re-renders (e.g. after mutating it with rotate()) keeps node ids
+// stable, which onNodeClick below relies on.
+function drawPhylogeny(input, options = {}) {
+  const {
     layout = "rect", // rect/radial/unrooted
     width = 800,
     height = 800,
@@ -938,12 +955,14 @@ function drawPhylogeny(
     tipLabels = true,
     labelFontSize = 10, // font size (px) for tip labels
     tipRadius, // px radius of tip dots; defaults to each layout's original size
-    internalNodeCircles = false, // draw a circle at every internal (non-tip) node
     internalNodeRadius = 3, // px radius for internal node circles
     nodeLabels = false, // draw text labels at internal nodes (e.g. clade/support labels)
     nodeLabelFontSize, // defaults to labelFontSize
     scaleBar = false, // false | true | number (branch-length units) | { length, x, y, label }
     alignTipLabels = false, // rect & radial only: align tip labels to a common column/ring, with dashed guide lines back to the true tip position
+    container, // DOM element or CSS selector — if given, drawPhylogeny mounts the SVG itself instead of just returning it
+    rotateOnClick = false, // click an internal node circle to rotate its children in place and auto-redraw (requires container)
+    onNodeClick, // (node, event) => void — fires when an internal node circle is clicked (requires internalNodeCircles: true)
     showTooltips = true,
     tooltipFormatter = (d, rtt) =>
       `${d.thisLabel ?? "(unnamed)"}\nroot→tip: ${(+rtt).toFixed(4)}`,
@@ -952,12 +971,46 @@ function drawPhylogeny(
     highlightTips = [], // array of tip labels or ids for static highlight (optional)
     highlightStroke = "#e63946",
     highlightWidth = 2.5
-  } = {}
-) {
+  } = options;
+  // rotateOnClick needs internal node circles to click on; auto-enable them
+  // unless the caller explicitly said otherwise.
+  const internalNodeCircles = options.internalNodeCircles ?? rotateOnClick;
 
   // shared helpers
   const isNumber = (x) => typeof x === "number" && Number.isFinite(x);
   const nodeLabelSize = nodeLabelFontSize ?? labelFontSize;
+  const parsedTree = (input && typeof input === "object" && Array.isArray(input.children))
+    ? input
+    : readTree(input);
+
+  const mountEl = typeof container === "string"
+    ? (typeof document !== "undefined" ? document.querySelector(container) : null)
+    : (container ?? null);
+
+  if (rotateOnClick && !mountEl) {
+    throw new Error(
+      "rotateOnClick requires a `container` (a DOM element or CSS selector) so drawPhylogeny can mount the redrawn SVG after each rotation."
+    );
+  }
+
+  // fires the caller's onNodeClick (if any), then handles the rotate+redraw itself
+  const effectiveOnNodeClick = (onNodeClick || rotateOnClick)
+    ? (node, event) => {
+      if (typeof onNodeClick === "function") onNodeClick(node, event);
+      if (rotateOnClick) {
+        rotate(parsedTree, node.thisId);
+        drawPhylogeny(parsedTree, options);
+      }
+    }
+    : undefined;
+
+  function mount(svgNode) {
+    if (mountEl) {
+      mountEl.innerHTML = "";
+      mountEl.appendChild(svgNode);
+    }
+    return svgNode;
+  }
   // Works for both radial (uses `r`) and rect (uses `x1`).
   // Falls back to summing branchLength up to the root if neither is present.
   function makeRootToTipGetter(byId, { prefer = "auto" } = {}) {
@@ -986,7 +1039,7 @@ function drawPhylogeny(
 
   if (layout === "rect") {
     // RECTANGULAR LAYOUT
-    const tree_df = rectangleLayout(readTree(treeText));
+    const tree_df = rectangleLayout(parsedTree);
     const horizontal = tree_df.horizontal_lines;
     const vertical = tree_df.vertical_lines;
     const tips = horizontal.filter((d) => d.isTip);
@@ -1097,6 +1150,12 @@ function drawPhylogeny(
           .append("title")
           .text((d) => tooltipFormatter(d, rootToTip(d.thisId)));
       }
+
+      if (effectiveOnNodeClick) {
+        internalDots
+          .style("cursor", "pointer")
+          .on("click", (event, d) => effectiveOnNodeClick(d, event));
+      }
     }
 
     // internal node labels (optional)
@@ -1226,13 +1285,12 @@ function drawPhylogeny(
       });
     }
 
-    return svg.node();
+    return mount(svg.node());
   } else if (layout === "radial") {
     // RADIAL LAYOUT
     if (width !== height) {
       throw new Error("width and height must be the same for radial layout");
     }
-    const parsedTree = readTree(treeText);
     const rad = radialLayout(parsedTree, {
       angleStrategy: "fan",
       arcsStyle: "fan"
@@ -1452,6 +1510,12 @@ function drawPhylogeny(
           .append("title")
           .text((d) => tooltipFormatter(d, rootToTip(d.thisId)));
       }
+
+      if (effectiveOnNodeClick) {
+        internalDots
+          .style("cursor", "pointer")
+          .on("click", (event, d) => effectiveOnNodeClick(d, event));
+      }
     }
 
     // ===== INTERNAL NODE LABELS (optional) =====
@@ -1649,10 +1713,9 @@ function drawPhylogeny(
       });
     }
 
-    return svg.node();
+    return mount(svg.node());
   } else if (layout === "unrooted") {
     // UNROOTED LAYOUT
-    const parsedTree = readTree(treeText);
     const unrootedPhylo = unrooted(parsedTree);
 
     const w = width;
@@ -1714,6 +1777,13 @@ function drawPhylogeny(
       .attr("stroke-width", 2)
       .attr("fill", (d) => (d.isTip ? "black" : "white"));
 
+    if (internalNodeCircles && effectiveOnNodeClick) {
+      nodes
+        .filter((d) => !d.isTip)
+        .style("cursor", "pointer")
+        .on("click", (event, d) => effectiveOnNodeClick(d, event));
+    }
+
     if (nodeLabels) {
       const labeledInternalNodes = unrootedPhylo.data.filter((d) => !d.isTip && d.thisLabel);
       group
@@ -1876,7 +1946,7 @@ function drawPhylogeny(
       });
     }
 
-    return svg.node();
+    return mount(svg.node());
   } else {
     throw new Error(
       "Unsupported layout type. Use 'rect', 'radial', or 'unrooted'."