月球大数据地理空间分析展示平台-【后端】-月球后台服务
13693261870
2023-09-22 666315f2a2c42d0339bd9e66f07b3062e9172239
添加无效值的判断
已修改1个文件
29 ■■■■ 文件已修改
src/main/java/com/moon/server/service/data/RasterAnalysisService.java 29 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/main/java/com/moon/server/service/data/RasterAnalysisService.java
@@ -7,6 +7,7 @@
import com.moon.server.entity.data.PublishEntity;
import com.moon.server.helper.GeoHelper;
import com.moon.server.helper.PathHelper;
import com.moon.server.helper.WebHelper;
import com.moon.server.service.all.WebSocketService;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
@@ -215,9 +216,13 @@
            for (int j = 1; j <= bandCount; j++) {
                double[] pixelValues = new double[1];
                ds.GetRasterBand(j).ReadRaster(xPixel, yPixel, 1, 1, pixelValues);
                vals.add(pixelValues[0]);
                if (!Double.isNaN(pixelValues[0])) {
                    vals.add(WebHelper.round(pixelValues[0], 3));
                }
            }
            entity.addPoint(xy[0], xy[1], len, vals);
            if (vals.size() > 0) {
                entity.addPoint(xy[0], xy[1], len, vals);
            }
        }
    }
@@ -230,12 +235,13 @@
        double minX = transform[0], pixelWidth = transform[1], maxY = transform[3], pixelHeight = Math.abs(transform[5]), rotationX = transform[2], rotationY = transform[4];
        double[] env = new double[4];
        geo.GetEnvelope(env);
        entity.addPoint(geo.Centroid().GetX(), geo.Centroid().GetY(), 0, null);
        int xMinPixel = Math.max((int) Math.floor((env[0] - minX) / pixelWidth), 1);
        int yMinPixel = Math.max((int) Math.floor((maxY - env[3]) / pixelHeight), 1);
        int xMaxPixel = Math.min((int) Math.floor((env[1] - minX) / pixelWidth), xSize);
        int yMaxPixel = Math.min((int) Math.floor((maxY - env[2]) / pixelHeight), ySize);
        if (xMaxPixel < 1 || yMaxPixel < 1 || xMaxPixel - xMinPixel < 1 || yMaxPixel - yMinPixel < 1) {
        if (xMaxPixel < 1 || yMaxPixel < 1 || xMaxPixel - xMinPixel < 0 || yMaxPixel - yMinPixel < 0) {
            setError(entity, "查询范围无效");
            return;
        }
@@ -243,7 +249,6 @@
        int bandCount = ds.getRasterCount();
        int width = xMaxPixel - xMinPixel;
        int height = yMaxPixel - yMinPixel;
        if (width * height > StaticData.I64 * StaticData.I64) {
            readRasterForBlocks(entity, ds, bandCount, xMinPixel, yMinPixel, width, height);
            return;
@@ -260,8 +265,8 @@
     * 按照块读取栅格数据
     */
    private void readRasterForBlocks(AnalysisResultEntity entity, Dataset ds, int bandCount, int xMinPixel, int yMinPixel, int width, int height) {
        List<Integer> xList = getSamples(xMinPixel, width);
        List<Integer> yList = getSamples(yMinPixel, height);
        List<Integer> xList = getSamples(xMinPixel, width - 1);
        List<Integer> yList = getSamples(yMinPixel, height - 1);
        double[] pixelValues = new double[1];
        for (int i = 1; i <= bandCount; i++) {
@@ -269,7 +274,9 @@
            for (Integer x : xList) {
                for (Integer y : yList) {
                    ds.GetRasterBand(i).ReadRaster(x, y, 1, 1, pixelValues);
                    list.add(pixelValues[0]);
                    if (!Double.isNaN(pixelValues[0])) {
                        list.add(pixelValues[0]);
                    }
                }
            }
            setBandVals(entity, list);
@@ -308,7 +315,9 @@
        List<Double> list = new ArrayList<>();
        for (double val : pixelValues) {
            list.add(val);
            if (!Double.isNaN(val)) {
                list.add(val);
            }
        }
        setBandVals(entity, list);
    }
@@ -317,6 +326,10 @@
     * 设置Band值
     */
    private void setBandVals(AnalysisResultEntity entity, List<Double> list) {
        if (null == list || list.isEmpty()) {
            return;
        }
        double min = Collections.min(list);
        double max = Collections.max(list);
        double avg = list.stream().mapToDouble(Double::valueOf).average().getAsDouble();