From 651bfc5cd284a259f429b50839c619dd1b94498a Mon Sep 17 00:00:00 2001
From: Jacky Zhao <j.zhao2k19@gmail.com>
Date: Mon, 07 Mar 2022 18:45:07 +0000
Subject: [PATCH] Merge pull request #62 from claudio4/hugo
---
assets/js/graph.js | 220 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 220 insertions(+), 0 deletions(-)
diff --git a/assets/js/graph.js b/assets/js/graph.js
new file mode 100644
index 0000000..c6c8718
--- /dev/null
+++ b/assets/js/graph.js
@@ -0,0 +1,220 @@
+async function drawGraph(url, baseUrl, pathColors, depth, enableDrag, enableLegend, enableZoom) {
+ const { index, links, content } = await fetchData
+ const curPage = url.replace(baseUrl, "")
+
+ const parseIdsFromLinks = (links) => [...(new Set(links.flatMap(link => ([link.source, link.target]))))]
+
+ const neighbours = new Set()
+ const wl = [curPage || "/", "__SENTINEL"]
+ if (depth >= 0) {
+ while (depth >= 0 && wl.length > 0) {
+ // compute neighbours
+ const cur = wl.shift()
+ if (cur === "__SENTINEL") {
+ depth--
+ 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))
+ }
+ }
+ } else {
+ parseIdsFromLinks(links).forEach(id => neighbours.add(id))
+ }
+
+ const data = {
+ nodes: [...neighbours].map(id => ({ id })),
+ links: links.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)"
+ }
+
+ for (const pathColor of pathColors) {
+ const path = Object.keys(pathColor)[0]
+ const colour = pathColor[path]
+ if (d.id.startsWith(path)) {
+ return colour
+ }
+ }
+
+ return "var(--g-node)"
+ }
+
+ const drag = simulation => {
+ function dragstarted(event, d) {
+ if (!event.active) simulation.alphaTarget(1).restart();
+ d.fx = d.x;
+ d.fy = d.y;
+ }
+
+ function dragged(event, d) {
+ d.fx = event.x;
+ d.fy = event.y;
+ }
+
+ function dragended(event, d) {
+ if (!event.active) simulation.alphaTarget(0);
+ d.fx = null;
+ d.fy = null;
+ }
+
+ const noop = () => { }
+ return d3.drag()
+ .on("start", enableDrag ? dragstarted : noop)
+ .on("drag", enableDrag ? dragged : noop)
+ .on("end", enableDrag ? dragended : noop);
+ }
+
+ const height = 250
+ const width = document.getElementById("graph-container").offsetWidth
+
+ const simulation = d3.forceSimulation(data.nodes)
+ .force("charge", d3.forceManyBody().strength(-30))
+ .force("link", d3.forceLink(data.links).id(d => d.id))
+ .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]);
+
+ if (enableLegend) {
+ 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)
+ svg.append("text").attr("x", -width / 2 + 40).attr("y", height / 2 - 30 * (i + 1)).text(key).style("font-size", "15px").attr("alignment-baseline", "middle")
+ })
+ }
+
+ // draw links between nodes
+ const link = svg.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)
+
+ // svg groups
+ const graphNode = svg.append("g")
+ .selectAll("g")
+ .data(data.nodes)
+ .enter().append("g")
+
+ // draw individual nodes
+ const node = graphNode.append("circle")
+ .attr("class", "node")
+ .attr("id", (d) => d.id)
+ .attr("r", (d) => {
+ const numOut = index.links[d.id]?.length || 0
+ const numIn = index.backlinks[d.id]?.length || 0
+ return 3 + (numOut + numIn) / 4
+ })
+ .attr("fill", color)
+ .style("cursor", "pointer")
+ .on("click", (_, d) => {
+ window.location.href = baseUrl + '/' + decodeURI(d.id).replace(/\s+/g, '-')
+ })
+ .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 currentId = d.id
+ const linkNodes = d3.selectAll(".link").filter(d => d.source.id === currentId || d.target.id === currentId)
+
+ // highlight neighbour nodes
+ neighbourNodes
+ .transition()
+ .duration(200)
+ .attr("fill", color)
+
+ // highlight links
+ linkNodes
+ .transition()
+ .duration(200)
+ .attr("stroke", "var(--g-link-active)")
+
+ // show text for self
+ d3.select(this.parentNode)
+ .select("text")
+ .raise()
+ .transition()
+ .duration(200)
+ .style("opacity", 1)
+ }).on("mouseleave", function (_, d) {
+ d3.selectAll(".node")
+ .transition()
+ .duration(200)
+ .attr("fill", color)
+
+ const currentId = d.id
+ const linkNodes = d3.selectAll(".link").filter(d => d.source.id === currentId || d.target.id === currentId)
+
+ linkNodes
+ .transition()
+ .duration(200)
+ .attr("stroke", "var(--g-link)")
+
+ d3.select(this.parentNode)
+ .select("text")
+ .transition()
+ .duration(200)
+ .style("opacity", 0)
+ })
+ .call(drag(simulation));
+
+ // draw labels
+ const labels = graphNode.append("text")
+ .attr("dx", 12)
+ .attr("dy", ".35em")
+ .text((d) => content[decodeURI(d.id).replace(/\s+/g, '-')]?.title || "Untitled")
+ .style("opacity", 0)
+ .style("pointer-events", "none")
+ .call(drag(simulation));
+
+ // set panning
+
+ if (enableZoom) {
+ svg.call(d3.zoom()
+ .extent([[0, 0], [width, height]])
+ .scaleExtent([0.25, 4])
+ .on("zoom", ({ transform }) => {
+ link.attr("transform", transform);
+ node.attr("transform", transform);
+ labels.attr("transform", transform);
+ }));
+ }
+
+ // progress the simulation
+ 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)
+ });
+}
--
Gitblit v1.10.0