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


@@ -11,12 +11,19 @@


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


import com.se.nsl.domain.vo.ConfigVo;


import com.se.nsl.helper.ComHelper;


import com.se.nsl.helper.GdalHelper;


import com.se.nsl.helper.StringHelper;


import com.se.nsl.helper.WebHelper;


import lombok.SneakyThrows;


import lombok.extern.slf4j.Slf4j;


import org.gdal.gdal.Band;


import org.gdal.gdal.Dataset;


import org.gdal.gdal.gdal;


import org.gdal.gdalconst.gdalconstConstants;


import org.gdal.ogr.Geometry;


import org.gdal.ogr.ogr;


import org.gdal.osr.CoordinateTransformation;


import org.gdal.osr.SpatialReference;


import org.springframework.stereotype.Service;


import org.springframework.util.StringUtils;




@@ -33,6 +40,8 @@


import java.nio.file.StandardCopyOption;


import java.sql.Timestamp;


import java.text.SimpleDateFormat;


import java.time.Instant;


import java.time.temporal.ChronoUnit;


import java.util.*;


import java.util.concurrent.ExecutorService;


import java.util.concurrent.Executors;


@@ -108,7 +117,7 @@


            SimuData data = JSONUtil.toBean(simu.getData(), SimuData.class);




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


            initArgs(data);


            initArgs(simu, data);


            createRainfallFile(simu, data);




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


@@ -138,18 +147,60 @@


    /**


     * 初始化参数


     */


    public void initArgs(SimuData data) throws IOException {


    public void initArgs(Simu simu, SimuData data) throws IOException {


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


        createDir(inPath);


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


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


        createDir(config.getOutPath() + File.separator + data.getOutPath());




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


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


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


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


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


        SpatialReference dstSR = GdalHelper.createSpatialReference(config.getEpsg());




        CoordinateTransformation ct = CoordinateTransformation.CreateCoordinateTransformation(GdalHelper.SR4326, dstSR);


        geom.Transform(ct);


        String wkt = geom.ExportToWkt();




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


        Dataset dsDem = gdal.Open(config.getSourceDem(), gdalconstConstants.GA_ReadOnly);


        ComHelper.Resample(dsDem, null, terrainFile, null, wkt, null, null);


        dsDem.delete();




        String landuseFile = inPath + File.separator + config.getLanduseFile();


        Dataset dsLanduse = gdal.Open(config.getSourceLanduse(), gdalconstConstants.GA_ReadOnly);


        ComHelper.Resample(dsLanduse, null, landuseFile, null, wkt, null, null);


        dsLanduse.delete();


    }




    public void updateTif(Simu simu, SimuData data2) throws IOException {


        Dataset ds = gdal.Open(config.getSourceLanduse(), gdalconstConstants.GA_Update); // 以读写模式打开TIFF文件




        Band band = ds.GetRasterBand(1);


        if (band.GetRasterDataType() != gdalconstConstants.GDT_Byte) {


            System.err.println("错误：非Byte类型数据");


            ds.delete();


            return;


        }




        int width = band.getXSize();


        int height = band.getYSize();




        // 读取Byte类型数据


        byte[] data = new byte[width * height];


        band.ReadRaster(0, 0, width, height, data);




        // 替换0值为8（注意Java的byte是有符号的，需处理0-255范围）


        for (int i = 0; i < data.length; i++) {


            if ((data[i] & 0xFF) == 0) {  // 无符号比较


                data[i] = (byte) 8;


            }


        }




        band.WriteRaster(0, 0, width, height, data); // 写回数据并保存


        band.SetNoDataValue(8.0);  // 设置新Nodata值


        band.FlushCache(); // 强制写入更改


        ds.delete();


    }




    private void createDir(String path) {


@@ -224,14 +275,22 @@




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


        if (null != data.getRainfalls() && data.getRainfalls().size() > 1) {


            duration = (int) (Math.abs(data.getRainfalls().get(data.getRainfalls().size() - 1).getTime().getTime() - data.getRainfalls().get(0).getTime().getTime()) / 60);


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


            int size = rainfalls.size();


            Rainfall last = rainfalls.get(size - 1);


            Rainfall first = rainfalls.get(0);


            Instant end = last.getTime().toInstant();


            Instant start = first.getTime().toInstant();


            long diff = ChronoUnit.SECONDS.between(end, start);


            duration = (int) (Math.abs(diff));


        }




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


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


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


        String landuseFile = inPath + File.separator + config.getLanduseFile();


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


        ConfigVo vo = new ConfigVo(terrainFile, landuseFile, 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));


@@ -248,7 +307,7 @@


        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 terrainFile = inPath + File.separator + config.getTerrainFile();


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




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


@@ -289,7 +348,7 @@


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


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


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


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


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


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




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