From b6899cd6d1bafb1c349f1fbd190e6632183719f5 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 01 十一月 2024 20:44:00 +0800
Subject: [PATCH] 11
---
src/main/java/com/se/simu/service/GedbService.java | 112 ++++++++++++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 109 insertions(+), 3 deletions(-)
diff --git a/src/main/java/com/se/simu/service/GedbService.java b/src/main/java/com/se/simu/service/GedbService.java
index a286a96..47e7584 100644
--- a/src/main/java/com/se/simu/service/GedbService.java
+++ b/src/main/java/com/se/simu/service/GedbService.java
@@ -18,7 +18,7 @@
import org.gdal.gdal.WarpOptions;
import org.gdal.gdal.gdal;
import org.gdal.gdalconst.gdalconst;
-import org.springframework.beans.factory.annotation.Value;
+import org.gdal.osr.SpatialReference;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;
import org.springframework.web.client.RestTemplate;
@@ -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,18 +226,115 @@
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 = String.format("%s\\%s.shp", basePath, layer.getShpName());
- if (!ShpHelper.createShp(path, layer)) {
+ String path = basePath + File.separator + layer.getShpName();
+ if (layer.getData().isEmpty() || !ShpHelper.createShp(path, layer)) {
throw new Exception(layer.getName() + "锛屽垱寤篠hapeFile鏂囦欢澶辫触锛�");
}
}
}
+ private JSONArray filterLayerData(JSONArray data) {
+ JSONArray arr = new JSONArray();
+ String[] strs = config.getJunctionFilter().split("=");
+ for (int i = 0, c = data.size(); i < c; i++) {
+ JSONObject obj = data.getJSONObject(i).getJSONObject("properties");
+ if (strs[1].equals(obj.getStr(strs[0]))) {
+ arr.put(data.getJSONObject(i));
+ }
+ }
+
+ return arr;
+ }
+
+ private void createZoneShp(String basePath, DataPo data, SpatialReference sr) {
+ 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 {
GeDb fileDb = getFileDb(token);
String fileId = getFileId(token, fileDb.getDbid());
--
Gitblit v1.9.3