[add]tif裁剪

张洋洋

2025-02-18 eaf2eddc15c08bc073c953630b727f6497dfee3a

 src/main/java/com/se/simu/controller/SimuController.java

@@ -244,10 +244,6 @@
                        Double minY = jsonObject.getDouble("minX");
                        Double maxY = jsonObject.getDouble("maxX");
                        // 3 判断范围值是否为空
                        // vo.setMinx(jsonObject.getDouble("minX"));
                        // vo.setMaxx(jsonObject.getDouble("maxX"));
                        // vo.setMiny(jsonObject.getDouble("minY"));
                        // vo.setMaxy(jsonObject.getDouble("maxY"));
                        vo.setMinx(jsonObject.getDouble("minY"));
                        vo.setMaxx(jsonObject.getDouble("maxY"));
                        vo.setMiny(jsonObject.getDouble("minX"));
@@ -340,6 +336,15 @@
                    return fail("防汛范围不能为空", false);
                }
            }
            if (StringUtils.isEmpty(vo.getSemUrl())){
                return fail("sem文件生成路径不能为空！", false);
            }
            List<GridDto> dtos = new ArrayList<>();
            dtos.add(new GridDto(vo.getMiny(), vo.getMinx()));
            dtos.add(new GridDto(vo.getMaxy(), vo.getMinx()));
            dtos.add(new GridDto(vo.getMaxy(), vo.getMaxx()));
            dtos.add(new GridDto(vo.getMiny(), vo.getMaxx()));
            createSem(tableName, dtos, vo.getSemUrl());
            // 开始模拟计算
            boolean flag = simuFilesService.createByfiles(vo);
            return success(flag, flag ? "成功" : "失败");
@@ -348,9 +353,36 @@
        }
    }

    public R<Boolean> createSem(String tableName, List<GridDto> dtos, String semUrl) throws Exception {
        boolean ends = semUrl.endsWith("\\");
        if (!ends) {
            semUrl = semUrl + "\\";
        }
        File file = new File(semUrl);
        if (!file.exists()) {
            file.mkdirs();
        }
        //降雨文件生成
        saveZarr(tableName,semUrl);
        String token = EntityLibraryUtils.login();
        //管点生成
        pointTosem(dtos, token,semUrl);
        //管线生成
        lineToSem(dtos, token,semUrl);
        //地形生成
        terrainToSem(dtos,semUrl);
        //范围生成
        gridToCityJson(dtos,semUrl);
        //河流生成
        riverToSem(semUrl);
        //土地利用生成
        landuseToSem(semUrl);
        return success(true);
    }

    @ApiOperation(value = "tarr文件组装")
    @GetMapping(value = "/saveZarr", produces = "application/json; charset=UTF-8")
    public R<String> saveZarr(@RequestParam("name") String tableName) throws Exception {
    public R<String> saveZarr(@RequestParam("name") String tableName,String semUrl) throws Exception {
        List<String> list = CsvToSQLiteUtils.getNameList(tableName);
        String path = outPath + "\\";
        String rainfall = "rainfall\\";
@@ -407,21 +439,13 @@
        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");
        SemUtils.cityJsonToSem(path + tableName + "\\降雨量.json", semUrl + "raingage.sem");
        return success(semUrl + "raingage.sem");
    }

    @ApiOperation(value = "管点转sem")
    @PostMapping(value = "/pointTosem", produces = "application/json; charset=UTF-8")
    public R<String> pointTosem(@RequestBody List<GridDto> dtos) throws Exception {
    public R<String> pointTosem(List<GridDto> dtos, String token,String semUrl) throws Exception {
        String path = outPath + "\\point\\";
        String publicKey = EntityLibraryUtils.getPublicKey();
        String token = EntityLibraryUtils.login();
        JSONArray jsonArrayList = EntityLibraryUtils.getPointInfo(dtos, token);
        JSONObject jsonObject = getModule("pointmodule.json");
        for (int s = 0; s < jsonArrayList.size(); s++) {
@@ -484,6 +508,50 @@
        FileWriter fileWriter = new FileWriter(pointPath);
        fileWriter.write(jsonObject.toJSONString());
        fileWriter.close();
        SemUtils.cityJsonToSem(pointPath, semUrl + "node.sem");
        return success(semUrl + "node.sem");
    }

    @ApiOperation(value = "管点转cityjson")
    @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()) {
@@ -492,61 +560,10 @@
        SemUtils.cityJsonToSem(pointPath, filePatn + "node.sem");
        return success(filePatn + "node.sem");
    }
//    @ApiOperation(value = "管点转cityjson")
//    @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 {
    public R<String> lineToSem(List<GridDto> dtos, String token,String semUrl) throws Exception {
        String path = outPath + "\\line\\";
        String publicKey = EntityLibraryUtils.getPublicKey();
        String token = EntityLibraryUtils.login();
        JSONArray jsonArrayList = EntityLibraryUtils.getLineInfo(dtos, token);
        if (jsonArrayList.size() == 0) {
            return fail("该区域不存在管线！");
@@ -635,13 +652,8 @@
        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");
        SemUtils.cityJsonToSem(pointPath, semUrl + "link.sem");
        return success(semUrl + "link.sem");
    }

    @ApiOperation(value = "管线转cityjson")
@@ -697,6 +709,49 @@
        return success(filePatn + "link.sem");
    }

    @ApiOperation(value = "地形转sem")
    @PostMapping(value = "/terrainToSem", produces = "application/json; charset=UTF-8")
    public R<String> terrainToSem(@RequestBody List<GridDto> dtos,String semUrl) throws Exception {
        long times = System.currentTimeMillis();
        String path = outPath + "\\terrain\\" + times + "\\";
        File dirFile = new File(path + "appearance");
        if (!dirFile.exists()) {
            dirFile.mkdirs();
        }
        double minx = 180.0;
        double maxx = 0.0;
        double miny = 180.0;
        double maxy = 0.0;
        for (GridDto dto : dtos
        ) {
            if (dto.getLon() > maxx) {
                maxx = dto.getLon();
            }
            if (dto.getLon() < minx) {
                minx = dto.getLon();
            }
            if (dto.getLat() > maxy) {
                maxy = dto.getLat();
            }
            if (dto.getLat() < miny) {
                miny = dto.getLat();
            }
        }
        JSONArray min = ProjectionToGeographicUtil.get4548Point(minx, miny);
        JSONArray max = ProjectionToGeographicUtil.get4548Point(maxx, maxy);
        String tifPath = path + "terrain.tif";
        TiffClipper.cropTiffByLatLon(config.getTifPath(), tifPath, min.getDouble(0), min.getDouble(1), max.getDouble(0), max.getDouble(1));
        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();
        SemUtils.cityJsonToSem(path + "terrain.json", semUrl + "terrain.sem");
        return success(semUrl + "terrain.sem");
    }

    @ApiOperation(value = "地形转cityjson")
    @GetMapping(value = "/terrainToCityJson", produces = "application/json; charset=UTF-8")
    public R<String> terrainToCityJson(@RequestParam("tifPath") String tifPath) throws Exception {
@@ -724,7 +779,7 @@

    @ApiOperation(value = "grid转cityjson")
    @PostMapping(value = "/gridToCityJson", produces = "application/json; charset=UTF-8")
    public R<String> gridToCityJson(@RequestBody List<GridDto> dtos) throws Exception {
    public R<String> gridToCityJson(@RequestBody List<GridDto> dtos,String semUrl) throws Exception {
        long times = System.currentTimeMillis();
        String path = outPath + "\\grid\\" + times + "\\";
        File dirFile = new File(path);
@@ -745,18 +800,13 @@
        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");
        SemUtils.cityJsonToSem(path + "grid.json", semUrl + "grid.sem");
        return success(semUrl + "grid.sem");
    }

    @ApiOperation(value = "river转sem")
    @GetMapping(value = "/riverToSem", produces = "application/json; charset=UTF-8")
    public R<String> riverToSem() throws Exception {
    public R<String> riverToSem(String semUrl) throws Exception {
        long times = System.currentTimeMillis();
        String path = outPath + "\\river\\" + times + "\\";
        File dirFile = new File(path);
@@ -767,18 +817,13 @@
        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");
        SemUtils.cityJsonToSem(path + "river.json", semUrl + "river.sem");
        return success(semUrl + "river.sem");
    }

    @ApiOperation(value = "landuse转sem")
    @GetMapping(value = "/landuseToSem", produces = "application/json; charset=UTF-8")
    public R<String> landuseToSem() throws Exception {
    public R<String> landuseToSem(String semUrl) throws Exception {
        long times = System.currentTimeMillis();
        String path = outPath + "\\landuse\\" + times + "\\";
        File dirFile = new File(path);
@@ -789,13 +834,8 @@
        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");
        SemUtils.cityJsonToSem(path + "landuse.json", semUrl + "landuse.sem");
        return success(semUrl + "landuse.sem");
    }

    public JSONObject getModule(String moduleName) {