Add prefetch to graph
This commit is contained in:
Aiden Bai
parent
0978df58f7
commit
c1b8d1965b
@ -1,12 +1,5 @@
|
||||
async function drawGraph(
|
||||
baseUrl,
|
||||
pathColors,
|
||||
depth,
|
||||
enableDrag,
|
||||
enableLegend,
|
||||
enableZoom
|
||||
) {
|
||||
const container = document.getElementById('graph-container')
|
||||
async function drawGraph(baseUrl, pathColors, depth, enableDrag, enableLegend, enableZoom) {
|
||||
const container = document.getElementById("graph-container")
|
||||
const { index, links, content } = await fetchData
|
||||
|
||||
// Use .pathname to remove hashes / searchParams / text fragments
|
||||
@ -23,22 +16,19 @@ async function drawGraph(
|
||||
const copyLinks = JSON.parse(JSON.stringify(links))
|
||||
|
||||
const neighbours = new Set()
|
||||
const wl = [curPage || '/', '__SENTINEL']
|
||||
const wl = [curPage || "/", "__SENTINEL"]
|
||||
if (depth >= 0) {
|
||||
while (depth >= 0 && wl.length > 0) {
|
||||
// compute neighbours
|
||||
const cur = wl.shift()
|
||||
if (cur === '__SENTINEL') {
|
||||
if (cur === "__SENTINEL") {
|
||||
depth--
|
||||
wl.push('__SENTINEL')
|
||||
wl.push("__SENTINEL")
|
||||
} else {
|
||||
neighbours.add(cur)
|
||||
const outgoing = index.links[cur] || []
|
||||
const incoming = index.backlinks[cur] || []
|
||||
wl.push(
|
||||
...outgoing.map((l) => l.target),
|
||||
...incoming.map((l) => l.source)
|
||||
)
|
||||
wl.push(...outgoing.map((l) => l.target), ...incoming.map((l) => l.source))
|
||||
}
|
||||
}
|
||||
} else {
|
||||
@ -47,14 +37,12 @@ async function drawGraph(
|
||||
|
||||
const data = {
|
||||
nodes: [...neighbours].map((id) => ({ id })),
|
||||
links: copyLinks.filter(
|
||||
(l) => neighbours.has(l.source) && neighbours.has(l.target)
|
||||
),
|
||||
links: copyLinks.filter((l) => neighbours.has(l.source) && neighbours.has(l.target)),
|
||||
}
|
||||
|
||||
const color = (d) => {
|
||||
if (d.id === curPage || (d.id === '/' && curPage === '')) {
|
||||
return 'var(--g-node-active)'
|
||||
if (d.id === curPage || (d.id === "/" && curPage === "")) {
|
||||
return "var(--g-node-active)"
|
||||
}
|
||||
|
||||
for (const pathColor of pathColors) {
|
||||
@ -65,7 +53,7 @@ async function drawGraph(
|
||||
}
|
||||
}
|
||||
|
||||
return 'var(--g-node)'
|
||||
return "var(--g-node)"
|
||||
}
|
||||
|
||||
const drag = (simulation) => {
|
||||
@ -89,9 +77,9 @@ async function drawGraph(
|
||||
const noop = () => {}
|
||||
return d3
|
||||
.drag()
|
||||
.on('start', enableDrag ? dragstarted : noop)
|
||||
.on('drag', enableDrag ? dragged : noop)
|
||||
.on('end', enableDrag ? dragended : noop)
|
||||
.on("start", enableDrag ? dragstarted : noop)
|
||||
.on("drag", enableDrag ? dragged : noop)
|
||||
.on("end", enableDrag ? dragended : noop)
|
||||
}
|
||||
|
||||
const height = Math.max(container.offsetHeight, 250)
|
||||
@ -99,67 +87,58 @@ async function drawGraph(
|
||||
|
||||
const simulation = d3
|
||||
.forceSimulation(data.nodes)
|
||||
.force('charge', d3.forceManyBody().strength(-30))
|
||||
.force("charge", d3.forceManyBody().strength(-30))
|
||||
.force(
|
||||
'link',
|
||||
"link",
|
||||
d3
|
||||
.forceLink(data.links)
|
||||
.id((d) => d.id)
|
||||
.distance(40)
|
||||
.distance(40),
|
||||
)
|
||||
.force('center', d3.forceCenter())
|
||||
.force("center", d3.forceCenter())
|
||||
|
||||
const svg = d3
|
||||
.select('#graph-container')
|
||||
.append('svg')
|
||||
.attr('width', width)
|
||||
.attr('height', height)
|
||||
.attr('viewBox', [-width / 2, -height / 2, width, height])
|
||||
.select("#graph-container")
|
||||
.append("svg")
|
||||
.attr("width", width)
|
||||
.attr("height", height)
|
||||
.attr("viewBox", [-width / 2, -height / 2, width, height])
|
||||
|
||||
if (enableLegend) {
|
||||
const legend = [
|
||||
{ Current: 'var(--g-node-active)' },
|
||||
{ Note: 'var(--g-node)' },
|
||||
...pathColors,
|
||||
]
|
||||
const legend = [{ Current: "var(--g-node-active)" }, { Note: "var(--g-node)" }, ...pathColors]
|
||||
legend.forEach((legendEntry, i) => {
|
||||
const key = Object.keys(legendEntry)[0]
|
||||
const colour = legendEntry[key]
|
||||
svg
|
||||
.append('circle')
|
||||
.attr('cx', -width / 2 + 20)
|
||||
.attr('cy', height / 2 - 30 * (i + 1))
|
||||
.attr('r', 6)
|
||||
.style('fill', colour)
|
||||
.append("circle")
|
||||
.attr("cx", -width / 2 + 20)
|
||||
.attr("cy", height / 2 - 30 * (i + 1))
|
||||
.attr("r", 6)
|
||||
.style("fill", colour)
|
||||
svg
|
||||
.append('text')
|
||||
.attr('x', -width / 2 + 40)
|
||||
.attr('y', height / 2 - 30 * (i + 1))
|
||||
.append("text")
|
||||
.attr("x", -width / 2 + 40)
|
||||
.attr("y", height / 2 - 30 * (i + 1))
|
||||
.text(key)
|
||||
.style('font-size', '15px')
|
||||
.attr('alignment-baseline', 'middle')
|
||||
.style("font-size", "15px")
|
||||
.attr("alignment-baseline", "middle")
|
||||
})
|
||||
}
|
||||
|
||||
// draw links between nodes
|
||||
const link = svg
|
||||
.append('g')
|
||||
.selectAll('line')
|
||||
.append("g")
|
||||
.selectAll("line")
|
||||
.data(data.links)
|
||||
.join('line')
|
||||
.attr('class', 'link')
|
||||
.attr('stroke', 'var(--g-link)')
|
||||
.attr('stroke-width', 2)
|
||||
.attr('data-source', (d) => d.source.id)
|
||||
.attr('data-target', (d) => d.target.id)
|
||||
.join("line")
|
||||
.attr("class", "link")
|
||||
.attr("stroke", "var(--g-link)")
|
||||
.attr("stroke-width", 2)
|
||||
.attr("data-source", (d) => d.source.id)
|
||||
.attr("data-target", (d) => d.target.id)
|
||||
|
||||
// svg groups
|
||||
const graphNode = svg
|
||||
.append('g')
|
||||
.selectAll('g')
|
||||
.data(data.nodes)
|
||||
.enter()
|
||||
.append('g')
|
||||
const graphNode = svg.append("g").selectAll("g").data(data.nodes).enter().append("g")
|
||||
|
||||
// calculate radius
|
||||
const nodeRadius = (d) => {
|
||||
@ -170,82 +149,68 @@ async function drawGraph(
|
||||
|
||||
// draw individual nodes
|
||||
const node = graphNode
|
||||
.append('circle')
|
||||
.attr('class', 'node')
|
||||
.attr('id', (d) => d.id)
|
||||
.attr('r', nodeRadius)
|
||||
.attr('fill', color)
|
||||
.style('cursor', 'pointer')
|
||||
.on('click', (_, d) => {
|
||||
.append("circle")
|
||||
.attr("class", "node")
|
||||
.attr("id", (d) => d.id)
|
||||
.attr("r", nodeRadius)
|
||||
.attr("fill", color)
|
||||
.style("cursor", "pointer")
|
||||
.on("click", (_, d) => {
|
||||
// SPA navigation
|
||||
window.navigate(
|
||||
new URL(`${baseUrl}${decodeURI(d.id).replace(/\s+/g, '-')}/`),
|
||||
'.singlePage'
|
||||
)
|
||||
window.Million.navigate(new URL(`${baseUrl}${decodeURI(d.id).replace(/\s+/g, "-")}/`), ".singlePage")
|
||||
})
|
||||
.on('mouseover', function(_, d) {
|
||||
d3.selectAll('.node')
|
||||
.transition()
|
||||
.duration(100)
|
||||
.attr('fill', 'var(--g-node-inactive)')
|
||||
.on("mouseover", function (_, d) {
|
||||
d3.selectAll(".node").transition().duration(100).attr("fill", "var(--g-node-inactive)")
|
||||
|
||||
const neighbours = parseIdsFromLinks([
|
||||
...(index.links[d.id] || []),
|
||||
...(index.backlinks[d.id] || []),
|
||||
])
|
||||
const neighbourNodes = d3
|
||||
.selectAll('.node')
|
||||
.filter((d) => neighbours.includes(d.id))
|
||||
const neighbourNodes = d3.selectAll(".node").filter((d) => neighbours.includes(d.id))
|
||||
const currentId = d.id
|
||||
window.Million.prefetch(new URL(`${baseUrl}${decodeURI(d.id).replace(/\s+/g, "-")}/`))
|
||||
const linkNodes = d3
|
||||
.selectAll('.link')
|
||||
.selectAll(".link")
|
||||
.filter((d) => d.source.id === currentId || d.target.id === currentId)
|
||||
|
||||
// highlight neighbour nodes
|
||||
neighbourNodes.transition().duration(200).attr('fill', color)
|
||||
neighbourNodes.transition().duration(200).attr("fill", color)
|
||||
|
||||
// highlight links
|
||||
linkNodes
|
||||
.transition()
|
||||
.duration(200)
|
||||
.attr('stroke', 'var(--g-link-active)')
|
||||
linkNodes.transition().duration(200).attr("stroke", "var(--g-link-active)")
|
||||
|
||||
// show text for self
|
||||
d3.select(this.parentNode)
|
||||
.raise()
|
||||
.select('text')
|
||||
.select("text")
|
||||
.transition()
|
||||
.duration(200)
|
||||
.style('opacity', 1)
|
||||
.style("opacity", 1)
|
||||
})
|
||||
.on('mouseleave', function(_, d) {
|
||||
d3.selectAll('.node').transition().duration(200).attr('fill', color)
|
||||
.on("mouseleave", function (_, d) {
|
||||
d3.selectAll(".node").transition().duration(200).attr("fill", color)
|
||||
|
||||
const currentId = d.id
|
||||
const linkNodes = d3
|
||||
.selectAll('.link')
|
||||
.selectAll(".link")
|
||||
.filter((d) => d.source.id === currentId || d.target.id === currentId)
|
||||
|
||||
linkNodes.transition().duration(200).attr('stroke', 'var(--g-link)')
|
||||
linkNodes.transition().duration(200).attr("stroke", "var(--g-link)")
|
||||
|
||||
d3.select(this.parentNode)
|
||||
.select('text')
|
||||
.transition()
|
||||
.duration(200)
|
||||
.style('opacity', 0)
|
||||
d3.select(this.parentNode).select("text").transition().duration(200).style("opacity", 0)
|
||||
})
|
||||
.call(drag(simulation))
|
||||
|
||||
// draw labels
|
||||
const labels = graphNode
|
||||
.append('text')
|
||||
.attr('dx', 0)
|
||||
.attr('dy', (d) => nodeRadius(d) + 8 + 'px')
|
||||
.attr('text-anchor', 'middle')
|
||||
.text((d) => content[d.id]?.title || d.id.replace('-', ' '))
|
||||
.style('opacity', 0)
|
||||
.style('pointer-events', 'none')
|
||||
.style('font-size', '0.4em')
|
||||
.append("text")
|
||||
.attr("dx", 0)
|
||||
.attr("dy", (d) => nodeRadius(d) + 8 + "px")
|
||||
.attr("text-anchor", "middle")
|
||||
.text((d) => content[d.id]?.title || d.id.replace("-", " "))
|
||||
.style("opacity", 0)
|
||||
.style("pointer-events", "none")
|
||||
.style("font-size", "0.4em")
|
||||
.raise()
|
||||
.call(drag(simulation))
|
||||
|
||||
@ -260,24 +225,24 @@ async function drawGraph(
|
||||
[width, height],
|
||||
])
|
||||
.scaleExtent([0.25, 4])
|
||||
.on('zoom', ({ transform }) => {
|
||||
link.attr('transform', transform)
|
||||
node.attr('transform', transform)
|
||||
.on("zoom", ({ transform }) => {
|
||||
link.attr("transform", transform)
|
||||
node.attr("transform", transform)
|
||||
const scale = transform.k
|
||||
const scaledOpacity = Math.max((scale - 1) / 3.75, 0)
|
||||
labels.attr('transform', transform).style('opacity', scaledOpacity)
|
||||
})
|
||||
labels.attr("transform", transform).style("opacity", scaledOpacity)
|
||||
}),
|
||||
)
|
||||
}
|
||||
|
||||
// progress the simulation
|
||||
simulation.on('tick', () => {
|
||||
simulation.on("tick", () => {
|
||||
link
|
||||
.attr('x1', (d) => d.source.x)
|
||||
.attr('y1', (d) => d.source.y)
|
||||
.attr('x2', (d) => d.target.x)
|
||||
.attr('y2', (d) => d.target.y)
|
||||
node.attr('cx', (d) => d.x).attr('cy', (d) => d.y)
|
||||
labels.attr('x', (d) => d.x).attr('y', (d) => d.y)
|
||||
.attr("x1", (d) => d.source.x)
|
||||
.attr("y1", (d) => d.source.y)
|
||||
.attr("x2", (d) => d.target.x)
|
||||
.attr("y2", (d) => d.target.y)
|
||||
node.attr("cx", (d) => d.x).attr("cy", (d) => d.y)
|
||||
labels.attr("x", (d) => d.x).attr("y", (d) => d.y)
|
||||
})
|
||||
}
|
||||
|
||||
@ -153,7 +153,7 @@ const highlight = (content, term) => {
|
||||
|
||||
const redir = (id, term) => {
|
||||
// SPA navigation
|
||||
window.navigate(
|
||||
window.Million.navigate(
|
||||
new URL(
|
||||
`${BASE_URL.replace(/\/$/g, "")}${id}#:~:text=${encodeURIComponent(term)}/`
|
||||
),
|
||||
|
||||
@ -107,7 +107,7 @@
|
||||
</script>
|
||||
{{else}}
|
||||
<script>
|
||||
window.navigate = (url) => window.location.href = url;
|
||||
window.Million.navigate = (url) => window.location.href = url;
|
||||
draw();
|
||||
</script>
|
||||
{{end}}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user