根据坐标查询积水深度功能第一次提交

wangyifei

2024-10-31 794f620632fca78834677d6b0890bfa0fe78eb02

根据坐标查询积水深度功能第一次提交

 src/main/java/com/se/simu/controller/WaterController.java

@@ -128,8 +128,7 @@
            }

            // 根据服务名+时间戳+坐标，查询对应的积水深度

            return 0.0;
            return waterService.getWaterHeight(serviceName, x, y, timestamp);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
            return null;

 src/main/java/com/se/simu/service/WaterService.java

@@ -3,6 +3,11 @@
import com.se.simu.config.PropertiesConfig;
import com.se.simu.domain.vo.*;
import lombok.extern.slf4j.Slf4j;
import org.gdal.gdal.Band;
import org.gdal.gdal.Dataset;
import org.gdal.gdal.gdal;
import org.gdal.gdalconst.gdalconstConstants;
import org.gdal.osr.SpatialReference;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
@@ -116,4 +121,66 @@
            return null;
        }
    }

    /**
     * 根据坐标查询积水深度
     */
    public Double getWaterHeight(String serviceName, Double x, Double y, Long timestamp) {
        String filePath = config.getOutPath() + File.separator + serviceName + File.separator + "waters"
                + File.separator + timestamp + File.separator + "water.tif";

        // 首先是所有gdal程序都需要的注册语句
        gdal.AllRegister();
        // 读取影像
        Dataset hDataset = gdal.Open(filePath, gdalconstConstants.GA_Update);

        // 设置数据集的投影
        SpatialReference srs = new SpatialReference();
        srs.ImportFromEPSG(4548);
        hDataset.SetProjection(srs.ExportToWkt());

        //获取栅格数量
        int numBands=hDataset.GetRasterCount();
        System.out.println("RasterCount: " + numBands);
        //构造仿射变换参数数组，并获取数据
        double[] gt = new double[6];
        hDataset.GetGeoTransform(gt);
        System.out.println("仿射变换参数"+ Arrays.toString(gt));

        //经纬度转换为栅格像素坐标
        int[] ColRow=Coordinates2ColRow(gt,x,y);

        //判断是否坐标超出范围
        if(ColRow[0]<0||ColRow[1]<0||ColRow[0]>hDataset.getRasterXSize()||ColRow[1]>hDataset.getRasterYSize()){
            System.out.println(Arrays.toString(ColRow)+"坐标值超出栅格范围！");
            return null;
        }

        //获取该点对应的波段的值
        Band band = hDataset.GetRasterBand(1);
        double[] values = new double[1];
        band.ReadRaster(ColRow[0], ColRow[1], 1, 1, values);
        double value = values[0];

        //释放资源
        hDataset.delete();
        return value;
    }

    /**
     * 将地图坐标转换为栅格像素坐标
     * @param gt 仿射变换参数
     * @param X 横坐标
     * @param Y 纵坐标
     * @return
     */
    public int[] Coordinates2ColRow(double[] gt, double X, double Y){
        int[] ints = new int[2];
        //向下取整,如果向上取整会导致计算结果偏大，从而在后面读取到邻近像元的数据
        double  Yline = Math.floor(((Y - gt[3])*gt[1] - (X-gt[0])*gt[4]) / (gt[5]*gt[1]-gt[2]*gt[4]));
        double  Xpixel = Math.floor((X-gt[0] - Yline*gt[2])/gt[1]);
        ints[0] = new Double(Xpixel).intValue();
        ints[1] = new Double(Yline).intValue();
        return ints;
    }
}