From 92fd3ce2a44b27dd931ee68a0f604e1af6d1b498 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 11 九月 2023 10:21:34 +0800
Subject: [PATCH] 修改栅格分析-1
---
src/main/java/com/moon/server/service/data/RasterAnalysisService.java | 148 +++++++++++++++++------------
src/main/java/com/moon/server/entity/data/AnalysisResultEntity.java | 55 +++++++++++
src/main/java/com/moon/server/controller/data/RasterAnalysisController.java | 84 ++++++++--------
3 files changed, 182 insertions(+), 105 deletions(-)
diff --git a/src/main/java/com/moon/server/controller/data/RasterAnalysisController.java b/src/main/java/com/moon/server/controller/data/RasterAnalysisController.java
index 58536b7..1d213f0 100644
--- a/src/main/java/com/moon/server/controller/data/RasterAnalysisController.java
+++ b/src/main/java/com/moon/server/controller/data/RasterAnalysisController.java
@@ -3,7 +3,7 @@
import com.moon.server.annotation.SysLog;
import com.moon.server.controller.all.BaseController;
import com.moon.server.entity.all.ResponseMsg;
-import com.moon.server.entity.data.ImageEntity;
+import com.moon.server.helper.StringHelper;
import com.moon.server.helper.WebHelper;
import com.moon.server.service.data.RasterAnalysisService;
import io.swagger.annotations.Api;
@@ -15,12 +15,8 @@
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
-import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
/**
* 鏍呮牸鍒嗘瀽
@@ -34,40 +30,32 @@
@Resource
RasterAnalysisService rasterService;
- @SysLog()
- @ApiOperation(value = "鏍规嵁geometry鑾峰彇瀵瑰簲鑼冨洿鐨勭伆搴﹀��")
- @PostMapping
- public ResponseMsg<Object> rasterAnalysis(@RequestBody List<ImageEntity> images) {
- // 杈撳叆鐨勫嚑浣旼eometry: POINT (75.772 2.498)
- String geometryString = images.get(0).getGeometryString();
- Geometry geometry = Geometry.CreateFromWkt(geometryString);
-
- for (ImageEntity i : images) {
- String path = i.getPath();
- if (geometry.GetGeometryType() == ogr.wkbPoint) {
- i.setResult(rasterService.processPoint(path, geometryString, i.getPointSize()));
- } else if (geometry.GetGeometryType() == ogr.wkbLineString) {
- i.setResult(rasterService.processLine(path, geometryString));
- } else if (geometry.GetGeometryType() == ogr.wkbPolygon) {
- i.setResult(rasterService.processPolygon(path, geometryString));
- }
- }
-
- return success(images);
- }
+ private final static List<Integer> pixels = new ArrayList<>(Arrays.asList(1, 2, 4, 8, 16, 32, 64, 128, 256));
@SysLog()
@ApiOperation(value = "鏌ヨ鐐瑰垎鏋�")
@ApiImplicitParams({
- @ApiImplicitParam(name = "wkt", value = "鐐筗KT", dataType = "String", example = "")
+ @ApiImplicitParam(name = "wkt", value = "鐐筗KT", dataType = "String", example = "POINT (115.94927385452 32.3754479115071)"),
+ @ApiImplicitParam(name = "pixel", value = "鍍忕礌鍊�", dataType = "Integer", example = "1")
})
@GetMapping(value = "/selectByPoint")
- public ResponseMsg<Object> selectByPoint(String wkt) {
+ public ResponseMsg<Object> selectByPoint(String wkt, Integer pixel) {
try {
- Map<String, Double> map = new HashMap<>(5);
- map.put("鍥惧眰鍚�", 0.0);
+ // Map<String, Double> map = new HashMap<>(5); map.put("鍥惧眰鍚�", 0.0); return success(map)
+ if (StringHelper.isEmpty(wkt)) {
+ return fail("WKT瀛楃涓蹭笉鑳戒负绌�");
+ }
+ Geometry geo = Geometry.CreateFromWkt(wkt);
+ if (null == geo || geo.GetGeometryType() != ogr.wkbPoint) {
+ return fail("WKT瀛楃涓蹭笉姝ｇ‘");
+ }
+ if (null == pixel || !pixels.contains(pixel)) {
+ return fail("鍍忕偣鍊煎彧鑳戒负锛�" + StringHelper.join(pixels, ", "));
+ }
- return success(map);
+ List<?> rs = rasterService.analysisPoint(wkt, pixel);
+
+ return success(rs.size(), rs);
} catch (Exception ex) {
return fail(ex, null);
}
@@ -76,17 +64,22 @@
@SysLog()
@ApiOperation(value = "鏌ヨ绾垮垎鏋�")
@ApiImplicitParams({
- @ApiImplicitParam(name = "wkt", value = "绾縒KT", dataType = "String", example = "")
+ @ApiImplicitParam(name = "wkt", value = "绾縒KT", dataType = "String", example = "LINESTRING(115.94927385452 32.3754479115071,121.989371092554 32.2766788010181,121.850621222894 29.6874200067864)")
})
@GetMapping(value = "/selectByPolyline")
public ResponseMsg<Object> selectByPolyline(String wkt) {
try {
- Map<String, List<Double>> map = new HashMap<>(5);
- List<Double> list = new ArrayList<>();
- list.add(0.0);
- map.put("鍥惧眰鍚�", list);
+ if (StringHelper.isEmpty(wkt)) {
+ return fail("WKT瀛楃涓蹭笉鑳戒负绌�");
+ }
+ Geometry geo = Geometry.CreateFromWkt(wkt);
+ if (null == geo || geo.GetGeometryType() != ogr.wkbLineString) {
+ return fail("WKT瀛楃涓蹭笉姝ｇ‘");
+ }
- return success(map);
+ List<?> rs = rasterService.analysisPolyline(wkt);
+
+ return success(rs.size(), rs);
} catch (Exception ex) {
return fail(ex, null);
}
@@ -95,15 +88,22 @@
@SysLog()
@ApiOperation(value = "鏌ヨ闈㈠垎鏋�")
@ApiImplicitParams({
- @ApiImplicitParam(name = "wkt", value = "闈KT", dataType = "String", example = "")
+ @ApiImplicitParam(name = "wkt", value = "闈KT", dataType = "String", example = "POLYGON ((115.94927385452 32.3754479115071,121.989371092554 32.2766788010181,121.850621222894 29.6874200067864,115.9727267226 29.7835368627922,115.94927385452 32.3754479115071))")
})
@GetMapping(value = "/selectByPolygon")
public ResponseMsg<Object> selectByPolygon(String wkt) {
try {
- Map<String, Double> map = new HashMap<>(5);
- map.put("鍥惧眰鍚�", 0.0);
+ if (StringHelper.isEmpty(wkt)) {
+ return fail("WKT瀛楃涓蹭笉鑳戒负绌�");
+ }
+ Geometry geo = Geometry.CreateFromWkt(wkt);
+ if (null == geo || geo.GetGeometryType() != ogr.wkbPolygon) {
+ return fail("WKT瀛楃涓蹭笉姝ｇ‘");
+ }
- return success(map);
+ List<?> rs = rasterService.analysisPolygon(wkt);
+
+ return success(rs.size(), rs);
} catch (Exception ex) {
return fail(ex, null);
}
diff --git a/src/main/java/com/moon/server/entity/data/AnalysisResultEntity.java b/src/main/java/com/moon/server/entity/data/AnalysisResultEntity.java
new file mode 100644
index 0000000..b5da1e1
--- /dev/null
+++ b/src/main/java/com/moon/server/entity/data/AnalysisResultEntity.java
@@ -0,0 +1,55 @@
+package com.moon.server.entity.data;
+
+import java.io.Serializable;
+
+/**
+ * 鍒嗘瀽缁撴灉
+ * @author WWW
+ * @date 2023-09-11
+ */
+public class AnalysisResultEntity implements Serializable {
+ private static final long serialVersionUID = -1237623414044281355L;
+
+ public AnalysisResultEntity() {
+ }
+
+ private String layerName;
+
+ private Double min;
+
+ private Double avg;
+
+ private Double max;
+
+ public String getLayerName() {
+ return layerName;
+ }
+
+ public void setLayerName(String layerName) {
+ this.layerName = layerName;
+ }
+
+ public Double getMin() {
+ return min;
+ }
+
+ public void setMin(Double min) {
+ this.min = min;
+ }
+
+ public Double getAvg() {
+ return avg;
+ }
+
+ public void setAvg(Double avg) {
+ this.avg = avg;
+ }
+
+ public Double getMax() {
+ return max;
+ }
+
+ public void setMax(Double max) {
+ this.max = max;
+ }
+}
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 dad196e..fcd5f35 100644
--- a/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
+++ b/src/main/java/com/moon/server/service/data/RasterAnalysisService.java
@@ -1,5 +1,6 @@
package com.moon.server.service.data;
+import com.moon.server.entity.data.AnalysisResultEntity;
import org.gdal.gdal.Band;
import org.gdal.gdal.Dataset;
import org.gdal.gdal.WarpOptions;
@@ -15,79 +16,30 @@
/**
* 鏍呮牸鍒嗘瀽鏈嶅姟
* @author LJW
- * @date 2023-9-1 6:16
+ * @date 2023-09-01 6:16
*/
@Service
public class RasterAnalysisService {
- 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");
+ public List<AnalysisResultEntity> analysisPoint(String wkt, Integer pixel) {
+ List<AnalysisResultEntity> rs = new ArrayList<>();
+ //
- return gdal.Warp("", new Dataset[]{dataset}, new WarpOptions(warpOptions));
+ return rs;
}
- public double[] processClippedDataByLine(String imagePath, String geometryString) {
- // 娉ㄥ唽GDAL椹卞姩
- gdal.AllRegister();
- // 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
- Dataset dataset = gdal.Open(imagePath);
- if (dataset == null) {
- throw new RuntimeException("Failed to open raster dataset.");
- }
+ public List<AnalysisResultEntity> analysisPolyline(String wkt) {
+ List<AnalysisResultEntity> rs = new ArrayList<>();
+ //
- Geometry geometry = Geometry.CreateFromWkt(geometryString);
+ return rs;
+ }
+ public List<AnalysisResultEntity> analysisPolygon(String wkt) {
+ List<AnalysisResultEntity> rs = new ArrayList<>();
+ //
- Dataset clippedDataset = clipRaster(dataset, geometry);
-
- int width = clippedDataset.GetRasterXSize();
- int height = clippedDataset.GetRasterYSize();
- int bandCount = clippedDataset.getRasterCount();
-
- List<double[]> sum = new ArrayList<>();
- for (int bandIndex = 1; bandIndex <= bandCount; bandIndex++) {
- Band band = clippedDataset.GetRasterBand(bandIndex);
-
- double[] pixelValues = new double[width * height];
- band.ReadRaster(0, 0, width, height, pixelValues);
-
- sum.add(pixelValues);
- }
-
- return calculateAverage(sum);
+ return rs;
}
- public double processClippedDataByPolygon(String imagePath, String geometryString) {
- // 娉ㄥ唽GDAL椹卞姩
- gdal.AllRegister();
- // 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
- Dataset dataset = gdal.Open(imagePath);
- if (dataset == null) {
- throw new RuntimeException("Failed to open raster dataset.");
- }
-
- Geometry geometry = Geometry.CreateFromWkt(geometryString);
-
- Dataset clippedDataset = clipRaster(dataset, geometry);
-
- int width = clippedDataset.GetRasterXSize();
- int height = clippedDataset.GetRasterYSize();
- int bandCount = clippedDataset.getRasterCount();
-
- double[] bandSum = new double[bandCount];
- for (int bandIndex = 1; bandIndex <= bandCount; bandIndex++) {
- Band band = clippedDataset.GetRasterBand(bandIndex);
-
- double[] pixelValues = new double[width * height];
- band.ReadRaster(0, 0, width, height, pixelValues);
-
- // 澶氭尝娈靛綊涓�鍖栧鐞�
- bandSum[bandIndex - 1] = average(pixelValues);
- }
- return average(bandSum);
- }
public double processPoint(String imagePath, String geometryString, int size) {
// 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
@@ -219,6 +171,76 @@
return average(calculateAverage(sum));
}
+ 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) {
+ // 娉ㄥ唽GDAL椹卞姩
+ gdal.AllRegister();
+ // 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
+ Dataset dataset = gdal.Open(imagePath);
+ if (dataset == null) {
+ throw new RuntimeException("Failed to open raster dataset.");
+ }
+
+ Geometry geometry = Geometry.CreateFromWkt(geometryString);
+
+ Dataset clippedDataset = clipRaster(dataset, geometry);
+
+ int width = clippedDataset.GetRasterXSize();
+ int height = clippedDataset.GetRasterYSize();
+ int bandCount = clippedDataset.getRasterCount();
+
+ List<double[]> sum = new ArrayList<>();
+ for (int bandIndex = 1; bandIndex <= bandCount; bandIndex++) {
+ Band band = clippedDataset.GetRasterBand(bandIndex);
+
+ double[] pixelValues = new double[width * height];
+ band.ReadRaster(0, 0, width, height, pixelValues);
+
+ sum.add(pixelValues);
+ }
+
+ return calculateAverage(sum);
+ }
+
+ public double processClippedDataByPolygon(String imagePath, String geometryString) {
+ // 娉ㄥ唽GDAL椹卞姩
+ gdal.AllRegister();
+ // 鎵撳紑鏍呮牸鍥惧儚鏁版嵁闆�
+ Dataset dataset = gdal.Open(imagePath);
+ if (dataset == null) {
+ throw new RuntimeException("Failed to open raster dataset.");
+ }
+
+ Geometry geometry = Geometry.CreateFromWkt(geometryString);
+
+ Dataset clippedDataset = clipRaster(dataset, geometry);
+
+ int width = clippedDataset.GetRasterXSize();
+ int height = clippedDataset.GetRasterYSize();
+ int bandCount = clippedDataset.getRasterCount();
+
+ double[] bandSum = new double[bandCount];
+ for (int bandIndex = 1; bandIndex <= bandCount; bandIndex++) {
+ Band band = clippedDataset.GetRasterBand(bandIndex);
+
+ double[] pixelValues = new double[width * height];
+ 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) {
--
Gitblit v1.9.3