#

luxiaotao1123

2024-10-24 69869a21a127d2951f83421dbb6315f2997187e3

 zy-acs-manager/src/main/java/com/zy/acs/manager/core/DispatcherTestController.java

@@ -49,60 +49,125 @@

        Set<String> visited = new HashSet<>();

        // 遍历所有节点，寻找度数不等于2的节点作为起点
        for (String codeData : codeDataList) {
            if (adjacencyCodeMap.get(codeData).size() != 2) {
                if (!visited.contains(codeData)) {
                    Lane lane = new Lane(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
                    dfsCalc(codeData, null, lane, visited, adjacencyCodeMap);
                    lanes.add(lane);
                List<String> neighbors = adjacencyCodeMap.get(codeData);
                for (String neighbor : neighbors) {
                    if (adjacencyCodeMap.get(neighbor).size() == 2 && !visited.contains(neighbor)) {
                        Lane lane = new Lane(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
                        lane.getCodes().add(codeData); // 包含起点
                        dfsCalcIncludingEnd(codeData, neighbor, lane, visited, adjacencyCodeMap);
                        lanes.add(lane);
                    }
                }
            }
        }

        // 处理独立的度数为2的环路或未连接到度数不等于2的节点的部分
        for (String codeData : codeDataList) {
            if (adjacencyCodeMap.get(codeData).size() == 2 && !visited.contains(codeData)) {
                // 检查是否为环路的一部分
                Lane lane = new Lane(String.valueOf(snowflakeIdWorker.nextId()).substring(3));
                dfsCalcForLoop(codeData, null, lane, visited, adjacencyCodeMap);
                lanes.add(lane);
            }
        }


        System.out.println(GsonUtils.toJson(lanes));
        return R.ok().add(lanes);
    }

    private void dfsCalcIncludingEnd(String start, String current, Lane lane, Set<String> visited, Map<String, List<String>> adjacencyCodeMap) {
        // 添加当前节点
        lane.getCodes().add(current);
        visited.add(current);

    private void dfsCalc(String code, String parent, Lane lane, Set<String> visited, Map<String, List<String>> adjacencyCodeMap) {
        visited.add(code);
        lane.getCodes().add(code);

        List<String> neighbors = adjacencyCodeMap.get(code);
        List<String> neighbors = adjacencyCodeMap.get(current);
        if (neighbors == null || neighbors.isEmpty()) {
            return;
        }

        for (String neighbor : neighbors) {
            if (parent != null && neighbor.equals(parent)) {
            if (neighbor.equals(start)) {
                continue;
            }
            if (!visited.contains(neighbor) && isSameDirection(code, neighbor, parent)) {
                dfsCalc(neighbor, code, lane, visited, adjacencyCodeMap);

            if (!visited.contains(neighbor)) {
                int degree = adjacencyCodeMap.get(neighbor).size();
                if (degree == 2) {
                    if (isSameDirection(current, neighbor, start)) {
                        dfsCalcIncludingEnd(current, neighbor, lane, visited, adjacencyCodeMap);
                    }
                } else {
                    // 终点或拐弯点，包含并停止
                    lane.getCodes().add(neighbor);
                    visited.add(neighbor);
                }
            }
        }
    }
    private void dfsCalcForLoop(String current, String parent, Lane lane, Set<String> visited, Map<String, List<String>> adjacencyCodeMap) {
        lane.getCodes().add(current);
        visited.add(current);

        List<String> neighbors = adjacencyCodeMap.get(current);
        if (neighbors == null || neighbors.isEmpty()) {
            return;
        }

        for (String neighbor : neighbors) {
            if (neighbor.equals(parent)) {
                continue;
            }

            if (!visited.contains(neighbor)) {
                int degree = adjacencyCodeMap.get(neighbor).size();
                if (degree == 2) {
                    if (isSameDirection(current, neighbor, parent)) {
                        dfsCalcForLoop(current, neighbor, lane, visited, adjacencyCodeMap);
                    }
                } else {
                    lane.getCodes().add(neighbor);
                    visited.add(neighbor);
                }
            }
        }
    }

    public boolean isSameDirection(String code, String neighbor, String parent) {
    public boolean isSameDirection(String current, String neighbor, String parent) {
        if (parent == null) {
            return true;
        }

        Code parentCode = codeService.selectByData(parent);
        Code currentCode = codeService.selectByData(code);
        Code currentCode = codeService.selectByData(current);
        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);
    }

    /**
     * 计算两个点之间的方向角
     *
     * @param from 起点
     * @param to 终点
     * @return 方向角（弧度）
     */
    public double calculateDirection(Code from, Code to) {
        double deltaX = to.getX() - from.getX();
        double deltaY = to.getY() - from.getY();
        return Math.atan2(deltaY, deltaX);
    }


}