#

vincentlu

2025-01-13 89c7f6e5bcc21b0e8f83a2bc6d680e2ffe431e6f

 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,113 +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) {
                    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);

            // 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) {
@@ -209,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) {
@@ -231,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);
@@ -287,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());
        }
    }