heatmaps: part 6

website/scripts/utils/chart/index.js

@@ -1339,6 +1339,8 @@ export function createChart({ parent, brk, fitContent }) {
 
   /** @type {Record<number, ReturnType<typeof createPersistedValue<"lin" | "log">>>} */
   const scalePersistedValues = {};
+  /** @type {Record<number, HTMLElement | undefined>} */
+  const scaleSelectorElements = {};
 
   /**
    * @param {number} paneIndex
@@ -1366,8 +1368,7 @@ export function createChart({ parent, brk, fitContent }) {
     const td = tr?.querySelector("td:last-child");
     if (!td) return;
 
-    // Remove previous if any
-    td.querySelector(":scope > .field")?.remove();
+    scaleSelectorElements[paneIndex]?.remove();
 
     /** @type {HTMLTableCellElement} */ (td).style.position = "relative";
 
@@ -1381,6 +1382,7 @@ export function createChart({ parent, brk, fitContent }) {
       },
       toTitle: (c) => (c === "lin" ? "Linear scale" : "Logarithmic scale"),
     });
+    scaleSelectorElements[paneIndex] = radios;
     td.append(radios);
   }
 
@@ -1586,7 +1588,7 @@ export function createChart({ parent, brk, fitContent }) {
           choices,
           groups,
           id: "index",
-        }),
+        }).element,
       );
 
       for (const preset of getRangePresets()) {
@@ -1649,7 +1651,7 @@ export function createChart({ parent, brk, fitContent }) {
               blueprints.panes[paneIndex].unit = unit;
               blueprints.rebuildPane(paneIndex);
             },
-          }),
+          }).element,
         );
       });