修改栅格分析的线分析功能-2

13693261870

2023-09-14 236b037b0d8d2673bb8f1f00013983459c6e669c

 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,9 +14,14 @@
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 sun.awt.IconInfo;

import javax.annotation.Resource;
import java.awt.geom.Point2D;
import java.io.File;
import java.util.*;

@@ -143,6 +149,33 @@
     * 分析线
     */
    public void analysisPolyline(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 = transform[5];

        int bandCount = ds.getRasterCount();
        double lineDis = getPolylineDistance(geo);
        double distance = lineDis / Math.max(geo.GetPointCount(), size);

        List<double[]> points = getPointsByDistance(geo, distance);
        for (double[] xy : points) {
            int xPixel = (int) Math.floor((xy[0] - minX) / pixelWidth);
            int yPixel = (int) Math.floor((maxY - xy[1]) / Math.abs(pixelHeight));

            List<Double> vals = new ArrayList<>();
            for (int i = 1; i <= bandCount; i++) {
                double[] pixelValues = new double[1];
                ds.GetRasterBand(i).ReadRaster(xPixel, yPixel, 1, 1, pixelValues);
                vals.add(pixelValues[0]);
            }
            entity.addPoint(xy[0], xy[1], vals);
        }
    }

    /**
     * 分析线
     */
    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 +325,49 @@
        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.getDistance(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<>();

        int width = clippedDataset.GetRasterXSize();
        int height = clippedDataset.GetRasterYSize();
        int bandCount = clippedDataset.getRasterCount();
        int c = geo.GetPointCount() - 1;
        for (int i = 0; i < c; i++) {
            double[] xy1 = geo.GetPoint(i);
            double[] xy2 = geo.GetPoint(i + 1);
            list.add(xy1);

        double[] bandSum = new double[bandCount];
        for (int bandIndex = 1; bandIndex <= bandCount; bandIndex++) {
            Band band = clippedDataset.GetRasterBand(bandIndex);
            double lineDis = GeoHelper.getDistance(xy1[0], xy1[1], xy2[0], xy2[1]);
            if (lineDis < distance) {
                continue;
            }

            double[] pixelValues = new double[width * height];
            band.ReadRaster(0, 0, width, height, pixelValues);

            // 多波段归一化处理
            bandSum[bandIndex - 1] = pixelValues[0];
            double d = distance;
            double angle = GeoHelper.getAngle(xy1[0], xy1[1], xy2[0], xy2[1]);
            while (d + distance * StaticData.D05 < lineDis) {
                Point2D point = GeoHelper.getPointByDistanceAndAngle(xy1[0], xy1[1], angle, d);
                list.add(new double[]{point.getX(), point.getY()});
                d += distance;
            }
        }

        list.add(geo.GetPoint(c));

        return list;
    }
}