From 89c7f6e5bcc21b0e8f83a2bc6d680e2ffe431e6f Mon Sep 17 00:00:00 2001
From: vincentlu <t1341870251@gmail.com>
Date: 星期一, 13 一月 2025 09:38:29 +0800
Subject: [PATCH] #

---
 zy-acs-manager/src/main/java/com/zy/acs/manager/core/service/AvoidWaveCalculator.java |  227 +++++++++++++++++++++++++++++---------------------------
 1 files changed, 119 insertions(+), 108 deletions(-)

diff --git a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/service/AvoidWaveCalculator.java b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/service/AvoidWaveCalculator.java
index ed017d2..08e7042 100644
--- a/zy-acs-manager/src/main/java/com/zy/acs/manager/core/service/AvoidWaveCalculator.java
+++ b/zy-acs-manager/src/main/java/com/zy/acs/manager/core/service/AvoidWaveCalculator.java
@@ -50,6 +50,8 @@
 
     private ExecutorService singleThreadExecutor;
 
+    private File pythonFile = null;
+
     @Autowired
     private MapDataDispatcher mapDataDispatcher;
     @Autowired
@@ -85,7 +87,8 @@
         });
     }
 
-    public void calcWaveScope() {
+
+    public boolean calcWaveScope() {
         Integer lev = MapDataDispatcher.MAP_DEFAULT_LEV;
         boolean lockAcquired = false;
 
@@ -95,114 +98,16 @@
         try {
             if (!(lockAcquired = lock.tryLock(LOCK_TIMEOUT, TimeUnit.SECONDS))) {
                 log.warn("AvoidWaveCalculator execute fail, cause can not acquire lock ...");
-                return;
+                return false;
             }
 
-            // python
-            AgvModel agvModel = agvModelService.selectByType(AgvModelType.CTU_BOX_TRANSPORT_AGV.toString());    // can be optimized
-            Double avoidDistance = MapDataUtils.getVehicleWaveSafeDistance(agvModel.getDiameter(), MapDataConstant.MAX_DISTANCE_BETWEEN_ADJACENT_AGV_FACTOR);
-
-            long startTime = System.currentTimeMillis();
-
-            Resource resource = new ClassPathResource("agv.py");
-
-            File tempScript = null;
-
-            try {
-                InputStream is = resource.getInputStream();
-                tempScript = File.createTempFile("agv", ".py");
-                tempScript.deleteOnExit();
-
-                Files.copy(is, tempScript.toPath(), StandardCopyOption.REPLACE_EXISTING);
-                tempScript.setExecutable(true);
-
-                ProcessBuilder processBuilder = new ProcessBuilder(
-                        "python" // 鎴栬�� "python3" 鍙栧喅浜庣郴缁熼厤缃�
-                        , tempScript.getAbsolutePath()
-                        , String.valueOf(avoidDistance)
-                        , redisProperties.getHost()
-                        , redisProperties.getPassword()
-                        , String.valueOf(redisProperties.getPort())
-                        , String.valueOf(redisProperties.getIndex())
-                );
-
-                processBuilder.redirectErrorStream(true);
-
-//            File file = resource.getFile();
-//            ProcessBuilder processBuilder = new ProcessBuilder(
-//                    "python"
-//                    , file.getAbsolutePath()
-//                    , String.valueOf(avoidDistance)
-//                    , redisProperties.getHost()
-//                    , redisProperties.getPassword()
-//                    , String.valueOf(redisProperties.getPort())
-//                    , String.valueOf(redisProperties.getIndex())
-//            );
-//            processBuilder.redirectErrorStream(true);
-
-                Process process = processBuilder.start();
-
-                BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
-                String line;
-                StringBuilder builder = new StringBuilder();
-                while ((line = reader.readLine()) != null) {
-                    builder.append(line);
-                }
-
-                int exitCode = process.waitFor();
-                if (exitCode != 0) {
-                    log.error("Python script exited with error code: {}", exitCode);
-                    log.error("python error:{}", builder.toString());
-                    return;
-                }
-                reader.close();
-
-                if (builder.length() <= 0) {
-                    return;
-                }
-
-                String result = builder.toString();
-
-                if (!Cools.isEmpty(result)) {
-                    if (!"1".equals(result)) {
-                        log.error("Fail python");
-                    }
-                }
-            } catch (IOException | InterruptedException e) {
-                e.printStackTrace();
-            }
-            log.error("python finish {}", System.currentTimeMillis() - startTime);
-
-            // java
-//            String[][] codeMatrix = mapDataDispatcher.getCodeMatrix(lev);
-//            String[][] waveMatrix = mapDataDispatcher.initWaveMatrix(lev);
-//
-//            // lock path
-//            DynamicNode[][] dynamicMatrix = mapDataDispatcher.getDynamicMatrix(lev);
-//            for (int i = 0; i < dynamicMatrix.length; i++) {
-//                for (int j = 0; j < dynamicMatrix[i].length; j++) {
-//                    DynamicNode dynamicNode = dynamicMatrix[i][j];
-//                    String vehicle = dynamicNode.getVehicle();
-//                    if (!DynamicNodeType.ACCESS.val.equals(vehicle) && !DynamicNodeType.BLOCK.val.equals(vehicle)) {
-//                        AgvModel agvModel = agvModelService.getById(agvService.selectByUuid(vehicle).getAgvModel());    // can be optimized
-//
-//                        Double avoidDistance = MapDataUtils.getVehicleWaveSafeDistance(agvModel.getDiameter(), MapDataConstant.MAX_DISTANCE_BETWEEN_ADJACENT_AGV_FACTOR);
-//                        List<NavigateNode> includeList = mapService.getWaveScopeByCode(lev, codeMatrix[i][j], avoidDistance);
-//
-//                        for (NavigateNode navigateNode : includeList) {
-//                            String waveNode = waveMatrix[navigateNode.getX()][navigateNode.getY()]; // overlay
-//                            waveMatrix[navigateNode.getX()][navigateNode.getY()] = MapDataUtils.generateWaveNode(waveNode, vehicle);
-//                        }
-//                    }
-//                }
-//            }
-//
-////            mapDataDispatcher.printMatrix(waveMatrix);
-//            mapDataDispatcher.setWaveMatrix(lev, waveMatrix);
+//            return this.calcWaveScopeByPython(lev);
+            return this.calcWaveScopeByJava(lev);
 
         } catch (Exception e) {
 
             log.error(this.getClass().getSimpleName(), e);
+            return false;
         } finally {
 
             if (lockAcquired) {
@@ -210,11 +115,117 @@
             }
 
             stopWatch.stop();
-            if (stopWatch.getTime() > 100) {
-                log.info("婊ゆ尝鍑芥暟鑺辫垂鏃堕棿涓猴細{}姣......", stopWatch.getTime());
+            if (stopWatch.getTime() > 0) {
+                if (stopWatch.getTime() > 50) {
+                    log.info("婊ゆ尝鍑芥暟鑺辫垂鏃堕棿涓猴細{}姣......", stopWatch.getTime());
+                }
             }
-
         }
+    }
+
+    private boolean calcWaveScopeByPython(Integer lev) throws Exception {
+        // python
+        AgvModel agvModel = agvModelService.selectByType(AgvModelType.CTU_BOX_TRANSPORT_AGV.toString());    // can be optimized
+        Double avoidDistance = MapDataUtils.getVehicleWaveSafeDistance(agvModel.getDiameter(), MapDataConstant.MAX_DISTANCE_BETWEEN_ADJACENT_AGV_FACTOR);
+
+        if (null == pythonFile) {
+            pythonFile = loadPythonFile();
+        }
+
+        ProcessBuilder processBuilder = new ProcessBuilder(
+                "python" // 鎴栬�� "python3" 鍙栧喅浜庣郴缁熼厤缃�
+                , pythonFile.getAbsolutePath()
+                , String.valueOf(avoidDistance)
+                , redisProperties.getHost()
+                , redisProperties.getPassword()
+                , String.valueOf(redisProperties.getPort())
+                , String.valueOf(redisProperties.getIndex())
+        );
+
+        processBuilder.redirectErrorStream(true);
+
+        Process process = processBuilder.start();
+
+        BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
+        String line;
+        StringBuilder builder = new StringBuilder();
+        while ((line = reader.readLine()) != null) {
+            builder.append(line);
+        }
+
+        int exitCode = process.waitFor();
+        if (exitCode != 0) {
+            log.error("Python script exited with error code: {}", exitCode);
+            log.error("python error:{}", builder.toString());
+            return false;
+        }
+        reader.close();
+
+        if (builder.length() <= 0) {
+            return false;
+        }
+
+        String result = builder.toString();
+
+        if (Cools.isEmpty(result)) {
+            return false;
+        }
+        if (!"1".equals(result)) {
+            log.error("Failed to call python");
+            return false;
+        }
+
+        return true;
+    }
+
+    private boolean calcWaveScopeByJava(Integer lev) throws Exception {
+        AgvModel agvModel = agvModelService.selectByType(AgvModelType.CTU_BOX_TRANSPORT_AGV.toString());    // can be optimized
+        Double avoidDistance = MapDataUtils.getVehicleWaveSafeDistance(agvModel.getDiameter(), MapDataConstant.MAX_DISTANCE_BETWEEN_ADJACENT_AGV_FACTOR);
+
+        // java
+        String[][] codeMatrix = mapDataDispatcher.getCodeMatrix(lev);
+        String[][] waveMatrix = mapDataDispatcher.initWaveMatrix(lev);
+
+        // lock path
+        DynamicNode[][] dynamicMatrix = mapDataDispatcher.getDynamicMatrix(lev);
+        for (int i = 0; i < dynamicMatrix.length; i++) {
+            for (int j = 0; j < dynamicMatrix[i].length; j++) {
+                DynamicNode dynamicNode = dynamicMatrix[i][j];
+                String vehicle = dynamicNode.getVehicle();
+                if (!DynamicNodeType.ACCESS.val.equals(vehicle) && !DynamicNodeType.BLOCK.val.equals(vehicle)) {
+
+                    List<NavigateNode> includeList = mapService.getWaveScopeByCode(lev, codeMatrix[i][j], avoidDistance);
+
+                    for (NavigateNode navigateNode : includeList) {
+                        String waveNode = waveMatrix[navigateNode.getX()][navigateNode.getY()]; // overlay
+                        waveMatrix[navigateNode.getX()][navigateNode.getY()] = MapDataUtils.generateWaveNode(waveNode, vehicle);
+                    }
+                }
+            }
+        }
+
+//            mapDataDispatcher.printMatrix(waveMatrix);
+        mapDataDispatcher.setWaveMatrix(lev, waveMatrix);
+
+        return true;
+    }
+
+    private File loadPythonFile() {
+        File scriptFile = null;
+        try {
+            Resource resource = new ClassPathResource("agv.py");
+
+            InputStream is = resource.getInputStream();
+            scriptFile = File.createTempFile("agv", ".py");
+            scriptFile.deleteOnExit();
+
+            Files.copy(is, scriptFile.toPath(), StandardCopyOption.REPLACE_EXISTING);
+
+            boolean executable = scriptFile.setExecutable(true);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+        return scriptFile;
     }
 
     public void syncWaveBySingleVeh(String agvNo, String codeData) {
@@ -232,7 +243,7 @@
             }
 
             Agv agv = agvService.selectByUuid(agvNo);
-            AgvModel agvModel = agvModelService.getById(agv.getAgvModel());
+            AgvModel agvModel = agvModelService.getByAgvNo(agvNo);
             Double avoidDistance = MapDataUtils.getVehicleWaveSafeDistance(agvModel.getDiameter(), MapDataConstant.MAX_DISTANCE_BETWEEN_ADJACENT_AGV_FACTOR);
 
             String[][] waveMatrix = mapDataDispatcher.getWaveMatrix(lev);
@@ -288,7 +299,7 @@
         DynamicNode dynamicNode = dynamicMatrix[codeMatrixIdx[0]][codeMatrixIdx[1]];
         String vehicle = dynamicNode.getVehicle();
         if (vehicle.equals(DynamicNodeType.ACCESS.val)) {
-            mapDataDispatcher.modifyDynamicMatrix(null, Utils.singletonList(code.getData()), agv.getUuid());
+            mapDataDispatcher.modifyDynamicMatrix(null, Utils.singletonList(codeMatrixIdx), agv.getUuid());
         }
     }
 

--
Gitblit v1.9.1