From 0890b7861feae74bdcfd1851e577db6b9f31d484 Mon Sep 17 00:00:00 2001
From: xingjinshuang <xingjs@qq.com>
Date: 星期四, 20 二月 2025 14:40:39 +0800
Subject: [PATCH] @xingjs@20250220@添加处理sww相关接口类
---
src/main/java/com/se/simu/service/WaterService.java | 223 ++++++++++++++++++++++++++++++++-----------------------
1 files changed, 129 insertions(+), 94 deletions(-)
diff --git a/src/main/java/com/se/simu/service/WaterService.java b/src/main/java/com/se/simu/service/WaterService.java
index 19a4736..46c7c96 100644
--- a/src/main/java/com/se/simu/service/WaterService.java
+++ b/src/main/java/com/se/simu/service/WaterService.java
@@ -1,32 +1,29 @@
package com.se.simu.service;
+import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONUtil;
import com.se.simu.config.PropertiesConfig;
import com.se.simu.domain.po.DataPo;
+import com.se.simu.domain.po.PondingPo;
import com.se.simu.domain.po.SimuPo;
import com.se.simu.domain.vo.*;
import com.se.simu.helper.GdalHelper;
+import com.se.simu.helper.StringHelper;
import lombok.extern.slf4j.Slf4j;
import org.gdal.gdal.Dataset;
import org.gdal.gdal.gdal;
import org.gdal.gdalconst.gdalconst;
+import org.gdal.ogr.*;
import org.gdal.osr.SpatialReference;
-import org.gdal.osr.osr;
import org.springframework.stereotype.Service;
+import org.springframework.util.CollectionUtils;
import javax.annotation.Resource;
import java.io.File;
import java.io.FileInputStream;
-import java.util.ArrayList;
-import java.util.Arrays;
import java.util.List;
+import java.util.stream.Collectors;
-/**
- * 鍐呮稘鏈嶅姟绫�
- *
- * @author WWW
- * @date 2024-07-16
- */
@Slf4j
@Service
@SuppressWarnings("ALL")
@@ -34,12 +31,9 @@
@Resource
PropertiesConfig config;
- /**
- * 鑾峰彇鍏冩暟鎹俊鎭�
- */
- public byte[] getLayerJson(String serviceName) {
+ public byte[] getson(String serviceName, String json) {
try {
- String filePath = config.getOutPath() + File.separator + serviceName + File.separator + "layer.json";
+ String filePath = config.getOutPath() + File.separator + serviceName + File.separator + json;
File dat = new File(filePath);
if (!dat.exists()) {
@@ -54,85 +48,27 @@
return bytes;
} catch (Exception ex) {
+ log.error(ex.getMessage(), ex);
return null;
}
}
- /**
- * 鑾峰彇鍦板舰楂樺害鍥�
- */
public String getTerraMap(String serviceName, Integer width, Integer height) {
return config.getOutPath() + File.separator + serviceName + File.separator + "terrain" + File.separator + width + "_" + height + ".png";
}
- /**
- * 鑾峰彇姘撮潰楂樺害鍥�
- */
public String getWaterMap(String serviceName, Integer width, Integer height, Long timestamp) {
return config.getOutPath() + File.separator + serviceName + File.separator + "waters" + File.separator + timestamp + File.separator + width + "_" + height + ".png";
}
- /**
- * 鑾峰彇姘存祦鍚戞祦閫熷浘
- */
public String getFlowMap(String serviceName, Integer width, Integer height, Long timestamp) {
return config.getOutPath() + File.separator + serviceName + File.separator + "flows" + File.separator + timestamp + File.separator + width + "_" + height + ".png";
}
- /**
- * 鑾峰彇鍥惧眰 *
- */
- public Layer getLayer(String serviceName) {
- Layer layer = new Layer();
- layer.setVersion(config.getVer());
- layer.setDuration(new Duration(1719812810225L, 1719812810225L));
- layer.setExtension(new Extension(2.11062743358, 0.53812160220, 2.11070827834, 0.53819799453, 1.151, 38.83));
-
- List<Integer[]> sizes = new ArrayList<>();
- sizes.add(new Integer[]{64, 64});
- sizes.add(new Integer[]{128, 128});
- sizes.add(new Integer[]{256, 256});
- sizes.add(new Integer[]{512, 512});
- sizes.add(new Integer[]{1024, 1024});
- sizes.add(new Integer[]{2048, 2048});
- layer.setTerrain(new Terrain(sizes));
-
- List<Long> data = new ArrayList<>(Arrays.asList(1719812812225L, 1719812812225L, 1719812812225L, 1719812812225L, 1719812812225L, 1719812812225L));
- layer.setWaters(new Water(data));
-
- return layer;
- }
-
- /**
- * 鑾峰彇闄嶆按鏇茬嚎鏂囦欢鏇茬嚎鍥�
- */
- public byte[] getRainfall(String serviceName) {
- try {
- String filePath = config.getOutPath() + File.separator + serviceName + File.separator + "rainfall.json";
-
- File rainfall = new File(filePath);
- if (!rainfall.exists()) {
- return null;
- }
-
- byte[] bytes = new byte[(int) rainfall.length()];
-
- FileInputStream fs = new FileInputStream(filePath);
- fs.read(bytes);
- fs.close();
-
- return bytes;
- } catch (Exception ex) {
- return null;
- }
- }
-
- /**
- * 鏍规嵁鍧愭爣鏌ヨ绉按娣卞害:gdalconst.GA_Update
- */
public Double getWaterHeight(SimuPo simu, double x, double y, Long timestamp) {
String filePath = config.getOutPath() + File.separator + simu.getServiceName() + File.separator + "waters"
+ File.separator + timestamp + File.separator + "water.tif";
+ if (!FileUtil.exist(filePath)) return null;
Dataset ds = null;
try {
@@ -152,11 +88,10 @@
return null;
}
- double[] values = new double[1];
- ds.GetRasterBand(1).ReadRaster(XY[0], XY[1], 1, 1, values);
- double val = values[0];
+ double[] vals = new double[1];
+ ds.GetRasterBand(1).ReadRaster(XY[0], XY[1], 1, 1, vals);
- return isValid(val) ? val : null;
+ return isValid(vals[0]) ? vals[0] : null;
} catch (Exception ex) {
log.error(ex.getMessage(), ex);
return null;
@@ -168,25 +103,10 @@
private SpatialReference getSpatialRef(SimuPo simu) {
DataPo data = JSONUtil.toBean(simu.getData(), DataPo.class);
- SpatialReference sr = new SpatialReference();
- sr.ImportFromEPSG(data.getEpsg());
- sr.SetAxisMappingStrategy(osr.OAMS_TRADITIONAL_GIS_ORDER);
-
- return sr;
+ return data.getSpatialReference();
}
- /**
- * 灏嗗湴鍥惧潗鏍囪浆鎹负鏍呮牸鍍忕礌鍧愭爣
- *
- * @param gt 浠垮皠鍙樻崲鍙傛暟
- * @param x 妯潗鏍�
- * @param y 绾靛潗鏍�
- * @return 鍍忕礌鍧愭爣
- */
public int[] coordinates2ColRow(double[] gt, double x, double y) {
- int[] ints = new int[2];
-
- // 鍚戜笅鍙栨暣,濡傛灉鍚戜笂鍙栨暣浼氬鑷磋绠楃粨鏋滃亸澶э紝浠庤�屽湪鍚庨潰璇诲彇鍒伴偦杩戝儚鍏冪殑鏁版嵁
//Double col = Math.floor(((y - gt[3]) * gt[1] - (x - gt[0]) * gt[4]) / (gt[5] * gt[1] - gt[2] * gt[4]));
Double col = Math.floor((y * gt[1] - x * gt[4] + gt[0] * gt[4] - gt[3] * gt[1]) / (gt[5] * gt[1] - gt[2] * gt[4]));
Double row = Math.floor((x - gt[0] - col * gt[2]) / gt[1]);
@@ -197,4 +117,119 @@
public static boolean isValid(double val) {
return !Double.isNaN(val) && val > Integer.MIN_VALUE;
}
+
+ public Double getWaterArea(SimuPo simu, double x, double y, Long timestamp) {
+ List<PondingPo> list = readWaterJson(simu.getServiceName(), timestamp);
+ if (CollectionUtils.isEmpty(list)) return null;
+
+ //DataPo data = JSONUtil.toBean(simu.getData(), DataPo.class);
+ //SpatialReference sr = GdalHelper.createSpatialReference(data.getEpsg());
+
+ Geometry p = new Geometry(ogr.wkbPoint);
+ p.AddPoint_2D(x, y);
+ p.AssignSpatialReference(GdalHelper.SR4326);
+
+ for (PondingPo po : list) {
+ if (StringHelper.isEmpty(po.getPolygon())) continue;
+
+ Geometry polygon = Geometry.CreateFromWkt(po.getPolygon());
+ polygon.AssignSpatialReference(GdalHelper.SR4326);
+ if (polygon.Contains(p)) {
+ return po.getArea();
+ }
+ }
+
+ return null;
+ }
+
+ private List<PondingPo> readWaterJson(String serviceName, Long timestamp) {
+ String filePath = config.getOutPath() + File.separator + serviceName + File.separator + "waters" + File.separator + timestamp + File.separator + "water.json";
+ String json = getText(filePath);
+ if (StringHelper.isEmpty(json)) {
+ return null;
+ }
+
+ return JSONUtil.toList(json, PondingPo.class);
+ }
+
+ public Double getWaterArea2(SimuPo simu, double x, double y, Long timestamp) {
+ String filePath = config.getOutPath() + File.separator + simu.getServiceName() + File.separator + "waters"
+ + File.separator + timestamp + File.separator + "water.geojson";
+ if (!FileUtil.exist(filePath)) return null;
+
+ Driver driver = null;
+ DataSource dataSource = null;
+ org.gdal.ogr.Layer layer = null;
+ try {
+ driver = ogr.GetDriverByName("GeoJSON");
+ if (null == driver) return null;
+
+ DataSource ds = driver.Open(filePath);
+ if (null == ds) return null;
+
+ layer = ds.GetLayer(0);
+ double[] xy = GdalHelper.fromWgs84(layer.GetSpatialRef(), x, y);
+
+ Geometry p = new Geometry(ogr.wkbPoint);
+ p.AddPoint_2D(xy[0], xy[1]);
+ p.AssignSpatialReference(layer.GetSpatialRef());
+
+ for (long i = 0, d = layer.GetFeatureCount(); i < d; i++) {
+ Feature f = layer.GetFeature(i);
+ if (f.GetGeometryRef().Intersects(p)) {
+ /*f.GetFieldAsDouble("val");
+ Geometry g = f.GetGeometryRef();
+ GdalHelper.fromWgs84(layer.GetSpatialRef(), g);
+ Double area= g.GetArea();*/
+
+ return f.GetGeometryRef().Area();
+ }
+ }
+
+ return null;
+ } catch (Exception ex) {
+ log.error(ex.getMessage(), ex);
+ return null;
+ } finally {
+ GdalHelper.delete(layer, dataSource, driver);
+ }
+ }
+
+ public List<BuildingDepthVo> getBuildingDepthBySeid(String serviceName, String seid) {
+ List<BuildingDepthVo> list = readBuildingJson(serviceName);
+ if (CollectionUtils.isEmpty(list)) return null;
+
+ return list.parallelStream()
+ .filter(b -> seid.equals(b.getId()))
+ .sorted((a, b) -> a.getTimestamp().compareTo(b.getTimestamp()))
+ .collect(Collectors.toList());
+ }
+
+ public List<BuildingDepthVo> getBuildingDepthByTime(String serviceName, Long timestamp) {
+ List<BuildingDepthVo> list = readBuildingJson(serviceName);
+ if (CollectionUtils.isEmpty(list)) {
+ return null;
+ }
+
+ return list.parallelStream().filter(b -> timestamp.equals(b.getTimestamp())).collect(Collectors.toList());
+ }
+
+ private List<BuildingDepthVo> readBuildingJson(String serviceName) {
+ String filePath = config.getOutPath() + File.separator + serviceName + File.separator + "building.json";
+ String json = getText(filePath);
+ if (StringHelper.isEmpty(json)) {
+ return null;
+ }
+
+ return JSONUtil.toList(json, BuildingDepthVo.class);
+ }
+
+ private String getText(String filePath) {
+ File file = new File(filePath);
+ if (!file.exists()) {
+ return null;
+ }
+
+ return FileUtil.readUtf8String(file);
+ }
}
--
Gitblit v1.9.3