From 1cae540cd2b99803808d31a3a5dd35e3216f19c7 Mon Sep 17 00:00:00 2001
From: wuww <252740454@qq.com>
Date: 星期三, 30 四月 2025 14:50:54 +0800
Subject: [PATCH] 插值处理-四邻域
---
src/main/java/com/se/nsl/service/TestService.java | 73 ++++++++++++++++++++++++++++++++++--
1 files changed, 68 insertions(+), 5 deletions(-)
diff --git a/src/main/java/com/se/nsl/service/TestService.java b/src/main/java/com/se/nsl/service/TestService.java
index 764be0e..c5d56d2 100644
--- a/src/main/java/com/se/nsl/service/TestService.java
+++ b/src/main/java/com/se/nsl/service/TestService.java
@@ -45,6 +45,8 @@
@Resource
ResultService resultService;
+ public final static double MIN_VAL = 0.00001;
+
public final static double MAX_X_OFFSET = 0;
public final static SimpleDateFormat SDF = new SimpleDateFormat("yyyy-MM-dd HH:mm");
@@ -285,7 +287,7 @@
for (int[] sizes : layer.getTerrain().getSize()) {
String fileName = ComHelper.getNameWithExt(ds.GetDescription()) + "_" + sizes[0] + "_" + sizes[1];
String tif = dto.getTemp() + File.separator + fileName + ".tif";
- ComHelper.Resample(ds, tif, sizes[0], sizes[1], layer);
+ ComHelper.Resample2(ds, tif, sizes[0], sizes[1], layer);
if (!new File(tif).exists()) continue;
String png = waterPath + File.separator + sizes[0] + "_" + sizes[1] + ".png";
@@ -303,6 +305,7 @@
float[] buffer = new float[width * height];
band.ReadRaster(0, 0, width, height, buffer);
double[] transform = ds.GetGeoTransform();
+ buffer = interpolateGrid(buffer, width);
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
//double differ = layer.getWaters().getMaxHeight() - layer.getWaters().getMinHeight(), minHeight = layer.getWaters().getMinHeight();
@@ -312,7 +315,7 @@
for (int y = 0; y < height; y++) {
int offset = x + y * width;
//if (Float.isNaN(buffer[offset]) || buffer[offset] < -999 || buffer[offset] < minHeight) continue;
- if (Float.isNaN(buffer[offset]) || buffer[offset] <= 0.00001 || Float.isNaN(ts[offset])) continue;
+ if (Float.isNaN(buffer[offset]) || buffer[offset] < MIN_VAL || Float.isNaN(ts[offset])) continue;
//double X = transform[0] + x * transform[1] + y * transform[2];
//double Y = transform[3] + x * transform[4] + y * transform[5];
@@ -337,6 +340,66 @@
} finally {
if (null != ds) ds.delete();
}
+ }
+
+ /**
+ * 鎻掑�煎鐞�-鍥涢偦鍩�
+ */
+ public float[] interpolateGrid(float[] buffer, int size) {
+ float[] tempBuffer = new float[size * size];
+ for (int i = 0; i < size; i++) {
+ for (int j = 0; j < size; j++) {
+ int index = i * size + j;
+ float current = buffer[index];
+ if (current == 0) {
+ float sum = 0;
+ int count = 0;
+ // 妫�鏌ヤ笂閭�
+ if (i > 0) {
+ float neighbor = buffer[(i - 1) * size + j];
+ if (neighbor > MIN_VAL) {
+ sum += neighbor;
+ count++;
+ }
+ }
+ // 妫�鏌ヤ笅閭�
+ if (i < size - 1) {
+ float neighbor = buffer[(i + 1) * size + j];
+ if (neighbor > MIN_VAL) {
+ sum += neighbor;
+ count++;
+ }
+ }
+ // 妫�鏌ュ乏閭�
+ if (j > 0) {
+ float neighbor = buffer[i * size + (j - 1)];
+ if (neighbor > MIN_VAL) {
+ sum += neighbor;
+ count++;
+ }
+ }
+ // 妫�鏌ュ彸閭�
+ if (j < size - 1) {
+ float neighbor = buffer[i * size + (j + 1)];
+ if (neighbor > MIN_VAL) {
+ sum += neighbor;
+ count++;
+ }
+ }
+ // 璁＄畻鏂板��
+ if (count > 0) {
+ tempBuffer[index] = (sum / count) * 0.5f;
+ } else {
+ tempBuffer[index] = 0;
+ }
+ } else {
+ tempBuffer[index] = current;
+ }
+ }
+ }
+ // 灏嗕复鏃舵暟缁勫鍒跺洖鍘熸暟缁�
+ //System.arraycopy(tempBuffer, 0, buffer, 0, tempBuffer.length);
+ return tempBuffer;
}
public List<BuildingDepthVo> processBuilding(ResultDto dto, List<String> files, LayerDto layer) {
@@ -541,7 +604,7 @@
int offset = x + y * width;
float fx = ComHelper.getFloatValue(vxBuffer[offset]);
float fy = ComHelper.getFloatValue(vyBuffer[offset]);
- if (Float.isNaN(fx) && Float.isNaN(fy) || (fx == 0.00001 && fy == 0.00001)) continue;
+ if (Float.isNaN(fx) && Float.isNaN(fy) || (fx < MIN_VAL && fy < MIN_VAL)) continue;
fx = Float.isNaN(fx) ? 0 : fx;
fy = Float.isNaN(fy) ? 0 : fy;
@@ -562,8 +625,8 @@
layer.getWaters().setFiles(null);
layer.getTerrain().setEpsg(null);
layer.getExtension().setDiffer(null);
- layer.setWaterUrl("/hls/w" + config.getOutPath() + ".m3u8");
- layer.setFlowUrl("//hls/f" + config.getOutPath() + ".m3u8");
+ layer.setWaterUrl("/hls/w" + dto.getOutPath() + ".m3u8");
+ layer.setFlowUrl("/hls/f" + dto.getOutPath() + ".m3u8");
String json = JSON.toJSONString(layer);
// String json = JSONUtil.toJsonPrettyStr(layer);
--
Gitblit v1.9.3