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/helper/ShpHelper.java | 30 ++++++++++++++++++++++++++++++
src/main/java/com/se/simu/service/ResultService.java | 19 +++++++++++++------
2 files changed, 43 insertions(+), 6 deletions(-)
diff --git a/src/main/java/com/se/simu/helper/ShpHelper.java b/src/main/java/com/se/simu/helper/ShpHelper.java
index b46120d..2af67f2 100644
--- a/src/main/java/com/se/simu/helper/ShpHelper.java
+++ b/src/main/java/com/se/simu/helper/ShpHelper.java
@@ -6,6 +6,8 @@
import com.se.simu.domain.dto.GeField;
import com.se.simu.domain.dto.GeLayer;
import lombok.extern.slf4j.Slf4j;
+import org.gdal.gdal.Dataset;
+import org.gdal.gdal.gdal;
import org.gdal.ogr.*;
import org.gdal.osr.SpatialReference;
@@ -299,4 +301,32 @@
private String getEpsg(SpatialReference sr) {
return sr.GetAuthorityCode(null);
}
+
+ public static boolean polygonize(Dataset ds, String filePath) {
+ Driver driver = null;
+ DataSource dataSource = null;
+ Layer layer = null;
+ try {
+ driver = ogr.GetDriverByName("ESRI shapefile");
+ if (null == driver) return false;
+
+ dataSource = driver.CreateDataSource(filePath, null);
+ if (null == dataSource) return false;
+
+ layer = dataSource.CreateLayer(FileUtil.getName(filePath), ds.GetSpatialRef(), ogr.wkbPolygon, getOptions());
+ if (null == layer) return false;
+
+ FieldDefn field = new FieldDefn("val", ogr.OFTReal);
+ layer.CreateField(field);
+
+ gdal.Polygonize(ds.GetRasterBand(1), null, layer, 0, null);
+
+ return true;
+ } catch (Exception ex) {
+ log.error(ex.getMessage(), ex);
+ return false;
+ } finally {
+ GdalHelper.delete(layer, dataSource, driver);
+ }
+ }
}
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