#

luxiaotao1123

2024-12-16 edd8ab14f72df3404b3a523e788eff2836238f3b

 zy-acs-manager/src/main/java/com/zy/acs/manager/core/service/AvoidWaveCalculator.java

@@ -3,6 +3,7 @@
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
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,16 @@
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.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
@@ -55,6 +63,8 @@
    private CodeService codeService;
    @Autowired
    private MapService mapService;
    @Autowired
    private RedisProperties redisProperties;

    @PostConstruct
    @SuppressWarnings("all")
@@ -89,78 +99,95 @@
                return;
            }

//            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();
//
//            ProcessBuilder processBuilder = new ProcessBuilder("python"
//                    , "D:\\tmp\\pyagv\\agv4.py"
//                    , String.valueOf(avoidDistance)
//            );
//            processBuilder.redirectErrorStream(true);
//
//            try {
//                Process process = processBuilder.start();
//
//                // 读取Python脚本的输出
//                BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
//                String line;
//                StringBuilder builder = new StringBuilder();
//                while ((line = reader.readLine()) != null) {
//                    builder.append(line);
//                }
//
//                // 等待Python脚本执行完成
//                int exitCode = process.waitFor();
//                if (exitCode != 0) {
//                    System.out.println("Python script exited with error code: " + exitCode);
//                    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);
            // 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);

            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
            long startTime = System.currentTimeMillis();

                        Double avoidDistance = MapDataUtils.getVehicleWaveSafeDistance(agvModel.getDiameter(), MapDataConstant.MAX_DISTANCE_BETWEEN_ADJACENT_AGV_FACTOR);
                        List<NavigateNode> includeList = mapService.getWaveScopeByCode(lev, codeMatrix[i][j], avoidDistance);
            Resource codeMatrixResource = new ClassPathResource("codeMatrix.txt");
            Resource cdaMatrixResource = new ClassPathResource("cdaMatrix.txt");

                        for (NavigateNode navigateNode : includeList) {
                            String waveNode = waveMatrix[navigateNode.getX()][navigateNode.getY()]; // overlay
                            waveMatrix[navigateNode.getX()][navigateNode.getY()] = MapDataUtils.generateWaveNode(waveNode, vehicle);
                        }
            File codeMatrixFile = codeMatrixResource.getFile();
            File cdaMatrixFile = cdaMatrixResource.getFile();

            String codeMatrixPath = codeMatrixFile.getAbsolutePath();
            String cdaMatrixPath = cdaMatrixFile.getAbsolutePath();

            Resource resource = new ClassPathResource("agv.py");
            File file = resource.getFile();
            ProcessBuilder processBuilder = new ProcessBuilder(
                    "python"
                    , file.getAbsolutePath()
                    , String.valueOf(avoidDistance)
                    , codeMatrixPath
                    , cdaMatrixPath
                    , redisProperties.getHost()
                    , redisProperties.getPassword()
                    , String.valueOf(redisProperties.getPort())
                    , String.valueOf(redisProperties.getIndex())
            );
            processBuilder.redirectErrorStream(true);

            try {
                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) {
                    System.out.println("Python script exited with error code: " + exitCode);
                    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);

//            mapDataDispatcher.printMatrix(waveMatrix);
            mapDataDispatcher.setWaveMatrix(lev, waveMatrix);
            // 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) {