月球大数据地理空间分析展示平台-【后端】-月球后台服务
13693261870
2023-09-11 e6a222fea235eb4ebb3fc47d95c61a07a2db5e15
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
package com.moon.server.controller.data;
 
import com.moon.server.annotation.SysLog;
import com.moon.server.controller.all.BaseController;
import com.moon.server.entity.all.ResponseMsg;
import com.moon.server.helper.StringHelper;
import com.moon.server.helper.WebHelper;
import com.moon.server.service.data.RasterAnalysisService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.gdal.ogr.Geometry;
import org.gdal.ogr.ogr;
import org.springframework.web.bind.annotation.*;
 
import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.util.*;
 
/**
 * 栅格分析
 * @author WWW
 * @date   2023-08-23
 */
@Api(tags = "数据管理\\栅格分析")
@RestController
@RequestMapping("/rasterAnalysis")
public class RasterAnalysisController extends BaseController {
    @Resource
    RasterAnalysisService rasterService;
 
    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 = "点WKT", 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, Integer pixel) {
        try {
            // 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, ", "));
            }
 
            List<?> rs = rasterService.analysis(geo, pixel);
 
            return success(rs.size(), rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }
 
    @SysLog()
    @ApiOperation(value = "查询线分析")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "wkt", value = "线WKT", 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 {
            if (StringHelper.isEmpty(wkt)) {
                return fail("WKT字符串不能为空");
            }
            Geometry geo = Geometry.CreateFromWkt(wkt);
            if (null == geo || geo.GetGeometryType() != ogr.wkbLineString) {
                return fail("WKT字符串不正确");
            }
 
            List<?> rs = rasterService.analysis(geo, 0);
 
            return success(rs.size(), rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }
 
    @SysLog()
    @ApiOperation(value = "查询面分析")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "wkt", value = "面WKT", 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 {
            if (StringHelper.isEmpty(wkt)) {
                return fail("WKT字符串不能为空");
            }
            Geometry geo = Geometry.CreateFromWkt(wkt);
            if (null == geo || geo.GetGeometryType() != ogr.wkbPolygon) {
                return fail("WKT字符串不正确");
            }
 
            List<?> rs = rasterService.analysis(geo, 0);
 
            return success(rs.size(), rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }
 
    @SysLog()
    @ApiOperation(value = "下载点分析")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "wkt", value = "点WKT", dataType = "String", example = "")
    })
    @GetMapping(value = "/downloadByPoint")
    public void downloadByPoint(String wkt, HttpServletResponse res) {
        try {
            // ...
 
            String filePath = "生成文件的路径";
            WebHelper.download(filePath, "文件名", res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }
    }
 
    @SysLog()
    @ApiOperation(value = "下载线分析")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "wkt", value = "线WKT", dataType = "String", example = "")
    })
    @GetMapping(value = "/downloadByPolyline")
    public void downloadByPolyline(String wkt, HttpServletResponse res) {
        try {
            // ...
 
            String filePath = "生成文件的路径";
            WebHelper.download(filePath, "文件名", res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }
    }
 
    @SysLog()
    @ApiOperation(value = "下载面分析")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "wkt", value = "面WKT", dataType = "String", example = "")
    })
    @GetMapping(value = "/downloadByPolygon")
    public void downloadByPolygon(String wkt, HttpServletResponse res) {
        try {
            // ...
 
            String filePath = "生成文件的路径";
            WebHelper.download(filePath, "文件名", res);
        } catch (Exception ex) {
            log.error(ex.getMessage(), ex);
        }
    }
}