内涝仿真服务
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
张洋洋
2025-01-10 f851dbe9fba3ff30605039b0cb1f69892e3483d0 
 src/main/java/com/se/simu/controller/SimuController.java


@@ -3,27 +3,31 @@


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.po.DataPo;


import com.se.simu.domain.po.SimuPo;


import com.se.simu.domain.vo.CreateFilesSimuVo;


import com.se.simu.domain.vo.CreateSimuVo;


import com.se.simu.domain.vo.R;


import com.se.simu.domain.vo.SimuVo;


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.ZipUtils;


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


@@ -271,13 +275,13 @@


        String stationRainFile = vo.getStationRainFile();


        log.info("站点雨量CSV文件地址：{}", floodFile);


        JSONArray array = new JSONArray();


        // 创建表名 时间戳


        String tableName = "station_rain_" + System.currentTimeMillis();


        try {


            // 判断地址不为空


            if (StringUtils.isEmpty(stationRainFile)) {


                return fail("站点雨量CSV文件地址不能为空", false);


            }


            // 创建表名 时间戳


            String tableName = "station_rain_" + System.currentTimeMillis();


            // 1 读取CSV 文件


            array = CsvToSQLiteUtils.readCsvSaveLocal(stationRainFile, tableName);


            // 获取仿真时间 duration


@@ -341,5 +345,79 @@


        }


    }




    @ApiOperation(value = "tarr文件组装")


    @GetMapping(value = "/testZarr", produces = "application/json; charset=UTF-8")


    public R<Boolean> testZarr(@RequestParam("name") String tableName) throws IOException {


        List<String> list = CsvToSQLiteUtils.getNameList(tableName);


        String path = "D:\\城市内涝\\sem\\雨量站点数据\\";


        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);


            ZipUtils.toZarr(basePath + src, basePath + src + ".zip");


            System.out.println(src + "的zarr数据生成=====================");


            //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);


    }




    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;


    }




}