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 | 102 ++++++++++++++++++++++++++++++++-------------------
1 files changed, 64 insertions(+), 38 deletions(-)
diff --git a/src/main/java/com/se/simu/helper/ShpHelper.java b/src/main/java/com/se/simu/helper/ShpHelper.java
index f9f8778..c205727 100644
--- a/src/main/java/com/se/simu/helper/ShpHelper.java
+++ b/src/main/java/com/se/simu/helper/ShpHelper.java
@@ -6,25 +6,18 @@
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.gdalconst.gdalconst;
import org.gdal.ogr.*;
-import org.gdal.osr.CoordinateTransformation;
import org.gdal.osr.SpatialReference;
-import org.gdal.osr.osr;
-import java.math.BigDecimal;
import java.sql.Timestamp;
-import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;
import java.util.Vector;
-/**
- * ShapeFile甯姪绫�
- *
- * @author WWW
- * @date 2024-09-13
- */
@Slf4j
@SuppressWarnings("ALL")
public class ShpHelper {
@@ -71,7 +64,7 @@
}
}
- private static void createFields(Layer layer,Map<String, Object> map) {
+ private static void createFields(Layer layer, Map<String, Object> map) {
for (String key : map.keySet()) {
Object val = map.get(key);
switch (val.getClass().getTypeName()) {
@@ -122,11 +115,11 @@
}
}
- 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 = fromWgs84(sr, minx, miny);
- double[] dmax = fromWgs84(sr, maxx, maxy);
+ double[] dmin = GdalHelper.fromWgs84(sr, minx, miny);
+ double[] dmax = GdalHelper.fromWgs84(sr, maxx, maxy);
minx = dmin[0];
miny = dmin[1];
maxx = dmax[0];
@@ -142,6 +135,7 @@
Geometry poly = new Geometry(ogr.wkbPolygon);
poly.AddGeometry(ring);
+ poly.AssignSpatialReference(sr);
return poly;
}
@@ -217,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");
@@ -304,30 +295,65 @@
return sr.GetAuthorityCode(null);
}
- /**
- * 杞崲涓篧GS84鍧愭爣
- */
- public static Geometry toWgs84(SpatialReference sr, double x, double y) {
- Geometry point = new Geometry(ogr.wkbPoint);
- point.AssignSpatialReference(sr);
- point.AddPoint(x, y);
+ public static boolean polygonize2Shp(Dataset ds, String filePath) {
+ Driver driver = null;
+ DataSource dataSource = null;
+ Layer layer = null;
+ try {
+ driver = ogr.GetDriverByName("ESRI shapefile");
+ if (null == driver) return false;
- point.TransformTo(GdalHelper.SR4326);
- //point.SwapXY();
+ dataSource = driver.CreateDataSource(filePath, null);
+ if (null == dataSource) return false;
- return point;
+ layer = dataSource.CreateLayer(FileUtil.getName(filePath), ds.GetSpatialRef(), ogr.wkbPolygon, getOptions());
+ 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);
+ }
}
- /**
- * WGS84杞崲涓虹洰鏍囧潗鏍�
- */
- public static double[] fromWgs84(SpatialReference sr, double x, double y) {
- // https://blog.csdn.net/weixin_34910922/article/details/129208661
- CoordinateTransformation ct = new CoordinateTransformation(GdalHelper.SR4326, sr);
- if (sr.IsProjected() != 1) {
- sr.SetAxisMappingStrategy(osr.OAMS_TRADITIONAL_GIS_ORDER);
- }
+ 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;
- return ct.TransformPoint(x, y);
+ 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