From f2063d52ffabc13c5027c4f4c4a415d6b67adaf5 Mon Sep 17 00:00:00 2001
From: Phodal Huang <h@phodal.com>
Date: Wed, 3 Mar 2021 17:11:30 +0800
Subject: [PATCH] feat(codeflower): init basic force layout code

---
 web/public/js/graph/cloc/code-flower.js | 67 ++++++++++++++++++++++++-
 web/public/js/index.js                  |  2 +-
 2 files changed, 66 insertions(+), 3 deletions(-)

diff --git a/web/public/js/graph/cloc/code-flower.js b/web/public/js/graph/cloc/code-flower.js
index ff9b6eaa..141c0da4 100644
--- a/web/public/js/graph/cloc/code-flower.js
+++ b/web/public/js/graph/cloc/code-flower.js
@@ -1,3 +1,66 @@
-function renderCodeFlower(data) {
-  console.log(data);
+function renderCodeFlower(data, selector) {
+  data = {
+    links: [
+      {"id": "Myriel", "group": 1},
+      {"id": "CountessdeLo", "group": 1},
+      {"id": "Geborand", "group": 1},
+      {"id": "Champtercier", "group": 1},
+    ],
+    nodes: [
+      {"source": "CountessdeLo", "target": "Myriel", "value": 1},
+      {"source": "Geborand", "target": "Myriel", "value": 1},
+      {"source": "Champtercier", "target": "Myriel", "value": 1},
+    ]
+  }
+
+  let w = GraphConfig.width;
+  let h = GraphConfig.height;
+
+  const links = data.links.map(d => Object.create(d));
+  const nodes = data.nodes.map(d => Object.create(d));
+
+  d3.select(selector).selectAll("svg").remove();
+  const svg = d3.select(selector).append("svg").attr("viewBox", [0, 0, w, h]);
+
+  const simulation = d3.forceSimulation(nodes)
+    .force("link", d3.forceLink().id(d => d.id))
+    .force("charge", d3.forceManyBody())
+    .force("center", d3.forceCenter(w / 2, h / 2));
+
+  let color = function () {
+    const scale = d3.scaleOrdinal(d3.schemeCategory10);
+    return d => scale(d.group);
+  }
+
+  const link = svg.append("g")
+    .attr("stroke", "#999")
+    .attr("stroke-opacity", 0.6)
+    .selectAll("line")
+    .data(links)
+    .join("line")
+    .attr("stroke-width", d => Math.sqrt(d.value));
+
+  const node = svg.append("g")
+    .attr("stroke", "#fff")
+    .attr("stroke-width", 1.5)
+    .selectAll("circle")
+    .data(nodes)
+    .join("circle")
+    .attr("r", 5)
+    .attr("fill", color)
+
+  node.append("title")
+    .text(d => d.id);
+
+  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);
+  });
 }
diff --git a/web/public/js/index.js b/web/public/js/index.js
index 4ae16856..cffbfd88 100644
--- a/web/public/js/index.js
+++ b/web/public/js/index.js
@@ -14,7 +14,7 @@ d3.json("data/cloc.json").then(function (json) {
 
   renderPacking(data["reports"])
   renderNestedTreemap(data["reports"])
-  renderCodeFlower(data["reports"]);
+  renderCodeFlower(data["reports"], '#code-flower');
 });
 
 d3.json("data/git.json").then(function (json) {