From 1ad7b12d159e4b69e8c23670e64de1700e2c8ed4 Mon Sep 17 00:00:00 2001
From: 张洋洋 <10611411+yang-yang-z@user.noreply.gitee.com>
Date: 星期二, 14 一月 2025 11:53:57 +0800
Subject: [PATCH] [add]管线json
---
src/main/java/com/se/simu/controller/SimuController.java | 178 ++++++++++++++++++++++++++++++++++++++++++++++++++++++----
1 files changed, 164 insertions(+), 14 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..6c0aa4e 100644
--- a/src/main/java/com/se/simu/controller/SimuController.java
+++ b/src/main/java/com/se/simu/controller/SimuController.java
@@ -3,6 +3,7 @@
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;
@@ -10,18 +11,20 @@
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.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Value;
+import org.springframework.core.io.ClassPathResource;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
-import java.io.IOException;
+import java.io.*;
+import java.net.URL;
+import java.nio.charset.StandardCharsets;
import java.util.*;
+import java.util.stream.Collectors;
@Api(tags = "浠跨湡绠＄悊")
@Slf4j
@@ -46,6 +49,9 @@
@Value("${simu-app.filePath}")
private String uploadedFolder;
+
+ @Value("${config.outPath}")
+ private String outPath;
@Resource
@@ -327,10 +333,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 +342,166 @@
}
@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<Boolean> saveZarr(@RequestParam("name") String tableName) throws IOException {
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());
+ metry.put("boundaries", boundarie);
+ geometry.add(metry);
+ cityObject.put("geometry", geometry);
+ jsonObject.getJSONObject("CityObjects").put(uuid, cityObject);
}
+ 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();
return success(true);
}
+ @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());
+ 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();
+ return success(pointPath);
+ }
+
+ @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());
+ }
+ JSONArray jsonArray=new JSONArray();
+ jsonArray.add(boundarie);
+ metry.put("boundaries", jsonArray);
+ cityObject.put("geometry",metry);
+ 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();
+ return success(pointPath);
+ }
+
+ 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