From 6030046a07c83b7b19f4765228f830bdc4b5c479 Mon Sep 17 00:00:00 2001
From: baeltaezaer <info@convierteweb.com.mx>
Date: Sun, 5 Jul 2026 14:38:24 -0600
Subject: [PATCH 1/2] fix(schematic-viewer): highlight nets on hover

---
 lib/components/SchematicViewer.tsx | 111 +++++++++++++++++++++++++++++
 1 file changed, 111 insertions(+)

diff --git a/lib/components/SchematicViewer.tsx b/lib/components/SchematicViewer.tsx
index 865a195..cd2e5b2 100644
--- a/lib/components/SchematicViewer.tsx
+++ b/lib/components/SchematicViewer.tsx
@@ -66,6 +66,21 @@ interface Props {
   onSchematicSheetChange?: (schematicSheetId: string) => void
 }
 
+const escapeCssAttributeValue = (value: string) =>
+  value.replace(/\\/g, "\\\\").replace(/"/g, '\\"')
+
+const buildTraceHoverStyles = (traceConnectivityKeys: string[]) =>
+  traceConnectivityKeys
+    .map((key) => {
+      const escapedKey = escapeCssAttributeValue(key)
+      const hoveredTraceSelector = `:is(g.trace[data-subcircuit-connectivity-map-key="${escapedKey}"], g.trace-overlays[data-subcircuit-connectivity-map-key="${escapedKey}"])`
+      return [
+        `svg:has(${hoveredTraceSelector}:hover) ${hoveredTraceSelector} { filter: invert(1); }`,
+        `svg:has(${hoveredTraceSelector}:hover) g.trace-overlays[data-subcircuit-connectivity-map-key="${escapedKey}"] .trace-crossing-outline { opacity: 0; }`,
+      ].join("\n")
+    })
+    .join("\n")
+
 export const SchematicViewer = ({
   circuitJson,
   containerStyle,
@@ -222,14 +237,19 @@ export const SchematicViewer = ({
 
   const [isHoveringClickablePort, setIsHoveringClickablePort] = useState(false)
   const hoveringPortsRef = useRef<Set<string>>(new Set())
+  const [hoveredPortsVersion, setHoveredPortsVersion] = useState(0)
 
   const handlePortHoverChange = useCallback(
     (portId: string, isHovering: boolean) => {
+      const hadPort = hoveringPortsRef.current.has(portId)
       if (isHovering) {
         hoveringPortsRef.current.add(portId)
       } else {
         hoveringPortsRef.current.delete(portId)
       }
+      if (hadPort !== hoveringPortsRef.current.has(portId)) {
+        setHoveredPortsVersion((version) => version + 1)
+      }
       setIsHoveringClickablePort(hoveringPortsRef.current.size > 0)
     },
     [],
@@ -281,6 +301,96 @@ export const SchematicViewer = ({
     }
   }, [circuitJsonKey, circuitJson, showSchematicPorts, activeSheetId])
 
+  const sourcePortConnectivityMap = useMemo(() => {
+    try {
+      const map = new Map<string, Set<string>>()
+      const sourceTraces = su(circuitJson).source_trace?.list() ?? []
+
+      for (const sourceTrace of sourceTraces) {
+        const connectivityKey = (sourceTrace as any)
+          .subcircuit_connectivity_map_key as string | undefined
+        if (!connectivityKey) continue
+
+        for (const sourcePortId of sourceTrace.connected_source_port_ids ??
+          []) {
+          const existingKeys = map.get(sourcePortId)
+          if (existingKeys) {
+            existingKeys.add(connectivityKey)
+          } else {
+            map.set(sourcePortId, new Set([connectivityKey]))
+          }
+        }
+      }
+
+      return map
+    } catch (err) {
+      console.error("Failed to derive source port connectivity map", err)
+      return new Map<string, Set<string>>()
+    }
+  }, [circuitJsonKey, circuitJson])
+
+  const traceConnectivityKeys = useMemo(() => {
+    try {
+      const keys = new Set<string>()
+      const sourceTraces = su(circuitJson).source_trace?.list() ?? []
+
+      for (const sourceTrace of sourceTraces) {
+        const connectivityKey = (sourceTrace as any)
+          .subcircuit_connectivity_map_key as string | undefined
+        if (connectivityKey) {
+          keys.add(connectivityKey)
+        }
+      }
+
+      return [...keys]
+    } catch (err) {
+      console.error("Failed to derive trace connectivity keys", err)
+      return []
+    }
+  }, [circuitJsonKey, circuitJson])
+
+  const traceHoverStyles = useMemo(
+    () => buildTraceHoverStyles(traceConnectivityKeys),
+    [traceConnectivityKeys],
+  )
+
+  const hoveredPortConnectivityKeys = useMemo(() => {
+    const keys = new Set<string>()
+    for (const portId of hoveringPortsRef.current) {
+      const portKeys = sourcePortConnectivityMap.get(portId)
+      if (!portKeys) continue
+      for (const key of portKeys) {
+        keys.add(key)
+      }
+    }
+    return [...keys]
+  }, [hoveredPortsVersion, sourcePortConnectivityMap])
+
+  useEffect(() => {
+    const svgDiv = svgDivRef.current
+    if (!svgDiv) return
+
+    const activeTraceElements = svgDiv.querySelectorAll<HTMLElement>(
+      '[data-net-hover-active="true"]',
+    )
+    for (const element of Array.from(activeTraceElements)) {
+      element.removeAttribute("data-net-hover-active")
+    }
+
+    if (hoveredPortConnectivityKeys.length === 0) {
+      return
+    }
+
+    for (const connectivityKey of hoveredPortConnectivityKeys) {
+      const escapedKey = escapeCssAttributeValue(connectivityKey)
+      const selector = `:is(g.trace[data-subcircuit-connectivity-map-key="${escapedKey}"], g.trace-overlays[data-subcircuit-connectivity-map-key="${escapedKey}"])`
+      const matchingElements = svgDiv.querySelectorAll<HTMLElement>(selector)
+      for (const element of Array.from(matchingElements)) {
+        element.setAttribute("data-net-hover-active", "true")
+      }
+    }
+  }, [hoveredPortConnectivityKeys, circuitJsonKey])
+
   const handleTouchStart = (e: React.TouchEvent) => {
     const touch = e.touches[0]
     touchStartRef.current = {
@@ -476,6 +586,7 @@ export const SchematicViewer = ({
 
   return (
     <MouseTracker>
+      {traceHoverStyles && <style>{traceHoverStyles}</style>}
       {onSchematicComponentClicked && (
         <style>
           {`.schematic-component-clickable [data-schematic-component-id]:hover { cursor: pointer !important; }`}

From 4818a06eb0c92e98f17c7d5e189f7159dbc6e297 Mon Sep 17 00:00:00 2001
From: baeltaezaer <info@convierteweb.com.mx>
Date: Sun, 5 Jul 2026 14:48:32 -0600
Subject: [PATCH 2/2] fix(schematic-viewer): highlight same-net traces

---
 lib/components/SchematicViewer.tsx            | 119 +-------------
 lib/hooks/useTraceHoverHighlighting.ts        |  78 ++++++++++
 .../get-schematic-trace-net-keys.test.ts      | 145 ++++++++++++++++++
 lib/utils/get-schematic-trace-net-keys.ts     | 114 ++++++++++++++
 4 files changed, 345 insertions(+), 111 deletions(-)
 create mode 100644 lib/hooks/useTraceHoverHighlighting.ts
 create mode 100644 lib/utils/get-schematic-trace-net-keys.test.ts
 create mode 100644 lib/utils/get-schematic-trace-net-keys.ts

diff --git a/lib/components/SchematicViewer.tsx b/lib/components/SchematicViewer.tsx
index cd2e5b2..b6520c6 100644
--- a/lib/components/SchematicViewer.tsx
+++ b/lib/components/SchematicViewer.tsx
@@ -25,6 +25,7 @@ import { SpiceSimulationIcon } from "./SpiceSimulationIcon"
 import { SpiceSimulationOverlay } from "./SpiceSimulationOverlay"
 import { zIndexMap } from "../utils/z-index-map"
 import { useSpiceSimulation } from "../hooks/useSpiceSimulation"
+import { useTraceHoverHighlighting } from "../hooks/useTraceHoverHighlighting"
 import { getSpiceFromCircuitJson } from "../utils/spice-utils"
 import {
   getStoredBoolean,
@@ -66,21 +67,6 @@ interface Props {
   onSchematicSheetChange?: (schematicSheetId: string) => void
 }
 
-const escapeCssAttributeValue = (value: string) =>
-  value.replace(/\\/g, "\\\\").replace(/"/g, '\\"')
-
-const buildTraceHoverStyles = (traceConnectivityKeys: string[]) =>
-  traceConnectivityKeys
-    .map((key) => {
-      const escapedKey = escapeCssAttributeValue(key)
-      const hoveredTraceSelector = `:is(g.trace[data-subcircuit-connectivity-map-key="${escapedKey}"], g.trace-overlays[data-subcircuit-connectivity-map-key="${escapedKey}"])`
-      return [
-        `svg:has(${hoveredTraceSelector}:hover) ${hoveredTraceSelector} { filter: invert(1); }`,
-        `svg:has(${hoveredTraceSelector}:hover) g.trace-overlays[data-subcircuit-connectivity-map-key="${escapedKey}"] .trace-crossing-outline { opacity: 0; }`,
-      ].join("\n")
-    })
-    .join("\n")
-
 export const SchematicViewer = ({
   circuitJson,
   containerStyle,
@@ -237,19 +223,14 @@ export const SchematicViewer = ({
 
   const [isHoveringClickablePort, setIsHoveringClickablePort] = useState(false)
   const hoveringPortsRef = useRef<Set<string>>(new Set())
-  const [hoveredPortsVersion, setHoveredPortsVersion] = useState(0)
 
   const handlePortHoverChange = useCallback(
     (portId: string, isHovering: boolean) => {
-      const hadPort = hoveringPortsRef.current.has(portId)
       if (isHovering) {
         hoveringPortsRef.current.add(portId)
       } else {
         hoveringPortsRef.current.delete(portId)
       }
-      if (hadPort !== hoveringPortsRef.current.has(portId)) {
-        setHoveredPortsVersion((version) => version + 1)
-      }
       setIsHoveringClickablePort(hoveringPortsRef.current.size > 0)
     },
     [],
@@ -301,96 +282,6 @@ export const SchematicViewer = ({
     }
   }, [circuitJsonKey, circuitJson, showSchematicPorts, activeSheetId])
 
-  const sourcePortConnectivityMap = useMemo(() => {
-    try {
-      const map = new Map<string, Set<string>>()
-      const sourceTraces = su(circuitJson).source_trace?.list() ?? []
-
-      for (const sourceTrace of sourceTraces) {
-        const connectivityKey = (sourceTrace as any)
-          .subcircuit_connectivity_map_key as string | undefined
-        if (!connectivityKey) continue
-
-        for (const sourcePortId of sourceTrace.connected_source_port_ids ??
-          []) {
-          const existingKeys = map.get(sourcePortId)
-          if (existingKeys) {
-            existingKeys.add(connectivityKey)
-          } else {
-            map.set(sourcePortId, new Set([connectivityKey]))
-          }
-        }
-      }
-
-      return map
-    } catch (err) {
-      console.error("Failed to derive source port connectivity map", err)
-      return new Map<string, Set<string>>()
-    }
-  }, [circuitJsonKey, circuitJson])
-
-  const traceConnectivityKeys = useMemo(() => {
-    try {
-      const keys = new Set<string>()
-      const sourceTraces = su(circuitJson).source_trace?.list() ?? []
-
-      for (const sourceTrace of sourceTraces) {
-        const connectivityKey = (sourceTrace as any)
-          .subcircuit_connectivity_map_key as string | undefined
-        if (connectivityKey) {
-          keys.add(connectivityKey)
-        }
-      }
-
-      return [...keys]
-    } catch (err) {
-      console.error("Failed to derive trace connectivity keys", err)
-      return []
-    }
-  }, [circuitJsonKey, circuitJson])
-
-  const traceHoverStyles = useMemo(
-    () => buildTraceHoverStyles(traceConnectivityKeys),
-    [traceConnectivityKeys],
-  )
-
-  const hoveredPortConnectivityKeys = useMemo(() => {
-    const keys = new Set<string>()
-    for (const portId of hoveringPortsRef.current) {
-      const portKeys = sourcePortConnectivityMap.get(portId)
-      if (!portKeys) continue
-      for (const key of portKeys) {
-        keys.add(key)
-      }
-    }
-    return [...keys]
-  }, [hoveredPortsVersion, sourcePortConnectivityMap])
-
-  useEffect(() => {
-    const svgDiv = svgDivRef.current
-    if (!svgDiv) return
-
-    const activeTraceElements = svgDiv.querySelectorAll<HTMLElement>(
-      '[data-net-hover-active="true"]',
-    )
-    for (const element of Array.from(activeTraceElements)) {
-      element.removeAttribute("data-net-hover-active")
-    }
-
-    if (hoveredPortConnectivityKeys.length === 0) {
-      return
-    }
-
-    for (const connectivityKey of hoveredPortConnectivityKeys) {
-      const escapedKey = escapeCssAttributeValue(connectivityKey)
-      const selector = `:is(g.trace[data-subcircuit-connectivity-map-key="${escapedKey}"], g.trace-overlays[data-subcircuit-connectivity-map-key="${escapedKey}"])`
-      const matchingElements = svgDiv.querySelectorAll<HTMLElement>(selector)
-      for (const element of Array.from(matchingElements)) {
-        element.setAttribute("data-net-hover-active", "true")
-      }
-    }
-  }, [hoveredPortConnectivityKeys, circuitJsonKey])
-
   const handleTouchStart = (e: React.TouchEvent) => {
     const touch = e.touches[0]
     touchStartRef.current = {
@@ -534,6 +425,13 @@ export const SchematicViewer = ({
     editEvents: editEventsWithUnappliedEditEvents,
   })
 
+  useTraceHoverHighlighting({
+    svgDivRef,
+    circuitJson,
+    circuitJsonKey,
+    enabled: !editModeEnabled,
+  })
+
   // Add group overlays when enabled. The key includes the active sheet so
   // overlays are recomputed against the freshly-rendered sheet's SVG.
   useSchematicGroupsOverlay({
@@ -586,7 +484,6 @@ export const SchematicViewer = ({
 
   return (
     <MouseTracker>
-      {traceHoverStyles && <style>{traceHoverStyles}</style>}
       {onSchematicComponentClicked && (
         <style>
           {`.schematic-component-clickable [data-schematic-component-id]:hover { cursor: pointer !important; }`}
diff --git a/lib/hooks/useTraceHoverHighlighting.ts b/lib/hooks/useTraceHoverHighlighting.ts
new file mode 100644
index 0000000..552ed0a
--- /dev/null
+++ b/lib/hooks/useTraceHoverHighlighting.ts
@@ -0,0 +1,78 @@
+import { useEffect, useMemo } from "react"
+import type { CircuitJson } from "circuit-json"
+import { getSchematicTraceNetKeyMap } from "lib/utils/get-schematic-trace-net-keys"
+
+const HOVER_FILTER =
+  "brightness(1.3) drop-shadow(0 0 3px rgba(255, 107, 53, 0.5))"
+
+export const useTraceHoverHighlighting = ({
+  svgDivRef,
+  circuitJson,
+  circuitJsonKey,
+  enabled,
+}: {
+  svgDivRef: React.RefObject<HTMLDivElement | null>
+  circuitJson: CircuitJson
+  circuitJsonKey: string
+  enabled: boolean
+}) => {
+  const schematicTraceNetKeyMap = useMemo(
+    () => getSchematicTraceNetKeyMap(circuitJson),
+    [circuitJsonKey, circuitJson],
+  )
+
+  useEffect(() => {
+    const svgRoot = svgDivRef.current
+    if (!svgRoot) return
+
+    const traceGroups = Array.from(
+      svgRoot.querySelectorAll<SVGGElement>(
+        '[data-circuit-json-type="schematic_trace"][data-schematic-trace-id]',
+      ),
+    )
+
+    const netKeyToElements = new Map<string, SVGGElement[]>()
+
+    for (const el of traceGroups) {
+      const id = el.dataset.schematicTraceId
+      if (!id) continue
+      const netKey = schematicTraceNetKeyMap.get(id)
+      const key = netKey ?? `single:${id}`
+      if (!netKeyToElements.has(key)) {
+        netKeyToElements.set(key, [])
+      }
+      netKeyToElements.get(key)!.push(el)
+    }
+
+    const cleanupCallbacks: Array<() => void> = []
+
+    for (const sameNetElements of netKeyToElements.values()) {
+      const applyHover = () => {
+        if (!enabled) return
+        for (const el of sameNetElements) {
+          el.style.filter = HOVER_FILTER
+        }
+      }
+
+      const removeHover = () => {
+        for (const el of sameNetElements) {
+          el.style.filter = ""
+        }
+      }
+
+      for (const el of sameNetElements) {
+        el.addEventListener("mouseenter", applyHover)
+        el.addEventListener("mouseleave", removeHover)
+        cleanupCallbacks.push(() => {
+          el.removeEventListener("mouseenter", applyHover)
+          el.removeEventListener("mouseleave", removeHover)
+          el.style.filter = ""
+        })
+      }
+    }
+
+    return () => {
+      for (const cleanup of cleanupCallbacks) cleanup()
+    }
+  }, [circuitJsonKey, svgDivRef, schematicTraceNetKeyMap, enabled])
+}
diff --git a/lib/utils/get-schematic-trace-net-keys.test.ts b/lib/utils/get-schematic-trace-net-keys.test.ts
new file mode 100644
index 0000000..5c86d24
--- /dev/null
+++ b/lib/utils/get-schematic-trace-net-keys.test.ts
@@ -0,0 +1,145 @@
+import { expect, test } from "bun:test"
+import {
+  getSchematicTraceNetKeyMap,
+  getSourceTraceNetKeyMap,
+  getSameNetSchematicTraceIdsMap,
+} from "./get-schematic-trace-net-keys"
+
+test("groups source traces by shared source ports transitively", () => {
+  const sourceTraceNetKeyMap = getSourceTraceNetKeyMap([
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_1",
+      connected_source_port_ids: ["source_port_1", "source_port_2"],
+      connected_source_net_ids: [],
+    },
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_2",
+      connected_source_port_ids: ["source_port_2", "source_port_3"],
+      connected_source_net_ids: [],
+    },
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_3",
+      connected_source_port_ids: ["source_port_4", "source_port_5"],
+      connected_source_net_ids: [],
+    },
+  ] as any)
+
+  expect(sourceTraceNetKeyMap.get("source_trace_1")!).toBe(
+    sourceTraceNetKeyMap.get("source_trace_2")!,
+  )
+  expect(sourceTraceNetKeyMap.get("source_trace_1")!).not.toBe(
+    sourceTraceNetKeyMap.get("source_trace_3")!,
+  )
+})
+
+test("maps schematic trace ids to their source trace net groups", () => {
+  const schematicTraceNetKeyMap = getSchematicTraceNetKeyMap([
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_1",
+      connected_source_port_ids: ["source_port_1", "source_port_2"],
+      connected_source_net_ids: [],
+    },
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_2",
+      connected_source_port_ids: ["source_port_2", "source_port_3"],
+      connected_source_net_ids: [],
+    },
+    {
+      type: "schematic_trace",
+      schematic_trace_id: "schematic_trace_1",
+      source_trace_id: "source_trace_1",
+    },
+    {
+      type: "schematic_trace",
+      schematic_trace_id: "schematic_trace_2",
+      source_trace_id: "source_trace_2",
+    },
+  ] as any)
+
+  expect(schematicTraceNetKeyMap.get("schematic_trace_1")!).toBe(
+    schematicTraceNetKeyMap.get("schematic_trace_2")!,
+  )
+})
+
+test("groups source traces by shared nets", () => {
+  const sourceTraceNetKeyMap = getSourceTraceNetKeyMap([
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_1",
+      connected_source_port_ids: [],
+      connected_source_net_ids: ["net_1"],
+    },
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_2",
+      connected_source_port_ids: [],
+      connected_source_net_ids: ["net_1"],
+    },
+  ] as any)
+
+  expect(sourceTraceNetKeyMap.get("source_trace_1")!).toBe(
+    sourceTraceNetKeyMap.get("source_trace_2")!,
+  )
+})
+
+test("groups source traces by subcircuit_connectivity_map_key", () => {
+  const sourceTraceNetKeyMap = getSourceTraceNetKeyMap([
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_1",
+      connected_source_port_ids: [],
+      connected_source_net_ids: [],
+      subcircuit_connectivity_map_key: "sub_1",
+    },
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_2",
+      connected_source_port_ids: [],
+      connected_source_net_ids: [],
+      subcircuit_connectivity_map_key: "sub_1",
+    },
+  ] as any)
+
+  expect(sourceTraceNetKeyMap.get("source_trace_1")!).toBe(
+    sourceTraceNetKeyMap.get("source_trace_2")!,
+  )
+})
+
+test("getSameNetSchematicTraceIdsMap returns same Set for same-net traces", () => {
+  const sameNetMap = getSameNetSchematicTraceIdsMap([
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_1",
+      connected_source_port_ids: ["source_port_1"],
+      connected_source_net_ids: [],
+    },
+    {
+      type: "source_trace",
+      source_trace_id: "source_trace_2",
+      connected_source_port_ids: ["source_port_1"],
+      connected_source_net_ids: [],
+    },
+    {
+      type: "schematic_trace",
+      schematic_trace_id: "schematic_trace_1",
+      source_trace_id: "source_trace_1",
+    },
+    {
+      type: "schematic_trace",
+      schematic_trace_id: "schematic_trace_2",
+      source_trace_id: "source_trace_2",
+    },
+  ] as any)
+
+  const set1 = sameNetMap.get("schematic_trace_1")!
+  const set2 = sameNetMap.get("schematic_trace_2")!
+  expect(set1).toBe(set2)
+  expect(set1.has("schematic_trace_1")).toBe(true)
+  expect(set1.has("schematic_trace_2")).toBe(true)
+  expect(set1.size).toBe(2)
+})
diff --git a/lib/utils/get-schematic-trace-net-keys.ts b/lib/utils/get-schematic-trace-net-keys.ts
new file mode 100644
index 0000000..c5f0684
--- /dev/null
+++ b/lib/utils/get-schematic-trace-net-keys.ts
@@ -0,0 +1,114 @@
+import type { CircuitJson } from "circuit-json"
+
+class DisjointSet {
+  private parent = new Map<string, string>()
+
+  add(id: string) {
+    if (!this.parent.has(id)) this.parent.set(id, id)
+  }
+
+  find(id: string): string {
+    this.add(id)
+    const parent = this.parent.get(id)!
+    if (parent === id) return id
+    const root = this.find(parent)
+    this.parent.set(id, root)
+    return root
+  }
+
+  union(a: string, b: string) {
+    const rootA = this.find(a)
+    const rootB = this.find(b)
+    if (rootA !== rootB) this.parent.set(rootB, rootA)
+  }
+}
+
+const getStringArray = (value: unknown): string[] =>
+  Array.isArray(value)
+    ? value.filter((item): item is string => typeof item === "string")
+    : []
+
+export const getSourceTraceNetKeyMap = (circuitJson: CircuitJson) => {
+  const disjointSet = new DisjointSet()
+  const sourceTraceIds: string[] = []
+
+  for (const element of circuitJson as Array<Record<string, unknown>>) {
+    if (element.type !== "source_trace") continue
+
+    const sourceTraceId = element.source_trace_id
+    if (typeof sourceTraceId !== "string") continue
+
+    sourceTraceIds.push(sourceTraceId)
+    disjointSet.add(sourceTraceId)
+
+    const connectedIds = [
+      ...getStringArray(element.connected_source_port_ids),
+      ...getStringArray(element.connected_source_net_ids),
+    ]
+
+    const subcircuitConnectivityMapKey =
+      typeof element.subcircuit_connectivity_map_key === "string"
+        ? element.subcircuit_connectivity_map_key
+        : null
+
+    if (subcircuitConnectivityMapKey) {
+      connectedIds.push(`subcircuit:${subcircuitConnectivityMapKey}`)
+    }
+
+    for (const connectedId of connectedIds) {
+      disjointSet.union(sourceTraceId, connectedId)
+    }
+  }
+
+  return new Map(
+    sourceTraceIds.map((sourceTraceId) => [
+      sourceTraceId,
+      disjointSet.find(sourceTraceId),
+    ]),
+  )
+}
+
+export const getSchematicTraceNetKeyMap = (circuitJson: CircuitJson) => {
+  const sourceTraceNetKeyMap = getSourceTraceNetKeyMap(circuitJson)
+  const schematicTraceNetKeyMap = new Map<string, string>()
+
+  for (const element of circuitJson as Array<Record<string, unknown>>) {
+    if (element.type !== "schematic_trace") continue
+
+    const schematicTraceId = element.schematic_trace_id
+    const sourceTraceId = element.source_trace_id
+    if (
+      typeof schematicTraceId !== "string" ||
+      typeof sourceTraceId !== "string"
+    ) {
+      continue
+    }
+
+    const netKey = sourceTraceNetKeyMap.get(sourceTraceId)
+    if (netKey) schematicTraceNetKeyMap.set(schematicTraceId, netKey)
+  }
+
+  return schematicTraceNetKeyMap
+}
+
+export const getSameNetSchematicTraceIdsMap = (circuitJson: CircuitJson) => {
+  const schematicTraceNetKeyMap = getSchematicTraceNetKeyMap(circuitJson)
+  const netKeyToTraceIds = new Map<string, Set<string>>()
+
+  for (const [traceId, netKey] of schematicTraceNetKeyMap) {
+    let traceIds = netKeyToTraceIds.get(netKey)
+    if (!traceIds) {
+      traceIds = new Set()
+      netKeyToTraceIds.set(netKey, traceIds)
+    }
+    traceIds.add(traceId)
+  }
+
+  const sameNetTraceIdsByTraceId = new Map<string, Set<string>>()
+  for (const [traceId, netKey] of schematicTraceNetKeyMap) {
+    const sameNet = netKeyToTraceIds.get(netKey)!
+    sameNetTraceIdsByTraceId.set(traceId, sameNet)
+  }
+
+  return sameNetTraceIdsByTraceId
+}