From 495292b588949bcb2cb15eaa9657619aca7f5579 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 30 十月 2024 16:18:41 +0800
Subject: [PATCH] 修改降雨文件的jar

---
 src/main/java/com/se/simu/service/GedbService.java |   97 ++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 84 insertions(+), 13 deletions(-)

diff --git a/src/main/java/com/se/simu/service/GedbService.java b/src/main/java/com/se/simu/service/GedbService.java
index 72b40a9..47e7584 100644
--- a/src/main/java/com/se/simu/service/GedbService.java
+++ b/src/main/java/com/se/simu/service/GedbService.java
@@ -177,6 +177,15 @@
         }
     }
 
+    public boolean queryBboxCount(String token, GeDb db, List<GeLayer> layers) {
+        for (GeLayer layer : layers) {
+            int count = getCount(token, db, layer);
+            if (0 == count) return false;
+        }
+
+        return true;
+    }
+
     private int getCount(String token, GeDb db, GeLayer layer) {
         Map<String, Object> map = new HashMap<>(6);
         map.put("token", token);
@@ -217,27 +226,84 @@
 
         List<GeLayer> layers = getLayers(token, db);
         queryData(token, db, layers);
+        checkData(layers);
         createShps(basePath, layers);
-
-
         createZoneShp(basePath, data, db.getSpatialReference());
         if (data.getPid() > 0) {
             createFloodShp(basePath, data, db.getSpatialReference());
         }
     }
 
+    private void checkData(List<GeLayer> layers) {
+        GeLayer point = getLayerByName(layers, config.getLayerNames().get(0));
+        GeLayer line = getLayerByName(layers, config.getLayerNames().get(1));
+
+        List<String> bsm = getValues(point, "bsm");
+        List<String> bsm2 = new ArrayList<>(bsm);
+        List<String> qdbsm = getValues(line, "qdbsm");
+        List<String> qdbsm2 = new ArrayList<>(qdbsm);
+        List<String> zdbsm = getValues(line, "zdbsm");
+        List<String> zdbsm2 = new ArrayList<>(zdbsm);
+
+        qdbsm.removeAll(bsm2);
+        zdbsm.removeAll(bsm2);
+        removeValues(line, "qdbsm", qdbsm);
+        removeValues(line, "zdbsm", zdbsm);
+
+        qdbsm = getValues(line, "qdbsm");
+        zdbsm = getValues(line, "zdbsm");
+        bsm.removeAll(qdbsm);
+        bsm.removeAll(zdbsm);
+        removeValues(point, "bsm", bsm);
+
+        GeLayer juncLayer = new GeLayer(point, filterLayerData(point.getData()));
+        juncLayer.setName("闆嗘按鐐�");
+        juncLayer.setShpName(config.getJunctionName());
+        layers.add(juncLayer);
+    }
+
+    private GeLayer getLayerByName(List<GeLayer> layers, String name) {
+        return layers.stream().filter(a -> a.getName().equals(name)).findFirst().orElse(null);
+    }
+
+    private List<String> getValues(GeLayer layer, String field) {
+        JSONArray data = layer.getData();
+        List<String> list = new ArrayList<>();
+
+        int i = 0;
+        while (i < data.size()) {
+            JSONObject obj = data.getJSONObject(i).getJSONObject("properties");
+            if (StringHelper.isEmpty(obj.getStr(field))) {
+                data.remove(i);
+                continue;
+            }
+            list.add(obj.getStr(field));
+            i++;
+        }
+
+        return list;
+    }
+
+    private void removeValues(GeLayer layer, String field, List<String> values) {
+        if (CollectionUtils.isEmpty(values)) return;
+
+        int i = 0;
+        JSONArray data = layer.getData();
+        while (i < data.size()) {
+            JSONObject obj = data.getJSONObject(i).getJSONObject("properties");
+            if (values.contains(obj.getStr(field))) {
+                data.remove(i);
+                continue;
+            }
+            i++;
+        }
+    }
+
     private void createShps(String basePath, List<GeLayer> layers) throws Exception {
         for (GeLayer layer : layers) {
             String path = basePath + File.separator + layer.getShpName();
-            if (!ShpHelper.createShp(path, layer)) {
+            if (layer.getData().isEmpty() || !ShpHelper.createShp(path, layer)) {
                 throw new Exception(layer.getName() + "锛屽垱寤篠hapeFile鏂囦欢澶辫触锛�");
-            }
-            // 绠$綉闆嗘按鐐�
-            if (layer.getShpName().equals(config.getShpNames().get(0))) {
-                GeLayer juncLayer = new GeLayer(layer, filterLayerData(layer.getData()));
-                if (!ShpHelper.createShp(basePath + File.separator + config.getJunctionName(), juncLayer)) {
-                    throw new Exception("闆嗘按鐐�" + "锛屽垱寤篠hapeFile鏂囦欢澶辫触锛�");
-                }
             }
         }
     }
@@ -256,12 +322,17 @@
     }
 
     private void createZoneShp(String basePath, DataPo data, SpatialReference sr) {
-        String filePath = basePath + File.separator + "";
-        ShpHelper.createShp(filePath, sr, data.getMinx(), data.getMiny(), data.getMaxx(), data.getMaxy());
+        String filePath = basePath + File.separator + config.getZoneName();
+        ShpHelper.createShp(filePath, null, sr, data.getMinx(), data.getMiny(), data.getMaxx(), data.getMaxy());
     }
 
     private void createFloodShp(String basePath, DataPo data, SpatialReference sr) {
-        //
+        String filePath = basePath + File.separator + config.getBarrierName();
+        Map<String, Object> map = new HashMap<>();
+        map.put("height", data.getFloodHeight());
+        map.put("type", data.getFloodType());
+
+        ShpHelper.createShp(filePath, map, sr, data.getFloodMinx(), data.getFloodMiny(), data.getFloodMaxx(), data.getFloodMaxy());
     }
 
     public void copeDem(String token, DataPo data) throws Exception {

--
Gitblit v1.9.3