From 49aa6ed510d18506cc11d24299899ab0b35047b3 Mon Sep 17 00:00:00 2001 From: 张洋洋 <10611411+yang-yang-z@user.noreply.gitee.com> Date: 星期二, 11 二月 2025 18:20:59 +0800 Subject: [PATCH] [add]cityjson转sem --- src/main/java/com/se/simu/controller/SimuController.java | 386 +++++++++++++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 371 insertions(+), 15 deletions(-) diff --git a/src/main/java/com/se/simu/controller/SimuController.java b/src/main/java/com/se/simu/controller/SimuController.java index f8d151d..891f0b1 100644 --- a/src/main/java/com/se/simu/controller/SimuController.java +++ b/src/main/java/com/se/simu/controller/SimuController.java @@ -3,25 +3,32 @@ import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.metadata.IPage; +import com.google.common.io.Resources; import com.se.simu.config.PropertiesConfig; import com.se.simu.domain.dto.GeDb; import com.se.simu.domain.dto.GeLayer; +import com.se.simu.domain.dto.GridDto; import com.se.simu.domain.po.DataPo; import com.se.simu.domain.po.SimuPo; import com.se.simu.domain.vo.*; import com.se.simu.service.*; -import com.se.simu.utils.CsvToSQLiteUtils; -import com.se.simu.utils.ShpToolUtils; -import com.se.simu.utils.ZarrUtils; +import com.se.simu.utils.*; import io.swagger.annotations.*; import lombok.extern.slf4j.Slf4j; +import org.apache.commons.codec.binary.Base64; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Value; +import org.springframework.core.io.ClassPathResource; import org.springframework.web.bind.annotation.*; +import org.springframework.web.util.UriUtils; import javax.annotation.Resource; -import java.io.IOException; +import java.io.*; +import java.net.URL; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; import java.util.*; +import java.util.stream.Collectors; @Api(tags = "浠跨湡绠$悊") @Slf4j @@ -46,6 +53,12 @@ @Value("${simu-app.filePath}") private String uploadedFolder; + + @Value("${config.outPath}") + private String outPath; + + @Value("${config.inPath}") + private String inPath; @Resource @@ -327,10 +340,6 @@ return fail("闃叉睕鑼冨洿涓嶈兘涓虹┖", false); } } - //todo 缁勮cityjson - JSONObject jsonObject = new JSONObject(); - jsonObject.put("shp", jsonArray); - jsonObject.put("csv", array); // 寮�濮嬫ā鎷熻绠� boolean flag = simuFilesService.createByfiles(vo); return success(flag, flag ? "鎴愬姛" : "澶辫触"); @@ -340,18 +349,365 @@ } @ApiOperation(value = "tarr鏂囦欢缁勮") - @GetMapping(value = "/testZarr", produces = "application/json; charset=UTF-8") - public R<Boolean> testZarr(@RequestParam("name") String tableName) throws IOException { + @GetMapping(value = "/saveZarr", produces = "application/json; charset=UTF-8") + public R<String> saveZarr(@RequestParam("name") String tableName) throws Exception { List<String> list = CsvToSQLiteUtils.getNameList(tableName); - String basePath="D:\\鍩庡競鍐呮稘\\sem\\闆ㄩ噺绔欑偣鏁版嵁\\闄嶉洦閲弞arr\\"; + String path = outPath + "\\"; + String rainfall = "rainfall\\"; + String basePath = path + tableName + "\\" + rainfall; + JSONObject jsonObject = getModule("rainfallmodule.json"); for (String src : list ) { + File directories = new File(basePath); + if (!directories.exists()) { + directories.mkdirs(); + System.out.println("Directories created successfully."); + } else { + System.out.println("Directories already exist."); + } List<StationRainVo> stationRainVos = CsvToSQLiteUtils.getList(tableName, src); - ZarrUtils.saveZarrRainfall(basePath+src,stationRainVos); - ZarrUtils.saveZarrTime(basePath+src,stationRainVos); - System.out.println(src+"鐨剒arr鏁版嵁鐢熸垚====================="); + ZarrUtils.saveZarrRainfall(basePath + src, stationRainVos); + ZarrUtils.saveZarrTime(basePath + src, stationRainVos); + ZipUtils.toZarr(basePath + src, basePath + src + ".zip"); + System.out.println(src + "鐨剒arr鏁版嵁鐢熸垚====================="); + //json鎷艰 + String uuid = "UUID_" + UUID.randomUUID().toString(); + //鎷艰zarr + JSONObject dynamizer = new JSONObject(); + dynamizer.put("url", rainfall.replace("\\", "/") + src + ".zarr"); + dynamizer.put("gmlId", uuid); + jsonObject.getJSONArray("Dynamizers").add(dynamizer); + //鎷艰鍧愭爣 + JSONArray vertice = new JSONArray(); + vertice.add(stationRainVos.get(0).getLongitude()); + vertice.add(stationRainVos.get(0).getLatitude()); + vertice.add(0.0); + jsonObject.getJSONArray("vertices").add(vertice); + //鎷艰鍩虹淇℃伅 + JSONObject cityObject = new JSONObject(); + cityObject.put("type", "+Rainfall"); + JSONObject attribute = new JSONObject(); + attribute.put("name", src); + cityObject.put("attributes", attribute); + JSONArray geometry = new JSONArray(); + JSONObject metry = new JSONObject(); + metry.put("type", "MultiPoint"); + metry.put("lod", 0); + JSONArray boundarie = new JSONArray(); + boundarie.add(jsonObject.getJSONArray("vertices").size() - 1); + metry.put("boundaries", boundarie); + geometry.add(metry); + cityObject.put("geometry", geometry); + jsonObject.getJSONObject("CityObjects").put(uuid, cityObject); } - return success(true); + File jsonFile = new File(path + tableName + "\\闄嶉洦閲�.json"); + if (jsonFile.exists()) { + jsonFile.createNewFile(); + } + FileWriter fileWriter = new FileWriter(path + tableName + "\\闄嶉洦閲�.json"); + fileWriter.write(jsonObject.toJSONString()); + fileWriter.close(); + String filePatn = inPath + "\\tongzhou\\"; + File file = new File(filePatn); + if (!file.exists()) { + file.mkdirs(); + } + SemUtils.cityJsonToSem(path + tableName + "\\闄嶉洦閲�.json", filePatn + "raingage.sem"); + return success(filePatn + "raingage.sem"); + } + + @ApiOperation(value = "绠$偣杞琧ityjson") + @GetMapping(value = "/pointToCityJson", produces = "application/json; charset=UTF-8") + public R<String> pointToCityJson(@RequestParam("shpPath") String shpPath) throws Exception { + String path = outPath + "\\point\\"; + List<Map<String, Object>> list = ShpReadUtils.readPointShp(shpPath); + JSONObject jsonObject = getModule("pointmodule.json"); + for (Map<String, Object> map : list + ) { + //鎷艰鍧愭爣 + JSONArray vertice = new JSONArray(); + vertice.add(map.get("lon")); + vertice.add(map.get("lat")); + vertice.add(0.0); + jsonObject.getJSONArray("vertices").add(vertice); + //鎷艰鍩虹淇℃伅 + JSONObject cityObject = new JSONObject(); + cityObject.put("type", "+PipePoint"); + JSONObject attribute = new JSONObject(); + attribute.put("name", map.get("fsw")); + cityObject.put("attributes", attribute); + JSONArray geometry = new JSONArray(); + JSONObject metry = new JSONObject(); + metry.put("type", "MultiPoint"); + metry.put("lod", 0); + JSONArray boundarie = new JSONArray(); + boundarie.add(jsonObject.getJSONArray("vertices").size() - 1); + metry.put("boundaries", boundarie); + geometry.add(metry); + cityObject.put("geometry", geometry); + jsonObject.getJSONObject("CityObjects").put("UUID_" + UUID.randomUUID().toString(), cityObject); + } + long times = System.currentTimeMillis(); + String pointPath = path + times + "\\绠$偣.json"; + File dirFile = new File(path + times); + if (!dirFile.exists()) { + dirFile.mkdirs(); + } + FileWriter fileWriter = new FileWriter(pointPath); + fileWriter.write(jsonObject.toJSONString()); + fileWriter.close(); + String filePatn = inPath + "\\tongzhou\\"; + File file = new File(filePatn); + if (!file.exists()) { + file.mkdirs(); + } + SemUtils.cityJsonToSem(pointPath, filePatn + "node.sem"); + return success(filePatn + "node.sem"); + } + + @ApiOperation(value = "绠$嚎sem鐢熸垚") + @PostMapping(value = "/lineToSem", produces = "application/json; charset=UTF-8") + public R<String> lineToSem(@RequestBody List<GridDto> dtos) throws Exception { + String path = outPath + "\\line\\"; + JSONArray jsonArrayList = EntityLibraryUtils.getPointInfo(dtos); + if (jsonArrayList.size() == 0) { + return fail("璇ュ尯鍩熶笉瀛樺湪绠$嚎锛�"); + } + StringBuffer stringBuffer = new StringBuffer("id in ("); + for (int i = 0; i < jsonArrayList.size(); i++) { + jsonArrayList.getJSONObject(i).getInteger("id"); + stringBuffer.append("'" + jsonArrayList.getJSONObject(i).getInteger("id") + "',"); + } + String sql = stringBuffer.toString(); + sql = sql.substring(0, sql.length() - 1) + ")"; + String publicKey = EntityLibraryUtils.getPublicKey(); + StringBuffer paramBuffer = new StringBuffer(""); + while (!"".equals(sql)) { + if (sql.length() > 110) { + String item = sql.substring(0, 110); + paramBuffer.append(EntityLibraryUtils.encrypt(item, publicKey) + ","); + sql = sql.replace(item, ""); + } else { + paramBuffer.append(EntityLibraryUtils.encrypt(sql, publicKey) + ","); + sql = ""; + } + } + String src=paramBuffer.toString().substring(0, paramBuffer.length() - 1); + JSONArray arrayDetail = EntityLibraryUtils.getPointDetail(src); + //todo 鏇挎崲mapencodedString + List<Map<String, Object>> list = new ArrayList<>(); + JSONObject jsonObject = getModule("linemodule.json"); + for (Map<String, Object> map : list + ) { + //鎷艰鍩虹淇℃伅 + JSONObject cityObject = new JSONObject(); + cityObject.put("type", "+PipeLine"); + JSONObject attribute = new JSONObject(); + attribute.put("name", map.get("msfs")); + JSONArray geometry = new JSONArray(); + JSONObject metry = new JSONObject(); + metry.put("type", "MultiLineString"); + metry.put("lod", 0); + JSONArray boundarie = new JSONArray(); + JSONArray array = JSONObject.parseObject(map.get("the_geom").toString()).getJSONArray("coordinates"); + for (int i = 0; i < array.size(); i++) { + JSONObject object = JSONObject.parseObject(array.get(i).toString()); + jsonObject.getJSONArray("vertices").add(ProjectionToGeographicUtil.getPoint(Double.valueOf(object.get("x").toString()), Double.valueOf(object.get("y").toString()))); + boundarie.add(jsonObject.getJSONArray("vertices").size() - 1); + } + JSONArray jsonArray = new JSONArray(); + jsonArray.add(boundarie); + metry.put("boundaries", jsonArray); + JSONArray metryArray = new JSONArray(); + metryArray.add(metry); + cityObject.put("geometry", metryArray); + cityObject.put("attributes", attribute); + cityObject.put("attributes", attribute); + geometry.add(metry); + jsonObject.getJSONObject("CityObjects").put("UUID_" + UUID.randomUUID().toString(), cityObject); + } + long times = System.currentTimeMillis(); + String pointPath = path + times + "\\绠$嚎.json"; + File dirFile = new File(path + times); + if (!dirFile.exists()) { + dirFile.mkdirs(); + } + FileWriter fileWriter = new FileWriter(pointPath); + fileWriter.write(jsonObject.toJSONString()); + fileWriter.close(); + String filePatn = inPath + "\\tongzhou\\"; + File file = new File(filePatn); + if (!file.exists()) { + file.mkdirs(); + } + SemUtils.cityJsonToSem(pointPath, filePatn + "link.sem"); + return success(filePatn + "link.sem"); + } +// @ApiOperation(value = "绠$嚎杞琧ityjson") +// @GetMapping(value = "/lineToCityJson", produces = "application/json; charset=UTF-8") +// public R<String> lineToCityJson(@RequestParam("shpPath") String shpPath) throws Exception { +// String path = outPath + "\\line\\"; +// List<Map<String, Object>> list = ShpReadUtils.readPointShp(shpPath); +// JSONObject jsonObject = getModule("linemodule.json"); +// for (Map<String, Object> map : list +// ) { +// //鎷艰鍩虹淇℃伅 +// JSONObject cityObject = new JSONObject(); +// cityObject.put("type", "+PipeLine"); +// JSONObject attribute = new JSONObject(); +// attribute.put("name", map.get("msfs")); +// JSONArray geometry = new JSONArray(); +// JSONObject metry = new JSONObject(); +// metry.put("type", "MultiLineString"); +// metry.put("lod", 0); +// JSONArray boundarie = new JSONArray(); +// JSONArray array = JSONObject.parseObject(map.get("the_geom").toString()).getJSONArray("coordinates"); +// for (int i = 0; i < array.size(); i++) { +// JSONObject object = JSONObject.parseObject(array.get(i).toString()); +// jsonObject.getJSONArray("vertices").add(ProjectionToGeographicUtil.getPoint(Double.valueOf(object.get("x").toString()), Double.valueOf(object.get("y").toString()))); +// boundarie.add(jsonObject.getJSONArray("vertices").size() - 1); +// } +// JSONArray jsonArray = new JSONArray(); +// jsonArray.add(boundarie); +// metry.put("boundaries", jsonArray); +// JSONArray metryArray = new JSONArray(); +// metryArray.add(metry); +// cityObject.put("geometry", metryArray); +// cityObject.put("attributes", attribute); +// cityObject.put("attributes", attribute); +// geometry.add(metry); +// jsonObject.getJSONObject("CityObjects").put("UUID_" + UUID.randomUUID().toString(), cityObject); +// } +// long times = System.currentTimeMillis(); +// String pointPath = path + times + "\\绠$嚎.json"; +// File dirFile = new File(path + times); +// if (!dirFile.exists()) { +// dirFile.mkdirs(); +// } +// FileWriter fileWriter = new FileWriter(pointPath); +// fileWriter.write(jsonObject.toJSONString()); +// fileWriter.close(); +// String filePatn = inPath + "\\tongzhou\\"; +// File file = new File(filePatn); +// if (!file.exists()) { +// file.mkdirs(); +// } +// SemUtils.cityJsonToSem(pointPath, filePatn+ "link.sem"); +// return success(filePatn + "link.sem"); +// } + + @ApiOperation(value = "鍦板舰杞琧ityjson") + @GetMapping(value = "/terrainToCityJson", produces = "application/json; charset=UTF-8") + public R<String> terrainToCityJson(@RequestParam("tifPath") String tifPath) throws Exception { + long times = System.currentTimeMillis(); + String path = outPath + "\\terrain\\" + times + "\\"; + File dirFile = new File(path + "appearance"); + if (!dirFile.exists()) { + dirFile.mkdirs(); + } + String pngPath = path + "appearance\\terrain.png"; + TiffToRGBUtil.tifToPng(tifPath, pngPath); + JSONObject jsonObject = getModule("terrainmodule.json"); + jsonObject.put("vertices", TiffCoordinateExtractorUtil.getCoordinate(tifPath)); + FileWriter fileWriter = new FileWriter(path + "terrain.json"); + fileWriter.write(jsonObject.toJSONString()); + fileWriter.close(); + String filePatn = inPath + "\\tongzhou\\"; + File file = new File(filePatn); + if (!file.exists()) { + file.mkdirs(); + } + SemUtils.cityJsonToSem(path + "terrain.json", filePatn + "terrain.sem"); + return success(filePatn + "terrain.sem"); + } + + @ApiOperation(value = "grid杞琧ityjson") + @PostMapping(value = "/gridToCityJson", produces = "application/json; charset=UTF-8") + public R<String> gridToCityJson(@RequestBody List<GridDto> dtos) throws Exception { + long times = System.currentTimeMillis(); + String path = outPath + "\\grid\\" + times + "\\"; + File dirFile = new File(path); + if (!dirFile.exists()) { + dirFile.mkdirs(); + } + JSONObject jsonObject = getModule("grid.json"); + JSONArray array = jsonObject.getJSONArray("vertices"); + for (GridDto dto : dtos + ) { + JSONArray jsonArray = new JSONArray(); + jsonArray.add(dto.getLon()); + jsonArray.add(dto.getLat()); + jsonArray.add(0); + array.add(jsonArray); + } + jsonObject.put("vertices", array); + FileWriter fileWriter = new FileWriter(path + "grid.json"); + fileWriter.write(jsonObject.toJSONString()); + fileWriter.close(); + String filePatn = inPath + "\\tongzhou\\"; + File file = new File(filePatn); + if (!file.exists()) { + file.mkdirs(); + } + SemUtils.cityJsonToSem(path + "grid.json", filePatn + "grid.sem"); + return success(filePatn + "grid.sem"); + } + + @ApiOperation(value = "river杞瑂em") + @GetMapping(value = "/riverToSem", produces = "application/json; charset=UTF-8") + public R<String> riverToSem() throws Exception { + long times = System.currentTimeMillis(); + String path = outPath + "\\river\\" + times + "\\"; + File dirFile = new File(path); + if (!dirFile.exists()) { + dirFile.mkdirs(); + } + JSONObject jsonObject = getModule("river.json"); + FileWriter fileWriter = new FileWriter(path + "river.json"); + fileWriter.write(jsonObject.toJSONString()); + fileWriter.close(); + String filePatn = inPath + "\\tongzhou\\"; + File file = new File(filePatn); + if (!file.exists()) { + file.mkdirs(); + } + SemUtils.cityJsonToSem(path + "river.json", filePatn + "river.sem"); + return success(filePatn + "river.sem"); + } + + @ApiOperation(value = "landuse杞瑂em") + @GetMapping(value = "/landuseToSem", produces = "application/json; charset=UTF-8") + public R<String> landuseToSem() throws Exception { + long times = System.currentTimeMillis(); + String path = outPath + "\\landuse\\" + times + "\\"; + File dirFile = new File(path); + if (!dirFile.exists()) { + dirFile.mkdirs(); + } + JSONObject jsonObject = getModule("landuse.json"); + FileWriter fileWriter = new FileWriter(path + "landuse.json"); + fileWriter.write(jsonObject.toJSONString()); + fileWriter.close(); + String filePatn = inPath + "\\tongzhou\\"; + File file = new File(filePatn); + if (!file.exists()) { + file.mkdirs(); + } + SemUtils.cityJsonToSem(path + "landuse.json", filePatn + "landuse.sem"); + return success(filePatn + "landuse.sem"); + } + + public JSONObject getModule(String moduleName) { + JSONObject jsonObject = new JSONObject(); + try { + URL resource = Resources.getResource(moduleName); + String fileContent = Resources.toString(resource, StandardCharsets.UTF_8); + jsonObject = JSONObject.parseObject(fileContent); + System.out.println(fileContent); + } catch (Exception e) { + e.printStackTrace(); + } + return jsonObject; } } -- Gitblit v1.9.3