| package com.zy.acs.fake.service; | 
|   | 
| import com.alibaba.fastjson.JSON; | 
| import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; | 
| import com.zy.acs.common.constant.RedisConstant; | 
| import com.zy.acs.common.utils.RedisSupport; | 
| import com.zy.acs.fake.domain.DynamicNode; | 
| import com.zy.acs.fake.entity.Code; | 
| import com.zy.acs.fake.enums.CodeNodeType; | 
| import com.zy.acs.fake.enums.DynamicNodeType; | 
| import com.zy.acs.fake.enums.MapNodeType; | 
| import com.zy.acs.framework.common.Cools; | 
| import com.zy.acs.framework.exception.CoolException; | 
| import lombok.extern.slf4j.Slf4j; | 
|   | 
| import java.util.ArrayList; | 
| import java.util.List; | 
| import java.util.Optional; | 
|   | 
| /** | 
|  * Created by vincent on 6/6/2024 | 
|  */ | 
| @Slf4j | 
| public class MapDataDispatcher { | 
|   | 
|     public static final Integer MAP_DEFAULT_LEV = 1; | 
|   | 
|     private final RedisSupport redis = RedisSupport.defaultRedisSupport; | 
|   | 
|     public static final Boolean CACHE_CODE_MATRIX = Boolean.FALSE; | 
|   | 
|     private String[][] codeMatrix; | 
|   | 
|     private final CodeService codeService; | 
|   | 
|     private final RouteService routeService; | 
|   | 
|     public MapDataDispatcher(CodeService codeService, RouteService routeService) { | 
|         this.codeService = codeService; | 
|         this.routeService = routeService; | 
|     } | 
|   | 
|     public DynamicNode[][] getDynamicMatrix(Integer lev) { | 
|         lev = Optional.ofNullable(lev).orElse(MAP_DEFAULT_LEV); | 
|         DynamicNode[][] dynamicMatrix; | 
|         String dynamicMatrixStr = redis.getValue(RedisConstant.AGV_MAP_ASTAR_DYNAMIC_FLAG, String.valueOf(lev)); | 
|         if (Cools.isEmpty(dynamicMatrixStr)) { | 
|             throw new CoolException("the floor " + lev + " cannot found dynamic matrix !!!"); | 
|         } else { | 
|             dynamicMatrix = JSON.parseObject(dynamicMatrixStr, DynamicNode[][].class); | 
|         } | 
|         if (null == dynamicMatrix) { | 
|             throw new CoolException("the floor " + lev + " cannot found dynamic matrix !!!"); | 
|         } | 
|         return dynamicMatrix; | 
|     } | 
|   | 
|     public void setDynamicMatrix(Integer lev, DynamicNode[][] dynamicMatrix) { | 
|         redis.setValue(RedisConstant.AGV_MAP_ASTAR_DYNAMIC_FLAG, String.valueOf(lev), JSON.toJSONString(dynamicMatrix)); | 
|     } | 
|   | 
|     public String[][] getCodeMatrix(Integer lev) { | 
|         lev = Optional.ofNullable(lev).orElse(MAP_DEFAULT_LEV); | 
|         String[][] codeMatrix; | 
|         if (!CACHE_CODE_MATRIX) { | 
|             if (null == this.codeMatrix) { | 
|                 codeMatrix = this.initCodeMatrix(lev); | 
|                 setCodeMatrix(lev, codeMatrix); | 
|             } else { | 
|                 codeMatrix = this.codeMatrix; | 
|             } | 
|         } else { | 
|             String codeMatrixStr = redis.getValue(RedisConstant.AGV_MAP_ASTAR_CODE_FLAG, String.valueOf(lev)); | 
|             if (Cools.isEmpty(codeMatrixStr)) { | 
|                 codeMatrix = this.initCodeMatrix(lev); | 
|                 setCodeMatrix(lev, codeMatrix); | 
|             } else { | 
|                 codeMatrix = JSON.parseObject(codeMatrixStr, String[][].class); | 
|             } | 
|         } | 
|         if (null == codeMatrix) { | 
|             throw new CoolException("the floor " + lev + " cannot found code matrix !!!"); | 
|         } | 
|         return codeMatrix; | 
|     } | 
|   | 
|     public void setCodeMatrix(Integer lev, String[][] codeMatrix) { | 
|         if (!CACHE_CODE_MATRIX) { | 
|             this.codeMatrix = codeMatrix; | 
|         } else { | 
|             redis.setValue(RedisConstant.AGV_MAP_ASTAR_CODE_FLAG, String.valueOf(lev), JSON.toJSONString(codeMatrix)); | 
|         } | 
|     } | 
|   | 
|   | 
|   | 
|   | 
|     public synchronized String[][] initCodeMatrix(Integer lev) { | 
|         log.info("【ASTAR】正在计算矩阵数据......"); | 
|         lev = Optional.ofNullable(lev).orElse(MAP_DEFAULT_LEV); | 
|         List<Code> codeList = codeService.list(new LambdaQueryWrapper<Code>() | 
|                 .eq(Code::getStatus, 1) | 
|                 .orderByAsc(Code::getY) | 
|                 .orderByAsc(Code::getX) | 
|         ); | 
| //        int xCount = codeService.selectDistinctCountFromX(); | 
| //        int yCount = codeService.selectDistinctCountFromY(); | 
| //        // [][] 第一个是排的数量 第二个是列的数量 | 
| //        String[][] codeMatrix = new String[yCount][xCount]; | 
|   | 
|         List<Double> xIdxList = new ArrayList<>(); | 
|         List<Double> yIdxList = new ArrayList<>(); | 
|         for (Code code : codeList) { | 
|             Double x = code.getX(); | 
|             Double y = code.getY(); | 
|             if (null == x || null == y) { continue; } | 
|             if (!xIdxList.contains(x)) { | 
|                 xIdxList.add(x); | 
|             } | 
|             if (!yIdxList.contains(y)) { | 
|                 yIdxList.add(y); | 
|             } | 
|         } | 
|         xIdxList.sort((o1, o2) -> (int) ((o1 * 100) - (o2 * 100))); | 
|         yIdxList.sort((o1, o2) -> (int) ((o1 * 100) - (o2 * 100))); | 
|   | 
|         // [][] 第一个是排的数量 第二个是列的数量 | 
|         String[][] codeMatrix = new String[yIdxList.size()][xIdxList.size()]; | 
|   | 
|         int idxX = -1; | 
|         int idxY = -1; | 
|         double lastX = -1; | 
|         double lastY = -1; | 
|         for (Code code : codeList) { | 
|             Double x = code.getX(); | 
|             Double y = code.getY(); | 
| //            if (y != lastY) { | 
| //                idxY ++; | 
| //                idxX = 0; | 
| //                lastY = y; | 
| //                lastX = x; | 
| //            } | 
| //            if (x != lastX) { | 
| //                idxX++; | 
| //                lastX = x; | 
| //            } | 
| //            codeMatrix[idxY][idxX] = code.getData(); | 
|   | 
|             codeMatrix[yIdxList.indexOf(y)][xIdxList.indexOf(x)] = code.getData(); | 
|         } | 
|   | 
|         for (int i = 0; i < codeMatrix.length; i++) { | 
|             for (int j = 0; j < codeMatrix[i].length; j++) { | 
|                 if (codeMatrix[i][j] == null) { | 
|                     codeMatrix[i][j] = CodeNodeType.NONE.val; | 
|                 } | 
|             } | 
|         } | 
|   | 
|         printMatrix(codeMatrix); | 
|   | 
|         return codeMatrix; | 
|     } | 
|   | 
|     public int[] getCodeMatrixIdx(Integer lev, String codeData) { | 
|         if (Cools.isEmpty(codeData)) { | 
|             return null; | 
|         } | 
|         lev = Optional.ofNullable(lev).orElse(MAP_DEFAULT_LEV); | 
|         String[][] codeMatrix = this.getCodeMatrix(lev); | 
|   | 
|         for (int i = 0; i < codeMatrix.length; i++) { | 
|             for (int j = 0; j < codeMatrix[i].length; j++) { | 
|                 if (codeMatrix[i][j].equals(codeData)) { | 
|                     return new int[]{i, j}; | 
|                 } | 
|             } | 
|         } | 
|         return null; | 
|     } | 
|   | 
|   | 
|     public void modifyDynamicMatrix(Integer lev, List<String> codeList, String vehicle) { | 
|         this.modifyDynamicMatrix(lev, codeList, vehicle, false); | 
|     } | 
|   | 
|     public synchronized void modifyDynamicMatrix(Integer lev, List<String> codeList, String vehicle, boolean reset) { | 
|         lev = Optional.ofNullable(lev).orElse(MAP_DEFAULT_LEV); | 
|         DynamicNode[][] dynamicMatrix = getDynamicMatrix(lev); | 
|   | 
|         if (!reset) { | 
|             if (Cools.isEmpty(codeList, vehicle)) { | 
|                 return; | 
|             } | 
|             int serial = 1; | 
|             for (String code : codeList) { | 
|                 int[] node = getCodeMatrixIdx(lev, code); | 
|                 dynamicMatrix[node[0]][node[1]] = new DynamicNode(vehicle, serial); | 
|                 serial++; | 
|             } | 
|         } else { | 
|             if (Cools.isEmpty(vehicle)) { | 
|                 return; | 
|             } | 
|             for (int i = 0; i < dynamicMatrix.length; i++) { | 
|                 for (int j = 0; j < dynamicMatrix[i].length; j++) { | 
|                     DynamicNode dynamicNode = dynamicMatrix[i][j]; | 
|                     if (vehicle.equals(dynamicNode.getVehicle())) { | 
|                         dynamicMatrix[i][j] = new DynamicNode(DynamicNodeType.ACCESS.val); | 
|                     } | 
|                 } | 
|             } | 
|         } | 
|   | 
|         setDynamicMatrix(lev, dynamicMatrix); | 
|     } | 
|   | 
|     public void clearDynamicMatrixByCodeList(Integer lev, List<String> codeList) { | 
|         this.modifyDynamicMatrix(lev, codeList, DynamicNodeType.ACCESS.val); | 
|     } | 
|   | 
|     public int[][] filterMapData(int[][] mapMatrix, Integer lev, List<String> lockNodes) { | 
|         String[][] codeMatrix = getCodeMatrix(lev); | 
|         for (int i = 0; i < codeMatrix.length; i++) { | 
|             for (int j = 0; j < codeMatrix[i].length; j++) { | 
|                 String code = codeMatrix[i][j]; | 
|                 if (lockNodes.contains(code)) { | 
|                     mapMatrix[i][j] = MapNodeType.LOCKED.val; | 
|                 } | 
|             } | 
|         } | 
|         return mapMatrix; | 
|     } | 
|   | 
|     public <T> void printMatrix(T[][] matrix) { | 
|         for (T[] list : matrix) { | 
|             for (T t : list) { | 
|                 System.out.format("%10s", t); | 
|             } | 
|             System.out.println(); | 
|         } | 
|     } | 
|   | 
|     public void printMatrix(int[][] matrix) { | 
|         for (int[] list : matrix) { | 
|             for (int t : list) { | 
|                 System.out.format("%10s", t); | 
|             } | 
|             System.out.println(); | 
|         } | 
|     } | 
|   | 
| } |