From 256c1b584af7ed549e3efc3909c8c8b55a7a3331 Mon Sep 17 00:00:00 2001
From: dcb <dcb@163.com>
Date: 星期四, 08 五月 2025 17:27:28 +0800
Subject: [PATCH] 生成流速图片
---
src/main/java/com/se/nsl/service/ResolveService.java | 22 ++++++++++
src/main/java/com/se/nsl/service/TestService.java | 67 +++++++++++++++++++++++++++++----
src/main/resources/application-dev.yml | 4 +-
3 files changed, 82 insertions(+), 11 deletions(-)
diff --git a/src/main/java/com/se/nsl/service/ResolveService.java b/src/main/java/com/se/nsl/service/ResolveService.java
index 8c84496..c4f0d3a 100644
--- a/src/main/java/com/se/nsl/service/ResolveService.java
+++ b/src/main/java/com/se/nsl/service/ResolveService.java
@@ -240,6 +240,26 @@
* 璋冪敤zarr2tif
*/
private String callZarr2tif(SimuData data) throws Exception {
+ String zarr2tifBat = config.getZarr2tifBat();
+ File dir = new File(zarr2tifBat).getParentFile();
+ String configJson = new File(dir, "config.json").getAbsolutePath();
+
+ String cmd = String.format("%s \"%s\"", zarr2tifBat, configJson);
+ String res = callBat(cmd);
+
+ File resultTifDir = new File(dir, "result_tif");
+ File targetDir = new File(config.getInPath(), data.getInPath() + File.separator + "depth");
+ System.out.println("targetDir:" + targetDir);
+ File[] files = resultTifDir.listFiles();
+ for (File file : files) {
+ File target = new File(targetDir, file.getName());
+ Files.copy(file.toPath(), target.toPath());
+ }
+
+ return res;
+ }
+
+ /*private String callZarr2tif(SimuData data) throws Exception {
File uwBat = new File(config.getUwSolverBat());
String zarrFile = uwBat.getParent() + File.separator + "result.zarr";
String inPath = config.getInPath() + File.separator + data.getInPath();
@@ -249,7 +269,7 @@
String cmd = String.format("%s \"%s\" \"%s\" \"%s\" \"%s\"", config.getZarr2tifBat(), "depth", zarrFile, terrainFile, waterPath);
return callBat(cmd);
- }
+ }*/
private String callBat(String cmd) {
try {
diff --git a/src/main/java/com/se/nsl/service/TestService.java b/src/main/java/com/se/nsl/service/TestService.java
index 32bb1bf..07575f8 100644
--- a/src/main/java/com/se/nsl/service/TestService.java
+++ b/src/main/java/com/se/nsl/service/TestService.java
@@ -87,7 +87,7 @@
copeTerrain(dto, layer);
//copeBuilding(dto, layer);
List<BuildingDepthVo> buildings = copeWater(dto, layer);
- //copeFlow(dto, layer);
+ copeFlow(dto, layer);
copeLayerJson(dto, layer);
//copeRainFallJson(dto, layer);
//copeBuildingDepthJson(dto, buildings);
@@ -564,9 +564,8 @@
}
public void copeFlow(ResultDto dto, LayerDto layer) {
- List<String> files = getFiles(dto.getFlowPath(), ".tif");
- if (null == files || files.size() != layer.getWaters().getData().size()) return;
-
+ List<String> files = layer.getWaters().getFiles();
+ if (null == files || files.isEmpty()) return;
for (int i = 0, c = files.size(); i < c; i++) {
Dataset ds = null;
try {
@@ -603,17 +602,69 @@
ds = gdal.Open(tif, gdalconstConstants.GA_ReadOnly);
if (null == ds || 0 == ds.getRasterCount()) return;
+ float[] depthBuffer = new float[width * height];
+ float[] vxBuffer = new float[width * height];
+ float[] vyBuffer = new float[width * height];
+
+ ds.GetRasterBand(1).ReadRaster(0, 0, width, height, depthBuffer);
+ ds.GetRasterBand(2).ReadRaster(0, 0, width, height, vxBuffer);
+ ds.GetRasterBand(3).ReadRaster(0, 0, width, height, vyBuffer);
+
+ createFlowPng(depthBuffer, vxBuffer, vyBuffer, png, width, height);
+ } finally {
+ if (null != ds) ds.delete();
+ }
+ }
+
+ public void createFlowPng(float[] depthBuffer, float[] vxBuffer, float[] vyBuffer, String png, int width, int height) {
+ BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
+ for (int x = 0; x < width; x++) {
+ for (int y = 0; y < height; y++) {
+ int offset = x + y * width;
+ float depth = ComHelper.getFloatValue(depthBuffer[offset]);
+ float fx = ComHelper.getFloatValue(vxBuffer[offset]);
+ float fy = ComHelper.getFloatValue(vyBuffer[offset]);
+ if (Float.isNaN(fx) && Float.isNaN(fy) || (fx < MIN_VAL && fy < MIN_VAL)) continue;
+
+ if (depth == 0) {
+ fx = 0f;
+ fy = 0f;
+ } else {
+ fx = Float.isNaN(fx) ? 0 : fx / depth;
+ fy = Float.isNaN(fy) ? 0 : fy / depth;
+ }
+ double dr = Math.sqrt(Math.pow(fx, 2) + Math.pow(fy, 2));
+
+ int r = (int) (dr / 4 * 255);
+ int g = (int) ((fx / dr * 0.5 + 0.5) * 255);
+ int b = (int) ((fy / dr * 0.5 + 0.5) * 255);
+
+ Color color = new Color(ComHelper.getSafeValue(r), ComHelper.getSafeValue(g), ComHelper.getSafeValue(b), 127);
+ image.setRGB(x, y, color.getRGB());
+ }
+ }
+ ComHelper.savePng(image, png);
+ }
+
+/*
+ public void vxyTif2Png(LayerDto layer, String tif, String png, int width, int height) {
+ Dataset ds = null;
+ try {
+ ds = gdal.Open(tif, gdalconstConstants.GA_ReadOnly);
+ if (null == ds || 0 == ds.getRasterCount()) return;
+
float[] vxBuffer = new float[width * height], vyBuffer = new float[width * height];
- ds.GetRasterBand(1).ReadRaster(0, 0, width, height, vxBuffer);
- ds.GetRasterBand(2).ReadRaster(0, 0, width, height, vyBuffer);
+ ds.GetRasterBand(2).ReadRaster(0, 0, width, height, vxBuffer);
+ ds.GetRasterBand(3).ReadRaster(0, 0, width, height, vyBuffer);
createFlowPng(vxBuffer, vyBuffer, png, width, height);
} finally {
if (null != ds) ds.delete();
}
}
+*/
- public void createFlowPng(float[] vxBuffer, float[] vyBuffer, String png, int width, int height) {
+ /*public void createFlowPng(float[] vxBuffer, float[] vyBuffer, String png, int width, int height) {
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
for (int x = 0; x < width; x++) {
for (int y = 0; y < height; y++) {
@@ -635,7 +686,7 @@
}
}
ComHelper.savePng(image, png);
- }
+ }*/
public void copeLayerJson(ResultDto dto, LayerDto layer) throws IOException {
layer.getWaters().setFiles(null);
diff --git a/src/main/resources/application-dev.yml b/src/main/resources/application-dev.yml
index e60c586..a65cbba 100644
--- a/src/main/resources/application-dev.yml
+++ b/src/main/resources/application-dev.yml
@@ -142,8 +142,8 @@
solverBat: D:\other\simu\uwsolver\run_solver.bat
sww2tifBat: D:\other\simu\uwsolver\sww2tif.bat
uwSolverBat: D:\other\simu\CudaUWSolver.Demo.NoVis.20250430\start.bat
- zarr2tifBat: D:\other\simu\zarr2tif1.0\start.bat
- #zarr2tifBat: D:\other\simu\zarr2tif-2.0-mkl\start.bat
+# zarr2tifBat: D:\other\simu\zarr2tif1.0\start.bat
+ zarr2tifBat: D:\other\simu\zarr2tif-2.0-mkl\start.bat
createRainfall: '"C:\Program Files\Python310\python.exe" D:\terrait\NslServer\data\CreatRainfall.py'
rainfallTitle: Station Longitude Latitude Year Month Day Hour Minute Intensity
rainfallSite: beijing
--
Gitblit v1.9.3