1

13693261870

2024-10-14 a6fe405947004d6571806edabd8f14357e144cfa

 src/main/java/com/se/simu/service/ResultService.java

@@ -1,11 +1,12 @@
package com.se.simu.service;

import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSON;
import com.se.simu.config.PropertiesConfig;
import com.se.simu.domain.dto.ExtensionDto;
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.helper.GdalHelper;
import lombok.extern.slf4j.Slf4j;
import org.gdal.gdal.Band;
@@ -39,7 +40,7 @@

    public final static double MAX_X_OFFSET = 0.002;

    public String test() throws Exception {
    public String testResuslt() throws Exception {
        ResultDto dto = new ResultDto(
                "202409",
                "D:\\simu\\input\\tongzhou-local-mesh2-terrain.tif",
@@ -48,15 +49,28 @@
                "D:\\simu\\input\\flows",
                "D:\\simu\\out",
                "");
        process(dto);
        LayerDto layer = new LayerDto(config.getVer(), 4528, config.getSizes());
        process(dto, layer);

        return "OK";
    }

    public void process(ResultDto dto) throws Exception {
        try {
            LayerDto layer = new LayerDto(config.getVer(), 4528, config.getSizes());
    public void process(DataPo data) throws Exception {
        String basePath = config.getInPath() + File.separator + data.getInPath() + File.separator;
        ResultDto dto = new ResultDto(
                data.getInPath(),
                basePath + config.getTerrainFile(),
                basePath + config.getBuildingFile(),
                basePath + config.getWaterPath(),
                basePath + config.getFlowPath(),
                config.getOutPath(),
                "");
        LayerDto layer = new LayerDto(config.getVer(), data.getEpsg(), config.getSizes());
        process(dto, layer);
    }

    private void process(ResultDto dto, LayerDto layer) throws Exception {
        try {
            copeTerrain(dto, layer);
            copeBuilding(dto, layer);
            copeWater(dto, layer);
@@ -160,7 +174,7 @@
            //band.ReadRaster(0, 0, width, height, buffer, width, height, 0, 0);
            band.ReadRaster(0, 0, width, height, buffer);

            BufferedImage image = createImage(width, height);
            BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
            double differ = layer.getExtension().getMaxHeight() - layer.getExtension().getMinHeight(), minHeight = layer.getExtension().getMinHeight();
            for (int x = 0; x < width; x++) {
                for (int y = 0; y < height; y++) {
@@ -257,6 +271,7 @@
    private static List<String> getFiles(String path, String suffix) {
        List<String> files = new ArrayList<>();
        getFiles(files, new File(path), suffix);
        files.sort((a, b) -> a.compareToIgnoreCase(b));

        return files;
    }
@@ -367,7 +382,7 @@
            band.ReadRaster(0, 0, width, height, buffer);
            float[] building = dto.getBuildings().get(width + "_" + height);

            BufferedImage image = createImage(width, height);
            BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
            double differ = layer.getExtension().getMaxHeight() - layer.getExtension().getMinHeight(), minHeight = layer.getExtension().getMinHeight();
            for (int x = 0; x < width; x++) {
                for (int y = 0; y < height; y++) {
@@ -399,8 +414,8 @@
        List<String> vyFiles = new ArrayList<>();
        List<String> files = getFiles(dto.getFlowPath(), ".tif");
        for (String file : files) {
            if (file.startsWith("vx")) vxFiles.add(file);
            if (file.startsWith("vy")) vyFiles.add(file);
            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;
@@ -460,7 +475,7 @@
    }

    private static void createFlowPng(float[] vxBuffer, float[] vyBuffer, String png, int width, int height) {
        BufferedImage image = createImage(width, height);
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
        // 用 R通道表示，流向为归一化的二维向量（x,y），G通道表示为 x *255 , B通道表示为 y * 255
        for (int x = 0; x < width; x++) {
            for (int y = 0; y < height; y++) {
@@ -499,7 +514,9 @@
     * 元数据
     */
    private void copeLayerJson(ResultDto dto, LayerDto layer) throws IOException {
        String json = JSONUtil.toJsonPrettyStr(layer);
        layer.getWaters().setFiles(null);
        //String json = JSONUtil.toJsonPrettyStr(layer);
        String json = JSON.toJSONString(layer);
        String filePath = dto.getOutPath() + File.separator + "layer.json";

        FileWriter fw = new FileWriter(filePath);