From fee67ca8a0760315047a52fc4101a8f4f80b7a7f Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期一, 11 十一月 2024 15:02:54 +0800 Subject: [PATCH] 1 --- src/main/java/com/moon/server/controller/data/RasterAnalysisController.java | 168 ++++++++++++++++++++++++++++++++++++------------------- 1 files changed, 109 insertions(+), 59 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 71e4217..7760150 100644 --- a/src/main/java/com/moon/server/controller/data/RasterAnalysisController.java +++ b/src/main/java/com/moon/server/controller/data/RasterAnalysisController.java @@ -3,43 +3,62 @@ 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.all.StaticData; +import com.moon.server.helper.StringHelper; import com.moon.server.helper.WebHelper; +import com.moon.server.service.data.RasterAnalysisService; +import com.moon.server.service.data.SlopeAnalysisService; import io.swagger.annotations.Api; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; import io.swagger.annotations.ApiOperation; -import org.springframework.web.bind.annotation.GetMapping; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RestController; +import org.gdal.ogr.Geometry; +import org.gdal.ogr.ogr; +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.*; -/** - * 鏍呮牸鍒嗘瀽 - * @author WWW - * @date 2023-08-23 - */ @Api(tags = "鏁版嵁绠$悊\\鏍呮牸鍒嗘瀽") @RestController +@SuppressWarnings("ALL") @RequestMapping("/rasterAnalysis") public class RasterAnalysisController extends BaseController { + @Resource + RasterAnalysisService rasterService; + + @Resource + SlopeAnalysisService slopeAnalysisService; + + private final static List<Integer> PIXELS = new ArrayList<>(Arrays.asList(1, 2, 4, 8, 16, 32, 64, 128, 256)); + + private final static List<Integer> NODES = new ArrayList<>(Arrays.asList(16, 32, 64, 96, 128, 192, 256, 384, 512, 768, 1024)); + @SysLog() @ApiOperation(value = "鏌ヨ鐐瑰垎鏋�") @ApiImplicitParams({ - @ApiImplicitParam(name = "wkt", value = "鐐筗KT", dataType = "String", example = "") + @ApiImplicitParam(name = "wkt", value = "鐐筗KT", dataType = "String", example = "POINT (157.750107 29.036669)"), + @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); + if (StringHelper.isEmpty(wkt)) { + return fail("WKT瀛楃涓蹭笉鑳戒负绌�"); + } + Geometry geo = Geometry.CreateFromWkt(wkt); + if (null == geo || geo.GetGeometryType() != ogr.wkbPoint) { + return fail("WKT瀛楃涓蹭笉姝g‘"); + } + if (null == pixel || !PIXELS.contains(pixel)) { + return fail("鍍忕礌鍊煎彧鑳戒负锛�" + StringHelper.join(PIXELS, ", ")); + } - return success(map); + List<?> rs = rasterService.analysis(geo, pixel); + + return success(rs.size(), rs); } catch (Exception ex) { return fail(ex, null); } @@ -48,17 +67,26 @@ @SysLog() @ApiOperation(value = "鏌ヨ绾垮垎鏋�") @ApiImplicitParams({ - @ApiImplicitParam(name = "wkt", value = "绾縒KT", dataType = "String", example = "") + @ApiImplicitParam(name = "wkt", value = "绾縒KT", dataType = "String", example = "LINESTRING(165.680851 31.333443,166.383982 31.283475,166.016355 30.908709)"), + @ApiImplicitParam(name = "nodes", value = "鑺傜偣鏁�", dataType = "Integer", example = "16") }) @GetMapping(value = "/selectByPolyline") - public ResponseMsg<Object> selectByPolyline(String wkt) { + public ResponseMsg<Object> selectByPolyline(String wkt, Integer nodes) { 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瀛楃涓蹭笉姝g‘"); + } + if (null == nodes || !NODES.contains(nodes)) { + return fail("鑺傜偣鏁板彧鑳戒负锛�" + StringHelper.join(NODES, ", ")); + } - return success(map); + List<?> rs = rasterService.analysis(geo, nodes); + + return success(rs.size(), rs); } catch (Exception ex) { return fail(ex, null); } @@ -67,67 +95,89 @@ @SysLog() @ApiOperation(value = "鏌ヨ闈㈠垎鏋�") @ApiImplicitParams({ - @ApiImplicitParam(name = "wkt", value = "闈KT", dataType = "String", example = "") + //@ApiImplicitParam(name = "wkt", value = "闈KT", dataType = "String", example = "POLYGON ((165.680851 31.333443,166.383982 31.283475,166.016355 30.908709,165.680851 31.333443))") + @ApiImplicitParam(name = "wkt", value = "闈KT", dataType = "String", example = "POLYGON ((56.61 33.94,115.04 33.56,114.09 -7.17,52.22 -6.22,56.61 33.94))") }) @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瀛楃涓蹭笉姝g‘"); + } - return success(map); + List<?> rs = rasterService.analysis(geo, 0); + + return success(rs.size(), rs); } catch (Exception ex) { return fail(ex, null); } } @SysLog() - @ApiOperation(value = "涓嬭浇鐐瑰垎鏋�") + @ApiOperation(value = "浣跨敤WKT鏌ヨ鍒嗘瀽锛岀粨鏋滀互娑堟伅鎺ㄩ��") @ApiImplicitParams({ - @ApiImplicitParam(name = "wkt", value = "鐐筗KT", dataType = "String", example = "") + @ApiImplicitParam(name = "wkt", value = "WKT瀛楃涓�", dataType = "String", example = "POLYGON ((56.61 33.94,115.04 33.56,114.09 -7.17,52.22 -6.22,56.61 33.94))"), + @ApiImplicitParam(name = "size", value = "鍍忕礌鍊�(鐐�)鎴栬妭鐐规暟(绾�)", dataType = "Integer", example = "16") }) - @RequestMapping(value = "/downloadByPoint") - public void downloadByPoint(String wkt, HttpServletRequest req, HttpServletResponse res) { + @GetMapping(value = "/selectByWktForPost") + public void selectByWktForPost(String wkt, Integer size, HttpServletRequest req, HttpServletResponse res) { try { - // ... + if (StringHelper.isEmpty(wkt)) { + WebHelper.writeStr2Page(res, "WKT瀛楃涓蹭笉鑳戒负绌�"); + return; + } + Geometry geo = Geometry.CreateFromWkt(wkt); + if (null == geo) { + WebHelper.writeStr2Page(res, "WKT瀛楃涓蹭笉姝g‘"); + return; + } + int wktType = geo.GetGeometryType(); + if (wktType != ogr.wkbPoint && wktType != ogr.wkbLineString && wktType != ogr.wkbPolygon) { + WebHelper.writeStr2Page(res, "WKT瀛楃涓插彧鏀寔鐐广�佺嚎涓插拰闈㈢被鍨�"); + return; + } + if (null == size || size < 0 || size > StaticData.I1024) { + WebHelper.writeStr2Page(res, "鍍忕礌鍊�(鐐�)鎴栬妭鐐规暟(绾�)鍊间笉姝g‘"); + return; + } - String filePath = "鐢熸垚鏂囦欢鐨勮矾寰�"; - WebHelper.download(filePath, "鏂囦欢鍚�", res); + String token = WebHelper.getToken(req); + rasterService.analysisForPost(geo, size, token); } catch (Exception ex) { log.error(ex.getMessage(), ex); } } @SysLog() - @ApiOperation(value = "涓嬭浇绾垮垎鏋�") + @ApiOperation(value = "涓嬭浇鍧″害鍒嗘瀽Excel") @ApiImplicitParams({ - @ApiImplicitParam(name = "wkt", value = "绾縒KT", dataType = "String", example = "") + @ApiImplicitParam(name = "wkt", value = "WKT瀛楃涓�", dataType = "String", example = "POLYGON ((-73.093 4.598,-72.494 4.669,-72.951 4.241,-73.093 4.598))") }) - @RequestMapping(value = "/downloadByPolyline") - public void downloadByPolyline(String wkt, HttpServletRequest req, HttpServletResponse res) { + @GetMapping(value = "/downloadSlopXls") + public void downloadSlopXls(String wkt, HttpServletRequest req, HttpServletResponse res) { try { - // ... + if (StringHelper.isEmpty(wkt)) { + WebHelper.writeStr2Page(res, "WKT瀛楃涓蹭笉鑳戒负绌�"); + return; + } + Geometry geo = Geometry.CreateFromWkt(wkt); + if (null == geo) { + WebHelper.writeStr2Page(res, "WKT瀛楃涓蹭笉姝g‘"); + return; + } + int wktType = geo.GetGeometryType(); + if (wktType != ogr.wkbPolygon) { + WebHelper.writeStr2Page(res, "WKT瀛楃涓插彧鏀寔闈㈢被鍨�"); + return; + } - String filePath = "鐢熸垚鏂囦欢鐨勮矾寰�"; - WebHelper.download(filePath, "鏂囦欢鍚�", res); + slopeAnalysisService.downloadSlopXls(geo, res); } catch (Exception ex) { - log.error(ex.getMessage(), ex); - } - } - - @SysLog() - @ApiOperation(value = "涓嬭浇闈㈠垎鏋�") - @ApiImplicitParams({ - @ApiImplicitParam(name = "wkt", value = "闈KT", dataType = "String", example = "") - }) - @RequestMapping(value = "/downloadByPolygon") - public void downloadByPolygon(String wkt, HttpServletRequest req, HttpServletResponse res) { - try { - // ... - - String filePath = "鐢熸垚鏂囦欢鐨勮矾寰�"; - WebHelper.download(filePath, "鏂囦欢鍚�", res); - } catch (Exception ex) { + WebHelper.writeStr2Page(res, ex.getMessage()); log.error(ex.getMessage(), ex); } } -- Gitblit v1.9.3