From 22f80ba79735ab9897530b79e34889d2082cedbc Mon Sep 17 00:00:00 2001
From: wuww <252740454@qq.com>
Date: 星期四, 08 五月 2025 16:25:40 +0800
Subject: [PATCH] 开发根据参数生成降雨文件接口
---
src/main/java/com/se/nsl/service/ResolveService.java | 94 +++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 87 insertions(+), 7 deletions(-)
diff --git a/src/main/java/com/se/nsl/service/ResolveService.java b/src/main/java/com/se/nsl/service/ResolveService.java
index 1234ecf..8c84496 100644
--- a/src/main/java/com/se/nsl/service/ResolveService.java
+++ b/src/main/java/com/se/nsl/service/ResolveService.java
@@ -22,6 +22,8 @@
import javax.annotation.Resource;
import java.io.File;
import java.io.IOException;
+import java.math.BigDecimal;
+import java.math.RoundingMode;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
@@ -31,6 +33,7 @@
import java.util.*;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
+import java.util.stream.Collectors;
@Slf4j
@Service
@@ -52,6 +55,8 @@
SimpleDateFormat YYYYMDHM = new SimpleDateFormat("yyyy M d H m ");
+ List<String> MODES = new ArrayList<>(Arrays.asList("姝ｆ�佸垎甯�", "骞冲潎鍒嗗竷", "娉㈠姩骞冲潎鍒嗗竷", "鎸佺画涓婂崌"));
+
public int start(Simu simu) {
Date now = new Date();
String date = StringHelper.YMDHMS2_FORMAT.format(now);
@@ -67,8 +72,8 @@
SimuData data = JSON.parseObject(simu.getData(), SimuData.class);
data.setInPath(date);
data.setOutPath(date);
- data.setEpsg(4548);
data.setEnvelope(envelope);
+ data.setEpsg(config.getEpsg());
simu.setData(JSON.toJSONString(data));
simu.setServiceName(date);
@@ -101,7 +106,7 @@
update(simu, 1, "鍒濆鍖栧弬鏁�");
initArgs(data);
- createRainfallFile(data);
+ createRainfallFile(simu, data);
update(simu, 2, "璋冪敤姹傝В鍣�");
callUwSolver(data);
@@ -152,9 +157,16 @@
f.mkdirs();
}
- private void createRainfallFile(SimuData data) throws Exception {
+ private void createRainfallFile(Simu simu, SimuData data) throws Exception {
List<Rainfall> rainfalls = data.getRainfalls();
- if (null == rainfalls || rainfalls.size() < 2) return; // CollUtil.isEmpty
+ if (null == rainfalls || rainfalls.size() < 2) createRainfall(simu);
+
+ String dat = config.getInPath() + File.separator + "Rainfalls" + File.separator + simu.getId() + ".dat";
+ String rainfallFile = config.getInPath() + File.separator + data.getInPath() + File.separator + "rainfall.dat";
+ if (new File(dat).exists()) {
+ Files.copy(Paths.get(dat), Paths.get(rainfallFile), StandardCopyOption.REPLACE_EXISTING);
+ return;
+ }
List<String> list = new ArrayList<>();
list.add(config.getRainfallTitle());
@@ -174,7 +186,6 @@
//list.add(prefix + YYYYMDHM.format(rainfalls.get(c).getTime()) + getMinVal(rainfalls.get(c).getIntensity() / unit, DIGIT));
list.add(String.format("%s%s%f", prefix, YYYYMDHM.format(rainfalls.get(c).getTime()), getMinVal(rainfalls.get(c).getIntensity() / unit, DIGIT)));
- String rainfallFile = config.getInPath() + File.separator + data.getInPath() + File.separator + "rainfall.dat";
Files.write(Paths.get(rainfallFile), list, StandardCharsets.UTF_8);
}
@@ -294,10 +305,79 @@
}
}
- public String createRainfallCsv(String csvPath, String type, double total, double intensity, int hours) {
+ public String createRainfallCsv(String csvPath, String mode, double total, double intensity, int hours) {
// python 鑴氭湰鍚�.py <鍙傛暟1-csv鏂囦欢鍚�> <鍙傛暟2-闄嶉洦妯″紡:姝ｆ�佸垎甯億骞冲潎鍒嗗竷|娉㈠姩骞冲潎鍒嗗竷|鎸佺画涓婂崌> <鍙傛暟3-闄嶉洦鎬婚噺> <鍙傛暟4-鏈�澶ч洦寮�> <鍙傛暟5-闄嶉洦鏃堕棿(鍒嗛挓)>
- String cmd = String.format("%s \"%s\" \"%s\" %f %f %d", config.getCreateRainfall(), csvPath, type, total, intensity, hours * 60);
+ String cmd = String.format("%s \"%s\" \"%s\" %f %f %d", config.getCreateRainfall(), csvPath, mode, total, intensity, hours * 60);
return callBat(cmd);
}
+
+ public void createRainfall(Simu simu) throws Exception {
+ SimuData data = JSON.parseObject(simu.getData(), SimuData.class);
+ if (null == data.getMode() || MODES.contains(data.getMode())) data.setMode(MODES.get(0));
+ if (StringUtils.isEmpty(data.getIntensityUnit())) data.setIntensityUnit("mm/h");
+
+ Geometry geom = Geometry.CreateFromWkt(simu.getGeom());
+ if (geom.GetGeometryType() == ogr.wkbMultiPolygon) geom = geom.GetGeometryRef(0);
+ double[] envelope = new double[4];
+ geom.GetEnvelope(envelope);
+ data.setEnvelope(envelope);
+ data.setEpsg(config.getEpsg());
+
+ String basePath = config.getInPath() + File.separator + "Rainfalls";
+ if (!new File(basePath).exists()) new File(basePath).mkdirs();
+ if (null == simu.getCreateTime()) simu.setCreateTime(new Timestamp(new Date().getTime()));
+
+ String csvPath = basePath + File.separator + simu.getId() + ".csv";
+ int unit = StringUtils.isEmpty(data.getIntensityUnit()) || "mm/h".equals(data.getIntensityUnit()) ? 60 : 5;
+ createRainfallCsv(csvPath, data.getMode(), data.getTotal(), data.getIntensity() / unit, data.getDuration());
+
+ List<Double> list = getValues(csvPath);
+ if (!CollUtil.isEmpty(list)) {
+ data.setRainfalls(new ArrayList<>());
+ setRainfalls(simu, data, list);
+ }
+
+ simu.setData(JSON.toJSONString(data));
+ }
+
+ private List<Double> getValues(String csvPath) throws Exception {
+ if (!new File(csvPath).exists()) return null;
+
+ List<String> list = Files.readAllLines(Paths.get(csvPath));
+ list.remove(0);
+ //list.remove(list.size() - 1);
+
+ return list.stream()
+ .map(s -> new BigDecimal(s).setScale(6, RoundingMode.HALF_DOWN).doubleValue())
+ .collect(Collectors.toList());
+ }
+
+ private void setRainfalls(Simu simu, SimuData data, List<Double> vals) throws Exception {
+ String basePath = config.getInPath() + File.separator + "Rainfalls";
+ String dat = basePath + File.separator + simu.getId() + ".dat";
+
+ Calendar cal = Calendar.getInstance();
+ cal.setTime(data.getStartTime());
+
+ List<String> list = new ArrayList<>();
+ list.add(config.getRainfallTitle());
+
+ double centerX = ComHelper.getMinVal((data.getMinx() + data.getMaxx()) / 2, DIGIT);
+ double centerY = ComHelper.getMinVal((data.getMiny() + data.getMaxy()) / 2, DIGIT);
+ String prefix = config.getRainfallSite() + " " + centerX + " " + centerY + " ";
+
+ Double total = 0.0;
+ for (int i = 0, c = vals.size(); i < c; i++) {
+ total += vals.get(i);
+ if (i % 15 == 0) {
+ data.getRainfalls().add(new Rainfall(cal.getTime(), vals.get(i), total));
+ }
+ list.add(String.format("%s%s%f", prefix, YYYYMDHM.format(cal.getTime()), vals.get(i)));
+
+ cal.add(Calendar.MINUTE, 1);
+ }
+
+ Files.write(Paths.get(dat), list, StandardCharsets.UTF_8);
+ }
}
--
Gitblit v1.9.3