From deeef732865c7dd901f07b7123e8f7a3c9f189c9 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 06 十一月 2024 14:53:52 +0800
Subject: [PATCH] 栅格矢量化

---
 src/main/java/com/se/simu/service/ResultService.java |   19 +++++++++++++------
 1 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/se/simu/service/ResultService.java b/src/main/java/com/se/simu/service/ResultService.java
index 793ee10..af93c9b 100644
--- a/src/main/java/com/se/simu/service/ResultService.java
+++ b/src/main/java/com/se/simu/service/ResultService.java
@@ -10,6 +10,7 @@
 import com.se.simu.domain.po.DataPo;
 import com.se.simu.domain.vo.BuildingDepthVo;
 import com.se.simu.helper.GdalHelper;
+import com.se.simu.helper.ShpHelper;
 import lombok.extern.slf4j.Slf4j;
 import org.gdal.gdal.Band;
 import org.gdal.gdal.Dataset;
@@ -359,6 +360,7 @@
 
                 createWaterPng(dto, ds, layer, layer.getWaters().getData().get(i));
                 copyWaterTif(dto, ds, layer.getWaters().getData().get(i));
+                createVectorShp(dto, ds, layer, layer.getWaters().getData().get(i));
             } finally {
                 if (null != ds) ds.delete();
             }
@@ -379,12 +381,6 @@
             String png = waterPath + File.separator + sizes[0] + "_" + sizes[1] + ".png";
             water2Png(dto, layer, tif, png, sizes[0], sizes[1]);
         }
-    }
-
-    private static void copyWaterTif(ResultDto dto, Dataset ds, long ticks) {
-        String source = ds.GetDescription();
-        String target = dto.getOutPath() + File.separator + "waters" + File.separator + ticks + File.separator + "water.tif";
-        FileUtil.copyFile(source, target);
     }
 
     private static String getNameWithExt(String file) {
@@ -502,6 +498,17 @@
         return !Double.isNaN(val) && val > Integer.MIN_VALUE;
     }
 
+    private static void copyWaterTif(ResultDto dto, Dataset ds, long ticks) {
+        String source = ds.GetDescription();
+        String target = dto.getOutPath() + File.separator + "waters" + File.separator + ticks + File.separator + "water.tif";
+        FileUtil.copyFile(source, target);
+    }
+
+    private static void createVectorShp(ResultDto dto, Dataset ds, LayerDto layer, Long ticks) {
+        String shpFile = dto.getOutPath() + File.separator + "waters" + File.separator + ticks + File.separator + "water.shp";
+        ShpHelper.polygonize(ds, shpFile);
+    }
+
     private void copeFlow(ResultDto dto, LayerDto layer) {
         List<String> files = getFiles(dto.getFlowPath(), ".tif");
         if (null == files || files.size() != layer.getWaters().getData().size()) return;

--
Gitblit v1.9.3