1

13693261870

2024-10-08 cdf4040c6cb8c7f3f200d64f754b6773554c6555

1

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

@@ -8,7 +8,6 @@
import com.se.simu.domain.dto.ResultDto;
import com.se.simu.helper.GdalHelper;
import lombok.extern.slf4j.Slf4j;
import nonapi.io.github.classgraph.utils.FileUtils;
import org.gdal.gdal.Band;
import org.gdal.gdal.Dataset;
import org.gdal.gdal.WarpOptions;
@@ -461,7 +460,35 @@
    }

    private static void createFlowPng(float[] vxBuffer, float[] vyBuffer, String png, int width, int height) {
        //
        BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
        Graphics2D graphic = image.createGraphics();
        Color transparent = new Color(0, 0, 0, 0);
        graphic.setColor(transparent);
        graphic.clearRect(0, 0, width, height);

        // 用 R通道表示，流向为归一化的二维向量（x,y），G通道表示为 x *255 , B通道表示为 y * 255
        for (int x = 0; x < width; x++) {
            for (int y = 0; y < height; y++) {
                int offset = x + y * width;
                float fx = getFloatValue(vxBuffer[offset]);
                float fy = getFloatValue(vyBuffer[offset]);
                if (Float.isNaN(fx) && Float.isNaN(fy) || (fx == 0 && fy == 0)) continue;

                fx = Float.isNaN(fx) ? 0 : fx;
                fy = Float.isNaN(fy) ? 0 : fy;
                double dr = Math.sqrt(Math.pow(fx, 2) + Math.pow(fy, 2));

                int r = (int) (dr / 4 * 255);
                int g = (int) ((fx / dr * 0.5 + 0.5) * 255);
                int b = (int) ((fy / dr * 0.5 + 0.5) * 255);

                Color color = new Color(getSafeValue(r), getSafeValue(g), getSafeValue(b), 127);
                //image.setRGB(x, y, color.getRGB());
                graphic.drawImage(image, x, y, 1, 1, color, null);
            }
        }
        graphic.dispose();
        savePng(image, png);
    }

    private static float getFloatValue(float val) {