From 1d480ee465a84fd08c71c96efb870ef38e7f3481 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 22 九月 2023 17:39:00 +0800
Subject: [PATCH] 删除无效代码

---
 src/main/java/com/moon/server/service/data/RasterAnalysisService.java |  441 +++++++++++++++++++++++++++++++++---------------------
 1 files changed, 271 insertions(+), 170 deletions(-)

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 abdca9d..fad1561 100644
--- a/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
+++ b/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
@@ -1,11 +1,14 @@
 package com.moon.server.service.data;
 
+import com.alibaba.fastjson.JSONObject;
 import com.moon.server.entity.all.StaticData;
 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 com.moon.server.helper.WebHelper;
+import com.moon.server.service.all.WebSocketService;
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
 import org.gdal.gdal.Band;
@@ -14,13 +17,12 @@
 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.awt.geom.Point2D;
 import java.io.File;
+import java.io.IOException;
 import java.util.*;
 
 /**
@@ -37,6 +39,70 @@
     PublishService publishService;
 
     private final static Log log = LogFactory.getLog(RasterAnalysisService.class);
+
+    /**
+     * 娴嬭瘯
+     */
+    public List<AnalysisResultEntity> test(Geometry geo, Integer size) {
+        List<AnalysisResultEntity> rs = new ArrayList<>();
+        AnalysisResultEntity entity = new AnalysisResultEntity();
+        entity.setLayerName("Test");
+
+        openRaster(entity, "D:\\Moon\\data\\DOM\\Lunar_LRO_LOLA_ClrShade_Global_128ppd_v04_2.tif", geo, size);
+        rs.add(entity);
+
+        return rs;
+    }
+
+    /**
+     * 浣跨敤WKT鏌ヨ鍒嗘瀽锛岀粨鏋滀互娑堟伅鎺ㄩ��
+     */
+    public void analysisForPost(Geometry geo, Integer size, String token) {
+        List<PublishEntity> pubs = publishService.selectRaster();
+        if (null == pubs || pubs.isEmpty()) {
+            return;
+        }
+
+        // for (PublishEntity pub : pubs) {
+        pubs.parallelStream().forEach(pub -> {
+            try {
+                AnalysisResultEntity entity = new AnalysisResultEntity(token);
+                entity.setLayerName(pub.getName());
+
+                List<MetaEntity> metas = publishService.selectMetasByPubid(pub.getId());
+                if (null == metas || metas.isEmpty()) {
+                    setError(entity, "鎵句笉鍒板彂甯冩暟鎹�");
+                    postInfo(entity);
+                    return;
+                }
+
+                String filePath = pathHelper.getConfig().getUploadPath() + File.separator + metas.get(0).getPath();
+                File file = new File(filePath);
+                if (!file.exists() || file.isDirectory()) {
+                    setError(entity, "婧愭暟鎹笉瀛樺湪");
+                    postInfo(entity);
+                    return;
+                }
+
+                openRaster(entity, filePath, geo, size);
+                postInfo(entity);
+            } catch (Exception ex) {
+                log.error(ex.getMessage(), ex);
+            }
+        });
+    }
+
+    /**
+     * 鎺ㄩ�佹秷鎭�
+     */
+    private void postInfo(AnalysisResultEntity entity) throws IOException {
+        JSONObject map = new JSONObject();
+        map.put("analysisForPost", entity);
+
+        String json = JSONObject.toJSONString(map);
+        // System.out.println(json)
+        WebSocketService.broadCastInfo(json);
+    }
 
     /**
      * 鍒嗘瀽鏂规硶
@@ -79,6 +145,7 @@
     private void openRaster(AnalysisResultEntity entity, String filePath, Geometry geo, int size) {
         Dataset ds = null;
         try {
+            // filePath = "D:\\Moon\\data\\DOM\\test.tif"
             ds = gdal.Open(filePath);
             if (null == ds) {
                 throw new Exception("鎵撳紑鏍呮牸鏁版嵁澶辫触");
@@ -119,7 +186,203 @@
     }
 
     /**
-     * 鍒嗘瀽鐐�
+     * 鍒嗘瀽绾�
+     */
+    public void analysisPolyline(AnalysisResultEntity entity, Dataset ds, Geometry geo, int size) {
+        double[] transform = ds.GetGeoTransform();
+        int xSize = ds.getRasterXSize(), ySize = ds.getRasterYSize();
+        double minX = transform[0], pixelWidth = transform[1], maxY = transform[3], pixelHeight = transform[5];
+
+        double len = 0;
+        int bandCount = ds.getRasterCount();
+        double lineDis = getPolylineDistance(geo);
+        double distance = lineDis / Math.max(geo.GetPointCount(), size);
+
+        List<double[]> points = getPointsByDistance(geo, distance);
+        for (int i = 0, c = points.size(); i < c; i++) {
+            double[] xy = points.get(i);
+            if (i > 0) {
+                double[] p = points.get(i - 1);
+                len += GeoHelper.getDistance(p[0], p[1], xy[0], xy[1]);
+            }
+
+            int xPixel = (int) Math.floor((xy[0] - minX) / pixelWidth);
+            int yPixel = (int) Math.floor((maxY - xy[1]) / Math.abs(pixelHeight));
+            if (xPixel < 1 || xPixel > xSize || yPixel < 1 || yPixel > ySize) {
+                continue;
+            }
+
+            List<Double> vals = new ArrayList<>();
+            for (int j = 1; j <= bandCount; j++) {
+                double[] pixelValues = new double[1];
+                ds.GetRasterBand(j).ReadRaster(xPixel, yPixel, 1, 1, pixelValues);
+                if (!Double.isNaN(pixelValues[0])) {
+                    vals.add(WebHelper.round(pixelValues[0], 3));
+                }
+            }
+            if (vals.size() > 0) {
+                entity.addPoint(xy[0], xy[1], len, vals);
+            }
+        }
+    }
+
+    /**
+     * 鍒嗘瀽闈�
+     */
+    public void analysisPolygon(AnalysisResultEntity entity, Dataset ds, Geometry geo) {
+        double[] transform = ds.GetGeoTransform();
+        int xSize = ds.getRasterXSize(), ySize = ds.getRasterYSize();
+        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 < 0 || yMaxPixel - yMinPixel < 0) {
+            setError(entity, "鏌ヨ鑼冨洿鏃犳晥");
+            return;
+        }
+
+        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;
+        }
+
+        for (int i = 1; i <= bandCount; i++) {
+            double[] pixelValues = new double[width * height];
+            ds.GetRasterBand(i).ReadRaster(xMinPixel, yMinPixel, width, height, pixelValues);
+            setBandVals(entity, pixelValues);
+        }
+    }
+
+    /**
+     * 鎸夌収鍧楄鍙栨爡鏍兼暟鎹�
+     */
+    private void readRasterForBlocks(AnalysisResultEntity entity, Dataset ds, int bandCount, int xMinPixel, int yMinPixel, int width, int 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++) {
+            List<Double> list = new ArrayList<>();
+            for (Integer x : xList) {
+                for (Integer y : yList) {
+                    ds.GetRasterBand(i).ReadRaster(x, y, 1, 1, pixelValues);
+                    if (!Double.isNaN(pixelValues[0])) {
+                        list.add(pixelValues[0]);
+                    }
+                }
+            }
+            setBandVals(entity, list);
+        }
+    }
+
+    /**
+     * 鑾峰彇鎶芥牱鍒楄〃
+     */
+    private List<Integer> getSamples(int start, int strip) {
+        List<Integer> list = new ArrayList<>();
+
+        double avg = 1.0 * strip / (StaticData.I64 - 1);
+        for (int i = 0; i < StaticData.I64; i++) {
+            list.add(start + (int) Math.ceil(avg * i));
+        }
+
+        return list;
+    }
+
+    /**
+     * 璁剧疆閿欒
+     */
+    private void setError(AnalysisResultEntity entity, String info) {
+        entity.setCode(StaticData.I500);
+        entity.setInfo(info);
+    }
+
+    /**
+     * 璁剧疆Band鍊�
+     */
+    private void setBandVals(AnalysisResultEntity entity, double[] pixelValues) {
+        if (null == pixelValues || pixelValues.length == 0) {
+            return;
+        }
+
+        List<Double> list = new ArrayList<>();
+        for (double val : pixelValues) {
+            if (!Double.isNaN(val)) {
+                list.add(val);
+            }
+        }
+        setBandVals(entity, list);
+    }
+
+    /**
+     * 璁剧疆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();
+        entity.addBandVals(min, avg, max);
+    }
+
+    /**
+     * 鑾峰彇鎶樼嚎璺濈
+     */
+    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]);
+        }
+
+        return dis;
+    }
+
+    /**
+     * 鏍规嵁璺濈鑾峰彇鑺傜偣
+     */
+    private List<double[]> getPointsByDistance(Geometry geo, double distance) {
+        List<double[]> list = new ArrayList<>();
+
+        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 lineDis = GeoHelper.getDistance(xy1[0], xy1[1], xy2[0], xy2[1]);
+            if (lineDis < distance) {
+                continue;
+            }
+
+            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;
+    }
+
+    /**
+     * 鍒嗘瀽鐐� 2
      */
     public void analysisPoint2(AnalysisResultEntity entity, Dataset ds, Geometry geo, int size) {
         double x = geo.GetX(), y = geo.GetY();
@@ -131,72 +394,15 @@
         int xPixel = (int) Math.floor((x - minX) / pixelWidth);
         int yPixel = (int) Math.floor((maxY - y) / Math.abs(pixelHeight));
 
-        List<double[]> list = new ArrayList<>();
         for (int i = 1; i <= bandCount; i++) {
             double[] pixelValues = new double[size * size];
             Band band = ds.GetRasterBand(i);
             band.ReadRaster(xPixel, yPixel, size, size, pixelValues);
-
-            list.add(pixelValues);
         }
-
-        processResult(entity, list);
     }
 
     /**
-     * 鍒嗘瀽绾�
-     */
-    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);
-    }
-
-    /**
-     * 鍒嗘瀽绾�
+     * 鍒嗘瀽绾� 2
      */
     public void analysisPolyline2(AnalysisResultEntity entity, Dataset ds, Geometry geo, int size) {
         double[] transform = ds.GetGeoTransform();
@@ -233,84 +439,10 @@
                 }
             }
         }
-
-        processResult(list, entity);
+        // processResult(list, entity)
     }
 
-    /**
-     * 鍒嗘瀽闈�
-     */
-    public void analysisPolygon(AnalysisResultEntity entity, Dataset ds, Geometry geo) {
-        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[] env = new double[4];
-        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 geoWidth = Math.abs(xMaxPixel - xMinPixel);
-        int geoHeight = Math.abs(yMaxPixel - yMinPixel);
-
-        List<double[]> list = new ArrayList<>();
-        for (int i = 1; i <= bandCount; i++) {
-            Band band = ds.GetRasterBand(i);
-            double[] pixelValues = new double[geoWidth * geoHeight];
-            band.ReadRaster(xMinPixel, yMinPixel, geoWidth, geoHeight, pixelValues);
-
-            list.add(pixelValues);
-        }
-
-        processResult(entity, list);
-    }
-
-    /**
-     * 璁剧疆閿欒
-     */
-    private void setError(AnalysisResultEntity entity, String info) {
-        entity.setCode(StaticData.I500);
-        entity.setInfo(info);
-    }
-
-    /**
-     * 澶勭悊缁撴灉
-     */
-    private void processResult(AnalysisResultEntity entity, List<double[]> list) {
-        if (null == list || list.isEmpty()) {
-            return;
-        }
-
-        List<Double> rs = new ArrayList<>();
-        for (double[] ds : list) {
-            if (null != ds && ds.length > 0) {
-                for (double d : ds) {
-                    rs.add(d);
-                }
-            }
-        }
-
-        processResult(rs, entity);
-    }
-
-    /**
-     * 澶勭悊缁撴灉
-     */
-    private void processResult(List<Double> rs, AnalysisResultEntity entity) {
-        if (null == rs || rs.isEmpty()) {
-            return;
-        }
-
-        double avg = rs.stream().mapToDouble(Double::valueOf).average().getAsDouble();
-        entity.setMin(Collections.min(rs));
-        entity.setMax(Collections.max(rs));
-        entity.setAvg(avg);
-    }
-
-    public void processClippedDataByLine(String imagePath, String geometryString) {
+    private void processClippedDataByLine(String imagePath, String geometryString) {
         // 娉ㄥ唽GDAL椹卞姩
         gdal.AllRegister();
         // 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
@@ -338,7 +470,7 @@
         }
     }
 
-    public static Dataset clipRaster(Dataset dataset, Geometry geometry) {
+    private Dataset clipRaster(Dataset dataset, Geometry geometry) {
         Vector<String> warpOptions = new Vector<>();
         warpOptions.add("-crop_to_cutline");
         warpOptions.add("-cutline");
@@ -346,36 +478,5 @@
         warpOptions.add("-dstalpha");
 
         return gdal.Warp("", new Dataset[]{dataset}, new WarpOptions(warpOptions));
-    }
-
-    /**
-     * 鑾峰彇鎶樼嚎璺濈
-     */
-    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]);
-        }
-
-        return dis;
-    }
-
-    /**
-     * 鏍规嵁璺濈鑾峰彇鑺傜偣
-     */
-    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);
-
-            double dis = GeoHelper.getDistance2(xy1[0], xy1[1], xy2[0], xy2[1]);
-
-        }
-
-        return list;
     }
 }

--
Gitblit v1.9.3