泥石流仿真服务
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
wuww
2025-05-14 ddee792c3e8991a72990dbb485bde26bdb968c93 
 src/main/java/com/se/nsl/service/ResolveService.java


@@ -5,6 +5,7 @@


import cn.hutool.json.JSONUtil;


import com.alibaba.fastjson.JSON;


import com.se.nsl.config.PropertiesConfig;


import com.se.nsl.domain.dto.Zarr2Tif;


import com.se.nsl.domain.po.Rainfall;


import com.se.nsl.domain.po.Simu;


import com.se.nsl.domain.po.SimuData;


@@ -20,8 +21,12 @@


import org.springframework.util.StringUtils;




import javax.annotation.Resource;


import java.io.BufferedReader;


import java.io.File;


import java.io.IOException;


import java.io.InputStreamReader;


import java.math.BigDecimal;


import java.math.RoundingMode;


import java.nio.charset.StandardCharsets;


import java.nio.file.Files;


import java.nio.file.Paths;


@@ -31,6 +36,7 @@


import java.util.*;


import java.util.concurrent.ExecutorService;


import java.util.concurrent.Executors;


import java.util.stream.Collectors;




@Slf4j


@Service


@@ -52,6 +58,8 @@




    SimpleDateFormat YYYYMDHM = new SimpleDateFormat("yyyy M d H m ");




    List<String> MODES = new ArrayList<>(Arrays.asList("正态分布", "平均分布", "波动平均分布", "持续上升"));




    public int start(Simu simu) {


        Date now = new Date();


        String date = StringHelper.YMDHMS2_FORMAT.format(now);


@@ -67,8 +75,8 @@


        SimuData data = JSON.parseObject(simu.getData(), SimuData.class);


        data.setInPath(date);


        data.setOutPath(date);


        data.setEpsg(4548);


        data.setEnvelope(envelope);


        data.setEpsg(config.getEpsg());




        simu.setData(JSON.toJSONString(data));


        simu.setServiceName(date);


@@ -101,7 +109,7 @@




            update(simu, 1, "初始化参数");


            initArgs(data);


            createRainfallFile(data);


            createRainfallFile(simu, data);




            update(simu, 2, "调用求解器");


            callUwSolver(data);


@@ -130,7 +138,7 @@


    /**


     * 初始化参数


     */


    private void initArgs(SimuData data) throws IOException {


    public void initArgs(SimuData data) throws IOException {


        String inPath = config.getInPath() + File.separator + data.getInPath();


        createDir(inPath);


        createDir(inPath + File.separator + "depth");


@@ -139,7 +147,7 @@




        // 临时复制高程tif，以后需要自行切割


        File uwBat = new File(config.getUwSolverBat());


        String sourceTif = uwBat.getParent() + File.separator + "data" + File.separator + "Hillzone.tif";


        String sourceTif = uwBat.getParent() + File.separator + "case1" + File.separator + "LiuLiMiaoZhen_5m_f32.tif";


        String targetTif = inPath + File.separator + config.getDemFile();


        Files.copy(Paths.get(sourceTif), Paths.get(targetTif), StandardCopyOption.REPLACE_EXISTING);


    }


@@ -152,9 +160,16 @@


        f.mkdirs();


    }




    private void createRainfallFile(SimuData data) throws Exception {


    public void createRainfallFile(Simu simu, SimuData data) throws Exception {


        List<Rainfall> rainfalls = data.getRainfalls();


        if (null == rainfalls || rainfalls.size() < 2) return; // CollUtil.isEmpty


        if (null == rainfalls || rainfalls.size() < 2) createRainfall(simu);




        String dat = config.getInPath() + File.separator + "Rainfalls" + File.separator + simu.getId() + ".dat";


        String rainfallFile = config.getInPath() + File.separator + data.getInPath() + File.separator + "rainfall.dat";


        if (new File(dat).exists()) {


            Files.copy(Paths.get(dat), Paths.get(rainfallFile), StandardCopyOption.REPLACE_EXISTING);


            return;


        }




        List<String> list = new ArrayList<>();


        list.add(config.getRainfallTitle());


@@ -174,7 +189,6 @@


        //list.add(prefix + YYYYMDHM.format(rainfalls.get(c).getTime()) + getMinVal(rainfalls.get(c).getIntensity() / unit, DIGIT));


        list.add(String.format("%s%s%f", prefix, YYYYMDHM.format(rainfalls.get(c).getTime()), getMinVal(rainfalls.get(c).getIntensity() / unit, DIGIT)));




        String rainfallFile = config.getInPath() + File.separator + data.getInPath() + File.separator + "rainfall.dat";


        Files.write(Paths.get(rainfallFile), list, StandardCharsets.UTF_8);


    }




@@ -204,7 +218,7 @@


    /**


     * 调用UWSolver


     */


    private String callUwSolver(SimuData data) throws Exception {


    public String callUwSolver(SimuData data) throws Exception {


        File uwBat = new File(config.getUwSolverBat());




        int duration = 3600 * data.getDuration(); // 秒数


@@ -213,22 +227,64 @@


        }




        String inPath = config.getInPath() + File.separator + data.getInPath();


        String terrainFile = (inPath + File.separator + config.getDemFile()).replace("\\", "/");


        String rainfallFile = (inPath + File.separator + "rainfall.dat").replace("\\", "/");


        ConfigVo vo = new ConfigVo(duration, config.getSaveFrames(), terrainFile, terrainFile, terrainFile, rainfallFile);


        String terrainFile = (inPath + File.separator + config.getDemFile());


        String rainfallFile = (inPath + File.separator + "rainfall.dat");


        String saveName = inPath + File.separator + "result.zarr";


        ConfigVo vo = new ConfigVo(terrainFile, terrainFile, terrainFile, rainfallFile, saveName, duration, config.getSaveFrames());




        String configFile = config.getInPath() + File.separator + data.getInPath() + File.separator + data.getInPath() + ".json";


        ComHelper.writeJson(configFile, JSON.toJSONString(vo));




        String cmd = String.format("%s %s", config.getUwSolverBat(), configFile);




        return callBat(cmd);


        return callBat2(cmd);


    }




    /**


     * 调用zarr2tif


     */


    private String callZarr2tif(SimuData data) throws Exception {


    public String callZarr2tif(SimuData data) throws Exception {


        String inPath = config.getInPath() + File.separator + data.getInPath();


        String zarrFile = inPath + File.separator + "result.zarr";


        String geotiffDir = inPath + File.separator + "depth";


        String terrainFile = inPath + File.separator + config.getDemFile();


        String jsonPath = inPath + File.separator + "zarr2tif.json";




        Zarr2Tif zarr2Tif = new Zarr2Tif(zarrFile, geotiffDir, terrainFile, data.getStartTime());


        ComHelper.writeJson(jsonPath, JSON.toJSONString(zarr2Tif));




        String cmd = String.format("%s \"%s\"", config.getZarr2tifBat(), jsonPath);




        return callBat2(cmd);


    }




    private String callBat2(String cmd) {


        try {


            ProcessBuilder pb = new ProcessBuilder("cmd", "/c", cmd);


            pb.redirectErrorStream(true); // 合并错误流到标准输出




            Process process = pb.start();


            process.getOutputStream().close();




            //StringBuilder sb = new StringBuilder();


            try (BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream(), "GBK"))) {


                String line;


                while ((line = reader.readLine()) != null) {


                    System.out.println(line);


                    //sb.append(line);


                }


            }




            int exitCode = process.waitFor();




            return "ok"; // sb.toString();


        } catch (Exception ex) {


            log.error(ex.getMessage(), ex);


            return null;


        }


    }




    /*private String callZarr2tif(SimuData data) throws Exception {


        File uwBat = new File(config.getUwSolverBat());


        String zarrFile = uwBat.getParent() + File.separator + "result.zarr";


        String inPath = config.getInPath() + File.separator + data.getInPath();


@@ -238,7 +294,7 @@


        String cmd = String.format("%s \"%s\" \"%s\" \"%s\" \"%s\"", config.getZarr2tifBat(), "depth", zarrFile, terrainFile, waterPath);




        return callBat(cmd);


    }


    }*/




    private String callBat(String cmd) {


        try {


@@ -294,10 +350,79 @@


        }


    }




    public String createRainfallCsv(String csvPath, String type, double total, double intensity, int hours) {


    public String createRainfallCsv(String csvPath, String mode, double total, double intensity, int hours) {


        // python 脚本名.py <参数1-csv文件名> <参数2-降雨模式:正态分布|平均分布|波动平均分布|持续上升> <参数3-降雨总量> <参数4-最大雨强> <参数5-降雨时间(分钟)>


        String cmd = String.format("%s \"%s\" \"%s\" %f %f %d", config.getCreateRainfall(), csvPath, type, total, intensity, hours * 60);


        String cmd = String.format("%s \"%s\" \"%s\" %f %f %d", config.getCreateRainfall(), csvPath, mode, total, intensity, hours * 60);




        return callBat(cmd);


    }




    public void createRainfall(Simu simu) throws Exception {


        SimuData data = JSON.parseObject(simu.getData(), SimuData.class);


        if (null == data.getMode() || MODES.contains(data.getMode())) data.setMode(MODES.get(0));


        if (StringUtils.isEmpty(data.getIntensityUnit())) data.setIntensityUnit("mm/h");




        Geometry geom = Geometry.CreateFromWkt(simu.getGeom());


        if (geom.GetGeometryType() == ogr.wkbMultiPolygon) geom = geom.GetGeometryRef(0);


        double[] envelope = new double[4];


        geom.GetEnvelope(envelope);


        data.setEnvelope(envelope);


        data.setEpsg(config.getEpsg());




        String basePath = config.getInPath() + File.separator + "Rainfalls";


        if (!new File(basePath).exists()) new File(basePath).mkdirs();


        if (null == simu.getCreateTime()) simu.setCreateTime(new Timestamp(new Date().getTime()));




        String csvPath = basePath + File.separator + simu.getId() + ".csv";


        int unit = StringUtils.isEmpty(data.getIntensityUnit()) || "mm/h".equals(data.getIntensityUnit()) ? 60 : 5;


        createRainfallCsv(csvPath, data.getMode(), data.getTotal(), data.getIntensity() / unit, data.getDuration());




        List<Double> list = getValues(csvPath);


        if (!CollUtil.isEmpty(list)) {


            data.setRainfalls(new ArrayList<>());


            setRainfalls(simu, data, list);


        }




        simu.setData(JSON.toJSONString(data));


    }




    private List<Double> getValues(String csvPath) throws Exception {


        if (!new File(csvPath).exists()) return null;




        List<String> list = Files.readAllLines(Paths.get(csvPath));


        list.remove(0);


        //list.remove(list.size() - 1);




        return list.stream()


                .map(s -> new BigDecimal(s).setScale(6, RoundingMode.HALF_DOWN).doubleValue())


                .collect(Collectors.toList());


    }




    private void setRainfalls(Simu simu, SimuData data, List<Double> vals) throws Exception {


        String basePath = config.getInPath() + File.separator + "Rainfalls";


        String dat = basePath + File.separator + simu.getId() + ".dat";




        Calendar cal = Calendar.getInstance();


        cal.setTime(data.getStartTime());




        List<String> list = new ArrayList<>();


        list.add(config.getRainfallTitle());




        double centerX = ComHelper.getMinVal((data.getMinx() + data.getMaxx()) / 2, DIGIT);


        double centerY = ComHelper.getMinVal((data.getMiny() + data.getMaxy()) / 2, DIGIT);


        String prefix = config.getRainfallSite() + " " + centerX + " " + centerY + " ";




        Double total = 0.0;


        for (int i = 0, c = vals.size(); i < c; i++) {


            total += vals.get(i);


            if (i % 15 == 0) {


                data.getRainfalls().add(new Rainfall(cal.getTime(), vals.get(i), total));


            }


            list.add(String.format("%s%s%f", prefix, YYYYMDHM.format(cal.getTime()), vals.get(i)));




            cal.add(Calendar.MINUTE, 1);


        }




        Files.write(Paths.get(dat), list, StandardCharsets.UTF_8);


    }


}