From 3b74a18403fa5004e0b3820b0c171d90a615ca21 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期四, 14 九月 2023 15:35:02 +0800
Subject: [PATCH] 修改栅格分析的线分析功能,添加GeoTools工具类

---
 src/main/java/com/moon/server/service/data/RasterAnalysisService.java |  100 +++++++++++++++----
 src/main/java/com/moon/server/helper/GdalHelper.java                  |    1 
 src/main/java/com/moon/server/helper/GeoHelper.java                   |  112 ++++++++++++++++++++++
 pom.xml                                                               |   63 +++++++++--
 4 files changed, 241 insertions(+), 35 deletions(-)

diff --git a/pom.xml b/pom.xml
index 00b069f..20edfb9 100644
--- a/pom.xml
+++ b/pom.xml
@@ -70,18 +70,6 @@
             <artifactId>mybatis-plus-boot-starter</artifactId>
             <version>3.5.1</version>
         </dependency>
-        <!--鏀寔PGIS鐨刧eometry绫诲瀷锛歨ttps://www.91mszl.com/zhangwuji/article/details/1350-->
-        <!--<dependency>
-            <groupId>com.eyougo</groupId>
-            <artifactId>mybatis-typehandlers-postgis</artifactId>
-            <version>1.0</version>
-        </dependency>-->
-        <!--mybatis-->
-        <!--<dependency>
-            <groupId>org.mybatis.spring.boot</groupId>
-            <artifactId>mybatis-spring-boot-starter</artifactId>
-            <version>2.2.0</version>
-        </dependency>-->
         <!--redis-->
         <dependency>
             <groupId>org.springframework.boot</groupId>
@@ -113,7 +101,7 @@
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-starter-aop</artifactId>
         </dependency>
-        <!--椤甸潰鐑儴缃�-->
+        <!--devtools-->
         <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-devtools</artifactId>
@@ -126,7 +114,7 @@
             <artifactId>fastjson</artifactId>
             <version>1.2.47</version>
         </dependency>
-        <!--鏁村悎druid鏁版嵁婧�-->
+        <!--druid-->
         <dependency>
             <groupId>com.alibaba</groupId>
             <artifactId>druid</artifactId>
@@ -265,6 +253,28 @@
             <artifactId>sqlite-jdbc</artifactId>
             <version>3.36.0.3</version>
         </dependency>
+        <!--geotools-->
+        <dependency>
+            <groupId>org.geotools</groupId>
+            <artifactId>gt-main</artifactId>
+            <version>22.0</version>
+        </dependency>
+        <dependency>
+            <groupId>org.geotools</groupId>
+            <artifactId>gt-epsg-hsql</artifactId>
+            <version>22.0</version>
+        </dependency>
+        <dependency>
+            <groupId>org.geotools</groupId>
+            <artifactId>gt-api</artifactId>
+            <version>20.0</version>
+        </dependency>
+        <dependency>
+            <groupId>org.geotools</groupId>
+            <artifactId>gt-opengis</artifactId>
+            <version>22.0</version>
+            <scope>compile</scope>
+        </dependency>
     </dependencies>
 
     <build>
@@ -280,4 +290,29 @@
             </plugin>
         </plugins>
     </build>
+
+    <repositories>
+        <repository>
+            <id>osgeo</id>
+            <name>OSGeo Release Repository</name>
+            <url>https://repo.osgeo.org/repository/release/</url>
+            <snapshots>
+                <enabled>false</enabled>
+            </snapshots>
+            <releases>
+                <enabled>true</enabled>
+            </releases>
+        </repository>
+        <repository>
+            <id>osgeo-snapshot</id>
+            <name>OSGeo Snapshot Repository</name>
+            <url>https://repo.osgeo.org/repository/snapshot/</url>
+            <snapshots>
+                <enabled>true</enabled>
+            </snapshots>
+            <releases>
+                <enabled>false</enabled>
+            </releases>
+        </repository>
+    </repositories>
 </project>
diff --git a/src/main/java/com/moon/server/helper/GdalHelper.java b/src/main/java/com/moon/server/helper/GdalHelper.java
index d92f91c..08ba1cc 100644
--- a/src/main/java/com/moon/server/helper/GdalHelper.java
+++ b/src/main/java/com/moon/server/helper/GdalHelper.java
@@ -53,6 +53,7 @@
         // 娉ㄥ唽鎵�鏈夌殑椹卞姩
         gdal.AllRegister();
         ogr.RegisterAll();
+        GeoHelper.initSr();
     }
 
     /**
diff --git a/src/main/java/com/moon/server/helper/GeoHelper.java b/src/main/java/com/moon/server/helper/GeoHelper.java
new file mode 100644
index 0000000..556c784
--- /dev/null
+++ b/src/main/java/com/moon/server/helper/GeoHelper.java
@@ -0,0 +1,112 @@
+package com.moon.server.helper;
+
+import com.moon.server.entity.all.StaticData;
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.gdal.osr.SpatialReference;
+import org.geotools.geometry.DirectPosition2D;
+import org.geotools.referencing.CRS;
+import org.geotools.referencing.GeodeticCalculator;
+import org.opengis.referencing.crs.CoordinateReferenceSystem;
+
+import java.awt.geom.Point2D;
+
+/**
+ * Geo甯姪绫�
+ * @author WWW
+ * @date 2023-09-14
+ */
+public class GeoHelper {
+    public static SpatialReference sr104903;
+
+    public final static double MOON_RADIUS = 1738000;
+
+    public static CoordinateReferenceSystem crs104903;
+
+    private final static Log log = LogFactory.getLog(GeoHelper.class);
+
+    /**
+     * 鍒濆鍖栧潗鏍囩郴
+     */
+    public static void initSr() {
+        try {
+            sr104903 = new SpatialReference(StaticData.MOON_2000_WKT);
+
+            crs104903 = CRS.parseWKT(StaticData.MOON_2000_WKT);
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+        }
+    }
+
+    /**
+     * 鑾峰彇璺濈1
+     */
+    public static double getDistance1(double lon1, double lat1, double lon2, double lat2) {
+        double radLat1 = Math.toRadians(lat1);
+        double radLat2 = Math.toRadians(lat2);
+        double a = radLat1 - radLat2;
+        double b = Math.toRadians(lon1) - Math.toRadians(lon2);
+        double s = 2 * Math.asin(Math.sqrt(Math.pow(Math.sin(a / 2), 2) + Math.cos(radLat1) * Math.cos(radLat2) * Math.pow(Math.sin(b / 2), 2)));
+
+        return s * MOON_RADIUS;
+    }
+
+    /**
+     * 鑾峰彇璺濈2
+     */
+    public static double getDistance2(double x1, double y1, double x2, double y2) {
+        GeodeticCalculator geodeticCalculator = new GeodeticCalculator(crs104903);
+        geodeticCalculator.setStartingGeographicPoint(x1, y1);
+        geodeticCalculator.setDestinationGeographicPoint(x2, y2);
+
+        return geodeticCalculator.getOrthodromicDistance();
+    }
+
+    /**
+     * 鑾峰彇鏂瑰悜瑙�
+     */
+    public static double getBearing(double x1, double y1, double x2, double y2) {
+        return (90 - Math.toDegrees(Math.atan2(x2 - x1, y2 - y1)) + 360) % 360;
+    }
+
+    /**
+     * 鑾峰彇瑙掑害1
+     */
+    public static double getAngle1(double x1, double y1, double x2, double y2) {
+        DirectPosition2D p1 = new DirectPosition2D(crs104903, x1, y1);
+        DirectPosition2D p2 = new DirectPosition2D(crs104903, x2, y2);
+
+        GeodeticCalculator gc = new GeodeticCalculator();
+        gc.setStartingGeographicPoint(p1);
+        gc.setDestinationGeographicPoint(p2);
+
+        return gc.getAzimuth();
+    }
+
+    /**
+     * 鑾峰彇瑙掑害2
+     */
+    public static double getAngle2(double x1, double y1, double x2, double y2) {
+        DirectPosition2D p1 = new DirectPosition2D(x1, y1);
+        DirectPosition2D p2 = new DirectPosition2D(x2, y2);
+
+        GeodeticCalculator gc = new GeodeticCalculator();
+        gc.setStartingGeographicPoint(p1);
+        gc.setDestinationGeographicPoint(p2);
+
+        return gc.getAzimuth();
+    }
+
+    /**
+     * 鏍规嵁璺濈鍜岃搴﹁幏鍙栫洰鏍囩偣
+     */
+    private static Point2D getPointByDistanceAndAngle(double x, double y, double angle, double distance) {
+        DirectPosition2D p1 = new DirectPosition2D(x, y);
+
+        GeodeticCalculator gc = new GeodeticCalculator();
+        gc.setStartingGeographicPoint(p1);
+        gc.setDirection(angle, distance);
+
+        return gc.getDestinationGeographicPoint();
+    }
+}
diff --git a/src/main/java/com/moon/server/service/data/RasterAnalysisService.java b/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
index e6531e3..abdca9d 100644
--- a/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
+++ b/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
@@ -4,6 +4,7 @@
 import com.moon.server.entity.data.AnalysisResultEntity;
 import com.moon.server.entity.data.MetaEntity;
 import com.moon.server.entity.data.PublishEntity;
+import com.moon.server.helper.GeoHelper;
 import com.moon.server.helper.PathHelper;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
@@ -13,7 +14,10 @@
 import org.gdal.gdal.gdal;
 import org.gdal.ogr.Geometry;
 import org.gdal.ogr.ogr;
+import org.gdal.osr.SpatialReference;
+import org.opengis.referencing.crs.CoordinateReferenceSystem;
 import org.springframework.stereotype.Service;
+import org.geotools.referencing.CRS;
 
 import javax.annotation.Resource;
 import java.io.File;
@@ -143,6 +147,58 @@
      * 鍒嗘瀽绾�
      */
     public void analysisPolyline(AnalysisResultEntity entity, Dataset ds, Geometry geo, int size) {
+        /*Double len = geo.Length();
+
+
+        double[] transform = ds.GetGeoTransform();
+        // double rotationX = transform[2]; double rotationY = transform[4]
+        double minX = transform[0], pixelWidth = transform[1], maxY = transform[3], pixelHeight = Math.abs(transform[5]);
+
+        double buffer = Math.max(pixelWidth, pixelHeight) * size;
+        double[] env = new double[4];
+        geo = geo.Buffer(buffer);
+        geo.GetEnvelope(env);
+
+        int xMinPixel = (int) Math.floor((env[0] - minX) / pixelWidth);
+        int yMinPixel = (int) Math.floor((maxY - env[3]) / pixelHeight);
+        int xMaxPixel = (int) Math.floor((env[1] - minX) / pixelWidth);
+        int yMaxPixel = (int) Math.floor((maxY - env[2]) / pixelHeight);
+
+        int bandCount = ds.getRasterCount();
+        int xMin = Math.min(xMinPixel, xMaxPixel);
+        int xMax = Math.max(xMinPixel, xMaxPixel);
+        int yMin = Math.min(yMinPixel, yMaxPixel);
+        int yMax = Math.max(yMinPixel, yMaxPixel);
+
+        List<Double> list = new ArrayList<>();
+        for (int y = yMin; y <= yMax; y++) {
+            for (int x = xMin; x <= xMax; x++) {
+                Geometry point = new Geometry(ogr.wkbPoint);
+                point.AddPoint(minX + pixelWidth * x, maxY - pixelHeight * y);
+                if (geo.Intersects(point)) {
+                    for (int i = 1; i <= bandCount; i++) {
+                        double[] values = new double[1];
+                        ds.GetRasterBand(i).ReadRaster(x, y, 1, 1, values);
+                        list.add(values[0]);
+                    }
+                }
+            }
+        }
+
+        processResult(list, entity);*/
+
+        //double dis = GeoHelper.getDistance1()
+
+        double lineDis = getPolylineDistance(geo);
+        double distance = lineDis / Math.max(geo.GetPointCount(), size);
+
+        List<double[]> list = getPointsByDistance(geo, distance);
+    }
+
+    /**
+     * 鍒嗘瀽绾�
+     */
+    public void analysisPolyline2(AnalysisResultEntity entity, Dataset ds, Geometry geo, int size) {
         double[] transform = ds.GetGeoTransform();
         // double rotationX = transform[2]; double rotationY = transform[4]
         double minX = transform[0], pixelWidth = transform[1], maxY = transform[3], pixelHeight = Math.abs(transform[5]);
@@ -292,32 +348,34 @@
         return gdal.Warp("", new Dataset[]{dataset}, new WarpOptions(warpOptions));
     }
 
-    public void processClippedDataByPolygon(String imagePath, String geometryString) {
-        // 娉ㄥ唽GDAL椹卞姩
-        gdal.AllRegister();
-        // 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
-        Dataset dataset = gdal.Open(imagePath);
-        if (dataset == null) {
-            throw new RuntimeException("Failed to open raster dataset.");
+    /**
+     * 鑾峰彇鎶樼嚎璺濈
+     */
+    private double getPolylineDistance(Geometry geo) {
+        double dis = 0;
+        for (int i = 0, c = geo.GetPointCount() - 1; i < c; i++) {
+            double[] xy1 = geo.GetPoint(i);
+            double[] xy2 = geo.GetPoint(i + 1);
+
+            dis += GeoHelper.getDistance2(xy1[0], xy1[1], xy2[0], xy2[1]);
         }
 
-        Geometry geometry = Geometry.CreateFromWkt(geometryString);
+        return dis;
+    }
 
-        Dataset clippedDataset = clipRaster(dataset, geometry);
+    /**
+     * 鏍规嵁璺濈鑾峰彇鑺傜偣
+     */
+    private List<double[]> getPointsByDistance(Geometry geo, double distance) {
+        List<double[]> list = new ArrayList<>();
+        for (int i = 0, c = geo.GetPointCount() - 1; i < c; i++) {
+            double[] xy1 = geo.GetPoint(i);
+            double[] xy2 = geo.GetPoint(i + 1);
 
-        int width = clippedDataset.GetRasterXSize();
-        int height = clippedDataset.GetRasterYSize();
-        int bandCount = clippedDataset.getRasterCount();
+            double dis = GeoHelper.getDistance2(xy1[0], xy1[1], xy2[0], xy2[1]);
 
-        double[] bandSum = new double[bandCount];
-        for (int bandIndex = 1; bandIndex <= bandCount; bandIndex++) {
-            Band band = clippedDataset.GetRasterBand(bandIndex);
-
-            double[] pixelValues = new double[width * height];
-            band.ReadRaster(0, 0, width, height, pixelValues);
-
-            // 澶氭尝娈靛綊涓�鍖栧鐞�
-            bandSum[bandIndex - 1] = pixelValues[0];
         }
+
+        return list;
     }
 }

--
Gitblit v1.9.3