优化

src/util/useD3.js

@@ -687,44 +687,51 @@ export default class RelationGraph {
     }) // x 坐标为两点中心距离减去自身长度一半
 
     // 5.修改节点的位置
-    this.circles.attr('cx', (d) => d.x).attr('cy', (d) => d.y)
-    this.squares
-      .attr('x', (d) => d.x - r)
-      .attr('y', (d) => d.y - r)
-      .attr('points', (d) => polygonPoints(d.x, d.y, r, 6))
-    this.pentagons
-      .attr('x', (d) => d.x - r)
-      .attr('y', (d) => d.y - r)
-      .attr('points', (d) => polygonPoints(d.x, d.y, r, 10))
-
     // 让menu随圆圈移动
-    this.circles.each(function () {
-      const menuEle = d3.select(this.nextSibling)
-      if (!menuEle.empty()) {
-        const self = d3.select(this)
-        const cx = self.attr('cx')
-        const cy = self.attr('cy')
-        menuEle.attr('transform', `translate(${cx},${cy})`).raise()
-      }
-    })
-    this.squares.each(function () {
-      const menuEle = d3.select(this.nextSibling)
-      if (!menuEle.empty()) {
-        const self = d3.select(this)
-        const cx = +self.attr('x') + r
-        const cy = +self.attr('y') + r
-        menuEle.attr('transform', `translate(${cx},${cy})`).raise()
-      }
-    })
-    this.pentagons.each(function () {
-      const menuEle = d3.select(this.nextSibling)
-      if (!menuEle.empty()) {
-        const self = d3.select(this)
-        const cx = +self.attr('x') + r
-        const cy = +self.attr('y') + r
-        menuEle.attr('transform', `translate(${cx},${cy})`).raise()
-      }
-    })
+    if (this.circles) {
+      this.circles.attr('cx', (d) => d.x).attr('cy', (d) => d.y)
+      this.circles.each(function () {
+        const menuEle = d3.select(this.nextSibling)
+        if (!menuEle.empty()) {
+          const self = d3.select(this)
+          const cx = self.attr('cx')
+          const cy = self.attr('cy')
+          menuEle.attr('transform', `translate(${cx},${cy})`).raise()
+        }
+      })
+    }
+    if (this.squares) {
+      this.squares
+        .attr('x', (d) => d.x - r)
+        .attr('y', (d) => d.y - r)
+        .attr('points', (d) => polygonPoints(d.x, d.y, r, 6))
+
+      this.squares.each(function () {
+        const menuEle = d3.select(this.nextSibling)
+        if (!menuEle.empty()) {
+          const self = d3.select(this)
+          const cx = +self.attr('x') + r
+          const cy = +self.attr('y') + r
+          menuEle.attr('transform', `translate(${cx},${cy})`).raise()
+        }
+      })
+    }
+    if (this.pentagons) {
+      this.pentagons
+        .attr('x', (d) => d.x - r)
+        .attr('y', (d) => d.y - r)
+        .attr('points', (d) => polygonPoints(d.x, d.y, r, 10))
+
+      this.pentagons.each(function () {
+        const menuEle = d3.select(this.nextSibling)
+        if (!menuEle.empty()) {
+          const self = d3.select(this)
+          const cx = +self.attr('x') + r
+          const cy = +self.attr('y') + r
+          menuEle.attr('transform', `translate(${cx},${cy})`).raise()
+        }
+      })
+    }
   }
 
   // 高亮元素及其相关的元素