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 ++++++++++++++++++++++++++++++
 1 files changed, 30 insertions(+), 0 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);
+        }
+    }
 }

--
Gitblit v1.9.3