fix markers on lines with variable aesthetics

src/marker.js

@@ -1,4 +1,6 @@
 import {create} from "./context.js";
+import {unset} from "./memoize.js";
+import {keyof} from "./options.js";
 
 export function markers(mark, {marker, markerStart = marker, markerMid = marker, markerEnd = marker} = {}) {
   mark.markerStart = maybeMarker(markerStart);
@@ -100,18 +102,56 @@ function markerTick(orient) {
 let nextMarkerId = 0;
 
 export function applyMarkers(path, mark, {stroke: S}, context) {
-  return applyMarkersColor(path, mark, S && ((i) => S[i]), context);
+  return applyMarkersColor(path, mark, S && ((i) => S[i]), null, context);
 }
 
-export function applyGroupedMarkers(path, mark, {stroke: S}, context) {
-  return applyMarkersColor(path, mark, S && (([i]) => S[i]), context);
+export function applyGroupedMarkers(path, mark, {stroke: S, z: Z}, context) {
+  return applyMarkersColor(path, mark, S && (([i]) => S[i]), Z, context);
 }
 
-function applyMarkersColor(path, {markerStart, markerMid, markerEnd, stroke}, strokeof = () => stroke, context) {
+const START = 1;
+const END = 2;
+
+/**
+ * When rendering lines or areas with variable aesthetics, a single series
+ * produces multiple path elements. The first path element is a START segment;
+ * the last path element is an END segment. When there is only a single path
+ * element, it is both a START and an END segment.
+ */
+function getGroupedOrientation(path, Z) {
+  const O = new Uint8Array(Z.length);
+  const D = path.data().filter((I) => I.length > 1);
+  const n = D.length;
+
+  // Forward pass to find start segments.
+  for (let i = 0, z = unset; i < n; ++i) {
+    const I = D[i];
+    if (I.length > 1) {
+      const i = I[0];
+      if (z !== (z = keyof(Z[i]))) O[i] |= START;
+    }
+  }
+
+  // Backwards pass to find end segments.
+  for (let i = n - 1, z = unset; i >= 0; --i) {
+    const I = D[i];
+    if (I.length > 1) {
+      const i = I[0];
+      if (z !== (z = keyof(Z[i]))) O[i] |= END;
+    }
+  }
+
+  return ([i]) => O[i];
+}
+
+function applyMarkersColor(path, {markerStart, markerMid, markerEnd, stroke}, strokeof = () => stroke, Z, context) {
+  if (!markerStart && !markerMid && !markerEnd) return;
   const iriByMarkerColor = new Map();
+  const orient = Z && getGroupedOrientation(path, Z);
 
-  function applyMarker(marker) {
+  function applyMarker(name, marker, filter) {
     return function (i) {
+      if (filter && !filter(i)) return;
       const color = strokeof(i);
       let iriByColor = iriByMarkerColor.get(marker);
       if (!iriByColor) iriByMarkerColor.set(marker, (iriByColor = new Map()));
@@ -122,11 +162,12 @@ function applyMarkersColor(path, {markerStart, markerMid, markerEnd, stroke}, st
         node.setAttribute("id", id);
         iriByColor.set(color, (iri = `url(#${id})`));
       }
-      return iri;
+      this.setAttribute(name, iri);
     };
   }
 
-  if (markerStart) path.attr("marker-start", applyMarker(markerStart));
-  if (markerMid) path.attr("marker-mid", applyMarker(markerMid));
-  if (markerEnd) path.attr("marker-end", applyMarker(markerEnd));
+  if (markerStart) path.each(applyMarker("marker-start", markerStart, orient && ((i) => orient(i) & START)));
+  if (markerMid && orient) path.each(applyMarker("marker-start", markerMid, (i) => !(orient(i) & START)));
+  if (markerMid) path.each(applyMarker("marker-mid", markerMid));
+  if (markerEnd) path.each(applyMarker("marker-end", markerEnd, orient && ((i) => orient(i) & END)));
 }

src/memoize.js

@@ -1,4 +1,4 @@
-const unset = Symbol("unset");
+export const unset = Symbol("unset");
 
 export function memoize1(compute) {
   return (compute.length === 1 ? memoize1Arg : memoize1Args)(compute);