内涝仿真服务
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
11
13693261870
2024-11-01 b6899cd6d1bafb1c349f1fbd190e6632183719f5 
 src/main/java/com/se/simu/service/ResultService.java


@@ -8,6 +8,7 @@


import com.se.simu.domain.dto.LayerDto;


import com.se.simu.domain.dto.ResultDto;


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


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


import com.se.simu.helper.GdalHelper;


import lombok.extern.slf4j.Slf4j;


import org.gdal.gdal.Band;


@@ -23,6 +24,10 @@


import java.awt.*;


import java.awt.image.BufferedImage;


import java.io.*;


import java.math.BigDecimal;


import java.math.RoundingMode;


import java.text.ParseException;


import java.text.SimpleDateFormat;


import java.util.*;


import java.util.List;




@@ -39,7 +44,9 @@


    @Resource


    PropertiesConfig config;




    public final static double MAX_X_OFFSET = 0.002;


    public final static double MAX_X_OFFSET = 0;




    private final static SimpleDateFormat SDF = new SimpleDateFormat("yyyy-MM-dd HH:mm");




    public void process(DataPo data) throws Exception {


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


@@ -61,6 +68,7 @@


            copeWater(dto, layer);


            copeFlow(dto, layer);


            copeLayerJson(dto, layer);


            copeRainFallJson(dto, layer);


        } finally {


            File dir = new File(dto.getTemp());


            if (dir.exists()) {


@@ -76,7 +84,7 @@


        Dataset ds = null;


        try {


            ds = gdal.Open(dto.getTerrainFile(), gdalconst.GA_ReadOnly);


            if (null == ds || 0 == ds.getRasterCount() || null == ds.GetSpatialRef()) return;


            if (null == ds || 0 == ds.getRasterCount()) return;




            setTerrainInfo(ds, layer);


            setWaterInfo(dto, layer);


@@ -124,20 +132,25 @@


        }


    }




    /**


     * 重采样


     */


    private static void Resample(Dataset ds, String dest, int width, int height, LayerDto layer) {


        Vector<String> vector = new Vector<>();


        vector.add("-t_srs");


        vector.add("EPSG:" + 4326);


        //vector.add("-s_srs");


        //vector.add("EPSG:" + 4548);


        //vector.add("-t_srs");


        //vector.add("EPSG:" + 4326);


        vector.add("-ts");


        vector.add("" + width);


        vector.add("" + height);


        vector.add("-te");


        vector.add("" + layer.getExtension().getMinx());


        vector.add("" + layer.getExtension().getMiny());


        vector.add("" + layer.getExtension().getMaxx());


        vector.add("" + layer.getExtension().getMaxy());


        vector.add("-te_srs");


        vector.add("EPSG:" + 4326);


        //vector.add("-te");


        //vector.add("" + layer.getExtension().getMinx());


        //vector.add("" + layer.getExtension().getMiny());


        //vector.add("" + layer.getExtension().getMaxx());


        //vector.add("" + layer.getExtension().getMaxy());


        //vector.add("-te_srs");


        //vector.add("EPSG:" + 4326);


        vector.add("-r");


        vector.add("bilinear");


        vector.add("-of");


@@ -292,10 +305,16 @@




        for (String file : files) {


            String fileName = getNameWithExt(file);


            int hour = Integer.parseInt(fileName.substring(0, 2));


            int minute = Integer.parseInt(fileName.substring(2, 4));


            int second = Integer.parseInt(fileName.substring(4, 6));


            int year = Integer.parseInt(fileName.substring(0, 4));


            int month = Integer.parseInt(fileName.substring(4, 6));


            int day = Integer.parseInt(fileName.substring(6, 8));


            int hour = Integer.parseInt(fileName.substring(8, 10));


            int minute = Integer.parseInt(fileName.substring(10, 12));


            int second = Integer.parseInt(fileName.substring(12, 14));




            calendar.set(Calendar.YEAR, year);


            calendar.set(Calendar.MONTH, month);


            calendar.set(Calendar.DAY_OF_MONTH, day);


            calendar.set(Calendar.HOUR_OF_DAY, hour);


            calendar.set(Calendar.MINUTE, minute);


            calendar.set(Calendar.SECOND, second);


@@ -311,7 +330,7 @@


            Dataset ds = null;


            try {


                ds = gdal.Open(file, gdalconst.GA_ReadOnly);


                if (null == ds || 0 == ds.getRasterCount() || null == ds.GetSpatialRef()) return;


                if (null == ds || 0 == ds.getRasterCount()) return;




                double[] mm = new double[2];


                ds.GetRasterBand(1).ComputeRasterMinMax(mm, 0);


@@ -329,9 +348,10 @@


            Dataset ds = null;


            try {


                ds = gdal.Open(files.get(i), gdalconst.GA_ReadOnly);


                if (null == ds || 0 == ds.getRasterCount() || null == ds.GetSpatialRef()) return;


                if (null == ds || 0 == ds.getRasterCount()) return;




                createWaterPng(dto, ds, layer, layer.getWaters().getData().get(i));


                copyWaterTif(dto, ds, layer.getWaters().getData().get(i));


            } finally {


                if (null != ds) ds.delete();


            }


@@ -352,6 +372,12 @@


            String png = waterPath + File.separator + sizes[0] + "_" + sizes[1] + ".png";


            water2Png(dto, layer, tif, png, sizes[0], sizes[1]);


        }


    }




    private static void copyWaterTif(ResultDto dto, Dataset ds, long ticks) {


        String source = ds.GetDescription();


        String target = dto.getOutPath() + File.separator + "waters" + File.separator + ticks + File.separator + "water.tif";


        FileUtil.copyFile(source, target);


    }




    private static String getNameWithExt(String file) {


@@ -378,8 +404,8 @@




                    double X = transform[0] + x * transform[1] + y * transform[2];


                    double Y = transform[3] + x * transform[4] + y * transform[5];


                    BuildingDto building = intersects(dto, X, Y);


                    if (null != building) continue;


                    //BuildingDto building = intersects(dto, X, Y);


                    //if (null != building) continue;




                    int r = 0, g, b;


                    if (buffer[offset] - layer.getExtension().getMaxHeight() > 0) {


@@ -413,67 +439,51 @@


    }




    private void copeFlow(ResultDto dto, LayerDto layer) {


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


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


        List<String> files = getFiles(dto.getFlowPath(), ".tif");


        for (String file : files) {


            if (file.contains(File.separator + "vx")) vxFiles.add(file);


            if (file.contains(File.separator + "vy")) vyFiles.add(file);


        }


        if (null == vxFiles || null == vyFiles || vxFiles.size() != vyFiles.size() || vxFiles.size() != layer.getWaters().getData().size())


            return;


        if (null == files || files.size() != layer.getWaters().getData().size()) return;




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


            Dataset vxDs = null, vyDs = null;


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


            Dataset ds = null;


            try {


                vxDs = gdal.Open(vxFiles.get(i), gdalconst.GA_ReadOnly);


                vyDs = gdal.Open(vyFiles.get(i), gdalconst.GA_ReadOnly);


                if (null == vxDs || 0 == vxDs.getRasterCount() || null == vxDs.GetSpatialRef() || null == vyDs || 0 == vyDs.getRasterCount() || null == vyDs.GetSpatialRef())


                    return;


                ds = gdal.Open(files.get(i), gdalconst.GA_ReadOnly);


                if (null == ds || 0 == ds.getRasterCount()) return;




                createFlowPng(dto, vxDs, vyDs, layer, layer.getWaters().getData().get(i));


                createFlowPng(dto, ds, layer, layer.getWaters().getData().get(i));


            } finally {


                if (null != vxDs) vxDs.delete();


                if (null != vyDs) vyDs.delete();


                if (null != ds) ds.delete();


            }


        }


    }




    private static void createFlowPng(ResultDto dto, Dataset vxDs, Dataset vyDs, LayerDto layer, long ticks) {


    private static void createFlowPng(ResultDto dto, Dataset ds, LayerDto layer, long ticks) {


        String flowPath = dto.getOutPath() + File.separator + "flows" + File.separator + ticks;


        File dir = new File(flowPath);


        if (!dir.exists() || !dir.isDirectory()) dir.mkdirs();




        for (int[] sizes : layer.getTerrain().getSize()) {


            String vxName = getNameWithExt(vxDs.GetDescription()) + "_" + sizes[0] + "_" + sizes[1];


            String vxTif = dto.getTemp() + File.separator + vxName + ".tif";


            Resample(vxDs, vxTif, sizes[0], sizes[1], layer);




            String vyName = getNameWithExt(vyDs.GetDescription()) + "_" + sizes[0] + "_" + sizes[1];


            String vyTif = dto.getTemp() + File.separator + vyName + ".tif";


            Resample(vyDs, vyTif, sizes[0], sizes[1], layer);


            if (!new File(vxTif).exists() || !new File(vyTif).exists()) continue;


            String name = getNameWithExt(ds.GetDescription()) + "_" + sizes[0] + "_" + sizes[1];


            String tif = dto.getTemp() + File.separator + name + ".tif";


            Resample(ds, tif, sizes[0], sizes[1], layer);


            if (!new File(tif).exists()) continue;




            String png = flowPath + File.separator + sizes[0] + "_" + sizes[1] + ".png";


            vxyTif2Png(layer, vxTif, vyTif, png, sizes[0], sizes[1]);


            vxyTif2Png(layer, tif, png, sizes[0], sizes[1]);


        }


    }




    private static void vxyTif2Png(LayerDto layer, String vxTif, String vyTif, String png, int width, int height) {


        Dataset vxDs = null, vyDs = null;


    private static void vxyTif2Png(LayerDto layer, String tif, String png, int width, int height) {


        Dataset ds = null;


        try {


            vxDs = gdal.Open(vxTif, gdalconst.GA_ReadOnly);


            vyDs = gdal.Open(vyTif, gdalconst.GA_ReadOnly);


            if (null == vxDs || 0 == vxDs.getRasterCount() || null == vyDs || 0 == vyDs.getRasterCount()) return;


            ds = gdal.Open(tif, gdalconst.GA_ReadOnly);


            if (null == ds || 0 == ds.getRasterCount()) return;




            float[] vxBuffer = new float[width * height], vyBuffer = new float[width * height];


            vxDs.GetRasterBand(1).ReadRaster(0, 0, width, height, vxBuffer);


            vyDs.GetRasterBand(1).ReadRaster(0, 0, width, height, vyBuffer);


            ds.GetRasterBand(1).ReadRaster(0, 0, width, height, vxBuffer);


            ds.GetRasterBand(2).ReadRaster(0, 0, width, height, vyBuffer);




            createFlowPng(vxBuffer, vyBuffer, png, width, height);


        } finally {


            if (null != vxDs) vxDs.delete();


            if (null != vyDs) vyDs.delete();


            if (null != ds) ds.delete();


        }


    }




@@ -518,10 +528,50 @@


     */


    private void copeLayerJson(ResultDto dto, LayerDto layer) throws IOException {


        layer.getWaters().setFiles(null);


        //String json = JSONUtil.toJsonPrettyStr(layer);


        layer.getTerrain().setEpsg(null);




        String json = JSON.toJSONString(layer);


        // String json = JSONUtil.toJsonPrettyStr(layer);


        String filePath = dto.getOutPath() + File.separator + "layer.json";




        writeJson(filePath, json);


    }




    /**


     * 处理降水曲线文件曲线图


     */


    public void copeRainFallJson(ResultDto dto, LayerDto layer) throws IOException, ParseException {


        String rainGageFilePath = config.getInPath() + File.separator + dto.getServiceName() + File.separator + "RainGage.dat";


        String filePath = dto.getOutPath() + File.separator + "rainfall.json";




        Map<String, Double> map = new LinkedHashMap<>();


        FileReader fr = new FileReader(rainGageFilePath);


        BufferedReader br = new BufferedReader(fr);




        String line = br.readLine();


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


            String[] rainFall = line.split(" ");


            if (rainFall.length < 7) continue;




            String sdt = rainFall[1] + "-" + rainFall[2] + "-" + rainFall[3] + " " + rainFall[4] + ":" + rainFall[5];


            BigDecimal num = new BigDecimal(rainFall[6]);


            map.put("" + SDF.parse(sdt).getTime(), num.setScale(2, RoundingMode.HALF_UP).doubleValue());


        }


        br.close();


        fr.close();




        String json = JSON.toJSONString(map);


        writeJson(filePath, json);


    }




    private void copeBuildingDepthJson(ResultDto dto, List<BuildingDepthVo> list) throws IOException {


        String json = JSON.toJSONString(list);


        String filePath = dto.getOutPath() + File.separator + "building.json";




        writeJson(filePath, json);


    }




    private void writeJson(String filePath, String json) throws IOException {


        FileWriter fw = new FileWriter(filePath);


        BufferedWriter bw = new BufferedWriter(fw);


        bw.write(json);