jbrowse-plugin-graphgenomeviewer 7.4.3 → 7.5.0

package/src/GraphGenomeView/model.test.ts

@@ -156,6 +156,7 @@ vi.mock('@jbrowse/core/util/io', () => ({
 }))
 
 const SIMPLE_GFA = 'H\tVN:Z:1.0\nS\t1\tACGT\nS\t2\tGGCC\nL\t1\t+\t2\t+\t0M\n'
+const WALKED_GFA = `${SIMPLE_GFA}W\tGRCh38\t0\tchr1\t0\t8\t>1>2\nW\tA\t1\tctg\t0\t8\t>1>2\n`
 
 const MOCK_LAYOUT = {
   nodePositions: {
@@ -218,7 +219,9 @@ function createAnchoredModel(props: { layers?: GraphLayers } = {}) {
 
 // A view attached to a tree, so afterAttach installs its reactions. A root
 // model never attaches.
-function attachedModel(props: { layoutMode?: LayoutModeValue } = {}) {
+function attachedModel(
+  props: { layoutMode?: LayoutModeValue; [key: string]: unknown } = {},
+) {
   return types
     .model({ view: stateModelFactory() })
     .create({ view: { type: 'GraphGenomeView', ...props } }).view
@@ -366,6 +369,30 @@ describe('restoring a stated cut', () => {
     )
   })
 
+  // Only a GBZ cut follows snarls past the window; a walk file cuts the same
+  // graph either way, so the strip's toggle has nothing to fetch.
+  test.each([
+    ['RgfaTabixAdapter', 0],
+    ['GbzBaseSyntenyAdapter', 1],
+  ])('the walk strip on a %s track cuts again %i times', async (type, n) => {
+    rpcRespond()
+    mockSession.tracks = [{ trackId: 'rgfa-track', adapter: { type } }]
+    const model = attachedModel({
+      layoutMode: 'force',
+      loadedTrackId: 'rgfa-track',
+      loadedRegion: TEST_REGION,
+    })
+    await vi.waitFor(() => {
+      expect(model.graph).toBeDefined()
+    })
+    const cuts = () =>
+      mockRpcCall.mock.calls.filter(c => c[1] === 'GetSubgraph').length
+    const before = cuts()
+    model.setWalkStrip(!model.walkStrip)
+    await layoutsSettled(model)
+    expect(cuts()).toBe(before + n)
+  })
+
   test('a stated track the session lacks is reported without a fetch', async () => {
     mockSession.tracks = []
     const model = restoredView()
@@ -874,27 +901,7 @@ describe('loadGFAFromLocation', () => {
     expect(model.graphRegion).toEqual(TEST_REGION)
   })
 
-  test('a graph read from a url gives the figure spec that reads it again', async () => {
-    rpcRespond()
-    mockReadFile.mockResolvedValue(SIMPLE_GFA)
-    const model = stateModelFactory().create({
-      type: 'GraphGenomeView',
-      gfaLocation: location,
-      loadedRegion: TEST_REGION,
-      layoutMode: 'auto',
-    })
-    await model.load()
-    expect(model.figureSpec()).toMatchObject({
-      gfa: location.uri,
-      region: 'chr1:1000-5000',
-      layout: 'auto',
-    })
-    expect(model.figure()).toContain(
-      '"gfa":"https://example.com/graphs/small.gfa"',
-    )
-  })
-
-  test('a gene track in tabix shorthand gives the figure spec its index', async () => {
+  test('a gene track in tabix shorthand turns genes on', async () => {
     rpcRespond()
     mockReadFile.mockResolvedValue(SIMPLE_GFA)
     mockSession.tracks = [
@@ -916,11 +923,7 @@ describe('loadGFAFromLocation', () => {
     })
     await model.load()
     expect(model.showGenes).toBe(true)
-    expect(model.figureSpec()?.genes).toEqual({
-      file: 'https://jbrowse.org/ucsc/hg38/ncbiRefSeq.gff.gz',
-      index: 'https://jbrowse.org/ucsc/hg38/ncbiRefSeq.gff.gz.csi',
-      format: 'gff3',
-    })
+    expect(model.geneTrack?.trackId).toBe('ncbiRefSeq')
     mockSession.tracks = []
   })
 
@@ -1160,7 +1163,7 @@ describe('layoutMode', () => {
     const consoleSpy = vi.spyOn(console, 'error').mockImplementation(() => {})
     rpcRespond()
     const model = attachedModel()
-    await model.loadGFA(SIMPLE_GFA, 'plain')
+    await model.loadGFA(WALKED_GFA, 'plain')
 
     model.setMaxGraphNodes(1)
     await layoutsSettled(model)
@@ -1458,7 +1461,7 @@ describe('node budget', () => {
     const consoleSpy = vi.spyOn(console, 'error').mockImplementation(() => {})
     rpcRespond()
     const model = withLimit(1)
-    await model.loadGFA(SIMPLE_GFA, 'two nodes')
+    await model.loadGFA(WALKED_GFA, 'two nodes')
     expect(String(model.error)).toMatch(/too large to draw/i)
 
     await model.switchLayout('walkrows')
@@ -1467,6 +1470,21 @@ describe('node budget', () => {
     consoleSpy.mockRestore()
   })
 
+  // walk rows hand a graph with no walks to the force engine, which draws
+  // every node
+  test('walk rows of a graph with no walks keep the budget', async () => {
+    const consoleSpy = vi.spyOn(console, 'error').mockImplementation(() => {})
+    rpcRespond()
+    const model = withLimit(1)
+    model.setLayoutMode('walkrows')
+    await model.loadGFA(SIMPLE_GFA, 'two nodes')
+    expect(String(model.error)).toMatch(/too large to draw/i)
+    expect(
+      mockRpcCall.mock.calls.filter(c => c[1] === 'GraphComputeLayout'),
+    ).toHaveLength(0)
+    consoleSpy.mockRestore()
+  })
+
   test('a declined graph does not keep the last graph drawn under it', async () => {
     const consoleSpy = vi.spyOn(console, 'error').mockImplementation(() => {})
     rpcRespond()
@@ -1490,7 +1508,7 @@ describe('node budget', () => {
     rpcRespond()
     const model = withLimit(1)
     model.setLayoutMode('walkrows')
-    await model.loadGFA(SIMPLE_GFA, 'two nodes')
+    await model.loadGFA(WALKED_GFA, 'two nodes')
     expect(model.error).toBeUndefined()
     expect(model.layoutResult).toBeDefined()
   })
@@ -2303,6 +2321,61 @@ describe('launching out of the graph', () => {
   // Where every contributor is a loaded assembly the graph offers one panel per
   // strain, and a synteny view of all of them — the item is disabled while no
   // synteny track aligns them, rather than absent.
+  // Sakai's allele is drawn bare unless Sakai's own annotation is read onto
+  // it: the reference's genes only reach the backbone
+  test("a loaded strain's own genes are read onto its nodes off the reference", async () => {
+    mockSession.assemblyNames = ['K12', 'Sakai']
+    mockSession.tracks = [
+      {
+        trackId: 'Sakai_genes',
+        name: 'Sakai genes',
+        assemblyNames: ['Sakai'],
+        adapter: { type: 'Gff3TabixAdapter' },
+      },
+    ]
+    const model = await loadedGraph(ECOLI_RGFA, {
+      refName: 'chr',
+      assemblyName: 'K12',
+      start: 1000,
+      end: 1020,
+    })
+    expect(model.ownGeneReads).toEqual([
+      {
+        stableRefName: 'Sakai#1#chr',
+        trackId: 'Sakai_genes',
+        region: {
+          assemblyName: 'Sakai',
+          refName: 'chr',
+          start: 90000,
+          end: 90004,
+        },
+      },
+    ])
+    mockRpcCall.mockImplementation((_sid: unknown, method: string) =>
+      method === 'CoreGetFeatures'
+        ? Promise.resolve([
+            {
+              name: 'stx2A',
+              refName: 'chr',
+              start: 90001,
+              end: 90003,
+              strand: 1,
+              subfeatures: [{ type: 'exon', start: 90001, end: 90003 }],
+            },
+          ])
+        : Promise.reject(new Error(`Unexpected RPC: ${method}`)),
+    )
+    await model.loadOwnGenes()
+    expect(model.ownGeneFeatures).toEqual([
+      expect.objectContaining({
+        name: 'stx2A · Sakai#1',
+        refName: 'Sakai#1#chr',
+        start: 90001,
+        end: 90003,
+      }),
+    ])
+  })
+
   test('loaded strains offer a per-strain linear view and a synteny item', async () => {
     mockSession.assemblyNames = ['K12', 'Sakai']
     const model = await loadedGraph(ECOLI_RGFA, {
@@ -2369,6 +2442,68 @@ describe('launching out of the graph', () => {
     expect(model.connectedViewId).toBe('view-1')
   })
 
+  test('an opened view carries one gene track, not every feature track', async () => {
+    mockSession.assemblyNames = ['hg38']
+    const feature = (trackId: string, name: string, type: string) => ({
+      trackId,
+      name,
+      type: 'FeatureTrack',
+      assemblyNames: ['hg38'],
+      adapter: { type },
+    })
+    mockSession.tracks = [
+      feature('rgfa-track', 'graph', 'RgfaTabixAdapter'),
+      feature('clinvar', 'ClinVar', 'BigBedAdapter'),
+      feature('ncbi', 'NCBI RefSeq genes', 'Gff3TabixAdapter'),
+      feature('gwas', 'GWAS catalog', 'BedTabixAdapter'),
+    ]
+    const model = await loadedGraph(HPRC_RGFA)
+    const { reference } = model.nodeLaunchTargets('1+')
+    model.showInLinearView(reference!)
+
+    expect(mockSession.addedViews).toEqual([
+      [
+        'LinearGenomeView',
+        expect.objectContaining({
+          init: expect.objectContaining({ tracks: ['rgfa-track', 'ncbi'] }),
+        }),
+      ],
+    ])
+  })
+
+  // a BED named for no annotation is not picked as genes, but beats `No
+  // tracks active`
+  test('a strain with no gene-looking track opens on its first feature track', async () => {
+    mockSession.assemblyNames = ['K12', 'Sakai']
+    mockSession.tracks = [
+      {
+        trackId: 'Sakai_segments',
+        name: 'Sakai segments',
+        type: 'FeatureTrack',
+        assemblyNames: ['Sakai'],
+        adapter: { type: 'BedAdapter' },
+      },
+    ]
+    const model = await loadedGraph(ECOLI_RGFA, {
+      refName: 'chr',
+      assemblyName: 'K12',
+      start: 1000,
+      end: 1020,
+    })
+    model.showInLinearView(model.nodeLaunchTargets('3+').own!)
+    expect(mockSession.addedViews).toEqual([
+      [
+        'LinearGenomeView',
+        expect.objectContaining({
+          init: expect.objectContaining({
+            assembly: 'Sakai',
+            tracks: ['Sakai_segments'],
+          }),
+        }),
+      ],
+    ])
+  })
+
   // A pairing is not stolen from a view still on screen: a graph launched from
   // a linear view came in on that sync. One whose view has been closed is held
   // by nothing, and kept, the view the graph opens next never got its hover.
@@ -3268,6 +3403,71 @@ describe('walk rows', () => {
     expect(model.walkStripRows).toBeUndefined()
   })
 
+  // node 2 is off GRCh38's path and both A#1 and B#1 walk it
+  test('the block two rows walk off the reference is lettered on both', async () => {
+    rpcRespond()
+    const model = stateModelFactory().create({
+      type: 'GraphGenomeView',
+      layoutMode: 'walkrows',
+    })
+    await model.loadGFA(WALKS_GFA, 'walks')
+    expect(model.sharedBlocks).toBeUndefined()
+    model.setLayer('walkBlocks', true)
+    const rows = [model.walkRowBars!.reference, ...model.walkRowBars!.rows]
+    const lettered = model.sharedBlocks!.map(b => ({
+      label: rows[b.row]!.label,
+      start: b.start,
+      bp: b.bp,
+      block: b.block,
+    }))
+    expect(lettered.sort((a, b) => a.label.localeCompare(b.label))).toEqual([
+      { label: 'A#1', start: 4, bp: 8, block: 0 },
+      { label: 'B#1', start: 4, bp: 8, block: 0 },
+    ])
+  })
+
+  test('the shared order stacks rows that walk the same nodes together', async () => {
+    rpcRespond()
+    const model = stateModelFactory().create({
+      type: 'GraphGenomeView',
+      layoutMode: 'walkrows',
+    })
+    await model.loadGFA(WALKS_GFA, 'walks')
+    model.setWalkRowSamples(['A', 'B'])
+    const before = model.walkRowBars!.rows.map(r => r.label)
+    model.setWalkRowOrder('shared')
+    const labels = model.walkRowBars!.rows.map(r => r.label)
+    expect(labels).not.toEqual(before)
+    // A#1 and B#1 share node 2, so B#1 stacks under A#1 rather than after
+    // A#2, which walks GRCh38's own route
+    expect(labels).toEqual(['A#1', 'B#1', 'A#2'])
+    // the sample filter and the order sit side by side in the grammar
+    expect(model.walkRowSamples).toEqual(['A', 'B'])
+    model.setWalkRowSamples(undefined)
+    expect(model.walkRowOrder).toBe('shared')
+  })
+
+  test('ribbons join neighbouring walk rows where they walk the same nodes', async () => {
+    rpcRespond()
+    const model = stateModelFactory().create({
+      type: 'GraphGenomeView',
+      layoutMode: 'walkrows',
+    })
+    await model.loadGFA(WALKS_GFA, 'walks')
+    expect(model.walkRowRibbons).toBeUndefined()
+    model.setLayer('walkRibbons', true)
+    const ribbons = model.walkRowRibbons!
+    expect(ribbons.length).toBeGreaterThan(0)
+    // every row passes nodes 1 and 3, so each pair of neighbours shares a
+    // stretch that starts at both bars' left ends
+    const rows = [model.walkRowBars!.reference, ...model.walkRowBars!.rows]
+    for (let row = 0; row < rows.length - 1; row++) {
+      expect(ribbons.some(r => r.row === row && r.a0 === 0 && r.b0 === 0)).toBe(
+        true,
+      )
+    }
+  })
+
   test('walk rows and tube maps take no strip', async () => {
     rpcRespond()
     const model = stateModelFactory().create({
@@ -3413,89 +3613,6 @@ describe('walk rows', () => {
     expect(svg).toContain("'s path")
   })
 
-  test("the figure spec keeps the strip's sample filter", async () => {
-    rpcRespond()
-    mockReadFile.mockResolvedValue(WALKS_GFA)
-    const model = stateModelFactory().create({
-      type: 'GraphGenomeView',
-      gfaLocation: {
-        uri: 'https://example.com/graphs/walks.gfa',
-        locationType: 'UriLocation',
-      },
-      loadedRegion: TEST_REGION,
-      layoutMode: 'ordered',
-      layers: { walkStrip: true },
-      rows: { kept: ['B'] },
-    })
-    await model.load()
-    expect(model.walkStripRows!.rows.map(r => r.label)).toEqual(['B#1'])
-    expect(model.figureSpec()).toMatchObject({
-      walkStrip: true,
-      walkRowSamples: ['B'],
-    })
-    model.setWalkStrip(false)
-    expect(model.figureSpec()?.walkRowSamples).toBeUndefined()
-  })
-
-  // bandage-figure shows deletion edges unless a spec says otherwise
-  test('the figure spec states deletion edges only when they are off', async () => {
-    rpcRespond()
-    mockReadFile.mockResolvedValue(WALKS_GFA)
-    const model = stateModelFactory().create({
-      type: 'GraphGenomeView',
-      gfaLocation: {
-        uri: 'https://example.com/graphs/walks.gfa',
-        locationType: 'UriLocation',
-      },
-      loadedRegion: TEST_REGION,
-      layoutMode: 'ordered',
-    })
-    await model.load()
-    expect(model.figureSpec()).not.toHaveProperty('showDeletionEdges')
-    model.setLayer('deletions', false)
-    expect(model.figureSpec()?.showDeletionEdges).toBe(false)
-  })
-
-  // bandage-figure laid a linear layout out as an ordinary one, and drew its
-  // own line widths, while the spec said nothing about either
-  test('the figure spec carries the settings that change the picture', async () => {
-    rpcRespond()
-    mockReadFile.mockResolvedValue(WALKS_GFA)
-    const model = stateModelFactory().create({
-      type: 'GraphGenomeView',
-      gfaLocation: {
-        uri: 'https://example.com/graphs/walks.gfa',
-        locationType: 'UriLocation',
-      },
-      loadedRegion: TEST_REGION,
-      layoutMode: 'ordered',
-      size: 9,
-    })
-    await model.load()
-    expect(model.figureSpec()).toMatchObject({ contigThickness: 9 })
-    expect(model.figureSpec()).not.toHaveProperty('connectorThickness')
-    expect(model.figureSpec()).not.toHaveProperty('linearLayout')
-    model.setLinearLayout(true)
-    expect(model.figureSpec()?.linearLayout).toBe(true)
-  })
-
-  test('walk rows offer no figure spec, since bandage-figure draws no walk rows', async () => {
-    rpcRespond()
-    mockReadFile.mockResolvedValue(WALKS_GFA)
-    const model = stateModelFactory().create({
-      type: 'GraphGenomeView',
-      gfaLocation: {
-        uri: 'https://example.com/graphs/walks.gfa',
-        locationType: 'UriLocation',
-      },
-      loadedRegion: TEST_REGION,
-      layoutMode: 'walkrows',
-    })
-    await model.load()
-    expect(model.walkRowBars).toBeDefined()
-    expect(model.figureSpecUnavailable).toMatch(/^bandage-figure draws no /)
-  })
-
   test("a picker's set keeps lifted walks this cut does not hold", async () => {
     rpcRespond()
     const model = stateModelFactory().create({
@@ -3908,6 +4025,33 @@ describe('walk rows', () => {
     model.setHoveredRowWalks([])
     expect(model.walkLift).toBeUndefined()
   })
+
+  // each lift rebuilds the drawing, and a sweep down a lane crosses many rows
+  test('moving between lane rows lifts the next once the pointer settles', async () => {
+    rpcRespond()
+    const model = stateModelFactory().create({
+      type: 'GraphGenomeView',
+      layoutMode: 'auto',
+    })
+    await model.loadGFA(WALKS_GFA, 'walks')
+    vi.useFakeTimers()
+    try {
+      model.setHoveredRowWalks(['A#1#ctg'])
+      expect(model.hoveredRowWalks).toEqual(['A#1#ctg'])
+      model.setHoveredRowWalks(['A#2#ctg'])
+      model.setHoveredRowWalks(['B#1#ctg'])
+      vi.advanceTimersByTime(40)
+      // more moves within the row do not put the lift off
+      model.setHoveredRowWalks(['B#1#ctg'])
+      expect(model.hoveredRowWalks).toEqual(['A#1#ctg'])
+      vi.advanceTimersByTime(60)
+      expect(model.hoveredRowWalks).toEqual(['B#1#ctg'])
+      model.setHoveredRowWalks([])
+      expect(model.hoveredRowWalks).toEqual([])
+    } finally {
+      vi.useRealTimers()
+    }
+  })
 })
 
 describe('annotation reads beside a cut', () => {
@@ -4007,6 +4151,43 @@ describe('annotation reads beside a cut', () => {
     expect(model.error).toBeUndefined()
   })
 
+  // most graphs have no bubble index, and a pan re-cuts many times
+  test('a bubble index that failed to read is not asked for again', async () => {
+    const warn = vi.spyOn(console, 'warn').mockImplementation(() => {})
+    const bubbleReads = () =>
+      mockRpcCall.mock.calls.filter(
+        c =>
+          (c[2] as { adapterConfig?: { type?: string } }).adapterConfig
+            ?.type === 'MinigraphBubbleAdapter',
+      ).length
+    const respond = mockRpcCall.getMockImplementation()!
+    mockRpcCall.mockImplementation(
+      (sid: unknown, method: string, args: Record<string, unknown>) =>
+        (args.adapterConfig as { type?: string } | undefined)?.type ===
+        'MinigraphBubbleAdapter'
+          ? Promise.reject(new Error('404'))
+          : respond(sid, method, args),
+    )
+    const model = attachedModel()
+    const settleAll = async () => {
+      await vi.waitFor(() => {
+        expect(reads.length).toBeGreaterThanOrEqual(2)
+      })
+      for (const read of reads.splice(0)) {
+        read.settle([])
+      }
+    }
+    let loaded = cut(model)
+    await settleAll()
+    await loaded
+    expect(bubbleReads()).toBe(1)
+    loaded = cut(model)
+    await settleAll()
+    await loaded
+    expect(bubbleReads()).toBe(1)
+    warn.mockRestore()
+  })
+
   test('a cut too large to draw still reads its genes, for when the limit is raised', async () => {
     vi.spyOn(console, 'error').mockImplementation(() => {})
     const model = attachedModel()

package/src/GraphGenomeView/model.ts

@@ -1,5 +1,3 @@
-import { cutsWholeWalks as wholeWalksRule } from '@jbrowse/bandage-core/layout/walkStrip'
-import { layoutModeByValue } from '@jbrowse/bandage-core/layoutModes'
 import { getSession } from '@jbrowse/core/util'
 
 import {
@@ -46,16 +44,11 @@ export const GraphPane = withLaunchActions
             canonicalAssemblyName(session, region.assemblyName),
           )
     },
-    // Walk rows, and the strip of them under a node layout, measure whole
-    // walks, which a GBZ cut only follows when asked, so a layout change
-    // that changes whether the drawing needs them cuts the track again.
+    // A layout change that changes the cut options cuts the track again:
+    // whole walks (cutsWholeWalksIn), or the walk-row step budget.
     cutChangedFrom(from: LayoutModeValue) {
-      const before = wholeWalksRule(layoutModeByValue(from), {
-        walkStrip: self.walkStrip,
-        host: !!self.host,
-      })
       return (
-        self.cutsWholeWalks !== before ||
+        self.cutsWholeWalks !== self.cutsWholeWalksIn(from) ||
         (from === 'walkrows') !== (self.chosenLayoutMode === 'walkrows')
       )
     },

package/src/GraphGenomeView/nodeSize.test.ts

@@ -1,14 +1,30 @@
-import { nodeSizeOf, sizeOfNodeWidth } from './nodeSize'
+import { nodeSizeOf, sizeOf } from './nodeSize'
 
 test('a number is a uniform thickness, an object scales by its field', () => {
-  expect(nodeSizeOf(9)).toEqual({ width: 'uniform', px: 9 })
+  expect(nodeSizeOf(9)).toEqual({ width: 'uniform', px: 9, links: 'uniform' })
   expect(nodeSizeOf({ field: 'depth', value: 4 })).toEqual({
     width: 'depth',
     px: 4,
+    links: 'uniform',
+  })
+  expect(nodeSizeOf(undefined)).toEqual({
+    width: 'depth',
+    px: 6,
+    links: 'uniform',
+  })
+  expect(nodeSizeOf({ value: 5, links: 'walks' })).toEqual({
+    width: 'uniform',
+    px: 5,
+    links: 'walks',
   })
-  expect(nodeSizeOf(undefined)).toEqual({ width: 'depth', px: 6 })
 })
 
-test.each(['depth', 'uniform'] as const)('%s survives the trip', width => {
-  expect(nodeSizeOf(sizeOfNodeWidth(width, 5))).toEqual({ width, px: 5 })
+test.each([
+  ['depth', 'uniform'],
+  ['uniform', 'uniform'],
+  ['depth', 'walks'],
+  ['uniform', 'walks'],
+] as const)('nodes %s, links %s survive the trip', (width, links) => {
+  const size = { width, px: 5, links }
+  expect(nodeSizeOf(sizeOf(size))).toEqual(size)
 })

package/src/GraphGenomeView/nodeSize.ts

@@ -1,25 +1,41 @@
 import { isRecord } from '../isRecord'
 
+import type { EdgeWidth } from '@jbrowse/bandage-core/edgeWidths'
 import type { NodeWidth } from '@jbrowse/bandage-core/nodeWidths'
 
-// The node thickness as a config or session writes it: a px number draws every
+// The mark thickness as a config or session writes it: a px number draws every
 // node that thick; `{ field: 'depth', value }` thickens a node the more paths
-// carry it, around `value` px. Unset is by depth around 6.
-export type NodeSize = number | { field?: 'depth'; value?: number }
+// carry it, around `value` px; `links: 'walks'` thickens a link the more walks
+// take it. Unset is nodes by depth around 6, links uniform.
+export type NodeSize =
+  number | { field?: 'depth'; value?: number; links?: 'walks' }
 
 const DEFAULT_PX = 6
 
-export function nodeSizeOf(size: unknown): { width: NodeWidth; px: number } {
+export interface MarkSize {
+  width: NodeWidth
+  px: number
+  links: EdgeWidth
+}
+
+export function nodeSizeOf(size: unknown): MarkSize {
   return typeof size === 'number'
-    ? { width: 'uniform', px: size }
+    ? { width: 'uniform', px: size, links: 'uniform' }
     : isRecord(size)
       ? {
           width: size.field === 'depth' ? 'depth' : 'uniform',
           px: typeof size.value === 'number' ? size.value : DEFAULT_PX,
+          links: size.links === 'walks' ? 'walks' : 'uniform',
         }
-      : { width: 'depth', px: DEFAULT_PX }
+      : { width: 'depth', px: DEFAULT_PX, links: 'uniform' }
 }
 
-export function sizeOfNodeWidth(width: NodeWidth, px: number): NodeSize {
-  return width === 'depth' ? { field: 'depth', value: px } : px
+export function sizeOf({ width, px, links }: MarkSize): NodeSize {
+  return width === 'uniform' && links === 'uniform'
+    ? px
+    : {
+        ...(width === 'depth' ? { field: 'depth' as const } : {}),
+        value: px,
+        ...(links === 'walks' ? { links } : {}),
+      }
 }