From dbb06623669bc69d934f9261e7f687c910a36176 Mon Sep 17 00:00:00 2001
From: DmitriyA <faleo1999@mail.ru>
Date: Mon, 3 Mar 2025 04:05:21 -0500
Subject: [PATCH] =?UTF-8?q?=D0=9F=D0=BE=D1=84=D0=B8=D0=BA=D1=81=D0=B8?=
 =?UTF-8?q?=D0=BB=20=D0=BE=D0=B1=D0=BD=D0=BE=D0=B2=D0=BB=D0=B5=D0=BD=D0=B8?=
 =?UTF-8?q?=D0=B5=20=D0=B3=D1=80=D0=B0=D1=84=D0=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Components/TreeChart/TreeChart.jsx | 109 ++++++++++++++-----------
 1 file changed, 60 insertions(+), 49 deletions(-)

diff --git a/src/Components/TreeChart/TreeChart.jsx b/src/Components/TreeChart/TreeChart.jsx
index 51c05b6..260854b 100644
--- a/src/Components/TreeChart/TreeChart.jsx
+++ b/src/Components/TreeChart/TreeChart.jsx
@@ -1,48 +1,68 @@
-import React, { useRef, useEffect } from "react";
+import React, { useRef, useEffect, useMemo } from "react";
 import * as d3 from "d3";
 import { getStatusColor } from "./dataUtils";
 
 const TreeChart = ({ data, onNodeClick }) => {
     const chartRef = useRef();
+    const simulationRef = useRef(null);
 
-    useEffect(() => {
-        if (!data || !data.items) return;
-
-        const width = 1000;
-        const height = 1000;
-
+    const { root, nodes, links } = useMemo(() => {
+        if (!data || !data.items) return { root: null, nodes: [], links: [] };
         const root = d3.hierarchy(data, (d) => d.items);
         const links = root.links();
         const nodes = root.descendants();
+        return { root, nodes, links };
+    }, [data]);
+
+    useEffect(() => {
+        if (!chartRef.current) return;
+
+        const svg = d3.select(chartRef.current)
+            .attr("width", 2000)
+            .attr("height", 1000)
+            .attr("viewBox", [-500, -500, 1000, 1000])
+            .attr("style", "max-width: 100%; height: auto;");
+    }, []);
+
+    useEffect(() => {
+        if (!root || !chartRef.current) return;
 
         const svg = d3.select(chartRef.current);
-        svg.selectAll("*").remove();
 
-        svg
-            .attr("width", width)
-            .attr("height", height)
-            .attr("viewBox", [-width / 2, -height / 2, width, height])
-            .attr("style", "max-width: 100%; height: auto;");
+        if (simulationRef.current) {
+            simulationRef.current.stop();
+        }
 
-        const link = svg.append("g")
-            .attr("stroke", "#999")
-            .attr("stroke-opacity", 0.6)
+        const link = svg.select(".links")
             .selectAll("line")
-            .data(links)
-            .join("line"); // Используем join для обновления связей
+            .data(links, (d) => `${d.source.data.id}-${d.target.data.id}`)
+            .join("line")
+            .attr("stroke", "#999")
+            .attr("stroke-opacity", 0.6);
 
-        const node = svg.append("g")
-            .attr("stroke", "#000")
-            .attr("stroke-width", 1.5)
+        const node = svg.select(".nodes")
             .selectAll("circle")
-            .data(nodes)
-            .join("circle")  // Используем join для обновления узлов
+            .data(nodes, (d) => d.data.id)
+            .join("circle")
             .attr("fill", (d) => getStatusColor(d.data.status))
             .attr("stroke", "#fff")
             .attr("r", 7)
             .call(drag());
 
-        // Обновляем только те узлы, которые нуждаются в изменении
+        node.on("click", (event, d) => {
+            if (onNodeClick) {
+                onNodeClick(d.data.id, d.data.title);
+            }
+        });
+
+        const text = svg.select(".labels")
+            .selectAll("text")
+            .data(nodes, (d) => d.data.id)
+            .join("text")
+            .text((d) => d.data.title)
+            .attr("dx", 12)
+            .attr("dy", 4);
+
         node.each(function (d) {
             if (d.data.status === "red") {
                 d3.select(this)
@@ -69,28 +89,8 @@ const TreeChart = ({ data, onNodeClick }) => {
             }
         });
 
-        const text = svg.append("g")
-            .attr("fill", "#000")
-            .attr("font-family", "Arial")
-            .attr("font-size", 12)
-            .attr("pointer-events", "none")
-            .selectAll("text")
-            .data(nodes)
-            .join("text")
-            .text((d) => d.data.title)
-            .attr("dx", 12)
-            .attr("dy", 4);
-
-        node.append("title").text((d) => d.data.title);
-
-        node.on("click", (event, d) => {
-            if (onNodeClick) {
-                onNodeClick(d.data.id, d.data.title);
-            }
-        });
-
         const simulation = d3.forceSimulation(nodes)
-            .force("link", d3.forceLink(links).id((d) => d.data.title).distance(80).strength(1))
+            .force("link", d3.forceLink(links).id((d) => d.data.id).distance(80).strength(1))
             .force("charge", d3.forceManyBody().strength(-500))
             .force("x", d3.forceX())
             .force("y", d3.forceY());
@@ -111,11 +111,14 @@ const TreeChart = ({ data, onNodeClick }) => {
                 .attr("y", (d) => d.y + 4);
         });
 
+        simulationRef.current = simulation;
+
         return () => simulation.stop();
-    }, [data, onNodeClick]);
+    }, [root, links, nodes, onNodeClick]);
 
     const drag = () => {
         function dragstarted(event, d) {
+            if (!event.active) simulationRef.current.alphaTarget(0.3).restart();
             d.fx = d.x;
             d.fy = d.y;
         }
@@ -126,14 +129,22 @@ const TreeChart = ({ data, onNodeClick }) => {
         }
 
         function dragended(event, d) {
-            d.fx = null;
-            d.fy = null;
+            if (!event.active) simulationRef.current.alphaTarget(0);
+            d.fx = d.x;
+            d.fy = d.y;
         }
 
         return d3.drag().on("start", dragstarted).on("drag", dragged).on("end", dragended);
     };
 
-    return <svg ref={chartRef}></svg>;
+    return (
+        <svg ref={chartRef}>
+            <g className="links" />
+            <g className="nodes" />
+            <g className="labels" />
+        </svg>
+    );
 };
 
 export default TreeChart;
+