From dbbfe5de7b6a80c143aa32e576b8bee589d7985a Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 13 九月 2023 13:52:08 +0800
Subject: [PATCH] 修改mn下的所有实体类
---
src/main/java/com/moon/server/service/data/RasterAnalysisService.java | 362 +++++++++++++++++++++++----------------------------
1 files changed, 166 insertions(+), 196 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 9088003..e6531e3 100644
--- a/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
+++ b/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
@@ -17,9 +17,7 @@
import javax.annotation.Resource;
import java.io.File;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Vector;
+import java.util.*;
/**
* 鏍呮牸鍒嗘瀽鏈嶅姟
@@ -37,9 +35,9 @@
private final static Log log = LogFactory.getLog(RasterAnalysisService.class);
/**
- * 鍒嗘瀽鐐�
+ * 鍒嗘瀽鏂规硶
*/
- public List<AnalysisResultEntity> analysisPoint(Geometry point, int size) {
+ public List<AnalysisResultEntity> analysis(Geometry geo, Integer size) {
List<AnalysisResultEntity> rs = new ArrayList<>();
List<PublishEntity> pubs = publishService.selectRaster();
@@ -51,7 +49,20 @@
AnalysisResultEntity entity = new AnalysisResultEntity();
entity.setLayerName(pub.getName());
- processPoint(entity, pub, point, size);
+ List<MetaEntity> metas = publishService.selectMetasByPubid(pub.getId());
+ if (null == metas || metas.isEmpty()) {
+ setError(entity, "鎵句笉鍒板彂甯冩暟鎹�");
+ continue;
+ }
+
+ String filePath = pathHelper.getConfig().getUploadPath() + File.separator + metas.get(0).getPath();
+ File file = new File(filePath);
+ if (!file.exists() || file.isDirectory()) {
+ setError(entity, "婧愭暟鎹笉瀛樺湪");
+ continue;
+ }
+
+ openRaster(entity, filePath, geo, size);
rs.add(entity);
}
@@ -59,37 +70,9 @@
}
/**
- * 璁剧疆閿欒
+ * 鎵撳紑鏍呮牸鏁版嵁
*/
- private void setError(AnalysisResultEntity entity, String info) {
- entity.setCode(StaticData.I500);
- entity.setInfo(info);
- }
-
- /**
- * 澶勭悊鐐�
- */
- private void processPoint(AnalysisResultEntity entity, PublishEntity pub, Geometry point, int size) {
- List<MetaEntity> metas = publishService.selectMetasByPubid(pub.getId());
- if (null == metas || metas.isEmpty()) {
- setError(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, "婧愭暟鎹笉瀛樺湪");
- return;
- }
-
- processPoint(entity, filePath, point, size);
- }
-
- /**
- * 澶勭悊鐐�
- */
- public void processPoint(AnalysisResultEntity entity, String filePath, Geometry point, int size) {
+ private void openRaster(AnalysisResultEntity entity, String filePath, Geometry geo, int size) {
Dataset ds = null;
try {
ds = gdal.Open(filePath);
@@ -97,26 +80,16 @@
throw new Exception("鎵撳紑鏍呮牸鏁版嵁澶辫触");
}
- double x = point.GetX(), y = point.GetY();
- 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 xPixel = (int) Math.floor((x - minX) / pixelWidth);
- int yPixel = (int) Math.floor((maxY - y) / Math.abs(pixelHeight));
-
- int bandCount = ds.getRasterCount();
-
- List<double[]> sum = new ArrayList<>();
- for (int bandIndex = 1; bandIndex <= bandCount; bandIndex++) {
- double[] pixelValues = new double[size * size];
- Band band = ds.GetRasterBand(bandIndex);
- band.ReadRaster(xPixel, yPixel, size, size, pixelValues);
-
- sum.add(pixelValues);
+ switch (geo.GetGeometryType()) {
+ case ogr.wkbPoint:
+ analysisPoint(entity, ds, geo, size);
+ break;
+ case ogr.wkbLineString:
+ analysisPolyline(entity, ds, geo, size);
+ break;
+ default:
+ analysisPolygon(entity, ds, geo);
}
-
- //return average(calculateAverage(sum));
} catch (Exception ex) {
setError(entity, ex.getMessage());
log.error(ex.getMessage(), ex);
@@ -129,127 +102,159 @@
}
/**
+ * 鍒嗘瀽鐐�
+ */
+ public void analysisPoint(AnalysisResultEntity entity, Dataset ds, Geometry geo, int size) {
+ double[] transform = ds.GetGeoTransform();
+ double pixelWidth = transform[1], pixelHeight = Math.abs(transform[5]);
+
+ double buffer = Math.max(pixelWidth, pixelHeight) * size;
+ geo = geo.Buffer(buffer);
+
+ analysisPolygon(entity, ds, geo);
+ }
+
+ /**
+ * 鍒嗘瀽鐐�
+ */
+ public void analysisPoint2(AnalysisResultEntity entity, Dataset ds, Geometry geo, int size) {
+ double x = geo.GetX(), y = geo.GetY();
+ 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();
+ 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 List<AnalysisResultEntity> analysisPolyline(Geometry polyline) {
- List<AnalysisResultEntity> rs = new ArrayList<>();
- //
+ 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 = Math.abs(transform[5]);
- return rs;
+ 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);
}
/**
* 鍒嗘瀽闈�
*/
- public List<AnalysisResultEntity> analysisPolygon(Geometry polygon) {
- List<AnalysisResultEntity> rs = new ArrayList<>();
- //
+ 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);
- return rs;
- }
+ 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);
- public List<Double> processLine(String imagePath, String geometryString) {
- // 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
- Dataset dataset = gdal.Open(imagePath);
- if (dataset == null) {
- throw new RuntimeException("Failed to open raster dataset.");
+ 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);
}
- Geometry geometry = Geometry.CreateFromWkt(geometryString);
+ processResult(entity, list);
+ }
- // 鏍呮牸鍍忕礌鑼冨洿
- double[] geoTransform = dataset.GetGeoTransform();
- double minX = geoTransform[0];
- double pixelWidth = geoTransform[1];
- double rotationX = geoTransform[2];
- double maxY = geoTransform[3];
- double rotationY = geoTransform[4];
- double pixelHeight = geoTransform[5];
+ /**
+ * 璁剧疆閿欒
+ */
+ private void setError(AnalysisResultEntity entity, String info) {
+ entity.setCode(StaticData.I500);
+ entity.setInfo(info);
+ }
- // geometry鍍忕礌鑼冨洿
- double[] bounds = new double[6];
- geometry.GetEnvelope(bounds);
- double startX = bounds[0];
- double endX = bounds[1];
- double startY = bounds[2];
- double endY = bounds[3];
+ /**
+ * 澶勭悊缁撴灉
+ */
+ private void processResult(AnalysisResultEntity entity, List<double[]> list) {
+ if (null == list || list.isEmpty()) {
+ return;
+ }
- int xStartPixel = (int) Math.floor((startX - minX) / pixelWidth);
- int ySstartPixel = (int) Math.floor((maxY - startY) / Math.abs(pixelHeight));
- int xEndPixel = (int) Math.floor((endX - minX) / pixelWidth);
- int yEndPixel = (int) Math.floor((maxY - endY) / Math.abs(pixelHeight));
-
- List<Double> values = new ArrayList<>();
- for (int y = Math.min(ySstartPixel, yEndPixel); y <= Math.max(ySstartPixel, yEndPixel); y++) {
- for (int x = Math.min(xStartPixel, xEndPixel); x <= Math.max(xStartPixel, xEndPixel); x++) {
- Geometry point = new Geometry(ogr.wkbPoint);
- point.AddPoint(minX + pixelWidth * x, maxY - pixelHeight * y);
- if (geometry.Contains(point)) {
- values.add(bandValueHandle(dataset, x, y));
+ List<Double> rs = new ArrayList<>();
+ for (double[] ds : list) {
+ if (null != ds && ds.length > 0) {
+ for (double d : ds) {
+ rs.add(d);
}
}
}
- return values;
+ processResult(rs, entity);
}
- public double processPolygon(String imagePath, String geometryString) {
- // 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
- Dataset dataset = gdal.Open(imagePath);
-
- if (dataset == null) {
- throw new RuntimeException("Failed to open raster dataset.");
+ /**
+ * 澶勭悊缁撴灉
+ */
+ private void processResult(List<Double> rs, AnalysisResultEntity entity) {
+ if (null == rs || rs.isEmpty()) {
+ return;
}
- Geometry geometry = Geometry.CreateFromWkt(geometryString);
-
- // 鏍呮牸鍍忕礌鑼冨洿
- double[] geoTransform = dataset.GetGeoTransform();
- double minX = geoTransform[0];
- double pixelWidth = geoTransform[1];
- double rotationX = geoTransform[2];
- double maxY = geoTransform[3];
- double rotationY = geoTransform[4];
- double pixelHeight = geoTransform[5];
-
- // geometry鍍忕礌鑼冨洿
- double[] env = new double[6];
- geometry.GetEnvelope(env);
- int xMinPixel = (int) Math.floor((env[0] - minX) / pixelWidth);
- int yMinPixel = (int) Math.floor((maxY - env[3]) / Math.abs(pixelHeight));
- int xMaxPixel = (int) Math.ceil((env[1] - minX) / pixelWidth);
- int yMaxPixel = (int) Math.ceil((maxY - env[2]) / Math.abs(pixelHeight));
-
- int bandCount = dataset.getRasterCount();
- int geometryWidth = Math.abs(xMaxPixel - xMinPixel);
- int geometryHeight = Math.abs(yMaxPixel - yMinPixel);
-
- List<double[]> sum = new ArrayList<>();
- for (int bandIndex = 1; bandIndex <= bandCount; bandIndex++) {
- Band band = dataset.GetRasterBand(bandIndex);
-
- double[] pixelValues = new double[geometryWidth * geometryHeight];
- band.ReadRaster(xMinPixel, yMinPixel, geometryWidth, geometryHeight, pixelValues);
-
- // 澶氭尝娈靛綊涓�鍖栧鐞�
- sum.add(pixelValues);
- }
-
- return average(calculateAverage(sum));
+ double avg = rs.stream().mapToDouble(Double::valueOf).average().getAsDouble();
+ entity.setMin(Collections.min(rs));
+ entity.setMax(Collections.max(rs));
+ entity.setAvg(avg);
}
- public static Dataset clipRaster(Dataset dataset, Geometry geometry) {
- Vector<String> warpOptions = new Vector<>();
- warpOptions.add("-crop_to_cutline");
- warpOptions.add("-cutline");
- warpOptions.add(geometry.ExportToWkt());
- warpOptions.add("-dstalpha");
-
- return gdal.Warp("", new Dataset[]{dataset}, new WarpOptions(warpOptions));
- }
-
- public double[] processClippedDataByLine(String imagePath, String geometryString) {
+ public void processClippedDataByLine(String imagePath, String geometryString) {
// 娉ㄥ唽GDAL椹卞姩
gdal.AllRegister();
// 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
@@ -275,11 +280,19 @@
sum.add(pixelValues);
}
-
- return calculateAverage(sum);
}
- public double processClippedDataByPolygon(String imagePath, String geometryString) {
+ public static Dataset clipRaster(Dataset dataset, Geometry geometry) {
+ Vector<String> warpOptions = new Vector<>();
+ warpOptions.add("-crop_to_cutline");
+ warpOptions.add("-cutline");
+ warpOptions.add(geometry.ExportToWkt());
+ warpOptions.add("-dstalpha");
+
+ return gdal.Warp("", new Dataset[]{dataset}, new WarpOptions(warpOptions));
+ }
+
+ public void processClippedDataByPolygon(String imagePath, String geometryString) {
// 娉ㄥ唽GDAL椹卞姩
gdal.AllRegister();
// 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
@@ -304,50 +317,7 @@
band.ReadRaster(0, 0, width, height, pixelValues);
// 澶氭尝娈靛綊涓�鍖栧鐞�
- bandSum[bandIndex - 1] = average(pixelValues);
- }
- return average(bandSum);
- }
-
- public static double average(double[] values) {
- double sum = 0;
- for (double value : values) {
- sum += value;
- }
- return sum / values.length;
- }
-
- public static double[] calculateAverage(List<double[]> list) {
- if (list == null || list.isEmpty()) {
- return new double[0];
- }
-
- int arrayLength = list.get(0).length;
- double[] outArray = new double[arrayLength];
-
- for (int i = 0; i < arrayLength; i++) {
- double sum = 0;
-
- for (double[] array : list) {
- sum += array[i];
- }
-
- outArray[i] = sum / list.size();
- }
-
- return outArray;
- }
-
- public static double bandValueHandle(Dataset dataset, int x, int y) {
- double[] bandSum = new double[dataset.getRasterCount()];
- for (int bandIndex = 1; bandIndex <= dataset.getRasterCount(); bandIndex++) {
- double[] pixelValues = new double[1];
- Band band = dataset.GetRasterBand(bandIndex);
- band.ReadRaster(x, y, 1, 1, pixelValues);
-
bandSum[bandIndex - 1] = pixelValues[0];
}
-
- return average(bandSum);
}
}
--
Gitblit v1.9.3