From 2bdbfd1ac24d4e9a87daf6fb8b99ec32264d5752 Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期六, 21 十二月 2024 11:25:27 +0800 Subject: [PATCH] # --- zy-acs-manager/src/main/java/com/zy/acs/manager/core/service/AvoidWaveCalculator.java | 154 ++++++++++++++++++++++++++++++++++++--------------- 1 files changed, 109 insertions(+), 45 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 af11655..2a6d863 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 @@ -1,8 +1,9 @@ package com.zy.acs.manager.core.service; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; -import com.zy.acs.framework.common.Cools; import com.zy.acs.common.utils.Utils; +import com.zy.acs.framework.common.Cools; +import com.zy.acs.manager.common.config.RedisProperties; import com.zy.acs.manager.common.utils.MapDataUtils; import com.zy.acs.manager.core.constant.MapDataConstant; import com.zy.acs.manager.core.service.astart.DynamicNodeType; @@ -13,6 +14,7 @@ import com.zy.acs.manager.manager.entity.AgvDetail; import com.zy.acs.manager.manager.entity.AgvModel; import com.zy.acs.manager.manager.entity.Code; +import com.zy.acs.manager.manager.enums.AgvModelType; import com.zy.acs.manager.manager.service.AgvDetailService; import com.zy.acs.manager.manager.service.AgvModelService; import com.zy.acs.manager.manager.service.AgvService; @@ -20,10 +22,15 @@ import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang.time.StopWatch; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.core.io.ClassPathResource; +import org.springframework.core.io.Resource; import org.springframework.stereotype.Component; import javax.annotation.PostConstruct; import javax.annotation.PreDestroy; +import java.io.*; +import java.nio.file.Files; +import java.nio.file.StandardCopyOption; import java.util.List; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; @@ -55,6 +62,8 @@ private CodeService codeService; @Autowired private MapService mapService; + @Autowired + private RedisProperties redisProperties; @PostConstruct @SuppressWarnings("all") @@ -66,14 +75,34 @@ this.calcDynamicNodeWhenBoot(); - while (!Thread.currentThread().isInterrupted()) { - - this.calcWaveScope(); - - try { Thread.sleep(500); } catch (InterruptedException ignore) {} - } +// while (!Thread.currentThread().isInterrupted()) { +// +// this.calcWaveScope(); +// +// try { Thread.sleep(500); } catch (InterruptedException ignore) {} +// } }); + } + + private File pythonFile = null; + + 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); + + scriptFile.setExecutable(true); + } catch (Exception e) { + throw new RuntimeException(e); + } + return scriptFile; } public void calcWaveScope() { @@ -89,47 +118,82 @@ return; } - String[][] codeMatrix = mapDataDispatcher.getCodeMatrix(lev); - String[][] waveMatrix = mapDataDispatcher.initWaveMatrix(lev); + // 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); - // vehicle -// for (Agv agv : agvService.list(new LambdaQueryWrapper<>())) { -// AgvDetail agvDetail = agvDetailService.selectByAgvId(agv.getId()); -// AgvModel agvModel = agvModelService.getById(agv.getAgvModel()); -// Code code = codeService.getById(agvDetail.getRecentCode()); -// -// Double avoidDistance = MapDataUtils.getVehicleWaveSafeDistance(agvModel.getDiameter(), MapDataConstant.MAX_DISTANCE_BETWEEN_ADJACENT_AGV_FACTOR); -// List<NavigateNode> includeList = mapService.getWaveScopeByCode(lev, code.getData(), avoidDistance); -// -// for (NavigateNode navigateNode : includeList) { -// String waveNode = waveMatrix[navigateNode.getX()][navigateNode.getY()]; -// waveMatrix[navigateNode.getX()][navigateNode.getY()] = MapDataUtils.generateWaveNode(waveNode, agv.getUuid()); -// } -// -// } - - // 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); - } - } - } + if (null == pythonFile) { + pythonFile = loadPythonFile(); } -// mapDataDispatcher.printMatrix(waveMatrix); - mapDataDispatcher.setWaveMatrix(lev, waveMatrix); + 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; + } + reader.close(); + + if (builder.length() <= 0) { + return; + } + + String result = builder.toString(); + + if (!Cools.isEmpty(result)) { + if (!"1".equals(result)) { + log.error("Failed to call python"); + } + } +// 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); } catch (Exception e) { -- Gitblit v1.9.1