From 28d9ed7bc5749bf0b20b537344573b79e142ff59 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@163.com>
Date: 星期四, 24 十月 2024 16:09:12 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/core/DispatcherTestController.java |   79 ++++++++++++++++++++++++++++++++++-----
 1 files changed, 68 insertions(+), 11 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/DispatcherTestController.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/DispatcherTestController.java
index 3e2db2d..708d4fb 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/DispatcherTestController.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/DispatcherTestController.java
@@ -42,21 +42,62 @@
                 return codeService.getById(node).getData();
             }).collect(Collectors.toList()));
         }
+        List<String> codeDataList = codeList.stream().map(Code::getData).collect(Collectors.toList());
+        System.out.println(codeDataList.toString());
+        System.out.println(adjacencyCodeMap.toString());
+
+
         Set<String> visited = new HashSet<>();
 
-        for (String codeData : codeList.stream().map(Code::getData).collect(Collectors.toList())) {
-            if (!visited.contains(codeData)) {
-                Lane lane = new Lane(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
-                dfsCalc(codeData, null, lane, visited, adjacencyCodeMap);
-                lanes.add(lane);
+        // 閬嶅巻鎵�鏈夎妭鐐癸紝瀵绘壘搴︽暟涓嶇瓑浜�2鐨勮妭鐐逛綔涓鸿捣鐐�
+        for (String codeData : codeDataList) {
+            if (adjacencyCodeMap.getOrDefault(codeData, Collections.emptyList()).size() != 2) {
+                List<String> neighbors = adjacencyCodeMap.getOrDefault(codeData, Collections.emptyList());
+                for (String neighbor : neighbors) {
+                    // 浠呴亶鍘嗗害鏁颁负2涓旀湭琚闂殑閭诲眳
+                    if (adjacencyCodeMap.getOrDefault(neighbor, Collections.emptyList()).size() == 2 && !visited.contains(neighbor)) {
+                        Lane lane = new Lane(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
+                        dfsCalc(neighbor, codeData, lane, visited, adjacencyCodeMap);
+                        lanes.add(lane);
+                    }
+                }
             }
         }
+
+        // 澶勭悊搴︽暟涓�2涓旇繛鎺ヤ袱涓嫄寮偣鐨勮妭鐐癸紙鐙珛宸烽亾锛�
+        for (String codeData : codeDataList) {
+            if (adjacencyCodeMap.getOrDefault(codeData, Collections.emptyList()).size() == 2 && !visited.contains(codeData)) {
+                List<String> neighbors = adjacencyCodeMap.getOrDefault(codeData, Collections.emptyList());
+                boolean connectedToTwoTurningPoints = true;
+                for (String neighbor : neighbors) {
+                    if (adjacencyCodeMap.getOrDefault(neighbor, Collections.emptyList()).size() == 2) {
+                        connectedToTwoTurningPoints = false;
+                        break;
+                    }
+                }
+                if (connectedToTwoTurningPoints) {
+                    Lane lane = new Lane(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
+                    lane.getCodes().add(codeData);
+                    lanes.add(lane);
+                    visited.add(codeData);
+                }
+            }
+        }
+
+
         System.out.println(GsonUtils.toJson(lanes));
         return R.ok().add(lanes);
     }
 
 
     private void dfsCalc(String code, String parent, Lane lane, Set<String> visited, Map<String, List<String>> adjacencyCodeMap) {
+        // 妫�鏌ュ綋鍓嶈妭鐐圭殑搴︽暟鏄惁涓�2
+        if (adjacencyCodeMap.getOrDefault(code, Collections.emptyList()).size() != 2) {
+            // 涓嶅寘鍚嫄寮偣
+            return;
+        }
+
+        // 鏍囪涓哄凡璁块棶骞跺姞鍏ュ贩閬�
         visited.add(code);
         lane.getCodes().add(code);
 
@@ -66,11 +107,15 @@
         }
 
         for (String neighbor : neighbors) {
-            if (parent != null && neighbor.equals(parent)) {
+            if (neighbor.equals(parent)) {
                 continue;
             }
-            if (!visited.contains(neighbor) && isSameDirection(code, neighbor, parent)) {
-                dfsCalc(neighbor, code, lane, visited, adjacencyCodeMap);
+
+            if (adjacencyCodeMap.getOrDefault(neighbor, Collections.emptyList()).size() == 2 && !visited.contains(neighbor)) {
+                // 妫�鏌ユ柟鍚戞槸鍚︿竴鑷�
+                if (isSameDirection(code, neighbor, parent)) {
+                    dfsCalc(neighbor, code, lane, visited, adjacencyCodeMap);
+                }
             }
         }
     }
@@ -79,9 +124,20 @@
         if (parent == null) {
             return true;
         }
-        double direction1 = calculateDirection(codeService.selectByData(parent) , codeService.selectByData(code));
-        double direction2 = calculateDirection(codeService.selectByData(code), codeService.selectByData(neighbor));
-        return direction1 == direction2;
+
+        Code parentCode = codeService.selectByData(parent);
+        Code currentCode = codeService.selectByData(code);
+        Code neighborCode = codeService.selectByData(neighbor);
+
+        double direction1 = calculateDirection(parentCode, currentCode);
+        double direction2 = calculateDirection(currentCode, neighborCode);
+        double angleDifference = Math.abs(direction1 - direction2);
+
+        // 瑙勮寖鍖栬搴﹀樊
+        angleDifference = Math.min(angleDifference, 2 * Math.PI - angleDifference);
+
+        // 璁剧疆瑙掑害宸槇鍊间负3搴�
+        return angleDifference < Math.toRadians(3);
     }
 
     public double calculateDirection(Code from, Code to) {
@@ -90,4 +146,5 @@
         return Math.atan2(deltaY, deltaX);
     }
 
+
 }

--
Gitblit v1.9.1