mid、mif文件操作工具类

秒速五厘米 2021-10-01 05:42 481阅读 0赞

# **微信搜索：“二十同学” 公众号，欢迎关注一条不一样的成长之路** #

# **读写mid、mif文件操作工具类** #

装载mid文件,将mid文件里的数据变为  List<List<String>> 形式的

public List<List<String>> loadMidFile(File file) {
            List<List<String>> midRows= new ArrayList<>();
            try {
                BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(file), "gbk"));
                String line;
                while ((line = reader.readLine()) != null) {
                    List<String> cols = Arrays.asList(line.split(",",-1));
                    if (cols.size() > 0) {
                        for (int i = 0; i < cols.size(); i++) {
                            String value = cols.get(i).replace("\"", "");
                            cols.set(i, value);
                        }
                        midRows.add(cols);
                    }
                }
                reader.close();
            } catch (FileNotFoundException e) {
                logger.error("loadMidFile()：", e);
            } catch (IOException e) {
                logger.error("loadMidFile()：", e);
            }
            return midRows;
        }

装载mif文件

public List<String> loadMifFile(File file, Boolean isLoadMifData) {
            //colField 为mid文件里各个值对应的字段
            List<String> colField = new ArrayList<>();
            try {
                BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(file), "gbk"));
                String line;
                while ((line = reader.readLine()) != null) {
                    if (line.toUpperCase().startsWith("COLUMNS")) {
                        int num = Integer.parseInt(line.substring("Columns ".length()));
                        for (int i = 0; i < num; i++) {
                            line = reader.readLine().trim();
                            List<String> cols = Arrays.asList(line.split(","));
                            if (cols.size() > 0) {
                                colField.add(cols.get(0).split(" ")[0].trim());
                            }
                        }
    
                        if (isLoadMifData) {
                            //装载mif里的geo数据(点，线，面不同类型)
                            loadMifFileGeoDatas(reader);
                        }
    
                        break;
                    }
                }
                reader.close();
    
            } catch (Exception e) {
                logger.error("loadMifFile()：", e);
            }
    
            return colField;
        }

//如果geo数据为面
    public Map<Integer, List<List<List<Double>>>> loadMifFileGeoDatas(BufferedReader reader) {
            Map<Integer, List<List<List<Double>>>> regions = new HashMap<>();
            Integer rowIndex = -1;
            try {
                String line;
                while ((line = reader.readLine()) != null) {
                    line = line.trim();
                    if (line.toUpperCase().contains("NONE")){
                        regions.put(++rowIndex, new ArrayList<>());
                    }
                    if (line.toUpperCase().startsWith("REGION")) {
                        int num = Integer.parseInt(line.substring("Region ".length()));
                        List<List<List<Double>>> dataGroups = new ArrayList<>();
    
                        for (int i = 0; i < num; i++) {
                            line = reader.readLine().trim();
                            int rowCount = Integer.parseInt(line);
                            List<List<Double>> datas = new ArrayList<>();
                            for (int j = 0; j < rowCount; j++) {
                                line = reader.readLine().trim();
                                List<String> xy = Arrays.asList(line.split(" "));
                                List<Double> points = new ArrayList<>();
    
                                if (xy.size() == 2) {
                                    points.add(Double.parseDouble(xy.get(0)));
                                    points.add(Double.parseDouble(xy.get(1)));
                                    datas.add(points);
                                }
                            }
                            dataGroups.add(datas);
                        }
                        regions.put(++rowIndex, dataGroups);
                    }
                }
                reader.close();
    
            } catch (Exception e) {
                logger.error("loadMifFileGeoDatas()：", e);
            }
            return regions;
        }
    
    
     public LinkedHashMap<String, Object> getGeometryMap(Map<Integer, List<List<List<Double>>>> regions,Integer rowIdx) {
    
            if (regions != null) {
                List<List<List<Double>>> coordinates = regions.getOrDefault(rowIdx, null);
                LinkedHashMap<String, Object> geometryMap = new LinkedHashMap<String, Object>() {
       {
                    put("type", "Polygon");
                    put("coordinates", coordinates);
                }};
    
                return geometryMap;
            }
    
            return null;
    
        }

//如果geo数据为点
    public Map<Integer, List<Double>> loadMifFileGeoDatas(BufferedReader reader) {
            Map<Integer, List<Double>> pointRows = new HashMap<>();
            Integer rowIndex = -1;
            try {
                String line;
                while ((line = reader.readLine()) != null) {
                    line = line.trim();
                    if (line.toUpperCase().contains("NONE")){
                        pointRows.put(++rowIndex, new ArrayList<>());
                    }
                    if (line.toUpperCase().startsWith("POINT")) {
                        List<String> point = Arrays.asList(line.split(" "));
                        if (point.size() == 3) {
                            List<Double> point1 = new ArrayList<>();
                            point1.add(Double.parseDouble(point.get(1)));
                            point1.add(Double.parseDouble(point.get(2)));
                            pointRows.put(++rowIndex, point1);
                        }
                    }
                }
                reader.close();
    
            } catch (Exception e) {
                logger.error("loadMifFileGeoDatas()：", e);
            }
            
            return pointRows;
    
        }
    
    
     public LinkedHashMap<String, Object> getGeometryMap(Map<Integer, List<Double>> pointRows,Integer rowIdx) {
            if (pointRows!= null) {
                List<Double> pointRow = pointRows.getOrDefault(rowIdx, null);
                LinkedHashMap<String, Object> geometryMap = new LinkedHashMap<String, Object>() {
       {
                    put("type", "Point");
                    put("coordinates", pointRow);
                }};
    
                return geometryMap;
            }
            return null;
        }

//如果geo数据为线
    public Map<Integer, List<List<Double>>> loadMifFileGeoDatas(BufferedReader reader)  {
            Map<Integer, List<List<Double>>> lineRows = new HashMap<>();
            Integer rowIndex = -1;
            try {
                String line;
                while ((line = reader.readLine()) != null) {
                    List<List<Double>> lines = new ArrayList<>();
                    line = line.trim();
                    if (line.toUpperCase().contains("NONE")){
                        lineRows.put(++rowIndex, lines);
                    }
                    if (line.toUpperCase().startsWith("LINE")) {
                        List<String> point = Arrays.asList(line.split(" "));
                        if (point.size() == 5) {
                            List<Double> point1 = new ArrayList<>();
                            point1.add(Double.parseDouble(point.get(1)));
                            point1.add(Double.parseDouble(point.get(2)));
                            lines.add(point1);
    
                            List<Double> point2 = new ArrayList<>();
                            point2.add(Double.parseDouble(point.get(3)));
                            point2.add(Double.parseDouble(point.get(4)));
                            lines.add(point2);
                        }
                        lineRows.put(++rowIndex, lines);
                    }
                    if (line.startsWith("Pline")) {
                        int num = Integer.parseInt(line.substring("Pline ".length()));
                        for (int i = 0; i < num; i++) {
                            line = reader.readLine().trim();
                            List<String> point = Arrays.asList(line.split(" "));
                            if (point.size() == 2) {
                                List<Double> points = new ArrayList<>();
                                points.add(Double.parseDouble(point.get(0)));
                                points.add(Double.parseDouble(point.get(1)));
                                lines.add(points);
                            }
                        }
                        lineRows.put(++rowIndex, lines);
                    }
                }
                reader.close();
    
            } catch (Exception e) {
                logger.error("loadMifFileGeoDatas()：", e);
            }
            return lineRows;
        }
    
    public LinkedHashMap<String, Object> getGeometryMap(Map<Integer, List<List<Double>>> lineRows,Integer rowIdx) {
            if (lineRows!= null) {
                List<List<Double>> line = lineRows.getOrDefault(rowIdx, null);
                LinkedHashMap<String, Object> geometryMap = new LinkedHashMap<String, Object>() {
       {
                    put("type", "LineString");
                    put("coordinates", line);
                }};
    
                return geometryMap;
            }
            return null;
        }