From 3417cf014a65765e02696c1d121ce58b2b4a8aed Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期二, 08 四月 2025 15:55:36 +0800 Subject: [PATCH] 修改pom.xml --- src/main/java/com/se/simu/helper/ShpHelper.java | 57 ++++++++++++++++++++++++++++++++++++++++++--------------- 1 files changed, 42 insertions(+), 15 deletions(-) diff --git a/src/main/java/com/se/simu/helper/ShpHelper.java b/src/main/java/com/se/simu/helper/ShpHelper.java index 2af67f2..c205727 100644 --- a/src/main/java/com/se/simu/helper/ShpHelper.java +++ b/src/main/java/com/se/simu/helper/ShpHelper.java @@ -8,6 +8,7 @@ import lombok.extern.slf4j.Slf4j; import org.gdal.gdal.Dataset; import org.gdal.gdal.gdal; +import org.gdal.gdalconst.gdalconst; import org.gdal.ogr.*; import org.gdal.osr.SpatialReference; @@ -17,12 +18,6 @@ import java.util.Map; import java.util.Vector; -/** - * ShapeFile甯姪绫� - * - * @author WWW - * @date 2024-09-13 - */ @Slf4j @SuppressWarnings("ALL") public class ShpHelper { @@ -120,7 +115,7 @@ } } - private static Geometry createPolygon(SpatialReference sr, Double minx, Double miny, Double maxx, Double maxy) { + public static Geometry createPolygon(SpatialReference sr, Double minx, Double miny, Double maxx, Double maxy) { String epsg = sr.GetAuthorityCode(null); if (!("4326".equals(epsg) || "4490".equals(epsg))) { double[] dmin = GdalHelper.fromWgs84(sr, minx, miny); @@ -140,6 +135,7 @@ Geometry poly = new Geometry(ogr.wkbPolygon); poly.AddGeometry(ring); + poly.AssignSpatialReference(sr); return poly; } @@ -215,10 +211,7 @@ f.SetField(i, local.getYear(), local.getMonthValue(), local.getDayOfMonth(), local.getHour(), local.getMinute(), local.getSecond(), 8); } - /** - * 鍒涘缓Geometry瀵硅薄 - */ - private static Geometry createGeometry(GeLayer geLayer, JSONObject geom) { + public static Geometry createGeometry(GeLayer geLayer, JSONObject geom) { String type = geom.getStr("type"); JSONArray cs = geom.getJSONArray("coordinates"); @@ -302,7 +295,7 @@ return sr.GetAuthorityCode(null); } - public static boolean polygonize(Dataset ds, String filePath) { + public static boolean polygonize2Shp(Dataset ds, String filePath) { Driver driver = null; DataSource dataSource = null; Layer layer = null; @@ -316,10 +309,9 @@ 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); + layer.CreateField(new FieldDefn("val", ogr.OFTReal)); - gdal.Polygonize(ds.GetRasterBand(1), null, layer, 0, null); + gdal.Polygonize(ds.GetRasterBand(1), ds.GetRasterBand(1).GetMaskBand(), layer, 0); return true; } catch (Exception ex) { @@ -329,4 +321,39 @@ GdalHelper.delete(layer, dataSource, driver); } } + + public static boolean polygonize2Geojson(Dataset ds, String filePath) { + Driver driver = null; + DataSource dataSource = null; + Layer layer = null; + try { + driver = ogr.GetDriverByName("GeoJSON"); + 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); + if (null == layer) return false; + + layer.CreateField(new FieldDefn("val", ogr.OFTReal)); + + gdal.Polygonize(ds.GetRasterBand(1), ds.GetRasterBand(1).GetMaskBand(), layer, 0); + + return true; + } catch (Exception ex) { + log.error(ex.getMessage(), ex); + return false; + } finally { + GdalHelper.delete(layer, dataSource, driver); + } + } + + public static void test() { + String path = "D:\\simu\\out\\20241010095328\\waters\\1730217635000\\"; + + Dataset ds = gdal.Open(path + "water.tif", gdalconst.GA_ReadOnly); + ds.SetSpatialRef(GdalHelper.createSpatialReference(4548)); + polygonize2Geojson(ds, path + "water.geojson"); + } } -- Gitblit v1.9.3